registry

type HFModelInfo struct {
	ID       string      `json:"id"`
	Siblings []HFSibling `json:"siblings"`
}

type HFPullOptions struct {
	// APIURL overrides the HuggingFace API endpoint.
	APIURL string
	// CDNURL overrides the HuggingFace CDN endpoint.
	CDNURL string
	// Token is an optional HuggingFace API token for gated models.
	Token string
	// OnProgress is called during file downloads.
	OnProgress ProgressFunc
	// Client overrides the HTTP client used for downloads.
	Client *http.Client
}

type HFSibling struct {
	Filename string `json:"rfilename"`
}

type LocalRegistry struct {
	// contains filtered or unexported fields
}

type ModelInfo struct {
	ID           string `json:"id"`
	Path         string `json:"path"`
	Architecture string `json:"architecture"`
	VocabSize    int    `json:"vocab_size"`
	MaxSeqLen    int    `json:"max_seq_len"`
	Size         int64  `json:"size"`
}

type ModelRegistry interface {
	// Pull downloads a model by ID and caches it locally.
	Pull(ctx context.Context, modelID string) (*ModelInfo, error)
	// Get returns a locally cached model by ID.
	Get(modelID string) (*ModelInfo, bool)
	// List returns all locally cached models.
	List() []ModelInfo
	// Delete removes a locally cached model.
	Delete(modelID string) error
}

type ProgressFunc func(downloaded, total int64)

type PullFunc func(ctx context.Context, modelID string, targetDir string) (*ModelInfo, error)