33 lines
715 B
Go
33 lines
715 B
Go
package agent
|
|
|
|
import (
|
|
"os"
|
|
|
|
"ollie/paths"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// modelsConfig holds the parsed contents of ~/.config/ollie/models.yaml.
|
|
type modelsConfig struct {
|
|
Compaction map[string]string `yaml:"compaction"`
|
|
Completion struct {
|
|
Model string `yaml:"model"`
|
|
Backend string `yaml:"backend"`
|
|
} `yaml:"completion"`
|
|
}
|
|
|
|
// loadModelsConfig reads and parses the models config file.
|
|
// Returns zero value if the file doesn't exist or is invalid.
|
|
func loadModelsConfig() modelsConfig {
|
|
data, err := os.ReadFile(paths.CfgDir() + "/models.yaml")
|
|
if err != nil {
|
|
return modelsConfig{}
|
|
}
|
|
var cfg modelsConfig
|
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
|
return modelsConfig{}
|
|
}
|
|
return cfg
|
|
}
|