npm - @dremio/js-sdk - Versions diffs - 0.39.1 → 0.40.1 - Mend

@dremio/js-sdk 0.39.1 → 0.40.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (74) hide show

package/dist/enterprise/ai/chat/{AgentChatPartialResponse.d.ts → eventWrappers/AgentChatPartialResponse.d.ts} RENAMED Viewed

@@ -1,4 +1,4 @@
-import { ChatSession } from "./ChatSession.ts";
+import { ChatSession } from "../ChatSession.ts";
 import * as z from "zod/mini";
 declare const agentPartialResponseTypeSchema: z.ZodMiniEnum<{
     delta: "delta";
@@ -16,9 +16,12 @@ export declare class AgentChatPartialResponse {
         content: string;
         id: string;
         session: {
+            createdAt: string;
+            id: string;
             modelName: string;
             modelProviderId: string;
-            sessionId: string;
+            updatedAt: string;
+            title?: string | undefined;
         };
         type: "delta" | "end" | "start";
     };
@@ -28,9 +31,12 @@ 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>;
-            sessionId: 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/{AgentChatPartialResponse.js → eventWrappers/AgentChatPartialResponse.js} RENAMED Viewed

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 import { nanoid } from "nanoid/non-secure";
-import { ChatSession } from "./ChatSession.js";
+import { ChatSession } from "../ChatSession.js";
 import * as z from "zod/mini";
 const agentPartialResponseTypeSchema = z.enum(["delta", "end", "start"]);
 export class AgentChatPartialResponse {

package/dist/enterprise/ai/chat/eventWrappers/AgentChatPartialResponse.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"AgentChatPartialResponse.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentChatPartialResponse.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,8BAA8B,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;AAEzE,MAAM,OAAO,wBAAwB;IAC1B,EAAE,GAAG,MAAM,EAAE,CAAC;IACd,OAAO,CAAc;IACrB,OAAO,CAAS;IAChB,IAAI,CAAiD;IAE9D,YACE,WAAgD,EAChD,IAAsC,EACtC,OAA4C,EAC5C,EAAmC;QAEnC,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC;QAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IAED,MAAM,CAAC,IAA8B;QACnC,OAAO,IAAI,wBAAwB,CACjC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,EAC3B,IAAI,CAAC,EAAE,CACR,CAAC;IACJ,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,wBAAwB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACxD,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,IAAI,KAAK,KAAK,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAa;QAC3B,qGAAqG;QACrG,OAAO,CAAC,CAAC,MAAM,CAAC,wBAAwB,CAAC,KAAK,EAAE,IAAW,CAAC,CAAC;IAC/D,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpB,CAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;QACnB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;QACd,OAAO,EAAE,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE;QACjC,IAAI,EAAE,8BAA8B;KACrC,CAAC,EACF,CAAC,CAAC,UAAU,CAAC,wBAAwB,CAAC,EACtC;QACE,MAAM,CAAC,CAAC;YACN,OAAO,IAAI,wBAAwB,CACjC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,EAC/B,CAAC,CAAC,IAAI,EACN,CAAC,CAAC,OAAO,EACT,CAAC,CAAC,EAAE,CACL,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,CAAC;YACN,OAAO;gBACL,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE;gBAC3B,IAAI,EAAE,CAAC,CAAC,IAAI;aACJ,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 { ChatSession } from \"../ChatSession.ts\";\nimport * as z from \"zod/mini\";\n\nconst agentPartialResponseTypeSchema = z.enum([\"delta\", \"end\", \"start\"]);\n\nexport class AgentChatPartialResponse {\n readonly id = nanoid();\n readonly session: ChatSession;\n readonly content: string;\n readonly type: z.infer<typeof agentPartialResponseTypeSchema>;\n\n constructor(\n chatSession: AgentChatPartialResponse[\"session\"],\n type: AgentChatPartialResponse[\"type\"],\n content: AgentChatPartialResponse[\"content\"],\n id?: AgentChatPartialResponse[\"id\"],\n ) {\n this.session = chatSession;\n this.type = type;\n this.content = content;\n\n if (id) {\n this.id = id;\n }\n }\n\n concat(next: AgentChatPartialResponse) {\n return new AgentChatPartialResponse(\n this.session,\n next.type,\n this.content + next.content,\n this.id,\n );\n }\n\n toJSON() {\n return z.encode(AgentChatPartialResponse.codec, this);\n }\n\n get isComplete() {\n return this.type === \"end\";\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(AgentChatPartialResponse.codec, data as any);\n }\n\n static codec = z.codec(\n z.object({\n content: z.string(),\n id: z.string(),\n session: ChatSession.codec.def.in,\n type: agentPartialResponseTypeSchema,\n }),\n z.instanceof(AgentChatPartialResponse),\n {\n decode(v) {\n return new AgentChatPartialResponse(\n ChatSession.fromJSON(v.session),\n v.type,\n v.content,\n v.id,\n );\n },\n encode(v) {\n return {\n content: v.content,\n id: v.id,\n session: v.session.toJSON(),\n type: v.type,\n } as const;\n },\n },\n );\n}\n"]}

package/dist/enterprise/ai/chat/{AgentChatResponse.d.ts → eventWrappers/AgentChatResponse.d.ts} RENAMED Viewed

@@ -1,8 +1,8 @@
-import { AgentErrorResponseContent } from "./AgentErrorResponseContent.ts";
-import { AgentModelResponseContent } from "./AgentModelResponseContent.ts";
-import { AgentToolRequestResponseContent } from "./AgentToolRequestResponseContent.ts";
-import { AgentToolResultResponseContent } from "./AgentToolResultResponseContent.ts";
-import { ChatSession } from "./ChatSession.ts";
+import { AgentErrorResponseContent } from "../responseContent/AgentErrorResponseContent.ts";
+import { AgentModelResponseContent } from "../responseContent/AgentModelResponseContent.ts";
+import { AgentToolRequestResponseContent } from "../responseContent/AgentToolRequestResponseContent.ts";
+import { AgentToolResultResponseContent } from "../responseContent/AgentToolResultResponseContent.ts";
+import { ChatSession } from "../ChatSession.ts";
 import * as z from "zod/mini";
 import { Temporal } from "temporal-polyfill";
 export declare class AgentChatResponse<T extends AgentResponseContent> {

package/dist/enterprise/ai/chat/{AgentChatResponse.js → eventWrappers/AgentChatResponse.js} RENAMED Viewed

@@ -14,14 +14,14 @@
  * 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 { AgentErrorResponseContent } from "../responseContent/AgentErrorResponseContent.js";
+import { AgentModelResponseContent } from "../responseContent/AgentModelResponseContent.js";
+import { AgentToolRequestResponseContent } from "../responseContent/AgentToolRequestResponseContent.js";
+import { AgentToolResultResponseContent } from "../responseContent/AgentToolResultResponseContent.js";
+import { ChatSession } from "../ChatSession.js";
 import * as z from "zod/mini";
 import { Temporal } from "temporal-polyfill";
-import { chatEventSchema } from "./chatEventSchema.js";
+import { chatEventSchema } from "../chatEventSchema.js";
 export class AgentChatResponse {
     createdAt = Temporal.Now.instant();
     id = nanoid();
@@ -75,7 +75,7 @@ export class AgentChatResponse {
                     modelProviderId: chatEvent.modelProviderId,
                 }),
                 ...("sessionId" in chatEvent && {
-                    sessionId: chatEvent.sessionId,
+                    id: chatEvent.sessionId,
                 }),
             });
             return new AgentChatResponse(chatSession, content, "createdAt" in chatEvent
@@ -98,7 +98,7 @@ export class AgentChatResponse {
                     modelProviderId: agentResponse.session.modelProviderId || "",
                     name: agentResponse.content.name,
                     result: agentResponse.content.result,
-                    sessionId: agentResponse.session.sessionId,
+                    sessionId: agentResponse.session.id,
                 };
             }
             else if (agentResponse.content instanceof AgentToolRequestResponseContent) {
@@ -114,7 +114,7 @@ export class AgentChatResponse {
                     modelName: agentResponse.session.modelName || "",
                     modelProviderId: agentResponse.session.modelProviderId || "",
                     name: agentResponse.content.name,
-                    sessionId: agentResponse.session.sessionId,
+                    sessionId: agentResponse.session.id,
                     summarizedTitle: agentResponse.content.summarizedTitle || undefined,
                 };
             }
@@ -128,7 +128,7 @@ export class AgentChatResponse {
                     modelProviderId: agentResponse.session.modelProviderId || "",
                     name: agentResponse.content.name,
                     result: agentResponse.content.result,
-                    sessionId: agentResponse.session.sessionId,
+                    sessionId: agentResponse.session.id,
                 };
             }
             else {

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

@@ -0,0 +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"]}

package/dist/enterprise/ai/chat/eventWrappers/AgentToolCall.d.ts CHANGED Viewed

@@ -1,6 +1,6 @@
-import { AgentChatResponse } from "../AgentChatResponse.ts";
-import { AgentToolRequestResponseContent } from "../AgentToolRequestResponseContent.ts";
-import { AgentToolResultResponseContent } from "../AgentToolResultResponseContent.ts";
+import { AgentChatResponse } from "./AgentChatResponse.ts";
+import { AgentToolRequestResponseContent } from "../responseContent/AgentToolRequestResponseContent.ts";
+import { AgentToolResultResponseContent } from "../responseContent/AgentToolResultResponseContent.ts";
 /**
  * Pairs a tool request with its associated tool result once completed.
  */

package/dist/enterprise/ai/chat/eventWrappers/AgentToolCall.js CHANGED Viewed

@@ -13,9 +13,9 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { AgentChatResponse } from "../AgentChatResponse.js";
-import { AgentToolRequestResponseContent } from "../AgentToolRequestResponseContent.js";
-import { AgentToolResultResponseContent } from "../AgentToolResultResponseContent.js";
+import { AgentChatResponse } from "./AgentChatResponse.js";
+import { AgentToolRequestResponseContent } from "../responseContent/AgentToolRequestResponseContent.js";
+import { AgentToolResultResponseContent } from "../responseContent/AgentToolResultResponseContent.js";
 /**
  * Pairs a tool request with its associated tool result once completed.
  */

package/dist/enterprise/ai/chat/eventWrappers/AgentToolCall.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"AgentToolCall.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentToolCall.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,~~yBAAyB~~,CAAC;~~AAC5D~~,OAAO,EAAE,+BAA+B,EAAE,MAAM,~~uCAAuC~~,CAAC;~~AACxF~~,OAAO,EAAE,8BAA8B,EAAE,MAAM,~~sCAAsC~~,CAAC;~~AAEtF~~;;GAEG;AACH,MAAM,OAAO,aAAa;IACf,OAAO,CAAqD;IAC5D,MAAM,CAAqD;IAEpE,YAAY,OAAiC,EAAE,MAAgC;QAC7E,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;IACrC,CAAC;IAED,IAAI,MAAM;QACR,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACjD,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,MAAM,KAAK,OAAO,CAAC;IACjC,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC;IACnC,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,QAA2D;QACpE,OAAO,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACnD,CAAC;IAED,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE;SACnB,CAAC;IACf,CAAC;CACF;AAED,MAAM,UAAU,aAAa,CAC3B,OAAgB;IAEhB,OAAO,CACL,OAAO,YAAY,iBAAiB;QACpC,OAAO,CAAC,OAAO,YAAY,+BAA+B,CAC3D,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,OAAgB;IAEhB,OAAO,CACL,OAAO,YAAY,iBAAiB;QACpC,OAAO,CAAC,OAAO,YAAY,8BAA8B,CAC1D,CAAC;AACJ,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 { AgentChatResponse } from \"~~../~~AgentChatResponse.ts\";\nimport { AgentToolRequestResponseContent } from \"../AgentToolRequestResponseContent.ts\";\nimport { AgentToolResultResponseContent } from \"../AgentToolResultResponseContent.ts\";\n\n/\n Pairs a tool request with its associated tool result once completed.\n */\nexport class AgentToolCall {\n readonly request: AgentChatResponse<AgentToolRequestResponseContent>;\n readonly result?: AgentChatResponse<AgentToolResultResponseContent>;\n\n constructor(request: AgentToolCall[\"request\"], result?: AgentToolCall[\"result\"]) {\n this.request = request;\n this.result = result;\n }\n\n get id() {\n return this.request.content.callId;\n }\n\n get status() {\n if (!this.result) {\n return \"pending\";\n }\n\n if (\"errorMessage\" in this.result.content.result) {\n return \"error\";\n }\n\n return \"success\";\n }\n\n get isError() {\n return this.status === \"error\";\n }\n\n get isPending() {\n return this.status === \"pending\";\n }\n\n get isSuccess() {\n return this.status === \"success\";\n }\n\n withResult(response: AgentChatResponse<AgentToolResultResponseContent>) {\n return new AgentToolCall(this.request, response);\n }\n\n toJSON() {\n return {\n id: this.id,\n request: this.request.toJSON(),\n result: this.result?.toJSON(),\n } as unknown;\n }\n}\n\nexport function isToolRequest(\n message: unknown,\n): message is AgentChatResponse<AgentToolRequestResponseContent> {\n return (\n message instanceof AgentChatResponse &&\n message.content instanceof AgentToolRequestResponseContent\n );\n}\n\nexport function isToolResult(\n message: unknown,\n): message is AgentChatResponse<AgentToolResultResponseContent> {\n return (\n message instanceof AgentChatResponse &&\n message.content instanceof AgentToolResultResponseContent\n );\n}\n"]}
1	+ {"version":3,"file":"AgentToolCall.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentToolCall.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,+BAA+B,EAAE,MAAM,uDAAuD,CAAC;AACxG,OAAO,EAAE,8BAA8B,EAAE,MAAM,sDAAsD,CAAC;AAEtG;;GAEG;AACH,MAAM,OAAO,aAAa;IACf,OAAO,CAAqD;IAC5D,MAAM,CAAqD;IAEpE,YAAY,OAAiC,EAAE,MAAgC;QAC7E,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;IACrC,CAAC;IAED,IAAI,MAAM;QACR,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACjD,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,MAAM,KAAK,OAAO,CAAC;IACjC,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC;IACnC,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,QAA2D;QACpE,OAAO,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACnD,CAAC;IAED,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE;SACnB,CAAC;IACf,CAAC;CACF;AAED,MAAM,UAAU,aAAa,CAC3B,OAAgB;IAEhB,OAAO,CACL,OAAO,YAAY,iBAAiB;QACpC,OAAO,CAAC,OAAO,YAAY,+BAA+B,CAC3D,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,OAAgB;IAEhB,OAAO,CACL,OAAO,YAAY,iBAAiB;QACpC,OAAO,CAAC,OAAO,YAAY,8BAA8B,CAC1D,CAAC;AACJ,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 { AgentChatResponse } from \"./AgentChatResponse.ts\";\nimport { AgentToolRequestResponseContent } from \"../responseContent/AgentToolRequestResponseContent.ts\";\nimport { AgentToolResultResponseContent } from \"../responseContent/AgentToolResultResponseContent.ts\";\n\n/\n Pairs a tool request with its associated tool result once completed.\n */\nexport class AgentToolCall {\n readonly request: AgentChatResponse<AgentToolRequestResponseContent>;\n readonly result?: AgentChatResponse<AgentToolResultResponseContent>;\n\n constructor(request: AgentToolCall[\"request\"], result?: AgentToolCall[\"result\"]) {\n this.request = request;\n this.result = result;\n }\n\n get id() {\n return this.request.content.callId;\n }\n\n get status() {\n if (!this.result) {\n return \"pending\";\n }\n\n if (\"errorMessage\" in this.result.content.result) {\n return \"error\";\n }\n\n return \"success\";\n }\n\n get isError() {\n return this.status === \"error\";\n }\n\n get isPending() {\n return this.status === \"pending\";\n }\n\n get isSuccess() {\n return this.status === \"success\";\n }\n\n withResult(response: AgentChatResponse<AgentToolResultResponseContent>) {\n return new AgentToolCall(this.request, response);\n }\n\n toJSON() {\n return {\n id: this.id,\n request: this.request.toJSON(),\n result: this.result?.toJSON(),\n } as unknown;\n }\n}\n\nexport function isToolRequest(\n message: unknown,\n): message is AgentChatResponse<AgentToolRequestResponseContent> {\n return (\n message instanceof AgentChatResponse &&\n message.content instanceof AgentToolRequestResponseContent\n );\n}\n\nexport function isToolResult(\n message: unknown,\n): message is AgentChatResponse<AgentToolResultResponseContent> {\n return (\n message instanceof AgentChatResponse &&\n message.content instanceof AgentToolResultResponseContent\n );\n}\n"]}

package/dist/enterprise/ai/chat/eventWrappers/AgentToolCallGroup.d.ts CHANGED Viewed

@@ -1,7 +1,7 @@
 import { AgentToolCall } from "./AgentToolCall.ts";
-import type { AgentChatResponse } from "../AgentChatResponse.ts";
-import type { AgentToolRequestResponseContent } from "../AgentToolRequestResponseContent.ts";
-import type { AgentToolResultResponseContent } from "../AgentToolResultResponseContent.ts";
+import type { AgentChatResponse } from "./AgentChatResponse.ts";
+import type { AgentToolRequestResponseContent } from "../responseContent/AgentToolRequestResponseContent.ts";
+import type { AgentToolResultResponseContent } from "../responseContent/AgentToolResultResponseContent.ts";
 /**
  * Contains a collection of `AgentToolCall` associated with an `AgentChatExchange`.
  * As tool calls stream in, this class is updated immutably but maintains

package/dist/enterprise/ai/chat/eventWrappers/AgentToolCallGroup.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"AgentToolCallGroup.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentToolCallGroup.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAKnD;;;;GAIG;AACH,MAAM,OAAO,kBAAkB;IACpB,EAAE,GAAG,MAAM,EAAE,CAAC;IACd,SAAS,CAAkB;IAEpC,YAAY,SAA0C,EAAE,EAA6B;QACnF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IAED,UAAU,CAAC,WAA+D;QACxE,OAAO,IAAI,kBAAkB,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,aAAa,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9F,CAAC;IAED,SAAS,CAAC,UAA6D;QACrE,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAC/C,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,UAAU,CAAC,OAAO,CAAC,MAAM,CACxD,CAAC;QAEF,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,YAAY,CAAE,CAAC;QAElD,OAAO,IAAI,kBAAkB,CAC3B;YACE,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC;YACxC,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC;YAC/B,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC;SAC1C,EACD,IAAI,CAAC,EAAE,CACR,CAAC;IACJ,CAAC;IAED,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;SAC/D,CAAC;IACJ,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 { nanoid } from \"nanoid/non-secure\";\nimport { AgentToolCall } from \"./AgentToolCall.ts\";\nimport type { AgentChatResponse } from \"~~../~~AgentChatResponse.ts\";\nimport type { AgentToolRequestResponseContent } from \"../AgentToolRequestResponseContent.ts\";\nimport type { AgentToolResultResponseContent } from \"../AgentToolResultResponseContent.ts\";\n\n/\n Contains a collection of `AgentToolCall` associated with an `AgentChatExchange`.\n * As tool calls stream in, this class is updated immutably but maintains\n * a stable `.id` property.\n */\nexport class AgentToolCallGroup {\n readonly id = nanoid();\n readonly toolCalls: AgentToolCall[];\n\n constructor(toolCalls: AgentToolCallGroup[\"toolCalls\"], id?: AgentToolCallGroup[\"id\"]) {\n this.toolCalls = toolCalls;\n if (id) {\n this.id = id;\n }\n }\n\n addRequest(toolRequest: AgentChatResponse<AgentToolRequestResponseContent>) {\n return new AgentToolCallGroup([...this.toolCalls, new AgentToolCall(toolRequest)], this.id);\n }\n\n addResult(toolResult: AgentChatResponse<AgentToolResultResponseContent>) {\n const requestIndex = this.toolCalls.findLastIndex(\n (toolCall) => toolCall.id === toolResult.content.callId,\n );\n\n if (requestIndex === -1) {\n return this;\n }\n\n const toolCall = this.toolCalls.at(requestIndex)!;\n\n return new AgentToolCallGroup(\n [\n ...this.toolCalls.slice(0, requestIndex),\n toolCall.withResult(toolResult),\n ...this.toolCalls.slice(requestIndex + 1),\n ],\n this.id,\n );\n }\n\n toJSON() {\n return {\n id: this.id,\n toolCalls: this.toolCalls.map((toolCall) => toolCall.toJSON()),\n };\n }\n}\n"]}
1	+ {"version":3,"file":"AgentToolCallGroup.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentToolCallGroup.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAKnD;;;;GAIG;AACH,MAAM,OAAO,kBAAkB;IACpB,EAAE,GAAG,MAAM,EAAE,CAAC;IACd,SAAS,CAAkB;IAEpC,YAAY,SAA0C,EAAE,EAA6B;QACnF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IAED,UAAU,CAAC,WAA+D;QACxE,OAAO,IAAI,kBAAkB,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,aAAa,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9F,CAAC;IAED,SAAS,CAAC,UAA6D;QACrE,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAC/C,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,UAAU,CAAC,OAAO,CAAC,MAAM,CACxD,CAAC;QAEF,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,YAAY,CAAE,CAAC;QAElD,OAAO,IAAI,kBAAkB,CAC3B;YACE,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC;YACxC,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC;YAC/B,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC;SAC1C,EACD,IAAI,CAAC,EAAE,CACR,CAAC;IACJ,CAAC;IAED,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;SAC/D,CAAC;IACJ,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 { nanoid } from \"nanoid/non-secure\";\nimport { AgentToolCall } from \"./AgentToolCall.ts\";\nimport type { AgentChatResponse } from \"./AgentChatResponse.ts\";\nimport type { AgentToolRequestResponseContent } from \"../responseContent/AgentToolRequestResponseContent.ts\";\nimport type { AgentToolResultResponseContent } from \"../responseContent/AgentToolResultResponseContent.ts\";\n\n/\n Contains a collection of `AgentToolCall` associated with an `AgentChatExchange`.\n * As tool calls stream in, this class is updated immutably but maintains\n * a stable `.id` property.\n */\nexport class AgentToolCallGroup {\n readonly id = nanoid();\n readonly toolCalls: AgentToolCall[];\n\n constructor(toolCalls: AgentToolCallGroup[\"toolCalls\"], id?: AgentToolCallGroup[\"id\"]) {\n this.toolCalls = toolCalls;\n if (id) {\n this.id = id;\n }\n }\n\n addRequest(toolRequest: AgentChatResponse<AgentToolRequestResponseContent>) {\n return new AgentToolCallGroup([...this.toolCalls, new AgentToolCall(toolRequest)], this.id);\n }\n\n addResult(toolResult: AgentChatResponse<AgentToolResultResponseContent>) {\n const requestIndex = this.toolCalls.findLastIndex(\n (toolCall) => toolCall.id === toolResult.content.callId,\n );\n\n if (requestIndex === -1) {\n return this;\n }\n\n const toolCall = this.toolCalls.at(requestIndex)!;\n\n return new AgentToolCallGroup(\n [\n ...this.toolCalls.slice(0, requestIndex),\n toolCall.withResult(toolResult),\n ...this.toolCalls.slice(requestIndex + 1),\n ],\n this.id,\n );\n }\n\n toJSON() {\n return {\n id: this.id,\n toolCalls: this.toolCalls.map((toolCall) => toolCall.toJSON()),\n };\n }\n}\n"]}

package/dist/enterprise/ai/chat/index.d.ts CHANGED Viewed

@@ -1,11 +1,11 @@
 import type { ChatSessionClient } from "./ChatSessionClient.ts";
 export type { ChatSessionClient };
-export * from "./AgentErrorResponseContent.ts";
-export * from "./AgentModelResponseContent.ts";
-export * from "./AgentChatPartialResponse.ts";
-export * from "./AgentChatResponse.ts";
-export * from "./AgentToolRequestResponseContent.ts";
-export * from "./AgentToolResultResponseContent.ts";
+export * from "./responseContent/AgentErrorResponseContent.ts";
+export * from "./responseContent/AgentModelResponseContent.ts";
+export * from "./eventWrappers/AgentChatPartialResponse.ts";
+export * from "./eventWrappers/AgentChatResponse.ts";
+export * from "./responseContent/AgentToolRequestResponseContent.ts";
+export * from "./responseContent/AgentToolResultResponseContent.ts";
 export * from "./ChatSession.ts";
 export * from "./UserChatMessage.ts";
 export * from "./UserChatMessageContent.ts";

package/dist/enterprise/ai/chat/index.js CHANGED Viewed

@@ -13,12 +13,12 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-export * from "./AgentErrorResponseContent.js";
-export * from "./AgentModelResponseContent.js";
-export * from "./AgentChatPartialResponse.js";
-export * from "./AgentChatResponse.js";
-export * from "./AgentToolRequestResponseContent.js";
-export * from "./AgentToolResultResponseContent.js";
+export * from "./responseContent/AgentErrorResponseContent.js";
+export * from "./responseContent/AgentModelResponseContent.js";
+export * from "./eventWrappers/AgentChatPartialResponse.js";
+export * from "./eventWrappers/AgentChatResponse.js";
+export * from "./responseContent/AgentToolRequestResponseContent.js";
+export * from "./responseContent/AgentToolResultResponseContent.js";
 export * from "./ChatSession.js";
 export * from "./UserChatMessage.js";
 export * from "./UserChatMessageContent.js";

package/dist/enterprise/ai/chat/index.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,cAAc,~~gCAAgC~~,CAAC;AAC/C,cAAc,~~gCAAgC~~,CAAC;AAC/C,cAAc~~,+BAA+B~~,CAAC;~~AAC9C~~,cAAc,~~wBAAwB~~,CAAC;~~AACvC~~,cAAc,~~sCAAsC~~,CAAC;~~AACrD~~,cAAc,~~qCAAqC~~,CAAC;~~AACpD~~,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC;AACrC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,sCAAsC,CAAC;AACrD,cAAc,iCAAiC,CAAC;AAChD,cAAc,kCAAkC,CAAC;AACjD,cAAc,uCAAuC,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 type { ChatSessionClient } from \"./ChatSessionClient.ts\";\nexport type { ChatSessionClient };\n\nexport from \"./AgentErrorResponseContent.ts\";\nexport * from \"./AgentModelResponseContent.ts\";\nexport * from \"./AgentChatPartialResponse.ts\";\nexport * from \"./AgentChatResponse.ts\";\nexport * from \"./AgentToolRequestResponseContent.ts\";\nexport * from \"./AgentToolResultResponseContent.ts\";\nexport * from \"./ChatSession.ts\";\nexport * from \"./UserChatMessage.ts\";\nexport * from \"./UserChatMessageContent.ts\";\nexport * from \"./eventWrappers/AgentChatExchange.ts\";\nexport * from \"./eventWrappers/AgentEndTurn.ts\";\nexport * from \"./eventWrappers/AgentToolCall.ts\";\nexport * from \"./eventWrappers/AgentToolCallGroup.ts\";\n"]}
1	+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,cAAc,gDAAgD,CAAC;AAC/D,cAAc,gDAAgD,CAAC;AAC/D,cAAc,6CAA6C,CAAC;AAC5D,cAAc,sCAAsC,CAAC;AACrD,cAAc,sDAAsD,CAAC;AACrE,cAAc,qDAAqD,CAAC;AACpE,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC;AACrC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,sCAAsC,CAAC;AACrD,cAAc,iCAAiC,CAAC;AAChD,cAAc,kCAAkC,CAAC;AACjD,cAAc,uCAAuC,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 type { ChatSessionClient } from \"./ChatSessionClient.ts\";\nexport type { ChatSessionClient };\n\nexport from \"./responseContent/AgentErrorResponseContent.ts\";\nexport * from \"./responseContent/AgentModelResponseContent.ts\";\nexport * from \"./eventWrappers/AgentChatPartialResponse.ts\";\nexport * from \"./eventWrappers/AgentChatResponse.ts\";\nexport * from \"./responseContent/AgentToolRequestResponseContent.ts\";\nexport * from \"./responseContent/AgentToolResultResponseContent.ts\";\nexport * from \"./ChatSession.ts\";\nexport * from \"./UserChatMessage.ts\";\nexport * from \"./UserChatMessageContent.ts\";\nexport * from \"./eventWrappers/AgentChatExchange.ts\";\nexport * from \"./eventWrappers/AgentEndTurn.ts\";\nexport * from \"./eventWrappers/AgentToolCall.ts\";\nexport * from \"./eventWrappers/AgentToolCallGroup.ts\";\n"]}

package/dist/enterprise/ai/chat/methods/sendChatMessage.d.ts CHANGED Viewed

@@ -4,7 +4,7 @@ import { Problem } from "../../../../common/Problem.ts";
 import type { ChatSession } from "../ChatSession.ts";
 import type { UserChatMessage } from "../UserChatMessage.ts";
 import { Err, Ok } from "ts-results-es";
-import { AgentChatResponse } from "../AgentChatResponse.ts";
+import { AgentChatResponse } from "../eventWrappers/AgentChatResponse.ts";
 type SendChatMessageProblemCodes = "https://api.dremio.dev/problems/ai-agent/chat/agent-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/input-too-large" | "https://api.dremio.dev/problems/ai-agent/chat/model-provider-not-found" | "https://api.dremio.dev/problems/ai-agent/chat/no-default-model-provider" | "https://api.dremio.dev/problems/ai-agent/chat/no-model-provider-privilege" | "https://api.dremio.dev/problems/ai-agent/chat/preview-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/rate-limit-exceeded";
-export declare const createSendChatMessage: (config: SonarV4Config) => (chatSession: ChatSession, userChatMessage: UserChatMessage) => import("rxjs").Observable<Ok<AgentChatResponse<import("../AgentChatResponse.ts").AgentResponseContent>> | Err<HttpError | Problem<SendChatMessageProblemCodes, undefined>>>;
+export declare const createSendChatMessage: (config: SonarV4Config) => (chatSession: ChatSession, userChatMessage: UserChatMessage) => import("rxjs").Observable<Ok<AgentChatResponse<import("../eventWrappers/AgentChatResponse.ts").AgentResponseContent>> | Err<HttpError | Problem<SendChatMessageProblemCodes, undefined>>>;
 export {};

package/dist/enterprise/ai/chat/methods/sendChatMessage.js CHANGED Viewed

@@ -17,7 +17,7 @@ import { catchError, map, of, share, tap } from "rxjs";
 import { fromTextEventStream } from "../../../../common/fromTextEventStream.js";
 import { Problem } from "../../../../common/Problem.js";
 import { Err, Ok } from "ts-results-es";
-import { AgentChatResponse } from "../AgentChatResponse.js";
+import { AgentChatResponse } from "../eventWrappers/AgentChatResponse.js";
 import * as z from "zod/mini";
 function refineChatErrors(e) {
     if (e.body instanceof Problem) {
@@ -30,7 +30,7 @@ export const createSendChatMessage = (config) => function sendChatMessage(chatSe
         .sonarV4Request("agent", {
         body: JSON.stringify(z.parse(sendChatMessageBodySchema, {
             message: userChatMessage.toString(),
-            sessionId: chatSession.sessionId,
+            sessionId: chatSession.id,
         })),
         headers: {
             Accept: "text/event-stream",
@@ -46,7 +46,7 @@ export const createSendChatMessage = (config) => function sendChatMessage(chatSe
         return result.value;
     })).pipe(map((event) => Ok(AgentChatResponse.fromJSON(JSON.parse(event.data)))), tap((agentChatResponse) => {
         if (agentChatResponse.isOk()) {
-            chatSession.assign(agentChatResponse.value.session);
+            chatSession.assign(agentChatResponse.value.session.toJSON());
         }
     }), catchError((e) => of(Err(refineChatErrors(e)))), share());
 };

package/dist/enterprise/ai/chat/methods/sendChatMessage.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"sendChatMessage.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/methods/sendChatMessage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAEvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,2CAA2C,CAAC;AAEhF,OAAO,EAAE,OAAO,EAAE,MAAM,+BAA+B,CAAC;AAGxD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,iBAAiB,EAAE,MAAM,~~yBAAyB~~,CAAC;~~AAC5D~~,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAW9B,SAAS,gBAAgB,CAAC,CAAY;IACpC,IAAI,CAAC,CAAC,IAAI,YAAY,OAAO,EAAE,CAAC;QAC9B,OAAO,CAAC,CAAC,IAA4C,CAAC;IACxD,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,MAAqB,EAAE,EAAE,CAC7D,SAAS,eAAe,CAAC,WAAwB,EAAE,eAAgC;IACjF,OAAO,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACxC,MAAM;SACH,cAAc,CAAC,OAAO,EAAE;QACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB,CAAC,CAAC,KAAK,CAAC,yBAAyB,EAAE;YACjC,OAAO,EAAE,eAAe,CAAC,QAAQ,EAAE;YACnC,SAAS,EAAE,WAAW,CAAC,~~SAAS~~;~~SACjC~~,CAAC,CACH;QACD,OAAO,EAAE;YACP,MAAM,EAAE,mBAAmB;YAC3B,cAAc,EAAE,kBAAkB;SACnC;QACD,MAAM,EAAE,MAAM;QACd,MAAM;KACP,CAAC;SACD,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;QACvB,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;YACnB,MAAM,MAAM,CAAC,KAAK,CAAC;QACrB,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC;IACtB,CAAC,CAAC,CACL,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EACtE,GAAG,CAAC,CAAC,iBAAiB,EAAE,EAAE;QACxB,IAAI,iBAAiB,CAAC,IAAI,EAAE,EAAE,CAAC;YAC7B,WAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;~~QACtD~~,CAAC;IACH,CAAC,CAAC,EACF,UAAU,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAC1D,KAAK,EAAE,CACR,CAAC;AACJ,CAAC,CAAC;AAEJ,MAAM,yBAAyB,GAAG,CAAC,CAAC,YAAY,CAAC;IAC/C,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACnD,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACjE,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACvE,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;CAClE,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 { catchError, map, of, share, tap } from \"rxjs\";\nimport type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport { fromTextEventStream } from \"../../../../common/fromTextEventStream.ts\";\nimport type { HttpError } from \"../../../../common/HttpError.ts\";\nimport { Problem } from \"../../../../common/Problem.ts\";\nimport type { ChatSession } from \"../ChatSession.ts\";\nimport type { UserChatMessage } from \"../UserChatMessage.ts\";\nimport { Err, Ok } from \"ts-results-es\";\nimport { AgentChatResponse } from \"../AgentChatResponse.ts\";\nimport as z from \"zod/mini\";\n\ntype SendChatMessageProblemCodes =\n \| \"https://api.dremio.dev/problems/ai-agent/chat/agent-disabled\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/input-too-large\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/model-provider-not-found\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/no-default-model-provider\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/no-model-provider-privilege\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/preview-disabled\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/rate-limit-exceeded\";\n\nfunction refineChatErrors(e: HttpError) {\n if (e.body instanceof Problem) {\n return e.body as Problem<SendChatMessageProblemCodes>;\n }\n return e;\n}\n\nexport const createSendChatMessage = (config: SonarV4Config) =>\n function sendChatMessage(chatSession: ChatSession, userChatMessage: UserChatMessage) {\n return fromTextEventStream(({ signal }) =>\n config\n .sonarV4Request(\"agent\", {\n body: JSON.stringify(\n z.parse(sendChatMessageBodySchema, {\n message: userChatMessage.toString(),\n sessionId: chatSession.~~sessionId~~,\n }),\n ),\n headers: {\n Accept: \"text/event-stream\",\n \"Content-Type\": \"application/json\",\n },\n method: \"POST\",\n signal,\n })\n .promise.then((result) => {\n if (result.isErr()) {\n throw result.error;\n }\n return result.value;\n }),\n ).pipe(\n map((event) => Ok(AgentChatResponse.fromJSON(JSON.parse(event.data)))),\n tap((agentChatResponse) => {\n if (agentChatResponse.isOk()) {\n chatSession.assign(agentChatResponse.value.session);\n }\n }),\n catchError((e: HttpError) => of(Err(refineChatErrors(e)))),\n share(),\n );\n };\n\nconst sendChatMessageBodySchema = z.strictObject({\n message: z.string().check(z.trim(), z.minLength(1)),\n modelName: z.optional(z.string().check(z.trim(), z.minLength(1))),\n modelProviderId: z.optional(z.string().check(z.trim(), z.minLength(1))),\n sessionId: z.optional(z.string().check(z.trim(), z.minLength(1))),\n});\n"]}
1	+ {"version":3,"file":"sendChatMessage.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/methods/sendChatMessage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAEvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,2CAA2C,CAAC;AAEhF,OAAO,EAAE,OAAO,EAAE,MAAM,+BAA+B,CAAC;AAGxD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAW9B,SAAS,gBAAgB,CAAC,CAAY;IACpC,IAAI,CAAC,CAAC,IAAI,YAAY,OAAO,EAAE,CAAC;QAC9B,OAAO,CAAC,CAAC,IAA4C,CAAC;IACxD,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,MAAqB,EAAE,EAAE,CAC7D,SAAS,eAAe,CAAC,WAAwB,EAAE,eAAgC;IACjF,OAAO,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACxC,MAAM;SACH,cAAc,CAAC,OAAO,EAAE;QACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB,CAAC,CAAC,KAAK,CAAC,yBAAyB,EAAE;YACjC,OAAO,EAAE,eAAe,CAAC,QAAQ,EAAE;YACnC,SAAS,EAAE,WAAW,CAAC,EAAE;SAC1B,CAAC,CACH;QACD,OAAO,EAAE;YACP,MAAM,EAAE,mBAAmB;YAC3B,cAAc,EAAE,kBAAkB;SACnC;QACD,MAAM,EAAE,MAAM;QACd,MAAM;KACP,CAAC;SACD,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;QACvB,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;YACnB,MAAM,MAAM,CAAC,KAAK,CAAC;QACrB,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC;IACtB,CAAC,CAAC,CACL,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EACtE,GAAG,CAAC,CAAC,iBAAiB,EAAE,EAAE;QACxB,IAAI,iBAAiB,CAAC,IAAI,EAAE,EAAE,CAAC;YAC7B,WAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC,CAAC,EACF,UAAU,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAC1D,KAAK,EAAE,CACR,CAAC;AACJ,CAAC,CAAC;AAEJ,MAAM,yBAAyB,GAAG,CAAC,CAAC,YAAY,CAAC;IAC/C,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACnD,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACjE,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACvE,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;CAClE,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 { catchError, map, of, share, tap } from \"rxjs\";\nimport type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport { fromTextEventStream } from \"../../../../common/fromTextEventStream.ts\";\nimport type { HttpError } from \"../../../../common/HttpError.ts\";\nimport { Problem } from \"../../../../common/Problem.ts\";\nimport type { ChatSession } from \"../ChatSession.ts\";\nimport type { UserChatMessage } from \"../UserChatMessage.ts\";\nimport { Err, Ok } from \"ts-results-es\";\nimport { AgentChatResponse } from \"../eventWrappers/AgentChatResponse.ts\";\nimport as z from \"zod/mini\";\n\ntype SendChatMessageProblemCodes =\n \| \"https://api.dremio.dev/problems/ai-agent/chat/agent-disabled\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/input-too-large\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/model-provider-not-found\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/no-default-model-provider\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/no-model-provider-privilege\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/preview-disabled\"\n \| \"https://api.dremio.dev/problems/ai-agent/chat/rate-limit-exceeded\";\n\nfunction refineChatErrors(e: HttpError) {\n if (e.body instanceof Problem) {\n return e.body as Problem<SendChatMessageProblemCodes>;\n }\n return e;\n}\n\nexport const createSendChatMessage = (config: SonarV4Config) =>\n function sendChatMessage(chatSession: ChatSession, userChatMessage: UserChatMessage) {\n return fromTextEventStream(({ signal }) =>\n config\n .sonarV4Request(\"agent\", {\n body: JSON.stringify(\n z.parse(sendChatMessageBodySchema, {\n message: userChatMessage.toString(),\n sessionId: chatSession.id,\n }),\n ),\n headers: {\n Accept: \"text/event-stream\",\n \"Content-Type\": \"application/json\",\n },\n method: \"POST\",\n signal,\n })\n .promise.then((result) => {\n if (result.isErr()) {\n throw result.error;\n }\n return result.value;\n }),\n ).pipe(\n map((event) => Ok(AgentChatResponse.fromJSON(JSON.parse(event.data)))),\n tap((agentChatResponse) => {\n if (agentChatResponse.isOk()) {\n chatSession.assign(agentChatResponse.value.session.toJSON());\n }\n }),\n catchError((e: HttpError) => of(Err(refineChatErrors(e)))),\n share(),\n );\n };\n\nconst sendChatMessageBodySchema = z.strictObject({\n message: z.string().check(z.trim(), z.minLength(1)),\n modelName: z.optional(z.string().check(z.trim(), z.minLength(1))),\n modelProviderId: z.optional(z.string().check(z.trim(), z.minLength(1))),\n sessionId: z.optional(z.string().check(z.trim(), z.minLength(1))),\n});\n"]}

package/dist/enterprise/ai/chat/responseContent/AgentErrorResponseContent.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"AgentErrorResponseContent.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/responseContent/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/responseContent/AgentModelResponseContent.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"AgentModelResponseContent.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/responseContent/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/responseContent/AgentToolRequestResponseContent.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"AgentToolRequestResponseContent.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/responseContent/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/responseContent/AgentToolResultResponseContent.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"AgentToolResultResponseContent.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/responseContent/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/index.d.ts CHANGED Viewed

@@ -1,3 +1,3 @@
-import { ModelProvider } from "./ModelProvider.ts";
+import { ModelProvider } from "./modelProvider/ModelProvider.ts";
 export * from "./chat/index.ts";
 export { ModelProvider };

package/dist/enterprise/ai/index.js CHANGED Viewed

@@ -13,7 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { ModelProvider } from "./ModelProvider.js";
+import { ModelProvider } from "./modelProvider/ModelProvider.js";
 export * from "./chat/index.js";
 export { ModelProvider };
 //# sourceMappingURL=index.js.map

package/dist/enterprise/ai/index.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/enterprise/ai/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,~~oBAAoB~~,CAAC;~~AACnD~~,cAAc,iBAAiB,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,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 { ModelProvider } from \"./ModelProvider.ts\";\nexport from \"./chat/index.ts\";\nexport { ModelProvider };\n"]}
1	+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/enterprise/ai/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,cAAc,iBAAiB,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,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 { ModelProvider } from \"./modelProvider/ModelProvider.ts\";\nexport from \"./chat/index.ts\";\nexport { ModelProvider };\n"]}

package/dist/enterprise/ai/{ModelProvider.d.ts → modelProvider/ModelProvider.d.ts} RENAMED Viewed

@@ -1,7 +1,7 @@
-import type { V4Config } from "../../common/Config.ts";
+import type { V4Config } from "../../../common/Config.ts";
 import * as z from "zod/mini";
 import { modelProviderRetrieveCodec, modelProviderUpdateCodec } from "./modelProviderCodec.ts";
-import { VersionConflictError } from "../../common/problems.ts";
+import { VersionConflictError } from "../../../common/problems.ts";
 export declare class ModelProvider implements Omit<ModelProviderProperties, "tag"> {
     #private;
     readonly config: ModelProviderProperties["config"];
@@ -15,9 +15,9 @@ export declare class ModelProvider implements Omit<ModelProviderProperties, "tag
     readonly name: ModelProviderProperties["name"];
     readonly sqlModelNames: ModelProviderProperties["sqlModelNames"];
     constructor(config: V4Config, properties: z.output<typeof modelProviderRetrieveCodec>);
-    delete(): import("ts-results-es").AsyncResult<void, import("../index.ts").HttpError>;
-    setDefault(): import("ts-results-es").AsyncResult<ModelProvider, VersionConflictError | import("../index.ts").HttpError>;
-    update(fields: Partial<z.output<typeof modelProviderUpdateCodec>>): import("ts-results-es").AsyncResult<ModelProvider, VersionConflictError | import("../index.ts").HttpError>;
+    delete(): import("ts-results-es").AsyncResult<void, import("../../index.ts").HttpError>;
+    setDefault(): import("ts-results-es").AsyncResult<ModelProvider, VersionConflictError | import("../../index.ts").HttpError>;
+    update(fields: Partial<z.output<typeof modelProviderUpdateCodec>>): import("ts-results-es").AsyncResult<ModelProvider, VersionConflictError | import("../../index.ts").HttpError>;
     static createSchema: z.ZodMiniObject<{
         name: z.ZodMiniString<string>;
         config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{

package/dist/enterprise/ai/{ModelProvider.js → modelProvider/ModelProvider.js} RENAMED Viewed

@@ -15,7 +15,7 @@
  */
 import * as z from "zod/mini";
 import { modelProviderCreateCodec, modelProviderRetrieveCodec, modelProviderUpdateCodec, } from "./modelProviderCodec.js";
-import { VersionConflictError } from "../../common/problems.js";
+import { VersionConflictError } from "../../../common/problems.js";
 export class ModelProvider {
     config;
     createdAt;

package/dist/enterprise/ai/modelProvider/ModelProvider.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"ModelProvider.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/ModelProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,EAC1B,wBAAwB,GACzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAEnE,MAAM,OAAO,aAAa;IACf,MAAM,CAAoC;IAC1C,SAAS,CAAuC;IAChD,SAAS,CAAuC;IAChD,gBAAgB,CAA8C;IAC9D,EAAE,CAAgC;IAClC,SAAS,CAAuC;IAChD,UAAU,CAAwC;IAClD,UAAU,CAAwC;IAClD,IAAI,CAAkC;IACtC,aAAa,CAA2C;IAExD,OAAO,CAAW;IAClB,IAAI,CAAS;IAEtB,YAAY,MAAgB,EAAE,UAAuD;QACnF,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,gBAAgB,CAAC;QACpD,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;QACxC,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,aAAa,GAAG,UAAU,CAAC,aAAa,CAAC;QAE9C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC;IAC7B,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,IAAI,CAAC,EAAE,EAAE,EAAE;YAC7C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,QAAQ;SACjB,CAAC;aACD,GAAG,CAAC,GAAG,EAAE,CAAC,SAAiB,CAAC,CAAC;IAClC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,MAA0D;QAC/D,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE;YAC9C,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM;YACpC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB;YAClE,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS;YAC7C,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI;YAC9B,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa;SAC1D,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,IAAI,CAAC,EAAE,EAAE,EAAE;YAC7C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACjD,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SACd,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAC1F;aACA,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;YACd,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO,IAAI,oBAAoB,EAAE,CAAC;YACpC,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,YAAY,GAAG,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAC;IACvD,MAAM,CAAC,YAAY,GAAG,wBAAwB,CAAC,GAAG,CAAC,GAAG,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 type { V4Config } from \"../../../common/Config.ts\";\nimport * as z from \"zod/mini\";\nimport {\n modelProviderCreateCodec,\n modelProviderRetrieveCodec,\n modelProviderUpdateCodec,\n} from \"./modelProviderCodec.ts\";\nimport { VersionConflictError } from \"../../../common/problems.ts\";\n\nexport class ModelProvider implements Omit<ModelProviderProperties, \"tag\"> {\n readonly config: ModelProviderProperties[\"config\"];\n readonly createdAt: ModelProviderProperties[\"createdAt\"];\n readonly createdBy: ModelProviderProperties[\"createdBy\"];\n readonly defaultModelName: ModelProviderProperties[\"defaultModelName\"];\n readonly id: ModelProviderProperties[\"id\"];\n readonly isDefault: ModelProviderProperties[\"isDefault\"];\n readonly modifiedAt: ModelProviderProperties[\"modifiedAt\"];\n readonly modifiedBy: ModelProviderProperties[\"modifiedBy\"];\n readonly name: ModelProviderProperties[\"name\"];\n readonly sqlModelNames: ModelProviderProperties[\"sqlModelNames\"];\n\n readonly #config: V4Config;\n readonly #tag: string;\n\n constructor(config: V4Config, properties: z.output<typeof modelProviderRetrieveCodec>) {\n this.config = properties.config;\n this.createdAt = properties.createdAt;\n this.createdBy = properties.createdBy;\n this.defaultModelName = properties.defaultModelName;\n this.id = properties.id;\n this.isDefault = properties.isDefault;\n this.modifiedAt = properties.modifiedAt;\n this.modifiedBy = properties.modifiedBy;\n this.name = properties.name;\n this.sqlModelNames = properties.sqlModelNames;\n\n this.#config = config;\n this.#tag = properties.tag;\n }\n\n delete() {\n return this.#config\n .v4Request(`model-provider/config/${this.id}`, {\n keepalive: true,\n method: \"DELETE\",\n })\n .map(() => undefined as void);\n }\n\n setDefault() {\n return this.update({ isDefault: true });\n }\n\n update(fields: Partial<z.output<typeof modelProviderUpdateCodec>>) {\n const body = z.encode(modelProviderUpdateCodec, {\n config: fields.config ?? this.config,\n defaultModelName: fields.defaultModelName ?? this.defaultModelName,\n isDefault: fields.isDefault ?? this.isDefault,\n name: fields.name ?? this.name,\n sqlModelNames: fields.sqlModelNames ?? this.sqlModelNames,\n });\n\n return this.#config\n .v4Request(`model-provider/config/${this.id}`, {\n body: JSON.stringify({ ...body, tag: this.#tag }),\n headers: {\n Accept: \"application/json\",\n \"Content-Type\": \"application/json\",\n },\n keepalive: true,\n method: \"PUT\",\n })\n .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n .map(\n (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n )\n .mapErr((val) => {\n if (val.status === 409) {\n return new VersionConflictError();\n }\n return val;\n });\n }\n\n static createSchema = modelProviderCreateCodec.def.out;\n static updateSchema = modelProviderUpdateCodec.def.out;\n}\n\ntype ModelProviderProperties = z.output<typeof modelProviderRetrieveCodec>;\n"]}

package/dist/enterprise/ai/modelProvider/modelProviderCodec.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"modelProviderCodec.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/modelProviderCodec.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,8CAA8C,EAC9C,sCAAsC,EACtC,4CAA4C,EAC5C,0CAA0C,EAC1C,kCAAkC,EAClC,4CAA4C,EAC5C,oCAAoC,EACpC,0CAA0C,EAC1C,6CAA6C,EAC7C,qCAAqC,EACrC,uCAAuC,EACvC,+BAA+B,GAChC,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE;IAC5B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE;IACtB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC9C,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;IACtB,mEAAmE;IACnE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACnD,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;CAChB,CAAC,CAAC;AAEH,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE;IACtE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC;QACd,CAAC,CAAC,YAAY,CAAC;YACb,aAAa,EAAE,8CAA8C;SAC9D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,0CAA0C;SACtD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,WAAW,EAAE,4CAA4C;SAC1D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,YAAY,EAAE,6CAA6C;SAC5D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,MAAM,EAAE,uCAAuC;SAChD,CAAC;KACH,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAC1C,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE;IAC9B,gBAAgB,EAAE,IAAI;IACtB,SAAS,EAAE,IAAI;IACf,IAAI,EAAE,IAAI;IACV,aAAa,EAAE,IAAI;CACpB,CAAC,EACF;IACE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC;QACd,CAAC,CAAC,YAAY,CAAC;YACb,aAAa,EAAE,sCAAsC;SACtD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,kCAAkC;SAC9C,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,WAAW,EAAE,oCAAoC;SAClD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,YAAY,EAAE,qCAAqC;SACpD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,MAAM,EAAE,+BAA+B;SACxC,CAAC;KACH,CAAC;CACH,CACF,CAAC;AAEF,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAC1C,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EACrD;IACE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC;QACd,CAAC,CAAC,YAAY,CAAC;YACb,aAAa,EAAE,4CAA4C;SAC5D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,kCAAkC,CAAC;SACzD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,WAAW,EAAE,0CAA0C;SACxD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,YAAY,EAAE,CAAC,CAAC,OAAO,CAAC,qCAAqC,CAAC;SAC/D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,+BAA+B,CAAC;SACnD,CAAC;KACH,CAAC;CACH,CACF,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAC/C,6BAA6B,EAC7B,CAAC,CAAC,MAAM,CACN,CAAC,CAAC,IAAI,CAAC,6BAA6B,EAAE;IACpC,SAAS,EAAE,IAAI;IACf,UAAU,EAAE,IAAI;IAChB,aAAa,EAAE,IAAI;CACpB,CAAC,EACF;IACE,SAAS,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC;IACzC,UAAU,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC;IAC1C,aAAa,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACjC,CACF,EACD;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAC7C,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC;YAC/C,aAAa,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC;SACxC,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;YACjC,UAAU,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,EAAE;YACnC,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF,CACF,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAC7C,2BAA2B,EAC3B,CAAC,CAAC,MAAM,CACN,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE;IAClC,aAAa,EAAE,IAAI;CACpB,CAAC,EACF;IACE,aAAa,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACjC,CACF,EACD;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC;SACxC,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF,CACF,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAC7C,2BAA2B,EAC3B,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE;IACrE,aAAa,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACjC,CAAC,EACF;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC;SACxC,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF,CACF,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\nimport {\n amazonBedrockModelProviderConfigRetrieveSchema,\n amazonBedrockModelProviderConfigSchema,\n amazonBedrockModelProviderConfigUpdateSchema,\n anthropicModelProviderConfigRetrieveSchema,\n anthropicModelProviderConfigSchema,\n azureOpenAiModelProviderConfigRetrieveSchema,\n azureOpenAiModelProviderConfigSchema,\n azureOpenAiModelProviderConfigUpdateSchema,\n googleGeminiModelProviderConfigRetrieveSchema,\n googleGeminiModelProviderConfigSchema,\n openAiModelProviderConfigRetrieveSchema,\n openAiModelProviderConfigSchema,\n} from \"./modelProviderConfigSchemas.ts\";\nimport { Temporal } from \"temporal-polyfill\";\n\nconst modelProviderBaseSchema = z.object({\n createdAt: z.string().check(z.iso.datetime()),\n createdBy: z.string(),\n defaultModelName: z.string(),\n id: z.string(),\n isDefault: z.boolean(),\n modifiedAt: z.string().check(z.iso.datetime()),\n modifiedBy: z.string(),\n // eslint-disable-next-line regexp/prefer-w, regexp/use-ignore-case\n name: z.string().check(z.regex(/^[a-zA-Z0-9_-]+$/)),\n sqlModelNames: z.array(z.string()),\n tag: z.string(),\n});\n\nconst modelProviderRetrieveInSchema = z.extend(modelProviderBaseSchema, {\n config: z.union([\n z.strictObject({\n amazonBedrock: amazonBedrockModelProviderConfigRetrieveSchema,\n }),\n z.strictObject({\n anthropic: anthropicModelProviderConfigRetrieveSchema,\n }),\n z.strictObject({\n azureOpenAi: azureOpenAiModelProviderConfigRetrieveSchema,\n }),\n z.strictObject({\n googleGemini: googleGeminiModelProviderConfigRetrieveSchema,\n }),\n z.strictObject({\n openAi: openAiModelProviderConfigRetrieveSchema,\n }),\n ]),\n});\n\nconst modelProviderCreateInSchema = z.extend(\n z.pick(modelProviderBaseSchema, {\n defaultModelName: true,\n isDefault: true,\n name: true,\n sqlModelNames: true,\n }),\n {\n config: z.union([\n z.strictObject({\n amazonBedrock: amazonBedrockModelProviderConfigSchema,\n }),\n z.strictObject({\n anthropic: anthropicModelProviderConfigSchema,\n }),\n z.strictObject({\n azureOpenAi: azureOpenAiModelProviderConfigSchema,\n }),\n z.strictObject({\n googleGemini: googleGeminiModelProviderConfigSchema,\n }),\n z.strictObject({\n openAi: openAiModelProviderConfigSchema,\n }),\n ]),\n },\n);\n\nconst modelProviderUpdateInSchema = z.extend(\n z.omit(modelProviderCreateInSchema, { config: true }),\n {\n config: z.union([\n z.strictObject({\n amazonBedrock: amazonBedrockModelProviderConfigUpdateSchema,\n }),\n z.strictObject({\n anthropic: z.partial(anthropicModelProviderConfigSchema),\n }),\n z.strictObject({\n azureOpenAi: azureOpenAiModelProviderConfigUpdateSchema,\n }),\n z.strictObject({\n googleGemini: z.partial(googleGeminiModelProviderConfigSchema),\n }),\n z.strictObject({\n openAi: z.partial(openAiModelProviderConfigSchema),\n }),\n ]),\n },\n);\n\nexport const modelProviderRetrieveCodec = z.codec(\n modelProviderRetrieveInSchema,\n z.extend(\n z.omit(modelProviderRetrieveInSchema, {\n createdAt: true,\n modifiedAt: true,\n sqlModelNames: true,\n }),\n {\n createdAt: z.instanceof(Temporal.Instant),\n modifiedAt: z.instanceof(Temporal.Instant),\n sqlModelNames: z.set(z.string()),\n },\n ),\n {\n decode(v) {\n return {\n ...v,\n createdAt: Temporal.Instant.from(v.createdAt),\n modifiedAt: Temporal.Instant.from(v.modifiedAt),\n sqlModelNames: new Set(v.sqlModelNames),\n };\n },\n encode(v) {\n return {\n ...v,\n createdAt: v.createdAt.toString(),\n modifiedAt: v.modifiedAt.toString(),\n sqlModelNames: Array.from(v.sqlModelNames),\n };\n },\n },\n);\n\nexport const modelProviderCreateCodec = z.codec(\n modelProviderCreateInSchema,\n z.extend(\n z.omit(modelProviderCreateInSchema, {\n sqlModelNames: true,\n }),\n {\n sqlModelNames: z.set(z.string()),\n },\n ),\n {\n decode(v) {\n return {\n ...v,\n sqlModelNames: new Set(v.sqlModelNames),\n };\n },\n encode(v) {\n return {\n ...v,\n sqlModelNames: Array.from(v.sqlModelNames),\n };\n },\n },\n);\n\nexport const modelProviderUpdateCodec = z.codec(\n modelProviderUpdateInSchema,\n z.extend(z.omit(modelProviderUpdateInSchema, { sqlModelNames: true }), {\n sqlModelNames: z.set(z.string()),\n }),\n {\n decode(v) {\n return {\n ...v,\n sqlModelNames: new Set(v.sqlModelNames),\n };\n },\n encode(v) {\n return {\n ...v,\n sqlModelNames: Array.from(v.sqlModelNames),\n };\n },\n },\n);\n"]}

package/dist/enterprise/ai/modelProvider/modelProviderConfigSchemas.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"modelProviderConfigSchemas.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/modelProviderConfigSchemas.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;AAEpD,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC;QAClB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;QACvB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;KAC5B,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACb,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7D,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,0BAA0B,EAAE,qBAAqB,CAAC,CAAC;IAClE,MAAM,EAAE,yBAAyB;CAClC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,8CAA8C,GAAG,CAAC,CAAC,MAAM,CAAC;IACrE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC;QACZ,CAAC,CAAC,MAAM,CAAC;YACP,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;SACzF,CAAC;QACF,qBAAqB;KACtB,CAAC;IACF,MAAM,EAAE,yBAAyB;CAClC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,4CAA4C,GAAG,CAAC,CAAC,MAAM,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,CAAC,CAAC,MAAM,CAAC;YACP,SAAS,EAAE,CAAC,CAAC,MAAM,CACjB,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,EAC7E;gBACE,eAAe,EAAE,CAAC,CAAC,QAAQ,CACzB,0BAA0B,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CACjE;aACF,CACF;SACF,CAAC;QACF,qBAAqB;KACtB,CAAC,CACH;IACD,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,yBAAyB,CAAC;CAC9C,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,kCAAkC,GAAG,CAAC,CAAC,MAAM,CAAC;IACzD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,gBAAgB,EAAE,CAAC,CAAC,MAAM,CAAC;QACzB,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;QACzB,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;QAC7B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;KAC1B,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oCAAoC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3D,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC;IACrC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,4CAA4C,GAAG,CAAC,CAAC,MAAM,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACb,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;KAC9F,CAAC;IACF,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0CAA0C,GAAG,CAAC,CAAC,MAAM,CAAC;IACjE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,oCAAoC,CAAC,KAAK,CAAC,IAAI;QAC/C,CAAC,CAAC,MAAM,CAAC;YACP,gBAAgB,EAAE,CAAC,CAAC,MAAM,CACxB,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,EAC3E;gBACE,iBAAiB,EAAE,CAAC,CAAC,QAAQ,CAC3B,eAAe,CAAC,KAAK,CAAC,gBAAgB,CAAC,KAAK,CAAC,iBAAiB,CAC/D;aACF,CACF;SACF,CAAC;KACH,CAAC,CACH;IACD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0CAA0C,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAEvE;;GAEG;AACH,MAAM,CAAC,MAAM,qCAAqC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,6CAA6C,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAE1E;;GAEG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;CACnD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,uCAAuC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9D,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC;IACrC,CAAC,CAAC,YAAY,CAAC;QACb,aAAa,EAAE,sCAAsC;KACtD,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,SAAS,EAAE,kCAAkC;KAC9C,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,WAAW,EAAE,oCAAoC;KAClD,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,YAAY,EAAE,qCAAqC;KACpD,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,MAAM,EAAE,+BAA+B;KACxC,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAAC;IAChD,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,sBAAsB;KAC/B,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;KAC9B,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 * Amazon Bedrock\n */\nexport const amazonBedrockRegionSchema = z.string();\n\nconst amazonBedrockAccessKeyAuth = z.object({\n accessKey: z.object({\n accessKeyId: z.string(),\n secretAccessKey: z.string(),\n }),\n});\n\nconst amazonBedrockRoleAuth = z.object({\n role: z.object({\n roleArn: z.string(),\n }),\n});\n\nexport const amazonBedrockModelProviderConfigSchema = z.object({\n auth: z.union([amazonBedrockAccessKeyAuth, amazonBedrockRoleAuth]),\n region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigRetrieveSchema = z.object({\n auth: z.union([\n z.object({\n accessKey: z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }),\n }),\n amazonBedrockRoleAuth,\n ]),\n region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigUpdateSchema = z.object({\n auth: z.optional(\n z.union([\n z.object({\n accessKey: z.extend(\n z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }),\n {\n secretAccessKey: z.optional(\n amazonBedrockAccessKeyAuth.shape.accessKey.shape.secretAccessKey,\n ),\n },\n ),\n }),\n amazonBedrockRoleAuth,\n ]),\n ),\n region: z.optional(amazonBedrockRegionSchema),\n});\n\n/**\n * Anthropic\n */\nexport const anthropicModelProviderConfigSchema = z.object({\n accessKey: z.string(),\n});\n\n/**\n * Azure OpenAI\n */\nconst azureOpenAiAuth = z.object({\n clientSecretAuth: z.object({\n azureClientId: z.string(),\n azureClientSecret: z.string(),\n azureTenantId: z.string(),\n }),\n});\n\nexport const azureOpenAiModelProviderConfigSchema = z.object({\n auth: z.object(azureOpenAiAuth.shape),\n resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigRetrieveSchema = z.object({\n auth: z.object({\n clientSecretAuth: z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }),\n }),\n resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigUpdateSchema = z.object({\n auth: z.optional(\n z.union([\n azureOpenAiModelProviderConfigSchema.shape.auth,\n z.object({\n clientSecretAuth: z.extend(\n z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }),\n {\n azureClientSecret: z.optional(\n azureOpenAiAuth.shape.clientSecretAuth.shape.azureClientSecret,\n ),\n },\n ),\n }),\n ]),\n ),\n resourceName: z.string(),\n});\n\nexport const anthropicModelProviderConfigRetrieveSchema = z.object({});\n\n/**\n * Google Gemini\n */\nexport const googleGeminiModelProviderConfigSchema = z.object({\n accessKey: z.string(),\n});\n\nexport const googleGeminiModelProviderConfigRetrieveSchema = z.object({});\n\n/**\n * OpenAI\n */\nexport const openAiModelProviderConfigSchema = z.object({\n accessKey: z.string(),\n organizationId: z.string(),\n region: z.optional(z.enum([\"GLOBAL\", \"US\", \"EU\"])),\n});\n\nexport const openAiModelProviderConfigRetrieveSchema = z.object({\n organizationId: z.string(),\n region: z.enum([\"GLOBAL\", \"US\", \"EU\"]),\n});\n\nconst listModelsAccessSchema = z.union([\n z.strictObject({\n amazonBedrock: amazonBedrockModelProviderConfigSchema,\n }),\n z.strictObject({\n anthropic: anthropicModelProviderConfigSchema,\n }),\n z.strictObject({\n azureOpenAi: azureOpenAiModelProviderConfigSchema,\n }),\n z.strictObject({\n googleGemini: googleGeminiModelProviderConfigSchema,\n }),\n z.strictObject({\n openAi: openAiModelProviderConfigSchema,\n }),\n]);\n\nexport const listModelsPropertiesSchema = z.union([\n z.object({\n access: listModelsAccessSchema,\n }),\n z.object({\n modelProviderName: z.string(),\n }),\n]);\n"]}