summaryrefslogtreecommitdiff
path: root/internal/executor/executor.go
diff options
context:
space:
mode:
authorPeter Stone <thepeterstone@gmail.com>2026-03-09 01:08:51 +0000
committerPeter Stone <thepeterstone@gmail.com>2026-03-09 01:08:51 +0000
commit5ad8356b5ce4d525af079f902791559d53b67bba (patch)
treee04ff2e9ffa4b0e94d67b0deaaa3092677104b07 /internal/executor/executor.go
parent4db4aaf61c0cc6161a4676ea7bd4a25a90e39b40 (diff)
executor: fix Claude rate-limit detection and prioritize Gemini when limited
Updated parseStream to detect 'rate_limit_event' and 'assistant' error:rate_limit messages from the Claude CLI. Updated Classifier to strongly prefer non-rate-limited agents. Added logging to Pool to track rate-limit status during classification.
Diffstat (limited to 'internal/executor/executor.go')
-rw-r--r--internal/executor/executor.go10
1 files changed, 9 insertions, 1 deletions
diff --git a/internal/executor/executor.go b/internal/executor/executor.go
index df222f8..f404ddf 100644
--- a/internal/executor/executor.go
+++ b/internal/executor/executor.go
@@ -238,6 +238,7 @@ func (p *Pool) executeResume(ctx context.Context, t *task.Task, exec *storage.Ex
retryAfter = 1 * time.Minute
}
p.rateLimited[agentType] = time.Now().Add(retryAfter)
+ p.logger.Info("agent rate limited", "agent", agentType, "retryAfter", retryAfter)
p.mu.Unlock()
}
@@ -295,12 +296,18 @@ func (p *Pool) execute(ctx context.Context, t *task.Task) {
now := time.Now()
for agent := range p.runners {
activeTasks[agent] = p.activePerAgent[agent]
- rateLimited[agent] = now.Before(p.rateLimited[agent])
+ activeUntil := p.rateLimited[agent]
+ isLimited := now.Before(activeUntil)
+ rateLimited[agent] = isLimited
+ if isLimited {
+ p.logger.Debug("agent rate limited", "agent", agent, "until", activeUntil)
+ }
}
status := SystemStatus{
ActiveTasks: activeTasks,
RateLimited: rateLimited,
}
+ p.logger.Debug("classifying task", "taskID", t.ID, "status", status)
p.mu.Unlock()
cls, err := p.Classifier.Classify(ctx, t.Name, t.Agent.Instructions, status)
@@ -433,6 +440,7 @@ func (p *Pool) execute(ctx context.Context, t *task.Task) {
retryAfter = 1 * time.Minute
}
p.rateLimited[agentType] = time.Now().Add(retryAfter)
+ p.logger.Info("agent rate limited", "agent", agentType, "retryAfter", retryAfter)
p.mu.Unlock()
}