Function insertUserMessageAndSetPending

coderd/x/chatd/chatd.go:4098–4150

insertUserMessageAndSetPending inserts a user message, transitions the chat to pending when needed, and returns the refreshed chat row.

(
	ctx context.Context,
	store database.Store,
	lockedChat database.Chat,
	modelConfigID uuid.UUID,
	content pqtype.NullRawMessage,
	createdBy uuid.UUID,
	apiKeyID string,
)

// insertUserMessageAndSetPending inserts a user message, transitions the
// chat to pending when needed, and returns the refreshed chat row.
func insertUserMessageAndSetPending(
	ctx context.Context,
	store database.Store,
	lockedChat database.Chat,
	modelConfigID uuid.UUID,
	content pqtype.NullRawMessage,
	createdBy uuid.UUID,
	apiKeyID string,
) (database.ChatMessage, database.Chat, error) {
	msgParams := database.InsertChatMessagesParams{ //nolint:exhaustruct // Fields populated by appendUserChatMessage.
		ChatID: lockedChat.ID,
	}
	insertUserMsg := newUserChatMessage(
		apiKeyID,
		content,
		database.ChatMessageVisibilityBoth,
		modelConfigID,
		chatprompt.CurrentContentVersion,
	)
	insertUserMsg = insertUserMsg.withCreatedBy(createdBy)
	appendUserChatMessage(&msgParams, insertUserMsg)
	messages, err := insertChatMessageWithStore(ctx, store, msgParams)
	if err != nil {
		return database.ChatMessage{}, database.Chat{}, err
	}
	message := messages[0]

	if lockedChat.Status == database.ChatStatusPending {
		if modelConfigID == uuid.Nil || lockedChat.LastModelConfigID == modelConfigID {
			return message, lockedChat, nil
		}
		// The InsertChatMessages CTE updates chats.last_model_config_id when
		// the message's model config differs. Reload to surface that change.
		updatedChat, err := store.GetChatByID(ctx, lockedChat.ID)
		if err != nil {
			return database.ChatMessage{}, database.Chat{}, xerrors.Errorf("get chat after model config update: %w", err)
		}
		return message, updatedChat, nil
	}

	updatedChat, err := store.UpdateChatStatus(ctx, database.UpdateChatStatusParams{
		ID:          lockedChat.ID,
		Status:      database.ChatStatusPending,
		WorkerID:    uuid.NullUUID{},
		StartedAt:   sql.NullTime{},
		HeartbeatAt: sql.NullTime{},
		LastError:   pqtype.NullRawMessage{},
	})
	if err != nil {
		return database.ChatMessage{}, database.Chat{}, xerrors.Errorf("set chat pending: %w", err)
	}
	return message, updatedChat, nil
}

// shouldQueueUserMessage reports whether a user message should be
// queued while a chat is active.