store

type BaseModel struct {
	gorm.Model

	TenantID string `gorm:"uniqueIndex:idx_base_model_model_id_tenant_id_project_id"`

	ModelID string `gorm:"uniqueIndex:idx_base_model_model_id_tenant_id_project_id"`

	// ProjectID is the ID of the project to which the model belongs. It is empty if
	// the model is globally scoped and it is not associated with any project.
	ProjectID string `gorm:"uniqueIndex:idx_base_model_model_id_tenant_id_project_id"`

	Path string

	Formats []byte

	// GGUFModelPath is the path to the GGUF model.
	GGUFModelPath string

	SourceRepository v1.SourceRepository

	LoadingStatus        v1.ModelLoadingStatus
	LoadingFailureReason string
	LoadingStatusMessage string
}

type HFModelRepo struct {
	gorm.Model

	Name      string `gorm:"uniqueIndex:idx_hf_model_repo_name_tenant_id_project_id"`
	ModelID   string `gorm:"uniqueIndex:idx_hf_model_repo_model_id_tenant_id_project_id"`
	TenantID  string `gorm:"uniqueIndex:idx_hf_model_repo_name_tenant_id_project_id;uniqueIndex:idx_hf_model_repo_model_id_tenant_id_project_id"`
	ProjectID string `gorm:"uniqueIndex:idx_hf_model_repo_name_tenant_id_project_id;uniqueIndex:idx_hf_model_repo_model_id_tenant_id_project_id"`
}

type Model struct {
	gorm.Model

	TenantID string `gorm:"uniqueIndex:idx_model_model_id_tenant_id"`
	ModelID  string `gorm:"uniqueIndex:idx_model_model_id_tenant_id"`

	OrganizationID string
	ProjectID      string `gorm:"index"`

	Path        string
	IsPublished bool

	BaseModelID  string `gorm:"index"`
	Adapter      v1.AdapterType
	Quantization v1.QuantizationType

	LoadingStatus        v1.ModelLoadingStatus
	LoadingFailureReason string
	LoadingStatusMessage string

	SourceRepository  v1.SourceRepository
	ModelFileLocation string
}

type ModelActivationStatus struct {
	gorm.Model

	TenantID string `gorm:"uniqueIndex:idx_model_activation_status_model_id_tenant_id_project_id"`
	ModelID  string `gorm:"uniqueIndex:idx_model_activation_status_model_id_tenant_id_project_id"`

	// ProjectID is set only for project-scoped base models. This is set to empty
	// for global-scoped base models and fine-tuned models. (Fine-tuned models have emtpy values
	// for backward compatibility.)
	ProjectID string `gorm:"uniqueIndex:idx_model_activation_status_model_id_tenant_id_project_id"`

	Status v1.ActivationStatus
}

type ModelConfig struct {
	gorm.Model

	TenantID string `gorm:"uniqueIndex:idx_model_config_model_id_tenant_id_project_id"`
	ModelID  string `gorm:"uniqueIndex:idx_model_config_model_id_tenant_id_project_id"`

	// ProjectID is set only for project-scoped base models. This is set to empty
	// for global-scoped base models and fine-tuned models.
	ProjectID string `gorm:"uniqueIndex:idx_model_config_model_id_tenant_id_project_id"`

	// EncodedConfig is a proto-encoded message that contains v1.ModelConfig.
	EncodedConfig []byte
}

type ModelKey struct {
	ModelID   string
	TenantID  string
	ProjectID string
}

type ModelSpec struct {
	ModelID           string
	TenantID          string
	OrganizationID    string
	ProjectID         string
	Path              string
	IsPublished       bool
	BaseModelID       string
	Adapter           v1.AdapterType
	Quantization      v1.QuantizationType
	LoadingStatus     v1.ModelLoadingStatus
	SourceRepository  v1.SourceRepository
	ModelFileLocation string
}

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

type StorageConfig struct {
	gorm.Model

	TenantID string `gorm:"uniqueIndex"`

	// PathPrefix is the prefix of the S3 path for storing models.
	PathPrefix string
}