ai_local

type CacheType string

type CreateLocalModel struct {
	Id       string
	Name     string
	Provider string
	State    int
}

type CreateLocalModelInstallState struct {
	Id       string
	Complete int64
	Total    int64
	State    int
	Msg      string
}

type CreateLocalModelPackage struct {
	Id          string
	Name        string
	Size        string
	Hash        string
	Description string
	Version     string
	Popular     bool
}

type EditLocalModel struct {
	State *int
}

type EditLocalModelInstallState struct {
	Complete *int64
	Total    *int64
	State    *int
	Msg      *string
}

type EditLocalModelPackage struct {
	Size        *string
	Hash        *string
	Description *string
	Version     *string
	Popular     *bool
}

type ILocalModelCacheService interface {
	List(ctx context.Context, model string, typ CacheType) ([]*LocalModelCache, error)
	Delete(ctx context.Context, model string) error
	Save(ctx context.Context, model string, typ CacheType, target string) error
	GetByTarget(ctx context.Context, typ CacheType, target string) (*LocalModelCache, error)
}

type ILocalModelInstallStateService interface {
	universally.IServiceGet[LocalModelInstallState]
	universally.IServiceCreate[CreateLocalModelInstallState]
	universally.IServiceEdit[EditLocalModelInstallState]
	universally.IServiceDelete
}

type ILocalModelPackageService interface {
	universally.IServiceGet[LocalModelPackage]
	universally.IServiceCreate[CreateLocalModelPackage]
	universally.IServiceEdit[EditLocalModelPackage]
	universally.IServiceDelete
}

type ILocalModelService interface {
	universally.IServiceGet[LocalModel]
	universally.IServiceCreate[CreateLocalModel]
	universally.IServiceEdit[EditLocalModel]
	universally.IServiceDelete
	DefaultModel(ctx context.Context) (*LocalModel, error)
	UpdateProvider(ctx context.Context, provider string, ids ...string) error
}

type LocalModel struct {
	Id       string
	Name     string
	Provider string
	State    int
	CreateAt time.Time
	UpdateAt time.Time
	Creator  string
	Updater  string
}

type LocalModelCache struct {
	Model  string
	Target string
	Type   CacheType
}

type LocalModelInstallState struct {
	Id       string
	Complete int64
	Total    int64
	State    int
	Msg      string
	UpdateAt time.Time
}

type LocalModelPackage struct {
	Id          string
	Name        string
	Size        string
	Hash        string
	Description string
	Version     string
	IsPopular   bool
}