@rudderhq/server 0.5.0 → 0.5.1

package/dist/services/chat-family-groups.js

@@ -0,0 +1,96 @@
+import { chatConversations, messengerCustomGroupEntries, messengerCustomGroups, } from "@rudderhq/db";
+import { MESSENGER_FORK_GROUP_DEFAULT_ICON } from "@rudderhq/shared";
+import { and, asc, desc, eq, sql } from "drizzle-orm";
+async function findChatFamilyGroup(client, orgId, userId, rootConversationId) {
+    const rootThreadKey = `chat:${rootConversationId}`;
+    return client
+        .select({ id: messengerCustomGroups.id })
+        .from(messengerCustomGroups)
+        .innerJoin(messengerCustomGroupEntries, eq(messengerCustomGroupEntries.groupId, messengerCustomGroups.id))
+        .where(and(eq(messengerCustomGroups.orgId, orgId), eq(messengerCustomGroups.userId, userId), eq(messengerCustomGroupEntries.orgId, orgId), eq(messengerCustomGroupEntries.userId, userId), eq(messengerCustomGroupEntries.threadKey, rootThreadKey)))
+        .orderBy(asc(messengerCustomGroups.sortOrder), asc(messengerCustomGroups.createdAt))
+        .limit(1)
+        .then((rows) => rows[0] ?? null);
+}
+async function createChatFamilyGroup(client, orgId, userId, name) {
+    const [lastGroup] = await client
+        .select({ sortOrder: messengerCustomGroups.sortOrder })
+        .from(messengerCustomGroups)
+        .where(and(eq(messengerCustomGroups.orgId, orgId), eq(messengerCustomGroups.userId, userId)))
+        .orderBy(desc(messengerCustomGroups.sortOrder))
+        .limit(1);
+    const now = new Date();
+    const [group] = await client
+        .insert(messengerCustomGroups)
+        .values({
+        orgId,
+        userId,
+        name: name.trim() || "Forked conversation",
+        icon: MESSENGER_FORK_GROUP_DEFAULT_ICON,
+        sortOrder: (lastGroup?.sortOrder ?? -1) + 1,
+        updatedAt: now,
+    })
+        .returning();
+    if (!group)
+        throw new Error("Failed to create Messenger conversation family group");
+    return group;
+}
+async function assignChatToFamilyGroup(client, orgId, userId, groupId, conversationId) {
+    const [lastEntry] = await client
+        .select({ sortOrder: messengerCustomGroupEntries.sortOrder })
+        .from(messengerCustomGroupEntries)
+        .where(and(eq(messengerCustomGroupEntries.orgId, orgId), eq(messengerCustomGroupEntries.userId, userId), eq(messengerCustomGroupEntries.groupId, groupId)))
+        .orderBy(desc(messengerCustomGroupEntries.sortOrder))
+        .limit(1);
+    const nextSortOrder = (lastEntry?.sortOrder ?? -1) + 1;
+    const now = new Date();
+    await client
+        .insert(messengerCustomGroupEntries)
+        .values({
+        orgId,
+        userId,
+        groupId,
+        threadKey: `chat:${conversationId}`,
+        sortOrder: nextSortOrder,
+        updatedAt: now,
+    })
+        .onConflictDoUpdate({
+        target: [
+            messengerCustomGroupEntries.orgId,
+            messengerCustomGroupEntries.userId,
+            messengerCustomGroupEntries.threadKey,
+        ],
+        set: {
+            groupId,
+            sortOrder: sql `CASE
+          WHEN ${messengerCustomGroupEntries.groupId} = ${groupId}
+            THEN ${messengerCustomGroupEntries.sortOrder}
+          ELSE ${nextSortOrder}
+        END`,
+            updatedAt: now,
+        },
+    });
+}
+export async function ensureChatFamilyGroup(client, input) {
+    // Family membership is user-scoped, but the root row is the stable lock shared
+    // by concurrent Fork and Side Chat promotions in the same conversation family.
+    await client.execute(sql `
+    SELECT ${chatConversations.id}
+    FROM ${chatConversations}
+    WHERE ${chatConversations.id} = ${input.rootConversationId}
+      AND ${chatConversations.orgId} = ${input.orgId}
+    FOR UPDATE
+  `);
+    const existing = await findChatFamilyGroup(client, input.orgId, input.userId, input.rootConversationId);
+    const group = existing ?? await createChatFamilyGroup(client, input.orgId, input.userId, input.groupName);
+    const familyConversationIds = new Set([
+        input.rootConversationId,
+        input.sourceConversationId,
+        input.childConversationId,
+    ]);
+    for (const conversationId of familyConversationIds) {
+        await assignChatToFamilyGroup(client, input.orgId, input.userId, group.id, conversationId);
+    }
+    return group;
+}
+//# sourceMappingURL=chat-family-groups.js.map

package/dist/services/chat-family-groups.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-family-groups.js","sourceRoot":"","sources":["../../src/services/chat-family-groups.ts"],"names":[],"mappings":"AACA,OAAO,EACL,iBAAiB,EACjB,2BAA2B,EAC3B,qBAAqB,GACtB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,iCAAiC,EAAE,MAAM,kBAAkB,CAAC;AACrE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAItD,KAAK,UAAU,mBAAmB,CAChC,MAA0B,EAC1B,KAAa,EACb,MAAc,EACd,kBAA0B;IAE1B,MAAM,aAAa,GAAG,QAAQ,kBAAkB,EAAE,CAAC;IACnD,OAAO,MAAM;SACV,MAAM,CAAC,EAAE,EAAE,EAAE,qBAAqB,CAAC,EAAE,EAAE,CAAC;SACxC,IAAI,CAAC,qBAAqB,CAAC;SAC3B,SAAS,CAAC,2BAA2B,EAAE,EAAE,CAAC,2BAA2B,CAAC,OAAO,EAAE,qBAAqB,CAAC,EAAE,CAAC,CAAC;SACzG,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,qBAAqB,CAAC,KAAK,EAAE,KAAK,CAAC,EACtC,EAAE,CAAC,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,EACxC,EAAE,CAAC,2BAA2B,CAAC,KAAK,EAAE,KAAK,CAAC,EAC5C,EAAE,CAAC,2BAA2B,CAAC,MAAM,EAAE,MAAM,CAAC,EAC9C,EAAE,CAAC,2BAA2B,CAAC,SAAS,EAAE,aAAa,CAAC,CACzD,CAAC;SACD,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;SACnF,KAAK,CAAC,CAAC,CAAC;SACR,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;AACrC,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,MAAqC,EACrC,KAAa,EACb,MAAc,EACd,IAAY;IAEZ,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,MAAM;SAC7B,MAAM,CAAC,EAAE,SAAS,EAAE,qBAAqB,CAAC,SAAS,EAAE,CAAC;SACtD,IAAI,CAAC,qBAAqB,CAAC;SAC3B,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,qBAAqB,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;SAC5F,OAAO,CAAC,IAAI,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;SAC9C,KAAK,CAAC,CAAC,CAAC,CAAC;IACZ,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM;SACzB,MAAM,CAAC,qBAAqB,CAAC;SAC7B,MAAM,CAAC;QACN,KAAK;QACL,MAAM;QACN,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,qBAAqB;QAC1C,IAAI,EAAE,iCAAiC;QACvC,SAAS,EAAE,CAAC,SAAS,EAAE,SAAS,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;QAC3C,SAAS,EAAE,GAAG;KACf,CAAC;SACD,SAAS,EAAE,CAAC;IACf,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IACpF,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,MAAqC,EACrC,KAAa,EACb,MAAc,EACd,OAAe,EACf,cAAsB;IAEtB,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,MAAM;SAC7B,MAAM,CAAC,EAAE,SAAS,EAAE,2BAA2B,CAAC,SAAS,EAAE,CAAC;SAC5D,IAAI,CAAC,2BAA2B,CAAC;SACjC,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,2BAA2B,CAAC,KAAK,EAAE,KAAK,CAAC,EAC5C,EAAE,CAAC,2BAA2B,CAAC,MAAM,EAAE,MAAM,CAAC,EAC9C,EAAE,CAAC,2BAA2B,CAAC,OAAO,EAAE,OAAO,CAAC,CACjD,CAAC;SACD,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,SAAS,CAAC,CAAC;SACpD,KAAK,CAAC,CAAC,CAAC,CAAC;IACZ,MAAM,aAAa,GAAG,CAAC,SAAS,EAAE,SAAS,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACvD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,MAAM;SACT,MAAM,CAAC,2BAA2B,CAAC;SACnC,MAAM,CAAC;QACN,KAAK;QACL,MAAM;QACN,OAAO;QACP,SAAS,EAAE,QAAQ,cAAc,EAAE;QACnC,SAAS,EAAE,aAAa;QACxB,SAAS,EAAE,GAAG;KACf,CAAC;SACD,kBAAkB,CAAC;QAClB,MAAM,EAAE;YACN,2BAA2B,CAAC,KAAK;YACjC,2BAA2B,CAAC,MAAM;YAClC,2BAA2B,CAAC,SAAS;SACtC;QACD,GAAG,EAAE;YACH,OAAO;YACP,SAAS,EAAE,GAAG,CAAQ;iBACb,2BAA2B,CAAC,OAAO,MAAM,OAAO;mBAC9C,2BAA2B,CAAC,SAAS;iBACvC,aAAa;YAClB;YACJ,SAAS,EAAE,GAAG;SACf;KACF,CAAC,CAAC;AACP,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,MAA6B,EAC7B,KAOC;IAED,+EAA+E;IAC/E,+EAA+E;IAC/E,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,CAAA;aACb,iBAAiB,CAAC,EAAE;WACtB,iBAAiB;YAChB,iBAAiB,CAAC,EAAE,MAAM,KAAK,CAAC,kBAAkB;YAClD,iBAAiB,CAAC,KAAK,MAAM,KAAK,CAAC,KAAK;;GAEjD,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CACxC,MAAM,EACN,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,kBAAkB,CACzB,CAAC;IACF,MAAM,KAAK,GAAG,QAAQ,IAAI,MAAM,qBAAqB,CACnD,MAAM,EACN,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,SAAS,CAChB,CAAC;IACF,MAAM,qBAAqB,GAAG,IAAI,GAAG,CAAC;QACpC,KAAK,CAAC,kBAAkB;QACxB,KAAK,CAAC,oBAAoB;QAC1B,KAAK,CAAC,mBAAmB;KAC1B,CAAC,CAAC;IACH,KAAK,MAAM,cAAc,IAAI,qBAAqB,EAAE,CAAC;QACnD,MAAM,uBAAuB,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;IAC7F,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/services/chat-generation-protocol.d.ts

@@ -1,26 +1,12 @@
 import type { Db } from "@rudderhq/db";
 import { chatControlActions, chatGenerationEvents, chatGenerations, chatGenerationTerminalOutbox, chatQueuedMessages } from "@rudderhq/db";
-import type { ChatControlDisposition, ChatGenerationEventKind, ChatGenerationStatus, ChatStreamTranscriptEntry } from "@rudderhq/shared";
+import type { ChatControlDisposition, ChatGenerationStatus, ChatStreamTranscriptEntry } from "@rudderhq/shared";
+import { type AppendEventFields, type ChatGenerationProtocolTransaction } from "./chat-generation-protocol.helpers.js";
+export { hashChatGenerationBody } from "./chat-generation-protocol.helpers.js";
 type GenerationRow = typeof chatGenerations.$inferSelect;
 type GenerationEventRow = typeof chatGenerationEvents.$inferSelect;
 type ControlActionRow = typeof chatControlActions.$inferSelect;
 type TerminalOutboxRow = typeof chatGenerationTerminalOutbox.$inferSelect;
-export type ChatGenerationProtocolTransaction = Parameters<Parameters<Db["transaction"]>[0]>[0];
-export declare function hashChatGenerationBody(body: string): string;
-type AppendEventFields = {
-    orgId: string;
-    generationId: string;
-    attemptEpoch: number;
-    eventKind: ChatGenerationEventKind;
-    payload?: Record<string, unknown>;
-    bodyOffset?: number | null;
-    bodyLength?: number | null;
-    assistantMessageId?: string | null;
-    runId?: string | null;
-    controlActionId?: string | null;
-    queueItemId?: string | null;
-    emittedAt?: Date | null;
-};
 type VisibleGenerationProjection = {
     body: string;
     transcript: ChatStreamTranscriptEntry[];
@@ -81,7 +67,7 @@ export declare function chatGenerationProtocolService(db: Db): {
             generationId: string;
             attemptEpoch: number;
             generationSeq: number;
-            eventKind: ChatGenerationEventKind;
+            eventKind: import("@rudderhq/shared").ChatGenerationEventKind;
             bodyOffset: number | null;
             bodyLength: number | null;
             queueItemId: string | null;
@@ -158,22 +144,22 @@ export declare function chatGenerationProtocolService(db: Db): {
             lateBytes: number;
         };
         message: {
-            status: string;
-            conversationId: string;
             id: string;
             orgId: string;
+            conversationId: string;
+            role: string;
             kind: string;
-            createdAt: Date;
-            updatedAt: Date;
-            runId: string | null;
+            status: string;
             body: string;
-            role: string;
-            approvalId: string | null;
             structuredPayload: Record<string, unknown> | null;
+            approvalId: string | null;
+            runId: string | null;
             replyingAgentId: string | null;
             chatTurnId: string | null;
             turnVariant: number;
             supersededAt: Date | null;
+            createdAt: Date;
+            updatedAt: Date;
         };
         event: {
             assistantMessageId: string | null;
@@ -185,7 +171,7 @@ export declare function chatGenerationProtocolService(db: Db): {
             generationId: string;
             attemptEpoch: number;
             generationSeq: number;
-            eventKind: ChatGenerationEventKind;
+            eventKind: import("@rudderhq/shared").ChatGenerationEventKind;
             bodyOffset: number | null;
             bodyLength: number | null;
             queueItemId: string | null;
@@ -308,7 +294,7 @@ export declare function chatGenerationProtocolService(db: Db): {
             generationId: string;
             attemptEpoch: number;
             generationSeq: number;
-            eventKind: ChatGenerationEventKind;
+            eventKind: import("@rudderhq/shared").ChatGenerationEventKind;
             bodyOffset: number | null;
             bodyLength: number | null;
             queueItemId: string | null;
@@ -324,6 +310,7 @@ export declare function chatGenerationProtocolService(db: Db): {
         expectedGenerationId: string;
         expectedAttemptEpoch: number;
         expectedControlVersion: number;
+        admissionControlVersion?: number;
         requestedRenderSeq: number;
         requestedBodyHash: string;
         now?: Date;
@@ -332,7 +319,7 @@ export declare function chatGenerationProtocolService(db: Db): {
         generation: GenerationRow;
         stopRequestedEvent: GenerationEventRow | null;
         outputCutoffEvent: GenerationEventRow | null;
-        outcome: "stop_applied" | "completion_committed";
+        outcome: "stop_applied" | "stop_in_progress" | "completion_committed" | "already_terminal";
         idempotent: boolean;
     }>;
     beginSteerFallbackCutoff: (input: {
@@ -456,7 +443,7 @@ export declare function chatGenerationProtocolService(db: Db): {
             generationId: string;
             attemptEpoch: number;
             generationSeq: number;
-            eventKind: ChatGenerationEventKind;
+            eventKind: import("@rudderhq/shared").ChatGenerationEventKind;
             bodyOffset: number | null;
             bodyLength: number | null;
             queueItemId: string | null;
@@ -620,7 +607,7 @@ export declare function chatGenerationProtocolService(db: Db): {
             generationId: string;
             attemptEpoch: number;
             generationSeq: number;
-            eventKind: ChatGenerationEventKind;
+            eventKind: import("@rudderhq/shared").ChatGenerationEventKind;
             bodyOffset: number | null;
             bodyLength: number | null;
             queueItemId: string | null;
@@ -670,5 +657,4 @@ export declare function chatGenerationProtocolService(db: Db): {
         outbox: TerminalOutboxRow;
     }[]>;
 };
-export {};
 //# sourceMappingURL=chat-generation-protocol.d.ts.map

package/dist/services/chat-generation-protocol.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-generation-protocol.d.ts","sourceRoot":"","sources":["../../src/services/chat-generation-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,EACf,4BAA4B,EAE5B,kBAAkB,EACnB,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EACV,sBAAsB,EACtB,uBAAuB,EACvB,oBAAoB,EACpB,yBAAyB,EAC1B,MAAM,kBAAkB,CAAC;AAO1B,KAAK,aAAa,GAAG,OAAO,eAAe,CAAC,YAAY,CAAC;AACzD,KAAK,kBAAkB,GAAG,OAAO,oBAAoB,CAAC,YAAY,CAAC;AACnE,KAAK,gBAAgB,GAAG,OAAO,kBAAkB,CAAC,YAAY,CAAC;AAC/D,KAAK,iBAAiB,GAAG,OAAO,4BAA4B,CAAC,YAAY,CAAC;AAE1E,MAAM,MAAM,iCAAiC,GAAG,UAAU,CACxD,UAAU,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CACjC,CAAC,CAAC,CAAC,CAAC;AAgCL,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAE3D;AAoED,KAAK,iBAAiB,GAAG;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,uBAAuB,CAAC;IACnC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;CACzB,CAAC;AAwFF,KAAK,2BAA2B,GAAG;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,yBAAyB,EAAE,CAAC;IACxC,kBAAkB,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtB,CAAC;AAiFF,MAAM,MAAM,2BAA2B,GAAG,iBAAiB,CAAC;AAE5D,wBAAgB,6BAA6B,CAAC,EAAE,EAAE,EAAE;wCACD;QAC/C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;KACtB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;mCAkC2C,iBAAiB,GAAG;QAC9D,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,SAAS,EAAE,SAAS,GAAG,SAAS,GAAG,YAAY,CAAC;QAChD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;0CAgDkD,IAAI,CACrD,iBAAiB,EACjB,cAAc,GAAG,oBAAoB,CACtC,GAAG;QACF,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC1B,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,yBAAyB,EAAE,CAAC;QACxC,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAChC,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;KACrB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;wCA+EgD;QAC/C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;KACtB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BA3SyB,MAAM;sBAAY,MAAM;;;;oCAoUL;QAC3C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,aAAa,EAAE,MAAM,CAAC;QACtB,gBAAgB,EAAE,MAAM,CAAC;QACzB,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6BAuDqC;QACpC,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,oBAAoB,EAAE,MAAM,CAAC;QAC7B,sBAAsB,EAAE,MAAM,CAAC;QAC/B,kBAAkB,EAAE,MAAM,CAAC;QAC3B,iBAAiB,EAAE,MAAM,CAAC;QAC1B,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ,KAAG,OAAO,CAAC;QACV,MAAM,EAAE,gBAAgB,CAAC;QACzB,UAAU,EAAE,aAAa,CAAC;QAC1B,kBAAkB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC9C,iBAAiB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC7C,OAAO,EAAE,cAAc,GAAG,sBAAsB,CAAC;QACjD,UAAU,EAAE,OAAO,CAAC;KACrB,CAAC;sCAwM6C;QAC7C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,eAAe,EAAE,MAAM,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ,KAAG,OAAO,CAAC;QACV,MAAM,EAAE,gBAAgB,CAAC;QACzB,UAAU,EAAE,aAAa,CAAC;QAC1B,IAAI,EAAE,OAAO,kBAAkB,CAAC,YAAY,CAAC;QAC7C,iBAAiB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC7C,iBAAiB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC7C,OAAO,EAAE,gBAAgB,GAAG,sBAAsB,CAAC;QACnD,UAAU,EAAE,OAAO,CAAC;KACrB,CAAC;mCAiU0C;QAC1C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,WAAW,EAAE,OAAO,CAAC,oBAAoB,EACvC,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,SAAS,GAAG,wBAAwB,CAAC,CAAC;QAC7E,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAChC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClC,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCA4H6C;QAC5C,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ,KAAG,OAAO,CAAC,2BAA2B,GAAG,IAAI,CAAC;8CAmDO;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;wCAmCvB;QAC/C,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,OAAO,CAAC,oBAAoB,EACxC,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,SAAS,GAAG,wBAAwB,GAAG,cAAc,CAAC,CAAC;QAC9F,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,kBAAkB,CAAC,EAAE,sBAAsB,CAAC;QAC5C,GAAG,CAAC,EAAE,IAAI,CAAC;QACX,OAAO,CAAC,EAAE,CACR,EAAE,EAAE,iCAAiC,EACrC,KAAK,EAAE,2BAA2B,KAC/B,OAAO,CAAC,IAAI,CAAC,CAAC;KACpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCA4R6C;QAC5C,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,IAAI,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ;;;;;;;;;;;;;;;;;;;;;;;uCA4N+C;QAC9C,GAAG,CAAC,EAAE,IAAI,CAAC;QACX,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,oBAAoB,CAAC,EAAE,OAAO,CAAC;KAChC;oBAmBe,aAAa;eAClB,kBAAkB;gBACjB,iBAAiB;;EAyF9B"}
+{"version":3,"file":"chat-generation-protocol.d.ts","sourceRoot":"","sources":["../../src/services/chat-generation-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,EACf,4BAA4B,EAE5B,kBAAkB,EACnB,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EACV,sBAAsB,EACtB,oBAAoB,EACpB,yBAAyB,EAC1B,MAAM,kBAAkB,CAAC;AAI1B,OAAO,EAOL,KAAK,iBAAiB,EACtB,KAAK,iCAAiC,EACvC,MAAM,uCAAuC,CAAC;AAI/C,OAAO,EAAE,sBAAsB,EAAE,MAAM,uCAAuC,CAAC;AAE/E,KAAK,aAAa,GAAG,OAAO,eAAe,CAAC,YAAY,CAAC;AACzD,KAAK,kBAAkB,GAAG,OAAO,oBAAoB,CAAC,YAAY,CAAC;AACnE,KAAK,gBAAgB,GAAG,OAAO,kBAAkB,CAAC,YAAY,CAAC;AAC/D,KAAK,iBAAiB,GAAG,OAAO,4BAA4B,CAAC,YAAY,CAAC;AAmH1E,KAAK,2BAA2B,GAAG;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,yBAAyB,EAAE,CAAC;IACxC,kBAAkB,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtB,CAAC;AAiFF,MAAM,MAAM,2BAA2B,GAAG,iBAAiB,CAAC;AAE5D,wBAAgB,6BAA6B,CAAC,EAAE,EAAE,EAAE;wCACD;QAC/C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;KACtB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;mCAkC2C,iBAAiB,GAAG;QAC9D,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,SAAS,EAAE,SAAS,GAAG,SAAS,GAAG,YAAY,CAAC;QAChD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;0CAgDkD,IAAI,CACrD,iBAAiB,EACjB,cAAc,GAAG,oBAAoB,CACtC,GAAG;QACF,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC1B,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,yBAAyB,EAAE,CAAC;QACxC,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAChC,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;KACrB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;wCA+EgD;QAC/C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;KACtB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BA3SyB,MAAM;sBAAY,MAAM;;;;oCAoUL;QAC3C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,aAAa,EAAE,MAAM,CAAC;QACtB,gBAAgB,EAAE,MAAM,CAAC;QACzB,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6BAuDqC;QACpC,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,oBAAoB,EAAE,MAAM,CAAC;QAC7B,sBAAsB,EAAE,MAAM,CAAC;QAC/B,uBAAuB,CAAC,EAAE,MAAM,CAAC;QACjC,kBAAkB,EAAE,MAAM,CAAC;QAC3B,iBAAiB,EAAE,MAAM,CAAC;QAC1B,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ,KAAG,OAAO,CAAC;QACV,MAAM,EAAE,gBAAgB,CAAC;QACzB,UAAU,EAAE,aAAa,CAAC;QAC1B,kBAAkB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC9C,iBAAiB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC7C,OAAO,EAAE,cAAc,GAAG,kBAAkB,GAAG,sBAAsB,GAAG,kBAAkB,CAAC;QAC3F,UAAU,EAAE,OAAO,CAAC;KACrB,CAAC;sCAoR6C;QAC7C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,eAAe,EAAE,MAAM,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ,KAAG,OAAO,CAAC;QACV,MAAM,EAAE,gBAAgB,CAAC;QACzB,UAAU,EAAE,aAAa,CAAC;QAC1B,IAAI,EAAE,OAAO,kBAAkB,CAAC,YAAY,CAAC;QAC7C,iBAAiB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC7C,iBAAiB,EAAE,kBAAkB,GAAG,IAAI,CAAC;QAC7C,OAAO,EAAE,gBAAgB,GAAG,sBAAsB,CAAC;QACnD,UAAU,EAAE,OAAO,CAAC;KACrB,CAAC;mCAiU0C;QAC1C,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACnC,WAAW,EAAE,OAAO,CAAC,oBAAoB,EACvC,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,SAAS,GAAG,wBAAwB,CAAC,CAAC;QAC7E,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAChC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClC,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCA4H6C;QAC5C,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ,KAAG,OAAO,CAAC,2BAA2B,GAAG,IAAI,CAAC;8CAmDO;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;wCAmCvB;QAC/C,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,OAAO,CAAC,oBAAoB,EACxC,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,SAAS,GAAG,wBAAwB,GAAG,cAAc,CAAC,CAAC;QAC9F,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,kBAAkB,CAAC,EAAE,sBAAsB,CAAC;QAC5C,GAAG,CAAC,EAAE,IAAI,CAAC;QACX,OAAO,CAAC,EAAE,CACR,EAAE,EAAE,iCAAiC,EACrC,KAAK,EAAE,2BAA2B,KAC/B,OAAO,CAAC,IAAI,CAAC,CAAC;KACpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCA4R6C;QAC5C,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,IAAI,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,GAAG,CAAC,EAAE,IAAI,CAAC;KACZ;;;;;;;;;;;;;;;;;;;;;;;uCA4N+C;QAC9C,GAAG,CAAC,EAAE,IAAI,CAAC;QACX,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,oBAAoB,CAAC,EAAE,OAAO,CAAC;KAChC;oBAmBe,aAAa;eAClB,kBAAkB;gBACjB,iBAAiB;;EAyF9B"}

package/dist/services/chat-generation-protocol.helpers.d.ts

@@ -0,0 +1,36 @@
+import type { Db } from "@rudderhq/db";
+import { chatGenerations } from "@rudderhq/db";
+import type { ChatGenerationEventKind } from "@rudderhq/shared";
+type GenerationRow = typeof chatGenerations.$inferSelect;
+export type ChatGenerationProtocolTransaction = Parameters<Parameters<Db["transaction"]>[0]>[0];
+export type AppendEventFields = {
+    orgId: string;
+    generationId: string;
+    attemptEpoch: number;
+    eventKind: ChatGenerationEventKind;
+    payload?: Record<string, unknown>;
+    bodyOffset?: number | null;
+    bodyLength?: number | null;
+    assistantMessageId?: string | null;
+    runId?: string | null;
+    controlActionId?: string | null;
+    queueItemId?: string | null;
+    emittedAt?: Date | null;
+};
+export declare const EMPTY_BODY_SHA256: string;
+export declare const SHA256_PATTERN: RegExp;
+export declare function hashChatGenerationBody(body: string): string;
+export declare function normalizeBodyHash(value: string): string;
+export declare function assertGenerationFence(generation: GenerationRow, input: {
+    conversationId?: string;
+    expectedAttemptEpoch: number;
+    expectedOwnerToken?: string | null;
+}): void;
+export declare function lockGeneration(tx: ChatGenerationProtocolTransaction, input: {
+    orgId: string;
+    generationId: string;
+    conversationId?: string;
+}): Promise<GenerationRow>;
+export declare function nextGenerationSeq(tx: ChatGenerationProtocolTransaction, generationId: string): Promise<number>;
+export {};
+//# sourceMappingURL=chat-generation-protocol.helpers.d.ts.map

package/dist/services/chat-generation-protocol.helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-generation-protocol.helpers.d.ts","sourceRoot":"","sources":["../../src/services/chat-generation-protocol.helpers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAwB,eAAe,EAAE,MAAM,cAAc,CAAC;AACrE,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAKhE,KAAK,aAAa,GAAG,OAAO,eAAe,CAAC,YAAY,CAAC;AAEzD,MAAM,MAAM,iCAAiC,GAAG,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAEhG,MAAM,MAAM,iBAAiB,GAAG;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,uBAAuB,CAAC;IACnC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;CACzB,CAAC;AAEF,eAAO,MAAM,iBAAiB,QAAgD,CAAC;AAC/E,eAAO,MAAM,cAAc,QAAoB,CAAC;AAEhD,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAE3D;AAED,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAKvD;AAED,wBAAgB,qBAAqB,CACnC,UAAU,EAAE,aAAa,EACzB,KAAK,EAAE;IACL,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACpC,QAWF;AAED,wBAAsB,cAAc,CAClC,EAAE,EAAE,iCAAiC,EACrC,KAAK,EAAE;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,cAAc,CAAC,EAAE,MAAM,CAAA;CAAE,GACtE,OAAO,CAAC,aAAa,CAAC,CAkBxB;AAED,wBAAsB,iBAAiB,CACrC,EAAE,EAAE,iCAAiC,EACrC,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,MAAM,CAAC,CAMjB"}

package/dist/services/chat-generation-protocol.helpers.js

@@ -0,0 +1,53 @@
+import { chatGenerationEvents, chatGenerations } from "@rudderhq/db";
+import { and, eq, sql } from "drizzle-orm";
+import { createHash } from "node:crypto";
+import { conflict, notFound, unprocessable } from "../errors.js";
+export const EMPTY_BODY_SHA256 = createHash("sha256").update("").digest("hex");
+export const SHA256_PATTERN = /^[a-f0-9]{64}$/i;
+export function hashChatGenerationBody(body) {
+    return createHash("sha256").update(body).digest("hex");
+}
+export function normalizeBodyHash(value) {
+    if (!SHA256_PATTERN.test(value)) {
+        throw unprocessable("Chat generation body hash must be a SHA-256 hex digest");
+    }
+    return value.toLowerCase();
+}
+export function assertGenerationFence(generation, input) {
+    if (input.conversationId && generation.conversationId !== input.conversationId) {
+        throw notFound("Chat generation not found");
+    }
+    if (generation.attemptEpoch !== input.expectedAttemptEpoch) {
+        throw conflict("Chat generation runtime attempt changed");
+    }
+    if (input.expectedOwnerToken !== undefined && generation.controlOwnerToken !== input.expectedOwnerToken) {
+        throw conflict("Chat generation control owner changed");
+    }
+}
+export async function lockGeneration(tx, input) {
+    await tx.execute(sql `
+    select ${chatGenerations.id}
+    from ${chatGenerations}
+    where ${chatGenerations.id} = ${input.generationId}
+      and ${chatGenerations.orgId} = ${input.orgId}
+    for update
+  `);
+    const generation = await tx
+        .select()
+        .from(chatGenerations)
+        .where(and(eq(chatGenerations.id, input.generationId), eq(chatGenerations.orgId, input.orgId)))
+        .limit(1)
+        .then((rows) => rows[0] ?? null);
+    if (!generation || (input.conversationId && generation.conversationId !== input.conversationId)) {
+        throw notFound("Chat generation not found");
+    }
+    return generation;
+}
+export async function nextGenerationSeq(tx, generationId) {
+    return tx
+        .select({ value: sql `coalesce(max(${chatGenerationEvents.generationSeq}), 0) + 1` })
+        .from(chatGenerationEvents)
+        .where(eq(chatGenerationEvents.generationId, generationId))
+        .then((rows) => Number(rows[0]?.value ?? 1));
+}
+//# sourceMappingURL=chat-generation-protocol.helpers.js.map

package/dist/services/chat-generation-protocol.helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-generation-protocol.helpers.js","sourceRoot":"","sources":["../../src/services/chat-generation-protocol.helpers.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAErE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAqBjE,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC/E,MAAM,CAAC,MAAM,cAAc,GAAG,iBAAiB,CAAC;AAEhD,MAAM,UAAU,sBAAsB,CAAC,IAAY;IACjD,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAAa;IAC7C,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAChC,MAAM,aAAa,CAAC,wDAAwD,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,UAAyB,EACzB,KAIC;IAED,IAAI,KAAK,CAAC,cAAc,IAAI,UAAU,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,EAAE,CAAC;QAC/E,MAAM,QAAQ,CAAC,2BAA2B,CAAC,CAAC;IAC9C,CAAC;IACD,IAAI,UAAU,CAAC,YAAY,KAAK,KAAK,CAAC,oBAAoB,EAAE,CAAC;QAC3D,MAAM,QAAQ,CAAC,yCAAyC,CAAC,CAAC;IAC5D,CAAC;IACD,IAAI,KAAK,CAAC,kBAAkB,KAAK,SAAS,IAAI,UAAU,CAAC,iBAAiB,KAAK,KAAK,CAAC,kBAAkB,EAAE,CAAC;QACxG,MAAM,QAAQ,CAAC,uCAAuC,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,EAAqC,EACrC,KAAuE;IAEvE,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,CAAA;aACT,eAAe,CAAC,EAAE;WACpB,eAAe;YACd,eAAe,CAAC,EAAE,MAAM,KAAK,CAAC,YAAY;YAC1C,eAAe,CAAC,KAAK,MAAM,KAAK,CAAC,KAAK;;GAE/C,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,MAAM,EAAE;SACxB,MAAM,EAAE;SACR,IAAI,CAAC,eAAe,CAAC;SACrB,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,KAAK,CAAC,YAAY,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;SAC9F,KAAK,CAAC,CAAC,CAAC;SACR,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;IACnC,IAAI,CAAC,UAAU,IAAI,CAAC,KAAK,CAAC,cAAc,IAAI,UAAU,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC;QAChG,MAAM,QAAQ,CAAC,2BAA2B,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,EAAqC,EACrC,YAAoB;IAEpB,OAAO,EAAE;SACN,MAAM,CAAC,EAAE,KAAK,EAAE,GAAG,CAAQ,gBAAgB,oBAAoB,CAAC,aAAa,WAAW,EAAE,CAAC;SAC3F,IAAI,CAAC,oBAAoB,CAAC;SAC1B,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;SAC1D,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC;AACjD,CAAC"}

package/dist/services/chat-generation-protocol.js

@@ -1,9 +1,11 @@
 import { chatControlActions, chatGenerationEvents, chatGenerations, chatGenerationTerminalOutbox, chatMessages, chatQueuedMessages, } from "@rudderhq/db";
 import { and, asc, desc, eq, inArray, isNotNull, isNull, lte, or, sql } from "drizzle-orm";
-import { createHash, randomUUID } from "node:crypto";
+import { randomUUID } from "node:crypto";
 import { conflict, notFound, unprocessable } from "../errors.js";
+import { assertGenerationFence, EMPTY_BODY_SHA256, lockGeneration, nextGenerationSeq, normalizeBodyHash, SHA256_PATTERN, } from "./chat-generation-protocol.helpers.js";
 import { withPersistedTranscript } from "./chats.helpers.js";
 import { normalizeLocalLibraryPathMarkdown } from "./library-path-markdown.js";
+export { hashChatGenerationBody } from "./chat-generation-protocol.helpers.js";
 const OUTPUT_ADMITTING_GENERATION_STATUSES = [
     "starting",
     "active",
@@ -27,55 +29,6 @@ const PROJECTABLE_TERMINAL_STATUSES = [
     "interrupted_unverified",
     "control_lost",
 ];
-const EMPTY_BODY_SHA256 = createHash("sha256").update("").digest("hex");
-const SHA256_PATTERN = /^[a-f0-9]{64}$/i;
-export function hashChatGenerationBody(body) {
-    return createHash("sha256").update(body).digest("hex");
-}
-function normalizeBodyHash(value) {
-    if (!SHA256_PATTERN.test(value)) {
-        throw unprocessable("Chat generation body hash must be a SHA-256 hex digest");
-    }
-    return value.toLowerCase();
-}
-function assertGenerationFence(generation, input) {
-    if (input.conversationId && generation.conversationId !== input.conversationId) {
-        throw notFound("Chat generation not found");
-    }
-    if (generation.attemptEpoch !== input.expectedAttemptEpoch) {
-        throw conflict("Chat generation runtime attempt changed");
-    }
-    if (input.expectedOwnerToken !== undefined
-        && generation.controlOwnerToken !== input.expectedOwnerToken) {
-        throw conflict("Chat generation control owner changed");
-    }
-}
-async function lockGeneration(tx, input) {
-    await tx.execute(sql `
-    select ${chatGenerations.id}
-    from ${chatGenerations}
-    where ${chatGenerations.id} = ${input.generationId}
-      and ${chatGenerations.orgId} = ${input.orgId}
-    for update
-  `);
-    const generation = await tx
-        .select()
-        .from(chatGenerations)
-        .where(and(eq(chatGenerations.id, input.generationId), eq(chatGenerations.orgId, input.orgId)))
-        .limit(1)
-        .then((rows) => rows[0] ?? null);
-    if (!generation || (input.conversationId && generation.conversationId !== input.conversationId)) {
-        throw notFound("Chat generation not found");
-    }
-    return generation;
-}
-async function nextGenerationSeq(tx, generationId) {
-    return tx
-        .select({ value: sql `coalesce(max(${chatGenerationEvents.generationSeq}), 0) + 1` })
-        .from(chatGenerationEvents)
-        .where(eq(chatGenerationEvents.generationId, generationId))
-        .then((rows) => Number(rows[0]?.value ?? 1));
-}
 async function appendEventLocked(tx, input) {
     const generationSeq = await nextGenerationSeq(tx, input.generationId);
     const [event] = await tx
@@ -422,6 +375,7 @@ export function chatGenerationProtocolService(db) {
     }
     async function beginStopAction(input) {
         const requestedBodyHash = normalizeBodyHash(input.requestedBodyHash);
+        const admissionControlVersion = input.admissionControlVersion ?? input.expectedControlVersion;
         return db.transaction(async (tx) => {
             const generation = await lockGeneration(tx, {
                 orgId: input.orgId,
@@ -439,6 +393,7 @@ export function chatGenerationProtocolService(db) {
                     || existingAction.actionKind !== "stop"
                     || existingAction.expectedGenerationId !== input.expectedGenerationId
                     || existingAction.expectedAttemptEpoch !== input.expectedAttemptEpoch
+                    || existingAction.expectedControlVersion !== input.expectedControlVersion
                     || existingAction.requestedRenderSeq !== input.requestedRenderSeq
                     || existingAction.requestedBodyHash !== requestedBodyHash) {
                     throw conflict("Control action id was already used for a different Stop request");
@@ -450,7 +405,11 @@ export function chatGenerationProtocolService(db) {
                     outputCutoffEvent: null,
                     outcome: existingAction.lastError === "generation_result_already_committed"
                         ? "completion_committed"
-                        : "stop_applied",
+                        : existingAction.lastError === "generation_runtime_already_terminal"
+                            ? "already_terminal"
+                            : existingAction.lastError === "generation_stop_already_in_progress"
+                                ? "stop_in_progress"
+                                : "stop_applied",
                     idempotent: true,
                 };
             }
@@ -498,12 +457,84 @@ export function chatGenerationProtocolService(db) {
                 };
             }
             if (generation.runtimeTerminalAt) {
-                throw conflict("Chat generation runtime is already terminal");
+                const now = input.now ?? new Date();
+                const generationWasStopped = generation.status === "stopped";
+                const [action] = await tx
+                    .insert(chatControlActions)
+                    .values({
+                    id: input.controlActionId,
+                    orgId: input.orgId,
+                    expectedGenerationId: generation.id,
+                    expectedAttemptEpoch: input.expectedAttemptEpoch,
+                    expectedControlVersion: input.expectedControlVersion,
+                    actionKind: "stop",
+                    localDisposition: generationWasStopped ? "stopped" : "cancelled",
+                    providerDisposition: "not_sent",
+                    controlOwnerToken: generation.controlOwnerToken,
+                    requestedRenderSeq: input.requestedRenderSeq,
+                    requestedBodyHash,
+                    acceptedThroughSeq: generation.acceptedThroughSeq ?? input.requestedRenderSeq,
+                    frozenBodyHash: generation.frozenBodyHash ?? requestedBodyHash,
+                    lastError: "generation_runtime_already_terminal",
+                    requestedAt: now,
+                    resolvedAt: now,
+                    createdAt: now,
+                    updatedAt: now,
+                })
+                    .returning();
+                if (!action)
+                    throw new Error("Failed to persist already-terminal Stop outcome");
+                return {
+                    action,
+                    generation,
+                    stopRequestedEvent: null,
+                    outputCutoffEvent: null,
+                    outcome: "already_terminal",
+                    idempotent: false,
+                };
+            }
+            if (generation.status === "stop_requested" || generation.status === "stopping") {
+                if (generation.controlVersion !== admissionControlVersion) {
+                    throw conflict("Chat generation control version changed");
+                }
+                const now = input.now ?? new Date();
+                const [action] = await tx
+                    .insert(chatControlActions)
+                    .values({
+                    id: input.controlActionId,
+                    orgId: input.orgId,
+                    expectedGenerationId: generation.id,
+                    expectedAttemptEpoch: input.expectedAttemptEpoch,
+                    expectedControlVersion: input.expectedControlVersion,
+                    actionKind: "stop",
+                    localDisposition: "stopping",
+                    providerDisposition: "not_sent",
+                    controlOwnerToken: generation.controlOwnerToken,
+                    requestedRenderSeq: input.requestedRenderSeq,
+                    requestedBodyHash,
+                    acceptedThroughSeq: generation.acceptedThroughSeq ?? input.requestedRenderSeq,
+                    frozenBodyHash: generation.frozenBodyHash ?? requestedBodyHash,
+                    lastError: "generation_stop_already_in_progress",
+                    requestedAt: now,
+                    createdAt: now,
+                    updatedAt: now,
+                })
+                    .returning();
+                if (!action)
+                    throw new Error("Failed to persist in-progress Stop outcome");
+                return {
+                    action,
+                    generation,
+                    stopRequestedEvent: null,
+                    outputCutoffEvent: null,
+                    outcome: "stop_in_progress",
+                    idempotent: false,
+                };
             }
             if (!CONTROL_ACTIVE_GENERATION_STATUSES.includes(generation.status)) {
                 throw conflict("Chat generation is no longer stoppable");
             }
-            if (generation.controlVersion !== input.expectedControlVersion) {
+            if (generation.controlVersion !== admissionControlVersion) {
                 throw conflict("Chat generation control version changed");
             }
             const durableCheckpointIsNewer = generation.lastClientCheckpointSeq !== null
@@ -534,7 +565,7 @@ export function chatGenerationProtocolService(db) {
                 throw conflict("Stop could not establish a durable rendered checkpoint");
             }
             const now = input.now ?? new Date();
-            const appliedControlVersion = generation.controlVersion + 1;
+            const appliedControlVersion = admissionControlVersion + 1;
             const [action] = await tx
                 .insert(chatControlActions)
                 .values({
@@ -599,7 +630,7 @@ export function chatGenerationProtocolService(db) {
                 stopRequestedAt: now,
                 updatedAt: now,
             })
-                .where(and(eq(chatGenerations.id, generation.id), eq(chatGenerations.attemptEpoch, input.expectedAttemptEpoch), eq(chatGenerations.controlVersion, input.expectedControlVersion)))
+                .where(and(eq(chatGenerations.id, generation.id), eq(chatGenerations.attemptEpoch, input.expectedAttemptEpoch), eq(chatGenerations.controlVersion, admissionControlVersion)))
                 .returning();
             if (!updatedGeneration)
                 throw conflict("Chat generation changed while applying Stop");