rpc_setmodel_test.go RawHistoryFolder

1package agent
2
3import (
4	"bytes"
5	"context"
6	"encoding/json"
7	"testing"
8
9	"github.com/patriceckhart/zot/packages/core"
10	"github.com/patriceckhart/zot/packages/provider"
11)
12
13func TestRPCSetModelUpdatesMaxTokens(t *testing.T) {
14	const to = "claude-opus-4-1-20250805"
15	want, err := provider.FindModel("anthropic", to)
16	if err != nil {
17		t.Fatal(err)
18	}
19	ag := core.NewAgent(&startCaptureClient{}, "claude-3-5-haiku-20241022", "", nil)
20	ag.MaxTokens = 4096
21	var out bytes.Buffer
22	s := &rpcServer{ctx: context.Background(), agent: ag, provider: "anthropic", model: ag.Model, out: &out}
23
24	raw, err := json.Marshal(map[string]string{"model": to})
25	if err != nil {
26		t.Fatal(err)
27	}
28	s.dispatch("set_model", "1", raw)
29	s.inFlight.Wait()
30
31	if ag.Model != to || ag.MaxTokens != want.MaxOutput {
32		t.Fatalf("after set_model: model=%q MaxTokens=%d, want %q %d (out=%s)", ag.Model, ag.MaxTokens, to, want.MaxOutput, out.String())
33	}
34}