Method PromoteQueued

coderd/x/chatd/chatd.go:2425–2616

PromoteQueued promotes a queued message into chat history. On a running chat with a fresh worker heartbeat the promote is deferred to the worker's persist+auto-promote so partial assistant output is not lost; otherwise it inserts the user message synchronously.

(
	ctx context.Context,
	opts PromoteQueuedOptions,
)

// to the worker's persist+auto-promote so partial assistant output
// is not lost; otherwise it inserts the user message synchronously.
func (p *Server) PromoteQueued(
	ctx context.Context,
	opts PromoteQueuedOptions,
) (PromoteQueuedResult, error) {
	if opts.ChatID == uuid.Nil {
		return PromoteQueuedResult{}, xerrors.New("chat_id is required")
	}

	var (
		result           PromoteQueuedResult
		promoted         database.ChatMessage
		updatedChat      database.Chat
		remainingQueue   []database.ChatQueuedMessage
		deferred         bool
		syntheticResults []database.ChatMessage
	)

	txErr := p.db.InTx(func(tx database.Store) error {
		lockedChat, err := tx.GetChatByIDForUpdate(ctx, opts.ChatID)
		if err != nil {
			return xerrors.Errorf("lock chat: %w", err)
		}

		if lockedChat.Archived {
			return ErrChatArchived
		}

		queuedMessages, err := tx.GetChatQueuedMessages(ctx, opts.ChatID)
		if err != nil {
			return xerrors.Errorf("get queued messages: %w", err)
		}

		var (
			targetContent       json.RawMessage
			targetModelConfigID uuid.NullUUID
			targetAPIKeyID      sql.NullString
			found               bool
		)
		for _, qm := range queuedMessages {
			if qm.ID == opts.QueuedMessageID {
				targetContent = qm.Content
				targetModelConfigID = qm.ModelConfigID
				targetAPIKeyID = qm.APIKeyID
				found = true
				break
			}
		}
		if !found {
			return xerrors.Errorf("queued message %d not found in chat %s", opts.QueuedMessageID, opts.ChatID)
		}

		// Setting pending would trip persistStep's ownership guard
		// and drop the worker's partial output. Set waiting and
		// reorder the queued row so the worker's auto-promote picks
		// it up after the persist.
		heartbeatFresh := lockedChat.HeartbeatAt.Valid &&
			p.clock.Now().Sub(lockedChat.HeartbeatAt.Time) < p.inFlightChatStaleAfter
		if lockedChat.Status == database.ChatStatusRunning && heartbeatFresh {