Paraclub/ai-mailer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: add default_language config parameter

Browse source
This commit is contained in:
Dominik Polakovics Polakovics 2026-03-19 15:15:42 +01:00
parent e88ac7caff
commit 1d03d584c2
3 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
config.yaml.example
View file

@ -13,6 +13,7 @@ ai:
model: "anthropic/claude-2" # Other options: "openai/gpt-4", "google/palm-2" model: "anthropic/claude-2" # Other options: "openai/gpt-4", "google/palm-2"
temperature: 0.7 # 0.0 to 1.0, lower for more focused responses temperature: 0.7 # 0.0 to 1.0, lower for more focused responses
max_tokens: 2000 # Adjust based on your needs and model limits max_tokens: 2000 # Adjust based on your needs and model limits
default_language: "English" # Fallback language when auto-detection fails (leave empty to disable)
context: context:
urls: urls:

1
config/config.go
View file

@ -35,6 +35,7 @@ type AIConfig struct {
Model string `yaml:"model"` Model string `yaml:"model"`
Temperature float32 `yaml:"temperature"` Temperature float32 `yaml:"temperature"`
MaxTokens int `yaml:"max_tokens"` MaxTokens int `yaml:"max_tokens"`
DefaultLanguage string `yaml:"default_language"`
} }
type ContextConfig struct { type ContextConfig struct {

10
internal/ai/ai.go
View file

@ -99,7 +99,15 @@ func (a *AI) GenerateReply(emailContent string, contextContent map[string]string
// First, detect the language // First, detect the language
lang, err := a.detectLanguage(emailContent) lang, err := a.detectLanguage(emailContent)
if err != nil { if err != nil {
return "", err if a.config.DefaultLanguage != "" {
logger.WithFields(logrus.Fields{
"error": err,
"defaultLanguage": a.config.DefaultLanguage,
}).Warn("Language detection failed, falling back to default language")
lang = a.config.DefaultLanguage
} else {
return "", err
}
} }
// Build prompts using exposed methods // Build prompts using exposed methods