session_deferred_tools_test.go RawHistoryFolder

1package core
2
3import (
4	"testing"
5
6	"github.com/patriceckhart/zot/packages/provider"
7)
8
9func TestSessionPreservesActivatedDeferredTools(t *testing.T) {
10	session, err := NewSession(t.TempDir(), "/workspace", "moonshotai", "kimi-k3", "test")
11	if err != nil {
12		t.Fatal(err)
13	}
14	message := provider.Message{
15		Role:           provider.RoleTool,
16		AddedToolNames: []string{"lookup_weather"},
17		Content: []provider.Content{provider.ToolResultBlock{
18			CallID:  "call-1",
19			Content: []provider.Content{provider.TextBlock{Text: "enabled"}},
20		}},
21	}
22	if err := session.AppendMessage(message); err != nil {
23		t.Fatal(err)
24	}
25	path := session.Path
26	if err := session.Close(); err != nil {
27		t.Fatal(err)
28	}
29
30	reopened, messages, err := OpenSession(path)
31	if err != nil {
32		t.Fatal(err)
33	}
34	defer reopened.Close()
35	if len(messages) != 1 || len(messages[0].AddedToolNames) != 1 || messages[0].AddedToolNames[0] != "lookup_weather" {
36		t.Fatalf("messages = %+v", messages)
37	}
38}