@dremio/js-sdk 0.36.0 → 0.36.2

package/dist/enterprise/ai/AIResource.js

@@ -21,11 +21,28 @@ import { ModelProvider } from "./ModelProvider.js";
 import * as z from "zod/mini";
 import { modelProviderCreateCodec, modelProviderRetrieveCodec } from "./modelProviderCodec.js";
 import { safeParseResultToResult } from "../../common/safeParseResultToResult.js";
+import { listModelsPropertiesSchema } from "./modelProviderConfigSchemas.js";
 import { Err, Ok } from "ts-results-es";
 import { Problem } from "../../common/Problem.js";
 import { AgentChatExchange } from "./chat/eventWrappers/AgentChatExchange.js";
 import { AgentEndTurn } from "../interfaces.js";
 import { ChatSessionClient } from "./chat/ChatSessionClient.js";
+export function listAvailableModels(config) {
+    return function listAvailableModels(properties, { signal } = {}) {
+        return config
+            .v4Request("model-provider:listModels", {
+            body: JSON.stringify(z.parse(listModelsPropertiesSchema, properties)),
+            headers: {
+                Accept: "application/json",
+                "Content-Type": "application/json",
+            },
+            method: "POST",
+            signal,
+        })
+            .map((res) => res.json())
+            .map((result) => new Set(result.models));
+    };
+}
 export class AIResource {
     #config;
     constructor(config) {
@@ -68,17 +85,7 @@ export class AIResource {
         };
     }
     listAvailableModels(properties) {
-        return this.#config
-            .v4Request("model-provider:listModels", {
-            body: JSON.stringify({ access: properties }),
-            headers: {
-                Accept: "application/json",
-                "Content-Type": "application/json",
-            },
-            method: "POST",
-        })
-            .map((res) => res.json())
-            .map((result) => new Set(result.models));
+        return listAvailableModels(this.#config)(properties);
     }
     retrieveModelProvider(id, { signal } = {}) {
         return this.#config

package/dist/enterprise/ai/AIResource.js.map

@@ -1 +1 @@
-{"version":3,"file":"AIResource.js","sourceRoot":"","sources":["../../../src/enterprise/ai/AIResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAEhE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AAE/F,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAElF,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2CAA2C,CAAC;AAC9E,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,MAAM,OAAO,UAAU;IACrB,OAAO,CAAoC;IAE3C,YAAY,MAAyC;QACnD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,mBAAmB,CAAC,UAAqD;QACvE,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE,UAAU,CAAC,CAAC;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,uBAAuB,EAAE;YAClC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,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,CAAC;IACN,CAAC;IAED,kBAAkB;QAChB,MAAM,OAAO,GAAG,CAAC,EAAE,MAAM,KAAkB,EAAE,EAAE,EAAE;YAC/C,OAAO,IAAI,CAAC,OAAO;iBAChB,SAAS,CAAC,uBAAuB,EAAE;gBAClC,OAAO,EAAE;oBACP,MAAM,EAAE,kBAAkB;iBAC3B;gBACD,MAAM;aACP,CAAC;iBACD,GAAG,CACF,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAwE,CAC1F;iBACA,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAClB,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACpC,uBAAuB,CAAC,CAAC,CAAC,UAAU,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAC3E,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAC5D,CACF;aACF,CAAC,CAAC,CAAC;QACR,CAAC,CAAC;QACF,OAAO;YACL,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,KAAkB,EAAE;gBACtC,KAAK,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACvF,CAAC;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,UAAsD;QACxE,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,2BAA2B,EAAE;YACtC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;YAC5C,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,MAAM,EAAE,MAAM;SACf,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAmC,CAAC;aACzD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED,qBAAqB,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,EAAE,EAAE,EAAE;YACxC,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;aAC3B;YACD,MAAM;SACP,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,CAAC;IACN,CAAC;IAED;;;OAGG;IACH,wBAAwB,CAAC,WAAwB,EAAE,eAAgC;QACjF,MAAM,wBAAwB,GAAG,IAAI,iBAAiB,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAE5E,OAAO,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,CAAC,IAAI,CAC5D,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAC1C,OAAO,CAAC,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC,EACrD,UAAU,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,YAAY,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAC9E,IAAI,CACF,CAAC,iBAAiB,EAAE,cAAc,EAAE,EAAE,CAAC,iBAAiB,CAAC,aAAa,CAAC,cAAc,CAAC,EACtF,wBAAwB,CACzB,EACD,SAAS,CAAC,wBAAwB,CAAC,EACnC,KAAK,EAAE,CACR,CAAC;IACJ,CAAC;IAED,uBAAuB,CAAC,WAAwB;QAC9C,OAAO,IAAI,iBAAiB,CAAC;YAC3B,WAAW;YACX,wBAAwB,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC;SACnE,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,WAAwB;QACtC,OAAO,CAAC,WAA4B,EAAE,EAAE,CACtC,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACjC,IAAI,CAAC,OAAO;aACT,cAAc,CAAC,OAAO,EAAE;YACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,OAAO,EAAE,WAAW,CAAC,QAAQ,EAAE;gBAC/B,SAAS,EAAE,WAAW,CAAC,SAAS;aAKjC,CAAC;YACF,OAAO,EAAE;gBACP,MAAM,EAAE,mBAAmB;gBAC3B,cAAc,EAAE,kBAAkB;aACnC;YACD,MAAM,EAAE,MAAM;YACd,MAAM;SACP,CAAC;aACD,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACvB,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,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;YACxB,IAAI,iBAAiB,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC7B,WAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACtD,CAAC;QACH,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;IACN,CAAC;CACF;AAWD,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","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, endWith, map, of, scan, share, startWith, tap } from \"rxjs\";\nimport { ChatSession } from \"./chat/ChatSession.ts\";\nimport type { UserChatMessage } from \"./chat/UserChatMessage.ts\";\nimport { AgentChatResponse } from \"./chat/AgentChatResponse.ts\";\nimport type { Logger, SonarV4Config, V4Config } from \"../../common/Config.ts\";\nimport { fromTextEventStream } from \"../../common/fromTextEventStream.ts\";\nimport { ModelProvider } from \"./ModelProvider.ts\";\n\nimport * as z from \"zod/mini\";\nimport { modelProviderCreateCodec, modelProviderRetrieveCodec } from \"./modelProviderCodec.js\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { safeParseResultToResult } from \"../../common/safeParseResultToResult.ts\";\nimport type { listModelsPropertiesSchema } from \"./modelProviderConfigSchemas.ts\";\nimport { Err, Ok } from \"ts-results-es\";\nimport type { HttpError } from \"../../common/HttpError.ts\";\nimport { Problem } from \"../../common/Problem.ts\";\nimport { AgentChatExchange } from \"./chat/eventWrappers/AgentChatExchange.ts\";\nimport { AgentEndTurn } from \"../interfaces.ts\";\nimport { ChatSessionClient } from \"./chat/ChatSessionClient.ts\";\nexport class AIResource {\n  #config: Logger & V4Config & SonarV4Config;\n\n  constructor(config: Logger & V4Config & SonarV4Config) {\n    this.#config = config;\n  }\n\n  createModelProvider(properties: z.output<typeof modelProviderCreateCodec>) {\n    const body = z.encode(modelProviderCreateCodec, properties);\n    return this.#config\n      .v4Request(`model-provider/config`, {\n        body: JSON.stringify(body),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      );\n  }\n\n  listModelProviders() {\n    const getPage = ({ signal }: SignalParam = {}) => {\n      return this.#config\n        .v4Request(\"model-provider/config\", {\n          headers: {\n            Accept: \"application/json\",\n          },\n          signal,\n        })\n        .map(\n          (res) => res.json() as Promise<{ configs: z.input<typeof modelProviderRetrieveCodec>[] }>,\n        )\n        .map((response) => ({\n          data: response.configs.map((entity) =>\n            safeParseResultToResult(z.safeDecode(modelProviderRetrieveCodec, entity)).map(\n              (properties) => new ModelProvider(this.#config, properties),\n            ),\n          ),\n        }));\n    };\n    return {\n      async *data({ signal }: SignalParam = {}) {\n        yield* (await getPage({ signal }).promise).map((response) => response.data).unwrap();\n      },\n      getPage,\n    };\n  }\n\n  listAvailableModels(properties: z.infer<typeof listModelsPropertiesSchema>) {\n    return this.#config\n      .v4Request(\"model-provider:listModels\", {\n        body: JSON.stringify({ access: properties }),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        method: \"POST\",\n      })\n      .map((res) => res.json() as Promise<{ models: string[] }>)\n      .map((result) => new Set(result.models));\n  }\n\n  retrieveModelProvider(id: string, { signal }: SignalParam = {}) {\n    return this.#config\n      .v4Request(`model-provider/config/${id}`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n        signal,\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  }\n\n  /**\n   * Creates an Observable that emits each time the `AgentChatExchange` is updated\n   * with events from the Agent.\n   */\n  createAgentChatExchange$(chatSession: ChatSession, userChatMessage: UserChatMessage) {\n    const initialAgentChatExchange = new AgentChatExchange(userChatMessage, []);\n\n    return this.sendChatMessage(chatSession)(userChatMessage).pipe(\n      map((eventResult) => eventResult.unwrap()),\n      endWith(new AgentEndTurn({ type: \"reply_finished\" })),\n      catchError((error: unknown) => of(new AgentEndTurn({ error, type: \"error\" }))),\n      scan(\n        (agentChatExchange, agentChatEvent) => agentChatExchange._addChatEvent(agentChatEvent),\n        initialAgentChatExchange,\n      ),\n      startWith(initialAgentChatExchange),\n      share(),\n    );\n  }\n\n  createChatSessionClient(chatSession: ChatSession) {\n    return new ChatSessionClient({\n      chatSession,\n      createAgentChatExchange$: this.createAgentChatExchange$.bind(this),\n    });\n  }\n\n  /**\n   * Sends a chat message to the Dremio AI Agent and returns an Observable containing\n   * Agent replies.\n   */\n  sendChatMessage(chatSession: ChatSession) {\n    return (chatMessage: UserChatMessage) =>\n      fromTextEventStream(({ signal }) =>\n        this.#config\n          .sonarV4Request(\"agent\", {\n            body: JSON.stringify({\n              message: chatMessage.toString(),\n              sessionId: chatSession.sessionId,\n            } satisfies {\n              message: string;\n              modelProviderId?: string;\n              sessionId?: string;\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}\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"]}
+{"version":3,"file":"AIResource.js","sourceRoot":"","sources":["../../../src/enterprise/ai/AIResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAEhE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AAE/F,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAC7E,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2CAA2C,CAAC;AAC9E,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAEhE,MAAM,UAAU,mBAAmB,CAAC,MAAgB;IAClD,OAAO,SAAS,mBAAmB,CACjC,UAAsD,EACtD,EAAE,MAAM,KAAkB,EAAE;QAE5B,OAAO,MAAM;aACV,SAAS,CAAC,2BAA2B,EAAE;YACtC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,0BAA0B,EAAE,UAAU,CAAC,CAAC;YACrE,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,MAAM,EAAE,MAAM;YACd,MAAM;SACP,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAmC,CAAC;aACzD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,UAAU;IACrB,OAAO,CAAoC;IAE3C,YAAY,MAAyC;QACnD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,mBAAmB,CAAC,UAAqD;QACvE,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE,UAAU,CAAC,CAAC;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,uBAAuB,EAAE;YAClC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,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,CAAC;IACN,CAAC;IAED,kBAAkB;QAChB,MAAM,OAAO,GAAG,CAAC,EAAE,MAAM,KAAkB,EAAE,EAAE,EAAE;YAC/C,OAAO,IAAI,CAAC,OAAO;iBAChB,SAAS,CAAC,uBAAuB,EAAE;gBAClC,OAAO,EAAE;oBACP,MAAM,EAAE,kBAAkB;iBAC3B;gBACD,MAAM;aACP,CAAC;iBACD,GAAG,CACF,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAwE,CAC1F;iBACA,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAClB,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACpC,uBAAuB,CAAC,CAAC,CAAC,UAAU,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAC3E,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAC5D,CACF;aACF,CAAC,CAAC,CAAC;QACR,CAAC,CAAC;QACF,OAAO;YACL,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,KAAkB,EAAE;gBACtC,KAAK,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACvF,CAAC;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,UAAsD;QACxE,OAAO,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC;IACvD,CAAC;IAED,qBAAqB,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,EAAE,EAAE,EAAE;YACxC,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;aAC3B;YACD,MAAM;SACP,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,CAAC;IACN,CAAC;IAED;;;OAGG;IACH,wBAAwB,CAAC,WAAwB,EAAE,eAAgC;QACjF,MAAM,wBAAwB,GAAG,IAAI,iBAAiB,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAE5E,OAAO,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,CAAC,IAAI,CAC5D,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAC1C,OAAO,CAAC,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC,EACrD,UAAU,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,YAAY,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAC9E,IAAI,CACF,CAAC,iBAAiB,EAAE,cAAc,EAAE,EAAE,CAAC,iBAAiB,CAAC,aAAa,CAAC,cAAc,CAAC,EACtF,wBAAwB,CACzB,EACD,SAAS,CAAC,wBAAwB,CAAC,EACnC,KAAK,EAAE,CACR,CAAC;IACJ,CAAC;IAED,uBAAuB,CAAC,WAAwB;QAC9C,OAAO,IAAI,iBAAiB,CAAC;YAC3B,WAAW;YACX,wBAAwB,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC;SACnE,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,WAAwB;QACtC,OAAO,CAAC,WAA4B,EAAE,EAAE,CACtC,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACjC,IAAI,CAAC,OAAO;aACT,cAAc,CAAC,OAAO,EAAE;YACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,OAAO,EAAE,WAAW,CAAC,QAAQ,EAAE;gBAC/B,SAAS,EAAE,WAAW,CAAC,SAAS;aAKjC,CAAC;YACF,OAAO,EAAE;gBACP,MAAM,EAAE,mBAAmB;gBAC3B,cAAc,EAAE,kBAAkB;aACnC;YACD,MAAM,EAAE,MAAM;YACd,MAAM;SACP,CAAC;aACD,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACvB,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,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;YACxB,IAAI,iBAAiB,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC7B,WAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACtD,CAAC;QACH,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;IACN,CAAC;CACF;AAWD,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","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, endWith, map, of, scan, share, startWith, tap } from \"rxjs\";\nimport { ChatSession } from \"./chat/ChatSession.ts\";\nimport type { UserChatMessage } from \"./chat/UserChatMessage.ts\";\nimport { AgentChatResponse } from \"./chat/AgentChatResponse.ts\";\nimport type { Logger, SonarV4Config, V4Config } from \"../../common/Config.ts\";\nimport { fromTextEventStream } from \"../../common/fromTextEventStream.ts\";\nimport { ModelProvider } from \"./ModelProvider.ts\";\n\nimport * as z from \"zod/mini\";\nimport { modelProviderCreateCodec, modelProviderRetrieveCodec } from \"./modelProviderCodec.js\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { safeParseResultToResult } from \"../../common/safeParseResultToResult.ts\";\nimport { listModelsPropertiesSchema } from \"./modelProviderConfigSchemas.js\";\nimport { Err, Ok } from \"ts-results-es\";\nimport type { HttpError } from \"../../common/HttpError.ts\";\nimport { Problem } from \"../../common/Problem.ts\";\nimport { AgentChatExchange } from \"./chat/eventWrappers/AgentChatExchange.ts\";\nimport { AgentEndTurn } from \"../interfaces.ts\";\nimport { ChatSessionClient } from \"./chat/ChatSessionClient.ts\";\n\nexport function listAvailableModels(config: V4Config) {\n  return function listAvailableModels(\n    properties: z.infer<typeof listModelsPropertiesSchema>,\n    { signal }: SignalParam = {},\n  ) {\n    return config\n      .v4Request(\"model-provider:listModels\", {\n        body: JSON.stringify(z.parse(listModelsPropertiesSchema, properties)),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        method: \"POST\",\n        signal,\n      })\n      .map((res) => res.json() as Promise<{ models: string[] }>)\n      .map((result) => new Set(result.models));\n  };\n}\n\nexport class AIResource {\n  #config: Logger & V4Config & SonarV4Config;\n\n  constructor(config: Logger & V4Config & SonarV4Config) {\n    this.#config = config;\n  }\n\n  createModelProvider(properties: z.output<typeof modelProviderCreateCodec>) {\n    const body = z.encode(modelProviderCreateCodec, properties);\n    return this.#config\n      .v4Request(`model-provider/config`, {\n        body: JSON.stringify(body),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      );\n  }\n\n  listModelProviders() {\n    const getPage = ({ signal }: SignalParam = {}) => {\n      return this.#config\n        .v4Request(\"model-provider/config\", {\n          headers: {\n            Accept: \"application/json\",\n          },\n          signal,\n        })\n        .map(\n          (res) => res.json() as Promise<{ configs: z.input<typeof modelProviderRetrieveCodec>[] }>,\n        )\n        .map((response) => ({\n          data: response.configs.map((entity) =>\n            safeParseResultToResult(z.safeDecode(modelProviderRetrieveCodec, entity)).map(\n              (properties) => new ModelProvider(this.#config, properties),\n            ),\n          ),\n        }));\n    };\n    return {\n      async *data({ signal }: SignalParam = {}) {\n        yield* (await getPage({ signal }).promise).map((response) => response.data).unwrap();\n      },\n      getPage,\n    };\n  }\n\n  listAvailableModels(properties: z.infer<typeof listModelsPropertiesSchema>) {\n    return listAvailableModels(this.#config)(properties);\n  }\n\n  retrieveModelProvider(id: string, { signal }: SignalParam = {}) {\n    return this.#config\n      .v4Request(`model-provider/config/${id}`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n        signal,\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  }\n\n  /**\n   * Creates an Observable that emits each time the `AgentChatExchange` is updated\n   * with events from the Agent.\n   */\n  createAgentChatExchange$(chatSession: ChatSession, userChatMessage: UserChatMessage) {\n    const initialAgentChatExchange = new AgentChatExchange(userChatMessage, []);\n\n    return this.sendChatMessage(chatSession)(userChatMessage).pipe(\n      map((eventResult) => eventResult.unwrap()),\n      endWith(new AgentEndTurn({ type: \"reply_finished\" })),\n      catchError((error: unknown) => of(new AgentEndTurn({ error, type: \"error\" }))),\n      scan(\n        (agentChatExchange, agentChatEvent) => agentChatExchange._addChatEvent(agentChatEvent),\n        initialAgentChatExchange,\n      ),\n      startWith(initialAgentChatExchange),\n      share(),\n    );\n  }\n\n  createChatSessionClient(chatSession: ChatSession) {\n    return new ChatSessionClient({\n      chatSession,\n      createAgentChatExchange$: this.createAgentChatExchange$.bind(this),\n    });\n  }\n\n  /**\n   * Sends a chat message to the Dremio AI Agent and returns an Observable containing\n   * Agent replies.\n   */\n  sendChatMessage(chatSession: ChatSession) {\n    return (chatMessage: UserChatMessage) =>\n      fromTextEventStream(({ signal }) =>\n        this.#config\n          .sonarV4Request(\"agent\", {\n            body: JSON.stringify({\n              message: chatMessage.toString(),\n              sessionId: chatSession.sessionId,\n            } satisfies {\n              message: string;\n              modelProviderId?: string;\n              sessionId?: string;\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}\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"]}

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

@@ -32,38 +32,7 @@ export declare class ModelProvider implements Omit<ModelProviderProperties, "tag
                         roleArn: z.ZodMiniString<string>;
                     }, z.core.$strict>;
                 }, z.core.$strict>]>;
-                region: z.ZodMiniEnum<{
-                    "ap-east-2": "ap-east-2";
-                    "ap-northeast-1": "ap-northeast-1";
-                    "ap-northeast-2": "ap-northeast-2";
-                    "ap-northeast-3": "ap-northeast-3";
-                    "ap-south-1": "ap-south-1";
-                    "ap-south-2": "ap-south-2";
-                    "ap-southeast-1": "ap-southeast-1";
-                    "ap-southeast-2": "ap-southeast-2";
-                    "ap-southeast-3": "ap-southeast-3";
-                    "ap-southeast-4": "ap-southeast-4";
-                    "ap-southeast-5": "ap-southeast-5";
-                    "ap-southeast-7": "ap-southeast-7";
-                    "ca-central-1": "ca-central-1";
-                    "eu-central-1": "eu-central-1";
-                    "eu-central-2": "eu-central-2";
-                    "eu-north-1": "eu-north-1";
-                    "eu-south-1": "eu-south-1";
-                    "eu-south-2": "eu-south-2";
-                    "eu-west-1": "eu-west-1";
-                    "eu-west-2": "eu-west-2";
-                    "eu-west-3": "eu-west-3";
-                    "il-central-1": "il-central-1";
-                    "me-central-1": "me-central-1";
-                    "sa-east-1": "sa-east-1";
-                    "us-east-1": "us-east-1";
-                    "us-east-2": "us-east-2";
-                    "us-gov-east-1": "us-gov-east-1";
-                    "us-gov-west-1": "us-gov-west-1";
-                    "us-west-1": "us-west-1";
-                    "us-west-2": "us-west-2";
-                }>;
+                region: z.ZodMiniString<string>;
             }, z.core.$strict>;
         }, z.core.$strict>, z.ZodMiniObject<{
             anthropic: z.ZodMiniObject<{
@@ -108,38 +77,7 @@ export declare class ModelProvider implements Omit<ModelProviderProperties, "tag
                         roleArn: z.ZodMiniString<string>;
                     }, z.core.$strict>;
                 }, z.core.$strict>]>>;
-                region: z.ZodMiniOptional<z.ZodMiniEnum<{
-                    "ap-east-2": "ap-east-2";
-                    "ap-northeast-1": "ap-northeast-1";
-                    "ap-northeast-2": "ap-northeast-2";
-                    "ap-northeast-3": "ap-northeast-3";
-                    "ap-south-1": "ap-south-1";
-                    "ap-south-2": "ap-south-2";
-                    "ap-southeast-1": "ap-southeast-1";
-                    "ap-southeast-2": "ap-southeast-2";
-                    "ap-southeast-3": "ap-southeast-3";
-                    "ap-southeast-4": "ap-southeast-4";
-                    "ap-southeast-5": "ap-southeast-5";
-                    "ap-southeast-7": "ap-southeast-7";
-                    "ca-central-1": "ca-central-1";
-                    "eu-central-1": "eu-central-1";
-                    "eu-central-2": "eu-central-2";
-                    "eu-north-1": "eu-north-1";
-                    "eu-south-1": "eu-south-1";
-                    "eu-south-2": "eu-south-2";
-                    "eu-west-1": "eu-west-1";
-                    "eu-west-2": "eu-west-2";
-                    "eu-west-3": "eu-west-3";
-                    "il-central-1": "il-central-1";
-                    "me-central-1": "me-central-1";
-                    "sa-east-1": "sa-east-1";
-                    "us-east-1": "us-east-1";
-                    "us-east-2": "us-east-2";
-                    "us-gov-east-1": "us-gov-east-1";
-                    "us-gov-west-1": "us-gov-west-1";
-                    "us-west-1": "us-west-1";
-                    "us-west-2": "us-west-2";
-                }>>;
+                region: z.ZodMiniOptional<z.ZodMiniString<string>>;
             }, z.core.$strict>;
         }, z.core.$strict>, z.ZodMiniObject<{
             anthropic: z.ZodMiniObject<{

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

@@ -42,7 +42,7 @@ export class AgentToolCallGroup {
             ...this.toolCalls.slice(0, requestIndex),
             toolCall.withResult(toolResult),
             ...this.toolCalls.slice(requestIndex + 1),
-        ]);
+        ], this.id);
     }
     toJSON() {
         return {

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

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

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

@@ -22,38 +22,7 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniObject<
                     roleArn: z.ZodMiniString<string>;
                 }, z.core.$strict>;
             }, z.core.$strict>]>;
-            region: z.ZodMiniEnum<{
-                "ap-east-2": "ap-east-2";
-                "ap-northeast-1": "ap-northeast-1";
-                "ap-northeast-2": "ap-northeast-2";
-                "ap-northeast-3": "ap-northeast-3";
-                "ap-south-1": "ap-south-1";
-                "ap-south-2": "ap-south-2";
-                "ap-southeast-1": "ap-southeast-1";
-                "ap-southeast-2": "ap-southeast-2";
-                "ap-southeast-3": "ap-southeast-3";
-                "ap-southeast-4": "ap-southeast-4";
-                "ap-southeast-5": "ap-southeast-5";
-                "ap-southeast-7": "ap-southeast-7";
-                "ca-central-1": "ca-central-1";
-                "eu-central-1": "eu-central-1";
-                "eu-central-2": "eu-central-2";
-                "eu-north-1": "eu-north-1";
-                "eu-south-1": "eu-south-1";
-                "eu-south-2": "eu-south-2";
-                "eu-west-1": "eu-west-1";
-                "eu-west-2": "eu-west-2";
-                "eu-west-3": "eu-west-3";
-                "il-central-1": "il-central-1";
-                "me-central-1": "me-central-1";
-                "sa-east-1": "sa-east-1";
-                "us-east-1": "us-east-1";
-                "us-east-2": "us-east-2";
-                "us-gov-east-1": "us-gov-east-1";
-                "us-gov-west-1": "us-gov-west-1";
-                "us-west-1": "us-west-1";
-                "us-west-2": "us-west-2";
-            }>;
+            region: z.ZodMiniString<string>;
         }, z.core.$strict>;
     }, z.core.$strict>, z.ZodMiniObject<{
         anthropic: z.ZodMiniObject<{}, z.core.$strict>;
@@ -91,38 +60,7 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniObject<
                     roleArn: z.ZodMiniString<string>;
                 }, z.core.$strict>;
             }, z.core.$strict>]>;
-            region: z.ZodMiniEnum<{
-                "ap-east-2": "ap-east-2";
-                "ap-northeast-1": "ap-northeast-1";
-                "ap-northeast-2": "ap-northeast-2";
-                "ap-northeast-3": "ap-northeast-3";
-                "ap-south-1": "ap-south-1";
-                "ap-south-2": "ap-south-2";
-                "ap-southeast-1": "ap-southeast-1";
-                "ap-southeast-2": "ap-southeast-2";
-                "ap-southeast-3": "ap-southeast-3";
-                "ap-southeast-4": "ap-southeast-4";
-                "ap-southeast-5": "ap-southeast-5";
-                "ap-southeast-7": "ap-southeast-7";
-                "ca-central-1": "ca-central-1";
-                "eu-central-1": "eu-central-1";
-                "eu-central-2": "eu-central-2";
-                "eu-north-1": "eu-north-1";
-                "eu-south-1": "eu-south-1";
-                "eu-south-2": "eu-south-2";
-                "eu-west-1": "eu-west-1";
-                "eu-west-2": "eu-west-2";
-                "eu-west-3": "eu-west-3";
-                "il-central-1": "il-central-1";
-                "me-central-1": "me-central-1";
-                "sa-east-1": "sa-east-1";
-                "us-east-1": "us-east-1";
-                "us-east-2": "us-east-2";
-                "us-gov-east-1": "us-gov-east-1";
-                "us-gov-west-1": "us-gov-west-1";
-                "us-west-1": "us-west-1";
-                "us-west-2": "us-west-2";
-            }>;
+            region: z.ZodMiniString<string>;
         }, z.core.$strict>;
     }, z.core.$strict>, z.ZodMiniObject<{
         anthropic: z.ZodMiniObject<{}, z.core.$strict>;
@@ -166,38 +104,7 @@ export declare const modelProviderCreateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
                     roleArn: z.ZodMiniString<string>;
                 }, z.core.$strict>;
             }, z.core.$strict>]>;
-            region: z.ZodMiniEnum<{
-                "ap-east-2": "ap-east-2";
-                "ap-northeast-1": "ap-northeast-1";
-                "ap-northeast-2": "ap-northeast-2";
-                "ap-northeast-3": "ap-northeast-3";
-                "ap-south-1": "ap-south-1";
-                "ap-south-2": "ap-south-2";
-                "ap-southeast-1": "ap-southeast-1";
-                "ap-southeast-2": "ap-southeast-2";
-                "ap-southeast-3": "ap-southeast-3";
-                "ap-southeast-4": "ap-southeast-4";
-                "ap-southeast-5": "ap-southeast-5";
-                "ap-southeast-7": "ap-southeast-7";
-                "ca-central-1": "ca-central-1";
-                "eu-central-1": "eu-central-1";
-                "eu-central-2": "eu-central-2";
-                "eu-north-1": "eu-north-1";
-                "eu-south-1": "eu-south-1";
-                "eu-south-2": "eu-south-2";
-                "eu-west-1": "eu-west-1";
-                "eu-west-2": "eu-west-2";
-                "eu-west-3": "eu-west-3";
-                "il-central-1": "il-central-1";
-                "me-central-1": "me-central-1";
-                "sa-east-1": "sa-east-1";
-                "us-east-1": "us-east-1";
-                "us-east-2": "us-east-2";
-                "us-gov-east-1": "us-gov-east-1";
-                "us-gov-west-1": "us-gov-west-1";
-                "us-west-1": "us-west-1";
-                "us-west-2": "us-west-2";
-            }>;
+            region: z.ZodMiniString<string>;
         }, z.core.$strict>;
     }, z.core.$strict>, z.ZodMiniObject<{
         anthropic: z.ZodMiniObject<{
@@ -238,38 +145,7 @@ export declare const modelProviderCreateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
                     roleArn: z.ZodMiniString<string>;
                 }, z.core.$strict>;
             }, z.core.$strict>]>;
-            region: z.ZodMiniEnum<{
-                "ap-east-2": "ap-east-2";
-                "ap-northeast-1": "ap-northeast-1";
-                "ap-northeast-2": "ap-northeast-2";
-                "ap-northeast-3": "ap-northeast-3";
-                "ap-south-1": "ap-south-1";
-                "ap-south-2": "ap-south-2";
-                "ap-southeast-1": "ap-southeast-1";
-                "ap-southeast-2": "ap-southeast-2";
-                "ap-southeast-3": "ap-southeast-3";
-                "ap-southeast-4": "ap-southeast-4";
-                "ap-southeast-5": "ap-southeast-5";
-                "ap-southeast-7": "ap-southeast-7";
-                "ca-central-1": "ca-central-1";
-                "eu-central-1": "eu-central-1";
-                "eu-central-2": "eu-central-2";
-                "eu-north-1": "eu-north-1";
-                "eu-south-1": "eu-south-1";
-                "eu-south-2": "eu-south-2";
-                "eu-west-1": "eu-west-1";
-                "eu-west-2": "eu-west-2";
-                "eu-west-3": "eu-west-3";
-                "il-central-1": "il-central-1";
-                "me-central-1": "me-central-1";
-                "sa-east-1": "sa-east-1";
-                "us-east-1": "us-east-1";
-                "us-east-2": "us-east-2";
-                "us-gov-east-1": "us-gov-east-1";
-                "us-gov-west-1": "us-gov-west-1";
-                "us-west-1": "us-west-1";
-                "us-west-2": "us-west-2";
-            }>;
+            region: z.ZodMiniString<string>;
         }, z.core.$strict>;
     }, z.core.$strict>, z.ZodMiniObject<{
         anthropic: z.ZodMiniObject<{
@@ -317,38 +193,7 @@ export declare const modelProviderUpdateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
                     roleArn: z.ZodMiniString<string>;
                 }, z.core.$strict>;
             }, z.core.$strict>]>>;
-            region: z.ZodMiniOptional<z.ZodMiniEnum<{
-                "ap-east-2": "ap-east-2";
-                "ap-northeast-1": "ap-northeast-1";
-                "ap-northeast-2": "ap-northeast-2";
-                "ap-northeast-3": "ap-northeast-3";
-                "ap-south-1": "ap-south-1";
-                "ap-south-2": "ap-south-2";
-                "ap-southeast-1": "ap-southeast-1";
-                "ap-southeast-2": "ap-southeast-2";
-                "ap-southeast-3": "ap-southeast-3";
-                "ap-southeast-4": "ap-southeast-4";
-                "ap-southeast-5": "ap-southeast-5";
-                "ap-southeast-7": "ap-southeast-7";
-                "ca-central-1": "ca-central-1";
-                "eu-central-1": "eu-central-1";
-                "eu-central-2": "eu-central-2";
-                "eu-north-1": "eu-north-1";
-                "eu-south-1": "eu-south-1";
-                "eu-south-2": "eu-south-2";
-                "eu-west-1": "eu-west-1";
-                "eu-west-2": "eu-west-2";
-                "eu-west-3": "eu-west-3";
-                "il-central-1": "il-central-1";
-                "me-central-1": "me-central-1";
-                "sa-east-1": "sa-east-1";
-                "us-east-1": "us-east-1";
-                "us-east-2": "us-east-2";
-                "us-gov-east-1": "us-gov-east-1";
-                "us-gov-west-1": "us-gov-west-1";
-                "us-west-1": "us-west-1";
-                "us-west-2": "us-west-2";
-            }>>;
+            region: z.ZodMiniOptional<z.ZodMiniString<string>>;
         }, z.core.$strict>;
     }, z.core.$strict>, z.ZodMiniObject<{
         anthropic: z.ZodMiniObject<{
@@ -395,38 +240,7 @@ export declare const modelProviderUpdateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
                     roleArn: z.ZodMiniString<string>;
                 }, z.core.$strict>;
             }, z.core.$strict>]>>;
-            region: z.ZodMiniOptional<z.ZodMiniEnum<{
-                "ap-east-2": "ap-east-2";
-                "ap-northeast-1": "ap-northeast-1";
-                "ap-northeast-2": "ap-northeast-2";
-                "ap-northeast-3": "ap-northeast-3";
-                "ap-south-1": "ap-south-1";
-                "ap-south-2": "ap-south-2";
-                "ap-southeast-1": "ap-southeast-1";
-                "ap-southeast-2": "ap-southeast-2";
-                "ap-southeast-3": "ap-southeast-3";
-                "ap-southeast-4": "ap-southeast-4";
-                "ap-southeast-5": "ap-southeast-5";
-                "ap-southeast-7": "ap-southeast-7";
-                "ca-central-1": "ca-central-1";
-                "eu-central-1": "eu-central-1";
-                "eu-central-2": "eu-central-2";
-                "eu-north-1": "eu-north-1";
-                "eu-south-1": "eu-south-1";
-                "eu-south-2": "eu-south-2";
-                "eu-west-1": "eu-west-1";
-                "eu-west-2": "eu-west-2";
-                "eu-west-3": "eu-west-3";
-                "il-central-1": "il-central-1";
-                "me-central-1": "me-central-1";
-                "sa-east-1": "sa-east-1";
-                "us-east-1": "us-east-1";
-                "us-east-2": "us-east-2";
-                "us-gov-east-1": "us-gov-east-1";
-                "us-gov-west-1": "us-gov-west-1";
-                "us-west-1": "us-west-1";
-                "us-west-2": "us-west-2";
-            }>>;
+            region: z.ZodMiniOptional<z.ZodMiniString<string>>;
         }, z.core.$strict>;
     }, z.core.$strict>, z.ZodMiniObject<{
         anthropic: z.ZodMiniObject<{