@dremio/js-sdk 0.42.0 → 0.43.0

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

@@ -1,3 +1,4 @@
+import { Temporal } from "temporal-polyfill";
 import * as z from "zod/mini";
 /**
  * Captures all of the possible chunks of data that the Agent can reply
@@ -6,8 +7,9 @@ import * as z from "zod/mini";
  * we validate that they match the most general type category they belong in.
  * For example, it's good enough to validate that the `callId` is a string
  * for type safety -- we don't need to verify that it's also a `z.uuid()`.
+ * @deprecated
  */
-export declare const chatEventSchema: z.ZodMiniDiscriminatedUnion<[z.ZodMiniObject<{
+export declare const chatEventV1Schema: z.ZodMiniDiscriminatedUnion<[z.ZodMiniObject<{
     chunkType: z.ZodMiniLiteral<"error">;
     /**
      * If the message is not present or very short, it won't be useful to show
@@ -17,19 +19,13 @@ export declare const chatEventSchema: z.ZodMiniDiscriminatedUnion<[z.ZodMiniObje
 }, z.core.$strip>, z.ZodMiniObject<{
     chunkType: z.ZodMiniLiteral<"model">;
     name: z.ZodMiniString<string>;
-    /**
-     * It's useful to indicate (to the type system) that the result contents
-     * at this stage can only be data (de)serializable from JSON. Because
-     * model responses can vary depending on runtime flags and edition, we can
-     * leave this field generic and refine it more precisely in later stages.
-     */
-    result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+    result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
     createdAt: z.ZodMiniString<string>;
     modelName: z.ZodMiniString<string>;
     modelProviderId: z.ZodMiniString<string>;
     sessionId: z.ZodMiniString<string>;
 }, z.core.$strip>, z.ZodMiniObject<{
-    arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+    arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
     callId: z.ZodMiniString<string>;
     chunkType: z.ZodMiniLiteral<"toolRequest">;
     commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
@@ -44,10 +40,172 @@ export declare const chatEventSchema: z.ZodMiniDiscriminatedUnion<[z.ZodMiniObje
     chunkType: z.ZodMiniLiteral<"toolResponse">;
     commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
     name: z.ZodMiniString<string>;
-    result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+    result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
     createdAt: z.ZodMiniString<string>;
     modelName: z.ZodMiniString<string>;
     modelProviderId: z.ZodMiniString<string>;
     sessionId: z.ZodMiniString<string>;
 }, z.core.$strip>], "chunkType">;
-export type ChatEvent = z.infer<typeof chatEventSchema>;
+/**
+ * @deprecated
+ */
+export type ChatEventV1 = z.infer<typeof chatEventV1Schema>;
+export declare const chatEventCodec: z.ZodMiniCodec<z.ZodMiniDiscriminatedUnion<[z.ZodMiniObject<{
+    chunkType: z.ZodMiniLiteral<"error">;
+    /**
+     * If the message is not present or very short, it won't be useful to show
+     * anyways, so the parser should throw to the generic response error handler.
+     */
+    message: z.ZodMiniString<string>;
+}, z.core.$strip>, z.ZodMiniObject<{
+    chunkType: z.ZodMiniLiteral<"conversationUpdate">;
+    summary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    title: z.ZodMiniOptional<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>;
+    result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
+    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<{
+    arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
+    callId: z.ZodMiniString<string>;
+    chunkType: z.ZodMiniLiteral<"toolRequest">;
+    commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    name: z.ZodMiniString<string>;
+    summarizedTitle: z.ZodMiniOptional<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<{
+    callId: z.ZodMiniString<string>;
+    chunkType: z.ZodMiniLiteral<"toolResponse">;
+    commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    name: z.ZodMiniString<string>;
+    result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
+    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<"userMessage">;
+    text: 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>], "chunkType">, z.ZodMiniUnion<readonly [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<"conversationUpdate">;
+        summary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        title: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    }, 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<"error">;
+        /**
+         * If the message is not present or very short, it won't be useful to show
+         * anyways, so the parser should throw to the generic response error handler.
+         */
+        message: z.ZodMiniString<string>;
+    }, 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<"model">;
+        name: z.ZodMiniString<string>;
+        /**
+         * It's useful to indicate (to the type system) that the result contents
+         * at this stage can only be data (de)serializable from JSON. Because
+         * model responses can vary depending on runtime flags and edition, we can
+         * leave this field generic and refine it more precisely in later stages.
+         */
+        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
+    }, 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<{
+        arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
+        callId: z.ZodMiniString<string>;
+        chunkType: z.ZodMiniLiteral<"toolRequest">;
+        commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        name: z.ZodMiniString<string>;
+        summarizedTitle: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    }, 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<{
+        callId: z.ZodMiniString<string>;
+        chunkType: z.ZodMiniLiteral<"toolResponse">;
+        commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        name: z.ZodMiniString<string>;
+        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
+    }, 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<"userMessage">;
+        text: z.ZodMiniString<string>;
+    }, z.core.$strip>;
+    role: z.ZodMiniLiteral<"user">;
+}, z.core.$strict>]>>;
+export type ChatEvent = z.output<typeof chatEventCodec>;
+export type ChatEventWithChunkType<T extends z.input<typeof chatEventCodec>["chunkType"]> = Extract<ChatEvent, {
+    content: {
+        chunkType: T;
+    };
+}>;

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

@@ -13,16 +13,72 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+import { Temporal } from "temporal-polyfill";
 import * as z from "zod/mini";
 /**
  * Shared chat event fields for all responses (except error response)
+ * @deprecated
  */
-const chatEventSharedSchema = {
+const chatEventV1SharedSchema = {
     createdAt: z.string().check(z.iso.datetime()),
     modelName: z.string().check(z.trim(), z.minLength(1)),
     modelProviderId: z.string().check(z.trim(), z.minLength(1)),
     sessionId: z.string().check(z.trim(), z.minLength(1)),
 };
+/**
+ * Shared chat event fields for all responses (except error response)
+ */
+const chatEventSharedSchema = {
+    conversationId: z.string(),
+    createdAt: z.string().check(z.iso.datetime()),
+    messageId: z.string(),
+    modelName: z.string(),
+    modelProviderId: z.string(),
+    runId: z.string(),
+};
+const errorChunkSchema = z.object({
+    chunkType: z.literal("error"),
+    /**
+     * If the message is not present or very short, it won't be useful to show
+     * anyways, so the parser should throw to the generic response error handler.
+     */
+    message: z.string().check(z.trim(), z.minLength(2)),
+});
+const conversationUpdateChunkSchema = z.object({
+    chunkType: z.literal("conversationUpdate"),
+    summary: z.optional(z.string()),
+    title: z.optional(z.string()),
+});
+const modelChunkSchema = z.object({
+    chunkType: z.literal("model"),
+    name: z.string().check(z.trim(), z.minLength(1)),
+    /**
+     * It's useful to indicate (to the type system) that the result contents
+     * at this stage can only be data (de)serializable from JSON. Because
+     * model responses can vary depending on runtime flags and edition, we can
+     * leave this field generic and refine it more precisely in later stages.
+     */
+    result: z.record(z.string(), z.unknown()),
+});
+const toolRequestChunkSchema = z.object({
+    arguments: z.record(z.string(), z.unknown()),
+    callId: z.string().check(z.trim(), z.minLength(1)),
+    chunkType: z.literal("toolRequest"),
+    commentary: z.optional(z.string()),
+    name: z.string(),
+    summarizedTitle: z.optional(z.string()),
+});
+const toolResponseChunkSchema = z.object({
+    callId: z.string(),
+    chunkType: z.literal("toolResponse"),
+    commentary: z.optional(z.string()),
+    name: z.string(),
+    result: z.record(z.string(), z.unknown()),
+});
+const userMessageChunkSchema = z.object({
+    chunkType: z.literal("userMessage"),
+    text: z.string(),
+});
 /**
  * Captures all of the possible chunks of data that the Agent can reply
  * with. We're lenient to extra/unexpected fields being present (`z.object`
@@ -30,44 +86,136 @@ const chatEventSharedSchema = {
  * we validate that they match the most general type category they belong in.
  * For example, it's good enough to validate that the `callId` is a string
  * for type safety -- we don't need to verify that it's also a `z.uuid()`.
+ * @deprecated
  */
-export const chatEventSchema = z.discriminatedUnion("chunkType", [
-    z.object({
-        chunkType: z.literal("error"),
-        /**
-         * If the message is not present or very short, it won't be useful to show
-         * anyways, so the parser should throw to the generic response error handler.
-         */
-        message: z.string().check(z.trim(), z.minLength(2)),
+export const chatEventV1Schema = z.discriminatedUnion("chunkType", [
+    errorChunkSchema,
+    z.extend(modelChunkSchema, chatEventV1SharedSchema),
+    z.extend(toolRequestChunkSchema, chatEventV1SharedSchema),
+    z.extend(toolResponseChunkSchema, chatEventV1SharedSchema),
+]);
+const chatEventInputSchema = z.discriminatedUnion("chunkType", [
+    errorChunkSchema,
+    z.extend(conversationUpdateChunkSchema, chatEventSharedSchema),
+    z.extend(modelChunkSchema, chatEventSharedSchema),
+    z.extend(toolRequestChunkSchema, chatEventSharedSchema),
+    z.extend(toolResponseChunkSchema, chatEventSharedSchema),
+    z.extend(userMessageChunkSchema, chatEventSharedSchema),
+]);
+const chatEventOutputSharedSchema = z.extend(z.omit(z.strictObject(chatEventSharedSchema), { createdAt: true, messageId: true }), {
+    createdAt: z.instanceof(Temporal.Instant),
+    id: z.string(),
+});
+const chatEventOutputSchema = z.union([
+    z.extend(chatEventOutputSharedSchema, {
+        content: conversationUpdateChunkSchema,
+        role: z.literal("agent"),
     }),
-    z.object({
-        ...chatEventSharedSchema,
-        chunkType: z.literal("model"),
-        name: z.string().check(z.trim(), z.minLength(1)),
-        /**
-         * It's useful to indicate (to the type system) that the result contents
-         * at this stage can only be data (de)serializable from JSON. Because
-         * model responses can vary depending on runtime flags and edition, we can
-         * leave this field generic and refine it more precisely in later stages.
-         */
-        result: z.record(z.string(), z.json()),
+    z.extend(chatEventOutputSharedSchema, { content: errorChunkSchema, role: z.literal("agent") }),
+    z.extend(chatEventOutputSharedSchema, { content: modelChunkSchema, role: z.literal("agent") }),
+    z.extend(chatEventOutputSharedSchema, {
+        content: toolRequestChunkSchema,
+        role: z.literal("agent"),
     }),
-    z.object({
-        ...chatEventSharedSchema,
-        arguments: z.record(z.string(), z.json()),
-        callId: z.string().check(z.trim(), z.minLength(1)),
-        chunkType: z.literal("toolRequest"),
-        commentary: z.optional(z.string()),
-        name: z.string(),
-        summarizedTitle: z.optional(z.string()),
+    z.extend(chatEventOutputSharedSchema, {
+        content: toolResponseChunkSchema,
+        role: z.literal("agent"),
     }),
-    z.object({
-        ...chatEventSharedSchema,
-        callId: z.string(),
-        chunkType: z.literal("toolResponse"),
-        commentary: z.optional(z.string()),
-        name: z.string(),
-        result: z.record(z.string(), z.json()),
+    z.extend(chatEventOutputSharedSchema, {
+        content: userMessageChunkSchema,
+        role: z.literal("user"),
     }),
 ]);
+export const chatEventCodec = z.codec(chatEventInputSchema, chatEventOutputSchema, {
+    decode(v) {
+        if (v.chunkType === "error") {
+            return {
+                content: {
+                    chunkType: "error",
+                    message: v.message,
+                },
+                conversationId: "",
+                createdAt: Temporal.Now.instant(),
+                id: "",
+                modelName: "",
+                modelProviderId: "",
+                role: "agent",
+                runId: "",
+            };
+        }
+        const sharedProperties = {
+            conversationId: v.conversationId,
+            createdAt: Temporal.Instant.from(v.createdAt),
+            id: v.messageId,
+            modelName: v.modelName,
+            modelProviderId: v.modelProviderId,
+            runId: v.runId,
+        };
+        switch (v.chunkType) {
+            case "conversationUpdate":
+                return {
+                    ...sharedProperties,
+                    content: {
+                        chunkType: "conversationUpdate",
+                        summary: v.summary,
+                        title: v.title,
+                    },
+                    role: "agent",
+                };
+            case "model":
+                return {
+                    ...sharedProperties,
+                    content: {
+                        chunkType: "model",
+                        name: v.name,
+                        result: v.result,
+                    },
+                    role: "agent",
+                };
+            case "toolRequest":
+                return {
+                    ...sharedProperties,
+                    content: {
+                        arguments: v.arguments,
+                        callId: v.callId,
+                        chunkType: "toolRequest",
+                        commentary: v.commentary,
+                        name: v.name,
+                        summarizedTitle: v.summarizedTitle,
+                    },
+                    role: "agent",
+                };
+            case "toolResponse":
+                return {
+                    ...sharedProperties,
+                    content: {
+                        callId: v.callId,
+                        chunkType: "toolResponse",
+                        commentary: v.commentary,
+                        name: v.name,
+                        result: v.result,
+                    },
+                    role: "agent",
+                };
+            case "userMessage":
+                return {
+                    ...sharedProperties,
+                    content: { chunkType: "userMessage", text: v.text },
+                    role: "user",
+                };
+        }
+    },
+    encode(v) {
+        return {
+            //@ts-ignore
+            conversationId: v.conversationId,
+            createdAt: v.createdAt.toString(),
+            messageId: v.id,
+            modelName: v.modelName,
+            modelProviderId: v.modelProviderId,
+            runId: v.runId,
+            ...v.content,
+        };
+    },
+});
 //# sourceMappingURL=chatEventSchema.js.map

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,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B;;GAEG;AACH,MAAM,qBAAqB,GAAG;IAC5B,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;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,CAAC,kBAAkB,CAAC,WAAW,EAAE;IAC/D,CAAC,CAAC,MAAM,CAAC;QACP,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QAC7B;;;WAGG;QACH,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;KACpD,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,GAAG,qBAAqB;QACxB,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QAC7B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAChD;;;;;WAKG;QACH,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;KACvC,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,GAAG,qBAAqB;QACxB,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACzC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAClD,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;QACnC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAClC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KACxC,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,GAAG,qBAAqB;QACxB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;QAClB,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;QACpC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAClC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;KACvC,CAAC;CACH,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 * as z from \"zod/mini\";\n\n/**\n * Shared chat event fields for all responses (except error response)\n */\nconst chatEventSharedSchema = {\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 * 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 */\nexport const chatEventSchema = z.discriminatedUnion(\"chunkType\", [\n  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  z.object({\n    ...chatEventSharedSchema,\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.json()),\n  }),\n  z.object({\n    ...chatEventSharedSchema,\n    arguments: z.record(z.string(), z.json()),\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  z.object({\n    ...chatEventSharedSchema,\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.json()),\n  }),\n]);\n\nexport type ChatEvent = z.infer<typeof chatEventSchema>;\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,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,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,gBAAgB;IAChB,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,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAC9F,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,IAAI,CAAC,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;YAC5B,OAAO;gBACL,OAAO,EAAE;oBACP,SAAS,EAAE,OAAO;oBAClB,OAAO,EAAE,CAAC,CAAC,OAAO;iBACnB;gBACD,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE;gBACjC,EAAE,EAAE,EAAE;gBACN,SAAS,EAAE,EAAE;gBACb,eAAe,EAAE,EAAE;gBACnB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE;aACgD,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;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,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,YAAY;YACZ,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 conversationUpdateChunkSchema = z.object({\n  chunkType: z.literal(\"conversationUpdate\"),\n  summary: z.optional(z.string()),\n  title: z.optional(z.string()),\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  errorChunkSchema,\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: errorChunkSchema, role: z.literal(\"agent\") }),\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    if (v.chunkType === \"error\") {\n      return {\n        content: {\n          chunkType: \"error\",\n          message: v.message,\n        },\n        conversationId: \"\",\n        createdAt: Temporal.Now.instant(),\n        id: \"\",\n        modelName: \"\",\n        modelProviderId: \"\",\n        role: \"agent\",\n        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    } 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 \"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      //@ts-ignore\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/chat/eventWrappers/AgentChatPartialResponse.d.ts

@@ -16,12 +16,9 @@ export declare class AgentChatPartialResponse {
         content: string;
         id: string;
         session: {
-            createdAt: string;
             id: string;
             modelName: string;
             modelProviderId: string;
-            updatedAt: string;
-            title?: string | undefined;
         };
         type: "delta" | "end" | "start";
     };
@@ -31,12 +28,9 @@ export declare class AgentChatPartialResponse {
         content: z.ZodMiniString<string>;
         id: z.ZodMiniString<string>;
         session: z.ZodMiniObject<{
-            createdAt: z.ZodMiniString<string>;
             id: z.ZodMiniString<string>;
             modelName: z.ZodMiniString<string>;
             modelProviderId: z.ZodMiniString<string>;
-            title: z.ZodMiniOptional<z.ZodMiniString<string>>;
-            updatedAt: z.ZodMiniString<string>;
         }, z.core.$strip>;
         type: z.ZodMiniEnum<{
             delta: "delta";

package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.d.ts

@@ -22,13 +22,13 @@ export declare class AgentChatResponse<T extends AgentResponseContent> {
     } | {
         chunkType: "model";
         name: string;
-        result: Record<string, z.core.util.JSONType>;
+        result: Record<string, unknown>;
         createdAt: string;
         modelName: string;
         modelProviderId: string;
         sessionId: string;
     } | {
-        arguments: Record<string, z.core.util.JSONType>;
+        arguments: Record<string, unknown>;
         callId: string;
         chunkType: "toolRequest";
         name: string;
@@ -42,7 +42,7 @@ export declare class AgentChatResponse<T extends AgentResponseContent> {
         callId: string;
         chunkType: "toolResponse";
         name: string;
-        result: Record<string, z.core.util.JSONType>;
+        result: Record<string, unknown>;
         createdAt: string;
         modelName: string;
         modelProviderId: string;
@@ -56,13 +56,13 @@ export declare class AgentChatResponse<T extends AgentResponseContent> {
     }, z.core.$strip>, z.ZodMiniObject<{
         chunkType: z.ZodMiniLiteral<"model">;
         name: z.ZodMiniString<string>;
-        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
         createdAt: z.ZodMiniString<string>;
         modelName: z.ZodMiniString<string>;
         modelProviderId: z.ZodMiniString<string>;
         sessionId: z.ZodMiniString<string>;
     }, z.core.$strip>, z.ZodMiniObject<{
-        arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+        arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
         callId: z.ZodMiniString<string>;
         chunkType: z.ZodMiniLiteral<"toolRequest">;
         commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
@@ -77,7 +77,7 @@ export declare class AgentChatResponse<T extends AgentResponseContent> {
         chunkType: z.ZodMiniLiteral<"toolResponse">;
         commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
         name: z.ZodMiniString<string>;
-        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
         createdAt: z.ZodMiniString<string>;
         modelName: z.ZodMiniString<string>;
         modelProviderId: z.ZodMiniString<string>;

package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.js

@@ -21,7 +21,7 @@ import { AgentToolResultResponseContent } from "../responseContent/AgentToolResu
 import { ChatSession } from "../ChatSession.js";
 import * as z from "zod/mini";
 import { Temporal } from "temporal-polyfill";
-import { chatEventSchema } from "../chatEventSchema.js";
+import { chatEventV1Schema } from "../chatEventSchema.js";
 export class AgentChatResponse {
     createdAt = Temporal.Now.instant();
     id = nanoid();
@@ -51,7 +51,7 @@ export class AgentChatResponse {
         // eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any
         return z.decode(AgentChatResponse.codec, data);
     }
-    static codec = z.codec(chatEventSchema, z.instanceof(AgentChatResponse), {
+    static codec = z.codec(chatEventV1Schema, z.instanceof(AgentChatResponse), {
         decode(chatEvent) {
             const chatSession = ChatSession.new();
             const content = (() => {

package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.js.map

@@ -1 +1 @@
-{"version":3,"file":"AgentChatResponse.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentChatResponse.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,yBAAyB,EAAE,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,yBAAyB,EAAE,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,+BAA+B,EAAE,MAAM,uDAAuD,CAAC;AACxG,OAAO,EAAE,8BAA8B,EAAE,MAAM,sDAAsD,CAAC;AACtG,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAkB,MAAM,uBAAuB,CAAC;AAExE,MAAM,OAAO,iBAAiB;IACnB,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnC,EAAE,GAAG,MAAM,EAAE,CAAC;IACd,OAAO,CAAc;IACrB,OAAO,CAAI;IAEpB,YAAY,WAAwB,EAAE,OAAU,EAAE,SAA4B,EAAE,EAAW;QACzF,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC;QAED,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,cAAc,CAAiC,OAAU;QACvD,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/E,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAa;QAC3B,qGAAqG;QACrG,OAAO,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAW,CAAC,CAAC;IACxD,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE;QACvE,MAAM,CAAC,SAAS;YACd,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,CAAC,GAAG,EAAE;gBACpB,QAAQ,SAAS,CAAC,SAAS,EAAE,CAAC;oBAC5B,KAAK,OAAO;wBACV,OAAO,IAAI,yBAAyB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;oBAC1D,KAAK,OAAO;wBACV,OAAO,IAAI,yBAAyB,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;oBACzE,KAAK,aAAa,CAAC,CAAC,CAAC;wBACnB,OAAO,IAAI,+BAA+B,CACxC,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,IAAI,EACd,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;6BAC7B,IAAI,EAAE;6BACN,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAE,CAAC,EAC1C,SAAS,CAAC,eAAe,IAAI,IAAI,EACjC,SAAS,CAAC,UAAU,IAAI,IAAI,CAC7B,CAAC;oBACJ,CAAC;oBACD,KAAK,cAAc;wBACjB,OAAO,IAAI,8BAA8B,CACvC,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,IAAI,EACd,SAAS,CAAC,MAAM,CACjB,CAAC;gBACN,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;YACL,WAAW,CAAC,MAAM,CAAC;gBACjB,GAAG,CAAC,WAAW,IAAI,SAAS,IAAI,EAAE,SAAS,EAAE,SAAS,CAAC,SAAS,EAAE,CAAC;gBACnE,GAAG,CAAC,iBAAiB,IAAI,SAAS,IAAI;oBACpC,eAAe,EAAE,SAAS,CAAC,eAAe;iBAC3C,CAAC;gBACF,GAAG,CAAC,WAAW,IAAI,SAAS,IAAI;oBAC9B,EAAE,EAAE,SAAS,CAAC,SAAS;iBACxB,CAAC;aACH,CAAC,CAAC;YACH,OAAO,IAAI,iBAAiB,CAC1B,WAAW,EACX,OAAO,EACP,WAAW,IAAI,SAAS;gBACtB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;gBAC5C,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAC3B,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,aAAa;YAClB,MAAM,SAAS,GAAG,aAAa,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC;YAEtD,IAAI,aAAa,CAAC,OAAO,YAAY,yBAAyB,EAAE,CAAC;gBAC/D,OAAO;oBACL,SAAS,EAAE,OAAO;oBAClB,GAAG,aAAa,CAAC,OAAO;iBACI,CAAC;YACjC,CAAC;iBAAM,IAAI,aAAa,CAAC,OAAO,YAAY,yBAAyB,EAAE,CAAC;gBACtE,OAAO;oBACL,SAAS,EAAE,OAAO;oBAClB,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE;oBAChD,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE;oBAC5D,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI;oBAChC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,EAAG;iBACR,CAAC;YACjC,CAAC;iBAAM,IAAI,aAAa,CAAC,OAAO,YAAY,+BAA+B,EAAE,CAAC;gBAC5E,OAAO;oBACL,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAC/C,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE;wBACjB,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;wBACxB,OAAO,KAAK,CAAC;oBACf,CAAC,EACD,EAA6B,CACU;oBACzC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,aAAa;oBACxB,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,UAAU,IAAI,SAAS;oBACzD,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE;oBAChD,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE;oBAC5D,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI;oBAChC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,EAAG;oBACpC,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,SAAS;iBACvC,CAAC;YACjC,CAAC;iBAAM,IAAI,aAAa,CAAC,OAAO,YAAY,8BAA8B,EAAE,CAAC;gBAC3E,OAAO;oBACL,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,cAAc;oBACzB,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,UAAU,IAAI,SAAS;oBACzD,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE;oBAChD,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE;oBAC5D,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI;oBAChC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,EAAG;iBACR,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;KACF,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 { AgentErrorResponseContent } from \"../responseContent/AgentErrorResponseContent.ts\";\nimport { AgentModelResponseContent } from \"../responseContent/AgentModelResponseContent.ts\";\nimport { AgentToolRequestResponseContent } from \"../responseContent/AgentToolRequestResponseContent.ts\";\nimport { AgentToolResultResponseContent } from \"../responseContent/AgentToolResultResponseContent.ts\";\nimport { ChatSession } from \"../ChatSession.ts\";\nimport * as z from \"zod/mini\";\nimport { Temporal } from \"temporal-polyfill\";\nimport { chatEventSchema, type ChatEvent } from \"../chatEventSchema.ts\";\n\nexport class AgentChatResponse<T extends AgentResponseContent> {\n  readonly createdAt = Temporal.Now.instant();\n  readonly id = nanoid();\n  readonly session: ChatSession;\n  readonly content: T;\n\n  constructor(chatSession: ChatSession, content: T, createdAt?: Temporal.Instant, id?: string) {\n    this.session = chatSession;\n    this.content = content;\n\n    if (createdAt) {\n      this.createdAt = createdAt;\n    }\n\n    if (id) {\n      this.id = id;\n    }\n  }\n\n  /**\n   * Creates a copy of the AgentResponse with a different content property.\n   * Preserves the original message's ID\n   */\n  replaceContent<R extends AgentResponseContent>(content: R): AgentChatResponse<R> {\n    return new AgentChatResponse(this.session, content, this.createdAt, this.id);\n  }\n\n  toJSON() {\n    return z.encode(AgentChatResponse.codec, this);\n  }\n\n  static fromJSON(data: unknown) {\n    // eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any\n    return z.decode(AgentChatResponse.codec, data as any);\n  }\n\n  static codec = z.codec(chatEventSchema, z.instanceof(AgentChatResponse), {\n    decode(chatEvent) {\n      const chatSession = ChatSession.new();\n      const content = (() => {\n        switch (chatEvent.chunkType) {\n          case \"error\":\n            return new AgentErrorResponseContent(chatEvent.message);\n          case \"model\":\n            return new AgentModelResponseContent(chatEvent.name, chatEvent.result);\n          case \"toolRequest\": {\n            return new AgentToolRequestResponseContent(\n              chatEvent.callId,\n              chatEvent.name,\n              Object.keys(chatEvent.arguments)\n                .sort()\n                .map((key) => chatEvent.arguments[key]!),\n              chatEvent.summarizedTitle || null,\n              chatEvent.commentary || null,\n            );\n          }\n          case \"toolResponse\":\n            return new AgentToolResultResponseContent(\n              chatEvent.callId,\n              chatEvent.name,\n              chatEvent.result,\n            );\n        }\n      })();\n      chatSession.assign({\n        ...(\"modelName\" in chatEvent && { modelName: chatEvent.modelName }),\n        ...(\"modelProviderId\" in chatEvent && {\n          modelProviderId: chatEvent.modelProviderId,\n        }),\n        ...(\"sessionId\" in chatEvent && {\n          id: chatEvent.sessionId,\n        }),\n      });\n      return new AgentChatResponse(\n        chatSession,\n        content,\n        \"createdAt\" in chatEvent\n          ? Temporal.Instant.from(chatEvent.createdAt)\n          : Temporal.Now.instant(),\n      );\n    },\n    encode(agentResponse) {\n      const createdAt = agentResponse.createdAt?.toString();\n\n      if (agentResponse.content instanceof AgentErrorResponseContent) {\n        return {\n          chunkType: \"error\",\n          ...agentResponse.content,\n        } as const satisfies ChatEvent;\n      } else if (agentResponse.content instanceof AgentModelResponseContent) {\n        return {\n          chunkType: \"model\",\n          createdAt: createdAt,\n          modelName: agentResponse.session.modelName || \"\",\n          modelProviderId: agentResponse.session.modelProviderId || \"\",\n          name: agentResponse.content.name,\n          result: agentResponse.content.result,\n          sessionId: agentResponse.session.id!,\n        } as const satisfies ChatEvent;\n      } else if (agentResponse.content instanceof AgentToolRequestResponseContent) {\n        return {\n          arguments: agentResponse.content.arguments.reduce(\n            (accum, curr, i) => {\n              accum[`arg${i}`] = curr;\n              return accum;\n            },\n            {} as Record<string, unknown>,\n          ) as Record<string, z.core.util.JSONType>,\n          callId: agentResponse.content.callId,\n          chunkType: \"toolRequest\",\n          commentary: agentResponse.content.commentary || undefined,\n          createdAt: createdAt,\n          modelName: agentResponse.session.modelName || \"\",\n          modelProviderId: agentResponse.session.modelProviderId || \"\",\n          name: agentResponse.content.name,\n          sessionId: agentResponse.session.id!,\n          summarizedTitle: agentResponse.content.summarizedTitle || undefined,\n        } as const satisfies ChatEvent;\n      } else if (agentResponse.content instanceof AgentToolResultResponseContent) {\n        return {\n          callId: agentResponse.content.callId,\n          chunkType: \"toolResponse\",\n          commentary: agentResponse.content.commentary || undefined,\n          createdAt: createdAt,\n          modelName: agentResponse.session.modelName || \"\",\n          modelProviderId: agentResponse.session.modelProviderId || \"\",\n          name: agentResponse.content.name,\n          result: agentResponse.content.result,\n          sessionId: agentResponse.session.id!,\n        } as const satisfies ChatEvent;\n      } else {\n        throw new Error(\"Unknown content type\");\n      }\n    },\n  });\n}\n\nexport type AgentResponseContent =\n  | AgentErrorResponseContent\n  | AgentModelResponseContent\n  | AgentToolRequestResponseContent\n  | AgentToolResultResponseContent;\n"]}
+{"version":3,"file":"AgentChatResponse.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentChatResponse.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,yBAAyB,EAAE,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,yBAAyB,EAAE,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,+BAA+B,EAAE,MAAM,uDAAuD,CAAC;AACxG,OAAO,EAAE,8BAA8B,EAAE,MAAM,sDAAsD,CAAC;AACtG,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAoB,MAAM,uBAAuB,CAAC;AAE5E,MAAM,OAAO,iBAAiB;IACnB,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnC,EAAE,GAAG,MAAM,EAAE,CAAC;IACd,OAAO,CAAc;IACrB,OAAO,CAAI;IAEpB,YAAY,WAAwB,EAAE,OAAU,EAAE,SAA4B,EAAE,EAAW;QACzF,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC;QAED,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,cAAc,CAAiC,OAAU;QACvD,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/E,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAa;QAC3B,qGAAqG;QACrG,OAAO,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAW,CAAC,CAAC;IACxD,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE;QACzE,MAAM,CAAC,SAAS;YACd,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,CAAC,GAAG,EAAE;gBACpB,QAAQ,SAAS,CAAC,SAAS,EAAE,CAAC;oBAC5B,KAAK,OAAO;wBACV,OAAO,IAAI,yBAAyB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;oBAC1D,KAAK,OAAO;wBACV,OAAO,IAAI,yBAAyB,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;oBACzE,KAAK,aAAa,CAAC,CAAC,CAAC;wBACnB,OAAO,IAAI,+BAA+B,CACxC,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,IAAI,EACd,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;6BAC7B,IAAI,EAAE;6BACN,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAE,CAAC,EAC1C,SAAS,CAAC,eAAe,IAAI,IAAI,EACjC,SAAS,CAAC,UAAU,IAAI,IAAI,CAC7B,CAAC;oBACJ,CAAC;oBACD,KAAK,cAAc;wBACjB,OAAO,IAAI,8BAA8B,CACvC,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,IAAI,EACd,SAAS,CAAC,MAAM,CACjB,CAAC;gBACN,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;YACL,WAAW,CAAC,MAAM,CAAC;gBACjB,GAAG,CAAC,WAAW,IAAI,SAAS,IAAI,EAAE,SAAS,EAAE,SAAS,CAAC,SAAS,EAAE,CAAC;gBACnE,GAAG,CAAC,iBAAiB,IAAI,SAAS,IAAI;oBACpC,eAAe,EAAE,SAAS,CAAC,eAAe;iBAC3C,CAAC;gBACF,GAAG,CAAC,WAAW,IAAI,SAAS,IAAI;oBAC9B,EAAE,EAAE,SAAS,CAAC,SAAS;iBACxB,CAAC;aACH,CAAC,CAAC;YACH,OAAO,IAAI,iBAAiB,CAC1B,WAAW,EACX,OAAO,EACP,WAAW,IAAI,SAAS;gBACtB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;gBAC5C,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAC3B,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,aAAa;YAClB,MAAM,SAAS,GAAG,aAAa,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC;YAEtD,IAAI,aAAa,CAAC,OAAO,YAAY,yBAAyB,EAAE,CAAC;gBAC/D,OAAO;oBACL,SAAS,EAAE,OAAO;oBAClB,GAAG,aAAa,CAAC,OAAO;iBACM,CAAC;YACnC,CAAC;iBAAM,IAAI,aAAa,CAAC,OAAO,YAAY,yBAAyB,EAAE,CAAC;gBACtE,OAAO;oBACL,SAAS,EAAE,OAAO;oBAClB,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE;oBAChD,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE;oBAC5D,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI;oBAChC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,EAAG;iBACN,CAAC;YACnC,CAAC;iBAAM,IAAI,aAAa,CAAC,OAAO,YAAY,+BAA+B,EAAE,CAAC;gBAC5E,OAAO;oBACL,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAC/C,CAAC,KAA8B,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE;wBAC1C,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;wBACxB,OAAO,KAAK,CAAC;oBACf,CAAC,EACD,EAA6B,CAC9B;oBACD,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,aAAa;oBACxB,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,UAAU,IAAI,SAAS;oBACzD,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE;oBAChD,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE;oBAC5D,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI;oBAChC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,EAAG;oBACpC,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,SAAS;iBACrC,CAAC;YACnC,CAAC;iBAAM,IAAI,aAAa,CAAC,OAAO,YAAY,8BAA8B,EAAE,CAAC;gBAC3E,OAAO;oBACL,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,cAAc;oBACzB,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,UAAU,IAAI,SAAS;oBACzD,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE;oBAChD,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE;oBAC5D,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI;oBAChC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;oBACpC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,EAAG;iBACN,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;KACF,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 { AgentErrorResponseContent } from \"../responseContent/AgentErrorResponseContent.ts\";\nimport { AgentModelResponseContent } from \"../responseContent/AgentModelResponseContent.ts\";\nimport { AgentToolRequestResponseContent } from \"../responseContent/AgentToolRequestResponseContent.ts\";\nimport { AgentToolResultResponseContent } from \"../responseContent/AgentToolResultResponseContent.ts\";\nimport { ChatSession } from \"../ChatSession.ts\";\nimport * as z from \"zod/mini\";\nimport { Temporal } from \"temporal-polyfill\";\nimport { chatEventV1Schema, type ChatEventV1 } from \"../chatEventSchema.ts\";\n\nexport class AgentChatResponse<T extends AgentResponseContent> {\n  readonly createdAt = Temporal.Now.instant();\n  readonly id = nanoid();\n  readonly session: ChatSession;\n  readonly content: T;\n\n  constructor(chatSession: ChatSession, content: T, createdAt?: Temporal.Instant, id?: string) {\n    this.session = chatSession;\n    this.content = content;\n\n    if (createdAt) {\n      this.createdAt = createdAt;\n    }\n\n    if (id) {\n      this.id = id;\n    }\n  }\n\n  /**\n   * Creates a copy of the AgentResponse with a different content property.\n   * Preserves the original message's ID\n   */\n  replaceContent<R extends AgentResponseContent>(content: R): AgentChatResponse<R> {\n    return new AgentChatResponse(this.session, content, this.createdAt, this.id);\n  }\n\n  toJSON() {\n    return z.encode(AgentChatResponse.codec, this);\n  }\n\n  static fromJSON(data: unknown) {\n    // eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any\n    return z.decode(AgentChatResponse.codec, data as any);\n  }\n\n  static codec = z.codec(chatEventV1Schema, z.instanceof(AgentChatResponse), {\n    decode(chatEvent) {\n      const chatSession = ChatSession.new();\n      const content = (() => {\n        switch (chatEvent.chunkType) {\n          case \"error\":\n            return new AgentErrorResponseContent(chatEvent.message);\n          case \"model\":\n            return new AgentModelResponseContent(chatEvent.name, chatEvent.result);\n          case \"toolRequest\": {\n            return new AgentToolRequestResponseContent(\n              chatEvent.callId,\n              chatEvent.name,\n              Object.keys(chatEvent.arguments)\n                .sort()\n                .map((key) => chatEvent.arguments[key]!),\n              chatEvent.summarizedTitle || null,\n              chatEvent.commentary || null,\n            );\n          }\n          case \"toolResponse\":\n            return new AgentToolResultResponseContent(\n              chatEvent.callId,\n              chatEvent.name,\n              chatEvent.result,\n            );\n        }\n      })();\n      chatSession.assign({\n        ...(\"modelName\" in chatEvent && { modelName: chatEvent.modelName }),\n        ...(\"modelProviderId\" in chatEvent && {\n          modelProviderId: chatEvent.modelProviderId,\n        }),\n        ...(\"sessionId\" in chatEvent && {\n          id: chatEvent.sessionId,\n        }),\n      });\n      return new AgentChatResponse(\n        chatSession,\n        content,\n        \"createdAt\" in chatEvent\n          ? Temporal.Instant.from(chatEvent.createdAt)\n          : Temporal.Now.instant(),\n      );\n    },\n    encode(agentResponse) {\n      const createdAt = agentResponse.createdAt?.toString();\n\n      if (agentResponse.content instanceof AgentErrorResponseContent) {\n        return {\n          chunkType: \"error\",\n          ...agentResponse.content,\n        } as const satisfies ChatEventV1;\n      } else if (agentResponse.content instanceof AgentModelResponseContent) {\n        return {\n          chunkType: \"model\",\n          createdAt: createdAt,\n          modelName: agentResponse.session.modelName || \"\",\n          modelProviderId: agentResponse.session.modelProviderId || \"\",\n          name: agentResponse.content.name,\n          result: agentResponse.content.result,\n          sessionId: agentResponse.session.id!,\n        } as const satisfies ChatEventV1;\n      } else if (agentResponse.content instanceof AgentToolRequestResponseContent) {\n        return {\n          arguments: agentResponse.content.arguments.reduce(\n            (accum: Record<string, unknown>, curr, i) => {\n              accum[`arg${i}`] = curr;\n              return accum;\n            },\n            {} as Record<string, unknown>,\n          ),\n          callId: agentResponse.content.callId,\n          chunkType: \"toolRequest\",\n          commentary: agentResponse.content.commentary || undefined,\n          createdAt: createdAt,\n          modelName: agentResponse.session.modelName || \"\",\n          modelProviderId: agentResponse.session.modelProviderId || \"\",\n          name: agentResponse.content.name,\n          sessionId: agentResponse.session.id!,\n          summarizedTitle: agentResponse.content.summarizedTitle || undefined,\n        } as const satisfies ChatEventV1;\n      } else if (agentResponse.content instanceof AgentToolResultResponseContent) {\n        return {\n          callId: agentResponse.content.callId,\n          chunkType: \"toolResponse\",\n          commentary: agentResponse.content.commentary || undefined,\n          createdAt: createdAt,\n          modelName: agentResponse.session.modelName || \"\",\n          modelProviderId: agentResponse.session.modelProviderId || \"\",\n          name: agentResponse.content.name,\n          result: agentResponse.content.result,\n          sessionId: agentResponse.session.id!,\n        } as const satisfies ChatEventV1;\n      } else {\n        throw new Error(\"Unknown content type\");\n      }\n    },\n  });\n}\n\nexport type AgentResponseContent =\n  | AgentErrorResponseContent\n  | AgentModelResponseContent\n  | AgentToolRequestResponseContent\n  | AgentToolResultResponseContent;\n"]}