opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(agent): use resolved light model from candidates

Browse source
This commit is contained in:
shadowinlife 2026-03-10 16:16:44 +08:00 • committed by 世行
parent 26f623ed32
commit 277b1c8c26
2 changed files with 34 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pkg/agent/loop.go
View file

@ -1357,7 +1357,11 @@ func (al *AgentLoop) selectCandidates(
"score": score,
"threshold": agent.Router.Threshold(),
})
return agent.LightCandidates, agent.Router.LightModel()
resolvedModel := agent.Router.LightModel()
if len(agent.LightCandidates) > 0 && strings.TrimSpace(agent.LightCandidates[0].Model) != "" {
resolvedModel = agent.LightCandidates[0].Model
}
return agent.LightCandidates, resolvedModel
}
// maybeSummarize triggers summarization if the session history exceeds thresholds.

29
pkg/agent/loop_test.go
View file

@ -575,6 +575,35 @@ func TestProcessMessage_SwitchModelShowModelConsistency(t *testing.T) {
}
}
func TestSelectCandidates_LightModelReturnsResolvedCandidateModel(t *testing.T) {
al := &AgentLoop{}
agent := &AgentInstance{
ID: "test-agent",
Model: "heavy-model",
Candidates: []providers.FallbackCandidate{{Provider: "openai", Model: "heavy-model"}},
Router: routing.New(routing.RouterConfig{
LightModel: "light-model-alias",
Threshold: 1,
}),
LightCandidates: []providers.FallbackCandidate{{
Provider: "openrouter",
Model: "stepfun/step-3.5-flash:free",
}},
}
candidates, model := al.selectCandidates(agent, "", nil)
if len(candidates) != 1 {
t.Fatalf("len(candidates) = %d, want 1", len(candidates))
}
if candidates[0].Model != "stepfun/step-3.5-flash:free" {
t.Fatalf("light candidate model = %q, want %q", candidates[0].Model, "stepfun/step-3.5-flash:free")
}
if model != "stepfun/step-3.5-flash:free" {
t.Fatalf("resolved model = %q, want %q", model, "stepfun/step-3.5-flash:free")
}
}
// TestToolResult_SilentToolDoesNotSendUserMessage verifies silent tools don't trigger outbound
func TestToolResult_SilentToolDoesNotSendUserMessage(t *testing.T) {
tmpDir, err := os.MkdirTemp("", "agent-test-*")