File tree Expand file tree Collapse file tree 1 file changed +16
-4
lines changed
Expand file tree Collapse file tree 1 file changed +16
-4
lines changed Original file line number Diff line number Diff line change @@ -201,11 +201,23 @@ func (m *model) updateGuard() {
201201 var judge detector.LLMJudge
202202 switch m .llmProvider {
203203 case "openai" :
204- judge = detector .NewOpenAIJudge (os .Getenv ("OPENAI_API_KEY" ), "gpt-5" )
204+ model := os .Getenv ("OPENAI_MODEL" )
205+ if model == "" {
206+ model = "gpt-5"
207+ }
208+ judge = detector .NewOpenAIJudge (os .Getenv ("OPENAI_API_KEY" ), model )
205209 case "openrouter" :
206- judge = detector .NewOpenRouterJudge (os .Getenv ("OPENROUTER_API_KEY" ), "anthropic/claude-sonnet-4.5" )
210+ model := os .Getenv ("OPENROUTER_MODEL" )
211+ if model == "" {
212+ model = "anthropic/claude-sonnet-4.5"
213+ }
214+ judge = detector .NewOpenRouterJudge (os .Getenv ("OPENROUTER_API_KEY" ), model )
207215 case "anthropic" :
208- judge = detector .NewAnthropicJudge (os .Getenv ("ANTHROPIC_API_KEY" ), "claude-sonnet-4-5-20250929" )
216+ model := os .Getenv ("ANTHROPIC_MODEL" )
217+ if model == "" {
218+ model = "claude-sonnet-4-5-20250929"
219+ }
220+ judge = detector .NewAnthropicJudge (os .Getenv ("ANTHROPIC_API_KEY" ), model )
209221 case "ollama" :
210222 ollamaHost := os .Getenv ("OLLAMA_HOST" )
211223 if ollamaHost == "" {
@@ -214,7 +226,7 @@ func (m *model) updateGuard() {
214226
215227 ollamaModel := os .Getenv ("OLLAMA_MODEL" )
216228 if ollamaModel == "" {
217- ollamaModel = "llama3.1 " // Default fallback
229+ ollamaModel = "llama3.3 " // Default fallback
218230 }
219231
220232 if ollamaHost != "" {
You can’t perform that action at this time.
0 commit comments