Function TestPromoteQueuedMessageUsesQueuedModelConfigID

coderd/x/chatd/chatd_test.go:2618–2673

(t *testing.T)

}

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

	db, ps := dbtestutil.NewDB(t)
	replica := newTestServer(t, db, ps, uuid.New())

	ctx := testutil.Context(t, testutil.WaitLong)
	user, org, modelConfigA := seedChatDependencies(t, db)
	modelConfigB := insertChatModelConfigWithCallConfig(
		t,
		db,
		user.ID,
		"openai",
		"gpt-4o-mini-promote-"+uuid.NewString(),
		codersdk.ChatModelCallConfig{},
	)

	chat := dbgen.Chat(t, db, database.Chat{
		OrganizationID:    org.ID,
		OwnerID:           user.ID,
		LastModelConfigID: modelConfigA.ID,
		Title:             "promote queued uses stored model",
	})

	queuedContent, err := json.Marshal([]codersdk.ChatMessagePart{codersdk.ChatMessageText("queued with model b")})
	require.NoError(t, err)
	queuedMessage, err := db.InsertChatQueuedMessage(ctx, database.InsertChatQueuedMessageParams{
		ChatID:  chat.ID,
		Content: queuedContent,
		ModelConfigID: uuid.NullUUID{
			UUID:  modelConfigB.ID,
			Valid: true,
		},
	})
	require.NoError(t, err)

	result, err := replica.PromoteQueued(ctx, chatd.PromoteQueuedOptions{
		ChatID:          chat.ID,
		QueuedMessageID: queuedMessage.ID,
		CreatedBy:       user.ID,
	})
	require.NoError(t, err)
	require.True(t, result.PromotedMessage.ModelConfigID.Valid)
	require.Equal(t, modelConfigB.ID, result.PromotedMessage.ModelConfigID.UUID)

	storedChat, err := db.GetChatByID(ctx, chat.ID)
	require.NoError(t, err)
	require.Equal(t, modelConfigB.ID, storedChat.LastModelConfigID)
	// The processor can pick up the pending chat immediately after
	// promotion, so this test only requires that promotion moved it out of
	// waiting and preserved the queued model configuration.
	require.Contains(t, []database.ChatStatus{
		database.ChatStatusPending,
		database.ChatStatusRunning,
	}, storedChat.Status)
}

func TestPromoteQueuedMessageReloadsChatWhenModelConfigChangesDuringPending(t *testing.T) {