summaryrefslogtreecommitdiff
path: root/internal/executor/classifier_test.go
diff options
context:
space:
mode:
authorPeter Stone <thepeterstone@gmail.com>2026-03-10 16:48:15 +0000
committerPeter Stone <thepeterstone@gmail.com>2026-03-10 16:48:15 +0000
commite0335047e063830ca000a8cb3a9ec31a8ab411a7 (patch)
tree9f38a4b669f35a0b9590b13b0fd0a0aeb37d99a9 /internal/executor/classifier_test.go
parente392f99727aa2f399033896f2cda5b22e3277700 (diff)
executor: explicit load balancing — code picks agent, classifier picks model
pickAgent() deterministically selects the agent with the fewest active tasks, skipping rate-limited agents. The classifier now only selects the model for the pre-assigned agent, so Gemini gets tasks from the start rather than only as a fallback when Claude's quota is exhausted. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Diffstat (limited to 'internal/executor/classifier_test.go')
-rw-r--r--internal/executor/classifier_test.go2
1 files changed, 1 insertions, 1 deletions
diff --git a/internal/executor/classifier_test.go b/internal/executor/classifier_test.go
index 631952f..83a9743 100644
--- a/internal/executor/classifier_test.go
+++ b/internal/executor/classifier_test.go
@@ -23,7 +23,7 @@ echo '{"response": "{\"agent_type\": \"gemini\", \"model\": \"gemini-2.5-flash-l
RateLimited: map[string]bool{"claude": false, "gemini": false},
}
- cls, err := c.Classify(context.Background(), "Test Task", "Test Instructions", status)
+ cls, err := c.Classify(context.Background(), "Test Task", "Test Instructions", status, "gemini")
if err != nil {
t.Fatalf("Classify failed: %v", err)
}