@dremio/js-sdk 0.62.1 → 0.64.0

package/dist/cloud/ai/AIResource.d.ts

@@ -157,6 +157,18 @@ export declare class AIResource {
                     chunkType: "interrupt";
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "contextUsage";
+                    contextUsagePercentage: number;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -334,6 +346,18 @@ export declare class AIResource {
                         chunkType: "interrupt";
                     };
                     role: "agent";
+                } | {
+                    conversationId: string;
+                    modelName: string;
+                    modelProviderId: string;
+                    runId: string;
+                    createdAt: import("temporal-polyfill").Temporal.Instant;
+                    id: string;
+                    content: {
+                        chunkType: "contextUsage";
+                        contextUsagePercentage: number;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -518,6 +542,18 @@ export declare class AIResource {
                 chunkType: "interrupt";
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "contextUsage";
+                contextUsagePercentage: number;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -719,6 +755,18 @@ export declare class AIResource {
                     chunkType: "interrupt";
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "contextUsage";
+                    contextUsagePercentage: number;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -900,6 +948,18 @@ export declare class AIResource {
                         chunkType: "interrupt";
                     };
                     role: "agent";
+                } | {
+                    conversationId: string;
+                    modelName: string;
+                    modelProviderId: string;
+                    runId: string;
+                    createdAt: import("temporal-polyfill").Temporal.Instant;
+                    id: string;
+                    content: {
+                        chunkType: "contextUsage";
+                        contextUsagePercentage: number;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -1096,6 +1156,18 @@ export declare class AIResource {
                 chunkType: "interrupt";
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "contextUsage";
+                contextUsagePercentage: number;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;

package/dist/common/fromTextEventStream.d.ts

@@ -1,8 +1,25 @@
 import { Observable } from "rxjs";
 import { type EventSourceMessage } from "eventsource-parser/stream";
 import type { SignalParam } from "./Params.ts";
+type TextEventStreamEvent = {
+    event: EventSourceMessage;
+    type: "event";
+};
+type TextEventStreamComment = {
+    comment: string;
+    type: "comment";
+};
+export type TextEventStreamActivity = TextEventStreamEvent | TextEventStreamComment;
 /**
  * Takes an instance of `Response` that will resolve to a `text/event-stream` and maps it
  * into an observable that emits for each event as it arrives
  */
 export declare const fromTextEventStream: (input: ({ signal }: SignalParam) => Promise<Response>) => Observable<EventSourceMessage>;
+/**
+ * Maps a `text/event-stream` response into all parsed stream activity.
+ *
+ * SSE comments are transport keepalive activity, not chat events. Use this helper when
+ * timeout/retry logic needs to observe comments before filtering them out.
+ */
+export declare const fromTextEventStreamActivity: (input: ({ signal }: SignalParam) => Promise<Response>) => Observable<TextEventStreamActivity>;
+export {};

package/dist/common/fromTextEventStream.js

@@ -13,14 +13,21 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { Observable } from "rxjs";
+import { filter, map, Observable } from "rxjs";
 import { EventSourceParserStream } from "eventsource-parser/stream";
 import { HttpError } from "./HttpError.js";
 /**
  * Takes an instance of `Response` that will resolve to a `text/event-stream` and maps it
  * into an observable that emits for each event as it arrives
  */
-export const fromTextEventStream = (input) => new Observable((subscriber) => {
+export const fromTextEventStream = (input) => fromTextEventStreamActivity(input).pipe(filter((activity) => activity.type === "event"), map((activity) => activity.event));
+/**
+ * Maps a `text/event-stream` response into all parsed stream activity.
+ *
+ * SSE comments are transport keepalive activity, not chat events. Use this helper when
+ * timeout/retry logic needs to observe comments before filtering them out.
+ */
+export const fromTextEventStreamActivity = (input) => new Observable((subscriber) => {
     /**
      * Create an internal AbortController so that the response stream
      * can be canceled early if the subscriber unsubscribes early
@@ -49,9 +56,13 @@ export const fromTextEventStream = (input) => new Observable((subscriber) => {
          * Pipe the raw byte stream through a text decoder, and then
          * through a parser that extracts individual SSE events
          */
-        return response.body
-            .pipeThrough(new TextDecoderStream())
-            .pipeThrough(new EventSourceParserStream());
+        return response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new EventSourceParserStream({
+            onComment(comment) {
+                if (!signal.aborted) {
+                    subscriber.next({ comment, type: "comment" });
+                }
+            },
+        }));
     })
         .then(async (readableStream) => {
         const reader = readableStream.getReader();
@@ -73,7 +84,7 @@ export const fromTextEventStream = (input) => new Observable((subscriber) => {
                 subscriber.complete();
                 break;
             }
-            subscriber.next(value);
+            subscriber.next({ event: value, type: "event" });
         }
         /**
          * Always call .cancel() on the reader even if the while loop

package/dist/common/fromTextEventStream.js.map

@@ -1 +1 @@
-{"version":3,"file":"fromTextEventStream.js","sourceRoot":"","sources":["../../src/common/fromTextEventStream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAClC,OAAO,EAA2B,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAG3C;;;GAGG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,KAAqD,EAAE,EAAE,CAC3F,IAAI,UAAU,CAAqB,CAAC,UAAU,EAAE,EAAE;IAChD;;;OAGG;IACH,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;IAE9B,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;SACd,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;QACvB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,MAAM,SAAS,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAED;;;WAGG;QACH,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,mBAAmB,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CACb,4DAA4D,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,GAAG,CACpG,CAAC;QACJ,CAAC;QAED;;WAEG;QACH,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED;;;WAGG;QACH,OAAO,QAAQ,CAAC,IAAI;aACjB,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;aACpC,WAAW,CAAC,IAAI,uBAAuB,EAAE,CAAC,CAAC;IAChD,CAAC,CAAC;SACD,IAAI,CAAC,KAAK,EAAE,cAAc,EAAE,EAAE;QAC7B,MAAM,MAAM,GAAG,cAAc,CAAC,SAAS,EAAE,CAAC;QAE1C;;;WAGG;QACH,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACvB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE5C;;;;eAIG;YACH,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM;YACR,CAAC;YAED,IAAI,IAAI,EAAE,CAAC;gBACT,UAAU,CAAC,QAAQ,EAAE,CAAC;gBACtB,MAAM;YACR,CAAC;YAED,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED;;;;WAIG;QACH,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;IACxB,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE;QACpB,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC,CAAC,CAAC;IAEL,OAAO,GAAG,EAAE;QACV,UAAU,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Observable } from \"rxjs\";\nimport { type EventSourceMessage, EventSourceParserStream } from \"eventsource-parser/stream\";\nimport { HttpError } from \"./HttpError.ts\";\nimport type { SignalParam } from \"./Params.ts\";\n\n/**\n * Takes an instance of `Response` that will resolve to a `text/event-stream` and maps it\n * into an observable that emits for each event as it arrives\n */\nexport const fromTextEventStream = (input: ({ signal }: SignalParam) => Promise<Response>) =>\n  new Observable<EventSourceMessage>((subscriber) => {\n    /**\n     * Create an internal AbortController so that the response stream\n     * can be canceled early if the subscriber unsubscribes early\n     */\n    const controller = new AbortController();\n    const { signal } = controller;\n\n    input({ signal })\n      .then(async (response) => {\n        if (!response.ok) {\n          throw await HttpError.fromResponse(response);\n        }\n\n        /**\n         * Even though the correct Content-Type header isn't strictly required,\n         * it's indicates either a client mistake or server implementation bug\n         */\n        if (response.headers.get(\"Content-Type\") !== \"text/event-stream\") {\n          throw new Error(\n            `Expected response type to be text/event-stream. Received ${response.headers.get(\"Content-Type\")}.`,\n          );\n        }\n\n        /**\n         * Response body should never be undefined for a text/event-stream\n         */\n        if (!response.body) {\n          throw new Error(`Response body was undefined`);\n        }\n\n        /**\n         * Pipe the raw byte stream through a text decoder, and then\n         * through a parser that extracts individual SSE events\n         */\n        return response.body\n          .pipeThrough(new TextDecoderStream())\n          .pipeThrough(new EventSourceParserStream());\n      })\n      .then(async (readableStream) => {\n        const reader = readableStream.getReader();\n\n        /**\n         * Check if the observable has been unsubscribed before reading\n         * each chunk\n         */\n        while (!signal.aborted) {\n          const { done, value } = await reader.read();\n\n          /**\n           * Check again after reading a chunk if the observable\n           * has been unsubscribed to avoid calling .next() or\n           * .complete() unnecessarily on the observable\n           */\n          if (signal.aborted) {\n            break;\n          }\n\n          if (done) {\n            subscriber.complete();\n            break;\n          }\n\n          subscriber.next(value);\n        }\n\n        /**\n         * Always call .cancel() on the reader even if the while loop\n         * was exited due to stream completion in order to indicate it\n         * can be fully cleaned up\n         */\n        await reader.cancel();\n      })\n      .catch((e: unknown) => {\n        subscriber.error(e);\n      });\n\n    return () => {\n      controller.abort();\n    };\n  });\n"]}
+{"version":3,"file":"fromTextEventStream.js","sourceRoot":"","sources":["../../src/common/fromTextEventStream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAC/C,OAAO,EAA2B,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAe3C;;;GAGG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,KAAqD,EAAE,EAAE,CAC3F,2BAA2B,CAAC,KAAK,CAAC,CAAC,IAAI,CACrC,MAAM,CAAC,CAAC,QAAQ,EAAoC,EAAE,CAAC,QAAQ,CAAC,IAAI,KAAK,OAAO,CAAC,EACjF,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAClC,CAAC;AAEJ;;;;;GAKG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,CACzC,KAAqD,EACrD,EAAE,CACF,IAAI,UAAU,CAA0B,CAAC,UAAU,EAAE,EAAE;IACrD;;;OAGG;IACH,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;IAE9B,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;SACd,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;QACvB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,MAAM,SAAS,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAED;;;WAGG;QACH,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,mBAAmB,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CACb,4DAA4D,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,GAAG,CACpG,CAAC;QACJ,CAAC;QAED;;WAEG;QACH,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED;;;WAGG;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC,CAAC,WAAW,CACnE,IAAI,uBAAuB,CAAC;YAC1B,SAAS,CAAC,OAAO;gBACf,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpB,UAAU,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;SACF,CAAC,CACH,CAAC;IACJ,CAAC,CAAC;SACD,IAAI,CAAC,KAAK,EAAE,cAAc,EAAE,EAAE;QAC7B,MAAM,MAAM,GAAG,cAAc,CAAC,SAAS,EAAE,CAAC;QAE1C;;;WAGG;QACH,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACvB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE5C;;;;eAIG;YACH,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM;YACR,CAAC;YAED,IAAI,IAAI,EAAE,CAAC;gBACT,UAAU,CAAC,QAAQ,EAAE,CAAC;gBACtB,MAAM;YACR,CAAC;YAED,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QACnD,CAAC;QAED;;;;WAIG;QACH,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;IACxB,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE;QACpB,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC,CAAC,CAAC;IAEL,OAAO,GAAG,EAAE;QACV,UAAU,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { filter, map, Observable } from \"rxjs\";\nimport { type EventSourceMessage, EventSourceParserStream } from \"eventsource-parser/stream\";\nimport { HttpError } from \"./HttpError.ts\";\nimport type { SignalParam } from \"./Params.ts\";\n\ntype TextEventStreamEvent = {\n  event: EventSourceMessage;\n  type: \"event\";\n};\n\ntype TextEventStreamComment = {\n  comment: string;\n  type: \"comment\";\n};\n\nexport type TextEventStreamActivity = TextEventStreamEvent | TextEventStreamComment;\n\n/**\n * Takes an instance of `Response` that will resolve to a `text/event-stream` and maps it\n * into an observable that emits for each event as it arrives\n */\nexport const fromTextEventStream = (input: ({ signal }: SignalParam) => Promise<Response>) =>\n  fromTextEventStreamActivity(input).pipe(\n    filter((activity): activity is TextEventStreamEvent => activity.type === \"event\"),\n    map((activity) => activity.event),\n  );\n\n/**\n * Maps a `text/event-stream` response into all parsed stream activity.\n *\n * SSE comments are transport keepalive activity, not chat events. Use this helper when\n * timeout/retry logic needs to observe comments before filtering them out.\n */\nexport const fromTextEventStreamActivity = (\n  input: ({ signal }: SignalParam) => Promise<Response>,\n) =>\n  new Observable<TextEventStreamActivity>((subscriber) => {\n    /**\n     * Create an internal AbortController so that the response stream\n     * can be canceled early if the subscriber unsubscribes early\n     */\n    const controller = new AbortController();\n    const { signal } = controller;\n\n    input({ signal })\n      .then(async (response) => {\n        if (!response.ok) {\n          throw await HttpError.fromResponse(response);\n        }\n\n        /**\n         * Even though the correct Content-Type header isn't strictly required,\n         * it's indicates either a client mistake or server implementation bug\n         */\n        if (response.headers.get(\"Content-Type\") !== \"text/event-stream\") {\n          throw new Error(\n            `Expected response type to be text/event-stream. Received ${response.headers.get(\"Content-Type\")}.`,\n          );\n        }\n\n        /**\n         * Response body should never be undefined for a text/event-stream\n         */\n        if (!response.body) {\n          throw new Error(`Response body was undefined`);\n        }\n\n        /**\n         * Pipe the raw byte stream through a text decoder, and then\n         * through a parser that extracts individual SSE events\n         */\n        return response.body.pipeThrough(new TextDecoderStream()).pipeThrough(\n          new EventSourceParserStream({\n            onComment(comment) {\n              if (!signal.aborted) {\n                subscriber.next({ comment, type: \"comment\" });\n              }\n            },\n          }),\n        );\n      })\n      .then(async (readableStream) => {\n        const reader = readableStream.getReader();\n\n        /**\n         * Check if the observable has been unsubscribed before reading\n         * each chunk\n         */\n        while (!signal.aborted) {\n          const { done, value } = await reader.read();\n\n          /**\n           * Check again after reading a chunk if the observable\n           * has been unsubscribed to avoid calling .next() or\n           * .complete() unnecessarily on the observable\n           */\n          if (signal.aborted) {\n            break;\n          }\n\n          if (done) {\n            subscriber.complete();\n            break;\n          }\n\n          subscriber.next({ event: value, type: \"event\" });\n        }\n\n        /**\n         * Always call .cancel() on the reader even if the while loop\n         * was exited due to stream completion in order to indicate it\n         * can be fully cleaned up\n         */\n        await reader.cancel();\n      })\n      .catch((e: unknown) => {\n        subscriber.error(e);\n      });\n\n    return () => {\n      controller.abort();\n    };\n  });\n"]}

package/dist/enterprise/ai/AIResource.d.ts

@@ -99,6 +99,18 @@ export declare class AIResource {
                     chunkType: "interrupt";
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "contextUsage";
+                    contextUsagePercentage: number;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -276,6 +288,18 @@ export declare class AIResource {
                         chunkType: "interrupt";
                     };
                     role: "agent";
+                } | {
+                    conversationId: string;
+                    modelName: string;
+                    modelProviderId: string;
+                    runId: string;
+                    createdAt: import("temporal-polyfill").Temporal.Instant;
+                    id: string;
+                    content: {
+                        chunkType: "contextUsage";
+                        contextUsagePercentage: number;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -460,6 +484,18 @@ export declare class AIResource {
                 chunkType: "interrupt";
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "contextUsage";
+                contextUsagePercentage: number;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -661,6 +697,18 @@ export declare class AIResource {
                     chunkType: "interrupt";
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "contextUsage";
+                    contextUsagePercentage: number;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -842,6 +890,18 @@ export declare class AIResource {
                         chunkType: "interrupt";
                     };
                     role: "agent";
+                } | {
+                    conversationId: string;
+                    modelName: string;
+                    modelProviderId: string;
+                    runId: string;
+                    createdAt: import("temporal-polyfill").Temporal.Instant;
+                    id: string;
+                    content: {
+                        chunkType: "contextUsage";
+                        contextUsagePercentage: number;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -1038,6 +1098,18 @@ export declare class AIResource {
                 chunkType: "interrupt";
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "contextUsage";
+                contextUsagePercentage: number;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;

package/dist/enterprise/ai/chat/UserChatMessage.d.ts

@@ -27,6 +27,7 @@ export declare class UserChatMessage {
                 }[];
             } | undefined;
             context?: string | undefined;
+            fileAttachmentIds?: string[] | undefined;
             skillIds?: string[] | undefined;
         }, "text"> | undefined;
     };
@@ -47,6 +48,7 @@ export declare const createConversationPromptSchema: z.ZodMiniObject<{
         }, z.core.$strip>>;
     }, z.core.$strip>>;
     context: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    fileAttachmentIds: z.ZodMiniOptional<z.ZodMiniArray<z.ZodMiniString<string>>>;
     skillIds: z.ZodMiniOptional<z.ZodMiniArray<z.ZodMiniString<string>>>;
     text: z.ZodMiniString<string>;
 }, z.core.$strict>;

package/dist/enterprise/ai/chat/UserChatMessage.js

@@ -57,6 +57,7 @@ export const createConversationPromptSchema = z.strictObject({
         })),
     })),
     context: z.optional(z.string()),
+    fileAttachmentIds: z.optional(z.array(z.string())),
     skillIds: z.optional(z.array(z.string())),
     text: z.string(),
 });

package/dist/enterprise/ai/chat/UserChatMessage.js.map

@@ -1 +1 @@
-{"version":3,"file":"UserChatMessage.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/UserChatMessage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,OAAO,eAAe;IACjB,SAAS,CAAyB;IAClC,OAAO,CAAS;IAChB,EAAE,CAAS;IACX,MAAM,CAAgE;IAE/E,YACE,EAAyB,EACzB,SAAuC,EACvC,OAAmC,EACnC,MAAkC;QAElC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,UAAU,CAAC,MAAiC;QAC1C,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,MAAM;QACJ,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE;YACpC,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,GAAG,CAAC,OAAmC,EAAE,MAAkC;QAChF,OAAO,IAAI,eAAe,CAAC,MAAM,EAAE,EAAE,QAAQ,CAAC,GAAG,CAAC,gBAAgB,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACzF,CAAC;CACF;AAED,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,CAAC,YAAY,CAAC;IAC3D,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAChD,SAAS,EAAE,CAAC,CAAC,QAAQ,CACnB,CAAC,CAAC,MAAM,CAAC;QACP,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE;QAC5B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;QACzB,aAAa,EAAE,CAAC,CAAC,KAAK,CACpB,CAAC,CAAC,MAAM,CAAC;YACP,mBAAmB,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5C,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE;YACrB,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5C,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5C,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;YACvB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;SACjB,CAAC,CACH;KACF,CAAC,CACH;IACD,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC/B,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IACzC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { nanoid } from \"nanoid/non-secure\";\nimport { Temporal } from \"temporal-polyfill\";\nimport * as z from \"zod/mini\";\n\nexport class UserChatMessage {\n  readonly createdAt: Temporal.ZonedDateTime;\n  readonly content: string;\n  readonly id: string;\n  readonly prompt?: Omit<z.infer<typeof createConversationPromptSchema>, \"text\">;\n\n  constructor(\n    id: UserChatMessage[\"id\"],\n    createdAt: UserChatMessage[\"createdAt\"],\n    content: UserChatMessage[\"content\"],\n    prompt?: UserChatMessage[\"prompt\"],\n  ) {\n    this.id = id;\n    this.createdAt = createdAt;\n    this.content = content;\n    this.prompt = prompt;\n  }\n\n  withPrompt(prompt: UserChatMessage[\"prompt\"]) {\n    return new UserChatMessage(this.id, this.createdAt, this.content, prompt);\n  }\n\n  toJSON() {\n    return {\n      content: this.content,\n      createdAt: this.createdAt.toString(),\n      id: this.id,\n      prompt: this.prompt,\n    };\n  }\n\n  static new(content: UserChatMessage[\"content\"], prompt?: UserChatMessage[\"prompt\"]) {\n    return new UserChatMessage(nanoid(), Temporal.Now.zonedDateTimeISO(), content, prompt);\n  }\n}\n\nexport const createConversationPromptSchema = z.strictObject({\n  acceptTaskNames: z.optional(z.array(z.string())),\n  approvals: z.optional(\n    z.object({\n      allToolsAllowed: z.boolean(),\n      approvalNonce: z.string(),\n      toolDecisions: z.array(\n        z.object({\n          allowInConversation: z.optional(z.boolean()),\n          approved: z.boolean(),\n          arguments: z.record(z.string(), z.unknown()),\n          catalogPath: z.optional(z.array(z.string())),\n          executionId: z.string(),\n          name: z.string(),\n        }),\n      ),\n    }),\n  ),\n  context: z.optional(z.string()),\n  skillIds: z.optional(z.array(z.string())),\n  text: z.string(),\n});\n"]}
+{"version":3,"file":"UserChatMessage.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/UserChatMessage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,OAAO,eAAe;IACjB,SAAS,CAAyB;IAClC,OAAO,CAAS;IAChB,EAAE,CAAS;IACX,MAAM,CAAgE;IAE/E,YACE,EAAyB,EACzB,SAAuC,EACvC,OAAmC,EACnC,MAAkC;QAElC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,UAAU,CAAC,MAAiC;QAC1C,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,MAAM;QACJ,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE;YACpC,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,GAAG,CAAC,OAAmC,EAAE,MAAkC;QAChF,OAAO,IAAI,eAAe,CAAC,MAAM,EAAE,EAAE,QAAQ,CAAC,GAAG,CAAC,gBAAgB,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACzF,CAAC;CACF;AAED,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,CAAC,YAAY,CAAC;IAC3D,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAChD,SAAS,EAAE,CAAC,CAAC,QAAQ,CACnB,CAAC,CAAC,MAAM,CAAC;QACP,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE;QAC5B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;QACzB,aAAa,EAAE,CAAC,CAAC,KAAK,CACpB,CAAC,CAAC,MAAM,CAAC;YACP,mBAAmB,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5C,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE;YACrB,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5C,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5C,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;YACvB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;SACjB,CAAC,CACH;KACF,CAAC,CACH;IACD,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC/B,iBAAiB,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAClD,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IACzC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { nanoid } from \"nanoid/non-secure\";\nimport { Temporal } from \"temporal-polyfill\";\nimport * as z from \"zod/mini\";\n\nexport class UserChatMessage {\n  readonly createdAt: Temporal.ZonedDateTime;\n  readonly content: string;\n  readonly id: string;\n  readonly prompt?: Omit<z.infer<typeof createConversationPromptSchema>, \"text\">;\n\n  constructor(\n    id: UserChatMessage[\"id\"],\n    createdAt: UserChatMessage[\"createdAt\"],\n    content: UserChatMessage[\"content\"],\n    prompt?: UserChatMessage[\"prompt\"],\n  ) {\n    this.id = id;\n    this.createdAt = createdAt;\n    this.content = content;\n    this.prompt = prompt;\n  }\n\n  withPrompt(prompt: UserChatMessage[\"prompt\"]) {\n    return new UserChatMessage(this.id, this.createdAt, this.content, prompt);\n  }\n\n  toJSON() {\n    return {\n      content: this.content,\n      createdAt: this.createdAt.toString(),\n      id: this.id,\n      prompt: this.prompt,\n    };\n  }\n\n  static new(content: UserChatMessage[\"content\"], prompt?: UserChatMessage[\"prompt\"]) {\n    return new UserChatMessage(nanoid(), Temporal.Now.zonedDateTimeISO(), content, prompt);\n  }\n}\n\nexport const createConversationPromptSchema = z.strictObject({\n  acceptTaskNames: z.optional(z.array(z.string())),\n  approvals: z.optional(\n    z.object({\n      allToolsAllowed: z.boolean(),\n      approvalNonce: z.string(),\n      toolDecisions: z.array(\n        z.object({\n          allowInConversation: z.optional(z.boolean()),\n          approved: z.boolean(),\n          arguments: z.record(z.string(), z.unknown()),\n          catalogPath: z.optional(z.array(z.string())),\n          executionId: z.string(),\n          name: z.string(),\n        }),\n      ),\n    }),\n  ),\n  context: z.optional(z.string()),\n  fileAttachmentIds: z.optional(z.array(z.string())),\n  skillIds: z.optional(z.array(z.string())),\n  text: z.string(),\n});\n"]}

package/dist/enterprise/ai/chat/chatEventSchema.d.ts

@@ -82,6 +82,15 @@ export declare const chatEventCodec: z.ZodMiniCodec<z.ZodMiniDiscriminatedUnion<
     modelName: z.ZodMiniString<string>;
     modelProviderId: z.ZodMiniString<string>;
     runId: z.ZodMiniString<string>;
+}, z.core.$strip>, z.ZodMiniObject<{
+    chunkType: z.ZodMiniLiteral<"contextUsage">;
+    contextUsagePercentage: z.ZodMiniNumber<number>;
+    conversationId: z.ZodMiniString<string>;
+    createdAt: z.ZodMiniString<string>;
+    messageId: z.ZodMiniString<string>;
+    modelName: z.ZodMiniString<string>;
+    modelProviderId: z.ZodMiniString<string>;
+    runId: z.ZodMiniString<string>;
 }, z.core.$strip>, z.ZodMiniObject<{
     chunkType: z.ZodMiniLiteral<"conversationUpdate">;
     summary: z.ZodMiniOptional<z.ZodMiniString<string>>;
@@ -230,6 +239,18 @@ export declare const chatEventCodec: z.ZodMiniCodec<z.ZodMiniDiscriminatedUnion<
         chunkType: z.ZodMiniLiteral<"interrupt">;
     }, z.core.$strip>;
     role: z.ZodMiniLiteral<"agent">;
+}, z.core.$strict>, z.ZodMiniObject<{
+    conversationId: z.ZodMiniString<string>;
+    modelName: z.ZodMiniString<string>;
+    modelProviderId: z.ZodMiniString<string>;
+    runId: z.ZodMiniString<string>;
+    createdAt: z.ZodMiniCustom<Temporal.Instant, Temporal.Instant>;
+    id: z.ZodMiniString<string>;
+    content: z.ZodMiniObject<{
+        chunkType: z.ZodMiniLiteral<"contextUsage">;
+        contextUsagePercentage: z.ZodMiniNumber<number>;
+    }, z.core.$strip>;
+    role: z.ZodMiniLiteral<"agent">;
 }, z.core.$strict>, z.ZodMiniObject<{
     conversationId: z.ZodMiniString<string>;
     modelName: z.ZodMiniString<string>;

package/dist/enterprise/ai/chat/chatEventSchema.js

@@ -118,6 +118,10 @@ const jobUpdateChunkSchema = z.object({
     toolExecutionId: z.string(),
     toolName: z.string(),
 });
+const contextUsageChunkSchema = z.object({
+    chunkType: z.literal("contextUsage"),
+    contextUsagePercentage: z.number().check(z.gte(0), z.lte(100)),
+});
 const selectedSkillInfoSchema = z.object({
     id: z.string(),
     name: z.string(),
@@ -149,6 +153,7 @@ const chatEventInputSchema = z.discriminatedUnion("chunkType", [
     z.extend(errorChunkV2Schema, chatEventSharedSchema),
     endOfStreamInputSchema,
     z.extend(interruptChunkSchema, chatEventSharedSchema),
+    z.extend(contextUsageChunkSchema, chatEventSharedSchema),
     z.extend(conversationUpdateChunkSchema, chatEventSharedSchema),
     z.extend(jobUpdateChunkSchema, chatEventSharedSchema),
     z.extend(modelChunkSchema, chatEventSharedSchema),
@@ -179,6 +184,10 @@ const chatEventOutputSchema = z.union([
         content: interruptChunkSchema,
         role: z.literal("agent"),
     }),
+    z.extend(chatEventOutputSharedSchema, {
+        content: contextUsageChunkSchema,
+        role: z.literal("agent"),
+    }),
     z.extend(chatEventOutputSharedSchema, {
         content: jobUpdateChunkSchema,
         role: z.literal("agent"),
@@ -231,6 +240,15 @@ export const chatEventCodec = z.codec(chatEventInputSchema, chatEventOutputSchem
             runId: v.runId,
         };
         switch (v.chunkType) {
+            case "contextUsage":
+                return {
+                    ...sharedProperties,
+                    content: {
+                        chunkType: "contextUsage",
+                        contextUsagePercentage: v.contextUsagePercentage,
+                    },
+                    role: "agent",
+                };
             case "conversationUpdate":
                 return {
                     ...sharedProperties,

package/dist/enterprise/ai/chat/chatEventSchema.js.map

@@ -1 +1 @@
-{"version":3,"file":"chatEventSchema.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/chatEventSchema.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B;;;GAGG;AACH,MAAM,uBAAuB,GAAG;IAC9B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACrD,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC3D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;CACtD,CAAC;AAEF;;GAEG;AACH,MAAM,qBAAqB,GAAG;IAC5B,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;IAC3B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CAClB,CAAC;AAEF,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IAC7B;;;OAGG;IACH,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;CACpD,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IAC7B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACnD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAEH,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC;IAC1C,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC/B,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAC9B,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;CACpC,CAAC,CAAC;AAEH,qGAAqG;AACrG,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;IACnC,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;IAC3B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CAClB,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;CAClC,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IAC7B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAChD;;;;;OAKG;IACH,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;CAC1C,CAAC,CAAC;AAEH,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC;IACvC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAEH,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAClD,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC;IACrC,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;IACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AACH,MAAM,0BAA0B,GAAG,wBAAwB,CAAC;AAC5D,MAAM,4BAA4B,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE,qBAAqB,CAAC,CAAC;AAC/F,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;AAEzD,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC5C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAClD,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;IACnC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;IACvC,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACvC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;IACpC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;CAC1C,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;IACjC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;IAC3B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;CACrB,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAIH,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;IACtC,cAAc,EAAE,CAAC,CAAC,KAAK,CAAC,uBAAuB,CAAC;CACjD,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;IACnC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAEH;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,kBAAkB,CAAC,WAAW,EAAE;IACjE,gBAAgB;IAChB,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE,uBAAuB,CAAC;IACnD,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,uBAAuB,CAAC;IACzD,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE,uBAAuB,CAAC;CAC3D,CAAC,CAAC;AAOH,MAAM,oBAAoB,GAAG,CAAC,CAAC,kBAAkB,CAAC,WAAW,EAAE;IAC7D,CAAC,CAAC,MAAM,CAAC,kBAAkB,EAAE,qBAAqB,CAAC;IACnD,sBAAsB;IACtB,CAAC,CAAC,MAAM,CAAC,oBAAoB,EAAE,qBAAqB,CAAC;IACrD,CAAC,CAAC,MAAM,CAAC,6BAA6B,EAAE,qBAAqB,CAAC;IAC9D,CAAC,CAAC,MAAM,CAAC,oBAAoB,EAAE,qBAAqB,CAAC;IACrD,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE,qBAAqB,CAAC;IACjD,CAAC,CAAC,MAAM,CAAC,yBAAyB,EAAE,qBAAqB,CAAC;IAC1D,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,qBAAqB,CAAC;IAC3D,4BAA4B;IAC5B,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,qBAAqB,CAAC;IACvD,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE,qBAAqB,CAAC;IACxD,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,qBAAqB,CAAC;CACxD,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAC1C,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,qBAAqB,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,EACnF;IACE,SAAS,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC;IACzC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;CACf,CACF,CAAC;AAEF,uEAAuE;AACvE,MAAM,uBAAuB,GAAG,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;AAElF,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CAAC;IACpC,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,6BAA6B;QACtC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAChG,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE;QAChC,OAAO,EAAE,sBAAsB;QAC/B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,oBAAoB;QAC7B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,oBAAoB;QAC7B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAC9F,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,yBAAyB;QAClC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,0BAA0B;QACnC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,0BAA0B;QACnC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,sBAAsB;QAC/B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,uBAAuB;QAChC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,sBAAsB;QAC/B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;KACxB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,oBAAoB,EAAE,qBAAqB,EAAE;IACjF,MAAM,CAAC,CAAC;QACN,IAAI,CAAC,CAAC,SAAS,KAAK,aAAa,EAAE,CAAC;YAClC,OAAO;gBACL,OAAO,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE;gBACrC,cAAc,EAAE,CAAC,CAAC,cAAc;gBAChC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,eAAe,EAAE,CAAC,CAAC,eAAe;gBAClC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,CAAC,KAAK;aAC2C,CAAC;QAC9D,CAAC;QAED,MAAM,gBAAgB,GAAG;YACvB,cAAc,EAAE,CAAC,CAAC,cAAc;YAChC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAC7C,EAAE,EAAE,CAAC,CAAC,SAAS;YACf,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;YAClC,KAAK,EAAE,CAAC,CAAC,KAAK;SACf,CAAC;QAEF,QAAQ,CAAC,CAAC,SAAS,EAAE,CAAC;YACpB,KAAK,oBAAoB;gBACvB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,oBAAoB;wBAC/B,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,KAAK,EAAE,CAAC,CAAC,KAAK;qBACf;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,OAAO;gBACV,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,OAAO;wBAClB,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;qBACb;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,WAAW;gBACd,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,WAAW;qBACvB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,WAAW;gBACd,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,WAAW;wBACtB,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,QAAQ,EAAE,CAAC,CAAC,QAAQ;wBACpB,eAAe,EAAE,CAAC,CAAC,eAAe;wBAClC,QAAQ,EAAE,CAAC,CAAC,QAAQ;qBACrB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,OAAO;gBACV,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,OAAO;wBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,MAAM,EAAE,CAAC,CAAC,MAAM;qBACjB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,gBAAgB;gBACnB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,gBAAgB;wBAC3B,cAAc,EAAE,CAAC,CAAC,cAAc;qBACjC;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,iBAAiB;gBACpB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,iBAAiB;wBAC5B,QAAQ,EAAE,CAAC,CAAC,QAAQ;wBACpB,IAAI,EAAE,CAAC,CAAC,IAAI;qBACb;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,eAAe;gBAClB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,SAAS,EAAE,eAAe;wBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;qBACb;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,aAAa;gBAChB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,CAAC,CAAC,SAAS;wBACtB,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,SAAS,EAAE,aAAa;wBACxB,UAAU,EAAE,CAAC,CAAC,UAAU;wBACxB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;wBAC1B,eAAe,EAAE,CAAC,CAAC,eAAe;wBAClC,UAAU,EAAE,CAAC,CAAC,UAAU;qBACzB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,cAAc;gBACjB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,SAAS,EAAE,cAAc;wBACzB,UAAU,EAAE,CAAC,CAAC,UAAU;wBACxB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,MAAM,EAAE,CAAC,CAAC,MAAM;qBACjB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,aAAa;gBAChB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;oBACnD,IAAI,EAAE,MAAM;iBAC6C,CAAC;QAChE,CAAC;IACH,CAAC;IACD,MAAM,CAAC,CAAC;QACN,IAAI,CAAC,CAAC,OAAO,CAAC,SAAS,KAAK,aAAa,EAAE,CAAC;YAC1C,OAAO;gBACL,SAAS,EAAE,aAAa;gBACxB,cAAc,EAAE,CAAC,CAAC,cAAc;gBAChC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;gBACjC,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,eAAe,EAAE,CAAC,CAAC,eAAe;gBAClC,KAAK,EAAE,CAAC,CAAC,KAAK;aACyC,CAAC;QAC5D,CAAC;QACD,uFAAuF;QACvF,MAAM,EAAE,GAAI,CAAqE,CAAC,EAAE,CAAC;QACrF,OAAO;YACL,cAAc,EAAE,CAAC,CAAC,cAAc;YAChC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;YACjC,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;YAClC,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,GAAG,CAAC,CAAC,OAAO;SAC2C,CAAC;IAC5D,CAAC;CACF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Temporal } from \"temporal-polyfill\";\nimport * as z from \"zod/mini\";\n\n/**\n * Shared chat event fields for all responses (except error response)\n * @deprecated\n */\nconst chatEventV1SharedSchema = {\n  createdAt: z.string().check(z.iso.datetime()),\n  modelName: z.string().check(z.trim(), z.minLength(1)),\n  modelProviderId: z.string().check(z.trim(), z.minLength(1)),\n  sessionId: z.string().check(z.trim(), z.minLength(1)),\n};\n\n/**\n * Shared chat event fields for all responses (except error response)\n */\nconst chatEventSharedSchema = {\n  conversationId: z.string(),\n  createdAt: z.string().check(z.iso.datetime()),\n  messageId: z.string(),\n  modelName: z.string(),\n  modelProviderId: z.string(),\n  runId: z.string(),\n};\n\nconst errorChunkSchema = z.object({\n  chunkType: z.literal(\"error\"),\n  /**\n   * If the message is not present or very short, it won't be useful to show\n   * anyways, so the parser should throw to the generic response error handler.\n   */\n  message: z.string().check(z.trim(), z.minLength(2)),\n});\n\nconst errorChunkV2Schema = z.object({\n  chunkType: z.literal(\"error\"),\n  message: z.string().check(z.trim(), z.minLength(2)),\n  type: z.string(),\n});\n\nconst conversationUpdateChunkSchema = z.object({\n  chunkType: z.literal(\"conversationUpdate\"),\n  summary: z.optional(z.string()),\n  title: z.optional(z.string()),\n});\n\nconst endOfStreamChunkSchema = z.object({\n  chunkType: z.literal(\"endOfStream\"),\n});\n\n// endOfStream omits messageId: the backend sends it absent or as \"\", neither maps to a meaningful id\nconst endOfStreamInputSchema = z.object({\n  chunkType: z.literal(\"endOfStream\"),\n  conversationId: z.string(),\n  createdAt: z.string().check(z.iso.datetime()),\n  modelName: z.string(),\n  modelProviderId: z.string(),\n  runId: z.string(),\n});\n\nconst interruptChunkSchema = z.object({\n  chunkType: z.literal(\"interrupt\"),\n});\n\nconst modelChunkSchema = z.object({\n  chunkType: z.literal(\"model\"),\n  name: z.string().check(z.trim(), z.minLength(1)),\n  /**\n   * It's useful to indicate (to the type system) that the result contents\n   * at this stage can only be data (de)serializable from JSON. Because\n   * model responses can vary depending on runtime flags and edition, we can\n   * leave this field generic and refine it more precisely in later stages.\n   */\n  result: z.record(z.string(), z.unknown()),\n});\n\nconst sandboxProgressChunkSchema = z.object({\n  chunkType: z.literal(\"sandboxProgress\"),\n  stepName: z.string(),\n  text: z.string(),\n});\n\nconst sandboxStdoutChunkSchema = z.object({\n  callId: z.string().check(z.trim(), z.minLength(1)),\n  chunkType: z.literal(\"sandboxStdout\"),\n  isFinal: z.boolean(),\n  text: z.string(),\n});\nconst sandboxStdoutContentSchema = sandboxStdoutChunkSchema;\nconst sandboxStdoutChatEventSchema = z.extend(sandboxStdoutChunkSchema, chatEventSharedSchema);\nconst toolPlanSchema = z.record(z.string(), z.unknown());\n\nconst toolRequestChunkSchema = z.object({\n  arguments: z.record(z.string(), z.unknown()),\n  callId: z.string().check(z.trim(), z.minLength(1)),\n  chunkType: z.literal(\"toolRequest\"),\n  commentary: z.optional(z.string()),\n  name: z.string(),\n  replacePlan: z.optional(toolPlanSchema),\n  summarizedTitle: z.optional(z.string()),\n  updatePlan: z.optional(toolPlanSchema),\n});\n\nconst toolResponseChunkSchema = z.object({\n  callId: z.string(),\n  chunkType: z.literal(\"toolResponse\"),\n  commentary: z.optional(z.string()),\n  name: z.string(),\n  result: z.record(z.string(), z.unknown()),\n});\n\nconst jobUpdateChunkSchema = z.object({\n  chunkType: z.literal(\"jobUpdate\"),\n  jobId: z.string(),\n  jobState: z.string(),\n  toolExecutionId: z.string(),\n  toolName: z.string(),\n});\n\nconst selectedSkillInfoSchema = z.object({\n  id: z.string(),\n  name: z.string(),\n});\n\nexport type SelectedSkillInfo = z.infer<typeof selectedSkillInfoSchema>;\n\nconst selectedSkillsChunkSchema = z.object({\n  chunkType: z.literal(\"selectedSkills\"),\n  selectedSkills: z.array(selectedSkillInfoSchema),\n});\n\nconst userMessageChunkSchema = z.object({\n  chunkType: z.literal(\"userMessage\"),\n  text: z.string(),\n});\n\n/**\n * Captures all of the possible chunks of data that the Agent can reply\n * with. We're lenient to extra/unexpected fields being present (`z.object`\n * instead of `z.strictObject`), but for the fields that are expected,\n * we validate that they match the most general type category they belong in.\n * For example, it's good enough to validate that the `callId` is a string\n * for type safety -- we don't need to verify that it's also a `z.uuid()`.\n * @deprecated\n */\nexport const chatEventV1Schema = z.discriminatedUnion(\"chunkType\", [\n  errorChunkSchema,\n  z.extend(modelChunkSchema, chatEventV1SharedSchema),\n  z.extend(toolRequestChunkSchema, chatEventV1SharedSchema),\n  z.extend(toolResponseChunkSchema, chatEventV1SharedSchema),\n]);\n\n/**\n * @deprecated\n */\nexport type ChatEventV1 = z.infer<typeof chatEventV1Schema>;\n\nconst chatEventInputSchema = z.discriminatedUnion(\"chunkType\", [\n  z.extend(errorChunkV2Schema, chatEventSharedSchema),\n  endOfStreamInputSchema,\n  z.extend(interruptChunkSchema, chatEventSharedSchema),\n  z.extend(conversationUpdateChunkSchema, chatEventSharedSchema),\n  z.extend(jobUpdateChunkSchema, chatEventSharedSchema),\n  z.extend(modelChunkSchema, chatEventSharedSchema),\n  z.extend(selectedSkillsChunkSchema, chatEventSharedSchema),\n  z.extend(sandboxProgressChunkSchema, chatEventSharedSchema),\n  sandboxStdoutChatEventSchema,\n  z.extend(toolRequestChunkSchema, chatEventSharedSchema),\n  z.extend(toolResponseChunkSchema, chatEventSharedSchema),\n  z.extend(userMessageChunkSchema, chatEventSharedSchema),\n]);\n\nconst chatEventOutputSharedSchema = z.extend(\n  z.omit(z.strictObject(chatEventSharedSchema), { createdAt: true, messageId: true }),\n  {\n    createdAt: z.instanceof(Temporal.Instant),\n    id: z.string(),\n  },\n);\n\n// endOfStream has no meaningful messageId on input, so no id on output\nconst endOfStreamOutputSchema = z.omit(chatEventOutputSharedSchema, { id: true });\n\nconst chatEventOutputSchema = z.union([\n  z.extend(chatEventOutputSharedSchema, {\n    content: conversationUpdateChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, { content: errorChunkV2Schema, role: z.literal(\"agent\") }),\n  z.extend(endOfStreamOutputSchema, {\n    content: endOfStreamChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: interruptChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: jobUpdateChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, { content: modelChunkSchema, role: z.literal(\"agent\") }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: selectedSkillsChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: sandboxProgressChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: sandboxStdoutContentSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: toolRequestChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: toolResponseChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: userMessageChunkSchema,\n    role: z.literal(\"user\"),\n  }),\n]);\n\nexport const chatEventCodec = z.codec(chatEventInputSchema, chatEventOutputSchema, {\n  decode(v) {\n    if (v.chunkType === \"endOfStream\") {\n      return {\n        content: { chunkType: \"endOfStream\" },\n        conversationId: v.conversationId,\n        createdAt: Temporal.Instant.from(v.createdAt),\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n        role: \"agent\",\n        runId: v.runId,\n      } as const satisfies z.output<typeof chatEventOutputSchema>;\n    }\n\n    const sharedProperties = {\n      conversationId: v.conversationId,\n      createdAt: Temporal.Instant.from(v.createdAt),\n      id: v.messageId,\n      modelName: v.modelName,\n      modelProviderId: v.modelProviderId,\n      runId: v.runId,\n    };\n\n    switch (v.chunkType) {\n      case \"conversationUpdate\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"conversationUpdate\",\n            summary: v.summary,\n            title: v.title,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"error\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"error\",\n            message: v.message,\n            type: v.type,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"interrupt\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"interrupt\",\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"jobUpdate\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"jobUpdate\",\n            jobId: v.jobId,\n            jobState: v.jobState,\n            toolExecutionId: v.toolExecutionId,\n            toolName: v.toolName,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"model\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"model\",\n            name: v.name,\n            result: v.result,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"selectedSkills\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"selectedSkills\",\n            selectedSkills: v.selectedSkills,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"sandboxProgress\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"sandboxProgress\",\n            stepName: v.stepName,\n            text: v.text,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"sandboxStdout\":\n        return {\n          ...sharedProperties,\n          content: {\n            callId: v.callId,\n            chunkType: \"sandboxStdout\",\n            isFinal: v.isFinal,\n            text: v.text,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"toolRequest\":\n        return {\n          ...sharedProperties,\n          content: {\n            arguments: v.arguments,\n            callId: v.callId,\n            chunkType: \"toolRequest\",\n            commentary: v.commentary,\n            name: v.name,\n            replacePlan: v.replacePlan,\n            summarizedTitle: v.summarizedTitle,\n            updatePlan: v.updatePlan,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"toolResponse\":\n        return {\n          ...sharedProperties,\n          content: {\n            callId: v.callId,\n            chunkType: \"toolResponse\",\n            commentary: v.commentary,\n            name: v.name,\n            result: v.result,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"userMessage\":\n        return {\n          ...sharedProperties,\n          content: { chunkType: \"userMessage\", text: v.text },\n          role: \"user\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n    }\n  },\n  encode(v) {\n    if (v.content.chunkType === \"endOfStream\") {\n      return {\n        chunkType: \"endOfStream\",\n        conversationId: v.conversationId,\n        createdAt: v.createdAt.toString(),\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n        runId: v.runId,\n      } as const satisfies z.infer<typeof chatEventInputSchema>;\n    }\n    // Safe: endOfStream (the only variant without id) is handled by the early return above\n    const id = (v as Extract<z.output<typeof chatEventOutputSchema>, { id: string }>).id;\n    return {\n      conversationId: v.conversationId,\n      createdAt: v.createdAt.toString(),\n      messageId: id,\n      modelName: v.modelName,\n      modelProviderId: v.modelProviderId,\n      runId: v.runId,\n      ...v.content,\n    } as const satisfies z.infer<typeof chatEventInputSchema>;\n  },\n});\n\nexport type ChatEvent = z.output<typeof chatEventCodec>;\nexport type ChatEventWithChunkType<T extends z.input<typeof chatEventCodec>[\"chunkType\"]> = Extract<\n  ChatEvent,\n  { content: { chunkType: T } }\n>;\n"]}
+{"version":3,"file":"chatEventSchema.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/chatEventSchema.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B;;;GAGG;AACH,MAAM,uBAAuB,GAAG;IAC9B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACrD,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC3D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;CACtD,CAAC;AAEF;;GAEG;AACH,MAAM,qBAAqB,GAAG;IAC5B,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;IAC3B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CAClB,CAAC;AAEF,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IAC7B;;;OAGG;IACH,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;CACpD,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IAC7B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACnD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAEH,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC;IAC1C,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC/B,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAC9B,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;CACpC,CAAC,CAAC;AAEH,qGAAqG;AACrG,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;IACnC,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;IAC3B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CAClB,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;CAClC,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IAC7B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAChD;;;;;OAKG;IACH,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;CAC1C,CAAC,CAAC;AAEH,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC;IACvC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAEH,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAClD,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC;IACrC,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;IACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AACH,MAAM,0BAA0B,GAAG,wBAAwB,CAAC;AAC5D,MAAM,4BAA4B,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE,qBAAqB,CAAC,CAAC;AAC/F,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;AAEzD,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC5C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAClD,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;IACnC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;IACvC,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACvC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;IACpC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;CAC1C,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;IACjC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;IAC3B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;CACrB,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;IACpC,sBAAsB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;CAC/D,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAIH,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;IACtC,cAAc,EAAE,CAAC,CAAC,KAAK,CAAC,uBAAuB,CAAC;CACjD,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;IACnC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC;AAEH;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,kBAAkB,CAAC,WAAW,EAAE;IACjE,gBAAgB;IAChB,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE,uBAAuB,CAAC;IACnD,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,uBAAuB,CAAC;IACzD,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE,uBAAuB,CAAC;CAC3D,CAAC,CAAC;AAOH,MAAM,oBAAoB,GAAG,CAAC,CAAC,kBAAkB,CAAC,WAAW,EAAE;IAC7D,CAAC,CAAC,MAAM,CAAC,kBAAkB,EAAE,qBAAqB,CAAC;IACnD,sBAAsB;IACtB,CAAC,CAAC,MAAM,CAAC,oBAAoB,EAAE,qBAAqB,CAAC;IACrD,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE,qBAAqB,CAAC;IACxD,CAAC,CAAC,MAAM,CAAC,6BAA6B,EAAE,qBAAqB,CAAC;IAC9D,CAAC,CAAC,MAAM,CAAC,oBAAoB,EAAE,qBAAqB,CAAC;IACrD,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE,qBAAqB,CAAC;IACjD,CAAC,CAAC,MAAM,CAAC,yBAAyB,EAAE,qBAAqB,CAAC;IAC1D,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,qBAAqB,CAAC;IAC3D,4BAA4B;IAC5B,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,qBAAqB,CAAC;IACvD,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE,qBAAqB,CAAC;IACxD,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,qBAAqB,CAAC;CACxD,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAC1C,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,qBAAqB,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,EACnF;IACE,SAAS,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC;IACzC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;CACf,CACF,CAAC;AAEF,uEAAuE;AACvE,MAAM,uBAAuB,GAAG,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;AAElF,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CAAC;IACpC,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,6BAA6B;QACtC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAChG,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE;QAChC,OAAO,EAAE,sBAAsB;QAC/B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,oBAAoB;QAC7B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,uBAAuB;QAChC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,oBAAoB;QAC7B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAC9F,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,yBAAyB;QAClC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,0BAA0B;QACnC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,0BAA0B;QACnC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,sBAAsB;QAC/B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,uBAAuB;QAChC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;KACzB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE;QACpC,OAAO,EAAE,sBAAsB;QAC/B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;KACxB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,oBAAoB,EAAE,qBAAqB,EAAE;IACjF,MAAM,CAAC,CAAC;QACN,IAAI,CAAC,CAAC,SAAS,KAAK,aAAa,EAAE,CAAC;YAClC,OAAO;gBACL,OAAO,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE;gBACrC,cAAc,EAAE,CAAC,CAAC,cAAc;gBAChC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,eAAe,EAAE,CAAC,CAAC,eAAe;gBAClC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,CAAC,KAAK;aAC2C,CAAC;QAC9D,CAAC;QAED,MAAM,gBAAgB,GAAG;YACvB,cAAc,EAAE,CAAC,CAAC,cAAc;YAChC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAC7C,EAAE,EAAE,CAAC,CAAC,SAAS;YACf,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;YAClC,KAAK,EAAE,CAAC,CAAC,KAAK;SACf,CAAC;QAEF,QAAQ,CAAC,CAAC,SAAS,EAAE,CAAC;YACpB,KAAK,cAAc;gBACjB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,cAAc;wBACzB,sBAAsB,EAAE,CAAC,CAAC,sBAAsB;qBACjD;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,oBAAoB;gBACvB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,oBAAoB;wBAC/B,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,KAAK,EAAE,CAAC,CAAC,KAAK;qBACf;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,OAAO;gBACV,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,OAAO;wBAClB,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;qBACb;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,WAAW;gBACd,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,WAAW;qBACvB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,WAAW;gBACd,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,WAAW;wBACtB,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,QAAQ,EAAE,CAAC,CAAC,QAAQ;wBACpB,eAAe,EAAE,CAAC,CAAC,eAAe;wBAClC,QAAQ,EAAE,CAAC,CAAC,QAAQ;qBACrB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,OAAO;gBACV,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,OAAO;wBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,MAAM,EAAE,CAAC,CAAC,MAAM;qBACjB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,gBAAgB;gBACnB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,gBAAgB;wBAC3B,cAAc,EAAE,CAAC,CAAC,cAAc;qBACjC;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,iBAAiB;gBACpB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,iBAAiB;wBAC5B,QAAQ,EAAE,CAAC,CAAC,QAAQ;wBACpB,IAAI,EAAE,CAAC,CAAC,IAAI;qBACb;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,eAAe;gBAClB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,SAAS,EAAE,eAAe;wBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;qBACb;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,aAAa;gBAChB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,CAAC,CAAC,SAAS;wBACtB,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,SAAS,EAAE,aAAa;wBACxB,UAAU,EAAE,CAAC,CAAC,UAAU;wBACxB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;wBAC1B,eAAe,EAAE,CAAC,CAAC,eAAe;wBAClC,UAAU,EAAE,CAAC,CAAC,UAAU;qBACzB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,cAAc;gBACjB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,SAAS,EAAE,cAAc;wBACzB,UAAU,EAAE,CAAC,CAAC,UAAU;wBACxB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,MAAM,EAAE,CAAC,CAAC,MAAM;qBACjB;oBACD,IAAI,EAAE,OAAO;iBAC4C,CAAC;YAC9D,KAAK,aAAa;gBAChB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;oBACnD,IAAI,EAAE,MAAM;iBAC6C,CAAC;QAChE,CAAC;IACH,CAAC;IACD,MAAM,CAAC,CAAC;QACN,IAAI,CAAC,CAAC,OAAO,CAAC,SAAS,KAAK,aAAa,EAAE,CAAC;YAC1C,OAAO;gBACL,SAAS,EAAE,aAAa;gBACxB,cAAc,EAAE,CAAC,CAAC,cAAc;gBAChC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;gBACjC,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,eAAe,EAAE,CAAC,CAAC,eAAe;gBAClC,KAAK,EAAE,CAAC,CAAC,KAAK;aACyC,CAAC;QAC5D,CAAC;QACD,uFAAuF;QACvF,MAAM,EAAE,GAAI,CAAqE,CAAC,EAAE,CAAC;QACrF,OAAO;YACL,cAAc,EAAE,CAAC,CAAC,cAAc;YAChC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;YACjC,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;YAClC,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,GAAG,CAAC,CAAC,OAAO;SAC2C,CAAC;IAC5D,CAAC;CACF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Temporal } from \"temporal-polyfill\";\nimport * as z from \"zod/mini\";\n\n/**\n * Shared chat event fields for all responses (except error response)\n * @deprecated\n */\nconst chatEventV1SharedSchema = {\n  createdAt: z.string().check(z.iso.datetime()),\n  modelName: z.string().check(z.trim(), z.minLength(1)),\n  modelProviderId: z.string().check(z.trim(), z.minLength(1)),\n  sessionId: z.string().check(z.trim(), z.minLength(1)),\n};\n\n/**\n * Shared chat event fields for all responses (except error response)\n */\nconst chatEventSharedSchema = {\n  conversationId: z.string(),\n  createdAt: z.string().check(z.iso.datetime()),\n  messageId: z.string(),\n  modelName: z.string(),\n  modelProviderId: z.string(),\n  runId: z.string(),\n};\n\nconst errorChunkSchema = z.object({\n  chunkType: z.literal(\"error\"),\n  /**\n   * If the message is not present or very short, it won't be useful to show\n   * anyways, so the parser should throw to the generic response error handler.\n   */\n  message: z.string().check(z.trim(), z.minLength(2)),\n});\n\nconst errorChunkV2Schema = z.object({\n  chunkType: z.literal(\"error\"),\n  message: z.string().check(z.trim(), z.minLength(2)),\n  type: z.string(),\n});\n\nconst conversationUpdateChunkSchema = z.object({\n  chunkType: z.literal(\"conversationUpdate\"),\n  summary: z.optional(z.string()),\n  title: z.optional(z.string()),\n});\n\nconst endOfStreamChunkSchema = z.object({\n  chunkType: z.literal(\"endOfStream\"),\n});\n\n// endOfStream omits messageId: the backend sends it absent or as \"\", neither maps to a meaningful id\nconst endOfStreamInputSchema = z.object({\n  chunkType: z.literal(\"endOfStream\"),\n  conversationId: z.string(),\n  createdAt: z.string().check(z.iso.datetime()),\n  modelName: z.string(),\n  modelProviderId: z.string(),\n  runId: z.string(),\n});\n\nconst interruptChunkSchema = z.object({\n  chunkType: z.literal(\"interrupt\"),\n});\n\nconst modelChunkSchema = z.object({\n  chunkType: z.literal(\"model\"),\n  name: z.string().check(z.trim(), z.minLength(1)),\n  /**\n   * It's useful to indicate (to the type system) that the result contents\n   * at this stage can only be data (de)serializable from JSON. Because\n   * model responses can vary depending on runtime flags and edition, we can\n   * leave this field generic and refine it more precisely in later stages.\n   */\n  result: z.record(z.string(), z.unknown()),\n});\n\nconst sandboxProgressChunkSchema = z.object({\n  chunkType: z.literal(\"sandboxProgress\"),\n  stepName: z.string(),\n  text: z.string(),\n});\n\nconst sandboxStdoutChunkSchema = z.object({\n  callId: z.string().check(z.trim(), z.minLength(1)),\n  chunkType: z.literal(\"sandboxStdout\"),\n  isFinal: z.boolean(),\n  text: z.string(),\n});\nconst sandboxStdoutContentSchema = sandboxStdoutChunkSchema;\nconst sandboxStdoutChatEventSchema = z.extend(sandboxStdoutChunkSchema, chatEventSharedSchema);\nconst toolPlanSchema = z.record(z.string(), z.unknown());\n\nconst toolRequestChunkSchema = z.object({\n  arguments: z.record(z.string(), z.unknown()),\n  callId: z.string().check(z.trim(), z.minLength(1)),\n  chunkType: z.literal(\"toolRequest\"),\n  commentary: z.optional(z.string()),\n  name: z.string(),\n  replacePlan: z.optional(toolPlanSchema),\n  summarizedTitle: z.optional(z.string()),\n  updatePlan: z.optional(toolPlanSchema),\n});\n\nconst toolResponseChunkSchema = z.object({\n  callId: z.string(),\n  chunkType: z.literal(\"toolResponse\"),\n  commentary: z.optional(z.string()),\n  name: z.string(),\n  result: z.record(z.string(), z.unknown()),\n});\n\nconst jobUpdateChunkSchema = z.object({\n  chunkType: z.literal(\"jobUpdate\"),\n  jobId: z.string(),\n  jobState: z.string(),\n  toolExecutionId: z.string(),\n  toolName: z.string(),\n});\n\nconst contextUsageChunkSchema = z.object({\n  chunkType: z.literal(\"contextUsage\"),\n  contextUsagePercentage: z.number().check(z.gte(0), z.lte(100)),\n});\n\nconst selectedSkillInfoSchema = z.object({\n  id: z.string(),\n  name: z.string(),\n});\n\nexport type SelectedSkillInfo = z.infer<typeof selectedSkillInfoSchema>;\n\nconst selectedSkillsChunkSchema = z.object({\n  chunkType: z.literal(\"selectedSkills\"),\n  selectedSkills: z.array(selectedSkillInfoSchema),\n});\n\nconst userMessageChunkSchema = z.object({\n  chunkType: z.literal(\"userMessage\"),\n  text: z.string(),\n});\n\n/**\n * Captures all of the possible chunks of data that the Agent can reply\n * with. We're lenient to extra/unexpected fields being present (`z.object`\n * instead of `z.strictObject`), but for the fields that are expected,\n * we validate that they match the most general type category they belong in.\n * For example, it's good enough to validate that the `callId` is a string\n * for type safety -- we don't need to verify that it's also a `z.uuid()`.\n * @deprecated\n */\nexport const chatEventV1Schema = z.discriminatedUnion(\"chunkType\", [\n  errorChunkSchema,\n  z.extend(modelChunkSchema, chatEventV1SharedSchema),\n  z.extend(toolRequestChunkSchema, chatEventV1SharedSchema),\n  z.extend(toolResponseChunkSchema, chatEventV1SharedSchema),\n]);\n\n/**\n * @deprecated\n */\nexport type ChatEventV1 = z.infer<typeof chatEventV1Schema>;\n\nconst chatEventInputSchema = z.discriminatedUnion(\"chunkType\", [\n  z.extend(errorChunkV2Schema, chatEventSharedSchema),\n  endOfStreamInputSchema,\n  z.extend(interruptChunkSchema, chatEventSharedSchema),\n  z.extend(contextUsageChunkSchema, chatEventSharedSchema),\n  z.extend(conversationUpdateChunkSchema, chatEventSharedSchema),\n  z.extend(jobUpdateChunkSchema, chatEventSharedSchema),\n  z.extend(modelChunkSchema, chatEventSharedSchema),\n  z.extend(selectedSkillsChunkSchema, chatEventSharedSchema),\n  z.extend(sandboxProgressChunkSchema, chatEventSharedSchema),\n  sandboxStdoutChatEventSchema,\n  z.extend(toolRequestChunkSchema, chatEventSharedSchema),\n  z.extend(toolResponseChunkSchema, chatEventSharedSchema),\n  z.extend(userMessageChunkSchema, chatEventSharedSchema),\n]);\n\nconst chatEventOutputSharedSchema = z.extend(\n  z.omit(z.strictObject(chatEventSharedSchema), { createdAt: true, messageId: true }),\n  {\n    createdAt: z.instanceof(Temporal.Instant),\n    id: z.string(),\n  },\n);\n\n// endOfStream has no meaningful messageId on input, so no id on output\nconst endOfStreamOutputSchema = z.omit(chatEventOutputSharedSchema, { id: true });\n\nconst chatEventOutputSchema = z.union([\n  z.extend(chatEventOutputSharedSchema, {\n    content: conversationUpdateChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, { content: errorChunkV2Schema, role: z.literal(\"agent\") }),\n  z.extend(endOfStreamOutputSchema, {\n    content: endOfStreamChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: interruptChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: contextUsageChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: jobUpdateChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, { content: modelChunkSchema, role: z.literal(\"agent\") }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: selectedSkillsChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: sandboxProgressChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: sandboxStdoutContentSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: toolRequestChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: toolResponseChunkSchema,\n    role: z.literal(\"agent\"),\n  }),\n  z.extend(chatEventOutputSharedSchema, {\n    content: userMessageChunkSchema,\n    role: z.literal(\"user\"),\n  }),\n]);\n\nexport const chatEventCodec = z.codec(chatEventInputSchema, chatEventOutputSchema, {\n  decode(v) {\n    if (v.chunkType === \"endOfStream\") {\n      return {\n        content: { chunkType: \"endOfStream\" },\n        conversationId: v.conversationId,\n        createdAt: Temporal.Instant.from(v.createdAt),\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n        role: \"agent\",\n        runId: v.runId,\n      } as const satisfies z.output<typeof chatEventOutputSchema>;\n    }\n\n    const sharedProperties = {\n      conversationId: v.conversationId,\n      createdAt: Temporal.Instant.from(v.createdAt),\n      id: v.messageId,\n      modelName: v.modelName,\n      modelProviderId: v.modelProviderId,\n      runId: v.runId,\n    };\n\n    switch (v.chunkType) {\n      case \"contextUsage\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"contextUsage\",\n            contextUsagePercentage: v.contextUsagePercentage,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"conversationUpdate\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"conversationUpdate\",\n            summary: v.summary,\n            title: v.title,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"error\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"error\",\n            message: v.message,\n            type: v.type,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"interrupt\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"interrupt\",\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"jobUpdate\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"jobUpdate\",\n            jobId: v.jobId,\n            jobState: v.jobState,\n            toolExecutionId: v.toolExecutionId,\n            toolName: v.toolName,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"model\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"model\",\n            name: v.name,\n            result: v.result,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"selectedSkills\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"selectedSkills\",\n            selectedSkills: v.selectedSkills,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"sandboxProgress\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"sandboxProgress\",\n            stepName: v.stepName,\n            text: v.text,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"sandboxStdout\":\n        return {\n          ...sharedProperties,\n          content: {\n            callId: v.callId,\n            chunkType: \"sandboxStdout\",\n            isFinal: v.isFinal,\n            text: v.text,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"toolRequest\":\n        return {\n          ...sharedProperties,\n          content: {\n            arguments: v.arguments,\n            callId: v.callId,\n            chunkType: \"toolRequest\",\n            commentary: v.commentary,\n            name: v.name,\n            replacePlan: v.replacePlan,\n            summarizedTitle: v.summarizedTitle,\n            updatePlan: v.updatePlan,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"toolResponse\":\n        return {\n          ...sharedProperties,\n          content: {\n            callId: v.callId,\n            chunkType: \"toolResponse\",\n            commentary: v.commentary,\n            name: v.name,\n            result: v.result,\n          },\n          role: \"agent\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n      case \"userMessage\":\n        return {\n          ...sharedProperties,\n          content: { chunkType: \"userMessage\", text: v.text },\n          role: \"user\",\n        } as const satisfies z.output<typeof chatEventOutputSchema>;\n    }\n  },\n  encode(v) {\n    if (v.content.chunkType === \"endOfStream\") {\n      return {\n        chunkType: \"endOfStream\",\n        conversationId: v.conversationId,\n        createdAt: v.createdAt.toString(),\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n        runId: v.runId,\n      } as const satisfies z.infer<typeof chatEventInputSchema>;\n    }\n    // Safe: endOfStream (the only variant without id) is handled by the early return above\n    const id = (v as Extract<z.output<typeof chatEventOutputSchema>, { id: string }>).id;\n    return {\n      conversationId: v.conversationId,\n      createdAt: v.createdAt.toString(),\n      messageId: id,\n      modelName: v.modelName,\n      modelProviderId: v.modelProviderId,\n      runId: v.runId,\n      ...v.content,\n    } as const satisfies z.infer<typeof chatEventInputSchema>;\n  },\n});\n\nexport type ChatEvent = z.output<typeof chatEventCodec>;\nexport type ChatEventWithChunkType<T extends z.input<typeof chatEventCodec>[\"chunkType\"]> = Extract<\n  ChatEvent,\n  { content: { chunkType: T } }\n>;\n"]}

package/dist/enterprise/ai/conversations/AgentConversation.d.ts

@@ -80,6 +80,18 @@ export declare class AgentConversation implements AgentConversationProperties {
             chunkType: "interrupt";
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "contextUsage";
+            contextUsagePercentage: number;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;
@@ -243,6 +255,18 @@ export declare class AgentConversation implements AgentConversationProperties {
             chunkType: "interrupt";
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "contextUsage";
+            contextUsagePercentage: number;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;