config

type ChromaConfig struct {
	URL      string
	Host     string
	Port     string
	Tenant   string
	Database string
	Timeout  time.Duration
}

type Config struct {
	Chroma  ChromaConfig
	Model   ModelConfig
	Logging LoggingConfig
}

type ConfigFile struct {
	Version  string             `yaml:"version"`
	Chroma   ConfigFileChroma   `yaml:"chroma"`
	Model    ConfigFileModel    `yaml:"model"`
	Logging  ConfigFileLogging  `yaml:"logging"`
	Features ConfigFileFeatures `yaml:"features"`
}

type ConfigFileChroma struct {
	Host     string `yaml:"host"`
	Port     string `yaml:"port"`
	Tenant   string `yaml:"tenant"`
	Database string `yaml:"database"`
	Timeout  int    `yaml:"timeout"`
}

type ConfigFileCreateCollection struct {
	AutoCreateDatabase bool `yaml:"auto_create_database"`
}

type ConfigFileFeatures struct {
	CreateCollection ConfigFileCreateCollection `yaml:"create_collection"`
}

type ConfigFileLogging struct {
	Level   string `yaml:"level"`
	Format  string `yaml:"format"`
	Verbose bool   `yaml:"verbose"`
}

type ConfigFileModel struct {
	ONNXModel string `yaml:"onnx_model"`
	Tokenizer string `yaml:"tokenizer"`
	ONNXLib   string `yaml:"onnx_lib"`
}

type LoggingConfig struct {
	Verbose bool
	Level   string
	Format  string
}

type ModelConfig struct {
	ONNXModel string
	Tokenizer string
	ONNXLib   string
}