@dremio/js-sdk 0.31.8 → 0.32.0

package/dist/enterprise/ai/AIResource.js

@@ -0,0 +1,123 @@
+/*
+ * 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 { catchError, map, of } from "rxjs";
+import { ChatSession } from "./chat/ChatSession.js";
+import { AgentChatResponse } from "./chat/AgentChatResponse.js";
+import { fromTextEventStream } from "../../common/fromTextEventStream.js";
+import { ModelProvider } from "./ModelProvider.js";
+import * as z from "zod/mini";
+import { modelProviderCreateCodec, modelProviderRetrieveCodec } from "./modelProviderCodec.js";
+import { safeParseResultToResult } from "../../common/safeParseResultToResult.js";
+import { Err, Ok } from "ts-results-es";
+import { Problem } from "../../common/Problem.js";
+export class AIResource {
+    #config;
+    constructor(config) {
+        this.#config = config;
+    }
+    createModelProvider(properties) {
+        const body = z.encode(modelProviderCreateCodec, properties);
+        return this.#config
+            .v4Request(`model-provider/config`, {
+            body: JSON.stringify(body),
+            headers: {
+                Accept: "application/json",
+                "Content-Type": "application/json",
+            },
+            keepalive: true,
+            method: "POST",
+        })
+            .map((res) => res.json())
+            .map((entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)));
+    }
+    listModelProviders() {
+        const getPage = ({ signal } = {}) => {
+            return this.#config
+                .v4Request("model-provider/config", {
+                headers: {
+                    Accept: "application/json",
+                },
+                signal,
+            })
+                .map((res) => res.json())
+                .map((response) => ({
+                data: response.configs.map((entity) => safeParseResultToResult(z.safeDecode(modelProviderRetrieveCodec, entity)).map((properties) => new ModelProvider(this.#config, properties))),
+            }));
+        };
+        return {
+            async *data({ signal } = {}) {
+                yield* (await getPage({ signal }).promise).map((response) => response.data).unwrap();
+            },
+            getPage,
+        };
+    }
+    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));
+    }
+    retrieveModelProvider(id, { signal } = {}) {
+        return this.#config
+            .v4Request(`model-provider/config/${id}`, {
+            headers: {
+                Accept: "application/json",
+            },
+            signal,
+        })
+            .map((res) => res.json())
+            .map((entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)));
+    }
+    /**
+     * Sends a chat message to the Dremio AI Agent and returns an Observable containing
+     * Agent replies.
+     */
+    sendChatMessage(chatSession) {
+        return (chatMessage) => fromTextEventStream(({ signal }) => this.#config
+            .sonarV4Request("agent", {
+            body: JSON.stringify({
+                message: chatMessage.toString(),
+                sessionId: chatSession.sessionId,
+            }),
+            headers: {
+                Accept: "text/event-stream",
+                "Content-Type": "application/json",
+            },
+            method: "POST",
+            signal,
+        })
+            .promise.then((result) => {
+            if (result.isErr()) {
+                throw result.error;
+            }
+            return result.value;
+        })).pipe(map((event) => Ok(AgentChatResponse.fromJSON(JSON.parse(event.data)))), catchError((e) => of(Err(refineChatErrors(e)))));
+    }
+}
+function refineChatErrors(e) {
+    if (e.body instanceof Problem) {
+        return e.body;
+    }
+    return e;
+}
+//# sourceMappingURL=AIResource.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AIResource.js","sourceRoot":"","sources":["../../../src/enterprise/ai/AIResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC3C,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,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,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,UAAU,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAC3D,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, map, of } 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\";\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   * 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        catchError((e: HttpError) => of(Err(refineChatErrors(e)))),\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

@@ -0,0 +1,181 @@
+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";
+export declare class ModelProvider implements Omit<ModelProviderProperties, "tag"> {
+    #private;
+    readonly config: ModelProviderProperties["config"];
+    readonly createdAt: ModelProviderProperties["createdAt"];
+    readonly createdBy: ModelProviderProperties["createdBy"];
+    readonly defaultModelName: ModelProviderProperties["defaultModelName"];
+    readonly id: ModelProviderProperties["id"];
+    readonly isDefault: ModelProviderProperties["isDefault"];
+    readonly modifiedAt: ModelProviderProperties["modifiedAt"];
+    readonly modifiedBy: ModelProviderProperties["modifiedBy"];
+    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>;
+    static createSchema: z.ZodMiniObject<{
+        name: z.ZodMiniString<string>;
+        config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+            amazonBedrock: z.ZodMiniObject<{
+                auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                    accessKey: z.ZodMiniObject<{
+                        accessKeyId: z.ZodMiniString<string>;
+                        secretAccessKey: z.ZodMiniString<string>;
+                    }, z.core.$strict>;
+                }, z.core.$strict>, z.ZodMiniObject<{
+                    role: 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";
+                }>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            anthropic: z.ZodMiniObject<{
+                accessKey: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            azureOpenAi: z.ZodMiniObject<{
+                auth: z.ZodMiniObject<{
+                    clientSecretAuth: z.ZodMiniObject<{
+                        azureClientId: z.ZodMiniString<string>;
+                        azureClientSecret: z.ZodMiniString<string>;
+                        azureTenantId: z.ZodMiniString<string>;
+                    }, z.core.$strict>;
+                }, z.core.$strict>;
+                resourceName: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            googleGemini: z.ZodMiniObject<{
+                accessKey: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            openAi: z.ZodMiniObject<{
+                accessKey: z.ZodMiniString<string>;
+                organizationId: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>]>;
+        defaultModelName: z.ZodMiniString<string>;
+        isDefault: z.ZodMiniBoolean<boolean>;
+        sqlModelNames: z.ZodMiniSet<z.ZodMiniString<string>>;
+    }, z.core.$strip>;
+    static updateSchema: z.ZodMiniObject<{
+        name: z.ZodMiniString<string>;
+        config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+            amazonBedrock: z.ZodMiniObject<{
+                auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                    accessKey: z.ZodMiniObject<{
+                        accessKeyId: z.ZodMiniString<string>;
+                        secretAccessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                    }, z.core.$strict>;
+                }, z.core.$strict>, z.ZodMiniObject<{
+                    role: 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";
+                }>>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            anthropic: z.ZodMiniObject<{
+                accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            azureOpenAi: z.ZodMiniObject<{
+                auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                    clientSecretAuth: z.ZodMiniObject<{
+                        azureClientId: z.ZodMiniString<string>;
+                        azureClientSecret: z.ZodMiniString<string>;
+                        azureTenantId: z.ZodMiniString<string>;
+                    }, z.core.$strict>;
+                }, z.core.$strict>, z.ZodMiniObject<{
+                    clientSecretAuth: z.ZodMiniObject<{
+                        azureClientId: z.ZodMiniString<string>;
+                        azureTenantId: z.ZodMiniString<string>;
+                        azureClientSecret: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                    }, z.core.$strict>;
+                }, z.core.$strict>]>>;
+                resourceName: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            googleGemini: z.ZodMiniObject<{
+                accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            openAi: z.ZodMiniObject<{
+                accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            }, z.core.$strict>;
+        }, z.core.$strict>]>;
+        defaultModelName: z.ZodMiniString<string>;
+        isDefault: z.ZodMiniBoolean<boolean>;
+        sqlModelNames: z.ZodMiniSet<z.ZodMiniString<string>>;
+    }, z.core.$strip>;
+}
+type ModelProviderProperties = z.output<typeof modelProviderRetrieveCodec>;
+export {};

package/dist/enterprise/ai/ModelProvider.js

@@ -0,0 +1,87 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import * as z from "zod/mini";
+import { modelProviderCreateCodec, modelProviderRetrieveCodec, modelProviderUpdateCodec, } from "./modelProviderCodec.js";
+import { VersionConflictError } from "../../common/problems.js";
+export class ModelProvider {
+    config;
+    createdAt;
+    createdBy;
+    defaultModelName;
+    id;
+    isDefault;
+    modifiedAt;
+    modifiedBy;
+    name;
+    sqlModelNames;
+    #config;
+    #tag;
+    constructor(config, properties) {
+        this.config = properties.config;
+        this.createdAt = properties.createdAt;
+        this.createdBy = properties.createdBy;
+        this.defaultModelName = properties.defaultModelName;
+        this.id = properties.id;
+        this.isDefault = properties.isDefault;
+        this.modifiedAt = properties.modifiedAt;
+        this.modifiedBy = properties.modifiedBy;
+        this.name = properties.name;
+        this.sqlModelNames = properties.sqlModelNames;
+        this.#config = config;
+        this.#tag = properties.tag;
+    }
+    delete() {
+        return this.#config
+            .v4Request(`model-provider/config/${this.id}`, {
+            keepalive: true,
+            method: "DELETE",
+        })
+            .map(() => undefined);
+    }
+    setDefault() {
+        return this.update({ isDefault: true });
+    }
+    update(fields) {
+        const body = z.encode(modelProviderUpdateCodec, {
+            config: fields.config ?? this.config,
+            defaultModelName: fields.defaultModelName ?? this.defaultModelName,
+            isDefault: fields.isDefault ?? this.isDefault,
+            name: fields.name ?? this.name,
+            sqlModelNames: fields.sqlModelNames ?? this.sqlModelNames,
+        });
+        return this.#config
+            .v4Request(`model-provider/config/${this.id}`, {
+            body: JSON.stringify({ ...body, tag: this.#tag }),
+            headers: {
+                Accept: "application/json",
+                "Content-Type": "application/json",
+            },
+            keepalive: true,
+            method: "PUT",
+        })
+            .map((res) => res.json())
+            .map((entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)))
+            .mapErr((val) => {
+            if (val.status === 409) {
+                return new VersionConflictError();
+            }
+            return val;
+        });
+    }
+    static createSchema = modelProviderCreateCodec.def.out;
+    static updateSchema = modelProviderUpdateCodec.def.out;
+}
+//# sourceMappingURL=ModelProvider.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelProvider.js","sourceRoot":"","sources":["../../../src/enterprise/ai/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,0BAA0B,CAAC;AAEhE,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/chat/AgentChatPartialResponse.d.ts

@@ -0,0 +1,42 @@
+import { ChatSession } from "./ChatSession.ts";
+import * as z from "zod/mini";
+declare const agentPartialResponseTypeSchema: z.ZodMiniEnum<{
+    delta: "delta";
+    end: "end";
+    start: "start";
+}>;
+export declare class AgentChatPartialResponse {
+    readonly id: string;
+    readonly session: ChatSession;
+    readonly content: string;
+    readonly type: z.infer<typeof agentPartialResponseTypeSchema>;
+    constructor(chatSession: AgentChatPartialResponse["session"], type: AgentChatPartialResponse["type"], content: AgentChatPartialResponse["content"], id?: AgentChatPartialResponse["id"]);
+    concat(next: AgentChatPartialResponse): AgentChatPartialResponse;
+    toJSON(): {
+        content: string;
+        id: string;
+        session: {
+            modelName?: string | undefined;
+            modelProviderId?: string | undefined;
+            sessionId?: string | undefined;
+        };
+        type: "delta" | "end" | "start";
+    };
+    get isComplete(): boolean;
+    static fromJSON(data: unknown): AgentChatPartialResponse;
+    static codec: z.ZodMiniCodec<z.ZodMiniObject<{
+        content: z.ZodMiniString<string>;
+        id: z.ZodMiniString<string>;
+        session: z.ZodMiniObject<{
+            modelName: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            modelProviderId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            sessionId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strip>;
+        type: z.ZodMiniEnum<{
+            delta: "delta";
+            end: "end";
+            start: "start";
+        }>;
+    }, z.core.$strip>, z.ZodMiniCustom<AgentChatPartialResponse, AgentChatPartialResponse>>;
+}
+export {};

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

@@ -0,0 +1,65 @@
+/*
+ * 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 { ChatSession } from "./ChatSession.js";
+import * as z from "zod/mini";
+const agentPartialResponseTypeSchema = z.enum(["delta", "end", "start"]);
+export class AgentChatPartialResponse {
+    id = nanoid();
+    session;
+    content;
+    type;
+    constructor(chatSession, type, content, id) {
+        this.session = chatSession;
+        this.type = type;
+        this.content = content;
+        if (id) {
+            this.id = id;
+        }
+    }
+    concat(next) {
+        return new AgentChatPartialResponse(this.session, next.type, this.content + next.content, this.id);
+    }
+    toJSON() {
+        return z.encode(AgentChatPartialResponse.codec, this);
+    }
+    get isComplete() {
+        return this.type === "end";
+    }
+    static fromJSON(data) {
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any
+        return z.decode(AgentChatPartialResponse.codec, data);
+    }
+    static codec = z.codec(z.object({
+        content: z.string(),
+        id: z.string(),
+        session: ChatSession.codec.def.in,
+        type: agentPartialResponseTypeSchema,
+    }), z.instanceof(AgentChatPartialResponse), {
+        decode(v) {
+            return new AgentChatPartialResponse(ChatSession.fromJSON(v.session), v.type, v.content, v.id);
+        },
+        encode(v) {
+            return {
+                content: v.content,
+                id: v.id,
+                session: v.session.toJSON(),
+                type: v.type,
+            };
+        },
+    });
+}
+//# sourceMappingURL=AgentChatPartialResponse.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AgentChatPartialResponse.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/AgentChatPartialResponse.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,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

@@ -0,0 +1,87 @@
+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 * as z from "zod/mini";
+import { Temporal } from "temporal-polyfill";
+export declare class AgentChatResponse<T extends AgentResponseContent> {
+    readonly createdAt: Temporal.Instant;
+    readonly id: string;
+    readonly session: ChatSession;
+    readonly content: T;
+    constructor(chatSession: ChatSession, content: T, createdAt?: Temporal.Instant, id?: string);
+    /**
+     * Creates a copy of the AgentResponse with a different content property.
+     * Preserves the original message's ID
+     */
+    replaceContent<R extends AgentResponseContent>(content: R): AgentChatResponse<R>;
+    toJSON(): {
+        chunkType: "error";
+        message: string;
+    } | {
+        chunkType: "model";
+        name: string;
+        result: Record<string, z.core.util.JSONType>;
+        createdAt: string;
+        modelName: string;
+        modelProviderId: string;
+        sessionId: string;
+    } | {
+        arguments: Record<string, z.core.util.JSONType>;
+        callId: string;
+        chunkType: "toolRequest";
+        name: string;
+        createdAt: string;
+        modelName: string;
+        modelProviderId: string;
+        sessionId: string;
+        commentary?: string | undefined;
+        summarizedTitle?: string | undefined;
+    } | {
+        callId: string;
+        chunkType: "toolResponse";
+        name: string;
+        result: Record<string, z.core.util.JSONType>;
+        createdAt: string;
+        modelName: string;
+        modelProviderId: string;
+        sessionId: string;
+        commentary?: string | undefined;
+    };
+    static fromJSON(data: unknown): AgentChatResponse<AgentResponseContent>;
+    static codec: z.ZodMiniCodec<z.ZodMiniDiscriminatedUnion<[z.ZodMiniObject<{
+        chunkType: z.ZodMiniLiteral<"error">;
+        message: z.ZodMiniString<string>;
+    }, z.core.$strip>, z.ZodMiniObject<{
+        chunkType: z.ZodMiniLiteral<"model">;
+        name: z.ZodMiniString<string>;
+        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+        createdAt: z.ZodMiniString<string>;
+        modelName: z.ZodMiniString<string>;
+        modelProviderId: z.ZodMiniString<string>;
+        sessionId: z.ZodMiniString<string>;
+    }, z.core.$strip>, z.ZodMiniObject<{
+        arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+        callId: z.ZodMiniString<string>;
+        chunkType: z.ZodMiniLiteral<"toolRequest">;
+        commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        name: z.ZodMiniString<string>;
+        summarizedTitle: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        createdAt: z.ZodMiniString<string>;
+        modelName: z.ZodMiniString<string>;
+        modelProviderId: z.ZodMiniString<string>;
+        sessionId: z.ZodMiniString<string>;
+    }, z.core.$strip>, z.ZodMiniObject<{
+        callId: z.ZodMiniString<string>;
+        chunkType: z.ZodMiniLiteral<"toolResponse">;
+        commentary: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        name: z.ZodMiniString<string>;
+        result: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniJSONSchema>;
+        createdAt: z.ZodMiniString<string>;
+        modelName: z.ZodMiniString<string>;
+        modelProviderId: z.ZodMiniString<string>;
+        sessionId: z.ZodMiniString<string>;
+    }, z.core.$strip>], "chunkType">, z.ZodMiniCustom<AgentChatResponse<AgentResponseContent>, AgentChatResponse<AgentResponseContent>>>;
+}
+export type AgentResponseContent = AgentErrorResponseContent | AgentModelResponseContent | AgentToolRequestResponseContent | AgentToolResultResponseContent;