@dremio/js-sdk 0.31.8 → 0.32.0

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

@@ -0,0 +1,138 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { nanoid } from "nanoid/non-secure";
+import { AgentErrorResponseContent } from "./AgentErrorResponseContent.js";
+import { AgentModelResponseContent } from "./AgentModelResponseContent.js";
+import { AgentToolRequestResponseContent } from "./AgentToolRequestResponseContent.js";
+import { AgentToolResultResponseContent } from "./AgentToolResultResponseContent.js";
+import { ChatSession } from "./ChatSession.js";
+import * as z from "zod/mini";
+import { Temporal } from "temporal-polyfill";
+import { chatEventSchema } from "./chatEventSchema.js";
+export class AgentChatResponse {
+    createdAt = Temporal.Now.instant();
+    id = nanoid();
+    session;
+    content;
+    constructor(chatSession, content, createdAt, id) {
+        this.session = chatSession;
+        this.content = content;
+        if (createdAt) {
+            this.createdAt = createdAt;
+        }
+        if (id) {
+            this.id = id;
+        }
+    }
+    /**
+     * Creates a copy of the AgentResponse with a different content property.
+     * Preserves the original message's ID
+     */
+    replaceContent(content) {
+        return new AgentChatResponse(this.session, content, this.createdAt, this.id);
+    }
+    toJSON() {
+        return z.encode(AgentChatResponse.codec, this);
+    }
+    static fromJSON(data) {
+        // 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), {
+        decode(chatEvent) {
+            const content = (() => {
+                switch (chatEvent.chunkType) {
+                    case "error":
+                        return new AgentErrorResponseContent(chatEvent.message);
+                    case "model":
+                        return new AgentModelResponseContent(chatEvent.name, chatEvent.result);
+                    case "toolRequest": {
+                        return new AgentToolRequestResponseContent(chatEvent.callId, chatEvent.name, Object.keys(chatEvent.arguments)
+                            .sort()
+                            .map((key) => chatEvent.arguments[key]), chatEvent.summarizedTitle || null, chatEvent.commentary || null);
+                    }
+                    case "toolResponse":
+                        return new AgentToolResultResponseContent(chatEvent.callId, chatEvent.name, chatEvent.result);
+                }
+            })();
+            return new AgentChatResponse(new ChatSession({
+                ...("modelName" in chatEvent && { modelName: chatEvent.modelName }),
+                ...("modelProviderId" in chatEvent && {
+                    modelProviderId: chatEvent.modelProviderId,
+                }),
+                ...("sessionId" in chatEvent && {
+                    sessionId: chatEvent.sessionId,
+                }),
+            }), content, "createdAt" in chatEvent
+                ? Temporal.Instant.from(chatEvent.createdAt)
+                : Temporal.Now.instant());
+        },
+        encode(agentResponse) {
+            const createdAt = agentResponse.createdAt?.toString();
+            if (agentResponse.content instanceof AgentErrorResponseContent) {
+                return {
+                    chunkType: "error",
+                    ...agentResponse.content,
+                };
+            }
+            else if (agentResponse.content instanceof AgentModelResponseContent) {
+                return {
+                    chunkType: "model",
+                    createdAt: createdAt,
+                    modelName: agentResponse.session.modelName || "",
+                    modelProviderId: agentResponse.session.modelProviderId || "",
+                    name: agentResponse.content.name,
+                    result: agentResponse.content.result,
+                    sessionId: agentResponse.session.sessionId,
+                };
+            }
+            else if (agentResponse.content instanceof AgentToolRequestResponseContent) {
+                return {
+                    arguments: agentResponse.content.arguments.reduce((accum, curr, i) => {
+                        accum[`arg${i}`] = curr;
+                        return accum;
+                    }, {}),
+                    callId: agentResponse.content.callId,
+                    chunkType: "toolRequest",
+                    commentary: agentResponse.content.commentary || undefined,
+                    createdAt: createdAt,
+                    modelName: agentResponse.session.modelName || "",
+                    modelProviderId: agentResponse.session.modelProviderId || "",
+                    name: agentResponse.content.name,
+                    sessionId: agentResponse.session.sessionId,
+                    summarizedTitle: agentResponse.content.summarizedTitle || undefined,
+                };
+            }
+            else if (agentResponse.content instanceof AgentToolResultResponseContent) {
+                return {
+                    callId: agentResponse.content.callId,
+                    chunkType: "toolResponse",
+                    commentary: agentResponse.content.commentary || undefined,
+                    createdAt: createdAt,
+                    modelName: agentResponse.session.modelName || "",
+                    modelProviderId: agentResponse.session.modelProviderId || "",
+                    name: agentResponse.content.name,
+                    result: agentResponse.content.result,
+                    sessionId: agentResponse.session.sessionId,
+                };
+            }
+            else {
+                throw new Error("Unknown content type");
+            }
+        },
+    });
+}
+//# sourceMappingURL=AgentChatResponse.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentChatResponse.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/AgentChatResponse.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAAE,+BAA+B,EAAE,MAAM,sCAAsC,CAAC;AACvF,OAAO,EAAE,8BAA8B,EAAE,MAAM,qCAAqC,CAAC;AACrF,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAkB,MAAM,sBAAsB,CAAC;AAEvE,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,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,OAAO,IAAI,iBAAiB,CAC1B,IAAI,WAAW,CAAC;gBACd,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,SAAS,EAAE,SAAS,CAAC,SAAS;iBAC/B,CAAC;aACH,CAAC,EACF,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,SAAU;iBACf,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,SAAU;oBAC3C,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,SAAU;iBACf,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 \"./AgentErrorResponseContent.ts\";\nimport { AgentModelResponseContent } from \"./AgentModelResponseContent.ts\";\nimport { AgentToolRequestResponseContent } from \"./AgentToolRequestResponseContent.ts\";\nimport { AgentToolResultResponseContent } from \"./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 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      return new AgentChatResponse(\n        new ChatSession({\n          ...(\"modelName\" in chatEvent && { modelName: chatEvent.modelName }),\n          ...(\"modelProviderId\" in chatEvent && {\n            modelProviderId: chatEvent.modelProviderId,\n          }),\n          ...(\"sessionId\" in chatEvent && {\n            sessionId: chatEvent.sessionId,\n          }),\n        }),\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.sessionId!,\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.sessionId!,\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.sessionId!,\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"]}

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

@@ -0,0 +1,4 @@
+export declare class AgentErrorResponseContent {
+    readonly message: string;
+    constructor(message: AgentErrorResponseContent["message"]);
+}

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

@@ -0,0 +1,22 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+export class AgentErrorResponseContent {
+    message;
+    constructor(message) {
+        this.message = message;
+    }
+}
+//# sourceMappingURL=AgentErrorResponseContent.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentErrorResponseContent.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/AgentErrorResponseContent.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,MAAM,OAAO,yBAAyB;IAC3B,OAAO,CAAS;IAEzB,YAAY,OAA6C;QACvD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF","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\nexport class AgentErrorResponseContent {\n  readonly message: string;\n\n  constructor(message: AgentErrorResponseContent[\"message\"]) {\n    this.message = message;\n  }\n}\n"]}

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

@@ -0,0 +1,6 @@
+import type { z } from "zod/mini";
+export declare class AgentModelResponseContent<N extends string = string, R extends Record<string, z.core.util.JSONType> = Record<string, z.core.util.JSONType>> {
+    readonly name: N;
+    readonly result: R;
+    constructor(name: N, result: R);
+}

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

@@ -0,0 +1,24 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+export class AgentModelResponseContent {
+    name;
+    result;
+    constructor(name, result) {
+        this.name = name;
+        this.result = result;
+    }
+}
+//# sourceMappingURL=AgentModelResponseContent.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentModelResponseContent.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/AgentModelResponseContent.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,MAAM,OAAO,yBAAyB;IAI3B,IAAI,CAAI;IACR,MAAM,CAAI;IAEnB,YAAY,IAAO,EAAE,MAAS;QAC5B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF","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 type { z } from \"zod/mini\";\n\nexport class AgentModelResponseContent<\n  N extends string = string,\n  R extends Record<string, z.core.util.JSONType> = Record<string, z.core.util.JSONType>,\n> {\n  readonly name: N;\n  readonly result: R;\n\n  constructor(name: N, result: R) {\n    this.name = name;\n    this.result = result;\n  }\n}\n"]}

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

@@ -0,0 +1,9 @@
+import * as z from "zod/mini";
+export declare class AgentToolRequestResponseContent {
+    readonly arguments: z.core.util.JSONType[];
+    readonly callId: string;
+    readonly commentary: string | null;
+    readonly name: string;
+    readonly summarizedTitle: string | null;
+    constructor(callId: AgentToolRequestResponseContent["callId"], name: AgentToolRequestResponseContent["name"], toolArguments: AgentToolRequestResponseContent["arguments"], summarizedTitle: AgentToolRequestResponseContent["summarizedTitle"], commentary: AgentToolRequestResponseContent["commentary"]);
+}

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

@@ -0,0 +1,35 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import * as z from "zod/mini";
+export class AgentToolRequestResponseContent {
+    arguments;
+    callId;
+    commentary = null;
+    name;
+    summarizedTitle = null;
+    constructor(callId, name, toolArguments, summarizedTitle, commentary) {
+        this.callId = callId;
+        this.name = name;
+        this.arguments = toolArguments;
+        if (summarizedTitle) {
+            this.summarizedTitle = summarizedTitle;
+        }
+        if (commentary) {
+            this.commentary = commentary;
+        }
+    }
+}
+//# sourceMappingURL=AgentToolRequestResponseContent.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentToolRequestResponseContent.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/AgentToolRequestResponseContent.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,OAAO,+BAA+B;IACjC,SAAS,CAAyB;IAClC,MAAM,CAAS;IACf,UAAU,GAAkB,IAAI,CAAC;IACjC,IAAI,CAAS;IACb,eAAe,GAAkB,IAAI,CAAC;IAE/C,YACE,MAAiD,EACjD,IAA6C,EAC7C,aAA2D,EAC3D,eAAmE,EACnE,UAAyD;QAEzD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,aAAa,CAAC;QAE/B,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACzC,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/B,CAAC;IACH,CAAC;CACF","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\nexport class AgentToolRequestResponseContent {\n  readonly arguments: z.core.util.JSONType[];\n  readonly callId: string;\n  readonly commentary: string | null = null;\n  readonly name: string;\n  readonly summarizedTitle: string | null = null;\n\n  constructor(\n    callId: AgentToolRequestResponseContent[\"callId\"],\n    name: AgentToolRequestResponseContent[\"name\"],\n    toolArguments: AgentToolRequestResponseContent[\"arguments\"],\n    summarizedTitle: AgentToolRequestResponseContent[\"summarizedTitle\"],\n    commentary: AgentToolRequestResponseContent[\"commentary\"],\n  ) {\n    this.callId = callId;\n    this.name = name;\n    this.arguments = toolArguments;\n\n    if (summarizedTitle) {\n      this.summarizedTitle = summarizedTitle;\n    }\n\n    if (commentary) {\n      this.commentary = commentary;\n    }\n  }\n}\n"]}

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

@@ -0,0 +1,8 @@
+import * as z from "zod/mini";
+export declare class AgentToolResultResponseContent {
+    readonly callId: string;
+    readonly commentary: string | null;
+    readonly name: string;
+    readonly result: Record<string, z.core.util.JSONType>;
+    constructor(callId: AgentToolResultResponseContent["callId"], name: AgentToolResultResponseContent["name"], result: AgentToolResultResponseContent["result"], commentary?: string);
+}

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

@@ -0,0 +1,31 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import * as z from "zod/mini";
+export class AgentToolResultResponseContent {
+    callId;
+    commentary = null;
+    name;
+    result;
+    constructor(callId, name, result, commentary) {
+        this.callId = callId;
+        this.name = name;
+        this.result = result;
+        if (commentary) {
+            this.commentary = commentary;
+        }
+    }
+}
+//# sourceMappingURL=AgentToolResultResponseContent.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentToolResultResponseContent.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/AgentToolResultResponseContent.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,OAAO,8BAA8B;IAChC,MAAM,CAAS;IACf,UAAU,GAAkB,IAAI,CAAC;IACjC,IAAI,CAAS;IACb,MAAM,CAAuC;IAEtD,YACE,MAAgD,EAChD,IAA4C,EAC5C,MAAgD,EAChD,UAAmB;QAEnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/B,CAAC;IACH,CAAC;CACF","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\nexport class AgentToolResultResponseContent {\n  readonly callId: string;\n  readonly commentary: string | null = null;\n  readonly name: string;\n  readonly result: Record<string, z.core.util.JSONType>;\n\n  constructor(\n    callId: AgentToolResultResponseContent[\"callId\"],\n    name: AgentToolResultResponseContent[\"name\"],\n    result: AgentToolResultResponseContent[\"result\"],\n    commentary?: string,\n  ) {\n    this.callId = callId;\n    this.name = name;\n    this.result = result;\n\n    if (commentary) {\n      this.commentary = commentary;\n    }\n  }\n}\n"]}

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

@@ -0,0 +1,32 @@
+import * as z from "zod/mini";
+declare const chatSessionInputSchema: z.ZodMiniObject<{
+    modelName: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    modelProviderId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    sessionId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+}, z.core.$strip>;
+/**
+ * Tracks an Agent chat session. Each field is optional and intentionally mutable
+ * so that if any of these fields become available during agent reply (such as which
+ * default model provider and name it picked, as well as the server-generated sessionId),
+ * they're mutated on the instance of the class.
+ */
+export declare class ChatSession {
+    modelName?: string;
+    modelProviderId?: string;
+    sessionId?: string;
+    constructor(properties?: z.infer<typeof chatSessionInputSchema>);
+    toJSON(): {
+        modelName?: string | undefined;
+        modelProviderId?: string | undefined;
+        sessionId?: string | undefined;
+    };
+    assign(properties: z.infer<typeof chatSessionInputSchema>): void;
+    equals(other: ChatSession): boolean;
+    static fromJSON(data: z.input<typeof ChatSession.codec>): ChatSession;
+    static codec: z.ZodMiniCodec<z.ZodMiniObject<{
+        modelName: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        modelProviderId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        sessionId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    }, z.core.$strip>, z.ZodMiniCustom<ChatSession, ChatSession>>;
+}
+export {};

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

@@ -0,0 +1,77 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import * as z from "zod/mini";
+const chatSessionInputSchema = z.object({
+    modelName: z.optional(z.string()),
+    modelProviderId: z.optional(z.string()),
+    sessionId: z.optional(z.string()),
+});
+/**
+ * Tracks an Agent chat session. Each field is optional and intentionally mutable
+ * so that if any of these fields become available during agent reply (such as which
+ * default model provider and name it picked, as well as the server-generated sessionId),
+ * they're mutated on the instance of the class.
+ */
+export class ChatSession {
+    modelName;
+    modelProviderId;
+    sessionId;
+    constructor(properties = {}) {
+        if (properties.modelName) {
+            this.modelName = properties.modelName;
+        }
+        if (properties.modelProviderId) {
+            this.modelProviderId = properties.modelProviderId;
+        }
+        if (properties.sessionId) {
+            this.sessionId = properties.sessionId;
+        }
+    }
+    toJSON() {
+        return z.encode(ChatSession.codec, this);
+    }
+    assign(properties) {
+        if (properties.modelName) {
+            this.modelName = properties.modelName;
+        }
+        if (properties.modelProviderId) {
+            this.modelProviderId = properties.modelProviderId;
+        }
+        if (properties.sessionId) {
+            this.sessionId = properties.sessionId;
+        }
+    }
+    equals(other) {
+        return (this === other ||
+            (this.sessionId === other.sessionId && (!!this.sessionId || !!other.sessionId)));
+    }
+    static fromJSON(data) {
+        return z.decode(ChatSession.codec, data);
+    }
+    static codec = z.codec(chatSessionInputSchema, z.instanceof(ChatSession), {
+        decode(v) {
+            return new ChatSession(v);
+        },
+        encode(v) {
+            return {
+                modelName: v.modelName,
+                modelProviderId: v.modelProviderId,
+                sessionId: v.sessionId,
+            };
+        },
+    });
+}
+//# sourceMappingURL=ChatSession.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ChatSession.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/ChatSession.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,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;IACvC,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAClC,CAAC,CAAC;AAEH;;;;;GAKG;AACH,MAAM,OAAO,WAAW;IACtB,SAAS,CAAU;IACnB,eAAe,CAAU;IACzB,SAAS,CAAU;IAEnB,YAAY,aAAqD,EAAE;QACjE,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACxC,CAAC;QACD,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;YAC/B,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;QACpD,CAAC;QAED,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACxC,CAAC;IACH,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,UAAkD;QACvD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACxC,CAAC;QACD,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;YAC/B,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;QACpD,CAAC;QAED,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACxC,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAkB;QACvB,OAAO,CACL,IAAI,KAAK,KAAK;YACd,CAAC,IAAI,CAAC,SAAS,KAAK,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAChF,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAuC;QACrD,OAAO,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,sBAAsB,EAAE,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE;QACxE,MAAM,CAAC,CAAC;YACN,OAAO,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC;QACD,MAAM,CAAC,CAAC;YACN,OAAO;gBACL,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,eAAe,EAAE,CAAC,CAAC,eAAe;gBAClC,SAAS,EAAE,CAAC,CAAC,SAAS;aACvB,CAAC;QACJ,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 * as z from \"zod/mini\";\n\nconst chatSessionInputSchema = z.object({\n  modelName: z.optional(z.string()),\n  modelProviderId: z.optional(z.string()),\n  sessionId: z.optional(z.string()),\n});\n\n/**\n * Tracks an Agent chat session. Each field is optional and intentionally mutable\n * so that if any of these fields become available during agent reply (such as which\n * default model provider and name it picked, as well as the server-generated sessionId),\n * they're mutated on the instance of the class.\n */\nexport class ChatSession {\n  modelName?: string;\n  modelProviderId?: string;\n  sessionId?: string;\n\n  constructor(properties: z.infer<typeof chatSessionInputSchema> = {}) {\n    if (properties.modelName) {\n      this.modelName = properties.modelName;\n    }\n    if (properties.modelProviderId) {\n      this.modelProviderId = properties.modelProviderId;\n    }\n\n    if (properties.sessionId) {\n      this.sessionId = properties.sessionId;\n    }\n  }\n\n  toJSON() {\n    return z.encode(ChatSession.codec, this);\n  }\n\n  assign(properties: z.infer<typeof chatSessionInputSchema>) {\n    if (properties.modelName) {\n      this.modelName = properties.modelName;\n    }\n    if (properties.modelProviderId) {\n      this.modelProviderId = properties.modelProviderId;\n    }\n\n    if (properties.sessionId) {\n      this.sessionId = properties.sessionId;\n    }\n  }\n\n  equals(other: ChatSession) {\n    return (\n      this === other ||\n      (this.sessionId === other.sessionId && (!!this.sessionId || !!other.sessionId))\n    );\n  }\n\n  static fromJSON(data: z.input<typeof ChatSession.codec>) {\n    return z.decode(ChatSession.codec, data);\n  }\n\n  static codec = z.codec(chatSessionInputSchema, z.instanceof(ChatSession), {\n    decode(v) {\n      return new ChatSession(v);\n    },\n    encode(v) {\n      return {\n        modelName: v.modelName,\n        modelProviderId: v.modelProviderId,\n        sessionId: v.sessionId,\n      };\n    },\n  });\n}\n"]}

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

@@ -0,0 +1,30 @@
+import * as z from "zod/mini";
+import { Temporal } from "temporal-polyfill";
+import { type UserChatMessageContent } from "./UserChatMessageContent.ts";
+export declare class UserChatMessage {
+    readonly createdAt: Temporal.Instant;
+    readonly content: UserChatMessageContent;
+    readonly id: string;
+    readonly context?: Record<string, z.core.util.JSONType>;
+    constructor(content: UserChatMessage["content"], context?: UserChatMessage["context"], id?: string, createdAt?: Temporal.Instant);
+    withContext(context: UserChatMessage["context"]): UserChatMessage;
+    mergeContext(context: NonNullable<UserChatMessage["context"]>): UserChatMessage;
+    toString(): string;
+    toJSON(): {
+        content: {
+            content: string;
+            type: "string";
+        };
+        createdAt: string;
+        id: string;
+    };
+    static fromJSON(data: z.input<typeof UserChatMessage.codec>): UserChatMessage;
+    static codec: z.ZodMiniCodec<z.ZodMiniObject<{
+        content: z.ZodMiniObject<{
+            content: z.ZodMiniString<string>;
+            type: z.ZodMiniLiteral<"string">;
+        }, z.core.$strip>;
+        createdAt: z.ZodMiniString<string>;
+        id: z.ZodMiniString<string>;
+    }, z.core.$strip>, z.ZodMiniCustom<UserChatMessage, UserChatMessage>>;
+}

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

@@ -0,0 +1,68 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { nanoid } from "nanoid/non-secure";
+import * as z from "zod/mini";
+import { Temporal } from "temporal-polyfill";
+import { userChatMessageContentCodec, } from "./UserChatMessageContent.js";
+export class UserChatMessage {
+    createdAt = Temporal.Now.instant();
+    content;
+    id = nanoid();
+    context;
+    constructor(content, context, id, createdAt) {
+        this.content = content;
+        this.context = context;
+        if (id) {
+            this.id = id;
+        }
+        if (createdAt) {
+            this.createdAt = createdAt;
+        }
+    }
+    withContext(context) {
+        return new UserChatMessage(this.content, context, this.id, this.createdAt);
+    }
+    mergeContext(context) {
+        return new UserChatMessage(this.content, { ...this.context, ...context }, this.id);
+    }
+    toString() {
+        return (this.content.content +
+            (this.context ? `\n\n<system-context>${JSON.stringify(this.context)}</system-context>` : ""));
+    }
+    toJSON() {
+        return z.encode(UserChatMessage.codec, this);
+    }
+    static fromJSON(data) {
+        return z.decode(UserChatMessage.codec, data);
+    }
+    static codec = z.codec(z.object({
+        content: userChatMessageContentCodec.def.in,
+        createdAt: z.string().check(z.iso.datetime()),
+        id: z.string(),
+    }), z.instanceof(UserChatMessage), {
+        decode(v) {
+            return new UserChatMessage(z.decode(userChatMessageContentCodec, v.content), undefined, v.id);
+        },
+        encode(v) {
+            return {
+                ...v,
+                content: z.encode(userChatMessageContentCodec, v.content),
+                createdAt: v.createdAt.toString(),
+            };
+        },
+    });
+}
+//# sourceMappingURL=UserChatMessage.js.map

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

@@ -0,0 +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,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EACL,2BAA2B,GAE5B,MAAM,6BAA6B,CAAC;AAErC,MAAM,OAAO,eAAe;IACjB,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnC,OAAO,CAAyB;IAChC,EAAE,GAAG,MAAM,EAAE,CAAC;IACd,OAAO,CAAwC;IAExD,YACE,OAAmC,EACnC,OAAoC,EACpC,EAAW,EACX,SAA4B;QAE5B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,WAAW,CAAC,OAAmC;QAC7C,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7E,CAAC;IAED,YAAY,CAAC,OAAgD;QAC3D,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,OAAO,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IACrF,CAAC;IAED,QAAQ;QACN,OAAO,CACL,IAAI,CAAC,OAAO,CAAC,OAAO;YACpB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC,CAC7F,CAAC;IACJ,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAA2C;QACzD,OAAO,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpB,CAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,2BAA2B,CAAC,GAAG,CAAC,EAAE;QAC3C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC7C,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;KACf,CAAC,EACF,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,EAC7B;QACE,MAAM,CAAC,CAAC;YACN,OAAO,IAAI,eAAe,CACxB,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC,CAAC,OAAO,CAAC,EAChD,SAAS,EACT,CAAC,CAAC,EAAE,CACL,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,CAAC;YACN,OAAO;gBACL,GAAG,CAAC;gBACJ,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC,CAAC,OAAO,CAAC;gBACzD,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;aACzB,CAAC;QACb,CAAC;KACF,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 { nanoid } from \"nanoid/non-secure\";\nimport * as z from \"zod/mini\";\nimport { Temporal } from \"temporal-polyfill\";\nimport {\n  userChatMessageContentCodec,\n  type UserChatMessageContent,\n} from \"./UserChatMessageContent.ts\";\n\nexport class UserChatMessage {\n  readonly createdAt = Temporal.Now.instant();\n  readonly content: UserChatMessageContent;\n  readonly id = nanoid();\n  readonly context?: Record<string, z.core.util.JSONType>;\n\n  constructor(\n    content: UserChatMessage[\"content\"],\n    context?: UserChatMessage[\"context\"],\n    id?: string,\n    createdAt?: Temporal.Instant,\n  ) {\n    this.content = content;\n    this.context = context;\n\n    if (id) {\n      this.id = id;\n    }\n\n    if (createdAt) {\n      this.createdAt = createdAt;\n    }\n  }\n\n  withContext(context: UserChatMessage[\"context\"]) {\n    return new UserChatMessage(this.content, context, this.id, this.createdAt);\n  }\n\n  mergeContext(context: NonNullable<UserChatMessage[\"context\"]>) {\n    return new UserChatMessage(this.content, { ...this.context, ...context }, this.id);\n  }\n\n  toString() {\n    return (\n      this.content.content +\n      (this.context ? `\\n\\n<system-context>${JSON.stringify(this.context)}</system-context>` : \"\")\n    );\n  }\n\n  toJSON() {\n    return z.encode(UserChatMessage.codec, this);\n  }\n\n  static fromJSON(data: z.input<typeof UserChatMessage.codec>) {\n    return z.decode(UserChatMessage.codec, data);\n  }\n\n  static codec = z.codec(\n    z.object({\n      content: userChatMessageContentCodec.def.in,\n      createdAt: z.string().check(z.iso.datetime()),\n      id: z.string(),\n    }),\n    z.instanceof(UserChatMessage),\n    {\n      decode(v) {\n        return new UserChatMessage(\n          z.decode(userChatMessageContentCodec, v.content),\n          undefined,\n          v.id,\n        );\n      },\n      encode(v) {\n        return {\n          ...v,\n          content: z.encode(userChatMessageContentCodec, v.content),\n          createdAt: v.createdAt.toString(),\n        } as const;\n      },\n    },\n  );\n}\n"]}

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

@@ -0,0 +1,19 @@
+import * as z from "zod/mini";
+export declare class UserChatMessageStringContent implements UserChatMessageContent {
+    readonly content: string;
+    constructor(content: string);
+    toJSON(): {
+        content: string;
+        type: "string";
+    };
+    static fromJSON(data: z.input<typeof UserChatMessageStringContent.codec>): UserChatMessageStringContent;
+    static codec: z.ZodMiniCodec<z.ZodMiniObject<{
+        content: z.ZodMiniString<string>;
+        type: z.ZodMiniLiteral<"string">;
+    }, z.core.$strip>, z.ZodMiniCustom<UserChatMessageStringContent, UserChatMessageStringContent>>;
+}
+export declare const userChatMessageContentCodec: z.ZodMiniCodec<z.ZodMiniObject<{
+    content: z.ZodMiniString<string>;
+    type: z.ZodMiniLiteral<"string">;
+}, z.core.$strip>, z.ZodMiniCustom<UserChatMessageStringContent, UserChatMessageStringContent>>;
+export type UserChatMessageContent = UserChatMessageStringContent;

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

@@ -0,0 +1,44 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import * as z from "zod/mini";
+export class UserChatMessageStringContent {
+    content;
+    constructor(content) {
+        this.content = content;
+    }
+    toJSON() {
+        return z.encode(UserChatMessageStringContent.codec, this);
+    }
+    static fromJSON(data) {
+        return z.decode(UserChatMessageStringContent.codec, data);
+    }
+    static codec = z.codec(z.object({
+        content: z.string(),
+        type: z.literal("string"),
+    }), z.instanceof(UserChatMessageStringContent), {
+        decode(v) {
+            return new UserChatMessageStringContent(v.content);
+        },
+        encode(v) {
+            return {
+                content: v.content,
+                type: "string",
+            };
+        },
+    });
+}
+export const userChatMessageContentCodec = UserChatMessageStringContent.codec;
+//# sourceMappingURL=UserChatMessageContent.js.map