Function TestMultipleContentChunks

pkg/runtime/runtime_test.go:350–394

(t *testing.T)

}

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

	stream := newStreamBuilder().
		AddContent("Hello ").
		AddContent("there, ").
		AddContent("how ").
		AddContent("are ").
		AddContent("you?").
		AddStopWithUsage(8, 12).
		Build()

	sess := session.New(session.WithUserMessage("Please greet me"))

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

	expectedEvents := []Event{
		TeamInfo([]AgentDetails{{Name: "root", Provider: "test", Model: "mock-model"}}, "root"),
		ToolsetInfo(0, false, "root"),
		UserMessage("Please greet me", sess.ID, nil, 0),
		StreamStarted(sess.ID, "root"),
		ToolsetInfo(0, false, "root"),
		AgentInfo("root", "test/mock-model", "", ""),
		AgentChoice("root", sess.ID, "Hello "),
		AgentChoice("root", sess.ID, "there, "),
		AgentChoice("root", sess.ID, "how "),
		AgentChoice("root", sess.ID, "are "),
		AgentChoice("root", sess.ID, "you?"),
		MessageAdded(sess.ID, msgAdded.Message, "root"),
		NewTokenUsageEvent(sess.ID, "root", &Usage{InputTokens: 8, OutputTokens: 12, ContextLength: 20, LastMessage: &MessageUsage{
			Usage:        chat.Usage{InputTokens: 8, OutputTokens: 12},
			Model:        "test/mock-model",
			FinishReason: chat.FinishReasonStop,
		}}),
		StreamStopped(sess.ID, "root", "normal"),
	}

	assertEventsEqual(t, expectedEvents, events)
}

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