@superblocksteam/vite-plugin-file-sync 1.99.99-next.2 → 2.0.0-SNAPSHOT.84af65cd215b6ca1

package/dist/ai-service/state-machine/helpers/peer.js

@@ -1,7 +1,345 @@
-export const sendUserMessageChannel = (clark) => (payload) => {
-    return void clark.context.peer?.call.aiPushMessage(payload);
+import { getLogger } from "../../../util/logger.js";
+import { OperationQueue } from "../../../util/operation-queue.js";
+import { sanitizeAiProviderResponse } from "../../chat/utils.js";
+import { emitLiveEditEvent } from "../../live-edit-events/bridge.js";
+import { catchBackgroundPeerDisconnect, isTransportCloseError, } from "./stable-peer.js";
+const PEER_PUSH_TIMEOUT_MS = 10_000;
+class PeerPushTimeoutError extends Error {
+}
+function withPeerPushTimeout(promise) {
+    return new Promise((resolve, reject) => {
+        const timer = setTimeout(() => reject(new PeerPushTimeoutError("peer push timed out")), PEER_PUSH_TIMEOUT_MS);
+        // Wrap with Promise.resolve: some callers (and test doubles) hand back a
+        // non-promise value synchronously; calling `.then` directly on that would
+        // throw instead of settling.
+        Promise.resolve(promise).then((v) => {
+            clearTimeout(timer);
+            resolve(v);
+        }, (e) => {
+            clearTimeout(timer);
+            reject(e);
+        });
+    });
+}
+/**
+ * Bounds a fire-and-forget peer push to {@link PEER_PUSH_TIMEOUT_MS} and
+ * classifies the outcome. Returns `true` when the push acked, `false` when the
+ * peer was disconnected or half-open (transport close, or the bound fired
+ * because the peer never acked). Re-throws genuine RPC-layer failures.
+ *
+ * Every push that shares the FSM's single `peerChannelQueue` MUST go through a
+ * bound like this: `OperationQueue` chains each op on the previous one, so an
+ * unbounded push wedged on a half-open peer would block every later queued op
+ * from starting — including a plan-delta whose own 10s bound then never arms,
+ * re-opening the 180s tool-timeout hang this fixes.
+ */
+async function deliverBoundedPush(promise, description) {
+    try {
+        await withPeerPushTimeout(promise);
+        return true;
+    }
+    catch (err) {
+        if (err instanceof PeerPushTimeoutError || isTransportCloseError(err)) {
+            getLogger().warn(`[ai-service] ${description} not delivered: peer disconnected/half-open`);
+            return false;
+        }
+        throw err;
+    }
+}
+const peerQueue = (clark) => {
+    const queue = clark.context.peerChannelQueue ?? new OperationQueue();
+    clark.updateContext({
+        peerChannelQueue: queue,
+    });
+    return (operation, delay = 0) => {
+        if (delay) {
+            return queue.enqueue(() => new Promise((resolve, reject) => {
+                setTimeout(() => {
+                    void operation().then(resolve, reject);
+                }, delay);
+            }));
+        }
+        else {
+            return queue.enqueue(operation);
+        }
+    };
 };
-export const sendUserCompletionChannel = (clark) => async (payload) => {
-    return void clark.context.peer?.call.aiSetCompleted(payload);
+const INTERACTIVE_PEER_PUSH_DROPPED = Symbol("INTERACTIVE_PEER_PUSH_DROPPED");
+async function pushInteractiveMessage(clark, payload) {
+    const peer = clark.context.peer;
+    if (!peer) {
+        getLogger().warn(`[ai-service] Cannot send interactive message: no peer. type=${payload.type}, state=${clark.state}`);
+        return INTERACTIVE_PEER_PUSH_DROPPED;
+    }
+    try {
+        // Bound the push like the plan-delta channel: a half-open peer at the
+        // first (interactive plan-card) push would otherwise await aiPushMessage
+        // indefinitely and ride the 180s tool timeout. A timeout is treated as a
+        // drop, so the caller's peerDisconnected handling kicks in.
+        await withPeerPushTimeout(peer.call.aiPushMessage(payload));
+    }
+    catch (err) {
+        if (err instanceof PeerPushTimeoutError || isTransportCloseError(err)) {
+            getLogger().info(`[ai-service] Interactive message push type=${payload.type} dropped because the editor peer disconnected`);
+            return INTERACTIVE_PEER_PUSH_DROPPED;
+        }
+        getLogger().warn(`[ai-service] Interactive message push type=${payload.type} failed`, err);
+        throw err;
+    }
+}
+export const sendInteractiveMessageChannel = (clark, chatSessionStore) => {
+    const queue = peerQueue(clark);
+    return async (payload, delay = 0, immediate = false, options = {}) => {
+        const messageId = payload.id ?? crypto.randomUUID();
+        const payloadWithId = {
+            ...payload,
+            id: messageId,
+            content: sanitizeAiProviderResponse(payload.content) ?? "",
+        };
+        const existing = clark.context.activeInteractiveMessage;
+        let recoveredFromStaleGuard = false;
+        if (existing) {
+            if (existing.id === messageId) {
+                return { success: true, alreadyActive: true };
+            }
+            if (existing.deliveredViaChannel) {
+                if (options.recoverStaleOnce?.isStale(existing)) {
+                    getLogger().warn(`[ai-service] sendInteractiveMessageChannel recovering from stale activeInteractiveMessage id=${existing.id} via recoverStaleOnce; clearing and delivering new payload`);
+                    clark.updateContext({ activeInteractiveMessage: undefined });
+                    recoveredFromStaleGuard = true;
+                }
+                else if (options.deferIfActive) {
+                    clark.updateContext({
+                        deferredInteractiveMessage: {
+                            payload: payloadWithId,
+                        },
+                    });
+                    return {
+                        success: false,
+                        deferred: true,
+                        message: "Interactive prompt deferred because another interactive message is still active.",
+                    };
+                }
+                else {
+                    getLogger().warn(`[ai-service] sendInteractiveMessageChannel displaced new payload; active prompt id=${existing.id} is still in flight (type=${existing.payload.type})`);
+                    return {
+                        success: false,
+                        displaced: true,
+                        message: "Cannot send interactive message: another prompt is currently awaiting user input",
+                    };
+                }
+            }
+            else {
+                getLogger().warn(`[ai-service] sendInteractiveMessageChannel recovering from stale activeInteractiveMessage id=${existing.id}; clearing and delivering new payload`);
+                clark.updateContext({ activeInteractiveMessage: undefined });
+                recoveredFromStaleGuard = true;
+            }
+        }
+        clark.updateContext({
+            activeInteractiveMessage: {
+                id: messageId,
+                payload: payloadWithId,
+                deliveredViaChannel: true,
+            },
+        });
+        try {
+            const deliver = async () => {
+                return pushInteractiveMessage(clark, payloadWithId);
+            };
+            const result = immediate ? await deliver() : await queue(deliver, delay);
+            if (result === INTERACTIVE_PEER_PUSH_DROPPED) {
+                clark.updateContext({
+                    activeInteractiveMessage: undefined,
+                });
+                return {
+                    success: false,
+                    peerDisconnected: true,
+                    message: "Interactive prompt could not be delivered because the editor peer disconnected.",
+                };
+            }
+            void chatSessionStore.recordAssistant(payloadWithId);
+            return recoveredFromStaleGuard
+                ? { success: true, recoverable: true }
+                : { success: true };
+        }
+        catch (error) {
+            clark.updateContext({
+                activeInteractiveMessage: undefined,
+            });
+            throw error;
+        }
+    };
+};
+export const flushDeferredInteractiveMessage = async (clark, chatSessionStore) => {
+    const deferred = clark.context.deferredInteractiveMessage;
+    if (!deferred || clark.context.activeInteractiveMessage) {
+        return false;
+    }
+    clark.updateContext({
+        deferredInteractiveMessage: undefined,
+    });
+    const sendInteractiveMessage = sendInteractiveMessageChannel(clark, chatSessionStore);
+    const result = await sendInteractiveMessage(deferred.payload);
+    return result.success;
+};
+// Frozen: this single instance is returned from every disconnect path, so a
+// caller must never mutate it (doing so would corrupt every other caller).
+const PEER_DISCONNECTED_RESULT = Object.freeze({
+    success: false,
+    peerDisconnected: true,
+    message: "Message could not be delivered because the editor peer disconnected.",
+});
+export const sendUserMessageChannel = (clark, chatSessionStore) => {
+    const queue = peerQueue(clark);
+    return async (payload, delay = 0, immediate = false, { persistBeforeDeliver = false } = {}) => {
+        // Generate ID upfront so both backend storage and frontend receive same ID
+        const messageId = payload.id ?? crypto.randomUUID();
+        const payloadWithId = {
+            ...payload,
+            id: messageId,
+            content: sanitizeAiProviderResponse(payload.content) ?? "",
+        };
+        const deliver = async () => {
+            if (persistBeforeDeliver) {
+                try {
+                    await chatSessionStore.recordAssistant(payloadWithId);
+                }
+                catch (err) {
+                    getLogger().warn(`[ai-service] recordAssistant failed; continuing with peer delivery. type=${payloadWithId.type}`, err);
+                }
+            }
+            else {
+                void chatSessionStore.recordAssistant(payloadWithId);
+            }
+            if (!clark.context.peer) {
+                getLogger().warn(`[ai-service] Cannot send message: no peer. type=${payloadWithId.type}, state=${clark.state}`);
+                return PEER_DISCONNECTED_RESULT;
+            }
+            try {
+                await withPeerPushTimeout(clark.context.peer.call.aiPushMessage(payloadWithId));
+                // A Gateway-driven turn has no editor on the far end of that push, so
+                // the same report goes to SABS for relay. Best-effort and non-blocking.
+                emitLiveEditEvent("chat_message", {
+                    messageId: payloadWithId.id,
+                    messageType: payloadWithId.type,
+                    role: payloadWithId.role,
+                    text: payloadWithId.content,
+                });
+                return { success: true };
+            }
+            catch (err) {
+                if (err instanceof PeerPushTimeoutError || isTransportCloseError(err)) {
+                    getLogger().warn(`[ai-service] message push type=${payloadWithId.type} not delivered: peer disconnected/half-open`);
+                    return PEER_DISCONNECTED_RESULT;
+                }
+                // Genuine RPC-layer failure (not a disconnect). Most call sites void
+                // this channel, so a bare rethrow would surface as a contextless
+                // unhandled rejection — log the payload type before propagating, the
+                // same way pushInteractiveMessage does.
+                getLogger().warn(`[ai-service] message push type=${payloadWithId.type} failed`, err);
+                throw err;
+            }
+        };
+        return immediate ? deliver() : queue(deliver, delay);
+    };
+};
+export const sendUserGenerationStateChannel = (clark) => {
+    return async (payload) => {
+        // Name the generation this state is about. The ai-service runs generations
+        // a caller never asked for — the v2->v3 migration turn, the new run an
+        // interrupt continues with — and each ends with its own terminal state. On
+        // a wire that does not say whose, a caller cannot tell its own turn ending
+        // from someone else's, which is what leaves consumers guessing with a
+        // timer. Only stamped when there is a prompt to name, so a state that
+        // belongs to no generation reads the same as it always did. (APPS-5161)
+        const attributed = payload.generationId === undefined && clark.context.promptId
+            ? { ...payload, generationId: clark.context.promptId }
+            : payload;
+        // Store generation state in context for replay on reconnect. The id is
+        // cached with it so a replayed state is as attributable as a live one.
+        // Cache `hasError` + `paywallReason` too — without them, a socket
+        // reconnect after the credit-paywall fires would replay an
+        // `aiSetGenerationState({state: CANCELLED})` with no paywall metadata,
+        // and the UI saga (`paywallReason && state === CANCELLED` ->
+        // openAiQuotaPaywallModal) would treat it as a plain cancel.
+        // (APPS-4080)
+        clark.updateContext({
+            lastGenerationState: {
+                state: attributed.state,
+                generationId: attributed.generationId,
+                hasSuggestions: attributed.hasSuggestions,
+                hasError: attributed.hasError,
+                paywallReason: attributed.paywallReason,
+            },
+        });
+        const peer = clark.context.peer;
+        if (!peer) {
+            getLogger().warn(`[ai-service] Cannot send generation state: no peer. state=${payload.state}`);
+            return;
+        }
+        // Reported before the push, not after, because the push is what ends the
+        // turn: StablePeer releases the binding as a terminal state goes out, so a
+        // correlator read on the far side of this await is already gone and the
+        // report is dropped as editor-driven. That loses exactly the event a
+        // Gateway caller is waiting for - the one saying the generation is over -
+        // and a caller whose socket has died has no other way to hear it.
+        //
+        // Safe in this order because the state is a fact about the generation
+        // rather than about its delivery, and forwarding never throws or blocks.
+        emitLiveEditEvent("generation_state", {
+            hasError: attributed.hasError ?? false,
+            hasSuggestions: attributed.hasSuggestions ?? false,
+            // SABS validates against lowercase snake_case state values.
+            state: String(attributed.state).toLowerCase(),
+        });
+        // Bounded: a half-open peer would otherwise hang this await indefinitely
+        // and, since callers void it, leak a run that never settles.
+        await deliverBoundedPush(peer.call.aiSetGenerationState(attributed), `generation state push state=${attributed.state}`);
+    };
+};
+/**
+ * Clears the pendingToolPermissionRequest from the FSM context and notifies the
+ * UI so it removes any stale permission prompt from the chat.
+ */
+export const clearPendingToolPermissionRequest = (clark, reason = "unspecified") => {
+    const pending = clark.context.pendingToolPermissionRequest;
+    if (!pending) {
+        getLogger().info(`[debug] clearPendingToolPermissionRequest skipped: no pending request (reason=${reason})`);
+        return;
+    }
+    getLogger().info(`[debug] clearPendingToolPermissionRequest: clearing pending request`, {
+        reason,
+        messageId: pending.messageId,
+        toolName: pending.toolName,
+    });
+    clark.updateContext({
+        pendingToolPermissionRequest: undefined,
+    });
+    if (!clark.context.peer) {
+        getLogger().warn(`[debug] Cannot notify UI of cleared tool permission: no peer (reason=${reason})`);
+        return;
+    }
+    getLogger().info(`[debug] Sending aiClearToolPermission to UI`, {
+        reason,
+        messageId: pending.messageId,
+        toolName: pending.toolName,
+    });
+    catchBackgroundPeerDisconnect(clark.context.peer.call.aiClearToolPermission(), "clear tool permission push");
+};
+export const sendChangeInfoChannel = (clark, chatSessionStore) => {
+    const queue = peerQueue(clark);
+    return async (payload, delay = 0) => {
+        return queue(async () => {
+            void chatSessionStore.recordAssistantChangeInfo(payload);
+            if (!clark.context.peer) {
+                getLogger().warn(`[ai-service] Cannot send change info: no peer`);
+                return;
+            }
+            // Bounded: this push shares the FSM's peerChannelQueue with plan-delta
+            // sends, so an unbounded await here on a half-open peer would wedge the
+            // queue and starve every later push.
+            await deliverBoundedPush(clark.context.peer.call.aiPushChangeInfo(payload), `change info push type=${payload.type}`);
+        }, delay);
+    };
 };
 //# sourceMappingURL=peer.js.map

package/dist/ai-service/state-machine/helpers/peer.js.map

@@ -1 +1 @@
-{"version":3,"file":"peer.js","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/peer.ts"],"names":[],"mappings":"AAGA,MAAM,CAAC,MAAM,sBAAsB,GACjC,CAAC,KAAY,EAAE,EAAE,CACjB,CAAC,OAA4D,EAAE,EAAE;IAC/D,OAAO,KAAK,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;AAC9D,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,yBAAyB,GACpC,CAAC,KAAY,EAAE,EAAE,CACjB,KAAK,EAAE,OAA6D,EAAE,EAAE;IACtE,OAAO,KAAK,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;AAC/D,CAAC,CAAC"}
+{"version":3,"file":"peer.js","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/peer.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAElE,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AAErE,OAAO,EACL,6BAA6B,EAC7B,qBAAqB,GACtB,MAAM,kBAAkB,CAAC;AAE1B,MAAM,oBAAoB,GAAG,MAAM,CAAC;AAEpC,MAAM,oBAAqB,SAAQ,KAAK;CAAG;AAE3C,SAAS,mBAAmB,CAAI,OAAmB;IACjD,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACxC,MAAM,KAAK,GAAG,UAAU,CACtB,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,oBAAoB,CAAC,qBAAqB,CAAC,CAAC,EAC7D,oBAAoB,CACrB,CAAC;QACF,yEAAyE;QACzE,0EAA0E;QAC1E,6BAA6B;QAC7B,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAC3B,CAAC,CAAC,EAAE,EAAE;YACJ,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO,CAAC,CAAC,CAAC,CAAC;QACb,CAAC,EACD,CAAC,CAAC,EAAE,EAAE;YACJ,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,CAAC,CAAC,CAAC;QACZ,CAAC,CACF,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;;;;;GAWG;AACH,KAAK,UAAU,kBAAkB,CAC/B,OAAyB,EACzB,WAAmB;IAEnB,IAAI,CAAC;QACH,MAAM,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,oBAAoB,IAAI,qBAAqB,CAAC,GAAG,CAAC,EAAE,CAAC;YACtE,SAAS,EAAE,CAAC,IAAI,CACd,gBAAgB,WAAW,6CAA6C,CACzE,CAAC;YACF,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,SAAS,GAAG,CAAC,KAAY,EAAE,EAAE;IACjC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,gBAAgB,IAAI,IAAI,cAAc,EAAE,CAAC;IACrE,KAAK,CAAC,aAAa,CAAC;QAClB,gBAAgB,EAAE,KAAK;KACxB,CAAC,CAAC;IAEH,OAAO,CAAI,SAA2B,EAAE,KAAK,GAAG,CAAC,EAAc,EAAE;QAC/D,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,KAAK,CAAC,OAAO,CAClB,GAAG,EAAE,CACH,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACjC,UAAU,CAAC,GAAG,EAAE;oBACd,KAAK,SAAS,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACzC,CAAC,EAAE,KAAK,CAAC,CAAC;YACZ,CAAC,CAAC,CACL,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,6BAA6B,GAAG,MAAM,CAAC,+BAA+B,CAAC,CAAC;AAE9E,KAAK,UAAU,sBAAsB,CACnC,KAAY,EACZ,OAAsB;IAEtB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;IAEhC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,SAAS,EAAE,CAAC,IAAI,CACd,+DAA+D,OAAO,CAAC,IAAI,WAAW,KAAK,CAAC,KAAK,EAAE,CACpG,CAAC;QACF,OAAO,6BAA6B,CAAC;IACvC,CAAC;IAED,IAAI,CAAC;QACH,sEAAsE;QACtE,yEAAyE;QACzE,yEAAyE;QACzE,4DAA4D;QAC5D,MAAM,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;IAC9D,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,oBAAoB,IAAI,qBAAqB,CAAC,GAAG,CAAC,EAAE,CAAC;YACtE,SAAS,EAAE,CAAC,IAAI,CACd,8CAA8C,OAAO,CAAC,IAAI,+CAA+C,CAC1G,CAAC;YACF,OAAO,6BAA6B,CAAC;QACvC,CAAC;QACD,SAAS,EAAE,CAAC,IAAI,CACd,8CAA8C,OAAO,CAAC,IAAI,SAAS,EACnE,GAAG,CACJ,CAAC;QACF,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AA8BD,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAC3C,KAAY,EACZ,gBAAkC,EAClC,EAAE;IACF,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,OAAO,KAAK,EACV,OAAsB,EACtB,KAAK,GAAG,CAAC,EACT,SAAS,GAAG,KAAK,EACjB,UAKI,EAAE,EACiC,EAAE;QACzC,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QACpD,MAAM,aAAa,GAAG;YACpB,GAAG,OAAO;YACV,EAAE,EAAE,SAAS;YACb,OAAO,EAAE,0BAA0B,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE;SAC3D,CAAC;QAEF,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;QACxD,IAAI,uBAAuB,GAAG,KAAK,CAAC;QAEpC,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,QAAQ,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC9B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;YAChD,CAAC;YAED,IAAI,QAAQ,CAAC,mBAAmB,EAAE,CAAC;gBACjC,IAAI,OAAO,CAAC,gBAAgB,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAChD,SAAS,EAAE,CAAC,IAAI,CACd,gGAAgG,QAAQ,CAAC,EAAE,4DAA4D,CACxK,CAAC;oBACF,KAAK,CAAC,aAAa,CAAC,EAAE,wBAAwB,EAAE,SAAS,EAAE,CAAC,CAAC;oBAC7D,uBAAuB,GAAG,IAAI,CAAC;gBACjC,CAAC;qBAAM,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;oBACjC,KAAK,CAAC,aAAa,CAAC;wBAClB,0BAA0B,EAAE;4BAC1B,OAAO,EAAE,aAAa;yBACvB;qBACF,CAAC,CAAC;oBACH,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,QAAQ,EAAE,IAAI;wBACd,OAAO,EACL,kFAAkF;qBACrF,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,SAAS,EAAE,CAAC,IAAI,CACd,sFAAsF,QAAQ,CAAC,EAAE,6BAA6B,QAAQ,CAAC,OAAO,CAAC,IAAI,GAAG,CACvJ,CAAC;oBACF,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,SAAS,EAAE,IAAI;wBACf,OAAO,EACL,kFAAkF;qBACrF,CAAC;gBACJ,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,SAAS,EAAE,CAAC,IAAI,CACd,gGAAgG,QAAQ,CAAC,EAAE,uCAAuC,CACnJ,CAAC;gBACF,KAAK,CAAC,aAAa,CAAC,EAAE,wBAAwB,EAAE,SAAS,EAAE,CAAC,CAAC;gBAC7D,uBAAuB,GAAG,IAAI,CAAC;YACjC,CAAC;QACH,CAAC;QAED,KAAK,CAAC,aAAa,CAAC;YAClB,wBAAwB,EAAE;gBACxB,EAAE,EAAE,SAAS;gBACb,OAAO,EAAE,aAAa;gBACtB,mBAAmB,EAAE,IAAI;aAC1B;SACF,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;gBACzB,OAAO,sBAAsB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;YACtD,CAAC,CAAC;YACF,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,MAAM,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACzE,IAAI,MAAM,KAAK,6BAA6B,EAAE,CAAC;gBAC7C,KAAK,CAAC,aAAa,CAAC;oBAClB,wBAAwB,EAAE,SAAS;iBACpC,CAAC,CAAC;gBACH,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,gBAAgB,EAAE,IAAI;oBACtB,OAAO,EACL,iFAAiF;iBACpF,CAAC;YACJ,CAAC;YACD,KAAK,gBAAgB,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;YACrD,OAAO,uBAAuB;gBAC5B,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE;gBACtC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,KAAK,CAAC,aAAa,CAAC;gBAClB,wBAAwB,EAAE,SAAS;aACpC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,+BAA+B,GAAG,KAAK,EAClD,KAAY,EACZ,gBAAkC,EAChB,EAAE;IACpB,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,0BAA0B,CAAC;IAC1D,IAAI,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,wBAAwB,EAAE,CAAC;QACxD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,KAAK,CAAC,aAAa,CAAC;QAClB,0BAA0B,EAAE,SAAS;KACtC,CAAC,CAAC;IACH,MAAM,sBAAsB,GAAG,6BAA6B,CAC1D,KAAK,EACL,gBAAgB,CACjB,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC9D,OAAO,MAAM,CAAC,OAAO,CAAC;AACxB,CAAC,CAAC;AAiCF,4EAA4E;AAC5E,2EAA2E;AAC3E,MAAM,wBAAwB,GAA0B,MAAM,CAAC,MAAM,CAAC;IACpE,OAAO,EAAE,KAAK;IACd,gBAAgB,EAAE,IAAI;IACtB,OAAO,EACL,sEAAsE;CACzE,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,KAAY,EACZ,gBAAkC,EAClC,EAAE;IACF,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,OAAO,KAAK,EACV,OAAsB,EACtB,KAAK,GAAG,CAAC,EACT,SAAS,GAAG,KAAK,EACjB,EAAE,oBAAoB,GAAG,KAAK,KAA6B,EAAE,EAC7B,EAAE;QAClC,2EAA2E;QAC3E,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QACpD,MAAM,aAAa,GAAG;YACpB,GAAG,OAAO;YACV,EAAE,EAAE,SAAS;YACb,OAAO,EAAE,0BAA0B,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE;SAC3D,CAAC;QAEF,MAAM,OAAO,GAAG,KAAK,IAAoC,EAAE;YACzD,IAAI,oBAAoB,EAAE,CAAC;gBACzB,IAAI,CAAC;oBACH,MAAM,gBAAgB,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;gBACxD,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,SAAS,EAAE,CAAC,IAAI,CACd,4EAA4E,aAAa,CAAC,IAAI,EAAE,EAChG,GAAG,CACJ,CAAC;gBACJ,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,KAAK,gBAAgB,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACxB,SAAS,EAAE,CAAC,IAAI,CACd,mDAAmD,aAAa,CAAC,IAAI,WAAW,KAAK,CAAC,KAAK,EAAE,CAC9F,CAAC;gBACF,OAAO,wBAAwB,CAAC;YAClC,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,mBAAmB,CACvB,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,CACrD,CAAC;gBACF,sEAAsE;gBACtE,wEAAwE;gBACxE,iBAAiB,CAAC,cAAc,EAAE;oBAChC,SAAS,EAAE,aAAa,CAAC,EAAE;oBAC3B,WAAW,EAAE,aAAa,CAAC,IAAI;oBAC/B,IAAI,EAAE,aAAa,CAAC,IAAI;oBACxB,IAAI,EAAE,aAAa,CAAC,OAAO;iBAC5B,CAAC,CAAC;gBACH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC3B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,GAAG,YAAY,oBAAoB,IAAI,qBAAqB,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtE,SAAS,EAAE,CAAC,IAAI,CACd,kCAAkC,aAAa,CAAC,IAAI,6CAA6C,CAClG,CAAC;oBACF,OAAO,wBAAwB,CAAC;gBAClC,CAAC;gBACD,qEAAqE;gBACrE,iEAAiE;gBACjE,qEAAqE;gBACrE,wCAAwC;gBACxC,SAAS,EAAE,CAAC,IAAI,CACd,kCAAkC,aAAa,CAAC,IAAI,SAAS,EAC7D,GAAG,CACJ,CAAC;gBACF,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACvD,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,KAAY,EAAE,EAAE;IAC7D,OAAO,KAAK,EACV,OAAmE,EACnE,EAAE;QACF,2EAA2E;QAC3E,uEAAuE;QACvE,2EAA2E;QAC3E,2EAA2E;QAC3E,sEAAsE;QACtE,sEAAsE;QACtE,wEAAwE;QACxE,MAAM,UAAU,GACd,OAAO,CAAC,YAAY,KAAK,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ;YAC1D,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE;YACtD,CAAC,CAAC,OAAO,CAAC;QAEd,uEAAuE;QACvE,uEAAuE;QACvE,kEAAkE;QAClE,2DAA2D;QAC3D,uEAAuE;QACvE,6DAA6D;QAC7D,6DAA6D;QAC7D,cAAc;QACd,KAAK,CAAC,aAAa,CAAC;YAClB,mBAAmB,EAAE;gBACnB,KAAK,EAAE,UAAU,CAAC,KAAK;gBACvB,YAAY,EAAE,UAAU,CAAC,YAAY;gBACrC,cAAc,EAAE,UAAU,CAAC,cAAc;gBACzC,QAAQ,EAAE,UAAU,CAAC,QAAQ;gBAC7B,aAAa,EAAE,UAAU,CAAC,aAAa;aACxC;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;QAChC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,SAAS,EAAE,CAAC,IAAI,CACd,6DAA6D,OAAO,CAAC,KAAK,EAAE,CAC7E,CAAC;YACF,OAAO;QACT,CAAC;QACD,yEAAyE;QACzE,2EAA2E;QAC3E,wEAAwE;QACxE,qEAAqE;QACrE,0EAA0E;QAC1E,kEAAkE;QAClE,EAAE;QACF,sEAAsE;QACtE,yEAAyE;QACzE,iBAAiB,CAAC,kBAAkB,EAAE;YACpC,QAAQ,EAAE,UAAU,CAAC,QAAQ,IAAI,KAAK;YACtC,cAAc,EAAE,UAAU,CAAC,cAAc,IAAI,KAAK;YAClD,4DAA4D;YAC5D,KAAK,EAAE,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE;SAC9C,CAAC,CAAC;QACH,yEAAyE;QACzE,6DAA6D;QAC7D,MAAM,kBAAkB,CACtB,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,EAC1C,+BAA+B,UAAU,CAAC,KAAK,EAAE,CAClD,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,iCAAiC,GAAG,CAC/C,KAAY,EACZ,MAAM,GAAG,aAAa,EACtB,EAAE;IACF,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,4BAA4B,CAAC;IAC3D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,SAAS,EAAE,CAAC,IAAI,CACd,iFAAiF,MAAM,GAAG,CAC3F,CAAC;QACF,OAAO;IACT,CAAC;IACD,SAAS,EAAE,CAAC,IAAI,CACd,qEAAqE,EACrE;QACE,MAAM;QACN,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,QAAQ,EAAE,OAAO,CAAC,QAAQ;KAC3B,CACF,CAAC;IACF,KAAK,CAAC,aAAa,CAAC;QAClB,4BAA4B,EAAE,SAAS;KACxC,CAAC,CAAC;IACH,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACxB,SAAS,EAAE,CAAC,IAAI,CACd,wEAAwE,MAAM,GAAG,CAClF,CAAC;QACF,OAAO;IACT,CAAC;IACD,SAAS,EAAE,CAAC,IAAI,CAAC,6CAA6C,EAAE;QAC9D,MAAM;QACN,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,QAAQ,EAAE,OAAO,CAAC,QAAQ;KAC3B,CAAC,CAAC;IACH,6BAA6B,CAC3B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAC/C,4BAA4B,CAC7B,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,KAAY,EACZ,gBAAkC,EAClC,EAAE;IACF,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,OAAO,KAAK,EAAE,OAAyB,EAAE,KAAK,GAAG,CAAC,EAAE,EAAE;QACpD,OAAO,KAAK,CAAC,KAAK,IAAI,EAAE;YACtB,KAAK,gBAAgB,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;YACzD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACxB,SAAS,EAAE,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;gBAClE,OAAO;YACT,CAAC;YACD,uEAAuE;YACvE,wEAAwE;YACxE,qCAAqC;YACrC,MAAM,kBAAkB,CACtB,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,EACjD,yBAAyB,OAAO,CAAC,IAAI,EAAE,CACxC,CAAC;QACJ,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC;AACJ,CAAC,CAAC"}

package/dist/ai-service/state-machine/helpers/policy-gate-usage-metrics.d.ts

@@ -0,0 +1,21 @@
+import type { AiPromptCostUsageEntry, AiQuotaPaywallReason } from "@superblocksteam/shared";
+import type { ChatSessionStore } from "../../chat/chat-session-store.js";
+import type { PolicyAgentErrorCode } from "../../policy-agent-metrics.js";
+import { type PromptFailure } from "./prompt-request-metrics.js";
+export declare function recordPolicyGateAttempt(params: {
+    coreModel?: string;
+}): void;
+export declare function policyGateFailureFromStream(errorCode: PolicyAgentErrorCode | undefined, quotaReason?: AiQuotaPaywallReason): PromptFailure;
+export declare function finalizePolicyGateUsageMetrics(params: {
+    aiAgentRunId: string;
+    chatSessionStore: ChatSessionStore;
+    coreModel?: string;
+    failure?: PromptFailure;
+    hadFallback?: boolean;
+    llmUsage: AiPromptCostUsageEntry[];
+    logger: {
+        warn: (...args: unknown[]) => void;
+    };
+    outcome: "failure" | "success";
+}): Promise<void>;
+//# sourceMappingURL=policy-gate-usage-metrics.d.ts.map

package/dist/ai-service/state-machine/helpers/policy-gate-usage-metrics.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"policy-gate-usage-metrics.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/policy-gate-usage-metrics.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,sBAAsB,EACtB,oBAAoB,EACrB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACzE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EAOL,KAAK,aAAa,EAEnB,MAAM,6BAA6B,CAAC;AAIrC,wBAAgB,uBAAuB,CAAC,MAAM,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAE5E;AAaD,wBAAgB,2BAA2B,CACzC,SAAS,EAAE,oBAAoB,GAAG,SAAS,EAC3C,WAAW,GAAE,oBAA8C,GAC1D,aAAa,CAyBf;AAED,wBAAsB,8BAA8B,CAAC,MAAM,EAAE;IAC3D,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,aAAa,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,QAAQ,EAAE,sBAAsB,EAAE,CAAC;IACnC,MAAM,EAAE;QAAE,IAAI,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAA;KAAE,CAAC;IAC/C,OAAO,EAAE,SAAS,GAAG,SAAS,CAAC;CAChC,GAAG,OAAO,CAAC,IAAI,CAAC,CAwChB"}

package/dist/ai-service/state-machine/helpers/policy-gate-usage-metrics.js

@@ -0,0 +1,71 @@
+import { promptFailureLlmProviderUnknown, promptFailurePaywall, promptFailurePlatform, recordPromptAttempt, recordPromptCredits, recordPromptOutcome, } from "./prompt-request-metrics.js";
+const POLICY_GATE_SURFACE = "policy_gate";
+export function recordPolicyGateAttempt(params) {
+    recordPromptAttempt({ ...params, surface: POLICY_GATE_SURFACE });
+}
+function policyGateMetricLabels(params) {
+    return {
+        coreModel: params.coreModel,
+        hadFallback: params.hadFallback ?? false,
+        surface: POLICY_GATE_SURFACE,
+    };
+}
+export function policyGateFailureFromStream(errorCode, quotaReason = "credit_limit_exceeded") {
+    switch (errorCode) {
+        case "llm_auth_error":
+            return { category: "auth_error", origin: "llm_provider" };
+        case "llm_overloaded":
+            return { category: "overloaded", origin: "llm_provider" };
+        case "llm_rate_limited":
+            return { category: "rate_limit", origin: "llm_provider" };
+        case "llm_refusal":
+        case "output_token_limit":
+            return { category: "no_output", origin: "llm_provider" };
+        case "llm_timeout":
+            return { category: "timeout", origin: "llm_provider" };
+        case "internal_error":
+            return promptFailureLlmProviderUnknown();
+        case "out_of_credits":
+            return promptFailurePaywall(quotaReason);
+        case "aborted":
+            return promptFailurePlatform("aborted");
+        case "incomplete_report":
+        case "invalid_report":
+        case "missing_report":
+        case undefined:
+            return promptFailurePlatform("handler_error");
+    }
+}
+export async function finalizePolicyGateUsageMetrics(params) {
+    const labels = policyGateMetricLabels(params);
+    recordPromptOutcome({
+        ...labels,
+        dedupeKey: params.aiAgentRunId,
+        failure: params.outcome === "failure" ? params.failure : undefined,
+        outcome: params.outcome,
+        // A policy-gate scan is not a user turn. Counting it toward (or clearing)
+        // the conversation streak pages a scan that failed twice, or lets a
+        // successful scan hide a stuck chat (ENG-5477 review).
+        streakEligible: false,
+    });
+    if (params.llmUsage.length === 0) {
+        return;
+    }
+    let totalCreditsUsed;
+    try {
+        totalCreditsUsed = await params.chatSessionStore.calculatePromptCost(params.aiAgentRunId, params.llmUsage);
+    }
+    catch (error) {
+        params.logger.warn("[billing:policy-gate-cost] Failed to calculate policy gate credits", {
+            aiAgentRunId: params.aiAgentRunId,
+            error: error instanceof Error ? error.message : String(error),
+        });
+        return;
+    }
+    recordPromptCredits({
+        ...labels,
+        credits: totalCreditsUsed,
+        dedupeKey: params.aiAgentRunId,
+    });
+}
+//# sourceMappingURL=policy-gate-usage-metrics.js.map

package/dist/ai-service/state-machine/helpers/policy-gate-usage-metrics.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policy-gate-usage-metrics.js","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/policy-gate-usage-metrics.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,+BAA+B,EAC/B,oBAAoB,EACpB,qBAAqB,EACrB,mBAAmB,EACnB,mBAAmB,EACnB,mBAAmB,GAGpB,MAAM,6BAA6B,CAAC;AAErC,MAAM,mBAAmB,GAAG,aAAsB,CAAC;AAEnD,MAAM,UAAU,uBAAuB,CAAC,MAA8B;IACpE,mBAAmB,CAAC,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,sBAAsB,CAAC,MAG/B;IACC,OAAO;QACL,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,KAAK;QACxC,OAAO,EAAE,mBAAmB;KAC7B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,SAA2C,EAC3C,cAAoC,uBAAuB;IAE3D,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,gBAAgB;YACnB,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QAC5D,KAAK,gBAAgB;YACnB,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QAC5D,KAAK,kBAAkB;YACrB,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QAC5D,KAAK,aAAa,CAAC;QACnB,KAAK,oBAAoB;YACvB,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QAC3D,KAAK,aAAa;YAChB,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QACzD,KAAK,gBAAgB;YACnB,OAAO,+BAA+B,EAAE,CAAC;QAC3C,KAAK,gBAAgB;YACnB,OAAO,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAC3C,KAAK,SAAS;YACZ,OAAO,qBAAqB,CAAC,SAAS,CAAC,CAAC;QAC1C,KAAK,mBAAmB,CAAC;QACzB,KAAK,gBAAgB,CAAC;QACtB,KAAK,gBAAgB,CAAC;QACtB,KAAK,SAAS;YACZ,OAAO,qBAAqB,CAAC,eAAe,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAAC,MASpD;IACC,MAAM,MAAM,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAE9C,mBAAmB,CAAC;QAClB,GAAG,MAAM;QACT,SAAS,EAAE,MAAM,CAAC,YAAY;QAC9B,OAAO,EAAE,MAAM,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QAClE,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,0EAA0E;QAC1E,oEAAoE;QACpE,uDAAuD;QACvD,cAAc,EAAE,KAAK;KACtB,CAAC,CAAC;IAEH,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO;IACT,CAAC;IAED,IAAI,gBAAoC,CAAC;IACzC,IAAI,CAAC;QACH,gBAAgB,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAClE,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,QAAQ,CAChB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,MAAM,CAAC,IAAI,CAChB,oEAAoE,EACpE;YACE,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CACF,CAAC;QACF,OAAO;IACT,CAAC;IAED,mBAAmB,CAAC;QAClB,GAAG,MAAM;QACT,OAAO,EAAE,gBAAgB;QACzB,SAAS,EAAE,MAAM,CAAC,YAAY;KAC/B,CAAC,CAAC;AACL,CAAC"}

package/dist/ai-service/state-machine/helpers/pre-session-metrics.d.ts

@@ -0,0 +1,38 @@
+import type { Logger } from "../../../util/logger.js";
+export declare const METRIC_PRE_SESSION_DURATION = "superblocks.clark.prompt.pre_session_duration_seconds";
+export declare const METRIC_PRE_SESSION_STALL_TOTAL = "superblocks.clark.prompt.pre_session_stall_total";
+type PreSessionOutcome = "canceled" | "ok";
+export declare const PRE_SESSION_BUDGET_MS = 30000;
+/**
+ * Second-scale bucket boundaries for the pre-session duration histogram.
+ * Defaults are a poor fit for a 30s watchdog budget (near-budget times
+ * collapse into one coarse band); keep resolution around the budget edge
+ * so the runbook P95 query is actionable.
+ */
+export declare const PRE_SESSION_DURATION_BUCKETS_SECONDS: readonly [0.05, 0.25, 1, 2, 5, 10, 15, 20, 25, 30, 45];
+export declare const PRE_SESSION_OUTCOME_TAG = "outcome";
+export declare function _resetPreSessionMetricsForTesting(): void;
+interface PreSessionWatchOptions {
+    applicationId?: string;
+    budgetMs?: number;
+    /** Frozen prompt dedupe key; stall notes this bucket instead of activeKey. */
+    degradationKey?: string;
+    logger: Pick<Logger, "warn" | "warnStructured">;
+    organizationId?: string;
+    /** Injectable clock for tests. */
+    nowMs?: () => number;
+    /** Injectable timer for tests. */
+    setTimer?: (callback: () => void, ms: number) => ReturnType<typeof setTimeout>;
+    clearTimer?: (handle: ReturnType<typeof setTimeout>) => void;
+}
+interface PreSessionWatch {
+    /** Disarm the watchdog and record the duration histogram. Idempotent. */
+    complete: (outcome: PreSessionOutcome) => void;
+}
+/**
+ * Arm an in-flight pre-session stall watchdog. Call {@link PreSessionWatch.complete}
+ * when streamText begins (`ok`) or the turn aborts before session (`canceled`).
+ */
+export declare function startPreSessionWatch(options: PreSessionWatchOptions): PreSessionWatch;
+export {};
+//# sourceMappingURL=pre-session-metrics.d.ts.map

package/dist/ai-service/state-machine/helpers/pre-session-metrics.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pre-session-metrics.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/pre-session-metrics.ts"],"names":[],"mappings":"AAeA,OAAO,KAAK,EAAiB,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAQrE,eAAO,MAAM,2BAA2B,0DACiB,CAAC;AAC1D,eAAO,MAAM,8BAA8B,qDACS,CAAC;AAErD,KAAK,iBAAiB,GAAG,UAAU,GAAG,IAAI,CAAC;AAE3C,eAAO,MAAM,qBAAqB,QAAS,CAAC;AAE5C;;;;;GAKG;AACH,eAAO,MAAM,oCAAoC,wDAEvC,CAAC;AAEX,eAAO,MAAM,uBAAuB,YAAY,CAAC;AAwDjD,wBAAgB,iCAAiC,IAAI,IAAI,CAExD;AAED,UAAU,sBAAsB;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,8EAA8E;IAC9E,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,gBAAgB,CAAC,CAAC;IAChD,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,kCAAkC;IAClC,KAAK,CAAC,EAAE,MAAM,MAAM,CAAC;IACrB,kCAAkC;IAClC,QAAQ,CAAC,EAAE,CACT,QAAQ,EAAE,MAAM,IAAI,EACpB,EAAE,EAAE,MAAM,KACP,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC;IACnC,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,KAAK,IAAI,CAAC;CAC9D;AAED,UAAU,eAAe;IACvB,yEAAyE;IACzE,QAAQ,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,CAAC;CAChD;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,sBAAsB,GAC9B,eAAe,CAuDjB"}

package/dist/ai-service/state-machine/helpers/pre-session-metrics.js

@@ -0,0 +1,131 @@
+/**
+ * Pre-session (LLMGenerating enter → streamText) observability
+ *
+ * IdleMonitor / `@event:stream_stall` only cover silence *after* Starting
+ * session. Empty Thinking hangs that block before the orchestrator starts
+ * need an in-flight watchdog: hung turns never finish a duration histogram.
+ *
+ * Metrics (no org/app labels — resource attrs only):
+ *   - `superblocks.clark.prompt.pre_session_duration_seconds` {outcome}
+ *   - `superblocks.clark.prompt.pre_session_stall_total`
+ *
+ * Structured warn: `event: "pre_session_stall"` (queryable like stream_stall).
+ */
+import { metrics } from "@opentelemetry/api";
+import { degradationKeyLogAttribute, notePromptDegradationForKey, } from "./prompt-degradation-metrics.js";
+const METER_NAME = "superblocks.clark.prompt";
+export const METRIC_PRE_SESSION_DURATION = "superblocks.clark.prompt.pre_session_duration_seconds";
+export const METRIC_PRE_SESSION_STALL_TOTAL = "superblocks.clark.prompt.pre_session_stall_total";
+export const PRE_SESSION_BUDGET_MS = 30_000;
+/**
+ * Second-scale bucket boundaries for the pre-session duration histogram.
+ * Defaults are a poor fit for a 30s watchdog budget (near-budget times
+ * collapse into one coarse band); keep resolution around the budget edge
+ * so the runbook P95 query is actionable.
+ */
+export const PRE_SESSION_DURATION_BUCKETS_SECONDS = [
+    0.05, 0.25, 1, 2, 5, 10, 15, 20, 25, 30, 45,
+];
+export const PRE_SESSION_OUTCOME_TAG = "outcome";
+class PreSessionMetrics {
+    cachedMeter;
+    cachedInstruments;
+    getInstruments() {
+        const meter = metrics.getMeter(METER_NAME);
+        if (meter === this.cachedMeter && this.cachedInstruments) {
+            return this.cachedInstruments;
+        }
+        this.cachedMeter = meter;
+        this.cachedInstruments = {
+            durationHistogram: meter.createHistogram(METRIC_PRE_SESSION_DURATION, {
+                description: "Wall time from LLMGenerating pre-stream critical section until streamText entry (or cancel).",
+                unit: "s",
+                advice: {
+                    explicitBucketBoundaries: [...PRE_SESSION_DURATION_BUCKETS_SECONDS],
+                },
+            }),
+            stallCounter: meter.createCounter(METRIC_PRE_SESSION_STALL_TOTAL, {
+                description: "In-flight pre-session watches that exceeded the stall budget before streamText.",
+                unit: "{stall}",
+            }),
+        };
+        return this.cachedInstruments;
+    }
+    recordDuration(outcome, durationSeconds) {
+        this.getInstruments().durationHistogram.record(durationSeconds, {
+            [PRE_SESSION_OUTCOME_TAG]: outcome,
+        });
+    }
+    recordStall() {
+        this.getInstruments().stallCounter.add(1);
+    }
+    _resetForTesting() {
+        this.cachedMeter = undefined;
+        this.cachedInstruments = undefined;
+    }
+}
+const preSessionMetrics = new PreSessionMetrics();
+export function _resetPreSessionMetricsForTesting() {
+    preSessionMetrics._resetForTesting();
+}
+/**
+ * Arm an in-flight pre-session stall watchdog. Call {@link PreSessionWatch.complete}
+ * when streamText begins (`ok`) or the turn aborts before session (`canceled`).
+ */
+export function startPreSessionWatch(options) {
+    const budgetMs = options.budgetMs ?? PRE_SESSION_BUDGET_MS;
+    const nowMs = options.nowMs ?? (() => Date.now());
+    const setTimer = options.setTimer ?? setTimeout;
+    const clearTimer = options.clearTimer ?? clearTimeout;
+    const startedAtMs = nowMs();
+    let settled = false;
+    const timer = setTimer(() => {
+        if (settled) {
+            return;
+        }
+        const elapsedMs = nowMs() - startedAtMs;
+        try {
+            preSessionMetrics.recordStall();
+            notePromptDegradationForKey(options.degradationKey, "pre_session_stall");
+        }
+        catch {
+            // Structured warn below remains the diagnostic source of truth.
+        }
+        const attributes = {
+            event: "pre_session_stall",
+            elapsed_ms: elapsedMs,
+            budget_ms: budgetMs,
+            ...degradationKeyLogAttribute(options.degradationKey),
+        };
+        if (options.organizationId) {
+            attributes.organization_id = options.organizationId;
+        }
+        if (options.applicationId) {
+            attributes.application_id = options.applicationId;
+        }
+        const message = "[llm-generating] Pre-session stall: streamText not reached within budget";
+        if (options.logger.warnStructured) {
+            options.logger.warnStructured(message, attributes);
+        }
+        else {
+            options.logger.warn(message, attributes);
+        }
+    }, budgetMs);
+    return {
+        complete(outcome) {
+            if (settled) {
+                return;
+            }
+            settled = true;
+            clearTimer(timer);
+            const durationSeconds = Math.max(0, (nowMs() - startedAtMs) / 1000);
+            try {
+                preSessionMetrics.recordDuration(outcome, durationSeconds);
+            }
+            catch {
+                // Best-effort; generation must continue.
+            }
+        },
+    };
+}
+//# sourceMappingURL=pre-session-metrics.js.map