@rudderhq/server 0.6.1 → 0.6.2-canary.0

package/dist/services/chats.js

@@ -17,11 +17,13 @@ import { createChatConversation, createChatWithInitialMessage } from "./chats.cr
 import { conversationMutability, nextForkTitle } from "./chats.fork-helpers.js";
 import { buildSearchSnippet, CHAT_TRANSCRIPT_KEY, chatTranscriptFromPayload, chatTranscriptSummaryFromEntries, contentPath, escapeLikePattern, incomingMessagePreviewSql, issueProposalFromPayload, isVisibleIncomingChatMessage, listContextLinksForConversationIds, listPrimaryIssues, operationProposalDecisionStatusFromPayload, operationProposalFromPayload, resolveContextEntities, safeTrim, stripChatMetadataFromPayload, textContains, truncatePreview, visibleIncomingMessageSql, withOperationProposalDecisionState, withPersistedTranscript, } from "./chats.helpers.js";
 import { copyForkChatMessages, listRecentUserChatMessages, updateTrustedInlineVisualMappings, } from "./chats.inline-visual-persistence.js";
+import { createQueuedMessageWithStagedAttachments as createQueuedMessageWithStagedAttachmentsInDb } from "./chats.queued-message-create.js";
+import { findQueuedMessageReplay, updateChatAgentRuntimeInvariant, } from "./chats.runtime-controls.js";
 import { issueApprovalService } from "./issue-approvals.js";
 import { issueService } from "./issues.js";
 import { normalizeLocalLibraryPathMarkdown } from "./library-path-markdown.js";
 import { organizationService } from "./orgs.js";
-import { isPostgresError } from "./postgres-errors.js";
+import { sanitizePostgresJsonValue } from "./postgres-json.js";
 const CHAT_TITLE_MAX_LENGTH = 200;
 class InvalidQueueDeliveryActionLinkError extends Error {
 }
@@ -419,9 +421,6 @@ export function chatService(db) {
             .where(inArray(approvals.id, approvalIds));
         return new Map(approvalRows.map((row) => [row.id, row]));
     }
-    function isQueuePositionConflict(error) {
-        return isPostgresError(error, "23505");
-    }
     async function createGeneration(orgId, conversationId) {
         const now = new Date();
         const [row] = await db
@@ -765,7 +764,7 @@ export function chatService(db) {
                 generationSeq: nextSeq,
                 attemptEpoch: input.attemptEpoch,
                 eventKind: input.eventKind,
-                payload: input.payload ?? {},
+                payload: sanitizePostgresJsonValue(input.payload ?? {}),
                 controlActionId: input.controlActionId ?? null,
                 queueItemId: input.queueItemId ?? null,
                 emittedAt: input.emittedAt ?? null,
@@ -1268,112 +1267,11 @@ export function chatService(db) {
         });
     }
     async function createQueuedMessageWithStagedAttachments(input) {
-        if ((input.payload.attachmentIds?.length ?? 0) > 0) {
-            throw unprocessable("Queued messages cannot reference client-provided attachment ids");
-        }
-        const payload = normalizeQueuedMessagePayload(input.payload);
-        if (!payload.body.trim() && (payload.inlineAnnotations?.length ?? 0) === 0) {
-            throw unprocessable("Queued message body or at least one inline annotation is required");
-        }
-        const fingerprint = queuedMessageMutationFingerprint({
-            payload,
-            stagedAttachments: input.stagedAttachments,
-            attachmentFileIndexesByAnnotationId: input.attachmentFileIndexesByAnnotationId,
-        });
-        for (let attempt = 0; attempt < 3; attempt += 1) {
-            try {
-                return await db.transaction(async (tx) => {
-                    const existing = await tx
-                        .select()
-                        .from(chatQueuedMessages)
-                        .where(and(eq(chatQueuedMessages.orgId, input.orgId), eq(chatQueuedMessages.conversationId, input.conversationId), eq(chatQueuedMessages.clientMutationId, input.clientMutationId)))
-                        .limit(1)
-                        .then((rows) => rows[0] ?? null);
-                    if (existing) {
-                        const existingPrivateState = queuedAnnotationAssetState(existing.payload);
-                        const existingFingerprint = existingPrivateState?.fingerprint
-                            ?? queuedMessageMutationFingerprint({
-                                payload: normalizeQueuedMessagePayload(existing.payload),
-                                stagedAttachments: [],
-                                attachmentFileIndexesByAnnotationId: new Map(),
-                            });
-                        if (existingFingerprint !== fingerprint) {
-                            throw conflict("Queued message idempotency key reused with a different payload");
-                        }
-                        return {
-                            item: hydrateQueuedMessage(existing),
-                            accepted: false,
-                            cleanupAttachments: [...input.stagedAttachments],
-                        };
-                    }
-                    await validateCanonicalChatInlineAnnotations(tx, {
-                        orgId: input.orgId,
-                        conversationId: input.conversationId,
-                        annotations: payload.inlineAnnotations ?? [],
-                        uploadedFileCount: input.stagedAttachments.length,
-                        attachmentFileIndexesByAnnotationId: input.attachmentFileIndexesByAnnotationId,
-                    });
-                    const assetIds = [];
-                    for (const attachment of input.stagedAttachments) {
-                        const [asset] = await tx
-                            .insert(assets)
-                            .values({
-                            orgId: input.orgId,
-                            provider: attachment.provider,
-                            objectKey: attachment.objectKey,
-                            contentType: attachment.contentType,
-                            byteSize: attachment.byteSize,
-                            sha256: attachment.sha256,
-                            originalFilename: attachment.originalFilename,
-                            createdByAgentId: attachment.createdByAgentId,
-                            createdByUserId: attachment.createdByUserId,
-                        })
-                            .returning({ id: assets.id });
-                        if (!asset)
-                            throw new Error("Failed to stage queued annotation asset");
-                        assetIds.push(asset.id);
-                    }
-                    const persistedPayload = withQueuedAnnotationAssetState({
-                        payload,
-                        fingerprint,
-                        assetIds,
-                        stagedAttachments: input.stagedAttachments,
-                        attachmentFileIndexesByAnnotationId: input.attachmentFileIndexesByAnnotationId,
-                    });
-                    const [positionRow] = await tx
-                        .select({
-                        nextPosition: sql `coalesce(max(${chatQueuedMessages.position}), 0) + 1`,
-                    })
-                        .from(chatQueuedMessages)
-                        .where(and(eq(chatQueuedMessages.orgId, input.orgId), eq(chatQueuedMessages.conversationId, input.conversationId)));
-                    const [row] = await tx
-                        .insert(chatQueuedMessages)
-                        .values({
-                        orgId: input.orgId,
-                        conversationId: input.conversationId,
-                        clientMutationId: input.clientMutationId,
-                        position: Number(positionRow?.nextPosition ?? 1),
-                        payload: persistedPayload,
-                        requestActor: input.requestActor ?? null,
-                        expectedGenerationId: input.expectedGenerationId ?? null,
-                    })
-                        .returning();
-                    if (!row)
-                        throw new Error("Failed to create queued chat message");
-                    return {
-                        item: hydrateQueuedMessage(row),
-                        accepted: true,
-                        cleanupAttachments: [],
-                    };
-                });
-            }
-            catch (error) {
-                if (attempt < 2 && isQueuePositionConflict(error))
-                    continue;
-                throw error;
-            }
-        }
-        throw new Error("Failed to create queued chat message");
+        return createQueuedMessageWithStagedAttachmentsInDb(db, input);
+    }
+    async function getQueuedMessageReplay(input) {
+        const existing = await findQueuedMessageReplay(db, input);
+        return existing ? hydrateQueuedMessage(existing) : null;
     }
     async function createQueuedMessage(input) {
         const result = await createQueuedMessageWithStagedAttachments({
@@ -1430,6 +1328,9 @@ export function chatService(db) {
                 inlineAnnotations: input.payload.inlineAnnotations === undefined
                     ? currentPayload.inlineAnnotations
                     : input.payload.inlineAnnotations,
+                // Editing changes content, not the Agent/runtime admission snapshot.
+                agentId: currentPayload.agentId, model: currentPayload.model,
+                effort: currentPayload.effort,
             });
             if (!payload.body.trim() && (payload.inlineAnnotations?.length ?? 0) === 0) {
                 throw unprocessable("Queued message body or at least one inline annotation is required");
@@ -1468,6 +1369,7 @@ export function chatService(db) {
                     payload,
                     stagedAttachments: input.stagedAttachments,
                     attachmentFileIndexesByAnnotationId: input.attachmentFileIndexesByAnnotationId,
+                    runtimeSnapshotVersion: current.runtimeSnapshotVersion,
                 });
                 const assetIds = [];
                 for (const attachment of input.stagedAttachments) {
@@ -2853,6 +2755,8 @@ export function chatService(db) {
                 title: childTitle,
                 summary: source.summary,
                 preferredAgentId: source.preferredAgentId,
+                modelOverride: null,
+                effortOverride: null,
                 routedAgentId: source.routedAgentId,
                 primaryIssueId: source.primaryIssueId,
                 forkedFromConversationId: source.id,
@@ -2954,6 +2858,10 @@ export function chatService(db) {
             return null;
         return getById(id);
     }
+    async function updateAgentModelInvariant(input) {
+        await updateChatAgentRuntimeInvariant(db, input);
+        return getById(input.id);
+    }
     async function updateDefaultTitle(id, title, expectedCurrentTitle = "New chat") {
         const [updated] = await db
             .update(chatConversations)
@@ -3906,6 +3814,7 @@ export function chatService(db) {
         create,
         createWithInitialMessage,
         update,
+        updateAgentModelInvariant,
         updateDefaultTitle,
         replaceSystemGeneratedTitle,
         listRecentUserMessages,
@@ -3928,6 +3837,7 @@ export function chatService(db) {
         getLatestGeneration,
         getQueueSnapshot,
         listQueuedMessages,
+        getQueuedMessageReplay,
         createQueuedMessage,
         createQueuedMessageWithStagedAttachments,
         updateQueuedMessage,