Function TestWithReasoning

pkg/runtime/runtime_test.go:396–436

(t *testing.T)

}

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

	stream := newStreamBuilder().
		AddReasoning("Let me think about this...").
		AddReasoning(" I should respond politely.").
		AddContent("Hello, how can I help you?").
		AddStopWithUsage(10, 15).
		Build()

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

	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, 12)
	msgAdded := events[9].(*MessageAddedEvent)
	require.NotNil(t, msgAdded.Message)

	expectedEvents := []Event{
		TeamInfo([]AgentDetails{{Name: "root", Provider: "test", Model: "mock-model"}}, "root"),
		ToolsetInfo(0, false, "root"),
		UserMessage("Hi", sess.ID, nil, 0),
		StreamStarted(sess.ID, "root"),
		ToolsetInfo(0, false, "root"),
		AgentInfo("root", "test/mock-model", "", ""),
		AgentChoiceReasoning("root", sess.ID, "Let me think about this..."),
		AgentChoiceReasoning("root", sess.ID, " I should respond politely."),
		AgentChoice("root", sess.ID, "Hello, how can I help you?"),
		MessageAdded(sess.ID, msgAdded.Message, "root"),
		NewTokenUsageEvent(sess.ID, "root", &Usage{InputTokens: 10, OutputTokens: 15, ContextLength: 25, LastMessage: &MessageUsage{
			Usage:        chat.Usage{InputTokens: 10, OutputTokens: 15},
			Model:        "test/mock-model",
			FinishReason: chat.FinishReasonStop,
		}}),
		StreamStopped(sess.ID, "root", "normal"),
	}

	assertEventsEqual(t, expectedEvents, events)
}

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