Function TestXMLToolCallFallback_WithPreamble

pkg/runtime/runtime_test.go:540–569

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

(t *testing.T)

// TestXMLToolCallFallback_WithPreamble verifies that preamble text before a
// <tool_call> block is emitted as AgentChoice while the XML is suppressed.
func TestXMLToolCallFallback_WithPreamble(t *testing.T) {
	t.Parallel()

	stream := newStreamBuilder().
		AddContent("I'll list the files for you.\n").
		AddContent(`<tool_call>{"name": "ls", "arguments": {"path": "/tmp"}}</tool_call>`).
		AddStopWithUsage(12, 18).
		Build()

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

	// Preamble text must have been emitted as AgentChoice.
	var choiceContents []string
	for _, ev := range events {
		if choice, ok := ev.(*AgentChoiceEvent); ok {
			choiceContents = append(choiceContents, choice.Content)
		}
	}
	require.NotEmpty(t, choiceContents, "Expected AgentChoice events for preamble")
	require.Contains(t, strings.Join(choiceContents, ""), "I'll list the files for you.")

	// XML itself must not leak into AgentChoice.
	for _, c := range choiceContents {
		require.NotContains(t, c, "<tool_call>")
	}

	// Tool must still be extracted.
	require.True(t, hasEventType(t, events, &PartialToolCallEvent{}))
}

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