Function TestEditMessageWithModelConfigOverride

coderd/x/chatd/chatd_test.go:8945–8993

TestEditMessageWithModelConfigOverride verifies that callers can change the model when editing a previous user message. The replacement message must persist with the new model and the chat's LastModelConfigID must be advanced so the assistant turn that follows runs against the new selection.

(t *testing.T)

// LastModelConfigID must be advanced so the assistant turn that follows
// runs against the new selection.
func TestEditMessageWithModelConfigOverride(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, modelA := seedChatDependencies(t, db)
	modelB := insertChatModelConfigWithCallConfig(
		t,
		db,
		user.ID,
		"openai",
		"gpt-4o-mini-edit-"+uuid.NewString(),
		codersdk.ChatModelCallConfig{},
	)

	chat, err := replica.CreateChat(ctx, chatd.CreateOptions{
		OwnerID:            user.ID,
		OrganizationID:     org.ID,
		Title:              "edit-with-model-override",
		ModelConfigID:      modelA.ID,
		InitialUserContent: []codersdk.ChatMessagePart{codersdk.ChatMessageText("original")},
	})
	require.NoError(t, err)

	initial, err := db.GetChatMessagesByChatID(ctx, database.GetChatMessagesByChatIDParams{
		ChatID:  chat.ID,
		AfterID: 0,
	})
	require.NoError(t, err)
	require.Len(t, initial, 1)
	require.Equal(t, modelA.ID, initial[0].ModelConfigID.UUID)

	result, err := replica.EditMessage(ctx, chatd.EditMessageOptions{
		ChatID:          chat.ID,
		EditedMessageID: initial[0].ID,
		Content:         []codersdk.ChatMessagePart{codersdk.ChatMessageText("edited")},
		ModelConfigID:   modelB.ID,
	})
	require.NoError(t, err)
	require.True(t, result.Message.ModelConfigID.Valid)
	require.Equal(t, modelB.ID, result.Message.ModelConfigID.UUID)

	storedChat, err := db.GetChatByID(ctx, chat.ID)
	require.NoError(t, err)
	require.Equal(t, modelB.ID, storedChat.LastModelConfigID,
		"edit must update last_model_config_id so the assistant turn picks up the new model")
}

// TestEditMessagePreservesModelConfigByDefault verifies that omitting
// ModelConfigID on edit keeps the original message's model. This is the