diff options
Diffstat (limited to 'internal/executor/classifier_test.go')
| -rw-r--r-- | internal/executor/classifier_test.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/internal/executor/classifier_test.go b/internal/executor/classifier_test.go index 4de44ca..631952f 100644 --- a/internal/executor/classifier_test.go +++ b/internal/executor/classifier_test.go @@ -11,7 +11,7 @@ func TestClassifier_Classify_Mock(t *testing.T) { // Create a temporary mock binary. mockBinary := filepathJoin(t.TempDir(), "mock-gemini") mockContent := `#!/bin/sh -echo '{"agent_type": "gemini", "model": "gemini-2.0-flash", "reason": "test reason"}' +echo '{"response": "{\"agent_type\": \"gemini\", \"model\": \"gemini-2.5-flash-lite\", \"reason\": \"test reason\"}"}' ` if err := os.WriteFile(mockBinary, []byte(mockContent), 0755); err != nil { t.Fatal(err) @@ -31,8 +31,8 @@ echo '{"agent_type": "gemini", "model": "gemini-2.0-flash", "reason": "test reas if cls.AgentType != "gemini" { t.Errorf("expected gemini, got %s", cls.AgentType) } - if cls.Model != "gemini-2.0-flash" { - t.Errorf("expected gemini-2.0-flash, got %s", cls.Model) + if cls.Model != "gemini-2.5-flash-lite" { + t.Errorf("expected gemini-2.5-flash-lite, got %s", cls.Model) } } |