@dremio/js-sdk 0.51.0 → 0.53.0

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

@@ -158,6 +158,21 @@ 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: "jobUpdate";
+                    jobId: string;
+                    jobState: string;
+                    toolExecutionId: string;
+                    toolName: string;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -277,6 +292,21 @@ 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: "jobUpdate";
+                        jobId: string;
+                        jobState: string;
+                        toolExecutionId: string;
+                        toolName: string;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -403,6 +433,21 @@ 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: "jobUpdate";
+                jobId: string;
+                jobState: string;
+                toolExecutionId: string;
+                toolName: string;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -546,6 +591,21 @@ 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: "jobUpdate";
+                    jobId: string;
+                    jobState: string;
+                    toolExecutionId: string;
+                    toolName: string;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -669,6 +729,21 @@ 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: "jobUpdate";
+                        jobId: string;
+                        jobState: string;
+                        toolExecutionId: string;
+                        toolName: string;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -807,6 +882,21 @@ 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: "jobUpdate";
+                jobId: string;
+                jobState: string;
+                toolExecutionId: string;
+                toolName: string;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;

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

@@ -100,6 +100,21 @@ 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: "jobUpdate";
+                    jobId: string;
+                    jobState: string;
+                    toolExecutionId: string;
+                    toolName: string;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -219,6 +234,21 @@ 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: "jobUpdate";
+                        jobId: string;
+                        jobState: string;
+                        toolExecutionId: string;
+                        toolName: string;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -345,6 +375,21 @@ 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: "jobUpdate";
+                jobId: string;
+                jobState: string;
+                toolExecutionId: string;
+                toolName: string;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -488,6 +533,21 @@ 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: "jobUpdate";
+                    jobId: string;
+                    jobState: string;
+                    toolExecutionId: string;
+                    toolName: string;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -611,6 +671,21 @@ 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: "jobUpdate";
+                        jobId: string;
+                        jobState: string;
+                        toolExecutionId: string;
+                        toolName: string;
+                    };
+                    role: "agent";
                 } | {
                     conversationId: string;
                     modelName: string;
@@ -749,6 +824,21 @@ 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: "jobUpdate";
+                jobId: string;
+                jobState: string;
+                toolExecutionId: string;
+                toolName: string;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;

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

@@ -23,6 +23,7 @@ export declare class UserChatMessage {
                     name: string;
                 }[];
             } | undefined;
+            acceptTaskNames?: string[] | undefined;
             context?: string | undefined;
             skillIds?: string[] | undefined;
         }, "text"> | undefined;
@@ -40,6 +41,7 @@ export declare const createConversationPromptSchema: z.ZodMiniObject<{
             name: z.ZodMiniString<string>;
         }, z.core.$strip>>;
     }, z.core.$strip>>;
+    acceptTaskNames: z.ZodMiniOptional<z.ZodMiniArray<z.ZodMiniString<string>>>;
     context: z.ZodMiniOptional<z.ZodMiniString<string>>;
     skillIds: z.ZodMiniOptional<z.ZodMiniArray<z.ZodMiniString<string>>>;
     text: z.ZodMiniString<string>;

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

@@ -53,6 +53,7 @@ export const createConversationPromptSchema = z.strictObject({
             name: z.string(),
         })),
     })),
+    acceptTaskNames: z.optional(z.array(z.string())),
     context: z.optional(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,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,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,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  approvals: z.optional(\n    z.object({\n      allToolsAllowed: z.boolean(),\n      approvalNonce: z.string(),\n      toolDecisions: z.array(\n        z.object({\n          approved: z.boolean(),\n          arguments: z.record(z.string(), z.unknown()),\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,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,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,MAAM,EAAE;YACvB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;SACjB,CAAC,CACH;KACF,CAAC,CACH;IACD,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAChD,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  approvals: z.optional(\n    z.object({\n      allToolsAllowed: z.boolean(),\n      approvalNonce: z.string(),\n      toolDecisions: z.array(\n        z.object({\n          approved: z.boolean(),\n          arguments: z.record(z.string(), z.unknown()),\n          executionId: z.string(),\n          name: z.string(),\n        }),\n      ),\n    }),\n  ),\n  acceptTaskNames: z.optional(z.array(z.string())),\n  context: z.optional(z.string()),\n  skillIds: z.optional(z.array(z.string())),\n  text: z.string(),\n});\n"]}

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

@@ -86,6 +86,18 @@ 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<"jobUpdate">;
+    jobId: z.ZodMiniString<string>;
+    jobState: z.ZodMiniString<string>;
+    toolExecutionId: z.ZodMiniString<string>;
+    toolName: z.ZodMiniString<string>;
+    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<"model">;
     name: z.ZodMiniString<string>;
@@ -178,6 +190,21 @@ 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<"jobUpdate">;
+        jobId: z.ZodMiniString<string>;
+        jobState: z.ZodMiniString<string>;
+        toolExecutionId: z.ZodMiniString<string>;
+        toolName: z.ZodMiniString<string>;
+    }, 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

@@ -86,6 +86,13 @@ const toolResponseChunkSchema = z.object({
     name: z.string(),
     result: z.record(z.string(), z.unknown()),
 });
+const jobUpdateChunkSchema = z.object({
+    chunkType: z.literal("jobUpdate"),
+    jobId: z.string(),
+    jobState: z.string(),
+    toolExecutionId: z.string(),
+    toolName: z.string(),
+});
 const userMessageChunkSchema = z.object({
     chunkType: z.literal("userMessage"),
     text: z.string(),
@@ -110,6 +117,7 @@ const chatEventInputSchema = z.discriminatedUnion("chunkType", [
     z.extend(endOfStreamChunkSchema, chatEventSharedSchema),
     z.extend(interruptChunkSchema, chatEventSharedSchema),
     z.extend(conversationUpdateChunkSchema, chatEventSharedSchema),
+    z.extend(jobUpdateChunkSchema, chatEventSharedSchema),
     z.extend(modelChunkSchema, chatEventSharedSchema),
     z.extend(toolRequestChunkSchema, chatEventSharedSchema),
     z.extend(toolResponseChunkSchema, chatEventSharedSchema),
@@ -133,6 +141,10 @@ const chatEventOutputSchema = z.union([
         content: interruptChunkSchema,
         role: z.literal("agent"),
     }),
+    z.extend(chatEventOutputSharedSchema, {
+        content: jobUpdateChunkSchema,
+        role: z.literal("agent"),
+    }),
     z.extend(chatEventOutputSharedSchema, { content: modelChunkSchema, role: z.literal("agent") }),
     z.extend(chatEventOutputSharedSchema, {
         content: toolRequestChunkSchema,
@@ -194,6 +206,18 @@ export const chatEventCodec = z.codec(chatEventInputSchema, chatEventOutputSchem
                     },
                     role: "agent",
                 };
+            case "jobUpdate":
+                return {
+                    ...sharedProperties,
+                    content: {
+                        chunkType: "jobUpdate",
+                        jobId: v.jobId,
+                        jobState: v.jobState,
+                        toolExecutionId: v.toolExecutionId,
+                        toolName: v.toolName,
+                    },
+                    role: "agent",
+                };
             case "model":
                 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,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,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,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACxC,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,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,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,qBAAqB,CAAC;IACvD,CAAC,CAAC,MAAM,CAAC,oBAAoB,EAAE,qBAAqB,CAAC;IACrD,CAAC,CAAC,MAAM,CAAC,6BAA6B,EAAE,qBAAqB,CAAC;IAC9D,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE,qBAAqB,CAAC;IACjD,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,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,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,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,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,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;SACqE,CAAC;QAEtF,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,aAAa;gBAChB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,aAAa;qBACzB;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,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,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,eAAe,EAAE,CAAC,CAAC,eAAe;qBACnC;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,OAAO;YACL,cAAc,EAAE,CAAC,CAAC,cAAc;YAChC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;YACjC,SAAS,EAAE,CAAC,CAAC,EAAE;YACf,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\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 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  summarizedTitle: z.optional(z.string()),\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 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  z.extend(endOfStreamChunkSchema, chatEventSharedSchema),\n  z.extend(interruptChunkSchema, chatEventSharedSchema),\n  z.extend(conversationUpdateChunkSchema, chatEventSharedSchema),\n  z.extend(modelChunkSchema, chatEventSharedSchema),\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\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(chatEventOutputSharedSchema, {\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, { content: modelChunkSchema, role: z.literal(\"agent\") }),\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    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    } as const satisfies Omit<z.output<typeof chatEventOutputSchema>, \"content\" | \"role\">;\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 \"endOfStream\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"endOfStream\",\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 \"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 \"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            summarizedTitle: v.summarizedTitle,\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    return {\n      conversationId: v.conversationId,\n      createdAt: v.createdAt.toString(),\n      messageId: v.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,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,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,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACxC,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,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,CAAC,CAAC,MAAM,CAAC,sBAAsB,EAAE,qBAAqB,CAAC;IACvD,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,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,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,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,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,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,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;SACqE,CAAC;QAEtF,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,aAAa;gBAChB,OAAO;oBACL,GAAG,gBAAgB;oBACnB,OAAO,EAAE;wBACP,SAAS,EAAE,aAAa;qBACzB;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,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,eAAe,EAAE,CAAC,CAAC,eAAe;qBACnC;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,OAAO;YACL,cAAc,EAAE,CAAC,CAAC,cAAc;YAChC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;YACjC,SAAS,EAAE,CAAC,CAAC,EAAE;YACf,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\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 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  summarizedTitle: z.optional(z.string()),\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 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  z.extend(endOfStreamChunkSchema, chatEventSharedSchema),\n  z.extend(interruptChunkSchema, chatEventSharedSchema),\n  z.extend(conversationUpdateChunkSchema, chatEventSharedSchema),\n  z.extend(jobUpdateChunkSchema, chatEventSharedSchema),\n  z.extend(modelChunkSchema, chatEventSharedSchema),\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\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(chatEventOutputSharedSchema, {\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: 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    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    } as const satisfies Omit<z.output<typeof chatEventOutputSchema>, \"content\" | \"role\">;\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 \"endOfStream\":\n        return {\n          ...sharedProperties,\n          content: {\n            chunkType: \"endOfStream\",\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 \"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            summarizedTitle: v.summarizedTitle,\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    return {\n      conversationId: v.conversationId,\n      createdAt: v.createdAt.toString(),\n      messageId: v.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

@@ -81,6 +81,21 @@ 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: "jobUpdate";
+            jobId: string;
+            jobState: string;
+            toolExecutionId: string;
+            toolName: string;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;
@@ -186,6 +201,21 @@ 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: "jobUpdate";
+            jobId: string;
+            jobState: string;
+            toolExecutionId: string;
+            toolName: string;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

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

@@ -116,6 +116,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                 chunkType: "interrupt";
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "jobUpdate";
+                jobId: string;
+                jobState: string;
+                toolExecutionId: string;
+                toolName: string;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -235,6 +250,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                     chunkType: "interrupt";
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "jobUpdate";
+                    jobId: string;
+                    jobState: string;
+                    toolExecutionId: string;
+                    toolName: string;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -361,6 +391,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
             chunkType: "interrupt";
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "jobUpdate";
+            jobId: string;
+            jobState: string;
+            toolExecutionId: string;
+            toolName: string;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;
@@ -504,6 +549,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                 chunkType: "interrupt";
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "jobUpdate";
+                jobId: string;
+                jobState: string;
+                toolExecutionId: string;
+                toolName: string;
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -627,6 +687,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                     chunkType: "interrupt";
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "jobUpdate";
+                    jobId: string;
+                    jobState: string;
+                    toolExecutionId: string;
+                    toolName: string;
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -765,6 +840,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
             chunkType: "interrupt";
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "jobUpdate";
+            jobId: string;
+            jobState: string;
+            toolExecutionId: string;
+            toolName: string;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

package/dist/enterprise/ai/conversations/methods/createConversation.d.ts

@@ -29,6 +29,7 @@ export declare const createConversationCodec: z.ZodMiniCodec<z.ZodMiniObject<{
                 name: z.ZodMiniString<string>;
             }, z.core.$strip>>;
         }, z.core.$strip>>;
+        acceptTaskNames: z.ZodMiniOptional<z.ZodMiniArray<z.ZodMiniString<string>>>;
         context: z.ZodMiniOptional<z.ZodMiniString<string>>;
         skillIds: z.ZodMiniOptional<z.ZodMiniArray<z.ZodMiniString<string>>>;
         text: z.ZodMiniString<string>;

package/dist/enterprise/ai/conversations/methods/createConversation.js

@@ -45,6 +45,7 @@ export const createConversationCodec = z.codec(createConversationInSchema, z.ext
             modelName: v.modelName,
             modelProviderId: v.modelProviderId,
             prompt: {
+                acceptTaskNames: v.message.prompt?.acceptTaskNames,
                 approvals: v.message.prompt?.approvals,
                 context: v.message.prompt?.context,
                 skillIds: v.message.prompt?.skillIds,
@@ -55,6 +56,7 @@ export const createConversationCodec = z.codec(createConversationInSchema, z.ext
     encode(v) {
         return {
             message: UserChatMessage.new(v.prompt.text, {
+                acceptTaskNames: v.prompt?.acceptTaskNames,
                 approvals: v.prompt?.approvals,
                 context: v.prompt?.context,
                 skillIds: v.prompt?.skillIds,

package/dist/enterprise/ai/conversations/methods/createConversation.js.map

@@ -1 +1 @@
-{"version":3,"file":"createConversation.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/createConversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,OAAO,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AAChF,OAAO,EAAE,8BAA8B,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAEhG,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,MAAqB,EAAE,EAAE,CAC1D,SAAS,kBAAkB,CAAC,IAA6C;IACvE,OAAO,MAAM;SACV,cAAc,CAAC,qBAAqB,EAAE;QACrC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAC;QAC7D,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,kBAAkB;SACnC;QACD,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,MAAM;KACf,CAAC;SACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA0D,CAAC;SAChF,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC,CAAC;AACpE,CAAC,CAAC;AAEJ,MAAM,0BAA0B,GAAG,CAAC,CAAC,YAAY,CAAC;IAChD,OAAO,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACjC,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACxC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAC5C,0BAA0B,EAC1B,CAAC,CAAC,MAAM,CACN,CAAC,CAAC,IAAI,CAAC,0BAA0B,EAAE;IACjC,OAAO,EAAE,IAAI;CACd,CAAC,EACF;IACE,MAAM,EAAE,8BAA8B;CACvC,CACF,EACD;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;YAClC,MAAM,EAAE;gBACN,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS;gBACtC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO;gBAClC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ;gBACpC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO;aACxB;SACF,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE;gBAC1C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,SAAS;gBAC9B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO;gBAC1B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,QAAQ;aAC7B,CAAC;YACF,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;SACnC,CAAC;IACJ,CAAC;CACF,CACF,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 * as z from \"zod/mini\";\nimport type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport { conversationPropertiesCodec } from \"../conversationPropertiesCodec.ts\";\nimport { createConversationPromptSchema, UserChatMessage } from \"../../chat/UserChatMessage.ts\";\n\nexport const createConversation = (config: SonarV4Config) =>\n  function createConversation(body: z.input<typeof createConversationCodec>) {\n    return config\n      .sonarV4Request(`agent/conversations`, {\n        body: JSON.stringify(z.decode(createConversationCodec, body)),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof conversationPropertiesCodec>>)\n      .map((entity) => z.decode(conversationPropertiesCodec, entity));\n  };\n\nconst createConversationInSchema = z.strictObject({\n  message: z.instanceof(UserChatMessage),\n  modelName: z.optional(z.string()),\n  modelProviderId: z.optional(z.string()),\n});\n\nexport const createConversationCodec = z.codec(\n  createConversationInSchema,\n  z.extend(\n    z.omit(createConversationInSchema, {\n      message: true,\n    }),\n    {\n      prompt: createConversationPromptSchema,\n    },\n  ),\n  {\n    decode(v) {\n      return {\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n        prompt: {\n          approvals: v.message.prompt?.approvals,\n          context: v.message.prompt?.context,\n          skillIds: v.message.prompt?.skillIds,\n          text: v.message.content,\n        },\n      };\n    },\n    encode(v) {\n      return {\n        message: UserChatMessage.new(v.prompt.text, {\n          approvals: v.prompt?.approvals,\n          context: v.prompt?.context,\n          skillIds: v.prompt?.skillIds,\n        }),\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n      };\n    },\n  },\n);\n"]}
+{"version":3,"file":"createConversation.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/createConversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,OAAO,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AAChF,OAAO,EAAE,8BAA8B,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAEhG,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,MAAqB,EAAE,EAAE,CAC1D,SAAS,kBAAkB,CAAC,IAA6C;IACvE,OAAO,MAAM;SACV,cAAc,CAAC,qBAAqB,EAAE;QACrC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAC;QAC7D,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,kBAAkB;SACnC;QACD,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,MAAM;KACf,CAAC;SACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA0D,CAAC;SAChF,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC,CAAC;AACpE,CAAC,CAAC;AAEJ,MAAM,0BAA0B,GAAG,CAAC,CAAC,YAAY,CAAC;IAChD,OAAO,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACjC,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACxC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAC5C,0BAA0B,EAC1B,CAAC,CAAC,MAAM,CACN,CAAC,CAAC,IAAI,CAAC,0BAA0B,EAAE;IACjC,OAAO,EAAE,IAAI;CACd,CAAC,EACF;IACE,MAAM,EAAE,8BAA8B;CACvC,CACF,EACD;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;YAClC,MAAM,EAAE;gBACN,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,eAAe;gBAClD,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS;gBACtC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO;gBAClC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ;gBACpC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO;aACxB;SACF,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE;gBAC1C,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,eAAe;gBAC1C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,SAAS;gBAC9B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO;gBAC1B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,QAAQ;aAC7B,CAAC;YACF,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,eAAe,EAAE,CAAC,CAAC,eAAe;SACnC,CAAC;IACJ,CAAC;CACF,CACF,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 * as z from \"zod/mini\";\nimport type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport { conversationPropertiesCodec } from \"../conversationPropertiesCodec.ts\";\nimport { createConversationPromptSchema, UserChatMessage } from \"../../chat/UserChatMessage.ts\";\n\nexport const createConversation = (config: SonarV4Config) =>\n  function createConversation(body: z.input<typeof createConversationCodec>) {\n    return config\n      .sonarV4Request(`agent/conversations`, {\n        body: JSON.stringify(z.decode(createConversationCodec, body)),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof conversationPropertiesCodec>>)\n      .map((entity) => z.decode(conversationPropertiesCodec, entity));\n  };\n\nconst createConversationInSchema = z.strictObject({\n  message: z.instanceof(UserChatMessage),\n  modelName: z.optional(z.string()),\n  modelProviderId: z.optional(z.string()),\n});\n\nexport const createConversationCodec = z.codec(\n  createConversationInSchema,\n  z.extend(\n    z.omit(createConversationInSchema, {\n      message: true,\n    }),\n    {\n      prompt: createConversationPromptSchema,\n    },\n  ),\n  {\n    decode(v) {\n      return {\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n        prompt: {\n          acceptTaskNames: v.message.prompt?.acceptTaskNames,\n          approvals: v.message.prompt?.approvals,\n          context: v.message.prompt?.context,\n          skillIds: v.message.prompt?.skillIds,\n          text: v.message.content,\n        },\n      };\n    },\n    encode(v) {\n      return {\n        message: UserChatMessage.new(v.prompt.text, {\n          acceptTaskNames: v.prompt?.acceptTaskNames,\n          approvals: v.prompt?.approvals,\n          context: v.prompt?.context,\n          skillIds: v.prompt?.skillIds,\n        }),\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n      };\n    },\n  },\n);\n"]}

package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.d.ts

@@ -53,6 +53,21 @@ export declare const retrieveConversationHistory: (config: SonarV4Config) => (id
             chunkType: "interrupt";
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "jobUpdate";
+            jobId: string;
+            jobState: string;
+            toolExecutionId: string;
+            toolName: string;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

package/dist-iife/cloud.js

@@ -16961,6 +16961,13 @@ var DremioCloud = (() => {
     name: string2(),
     result: record(string2(), unknown())
   });
+  var jobUpdateChunkSchema = object({
+    chunkType: literal("jobUpdate"),
+    jobId: string2(),
+    jobState: string2(),
+    toolExecutionId: string2(),
+    toolName: string2()
+  });
   var userMessageChunkSchema = object({
     chunkType: literal("userMessage"),
     text: string2()
@@ -16976,6 +16983,7 @@ var DremioCloud = (() => {
     extend2(endOfStreamChunkSchema, chatEventSharedSchema),
     extend2(interruptChunkSchema, chatEventSharedSchema),
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
+    extend2(jobUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
     extend2(toolResponseChunkSchema, chatEventSharedSchema),
@@ -16999,6 +17007,10 @@ var DremioCloud = (() => {
       content: interruptChunkSchema,
       role: literal("agent")
     }),
+    extend2(chatEventOutputSharedSchema, {
+      content: jobUpdateChunkSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, { content: modelChunkSchema, role: literal("agent") }),
     extend2(chatEventOutputSharedSchema, {
       content: toolRequestChunkSchema,
@@ -17060,6 +17072,18 @@ var DremioCloud = (() => {
             },
             role: "agent"
           };
+        case "jobUpdate":
+          return {
+            ...sharedProperties,
+            content: {
+              chunkType: "jobUpdate",
+              jobId: v2.jobId,
+              jobState: v2.jobState,
+              toolExecutionId: v2.toolExecutionId,
+              toolName: v2.toolName
+            },
+            role: "agent"
+          };
         case "model":
           return {
             ...sharedProperties,
@@ -17262,6 +17286,7 @@ var DremioCloud = (() => {
         name: string2()
       }))
     })),
+    acceptTaskNames: optional(array(string2())),
     context: optional(string2()),
     skillIds: optional(array(string2())),
     text: string2()
@@ -17729,6 +17754,7 @@ var DremioCloud = (() => {
         modelName: v2.modelName,
         modelProviderId: v2.modelProviderId,
         prompt: {
+          acceptTaskNames: v2.message.prompt?.acceptTaskNames,
           approvals: v2.message.prompt?.approvals,
           context: v2.message.prompt?.context,
           skillIds: v2.message.prompt?.skillIds,
@@ -17739,6 +17765,7 @@ var DremioCloud = (() => {
     encode(v2) {
       return {
         message: UserChatMessage.new(v2.prompt.text, {
+          acceptTaskNames: v2.prompt?.acceptTaskNames,
           approvals: v2.prompt?.approvals,
           context: v2.prompt?.context,
           skillIds: v2.prompt?.skillIds

package/dist-iife/enterprise.js

@@ -16082,6 +16082,13 @@ var DremioEnterprise = (() => {
     name: string2(),
     result: record(string2(), unknown())
   });
+  var jobUpdateChunkSchema = object({
+    chunkType: literal("jobUpdate"),
+    jobId: string2(),
+    jobState: string2(),
+    toolExecutionId: string2(),
+    toolName: string2()
+  });
   var userMessageChunkSchema = object({
     chunkType: literal("userMessage"),
     text: string2()
@@ -16097,6 +16104,7 @@ var DremioEnterprise = (() => {
     extend2(endOfStreamChunkSchema, chatEventSharedSchema),
     extend2(interruptChunkSchema, chatEventSharedSchema),
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
+    extend2(jobUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
     extend2(toolResponseChunkSchema, chatEventSharedSchema),
@@ -16120,6 +16128,10 @@ var DremioEnterprise = (() => {
       content: interruptChunkSchema,
       role: literal("agent")
     }),
+    extend2(chatEventOutputSharedSchema, {
+      content: jobUpdateChunkSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, { content: modelChunkSchema, role: literal("agent") }),
     extend2(chatEventOutputSharedSchema, {
       content: toolRequestChunkSchema,
@@ -16181,6 +16193,18 @@ var DremioEnterprise = (() => {
             },
             role: "agent"
           };
+        case "jobUpdate":
+          return {
+            ...sharedProperties,
+            content: {
+              chunkType: "jobUpdate",
+              jobId: v2.jobId,
+              jobState: v2.jobState,
+              toolExecutionId: v2.toolExecutionId,
+              toolName: v2.toolName
+            },
+            role: "agent"
+          };
         case "model":
           return {
             ...sharedProperties,
@@ -16469,6 +16493,7 @@ var DremioEnterprise = (() => {
         name: string2()
       }))
     })),
+    acceptTaskNames: optional(array(string2())),
     context: optional(string2()),
     skillIds: optional(array(string2())),
     text: string2()
@@ -16501,6 +16526,7 @@ var DremioEnterprise = (() => {
         modelName: v2.modelName,
         modelProviderId: v2.modelProviderId,
         prompt: {
+          acceptTaskNames: v2.message.prompt?.acceptTaskNames,
           approvals: v2.message.prompt?.approvals,
           context: v2.message.prompt?.context,
           skillIds: v2.message.prompt?.skillIds,
@@ -16511,6 +16537,7 @@ var DremioEnterprise = (() => {
     encode(v2) {
       return {
         message: UserChatMessage.new(v2.prompt.text, {
+          acceptTaskNames: v2.prompt?.acceptTaskNames,
           approvals: v2.prompt?.approvals,
           context: v2.prompt?.context,
           skillIds: v2.prompt?.skillIds

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dremio/js-sdk",
-  "version": "0.51.0",
+  "version": "0.53.0",
   "description": "JavaScript library for the Dremio API",
   "keywords": [
     "dremio",