Documentation
¶
Index ¶
- func ExtractQuantization(filename string) string
- func ParseCivitAIModelID(url string) (string, error)
- func ParseCivitAIVersionID(url string) (int64, error)
- type CivitAIFetcher
- func (f *CivitAIFetcher) CanHandle(url string) bool
- func (f *CivitAIFetcher) CheckConnectivity(ctx context.Context) error
- func (f *CivitAIFetcher) FetchMetadata(ctx context.Context, url string) (*state.ModelMetadata, error)
- func (f *CivitAIFetcher) SetAPIKey(key string)
- func (f *CivitAIFetcher) Source() string
- func (f *CivitAIFetcher) ValidateAPIKey(ctx context.Context, apiKey string) error
- type Fetcher
- type HuggingFaceFetcher
- type Registry
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func ExtractQuantization ¶
ExtractQuantization extracts quantization info from filename (exported for scanner)
func ParseCivitAIModelID ¶
ParseModelIDFromURL extracts the model ID from a CivitAI URL
func ParseCivitAIVersionID ¶
GetModelVersionID extracts version ID from download URLs like /api/download/models/{versionId}
Types ¶
type CivitAIFetcher ¶
type CivitAIFetcher struct {
// contains filtered or unexported fields
}
CivitAIFetcher fetches metadata from CivitAI
func NewCivitAIFetcher ¶
func NewCivitAIFetcher(client *http.Client) *CivitAIFetcher
NewCivitAIFetcher creates a new CivitAI metadata fetcher
func (*CivitAIFetcher) CanHandle ¶
func (f *CivitAIFetcher) CanHandle(url string) bool
func (*CivitAIFetcher) CheckConnectivity ¶
func (f *CivitAIFetcher) CheckConnectivity(ctx context.Context) error
CheckConnectivity tests if CivitAI is accessible (useful for VPN detection)
func (*CivitAIFetcher) FetchMetadata ¶
func (f *CivitAIFetcher) FetchMetadata(ctx context.Context, url string) (*state.ModelMetadata, error)
func (*CivitAIFetcher) SetAPIKey ¶
func (f *CivitAIFetcher) SetAPIKey(key string)
SetAPIKey sets the CivitAI API key for authenticated requests
func (*CivitAIFetcher) Source ¶
func (f *CivitAIFetcher) Source() string
func (*CivitAIFetcher) ValidateAPIKey ¶
func (f *CivitAIFetcher) ValidateAPIKey(ctx context.Context, apiKey string) error
ValidateAPIKey checks if the provided API key is valid
type Fetcher ¶
type Fetcher interface {
// CanHandle returns true if this fetcher can handle the given URL
CanHandle(url string) bool
// FetchMetadata retrieves metadata for the given URL
FetchMetadata(ctx context.Context, url string) (*state.ModelMetadata, error)
// Source returns the name of this metadata source
Source() string
}
Fetcher is the interface for fetching model metadata from various sources
type HuggingFaceFetcher ¶
type HuggingFaceFetcher struct {
// contains filtered or unexported fields
}
HuggingFaceFetcher fetches metadata from HuggingFace
func NewHuggingFaceFetcher ¶
func NewHuggingFaceFetcher(client *http.Client) *HuggingFaceFetcher
NewHuggingFaceFetcher creates a new HuggingFace metadata fetcher
func (*HuggingFaceFetcher) CanHandle ¶
func (f *HuggingFaceFetcher) CanHandle(url string) bool
func (*HuggingFaceFetcher) FetchMetadata ¶
func (f *HuggingFaceFetcher) FetchMetadata(ctx context.Context, url string) (*state.ModelMetadata, error)
func (*HuggingFaceFetcher) Source ¶
func (f *HuggingFaceFetcher) Source() string
type Registry ¶
type Registry struct {
// contains filtered or unexported fields
}
Registry holds all available metadata fetchers
func NewRegistry ¶
func NewRegistry() *Registry
NewRegistry creates a new fetcher registry with default fetchers
func (*Registry) CanHandle ¶
CanHandle returns true if any registered fetcher can handle the given URL
func (*Registry) FetchMetadata ¶
FetchMetadata attempts to fetch metadata using the appropriate fetcher
Source Files