@dremio/js-sdk 0.32.0 → 0.33.0

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

@@ -0,0 +1,21 @@
+import { AgentEndTurn } from "./AgentEndTurn.ts";
+import { AgentChatResponse, UserChatMessage, type AgentResponseContent } from "@dremio/js-sdk/enterprise";
+import { type Option } from "ts-results-es";
+import { AgentToolCallGroup } from "./AgentToolCallGroup.ts";
+import type { AgentToolCall } from "./AgentToolCall.ts";
+/**
+ * Represents a single chat exchange, starting with a `UserChatMessage` and followed by
+ * 0 or more agent chat events. As chat events stream in, this class is updated
+ * immutably but maintains a stable `.id` property.
+ */
+export declare class AgentChatExchange {
+    readonly agentChatEvents: AgentChatExchangeEvent[];
+    readonly agentToolCallGroup: AgentToolCallGroup;
+    readonly id: string;
+    readonly userChatMessage: UserChatMessage;
+    constructor(userChatMessage: AgentChatExchange["userChatMessage"], agentChatEvents: AgentChatExchange["agentChatEvents"], agentToolCallGroup?: AgentToolCallGroup, id?: string);
+    get isComplete(): boolean;
+    get completeReason(): Option<AgentEndTurn["reason"]>;
+    addChatEvent(agentChatEvent: AgentChatExchangeEvent): AgentChatExchange;
+}
+export type AgentChatExchangeEvent = AgentChatResponse<AgentResponseContent> | AgentEndTurn | AgentToolCall | AgentToolCallGroup;

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

@@ -0,0 +1,58 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { nanoid } from "nanoid/non-secure";
+import { AgentEndTurn } from "./AgentEndTurn.js";
+import { AgentChatResponse, UserChatMessage, } from "@dremio/js-sdk/enterprise";
+import { None, Some } from "ts-results-es";
+import { AgentToolCallGroup } from "./AgentToolCallGroup.js";
+import { isToolRequest, isToolResult } from "./AgentToolCall.js";
+/**
+ * Represents a single chat exchange, starting with a `UserChatMessage` and followed by
+ * 0 or more agent chat events. As chat events stream in, this class is updated
+ * immutably but maintains a stable `.id` property.
+ */
+export class AgentChatExchange {
+    agentChatEvents;
+    agentToolCallGroup;
+    id;
+    userChatMessage;
+    constructor(userChatMessage, agentChatEvents, agentToolCallGroup = new AgentToolCallGroup([]), id = nanoid()) {
+        this.agentChatEvents = agentChatEvents;
+        this.agentToolCallGroup = agentToolCallGroup;
+        this.id = id;
+        this.userChatMessage = userChatMessage;
+    }
+    get isComplete() {
+        return this.agentChatEvents.at(-1) instanceof AgentEndTurn;
+    }
+    get completeReason() {
+        const lastEvent = this.agentChatEvents.at(-1);
+        if (lastEvent instanceof AgentEndTurn) {
+            return Some(lastEvent.reason);
+        }
+        return None;
+    }
+    addChatEvent(agentChatEvent) {
+        if (isToolRequest(agentChatEvent)) {
+            return new AgentChatExchange(this.userChatMessage, this.agentChatEvents, this.agentToolCallGroup.addRequest(agentChatEvent), this.id);
+        }
+        if (isToolResult(agentChatEvent)) {
+            return new AgentChatExchange(this.userChatMessage, this.agentChatEvents, this.agentToolCallGroup.addResult(agentChatEvent), this.id);
+        }
+        return new AgentChatExchange(this.userChatMessage, [...this.agentChatEvents, agentChatEvent], this.agentToolCallGroup, this.id);
+    }
+}
+//# sourceMappingURL=AgentChatExchange.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentChatExchange.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentChatExchange.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EACL,iBAAiB,EACjB,eAAe,GAEhB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAe,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGjE;;;;GAIG;AACH,MAAM,OAAO,iBAAiB;IACnB,eAAe,CAA2B;IAC1C,kBAAkB,CAAqB;IACvC,EAAE,CAAS;IACX,eAAe,CAAkB;IAE1C,YACE,eAAqD,EACrD,eAAqD,EACrD,kBAAkB,GAAG,IAAI,kBAAkB,CAAC,EAAE,CAAC,EAC/C,EAAE,GAAG,MAAM,EAAE;QAEb,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAC7C,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IACzC,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY,YAAY,CAAC;IAC7D,CAAC;IAED,IAAI,cAAc;QAChB,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9C,IAAI,SAAS,YAAY,YAAY,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,YAAY,CAAC,cAAsC;QACjD,IAAI,aAAa,CAAC,cAAc,CAAC,EAAE,CAAC;YAClC,OAAO,IAAI,iBAAiB,CAC1B,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,cAAc,CAAC,EAClD,IAAI,CAAC,EAAE,CACR,CAAC;QACJ,CAAC;QACD,IAAI,YAAY,CAAC,cAAc,CAAC,EAAE,CAAC;YACjC,OAAO,IAAI,iBAAiB,CAC1B,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,cAAc,CAAC,EACjD,IAAI,CAAC,EAAE,CACR,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,iBAAiB,CAC1B,IAAI,CAAC,eAAe,EACpB,CAAC,GAAG,IAAI,CAAC,eAAe,EAAE,cAAc,CAAC,EACzC,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,EAAE,CACR,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 { AgentEndTurn } from \"./AgentEndTurn.ts\";\nimport {\n  AgentChatResponse,\n  UserChatMessage,\n  type AgentResponseContent,\n} from \"@dremio/js-sdk/enterprise\";\nimport { None, Some, type Option } from \"ts-results-es\";\nimport { AgentToolCallGroup } from \"./AgentToolCallGroup.ts\";\nimport { isToolRequest, isToolResult } from \"./AgentToolCall.ts\";\nimport type { AgentToolCall } from \"./AgentToolCall.ts\";\n\n/**\n * Represents a single chat exchange, starting with a `UserChatMessage` and followed by\n * 0 or more agent chat events. As chat events stream in, this class is updated\n * immutably but maintains a stable `.id` property.\n */\nexport class AgentChatExchange {\n  readonly agentChatEvents: AgentChatExchangeEvent[];\n  readonly agentToolCallGroup: AgentToolCallGroup;\n  readonly id: string;\n  readonly userChatMessage: UserChatMessage;\n\n  constructor(\n    userChatMessage: AgentChatExchange[\"userChatMessage\"],\n    agentChatEvents: AgentChatExchange[\"agentChatEvents\"],\n    agentToolCallGroup = new AgentToolCallGroup([]),\n    id = nanoid(),\n  ) {\n    this.agentChatEvents = agentChatEvents;\n    this.agentToolCallGroup = agentToolCallGroup;\n    this.id = id;\n    this.userChatMessage = userChatMessage;\n  }\n\n  get isComplete() {\n    return this.agentChatEvents.at(-1) instanceof AgentEndTurn;\n  }\n\n  get completeReason(): Option<AgentEndTurn[\"reason\"]> {\n    const lastEvent = this.agentChatEvents.at(-1);\n\n    if (lastEvent instanceof AgentEndTurn) {\n      return Some(lastEvent.reason);\n    }\n\n    return None;\n  }\n\n  addChatEvent(agentChatEvent: AgentChatExchangeEvent) {\n    if (isToolRequest(agentChatEvent)) {\n      return new AgentChatExchange(\n        this.userChatMessage,\n        this.agentChatEvents,\n        this.agentToolCallGroup.addRequest(agentChatEvent),\n        this.id,\n      );\n    }\n    if (isToolResult(agentChatEvent)) {\n      return new AgentChatExchange(\n        this.userChatMessage,\n        this.agentChatEvents,\n        this.agentToolCallGroup.addResult(agentChatEvent),\n        this.id,\n      );\n    }\n    return new AgentChatExchange(\n      this.userChatMessage,\n      [...this.agentChatEvents, agentChatEvent],\n      this.agentToolCallGroup,\n      this.id,\n    );\n  }\n}\n\nexport type AgentChatExchangeEvent =\n  | AgentChatResponse<AgentResponseContent>\n  | AgentEndTurn\n  | AgentToolCall\n  | AgentToolCallGroup;\n"]}

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

@@ -0,0 +1,42 @@
+import * as z from "zod/mini";
+declare const endTurnReasonSchema: z.ZodMiniDiscriminatedUnion<[z.ZodMiniObject<{
+    error: z.ZodMiniUnknown;
+    type: z.ZodMiniLiteral<"error">;
+}, z.core.$strip>, z.ZodMiniObject<{
+    type: z.ZodMiniLiteral<"reply_finished">;
+}, z.core.$strip>, z.ZodMiniObject<{
+    type: z.ZodMiniLiteral<"user_canceled">;
+}, z.core.$strip>], "type">;
+/**
+ * A marker emitted from the Agent event stream to
+ * track when the Agent has completed its turn.
+ */
+export declare class AgentEndTurn {
+    readonly id: string;
+    readonly reason: z.input<typeof endTurnReasonSchema>;
+    constructor(reason: AgentEndTurn["reason"], id?: string);
+    toJSON(): {
+        id: string;
+        reason: {
+            error: unknown;
+            type: "error";
+        } | {
+            type: "reply_finished";
+        } | {
+            type: "user_canceled";
+        };
+    };
+    static fromJSON(data: unknown): AgentEndTurn;
+    static codec: z.ZodMiniCodec<z.ZodMiniObject<{
+        id: z.ZodMiniString<string>;
+        reason: z.ZodMiniDiscriminatedUnion<[z.ZodMiniObject<{
+            error: z.ZodMiniUnknown;
+            type: z.ZodMiniLiteral<"error">;
+        }, z.core.$strip>, z.ZodMiniObject<{
+            type: z.ZodMiniLiteral<"reply_finished">;
+        }, z.core.$strip>, z.ZodMiniObject<{
+            type: z.ZodMiniLiteral<"user_canceled">;
+        }, z.core.$strip>], "type">;
+    }, z.core.$strip>, z.ZodMiniCustom<AgentEndTurn, AgentEndTurn>>;
+}
+export {};

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

@@ -0,0 +1,56 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { nanoid } from "nanoid/non-secure";
+import * as z from "zod/mini";
+const endTurnReasonSchema = z.discriminatedUnion("type", [
+    z.object({ error: z.unknown(), type: z.literal("error") }),
+    z.object({ type: z.literal("reply_finished") }),
+    z.object({ type: z.literal("user_canceled") }),
+]);
+/**
+ * A marker emitted from the Agent event stream to
+ * track when the Agent has completed its turn.
+ */
+export class AgentEndTurn {
+    id;
+    reason;
+    constructor(reason, id = nanoid()) {
+        this.reason = reason;
+        this.id = id;
+    }
+    toJSON() {
+        return z.encode(AgentEndTurn.codec, this);
+    }
+    static fromJSON(data) {
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any
+        return z.decode(AgentEndTurn.codec, data);
+    }
+    static codec = z.codec(z.object({
+        id: z.string(),
+        reason: endTurnReasonSchema,
+    }), z.instanceof(AgentEndTurn), {
+        decode(v) {
+            return new AgentEndTurn(v.reason, v.id);
+        },
+        encode(v) {
+            return {
+                id: v.id,
+                reason: v.reason,
+            };
+        },
+    });
+}
+//# sourceMappingURL=AgentEndTurn.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentEndTurn.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentEndTurn.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,mBAAmB,GAAG,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;IACvD,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAC1D,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;IAC/C,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;CAC/C,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,OAAO,YAAY;IACd,EAAE,CAAC;IACH,MAAM,CAAsC;IAErD,YAAY,MAA8B,EAAE,EAAE,GAAG,MAAM,EAAE;QACvD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAa;QAC3B,qGAAqG;QACrG,OAAO,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,EAAE,IAAW,CAAC,CAAC;IACnD,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpB,CAAC,CAAC,MAAM,CAAC;QACP,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;QACd,MAAM,EAAE,mBAAmB;KAC5B,CAAC,EACF,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,EAC1B;QACE,MAAM,CAAC,CAAC;YACN,OAAO,IAAI,YAAY,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAC1C,CAAC;QACD,MAAM,CAAC,CAAC;YACN,OAAO;gBACL,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,MAAM,EAAE,CAAC,CAAC,MAAM;aACR,CAAC;QACb,CAAC;KACF,CACF,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { nanoid } from \"nanoid/non-secure\";\nimport * as z from \"zod/mini\";\n\nconst endTurnReasonSchema = z.discriminatedUnion(\"type\", [\n  z.object({ error: z.unknown(), type: z.literal(\"error\") }),\n  z.object({ type: z.literal(\"reply_finished\") }),\n  z.object({ type: z.literal(\"user_canceled\") }),\n]);\n\n/**\n * A marker emitted from the Agent event stream to\n * track when the Agent has completed its turn.\n */\nexport class AgentEndTurn {\n  readonly id;\n  readonly reason: z.input<typeof endTurnReasonSchema>;\n\n  constructor(reason: AgentEndTurn[\"reason\"], id = nanoid()) {\n    this.reason = reason;\n    this.id = id;\n  }\n\n  toJSON() {\n    return z.encode(AgentEndTurn.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(AgentEndTurn.codec, data as any);\n  }\n\n  static codec = z.codec(\n    z.object({\n      id: z.string(),\n      reason: endTurnReasonSchema,\n    }),\n    z.instanceof(AgentEndTurn),\n    {\n      decode(v) {\n        return new AgentEndTurn(v.reason, v.id);\n      },\n      encode(v) {\n        return {\n          id: v.id,\n          reason: v.reason,\n        } as const;\n      },\n    },\n  );\n}\n"]}

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

@@ -0,0 +1,18 @@
+import { AgentChatResponse, AgentToolRequestResponseContent, AgentToolResultResponseContent } from "@dremio/js-sdk/enterprise";
+/**
+ * Pairs a tool request with its associated tool result once completed.
+ */
+export declare class AgentToolCall {
+    readonly request: AgentChatResponse<AgentToolRequestResponseContent>;
+    readonly result?: AgentChatResponse<AgentToolResultResponseContent>;
+    constructor(request: AgentToolCall["request"], result?: AgentToolCall["result"]);
+    get id(): string;
+    get status(): "success" | "error" | "pending";
+    get isError(): boolean;
+    get isPending(): boolean;
+    get isSuccess(): boolean;
+    withResult(response: AgentChatResponse<AgentToolResultResponseContent>): AgentToolCall;
+    toJSON(): unknown;
+}
+export declare function isToolRequest(message: unknown): message is AgentChatResponse<AgentToolRequestResponseContent>;
+export declare function isToolResult(message: unknown): message is AgentChatResponse<AgentToolResultResponseContent>;

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

@@ -0,0 +1,67 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { AgentChatResponse, AgentToolRequestResponseContent, AgentToolResultResponseContent, } from "@dremio/js-sdk/enterprise";
+/**
+ * Pairs a tool request with its associated tool result once completed.
+ */
+export class AgentToolCall {
+    request;
+    result;
+    constructor(request, result) {
+        this.request = request;
+        this.result = result;
+    }
+    get id() {
+        return this.request.content.callId;
+    }
+    get status() {
+        if (!this.result) {
+            return "pending";
+        }
+        if ("errorMessage" in this.result.content.result) {
+            return "error";
+        }
+        return "success";
+    }
+    get isError() {
+        return this.status === "error";
+    }
+    get isPending() {
+        return this.status === "pending";
+    }
+    get isSuccess() {
+        return this.status === "success";
+    }
+    withResult(response) {
+        return new AgentToolCall(this.request, response);
+    }
+    toJSON() {
+        return {
+            id: this.id,
+            request: this.request.toJSON(),
+            result: this.result?.toJSON(),
+        };
+    }
+}
+export function isToolRequest(message) {
+    return (message instanceof AgentChatResponse &&
+        message.content instanceof AgentToolRequestResponseContent);
+}
+export function isToolResult(message) {
+    return (message instanceof AgentChatResponse &&
+        message.content instanceof AgentToolResultResponseContent);
+}
+//# sourceMappingURL=AgentToolCall.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentToolCall.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/chat/eventWrappers/AgentToolCall.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,iBAAiB,EACjB,+BAA+B,EAC/B,8BAA8B,GAC/B,MAAM,2BAA2B,CAAC;AAEnC;;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 {\n  AgentChatResponse,\n  AgentToolRequestResponseContent,\n  AgentToolResultResponseContent,\n} from \"@dremio/js-sdk/enterprise\";\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

@@ -0,0 +1,18 @@
+import { AgentToolCall } from "./AgentToolCall.ts";
+import type { AgentChatResponse, AgentToolRequestResponseContent, AgentToolResultResponseContent } from "@dremio/js-sdk/enterprise";
+/**
+ * Contains a collection of `AgentToolCall` associated with an `AgentChatExchange`.
+ * As tool calls stream in, this class is updated immutably but maintains
+ * a stable `.id` property.
+ */
+export declare class AgentToolCallGroup {
+    readonly id: string;
+    readonly toolCalls: AgentToolCall[];
+    constructor(toolCalls: AgentToolCallGroup["toolCalls"], id?: AgentToolCallGroup["id"]);
+    addRequest(toolRequest: AgentChatResponse<AgentToolRequestResponseContent>): AgentToolCallGroup;
+    addResult(toolResult: AgentChatResponse<AgentToolResultResponseContent>): AgentToolCallGroup;
+    toJSON(): {
+        id: string;
+        toolCalls: unknown[];
+    };
+}

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

@@ -0,0 +1,54 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { nanoid } from "nanoid/non-secure";
+import { AgentToolCall } from "./AgentToolCall.js";
+/**
+ * Contains a collection of `AgentToolCall` associated with an `AgentChatExchange`.
+ * As tool calls stream in, this class is updated immutably but maintains
+ * a stable `.id` property.
+ */
+export class AgentToolCallGroup {
+    id = nanoid();
+    toolCalls;
+    constructor(toolCalls, id) {
+        this.toolCalls = toolCalls;
+        if (id) {
+            this.id = id;
+        }
+    }
+    addRequest(toolRequest) {
+        return new AgentToolCallGroup([...this.toolCalls, new AgentToolCall(toolRequest)], this.id);
+    }
+    addResult(toolResult) {
+        const requestIndex = this.toolCalls.findLastIndex((toolCall) => toolCall.id === toolResult.content.callId);
+        if (requestIndex === -1) {
+            return this;
+        }
+        const toolCall = this.toolCalls.at(requestIndex);
+        return new AgentToolCallGroup([
+            ...this.toolCalls.slice(0, requestIndex),
+            toolCall.withResult(toolResult),
+            ...this.toolCalls.slice(requestIndex + 1),
+        ]);
+    }
+    toJSON() {
+        return {
+            id: this.id,
+            toolCalls: this.toolCalls.map((toolCall) => toolCall.toJSON()),
+        };
+    }
+}
+//# sourceMappingURL=AgentToolCallGroup.js.map

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

@@ -0,0 +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;AAOnD;;;;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,CAAC;YAC5B,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,CAAC,CAAC;IACL,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 {\n  AgentChatResponse,\n  AgentToolRequestResponseContent,\n  AgentToolResultResponseContent,\n} from \"@dremio/js-sdk/enterprise\";\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      ...this.toolCalls.slice(0, requestIndex),\n      toolCall.withResult(toolResult),\n      ...this.toolCalls.slice(requestIndex + 1),\n    ]);\n  }\n\n  toJSON() {\n    return {\n      id: this.id,\n      toolCalls: this.toolCalls.map((toolCall) => toolCall.toJSON()),\n    };\n  }\n}\n"]}

package/package.json

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