core/mcp
8
0
Fork 0
Code Issues 4 Pull requests 2 Projects Releases Packages Wiki Activity Actions

fix(agentic): match concurrency limits on base agent type

Browse source 
gemini:flash, gemini:pro now match the "gemini" concurrency limit.
Strips model variant before checking config. Prevents unlimited
spawning when using model variants.

Co-Authored-By: Virgil <virgil@lethean.io>
This commit is contained in:
Snider 2026-03-15 19:00:19 +00:00
parent cb8c126d78
commit f8bee4b4ad
1 changed files with 15 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
pkg/mcp/agentic/queue.go
View file

@ -7,6 +7,7 @@ import (
"os"
"os/exec"
"path/filepath"
"strings"
"syscall"
"time"
@ -118,7 +119,12 @@ func (s *PrepSubsystem) countRunningByAgent(agent string) int {
}
st, err := readStatus(filepath.Join(wsRoot, entry.Name()))
if err != nil || st.Status != "running" || st.Agent != agent {
if err != nil || st.Status != "running" {
continue
}
// Match on base agent type (gemini:flash matches gemini)
stBase := strings.SplitN(st.Agent, ":", 2)[0]
if stBase != agent {
continue
}
@ -133,14 +139,20 @@ func (s *PrepSubsystem) countRunningByAgent(agent string) int {
return count
}
// baseAgent strips the model variant (gemini:flash → gemini).
func baseAgent(agent string) string {
return strings.SplitN(agent, ":", 2)[0]
}
// canDispatchAgent checks if we're under the concurrency limit for a specific agent type.
func (s *PrepSubsystem) canDispatchAgent(agent string) bool {
cfg := s.loadAgentsConfig()
limit, ok := cfg.Concurrency[agent]
base := baseAgent(agent)
limit, ok := cfg.Concurrency[base]
if !ok || limit <= 0 {
return true
}
return s.countRunningByAgent(agent) < limit
return s.countRunningByAgent(base) < limit
}
// canDispatch is kept for backwards compat.