cli_session_test.go RawHistoryFolder

1package agent
2
3import (
4	"testing"
5
6	"github.com/patriceckhart/zot/packages/agent/modes"
7	"github.com/patriceckhart/zot/packages/core"
8	"github.com/patriceckhart/zot/packages/provider"
9)
10
11func TestLiveInteractiveAgentUsesReplacementAgentForSessionResume(t *testing.T) {
12	startup := core.NewAgent(nil, "startup-model", "", nil)
13	startup.SetMessages([]provider.Message{{
14		Role:    provider.RoleUser,
15		Content: []provider.Content{provider.TextBlock{Text: "startup transcript"}},
16	}})
17
18	replacement := core.NewAgent(nil, "replacement-model", "", nil)
19	iv := modes.NewInteractive(modes.InteractiveConfig{Agent: replacement})
20
21	resumed := []provider.Message{{
22		Role:    provider.RoleUser,
23		Content: []provider.Content{provider.TextBlock{Text: "resumed transcript"}},
24	}}
25	liveInteractiveAgent(iv, startup).SetMessages(resumed)
26
27	if got := firstMessageText(replacement.Messages()); got != "resumed transcript" {
28		t.Fatalf("replacement agent transcript = %q, want resumed transcript", got)
29	}
30	if got := firstMessageText(startup.Messages()); got != "startup transcript" {
31		t.Fatalf("startup agent transcript changed to %q", got)
32	}
33}
34
35func TestLiveInteractiveAgentFallsBackBeforeInteractiveConstruction(t *testing.T) {
36	startup := core.NewAgent(nil, "startup-model", "", nil)
37	if got := liveInteractiveAgent(nil, startup); got != startup {
38		t.Fatalf("fallback agent = %p, want %p", got, startup)
39	}
40}
41
42func firstMessageText(messages []provider.Message) string {
43	if len(messages) == 0 || len(messages[0].Content) == 0 {
44		return ""
45	}
46	text, _ := messages[0].Content[0].(provider.TextBlock)
47	return text.Text
48}