interactive_model_swap_test.go RawHistoryFolder

1package modes
2
3import (
4	"testing"
5
6	"github.com/patriceckhart/zot/packages/core"
7	"github.com/patriceckhart/zot/packages/provider"
8)
9
10// A same-provider /model switch reuses the agent, so it must also adopt
11// the new model's output budget. Keeping the previous model's MaxTokens
12// truncated responses (e.g. a 4096 budget carried onto a 128k model).
13func TestSameProviderModelSwapUpdatesMaxTokens(t *testing.T) {
14	const from, to = "claude-3-5-haiku-20241022", "claude-opus-4-1-20250805"
15	fromModel, err := provider.FindModel("anthropic", from)
16	if err != nil {
17		t.Fatal(err)
18	}
19	toModel, err := provider.FindModel("anthropic", to)
20	if err != nil {
21		t.Fatal(err)
22	}
23	if fromModel.MaxOutput == toModel.MaxOutput || toModel.MaxOutput <= 0 {
24		t.Fatalf("test needs distinct positive limits, got %d and %d", fromModel.MaxOutput, toModel.MaxOutput)
25	}
26
27	ag := core.NewAgent(nil, from, "", nil)
28	ag.MaxTokens = fromModel.MaxOutput
29	i := &Interactive{agent: ag, cfg: InteractiveConfig{Provider: "anthropic", Model: from}}
30
31	i.swapModel("anthropic", to, nil, false)
32
33	if ag.Model != to {
34		t.Fatalf("agent model = %q, want %q", ag.Model, to)
35	}
36	if ag.MaxTokens != toModel.MaxOutput {
37		t.Fatalf("agent MaxTokens = %d, want %d", ag.MaxTokens, toModel.MaxOutput)
38	}
39}