Function TestSessionFromEventsWithToolDefinitions

pkg/evaluation/save_test.go:517–571

(t *testing.T)

}

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

	events := []map[string]any{
		{"type": "agent_choice", "content": "Let me read that file.", "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"}`,
				},
			},
			"tool_definition": map[string]any{
				"name":        "read_file",
				"category":    "filesystem",
				"description": "Read the contents of a file",
			},
		},
		{
			"type":         "tool_call_response",
			"tool_call_id": "call_123",
			"response":     "file content",
		},
		{"type": "stream_stopped"},
	}

	sess := SessionFromEvents(events, "test", []string{"read the file"})

	// Find the assistant message with tool calls
	var assistantMsg *session.Message
	for _, item := range sess.Messages {
		if item.Message != nil && item.Message.Message.Role == chat.MessageRoleAssistant && len(item.Message.Message.ToolCalls) > 0 {
			assistantMsg = item.Message
			break
		}
	}

	require.NotNil(t, assistantMsg, "should have assistant message with tool calls")
	assert.Len(t, assistantMsg.Message.ToolCalls, 1)
	assert.Len(t, assistantMsg.Message.ToolDefinitions, 1)

	// Verify tool call
	toolCall := assistantMsg.Message.ToolCalls[0]
	assert.Equal(t, "call_123", toolCall.ID)
	assert.Equal(t, "read_file", toolCall.Function.Name)

	// Verify tool definition
	toolDef := assistantMsg.Message.ToolDefinitions[0]
	assert.Equal(t, "read_file", toolDef.Name)
	assert.Equal(t, "filesystem", toolDef.Category)
	assert.Equal(t, "Read the contents of a file", toolDef.Description)
}

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