Function TestMixedContentAndReasoning

pkg/runtime/runtime_test.go:438–480

(t *testing.T)

}

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

	stream := newStreamBuilder().
		AddReasoning("The user wants a greeting").
		AddContent("Hello!").
		AddReasoning(" I should be friendly").
		AddContent(" How can I help you today?").
		AddStopWithUsage(15, 20).
		Build()

	sess := session.New(session.WithUserMessage("Hi there"))

	events := runSession(t, sess, stream)

	// Extract the actual message from MessageAddedEvent to use in comparison
	// (it contains dynamic fields like CreatedAt that we can't predict)
	require.Len(t, events, 13)
	msgAdded := events[10].(*MessageAddedEvent)
	require.NotNil(t, msgAdded.Message)

	expectedEvents := []Event{
		TeamInfo([]AgentDetails{{Name: "root", Provider: "test", Model: "mock-model"}}, "root"),
		ToolsetInfo(0, false, "root"),
		UserMessage("Hi there", sess.ID, nil, 0),
		StreamStarted(sess.ID, "root"),
		ToolsetInfo(0, false, "root"),
		AgentInfo("root", "test/mock-model", "", ""),
		AgentChoiceReasoning("root", sess.ID, "The user wants a greeting"),
		AgentChoice("root", sess.ID, "Hello!"),
		AgentChoiceReasoning("root", sess.ID, " I should be friendly"),
		AgentChoice("root", sess.ID, " How can I help you today?"),
		MessageAdded(sess.ID, msgAdded.Message, "root"),
		NewTokenUsageEvent(sess.ID, "root", &Usage{InputTokens: 15, OutputTokens: 20, ContextLength: 35, LastMessage: &MessageUsage{
			Usage:        chat.Usage{InputTokens: 15, OutputTokens: 20},
			Model:        "test/mock-model",
			FinishReason: chat.FinishReasonStop,
		}}),
		StreamStopped(sess.ID, "root", "normal"),
	}

	assertEventsEqual(t, expectedEvents, events)
}

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