Function TestXMLToolCallFallback

pkg/runtime/runtime_test.go:504–536

TestXMLToolCallFallback verifies that blocks in text content are extracted as tool calls and not leaked as AgentChoice events.

(t *testing.T)

// TestXMLToolCallFallback verifies that <tool_call> blocks in text content
// are extracted as tool calls and not leaked as AgentChoice events.
func TestXMLToolCallFallback(t *testing.T) {
	t.Parallel()

	xmlPayload := `<tool_call>
{"name": "shell_exec", "arguments": {"cmd": "ls -la"}}
</tool_call>`

	stream := newStreamBuilder().
		AddContent(xmlPayload).
		AddStopWithUsage(10, 15).
		Build()

	sess := session.New(session.WithUserMessage("list files"))
	events := runSession(t, sess, stream)

	// The XML should be promoted to a PartialToolCall, not remain as plain text.
	require.True(t, hasEventType(t, events, &PartialToolCallEvent{}), "Expected PartialToolCallEvent from XML extraction")

	// No raw XML should have been emitted as an AgentChoice event.
	for _, ev := range events {
		if choice, ok := ev.(*AgentChoiceEvent); ok {
			require.NotContains(t, choice.Content, "<tool_call>", "XML tool call block must not appear in AgentChoice events")
		}
	}

	// Verify the extracted tool call fields.
	for _, ev := range events {
		if partial, ok := ev.(*PartialToolCallEvent); ok {
			require.Equal(t, "shell_exec", partial.ToolCall.Function.Name)
			require.JSONEq(t, `{"cmd": "ls -la"}`, partial.ToolCall.Function.Arguments)
		}
	}
}

// TestXMLToolCallFallback_WithPreamble verifies that preamble text before a
// <tool_call> block is emitted as AgentChoice while the XML is suppressed.