Function TestSessionFromEvents

pkg/evaluation/save_test.go:344–446

(t *testing.T)

}

func TestSessionFromEvents(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name         string
		events       []map[string]any
		title        string
		question     string
		wantMessages int
		wantContent  string
	}{
		{
			name:         "empty events",
			events:       []map[string]any{},
			title:        "test",
			question:     "hello",
			wantMessages: 1, // just the user message
			wantContent:  "",
		},
		{
			name: "agent choice events",
			events: []map[string]any{
				{"type": "agent_choice", "content": "Hello ", "agent_name": "root"},
				{"type": "agent_choice", "content": "world!"},
				{"type": "stream_stopped"},
			},
			title:        "test",
			question:     "greet me",
			wantMessages: 2, // user + assistant
			wantContent:  "Hello world!",
		},
		{
			name: "tool calls and responses",
			events: []map[string]any{
				{"type": "agent_choice", "content": "Let me help.", "agent_name": "root"},
				{
					"type": "tool_call",
					"tool_call": map[string]any{
						"id":   "call_123",
						"type": "function",
						"function": map[string]any{
							"name":      "read_file",
							"arguments": `{"path": "test.txt"}`,
						},
					},
				},
				{
					"type":         "tool_call_response",
					"tool_call_id": "call_123",
					"response":     "file content",
				},
				{"type": "agent_choice", "content": "Done!"},
				{"type": "stream_stopped"},
			},
			title:        "test",
			question:     "read file",
			wantMessages: 4, // user + assistant (with tool call) + tool response + assistant
			wantContent:  "Done!",