Skip to content

Commit 43182f8

Browse files
committed
feat: add model selection env vars for all LLM providers
1 parent e609d8f commit 43182f8

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

cmd/go-promptguard/interactive.go

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff 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 != "" {

0 commit comments

Comments
 (0)