@superblocksteam/vite-plugin-file-sync 1.99.99-next.2 → 2.0.0-SNAPSHOT.2143e2e95d0107ab

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

@@ -1,10 +1,15 @@
-import { USER_CANCELED } from "../clark-fsm.js";
+import { AGENT_CANCELED, USER_CANCELED } from "../clark-fsm.js";
+import { promptFailureFromStreamAbort } from "./prompt-failure.js";
 export const transitionFrom = (clark) => {
     return (event) => {
         if (clark.context.abortController?.signal.aborted &&
-            event.type !== USER_CANCELED) {
+            event.type !== USER_CANCELED &&
+            event.type !== AGENT_CANCELED) {
             // abort signal was not explicitly handled by previous state
-            return clark.send({ type: USER_CANCELED });
+            return clark.send({
+                type: USER_CANCELED,
+                failure: promptFailureFromStreamAbort(clark.context),
+            });
         }
         return clark.send(event);
     };

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

@@ -1 +1 @@
-{"version":3,"file":"transition.js","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/transition.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAwBhD,MAAM,CAAC,MAAM,cAAc,GAAG,CAAyB,KAAY,EAAE,EAAE;IACrE,OAAO,CACL,KAAwC,EACxC,EAAE;QACF,IACE,KAAK,CAAC,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC,OAAO;YAC7C,KAAK,CAAC,IAAI,KAAK,aAAa,EAC5B,CAAC;YACD,4DAA4D;YAC5D,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"transition.js","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/transition.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAQhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,qBAAqB,CAAC;AAiBnE,MAAM,CAAC,MAAM,cAAc,GAAG,CAAyB,KAAY,EAAE,EAAE;IACrE,OAAO,CACL,KAAwC,EACxC,EAAE;QACF,IACE,KAAK,CAAC,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC,OAAO;YAC7C,KAAK,CAAC,IAAI,KAAK,aAAa;YAC5B,KAAK,CAAC,IAAI,KAAK,cAAc,EAC7B,CAAC;YACD,4DAA4D;YAC5D,OAAO,KAAK,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,4BAA4B,CAAC,KAAK,CAAC,OAAO,CAAC;aACrD,CAAC,CAAC;QACL,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC,CAAC;AACJ,CAAC,CAAC"}

package/dist/ai-service/state-machine/helpers/user-preferences.d.ts

@@ -0,0 +1,8 @@
+import { type ClarkUserPreferences } from "@superblocksteam/library-shared/types";
+/**
+ * Merge partial {@link ClarkUserPreferences} from a user prompt with
+ * product defaults so callers can treat every field as required without
+ * scattering `?? defaultX` checks throughout the FSM.
+ */
+export declare const resolveUserPreferences: (prefs: Partial<ClarkUserPreferences> | undefined) => ClarkUserPreferences;
+//# sourceMappingURL=user-preferences.d.ts.map

package/dist/ai-service/state-machine/helpers/user-preferences.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"user-preferences.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/user-preferences.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,oBAAoB,EAC1B,MAAM,uCAAuC,CAAC;AAE/C;;;;GAIG;AACH,eAAO,MAAM,sBAAsB,GACjC,OAAO,OAAO,CAAC,oBAAoB,CAAC,GAAG,SAAS,KAC/C,oBAGD,CAAC"}

package/dist/ai-service/state-machine/helpers/user-preferences.js

@@ -0,0 +1,11 @@
+import { CLARK_USER_PREFERENCES_DEFAULTS, } from "@superblocksteam/library-shared/types";
+/**
+ * Merge partial {@link ClarkUserPreferences} from a user prompt with
+ * product defaults so callers can treat every field as required without
+ * scattering `?? defaultX` checks throughout the FSM.
+ */
+export const resolveUserPreferences = (prefs) => ({
+    ...CLARK_USER_PREFERENCES_DEFAULTS,
+    ...(prefs ?? {}),
+});
+//# sourceMappingURL=user-preferences.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"user-preferences.js","sourceRoot":"","sources":["../../../../src/ai-service/state-machine/helpers/user-preferences.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,+BAA+B,GAEhC,MAAM,uCAAuC,CAAC;AAE/C;;;;GAIG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,KAAgD,EAC1B,EAAE,CAAC,CAAC;IAC1B,GAAG,+BAA+B;IAClC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;CACjB,CAAC,CAAC"}

package/dist/ai-service/state-machine/mocks.d.ts

@@ -1,7 +1,7 @@
+import type { EditorClient } from "../types.js";
 import type { Clark, ClarkStateHandlerParams } from "./clark-fsm.js";
-import type { EditorClientMethods } from "@superblocksteam/library-shared/types";
-import type { ISocketClient } from "@superblocksteam/shared";
-export declare const mockPeer: ISocketClient<EditorClientMethods>;
+export declare const mockJwt: string;
+export declare const mockPeer: EditorClient;
 export declare const mockClark: Clark;
 export declare const mockHandlerParams: ClarkStateHandlerParams;
 //# sourceMappingURL=mocks.d.ts.map

package/dist/ai-service/state-machine/mocks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mocks.d.ts","sourceRoot":"","sources":["../../../src/ai-service/state-machine/mocks.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,KAAK,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AACrE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AACjF,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE7D,eAAO,MAAM,QAAQ,EAML,aAAa,CAAC,mBAAmB,CAAC,CAAC;AAEnD,eAAO,MAAM,SAAS,EAQN,KAAK,CAAC;AAEtB,eAAO,MAAM,iBAAiB,EAqDd,uBAAuB,CAAC"}
+{"version":3,"file":"mocks.d.ts","sourceRoot":"","sources":["../../../src/ai-service/state-machine/mocks.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,KAAK,EAAE,KAAK,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAGrE,eAAO,MAAM,OAAO,QAKY,CAAC;AAgBjC,eAAO,MAAM,QAAQ,EAmBL,YAAY,CAAC;AAE7B,eAAO,MAAM,SAAS,EASN,KAAK,CAAC;AAEtB,eAAO,MAAM,iBAAiB,EAiId,uBAAuB,CAAC"}

package/dist/ai-service/state-machine/mocks.js

@@ -1,17 +1,50 @@
 import { simulateReadableStream } from "ai";
-import { MockLanguageModelV1 } from "ai/test";
+import { MockLanguageModelV3 } from "ai/test";
 import { vi } from "vitest";
+import { AiServiceFeatureFlags } from "../features.js";
+// A properly formatted mock JWT with user_id in payload for tests that need JWT parsing
+export const mockJwt = `header.${Buffer.from(JSON.stringify({
+    user_id: "test-user-123",
+    org_id: "test-org-456",
+})).toString("base64")}.signature`;
+const createV3Usage = (inputTokens, outputTokens) => ({
+    inputTokens: {
+        total: inputTokens,
+        noCache: inputTokens,
+        cacheRead: 0,
+        cacheWrite: 0,
+    },
+    outputTokens: {
+        total: outputTokens,
+        text: outputTokens,
+        reasoning: 0,
+    },
+});
 export const mockPeer = {
+    activeRouteLiveness: {
+        activityVersion: 0,
+        available: true,
+        generation: 1,
+    },
     call: {
         aiPushMessage: vi.fn(),
-        aiSetCompleted: vi.fn(),
+        aiPushChangeInfo: vi.fn(),
+        aiSetGenerationState: vi.fn(),
+        aiPushPromptInterpretation: vi.fn(),
     },
+    connectionGeneration: 1,
+    invalidateConnection: vi.fn(),
+    pushMessageToObservers: vi.fn(),
+    reportRpcTimeout: vi.fn(),
+    whenConnected: vi.fn(async () => false),
+    whenConnectedAfter: vi.fn(async () => false),
     close: vi.fn(),
 };
 export const mockClark = {
     state: "Idle",
     context: { peer: mockPeer },
     getMetadata: vi.fn(),
+    resetContext: vi.fn(),
     updateContext: vi.fn(),
     updateMetadata: vi.fn(),
     resetAllMetadata: vi.fn(),
@@ -32,41 +65,113 @@ export const mockHandlerParams = {
     },
     signals: {
         emit: vi.fn(),
+        once: vi
+            .fn()
+            .mockImplementation((_event, callback) => {
+            // Simulate checkpoint uploaded immediately (next microtask)
+            void Promise.resolve().then(() => callback());
+        }),
+    },
+    sourceTrackerInterface: {
+        getElementToFilePathMap: vi.fn(),
+        getElementToFilePath: vi.fn(),
+        getElementToLocation: vi.fn(),
+        getScopeDefinitionForPage: vi.fn(),
+        isCustomComponent: vi.fn(),
     },
     appShell: {
         appRootDirPath: "/test/app",
-        readFiles: vi.fn(),
-        runTypecheck: vi.fn(),
-    },
-    artifactProcessor: {
-        process: vi.fn(),
-        extractFileArtifacts: vi.fn().mockReturnValue([
+        readFiles: vi.fn().mockReturnValue([
             {
                 name: "test.ts",
                 content: 'console.log("test");',
             },
         ]),
-        createArtifactContext: vi.fn().mockReturnValue({
-            paths: ["test.ts"],
-            content: "test file content",
-        }),
+        runTypecheck: vi.fn(),
+        runLinter: vi.fn(),
+        warmLinterCache: vi.fn(),
+        clearScratchDirectory: vi.fn(),
     },
-    anthropicProvider: vi.fn().mockReturnValue(new MockLanguageModelV1({
-        doStream: async () => ({
+    clarkProvider: vi.fn().mockReturnValue(new MockLanguageModelV3({
+        doStream: async (_options) => ({
             stream: simulateReadableStream({
                 chunks: [
-                    { type: "text-delta", textDelta: "Hello" },
-                    { type: "text-delta", textDelta: " world!" },
+                    { type: "text-delta", id: "1", delta: "Hello" },
+                    { type: "text-delta", id: "2", delta: " world!" },
                     {
                         type: "finish",
-                        finishReason: "stop",
-                        logprobs: undefined,
-                        usage: { completionTokens: 10, promptTokens: 3 },
+                        finishReason: { unified: "stop", raw: "stop" },
+                        usage: createV3Usage(13, 10),
                     },
                 ],
             }),
-            rawCall: { rawPrompt: null, rawSettings: {} },
+            request: { body: null },
+            response: { headers: {} },
         }),
     })),
+    integrationStore: {
+        getOpenApiSpec: vi.fn(),
+        getIntegrationForPromptContext: vi.fn(),
+    },
+    knowledgeManager: {
+        loadFactsPromptSections: vi.fn().mockResolvedValue({
+            systemPromptSection: "",
+            userMessageSection: "",
+            onDemandSkillIndex: "",
+            facts: {
+                org: [],
+                user: [],
+                integrations: {},
+            },
+        }),
+        getNetNewFacts: vi.fn().mockReturnValue({
+            org: [],
+            user: [],
+            integrations: {},
+        }),
+        hasAnyFacts: vi.fn().mockReturnValue(false),
+        getFactById: vi.fn().mockReturnValue(undefined),
+    },
+    appContextStore: {
+        componentSemanticsById: {},
+        apiSemanticsByName: {},
+        getComponent: vi.fn().mockReturnValue({}),
+    },
+    chatSessionStore: {
+        getMessages: vi.fn(),
+        newChat: vi.fn(),
+        recordUser: vi.fn(),
+        recordUserAcceptance: vi.fn(),
+        recordUserRejection: vi.fn(),
+        recordAssistant: vi.fn(),
+        recordAssistantChangeInfo: vi.fn(),
+        getLastUserMessage: vi.fn(),
+        flushPendingPushes: vi.fn().mockResolvedValue(undefined),
+    },
+    features: AiServiceFeatureFlags.create({ "ai.suggest-whats-next": true }),
+    llmProvider: {
+        name: "test-provider",
+        modelForTask: vi.fn().mockReturnValue(new MockLanguageModelV3({
+            doStream: async (_options) => ({
+                stream: simulateReadableStream({
+                    chunks: [
+                        {
+                            type: "finish",
+                            finishReason: { unified: "stop", raw: "stop" },
+                            usage: createV3Usage(1, 1),
+                        },
+                    ],
+                }),
+                request: { body: null },
+                response: { headers: {} },
+            }),
+        })),
+    },
+    logger: {
+        info: vi.fn(),
+        error: vi.fn(),
+        debug: vi.fn(),
+        warn: vi.fn(),
+    },
 };
 //# sourceMappingURL=mocks.js.map

package/dist/ai-service/state-machine/mocks.js.map

@@ -1 +1 @@
-{"version":3,"file":"mocks.js","sourceRoot":"","sources":["../../../src/ai-service/state-machine/mocks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,IAAI,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAK5B,MAAM,CAAC,MAAM,QAAQ,GAAG;IACtB,IAAI,EAAE;QACJ,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;QACtB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;KACxB;IACD,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;CACkC,CAAC;AAEnD,MAAM,CAAC,MAAM,SAAS,GAAG;IACvB,KAAK,EAAE,MAAM;IACb,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;IAC3B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;IACpB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;IACtB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;IACvB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;IACzB,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;CACM,CAAC;AAEtB,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,cAAc,EAAE;QACd,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;KACtB;IACD,cAAc,EAAE,WAAW;IAC3B,gBAAgB,EAAE;QAChB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;QACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;KACxD;IACD,gBAAgB,EAAE;QAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;KAChB;IACD,OAAO,EAAE;QACP,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;KACd;IACD,QAAQ,EAAE;QACR,cAAc,EAAE,WAAW;QAC3B,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;QAClB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;KACtB;IACD,iBAAiB,EAAE;QACjB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAC5C;gBACE,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,sBAAsB;aAChC;SACF,CAAC;QACF,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAC7C,KAAK,EAAE,CAAC,SAAS,CAAC;YAClB,OAAO,EAAE,mBAAmB;SAC7B,CAAC;KACH;IACD,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CACxC,IAAI,mBAAmB,CAAC;QACtB,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACrB,MAAM,EAAE,sBAAsB,CAAC;gBAC7B,MAAM,EAAE;oBACN,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,OAAO,EAAE;oBAC1C,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE;oBAC5C;wBACE,IAAI,EAAE,QAAQ;wBACd,YAAY,EAAE,MAAM;wBACpB,QAAQ,EAAE,SAAS;wBACnB,KAAK,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE;qBACjD;iBACF;aACF,CAAC;YACF,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE;SAC9C,CAAC;KACH,CAAC,CACH;CACoC,CAAC"}
+{"version":3,"file":"mocks.js","sourceRoot":"","sources":["../../../src/ai-service/state-machine/mocks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,IAAI,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AAIvD,wFAAwF;AACxF,MAAM,CAAC,MAAM,OAAO,GAAG,UAAU,MAAM,CAAC,IAAI,CAC1C,IAAI,CAAC,SAAS,CAAC;IACb,OAAO,EAAE,eAAe;IACxB,MAAM,EAAE,cAAc;CACvB,CAAC,CACH,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;AAEjC,MAAM,aAAa,GAAG,CAAC,WAAmB,EAAE,YAAoB,EAAE,EAAE,CAAC,CAAC;IACpE,WAAW,EAAE;QACX,KAAK,EAAE,WAAW;QAClB,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;KACd;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,YAAY;QACnB,IAAI,EAAE,YAAY;QAClB,SAAS,EAAE,CAAC;KACb;CACF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,QAAQ,GAAG;IACtB,mBAAmB,EAAE;QACnB,eAAe,EAAE,CAAC;QAClB,SAAS,EAAE,IAAI;QACf,UAAU,EAAE,CAAC;KACd;IACD,IAAI,EAAE;QACJ,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;QACtB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC7B,0BAA0B,EAAE,EAAE,CAAC,EAAE,EAAE;KACpC;IACD,oBAAoB,EAAE,CAAC;IACvB,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC7B,sBAAsB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC/B,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;IACzB,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,CAAC;IACvC,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,CAAC;IAC5C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;CACY,CAAC;AAE7B,MAAM,CAAC,MAAM,SAAS,GAAG;IACvB,KAAK,EAAE,MAAM;IACb,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;IAC3B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;IACpB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;IACrB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;IACtB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;IACvB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;IACzB,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;CACM,CAAC;AAEtB,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,cAAc,EAAE;QACd,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;KACtB;IACD,cAAc,EAAE,WAAW;IAC3B,gBAAgB,EAAE;QAChB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;QACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;KACxD;IACD,gBAAgB,EAAE;QAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;KAChB;IACD,OAAO,EAAE;QACP,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE;aACL,EAAE,EAAE;aACJ,kBAAkB,CACjB,CAAC,MAAc,EAAE,QAAsC,EAAE,EAAE;YACzD,4DAA4D;YAC5D,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC;QAChD,CAAC,CACF;KACJ;IACD,sBAAsB,EAAE;QACtB,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE;QAChC,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC7B,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC7B,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE;QAClC,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC3B;IACD,QAAQ,EAAE;QACR,cAAc,EAAE,WAAW;QAC3B,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YACjC;gBACE,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,sBAAsB;aAChC;SACF,CAAC;QACF,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;QAClB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC/B;IACD,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CACpC,IAAI,mBAAmB,CAAC;QACtB,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YAC7B,MAAM,EAAE,sBAAsB,CAAC;gBAC7B,MAAM,EAAE;oBACN,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE;oBAC/C,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE;oBACjD;wBACE,IAAI,EAAE,QAAQ;wBACd,YAAY,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE;wBAC9C,KAAK,EAAE,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC;qBAC7B;iBACF;aACF,CAAC;YACF,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YACvB,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;SAC1B,CAAC;KACH,CAAC,CACH;IACD,gBAAgB,EAAE;QAChB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;QACvB,8BAA8B,EAAE,EAAE,CAAC,EAAE,EAAE;KACxC;IACD,gBAAgB,EAAE;QAChB,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACjD,mBAAmB,EAAE,EAAE;YACvB,kBAAkB,EAAE,EAAE;YACtB,kBAAkB,EAAE,EAAE;YACtB,KAAK,EAAE;gBACL,GAAG,EAAE,EAAE;gBACP,IAAI,EAAE,EAAE;gBACR,YAAY,EAAE,EAAE;aACjB;SACF,CAAC;QACF,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YACtC,GAAG,EAAE,EAAE;YACP,IAAI,EAAE,EAAE;YACR,YAAY,EAAE,EAAE;SACjB,CAAC;QACF,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC3C,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;KAChD;IACD,eAAe,EAAE;QACf,sBAAsB,EAAE,EAAE;QAC1B,kBAAkB,EAAE,EAAE;QACtB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;KAC1C;IACD,gBAAgB,EAAE;QAChB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;QACpB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC7B,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC5B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE;QAClC,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC3B,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KACzD;IACD,QAAQ,EAAE,qBAAqB,CAAC,MAAM,CAAC,EAAE,uBAAuB,EAAE,IAAI,EAAE,CAAC;IACzE,WAAW,EAAE;QACX,IAAI,EAAE,eAAe;QACrB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CACnC,IAAI,mBAAmB,CAAC;YACtB,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAC7B,MAAM,EAAE,sBAAsB,CAAC;oBAC7B,MAAM,EAAE;wBACN;4BACE,IAAI,EAAE,QAAQ;4BACd,YAAY,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE;4BAC9C,KAAK,EAAE,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC;yBAC3B;qBACF;iBACF,CAAC;gBACF,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;gBACvB,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;aAC1B,CAAC;SACH,CAAC,CACH;KACF;IACD,MAAM,EAAE;QACN,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;KACd;CACoC,CAAC"}

package/dist/ai-service/state-machine/traced-fsm.d.ts

@@ -0,0 +1,89 @@
+import { type Logger } from "../../util/logger.js";
+import { type ExportedLLMObsSpan, type LLMObsNamedSpanOptions, LLMObsTracer } from "../llmobs/index.js";
+import { FSM, type FSMOnTransitionParams, type FSMConfig, type FSMState, type FSMEventBase } from "./fsm.js";
+export type TracedFSMConfig<S extends FSMState, E extends FSMEventBase, C> = FSMConfig<S, E, C> & {
+    /**
+     * Create a provider function that returns tags to be added to every span.
+     * This can be used to attach contextual metadata such as request IDs,
+     * user IDs, etc. It is called at the termination of each non-terminal state,
+     * and at the end of the FSM lifecycle.
+     * @param fsm The TracedFSM instance
+     */
+    createTagProvider?: TracedFSMTagProviderFactory<S, E, C>;
+    /** Optional session ID to associate with all spans created by this FSM instance. */
+    sessionId?: string;
+};
+export type TracedFSMStateSpanOptions = LLMObsNamedSpanOptions | "terminal";
+export type TracedFSMTagProviderFactory<S extends FSMState, E extends FSMEventBase, C> = (fsm: TracedFSM<S, E, C>) => TagProvider;
+/**
+ * FSM wrapper that emits tracing spans for each state transition.
+ *
+ * ## Span Hierarchy
+ *
+ * The TracedFSM creates a hierarchical span structure:
+ * - **Root Span** (optional): Tracks the entire FSM lifecycle from initialization to terminal state
+ * - **State Spans**: Individual spans for each non-terminal state
+ *
+ * ## Terminal States
+ *
+ * States marked as "terminal" in the configuration:
+ * - Do not create their own spans
+ * - Trigger the completion of the root span (if configured)
+ * - Represent final states in the FSM lifecycle
+ *
+ * ## Error Handling
+ *
+ * Tracing failures are logged but do not interrupt FSM operation. The FSM will continue
+ * to function normally even if the tracing infrastructure is unavailable.
+ */
+export declare abstract class TracedFSM<S extends string, E extends {
+    type: string;
+}, C> extends FSM<S, E, C> {
+    private spans;
+    protected sessionId?: string;
+    tracer: LLMObsTracer;
+    logger: Logger;
+    private getTags;
+    constructor(config: TracedFSMConfig<S, E, C>, tracer?: LLMObsTracer, logger?: Logger);
+    /**
+     * Exported representation of the active root span (FSM lifecycle), if any.
+     */
+    get rootSpan(): ExportedLLMObsSpan | undefined;
+    /**
+     * Supply the span options for the long-lived root span that wraps
+     * the entire FSM lifecycle. Invoked once when the first state transition
+     * runs.
+     */
+    protected abstract getRootSpanOptions(initialEvent: E): LLMObsNamedSpanOptions;
+    /**
+     * Provide the span options for the state being entered.
+     * Return "terminal" to finish the root span instead of
+     * creating a new state span.
+     */
+    protected abstract getStateSpanOptions(state: S): TracedFSMStateSpanOptions;
+    /**
+     * Sanitize the event object to remove sensitive information
+     * before attaching it to the trace.
+     * */
+    protected abstract sanitizeEvent(event: E): Partial<E>;
+    /**
+     * Extra tags applied when finishing the root span on a terminal transition.
+     * Invoked before the terminal-state handler runs.
+     */
+    protected getRootSpanFinishTags(_params: FSMOnTransitionParams<S, E>): Tags;
+    private handleTransitionTracing;
+    private startRootSpanIfNeeded;
+    private finishRootSpanIfOpen;
+    private startStateSpan;
+    private finishStateSpanIfOpen;
+    private startSpan;
+    private finishSpan;
+    private isTransitionParams;
+    private isErrorTransitionParams;
+    private getErrorData;
+}
+export type Tags = {
+    [key: string]: any;
+};
+export type TagProvider = () => Tags;
+//# sourceMappingURL=traced-fsm.d.ts.map

package/dist/ai-service/state-machine/traced-fsm.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"traced-fsm.d.ts","sourceRoot":"","sources":["../../../src/ai-service/state-machine/traced-fsm.ts"],"names":[],"mappings":"AACA,OAAO,EAA2B,KAAK,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC5E,OAAO,EAEL,KAAK,kBAAkB,EACvB,KAAK,sBAAsB,EAC3B,YAAY,EACb,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,GAAG,EACH,KAAK,qBAAqB,EAC1B,KAAK,SAAS,EACd,KAAK,QAAQ,EACb,KAAK,YAAY,EAClB,MAAM,UAAU,CAAC;AAElB,MAAM,MAAM,eAAe,CACzB,CAAC,SAAS,QAAQ,EAClB,CAAC,SAAS,YAAY,EACtB,CAAC,IACC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG;IACvB;;;;;;OAMG;IACH,iBAAiB,CAAC,EAAE,2BAA2B,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACzD,oFAAoF;IACpF,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,sBAAsB,GAAG,UAAU,CAAC;AAE5E,MAAM,MAAM,2BAA2B,CACrC,CAAC,SAAS,QAAQ,EAClB,CAAC,SAAS,YAAY,EACtB,CAAC,IACC,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,WAAW,CAAC;AAE7C;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,8BAAsB,SAAS,CAC7B,CAAC,SAAS,MAAM,EAChB,CAAC,SAAS;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,EAC1B,CAAC,CACD,SAAQ,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACpB,OAAO,CAAC,KAAK,CAIX;IAEF,SAAS,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,YAAY,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,OAAO,CAAc;gBAG3B,MAAM,EAAE,eAAe,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAChC,MAAM,GAAE,YAAoC,EAC5C,MAAM,GAAE,MAAoB;IAiC9B;;OAEG;IACH,IAAW,QAAQ,IAAI,kBAAkB,GAAG,SAAS,CAEpD;IAED;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,kBAAkB,CACnC,YAAY,EAAE,CAAC,GACd,sBAAsB;IAEzB;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC,GAAG,yBAAyB;IAE3E;;;SAGK;IACL,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAEtD;;;OAGG;IACH,SAAS,CAAC,qBAAqB,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI;YAI7D,uBAAuB;IA2BrC,OAAO,CAAC,qBAAqB;IAY7B,OAAO,CAAC,oBAAoB;IAgB5B,OAAO,CAAC,cAAc;IAOtB,OAAO,CAAC,qBAAqB;IAY7B,OAAO,CAAC,SAAS;IAgBjB,OAAO,CAAC,UAAU;IAwBlB,OAAO,CAAC,kBAAkB;IAY1B,OAAO,CAAC,uBAAuB;IAS/B,OAAO,CAAC,YAAY;CASrB;AACD,MAAM,MAAM,IAAI,GAAG;IAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;CAAE,CAAC;AAE1C,MAAM,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC"}

package/dist/ai-service/state-machine/traced-fsm.js

@@ -0,0 +1,179 @@
+import { safeJsonStringify } from "../../util/log-sanitizer.js";
+import { getErrorMeta, getLogger } from "../../util/logger.js";
+import { LLMObsTracer, } from "../llmobs/index.js";
+import { FSM, } from "./fsm.js";
+/**
+ * FSM wrapper that emits tracing spans for each state transition.
+ *
+ * ## Span Hierarchy
+ *
+ * The TracedFSM creates a hierarchical span structure:
+ * - **Root Span** (optional): Tracks the entire FSM lifecycle from initialization to terminal state
+ * - **State Spans**: Individual spans for each non-terminal state
+ *
+ * ## Terminal States
+ *
+ * States marked as "terminal" in the configuration:
+ * - Do not create their own spans
+ * - Trigger the completion of the root span (if configured)
+ * - Represent final states in the FSM lifecycle
+ *
+ * ## Error Handling
+ *
+ * Tracing failures are logged but do not interrupt FSM operation. The FSM will continue
+ * to function normally even if the tracing infrastructure is unavailable.
+ */
+export class TracedFSM extends FSM {
+    spans;
+    sessionId;
+    tracer;
+    logger;
+    getTags;
+    constructor(config, tracer = LLMObsTracer.instance, logger = getLogger()) {
+        const { sessionId, createTransitionHandler, createTagProvider, ...rest } = config;
+        const wrappedConfig = {
+            ...rest,
+            createTransitionHandler: (fsm) => {
+                const tracedFsm = fsm;
+                const originalHandler = createTransitionHandler(tracedFsm);
+                return async (params) => {
+                    return await tracedFsm.handleTransitionTracing(originalHandler, params);
+                };
+            },
+        };
+        super(wrappedConfig);
+        this.spans = {
+            stateSpan: undefined,
+            rootSpan: undefined,
+            exportedRootSpan: undefined,
+        };
+        this.sessionId = sessionId;
+        this.tracer = tracer;
+        this.logger = logger;
+        this.getTags = createTagProvider?.(this) ?? (() => ({}));
+    }
+    /**
+     * Exported representation of the active root span (FSM lifecycle), if any.
+     */
+    get rootSpan() {
+        return this.spans.exportedRootSpan;
+    }
+    /**
+     * Extra tags applied when finishing the root span on a terminal transition.
+     * Invoked before the terminal-state handler runs.
+     */
+    getRootSpanFinishTags(_params) {
+        return {};
+    }
+    async handleTransitionTracing(next, params) {
+        this.finishStateSpanIfOpen(params);
+        const options = this.getStateSpanOptions(params.to);
+        if (options == "terminal") {
+            this.finishRootSpanIfOpen(params);
+        }
+        else {
+            this.startRootSpanIfNeeded(params);
+            this.startStateSpan(options, params);
+        }
+        try {
+            await next(params);
+        }
+        catch (error) {
+            this.logger.error("Traced FSM transition threw an error", getErrorMeta(error));
+            this.finishStateSpanIfOpen(error);
+            this.finishRootSpanIfOpen(error);
+            throw error;
+        }
+    }
+    startRootSpanIfNeeded(params) {
+        if (this.spans.rootSpan) {
+            // Root span already active
+            return;
+        }
+        const config = this.getRootSpanOptions(params.event);
+        const span = this.startSpan(config, params);
+        this.spans.rootSpan = span;
+        this.spans.exportedRootSpan = this.tracer.exportSpan(span);
+        this.tracer.rootSpan = span;
+    }
+    finishRootSpanIfOpen(paramsOrError) {
+        const span = this.spans.rootSpan;
+        if (!span) {
+            // No active root span to finish
+            return;
+        }
+        const extraTags = this.isTransitionParams(paramsOrError)
+            ? this.getRootSpanFinishTags(paramsOrError)
+            : {};
+        this.finishSpan(span, paramsOrError, extraTags);
+        this.spans.rootSpan = undefined;
+        this.tracer.rootSpan = undefined;
+    }
+    startStateSpan(options, params) {
+        this.spans.stateSpan = this.startSpan(options, params);
+    }
+    finishStateSpanIfOpen(paramsOrError) {
+        const stateSpan = this.spans.stateSpan;
+        if (!stateSpan) {
+            // No active state span to finish. This should only occur on initial transition.
+            return;
+        }
+        this.finishSpan(stateSpan, paramsOrError);
+        this.spans.stateSpan = undefined;
+    }
+    startSpan(options, params) {
+        if (this.sessionId) {
+            options.sessionId = this.sessionId;
+        }
+        const span = this.tracer.startActiveSpan(options);
+        const sanitizedEvent = this.sanitizeEvent(params.event);
+        this.tracer.annotate(span, {
+            inputData: safeJsonStringify(sanitizedEvent),
+            tags: { "fsm.fromState": params.from },
+        });
+        return span;
+    }
+    finishSpan(span, paramsOrError, extraTags = {}) {
+        const tags = { ...this.getTags(), ...extraTags };
+        if (this.isTransitionParams(paramsOrError)) {
+            const sanitizedEvent = this.sanitizeEvent(paramsOrError.event);
+            this.tracer.annotate(span, {
+                outputData: safeJsonStringify(sanitizedEvent),
+                tags: { "fsm.toState": paramsOrError.to, ...tags },
+                error: this.isErrorTransitionParams(paramsOrError)
+                    ? safeJsonStringify(this.getErrorData(paramsOrError))
+                    : undefined,
+            });
+        }
+        else {
+            this.tracer.annotate(span, {
+                error: paramsOrError,
+                tags: { ...tags },
+            });
+        }
+        span.finish();
+    }
+    isTransitionParams(value) {
+        return (value !== null &&
+            typeof value === "object" &&
+            "event" in value &&
+            "from" in value &&
+            "to" in value);
+    }
+    isErrorTransitionParams(value) {
+        if (!this.isTransitionParams(value)) {
+            return false;
+        }
+        return "error" in value.event;
+    }
+    getErrorData(value) {
+        if (!this.isErrorTransitionParams(value)) {
+            return undefined;
+        }
+        if ("error" in value.event) {
+            return value.event.error;
+        }
+        return value.event;
+    }
+}
+//# sourceMappingURL=traced-fsm.js.map

package/dist/ai-service/state-machine/traced-fsm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traced-fsm.js","sourceRoot":"","sources":["../../../src/ai-service/state-machine/traced-fsm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,SAAS,EAAe,MAAM,sBAAsB,CAAC;AAC5E,OAAO,EAIL,YAAY,GACb,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,GAAG,GAKJ,MAAM,UAAU,CAAC;AA2BlB;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,OAAgB,SAIpB,SAAQ,GAAY;IACZ,KAAK,CAIX;IAEQ,SAAS,CAAU;IACtB,MAAM,CAAe;IACrB,MAAM,CAAS;IACd,OAAO,CAAc;IAE7B,YACE,MAAgC,EAChC,SAAuB,YAAY,CAAC,QAAQ,EAC5C,SAAiB,SAAS,EAAE;QAE5B,MAAM,EAAE,SAAS,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,GAAG,IAAI,EAAE,GACtE,MAAM,CAAC;QAET,MAAM,aAAa,GAAuB;YACxC,GAAG,IAAI;YACP,uBAAuB,EAAE,CAAC,GAAiB,EAAE,EAAE;gBAC7C,MAAM,SAAS,GAAG,GAAyB,CAAC;gBAC5C,MAAM,eAAe,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC;gBAC3D,OAAO,KAAK,EAAE,MAAmC,EAAE,EAAE;oBACnD,OAAO,MAAM,SAAS,CAAC,uBAAuB,CAC5C,eAAe,EACf,MAAM,CACP,CAAC;gBACJ,CAAC,CAAC;YACJ,CAAC;SACF,CAAC;QAEF,KAAK,CAAC,aAAa,CAAC,CAAC;QAErB,IAAI,CAAC,KAAK,GAAG;YACX,SAAS,EAAE,SAAS;YACpB,QAAQ,EAAE,SAAS;YACnB,gBAAgB,EAAE,SAAS;SAC5B,CAAC;QAEF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,iBAAiB,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC;IACrC,CAAC;IAwBD;;;OAGG;IACO,qBAAqB,CAAC,OAAoC;QAClE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,KAAK,CAAC,uBAAuB,CACnC,IAAmE,EACnE,MAAmC;QAEnC,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACpD,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;YAC1B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,sCAAsC,EACtC,YAAY,CAAC,KAAK,CAAC,CACpB,CAAC;YACF,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;YAEjC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,qBAAqB,CAAC,MAAmC;QAC/D,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YACxB,2BAA2B;YAC3B,OAAO;QACT,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACrD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC5C,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;IAC9B,CAAC;IAEO,oBAAoB,CAC1B,aAAoD;QAEpD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,gCAAgC;YAChC,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC;YACtD,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,aAAa,CAAC;YAC3C,CAAC,CAAC,EAAE,CAAC;QACP,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC;QAChC,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,SAAS,CAAC;IACnC,CAAC;IAEO,cAAc,CACpB,OAA+B,EAC/B,MAAmC;QAEnC,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACzD,CAAC;IAEO,qBAAqB,CAC3B,aAAoD;QAEpD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;QACvC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,gFAAgF;YAChF,OAAO;QACT,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;IACnC,CAAC;IAEO,SAAS,CACf,OAA+B,EAC/B,MAAmC;QAEnC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QACrC,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE;YACzB,SAAS,EAAE,iBAAiB,CAAC,cAAc,CAAC;YAC5C,IAAI,EAAE,EAAE,eAAe,EAAE,MAAM,CAAC,IAAI,EAAE;SACvC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,UAAU,CAChB,IAAgB,EAChB,aAAoD,EACpD,YAAkB,EAAE;QAEpB,MAAM,IAAI,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,GAAG,SAAS,EAAE,CAAC;QACjD,IAAI,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,EAAE,CAAC;YAC3C,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YAC/D,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE;gBACzB,UAAU,EAAE,iBAAiB,CAAC,cAAc,CAAC;gBAC7C,IAAI,EAAE,EAAE,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE;gBAClD,KAAK,EAAE,IAAI,CAAC,uBAAuB,CAAC,aAAa,CAAC;oBAChD,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;oBACrD,CAAC,CAAC,SAAS;aACd,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE;gBACzB,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE;aAClB,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAEO,kBAAkB,CACxB,KAA4C;QAE5C,OAAO,CACL,KAAK,KAAK,IAAI;YACd,OAAO,KAAK,KAAK,QAAQ;YACzB,OAAO,IAAI,KAAK;YAChB,MAAM,IAAI,KAAK;YACf,IAAI,IAAI,KAAK,CACd,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAC7B,KAA4C;QAE5C,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC;IAChC,CAAC;IAEO,YAAY,CAAC,KAA4C;QAC/D,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,EAAE,CAAC;YACzC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,OAAO,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAC3B,OAAO,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC;QAC3B,CAAC;QACD,OAAO,KAAK,CAAC,KAAK,CAAC;IACrB,CAAC;CACF"}

package/dist/ai-service/tasks/access.d.ts

@@ -0,0 +1,23 @@
+import type { RunningTask } from "./types.js";
+/**
+ * Returns true when a task may be inspected (`checkTask`, `waitForTask`,
+ * `listTasks`) or actuated (`cancelTask`) by the caller currently
+ * bound to the AiService.
+ *
+ * The contract:
+ *   - Tasks with no recorded `ownerUserId` (legacy or local-dev with no
+ *     JWT) are accessible to anyone in the same AiService session —
+ *     preserves the pre-scoping behavior so adding ownership does not
+ *     silently break tasks created before the upgrade.
+ *   - Tasks with a recorded `ownerUserId` are accessible only when the
+ *     caller's identity (`callerUserId`) matches exactly. A caller with
+ *     no identity (anonymous reconnect, JWT lost) cannot inspect
+ *     identified tasks.
+ *
+ * Centralized so `checkTask` / `waitForTask` / `cancelTask` / `listTasks`
+ * apply the same denylist semantics — any divergence between them
+ * would re-introduce the cross-peer info-leak Matt flagged on PR
+ * #19718.
+ */
+export declare function isTaskAccessibleTo(task: Pick<RunningTask, "ownerUserId">, callerUserId: string | undefined): boolean;
+//# sourceMappingURL=access.d.ts.map

package/dist/ai-service/tasks/access.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"access.d.ts","sourceRoot":"","sources":["../../../src/ai-service/tasks/access.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE9C;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,kBAAkB,CAChC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,EACtC,YAAY,EAAE,MAAM,GAAG,SAAS,GAC/B,OAAO,CAGT"}

package/dist/ai-service/tasks/access.js

@@ -0,0 +1,26 @@
+/**
+ * Returns true when a task may be inspected (`checkTask`, `waitForTask`,
+ * `listTasks`) or actuated (`cancelTask`) by the caller currently
+ * bound to the AiService.
+ *
+ * The contract:
+ *   - Tasks with no recorded `ownerUserId` (legacy or local-dev with no
+ *     JWT) are accessible to anyone in the same AiService session —
+ *     preserves the pre-scoping behavior so adding ownership does not
+ *     silently break tasks created before the upgrade.
+ *   - Tasks with a recorded `ownerUserId` are accessible only when the
+ *     caller's identity (`callerUserId`) matches exactly. A caller with
+ *     no identity (anonymous reconnect, JWT lost) cannot inspect
+ *     identified tasks.
+ *
+ * Centralized so `checkTask` / `waitForTask` / `cancelTask` / `listTasks`
+ * apply the same denylist semantics — any divergence between them
+ * would re-introduce the cross-peer info-leak Matt flagged on PR
+ * #19718.
+ */
+export function isTaskAccessibleTo(task, callerUserId) {
+    if (task.ownerUserId === undefined)
+        return true;
+    return task.ownerUserId === callerUserId;
+}
+//# sourceMappingURL=access.js.map

package/dist/ai-service/tasks/access.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"access.js","sourceRoot":"","sources":["../../../src/ai-service/tasks/access.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,UAAU,kBAAkB,CAChC,IAAsC,EACtC,YAAgC;IAEhC,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IAChD,OAAO,IAAI,CAAC,WAAW,KAAK,YAAY,CAAC;AAC3C,CAAC"}