botcmd_limits_test.go RawHistoryFolder

1package agent
2
3import (
4	"testing"
5
6	"github.com/bnema/zut/packages/core"
7)
8
9func TestApplyResolvedModelAdoptsLimits(t *testing.T) {
10	ag := core.NewAgent(nil, "old", "", nil)
11	ag.MaxTokens = 4096
12	ag.ContextWindow = 8000
13
14	applyResolvedModel(ag, Resolved{Model: "new", MaxOutput: 64000, ContextWindow: 200000})
15	if ag.Model != "new" || ag.MaxTokens != 64000 || ag.ContextWindow != 200000 {
16		t.Fatalf("agent = model %q max %d ctx %d", ag.Model, ag.MaxTokens, ag.ContextWindow)
17	}
18
19	// Missing output metadata must not keep the previous model's cap;
20	// a missing context window keeps the existing one.
21	applyResolvedModel(ag, Resolved{Model: "other"})
22	if ag.Model != "other" || ag.MaxTokens != 0 || ag.ContextWindow != 200000 {
23		t.Fatalf("agent = model %q max %d ctx %d", ag.Model, ag.MaxTokens, ag.ContextWindow)
24	}
25}