24 lines
507 B
Go
24 lines
507 B
Go
package main
|
|
|
|
import "io/ioutil"
|
|
import "gopkg.in/yaml.v3"
|
|
|
|
// Config holds all prompts and settings
|
|
type Config struct {
|
|
LLM struct {
|
|
ExtractKeywordsPrompt string `yaml:"extract_keywords_prompt"`
|
|
DisambiguatePrompt string `yaml:"disambiguate_prompt"`
|
|
TranslatePrompt string `yaml:"translate_prompt"`
|
|
} `yaml:"llm"`
|
|
}
|
|
|
|
var appConfig Config
|
|
|
|
func loadConfig(path string) error {
|
|
data, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return yaml.Unmarshal(data, &appConfig)
|
|
}
|