config

type Config struct {
	Database db.Config `yaml:"database"`

	ObjectStore ObjectStoreConfig `yaml:"objectStore"`

	// BaseModels is the list of base models to load. Currently each model follows Hugging Face's model format.
	BaseModels []string `yaml:"baseModels"`

	ModelLoadInterval time.Duration `yaml:"modelLoadInterval"`

	// RunOnce is set to true when models are loaded only once.
	RunOnce bool `yaml:"runOnce"`

	// SkipDBUpdate is set to true when the loader skips updateing the "base-models" table.
	SkipDBUpdate bool `yaml:"skipDbUpdate"`

	Downloader DownloaderConfig `yaml:"downloader"`

	Debug DebugConfig `yaml:"debug"`
}

type DebugConfig struct {
	Standalone bool   `yaml:"standalone"`
	SqlitePath string `yaml:"sqlitePath"`
}

type DownloaderConfig struct {
	HuggingFace HuggingFaceDownloaderConfig `yaml:"huggingFace"`
}

type HuggingFaceDownloaderConfig struct {
	CacheDir string `yaml:"cacheDir"`
}

type ObjectStoreConfig struct {
	S3 S3Config `yaml:"s3"`
}

type S3Config struct {
	EndpointURL string `yaml:"endpointUrl"`
	Region      string `yaml:"region"`

	Bucket string `yaml:"bucket"`

	PathPrefix string `yaml:"pathPrefix"`

	// BaseModelPathPrefix is the path prefix for the base models in the object store. A model is stored under
	// <ObjectStore.S3.PathPrefix>/<BaseModelPathPrefix>.
	BaseModelPathPrefix string `yaml:"baseModelPathPrefix"`
}