@dremio/js-sdk 0.38.0 → 0.39.1

package/dist/cloud/ai/AIResource.d.ts

@@ -1,6 +1,5 @@
 import { ChatSession } from "../../enterprise/ai/chat/ChatSession.ts";
 import type { UserChatMessage } from "../../enterprise/ai/chat/UserChatMessage.ts";
-import { AgentChatResponse } from "../../enterprise/ai/chat/AgentChatResponse.ts";
 import type { Logger, SonarV4Config, V4Config } from "../../common/Config.ts";
 import { ModelProvider } from "../../enterprise/ai/ModelProvider.ts";
 import * as z from "zod/mini";
@@ -8,6 +7,10 @@ import { modelProviderCreateCodec } from "./modelProviderCodec.js";
 import { DremioModelProvider } from "./DremioModelProvider.ts";
 import { listAvailableModels } from "../../enterprise/ai/AIResource.ts";
 import type { SignalParam } from "../../common/Params.ts";
+import type { AgentChatResponse, AgentResponseContent } from "../../enterprise/ai/chat/AgentChatResponse.ts";
+import { type Observable } from "rxjs";
+import { AgentChatExchange } from "../../enterprise/ai/chat/eventWrappers/AgentChatExchange.ts";
+import { ChatSessionClient } from "../../enterprise/ai/chat/ChatSessionClient.ts";
 export declare class AIResource {
     #private;
     constructor(config: Logger & V4Config & SonarV4Config);
@@ -47,6 +50,7 @@ export declare class AIResource {
         } | {
             openAi: {
                 organizationId: string;
+                region: "GLOBAL" | "US" | "EU";
             };
         };
         defaultModelName: string;
@@ -102,6 +106,7 @@ export declare class AIResource {
             } | {
                 openAi: {
                     organizationId: string;
+                    region: "GLOBAL" | "US" | "EU";
                 };
             };
             defaultModelName: string;
@@ -157,6 +162,7 @@ export declare class AIResource {
                 } | {
                     openAi: {
                         organizationId: string;
+                        region: "GLOBAL" | "US" | "EU";
                     };
                 };
                 defaultModelName: string;
@@ -214,6 +220,7 @@ export declare class AIResource {
         } | {
             openAi: {
                 organizationId: string;
+                region: "GLOBAL" | "US" | "EU";
             };
         };
         defaultModelName: string;
@@ -236,5 +243,13 @@ export declare class AIResource {
      * Sends a chat message to the Dremio AI Agent and returns an Observable containing
      * Agent replies.
      */
-    sendChatMessage(chatSession: ChatSession): (chatMessage: UserChatMessage) => import("rxjs").Observable<AgentChatResponse<import("../../enterprise/ai/chat/AgentChatResponse.ts").AgentResponseContent>>;
+    sendChatMessage(chatSession: ChatSession, userChatMessage: UserChatMessage): Observable<import("ts-results-es").Ok<AgentChatResponse<AgentResponseContent>> | import("ts-results-es").Err<import("../index.ts").HttpError | import("../index.ts").Problem<"https://api.dremio.dev/problems/ai-agent/chat/agent-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/input-too-large" | "https://api.dremio.dev/problems/ai-agent/chat/model-provider-not-found" | "https://api.dremio.dev/problems/ai-agent/chat/no-default-model-provider" | "https://api.dremio.dev/problems/ai-agent/chat/no-model-provider-privilege" | "https://api.dremio.dev/problems/ai-agent/chat/preview-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/rate-limit-exceeded", undefined>>>;
+    /**
+     * Creates an Observable that emits each time the `AgentChatExchange` is updated
+     * with events from the Agent.
+     */
+    createAgentChatExchange$<T extends AgentResponseContent>(chatSession: ChatSession, userChatMessage: UserChatMessage, options?: {
+        mapChatEvents?: (v: AgentChatResponse<AgentResponseContent>) => Observable<AgentChatResponse<T>>;
+    }): Observable<AgentChatExchange>;
+    createChatSessionClient(chatSession: ChatSession): ChatSessionClient;
 }

package/dist/cloud/ai/AIResource.js

@@ -13,10 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { map } from "rxjs";
 import { ChatSession } from "../../enterprise/ai/chat/ChatSession.js";
-import { AgentChatResponse } from "../../enterprise/ai/chat/AgentChatResponse.js";
-import { fromTextEventStream } from "../../common/fromTextEventStream.js";
 import { ModelProvider } from "../../enterprise/ai/ModelProvider.js";
 import * as z from "zod/mini";
 import { dremioProvidedSchema, modelProviderCreateCodec, modelProviderRetrieveCodec, safeDecodeModelProvider, } from "./modelProviderCodec.js";
@@ -24,6 +21,11 @@ import { modelProviderRetrieveCodec as baseModelProviderRetrieveCodec } from "..
 import { DremioModelProvider } from "./DremioModelProvider.js";
 import { safeParseResultToResult } from "../../common/safeParseResultToResult.js";
 import { listAvailableModels } from "../../enterprise/ai/AIResource.js";
+import { createSendChatMessage } from "../../enterprise/ai/chat/methods/sendChatMessage.js";
+import { catchError, concatMap, endWith, map, of, scan, share, startWith, } from "rxjs";
+import { AgentEndTurn } from "../../enterprise/ai/chat/eventWrappers/AgentEndTurn.js";
+import { AgentChatExchange } from "../../enterprise/ai/chat/eventWrappers/AgentChatExchange.js";
+import { ChatSessionClient } from "../../enterprise/ai/chat/ChatSessionClient.js";
 export class AIResource {
     #config;
     constructor(config) {
@@ -87,26 +89,22 @@ export class AIResource {
      * 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) => AgentChatResponse.fromJSON(JSON.parse(event.data))));
+    sendChatMessage(chatSession, userChatMessage) {
+        return createSendChatMessage(this.#config)(chatSession, userChatMessage);
+    }
+    /**
+     * Creates an Observable that emits each time the `AgentChatExchange` is updated
+     * with events from the Agent.
+     */
+    createAgentChatExchange$(chatSession, userChatMessage, options = {}) {
+        const initialAgentChatExchange = new AgentChatExchange(userChatMessage, []);
+        return this.sendChatMessage(chatSession, userChatMessage).pipe(map((eventResult) => eventResult.unwrap()), concatMap((agentChatEvent) => options.mapChatEvents?.(agentChatEvent) || of(agentChatEvent)), endWith(new AgentEndTurn({ type: "reply_finished" })), catchError((error) => of(new AgentEndTurn({ error, type: "error" }))), scan((agentChatExchange, agentChatEvent) => agentChatExchange._addChatEvent(agentChatEvent), initialAgentChatExchange), startWith(initialAgentChatExchange), share());
+    }
+    createChatSessionClient(chatSession) {
+        return new ChatSessionClient({
+            chatSession,
+            createAgentChatExchange$: this.createAgentChatExchange$.bind(this),
+        });
     }
 }
 //# sourceMappingURL=AIResource.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"AIResource.js","sourceRoot":"","sources":["../../../src/cloud/ai/AIResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAC3B,OAAO,EAAE,WAAW,EAAE,MAAM,yCAAyC,CAAC;AAEtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,+CAA+C,CAAC;AAElF,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AACrE,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,oBAAoB,EACpB,wBAAwB,EACxB,0BAA0B,EAC1B,uBAAuB,GAExB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,0BAA0B,IAAI,8BAA8B,EAAE,MAAM,2CAA2C,CAAC;AACzH,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAGxE,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,OAAO,CACN,CAAC,MAAM,EAAE,EAAE,CACT,uBAAuB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAqC,CACpF,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;oBACb,IAAI,gBAAgB,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;wBAC1C,OAAO,IAAI,mBAAmB,CAC5B,IAAI,CAAC,OAAO,EACZ,UAAkD,CACnD,CAAC;oBACJ,CAAC;oBACD,OAAO,IAAI,aAAa,CACtB,IAAI,CAAC,OAAO,EACZ,UAA6D,CAC9D,CAAC;gBACJ,CAAC,CACF,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,GAAG,IAAwD;QAC7E,OAAO,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IACpD,CAAC;IAED,qBAAqB,CAAC,EAAU;QAC9B,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,EAAE,EAAE,EAAE;YACxC,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;aAC3B;SACF,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,uBAAuB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;IACxE,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,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/E,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 { map } from \"rxjs\";\nimport { ChatSession } from \"../../enterprise/ai/chat/ChatSession.ts\";\nimport type { UserChatMessage } from \"../../enterprise/ai/chat/UserChatMessage.ts\";\nimport { AgentChatResponse } from \"../../enterprise/ai/chat/AgentChatResponse.ts\";\nimport type { Logger, SonarV4Config, V4Config } from \"../../common/Config.ts\";\nimport { fromTextEventStream } from \"../../common/fromTextEventStream.ts\";\nimport { ModelProvider } from \"../../enterprise/ai/ModelProvider.ts\";\nimport * as z from \"zod/mini\";\nimport {\n  dremioProvidedSchema,\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec,\n  safeDecodeModelProvider,\n  type SafeDecodeResultWithoutDremioLLM,\n} from \"./modelProviderCodec.js\";\nimport { modelProviderRetrieveCodec as baseModelProviderRetrieveCodec } from \"../../enterprise/ai/modelProviderCodec.ts\";\nimport { DremioModelProvider } from \"./DremioModelProvider.ts\";\nimport { safeParseResultToResult } from \"../../common/safeParseResultToResult.ts\";\nimport { listAvailableModels } from \"../../enterprise/ai/AIResource.ts\";\nimport type { SignalParam } from \"../../common/Params.ts\";\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      .andThen(\n        (entity) =>\n          safeDecodeModelProvider(this.#config, entity) as SafeDecodeResultWithoutDremioLLM,\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) => {\n                if (\"dremioProvided\" in properties.config) {\n                  return new DremioModelProvider(\n                    this.#config,\n                    properties as z.infer<typeof dremioProvidedSchema>,\n                  );\n                }\n                return new ModelProvider(\n                  this.#config,\n                  properties as z.output<typeof baseModelProviderRetrieveCodec>,\n                );\n              },\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(...args: Parameters<ReturnType<typeof listAvailableModels>>) {\n    return listAvailableModels(this.#config)(...args);\n  }\n\n  retrieveModelProvider(id: string) {\n    return this.#config\n      .v4Request(`model-provider/config/${id}`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .andThen((entity) => safeDecodeModelProvider(this.#config, entity));\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(map((event) => AgentChatResponse.fromJSON(JSON.parse(event.data))));\n  }\n}\n"]}
+{"version":3,"file":"AIResource.js","sourceRoot":"","sources":["../../../src/cloud/ai/AIResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,yCAAyC,CAAC;AAGtE,OAAO,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AACrE,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,oBAAoB,EACpB,wBAAwB,EACxB,0BAA0B,EAC1B,uBAAuB,GAExB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,0BAA0B,IAAI,8BAA8B,EAAE,MAAM,2CAA2C,CAAC;AACzH,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAExE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qDAAqD,CAAC;AAK5F,OAAO,EACL,UAAU,EACV,SAAS,EACT,OAAO,EACP,GAAG,EACH,EAAE,EACF,IAAI,EACJ,KAAK,EACL,SAAS,GAEV,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,YAAY,EAAE,MAAM,wDAAwD,CAAC;AACtF,OAAO,EAAE,iBAAiB,EAAE,MAAM,6DAA6D,CAAC;AAChG,OAAO,EAAE,iBAAiB,EAAE,MAAM,+CAA+C,CAAC;AAElF,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,OAAO,CACN,CAAC,MAAM,EAAE,EAAE,CACT,uBAAuB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAqC,CACpF,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;oBACb,IAAI,gBAAgB,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;wBAC1C,OAAO,IAAI,mBAAmB,CAC5B,IAAI,CAAC,OAAO,EACZ,UAAkD,CACnD,CAAC;oBACJ,CAAC;oBACD,OAAO,IAAI,aAAa,CACtB,IAAI,CAAC,OAAO,EACZ,UAA6D,CAC9D,CAAC;gBACJ,CAAC,CACF,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,GAAG,IAAwD;QAC7E,OAAO,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IACpD,CAAC;IAED,qBAAqB,CAAC,EAAU;QAC9B,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,EAAE,EAAE,EAAE;YACxC,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;aAC3B;SACF,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,uBAAuB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;IACxE,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,WAAwB,EAAE,eAAgC;QACxE,OAAO,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAC3E,CAAC;IAED;;;OAGG;IACH,wBAAwB,CACtB,WAAwB,EACxB,eAAgC,EAChC,UAII,EAAE;QAEN,MAAM,wBAAwB,GAAG,IAAI,iBAAiB,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAE5E,OAAO,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC,IAAI,CAC5D,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAC1C,SAAS,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,cAAc,CAAC,CAAC,EAC5F,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;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 { ChatSession } from \"../../enterprise/ai/chat/ChatSession.ts\";\nimport type { UserChatMessage } from \"../../enterprise/ai/chat/UserChatMessage.ts\";\nimport type { Logger, SonarV4Config, V4Config } from \"../../common/Config.ts\";\nimport { ModelProvider } from \"../../enterprise/ai/ModelProvider.ts\";\nimport * as z from \"zod/mini\";\nimport {\n  dremioProvidedSchema,\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec,\n  safeDecodeModelProvider,\n  type SafeDecodeResultWithoutDremioLLM,\n} from \"./modelProviderCodec.js\";\nimport { modelProviderRetrieveCodec as baseModelProviderRetrieveCodec } from \"../../enterprise/ai/modelProviderCodec.ts\";\nimport { DremioModelProvider } from \"./DremioModelProvider.ts\";\nimport { safeParseResultToResult } from \"../../common/safeParseResultToResult.ts\";\nimport { listAvailableModels } from \"../../enterprise/ai/AIResource.ts\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { createSendChatMessage } from \"../../enterprise/ai/chat/methods/sendChatMessage.ts\";\nimport type {\n  AgentChatResponse,\n  AgentResponseContent,\n} from \"../../enterprise/ai/chat/AgentChatResponse.ts\";\nimport {\n  catchError,\n  concatMap,\n  endWith,\n  map,\n  of,\n  scan,\n  share,\n  startWith,\n  type Observable,\n} from \"rxjs\";\nimport { AgentEndTurn } from \"../../enterprise/ai/chat/eventWrappers/AgentEndTurn.ts\";\nimport { AgentChatExchange } from \"../../enterprise/ai/chat/eventWrappers/AgentChatExchange.ts\";\nimport { ChatSessionClient } from \"../../enterprise/ai/chat/ChatSessionClient.ts\";\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      .andThen(\n        (entity) =>\n          safeDecodeModelProvider(this.#config, entity) as SafeDecodeResultWithoutDremioLLM,\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) => {\n                if (\"dremioProvided\" in properties.config) {\n                  return new DremioModelProvider(\n                    this.#config,\n                    properties as z.infer<typeof dremioProvidedSchema>,\n                  );\n                }\n                return new ModelProvider(\n                  this.#config,\n                  properties as z.output<typeof baseModelProviderRetrieveCodec>,\n                );\n              },\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(...args: Parameters<ReturnType<typeof listAvailableModels>>) {\n    return listAvailableModels(this.#config)(...args);\n  }\n\n  retrieveModelProvider(id: string) {\n    return this.#config\n      .v4Request(`model-provider/config/${id}`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .andThen((entity) => safeDecodeModelProvider(this.#config, entity));\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, userChatMessage: UserChatMessage) {\n    return createSendChatMessage(this.#config)(chatSession, userChatMessage);\n  }\n\n  /**\n   * Creates an Observable that emits each time the `AgentChatExchange` is updated\n   * with events from the Agent.\n   */\n  createAgentChatExchange$<T extends AgentResponseContent>(\n    chatSession: ChatSession,\n    userChatMessage: UserChatMessage,\n    options: {\n      mapChatEvents?: (\n        v: AgentChatResponse<AgentResponseContent>,\n      ) => Observable<AgentChatResponse<T>>;\n    } = {},\n  ) {\n    const initialAgentChatExchange = new AgentChatExchange(userChatMessage, []);\n\n    return this.sendChatMessage(chatSession, userChatMessage).pipe(\n      map((eventResult) => eventResult.unwrap()),\n      concatMap((agentChatEvent) => options.mapChatEvents?.(agentChatEvent) || of(agentChatEvent)),\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"]}

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

@@ -8,7 +8,7 @@ export { modelProviderCreateCodec, modelProviderUpdateCodec };
 export declare const dremioProvidedSchema: z.ZodMiniObject<{
     config: z.ZodMiniObject<{
         dremioProvided: z.ZodMiniObject<Record<never, z.core.SomeType>, z.core.$strip>;
-    }, z.core.$strict>;
+    }, z.core.$strip>;
     defaultModelName: z.ZodMiniString<string>;
     id: z.ZodMiniString<string>;
     isDefault: z.ZodMiniBoolean<boolean>;
@@ -32,37 +32,42 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniUnion<r
             auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
                 accessKey: z.ZodMiniObject<{
                     accessKeyId: z.ZodMiniString<string>;
-                }, z.core.$strict>;
-            }, z.core.$strict>, z.ZodMiniObject<{
+                }, z.core.$strip>;
+            }, z.core.$strip>, z.ZodMiniObject<{
                 role: z.ZodMiniObject<{
                     roleArn: z.ZodMiniString<string>;
-                }, z.core.$strict>;
-            }, z.core.$strict>]>;
+                }, z.core.$strip>;
+            }, z.core.$strip>]>;
             region: z.ZodMiniString<string>;
-        }, z.core.$strict>;
+        }, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
-        anthropic: z.ZodMiniObject<{}, z.core.$strict>;
+        anthropic: z.ZodMiniObject<{}, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
         azureOpenAi: z.ZodMiniObject<{
             auth: z.ZodMiniObject<{
                 clientSecretAuth: z.ZodMiniObject<{
                     azureClientId: z.ZodMiniString<string>;
                     azureTenantId: z.ZodMiniString<string>;
-                }, z.core.$strict>;
-            }, z.core.$strict>;
+                }, z.core.$strip>;
+            }, z.core.$strip>;
             resourceName: z.ZodMiniString<string>;
-        }, z.core.$strict>;
+        }, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
-        googleGemini: z.ZodMiniObject<{}, z.core.$strict>;
+        googleGemini: z.ZodMiniObject<{}, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
         openAi: z.ZodMiniObject<{
             organizationId: z.ZodMiniString<string>;
-        }, z.core.$strict>;
+            region: z.ZodMiniEnum<{
+                GLOBAL: "GLOBAL";
+                US: "US";
+                EU: "EU";
+            }>;
+        }, z.core.$strip>;
     }, z.core.$strict>]>;
 }, z.core.$strip>, z.ZodMiniObject<{
     config: z.ZodMiniObject<{
         dremioProvided: z.ZodMiniObject<Record<never, z.core.SomeType>, z.core.$strip>;
-    }, z.core.$strict>;
+    }, z.core.$strip>;
     defaultModelName: z.ZodMiniString<string>;
     id: z.ZodMiniString<string>;
     isDefault: z.ZodMiniBoolean<boolean>;
@@ -80,32 +85,37 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniUnion<r
             auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
                 accessKey: z.ZodMiniObject<{
                     accessKeyId: z.ZodMiniString<string>;
-                }, z.core.$strict>;
-            }, z.core.$strict>, z.ZodMiniObject<{
+                }, z.core.$strip>;
+            }, z.core.$strip>, z.ZodMiniObject<{
                 role: z.ZodMiniObject<{
                     roleArn: z.ZodMiniString<string>;
-                }, z.core.$strict>;
-            }, z.core.$strict>]>;
+                }, z.core.$strip>;
+            }, z.core.$strip>]>;
             region: z.ZodMiniString<string>;
-        }, z.core.$strict>;
+        }, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
-        anthropic: z.ZodMiniObject<{}, z.core.$strict>;
+        anthropic: z.ZodMiniObject<{}, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
         azureOpenAi: z.ZodMiniObject<{
             auth: z.ZodMiniObject<{
                 clientSecretAuth: z.ZodMiniObject<{
                     azureClientId: z.ZodMiniString<string>;
                     azureTenantId: z.ZodMiniString<string>;
-                }, z.core.$strict>;
-            }, z.core.$strict>;
+                }, z.core.$strip>;
+            }, z.core.$strip>;
             resourceName: z.ZodMiniString<string>;
-        }, z.core.$strict>;
+        }, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
-        googleGemini: z.ZodMiniObject<{}, z.core.$strict>;
+        googleGemini: z.ZodMiniObject<{}, z.core.$strip>;
     }, z.core.$strict>, z.ZodMiniObject<{
         openAi: z.ZodMiniObject<{
             organizationId: z.ZodMiniString<string>;
-        }, z.core.$strict>;
+            region: z.ZodMiniEnum<{
+                GLOBAL: "GLOBAL";
+                US: "US";
+                EU: "EU";
+            }>;
+        }, z.core.$strip>;
     }, z.core.$strict>]>;
     defaultModelName: z.ZodMiniString<string>;
     isDefault: z.ZodMiniBoolean<boolean>;
@@ -115,7 +125,7 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniUnion<r
 }, z.core.$strip>, z.ZodMiniObject<{
     config: z.ZodMiniObject<{
         dremioProvided: z.ZodMiniObject<Record<never, z.core.SomeType>, z.core.$strip>;
-    }, z.core.$strict>;
+    }, z.core.$strip>;
     defaultModelName: z.ZodMiniString<string>;
     id: z.ZodMiniString<string>;
     isDefault: z.ZodMiniBoolean<boolean>;

package/dist/cloud/ai/modelProviderCodec.js

@@ -20,7 +20,7 @@ import { DremioModelProvider } from "./DremioModelProvider.js";
 import { ModelProvider } from "../interfaces.js";
 export { modelProviderCreateCodec, modelProviderUpdateCodec };
 export const dremioProvidedSchema = z.object({
-    config: z.strictObject({
+    config: z.object({
         dremioProvided: z.object(),
     }),
     defaultModelName: z.string(),

package/dist/cloud/ai/modelProviderCodec.js.map

@@ -1 +1 @@
-{"version":3,"file":"modelProviderCodec.js","sourceRoot":"","sources":["../../../src/cloud/ai/modelProviderCodec.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,IAAI,8BAA8B,EAC5D,wBAAwB,GACzB,MAAM,2CAA2C,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE/D,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,OAAO,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,CAAC;AAE9D,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,MAAM,EAAE,CAAC,CAAC,YAAY,CAAC;QACrB,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;KAC3B,CAAC;IACF,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE;IAC5B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE;IACtB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;CAChB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAC/C,CAAC,CAAC,KAAK,CAAC,CAAC,8BAA8B,CAAC,GAAG,CAAC,EAAE,EAAE,oBAAoB,CAAC,CAAC,EACtE,CAAC,CAAC,KAAK,CAAC,CAAC,8BAA8B,CAAC,GAAG,CAAC,GAAG,EAAE,oBAAoB,CAAC,CAAC,EACvE;IACE,MAAM,CAAC,CAAC;QACN,IAAI,gBAAgB,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACjC,OAAO,CAAyC,CAAC;QACnD,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CACb,8BAA8B,EAC9B,CAAmD,CACpD,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,IAAI,gBAAgB,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACjC,OAAO,CAAyC,CAAC;QACnD,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CACb,8BAA8B,EAC9B,CAAoD,CACrD,CAAC;IACJ,CAAC;CACF,CACF,CAAC;AAEF,MAAM,UAAU,uBAAuB,CACrC,MAAgB,EAChB,MAAkD;IAKlD,MAAM,YAAY,GAAG,CAAC,CAAC,UAAU,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC;IAEtE,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;QACzB,IAAI,gBAAgB,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjD,OAAO,EAAE,CACP,IAAI,mBAAmB,CAAC,MAAM,EAAE,YAAY,CAAC,IAA4C,CAAC,CAC3F,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,CACP,IAAI,aAAa,CACf,MAAM,EACN,YAAY,CAAC,IAAuD,CACrE,CACF,CAAC;IACJ,CAAC;IAED,OAAO,GAAG,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\nimport {\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec as baseModelProviderRetrieveCodec,\n  modelProviderUpdateCodec,\n} from \"../../enterprise/ai/modelProviderCodec.ts\";\nimport { Err, Ok, Result } from \"ts-results-es\";\nimport { DremioModelProvider } from \"./DremioModelProvider.ts\";\nimport type { V4Config } from \"../../common/Config.ts\";\nimport { ModelProvider } from \"../interfaces.ts\";\n\nexport { modelProviderCreateCodec, modelProviderUpdateCodec };\n\nexport const dremioProvidedSchema = z.object({\n  config: z.strictObject({\n    dremioProvided: z.object(),\n  }),\n  defaultModelName: z.string(),\n  id: z.string(),\n  isDefault: z.boolean(),\n  name: z.string(),\n  sqlModelNames: z.array(z.string()),\n  tag: z.string(),\n});\n\nexport const modelProviderRetrieveCodec = z.codec(\n  z.union([baseModelProviderRetrieveCodec.def.in, dremioProvidedSchema]),\n  z.union([baseModelProviderRetrieveCodec.def.out, dremioProvidedSchema]),\n  {\n    decode(v) {\n      if (\"dremioProvided\" in v.config) {\n        return v as z.infer<typeof dremioProvidedSchema>;\n      }\n\n      return z.decode(\n        baseModelProviderRetrieveCodec,\n        v as z.input<typeof baseModelProviderRetrieveCodec>,\n      );\n    },\n    encode(v) {\n      if (\"dremioProvided\" in v.config) {\n        return v as z.infer<typeof dremioProvidedSchema>;\n      }\n\n      return z.encode(\n        baseModelProviderRetrieveCodec,\n        v as z.output<typeof baseModelProviderRetrieveCodec>,\n      );\n    },\n  },\n);\n\nexport function safeDecodeModelProvider(\n  config: V4Config,\n  entity: z.input<typeof modelProviderRetrieveCodec>,\n): Result<\n  DremioModelProvider | ModelProvider,\n  z.core.util.SafeParseResult<z.output<typeof modelProviderRetrieveCodec>>[\"error\"]\n> {\n  const decodeResult = z.safeDecode(modelProviderRetrieveCodec, entity);\n\n  if (decodeResult.success) {\n    if (\"dremioProvided\" in decodeResult.data.config) {\n      return Ok(\n        new DremioModelProvider(config, decodeResult.data as z.infer<typeof dremioProvidedSchema>),\n      );\n    }\n    return Ok(\n      new ModelProvider(\n        config,\n        decodeResult.data as z.output<typeof baseModelProviderRetrieveCodec>,\n      ),\n    );\n  }\n\n  return Err(decodeResult.error);\n}\n\nexport type SafeDecodeResultWithoutDremioLLM = Result<\n  ModelProvider,\n  z.core.util.SafeParseResult<z.output<typeof modelProviderRetrieveCodec>>[\"error\"]\n>;\n"]}
+{"version":3,"file":"modelProviderCodec.js","sourceRoot":"","sources":["../../../src/cloud/ai/modelProviderCodec.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,IAAI,8BAA8B,EAC5D,wBAAwB,GACzB,MAAM,2CAA2C,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE/D,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,OAAO,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,CAAC;AAE9D,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;KAC3B,CAAC;IACF,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE;IAC5B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE;IACtB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;CAChB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAC/C,CAAC,CAAC,KAAK,CAAC,CAAC,8BAA8B,CAAC,GAAG,CAAC,EAAE,EAAE,oBAAoB,CAAC,CAAC,EACtE,CAAC,CAAC,KAAK,CAAC,CAAC,8BAA8B,CAAC,GAAG,CAAC,GAAG,EAAE,oBAAoB,CAAC,CAAC,EACvE;IACE,MAAM,CAAC,CAAC;QACN,IAAI,gBAAgB,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACjC,OAAO,CAAyC,CAAC;QACnD,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CACb,8BAA8B,EAC9B,CAAmD,CACpD,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,IAAI,gBAAgB,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACjC,OAAO,CAAyC,CAAC;QACnD,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CACb,8BAA8B,EAC9B,CAAoD,CACrD,CAAC;IACJ,CAAC;CACF,CACF,CAAC;AAEF,MAAM,UAAU,uBAAuB,CACrC,MAAgB,EAChB,MAAkD;IAKlD,MAAM,YAAY,GAAG,CAAC,CAAC,UAAU,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC;IAEtE,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;QACzB,IAAI,gBAAgB,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjD,OAAO,EAAE,CACP,IAAI,mBAAmB,CAAC,MAAM,EAAE,YAAY,CAAC,IAA4C,CAAC,CAC3F,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,CACP,IAAI,aAAa,CACf,MAAM,EACN,YAAY,CAAC,IAAuD,CACrE,CACF,CAAC;IACJ,CAAC;IAED,OAAO,GAAG,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\nimport {\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec as baseModelProviderRetrieveCodec,\n  modelProviderUpdateCodec,\n} from \"../../enterprise/ai/modelProviderCodec.ts\";\nimport { Err, Ok, Result } from \"ts-results-es\";\nimport { DremioModelProvider } from \"./DremioModelProvider.ts\";\nimport type { V4Config } from \"../../common/Config.ts\";\nimport { ModelProvider } from \"../interfaces.ts\";\n\nexport { modelProviderCreateCodec, modelProviderUpdateCodec };\n\nexport const dremioProvidedSchema = z.object({\n  config: z.object({\n    dremioProvided: z.object(),\n  }),\n  defaultModelName: z.string(),\n  id: z.string(),\n  isDefault: z.boolean(),\n  name: z.string(),\n  sqlModelNames: z.array(z.string()),\n  tag: z.string(),\n});\n\nexport const modelProviderRetrieveCodec = z.codec(\n  z.union([baseModelProviderRetrieveCodec.def.in, dremioProvidedSchema]),\n  z.union([baseModelProviderRetrieveCodec.def.out, dremioProvidedSchema]),\n  {\n    decode(v) {\n      if (\"dremioProvided\" in v.config) {\n        return v as z.infer<typeof dremioProvidedSchema>;\n      }\n\n      return z.decode(\n        baseModelProviderRetrieveCodec,\n        v as z.input<typeof baseModelProviderRetrieveCodec>,\n      );\n    },\n    encode(v) {\n      if (\"dremioProvided\" in v.config) {\n        return v as z.infer<typeof dremioProvidedSchema>;\n      }\n\n      return z.encode(\n        baseModelProviderRetrieveCodec,\n        v as z.output<typeof baseModelProviderRetrieveCodec>,\n      );\n    },\n  },\n);\n\nexport function safeDecodeModelProvider(\n  config: V4Config,\n  entity: z.input<typeof modelProviderRetrieveCodec>,\n): Result<\n  DremioModelProvider | ModelProvider,\n  z.core.util.SafeParseResult<z.output<typeof modelProviderRetrieveCodec>>[\"error\"]\n> {\n  const decodeResult = z.safeDecode(modelProviderRetrieveCodec, entity);\n\n  if (decodeResult.success) {\n    if (\"dremioProvided\" in decodeResult.data.config) {\n      return Ok(\n        new DremioModelProvider(config, decodeResult.data as z.infer<typeof dremioProvidedSchema>),\n      );\n    }\n    return Ok(\n      new ModelProvider(\n        config,\n        decodeResult.data as z.output<typeof baseModelProviderRetrieveCodec>,\n      ),\n    );\n  }\n\n  return Err(decodeResult.error);\n}\n\nexport type SafeDecodeResultWithoutDremioLLM = Result<\n  ModelProvider,\n  z.core.util.SafeParseResult<z.output<typeof modelProviderRetrieveCodec>>[\"error\"]\n>;\n"]}

package/dist/cloud/projects/Project.d.ts

@@ -4,6 +4,8 @@ import { projectUpdateSchema, type projectPropertiesCodec } from "./projectPrope
 import * as z from "zod/mini";
 import type { V3Config } from "../../common/Config.ts";
 import type { HttpError } from "../../common/HttpError.ts";
+import type { SignalParam } from "../../common/Params.ts";
+import { Grant } from "../../enterprise/grants/Grant.ts";
 export declare class Project implements ProjectInterface {
     #private;
     readonly catalog: ProjectInterface["catalog"];
@@ -26,6 +28,10 @@ export declare class Project implements ProjectInterface {
     activate(): Promise<Result<undefined, HttpError>>;
     delete(): Promise<Result<undefined, HttpError>>;
     update(properties: z.output<typeof projectUpdateSchema>): AsyncResult<Project, HttpError | z.core.util.SafeParseResult<z.output<typeof projectUpdateSchema>>["error"]>;
+    grants({ signal }?: SignalParam): AsyncResult<{
+        readonly availablePrivileges: Set<"DELETE" | "ALTER" | "INSERT" | "SELECT" | "TRUNCATE" | "UPDATE" | "DROP" | "MODIFY" | "MANAGE_GRANTS" | "ALTER_REFLECTION" | "CREATE_SOURCE" | "CREATE_TABLE" | "EXECUTE" | "EXTERNAL_QUERY" | "MONITOR" | "OPERATE" | "READ_METADATA" | "UPLOAD_FILE" | "USAGE" | "VIEW_JOB_HISTORY" | "VIEW_REFLECTION">;
+        readonly grants: Grant<Set<"DELETE" | "ALTER" | "INSERT" | "SELECT" | "TRUNCATE" | "UPDATE" | "DROP" | "MODIFY" | "MANAGE_GRANTS" | "ALTER_REFLECTION" | "CREATE_SOURCE" | "CREATE_TABLE" | "EXECUTE" | "EXTERNAL_QUERY" | "MONITOR" | "OPERATE" | "READ_METADATA" | "UPLOAD_FILE" | "USAGE" | "VIEW_JOB_HISTORY" | "VIEW_REFLECTION">, "PROJECT">[];
+    }, HttpError>;
     get settled(): boolean;
     static updateSchema: z.ZodMiniObject<{
         credentials: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniDiscriminatedUnion<[z.ZodMiniObject<{

package/dist/cloud/projects/Project.js

@@ -16,6 +16,8 @@
 import { AsyncResult, Err } from "ts-results-es";
 import { projectUpdateSchema } from "./projectPropertiesCodec.js";
 import * as z from "zod/mini";
+import { Grant } from "../../enterprise/grants/Grant.js";
+import { RoleGrantee, UserGrantee } from "../../enterprise/Grantee.js";
 export class Project {
     #config;
     catalog;
@@ -79,9 +81,71 @@ export class Project {
             .map((res) => res.json())
             .map((entity) => this.#projectsResource._createFromEntity(entity));
     }
+    grants({ signal } = {}) {
+        return this.#config
+            .v3Request(`projects/${this.id}/grants`, {
+            headers: {
+                Accept: "application/json",
+            },
+            signal,
+        })
+            .map((res) => res.json())
+            .map((v) => {
+            return {
+                availablePrivileges: new Set(v.availablePrivileges),
+                grants: v.grants.map((grant) => {
+                    const grantee = grant.granteeType === "USER"
+                        ? new UserGrantee(grant.granteeId)
+                        : new RoleGrantee(grant.granteeId);
+                    return new Grant({
+                        grantee,
+                        privileges: new Set(grant.privileges),
+                        tag: grant.tag,
+                        type: "PROJECT",
+                    });
+                }),
+            };
+        });
+    }
     get settled() {
         return this.state === "ACTIVE" || this.state === "ARCHIVED" || this.state === "INACTIVE";
     }
     static updateSchema = projectUpdateSchema;
 }
+const projectAvailablePrivileges = z.enum([
+    "ALTER",
+    "ALTER_REFLECTION",
+    "CREATE_SOURCE",
+    "CREATE_TABLE",
+    "DELETE",
+    "DROP",
+    "EXECUTE",
+    "EXTERNAL_QUERY",
+    "INSERT",
+    "MANAGE_GRANTS",
+    "MODIFY",
+    "MONITOR",
+    "OPERATE",
+    "READ_METADATA",
+    "SELECT",
+    "TRUNCATE",
+    "UPDATE",
+    "UPLOAD_FILE",
+    "USAGE",
+    "VIEW_JOB_HISTORY",
+    "VIEW_REFLECTION",
+]);
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+const projectGrantsInSchema = z.object({
+    availablePrivileges: z.array(projectAvailablePrivileges),
+    grants: z.array(z.object({
+        granteeId: z.string(),
+        granteeType: z.enum(["USER", "ROLE"]),
+        grantType: z.literal("PROJECT"),
+        name: z.string(),
+        privileges: z.array(projectAvailablePrivileges),
+        tag: z.string(),
+    })),
+    totalResults: z.int().check(z.minimum(0)),
+});
 //# sourceMappingURL=Project.js.map

package/dist/cloud/projects/Project.js.map

@@ -1 +1 @@
-{"version":3,"file":"Project.js","sourceRoot":"","sources":["../../../src/cloud/projects/Project.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,WAAW,EAAE,GAAG,EAAe,MAAM,eAAe,CAAC;AAE9D,OAAO,EAAE,mBAAmB,EAA+B,MAAM,6BAA6B,CAAC;AAC/F,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAI9B,MAAM,OAAO,OAAO;IACT,OAAO,CAAW;IAClB,OAAO,CAA8B;IACrC,OAAO,CAA8B;IACrC,SAAS,CAAgC;IACzC,SAAS,CAAgC;IACzC,SAAS,CAAgC;IACzC,WAAW,CAAkC;IAC7C,EAAE,CAAyB;IAC3B,cAAc,CAAqC;IACnD,UAAU,CAAiC;IAC3C,UAAU,CAAiC;IAC3C,IAAI,CAA2B;IAC/B,eAAe,CAAsC;IACrD,cAAc,CAAqC;IACnD,YAAY,CAAmC;IAC/C,KAAK,CAA4B;IACjC,IAAI,CAA2B;IAExC,iBAAiB,CAAmB;IAEpC,YAAY,MAAgB,EAAE,UAA6B,EAAE,gBAAkC;QAC7F,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAC1C,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,cAAc,CAAC;QAChD,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,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;QAClD,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,cAAc,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC;QAC5C,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,iBAAiB,GAAG,gBAAgB,CAAC;IAC5C,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CACJ,UAAgD;QAKhD,MAAM,gBAAgB,GAAG,CAAC,CAAC,UAAU,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;QACvE,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;YAC9B,OAAO,IAAI,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvE,CAAC;QACD,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,YAAY,IAAI,CAAC,EAAE,EAAE,EAAE;YAChC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC;YAC3C,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,EAAqD,CAAC;aAC3E,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,KAAK,UAAU,IAAI,IAAI,CAAC,KAAK,KAAK,UAAU,CAAC;IAC3F,CAAC;IAED,MAAM,CAAC,YAAY,GAAG,mBAAmB,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 { AsyncResult, Err, type Result } from \"ts-results-es\";\nimport type { ProjectsResource } from \"./ProjectsResource.ts\";\nimport { projectUpdateSchema, type projectPropertiesCodec } from \"./projectPropertiesCodec.ts\";\nimport * as z from \"zod/mini\";\nimport type { V3Config } from \"../../common/Config.ts\";\nimport type { HttpError } from \"../../common/HttpError.ts\";\n\nexport class Project implements ProjectInterface {\n  readonly #config: V3Config;\n  readonly catalog: ProjectInterface[\"catalog\"];\n  readonly cloudId: ProjectInterface[\"cloudId\"];\n  readonly cloudType: ProjectInterface[\"cloudType\"];\n  readonly createdAt: ProjectInterface[\"createdAt\"];\n  readonly createdBy: ProjectInterface[\"createdBy\"];\n  readonly credentials: ProjectInterface[\"credentials\"];\n  readonly id: ProjectInterface[\"id\"];\n  readonly lastStateError: ProjectInterface[\"lastStateError\"];\n  readonly modifiedAt: ProjectInterface[\"modifiedAt\"];\n  readonly modifiedBy: ProjectInterface[\"modifiedBy\"];\n  readonly name: ProjectInterface[\"name\"];\n  readonly numberOfEngines: ProjectInterface[\"numberOfEngines\"];\n  readonly primaryCatalog: ProjectInterface[\"primaryCatalog\"];\n  readonly projectStore: ProjectInterface[\"projectStore\"];\n  readonly state: ProjectInterface[\"state\"];\n  readonly type: ProjectInterface[\"type\"];\n\n  #projectsResource: ProjectsResource;\n\n  constructor(config: V3Config, properties: ProjectProperties, projectsResource: ProjectsResource) {\n    this.#config = config;\n    this.catalog = properties.catalog;\n    this.cloudId = properties.cloudId;\n    this.cloudType = properties.cloudType;\n    this.createdAt = properties.createdAt;\n    this.createdBy = properties.createdBy;\n    this.credentials = properties.credentials;\n    this.id = properties.id;\n    this.lastStateError = properties.lastStateError;\n    this.modifiedAt = properties.modifiedAt;\n    this.modifiedBy = properties.modifiedBy;\n    this.name = properties.name;\n    this.numberOfEngines = properties.numberOfEngines;\n    this.primaryCatalog = properties.primaryCatalog;\n    this.projectStore = properties.projectStore;\n    this.state = properties.state;\n    this.type = properties.type;\n    this.#projectsResource = projectsResource;\n  }\n\n  activate() {\n    return this.#projectsResource.activate(this.id);\n  }\n\n  delete() {\n    return this.#projectsResource.delete(this.id);\n  }\n\n  update(\n    properties: z.output<typeof projectUpdateSchema>,\n  ): AsyncResult<\n    Project,\n    HttpError | z.core.util.SafeParseResult<z.output<typeof projectUpdateSchema>>[\"error\"]\n  > {\n    const parsedProperties = z.safeEncode(projectUpdateSchema, properties);\n    if (!parsedProperties.success) {\n      return new AsyncResult(Promise.resolve(Err(parsedProperties.error)));\n    }\n    return this.#config\n      .v3Request(`projects/${this.id}`, {\n        body: JSON.stringify(parsedProperties.data),\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 projectPropertiesCodec>>)\n      .map((entity) => this.#projectsResource._createFromEntity(entity));\n  }\n\n  get settled() {\n    return this.state === \"ACTIVE\" || this.state === \"ARCHIVED\" || this.state === \"INACTIVE\";\n  }\n\n  static updateSchema = projectUpdateSchema;\n}\n\nexport type ProjectProperties = z.output<typeof projectPropertiesCodec>;\n\nexport type ProjectMethods = {\n  delete(): Promise<Result<undefined, unknown>>;\n  get settled(): boolean;\n};\n\nexport type ProjectInterface = ProjectProperties & ProjectMethods;\n"]}
+{"version":3,"file":"Project.js","sourceRoot":"","sources":["../../../src/cloud/projects/Project.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,WAAW,EAAE,GAAG,EAAe,MAAM,eAAe,CAAC;AAE9D,OAAO,EAAE,mBAAmB,EAA+B,MAAM,6BAA6B,CAAC;AAC/F,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAI9B,OAAO,EAAE,KAAK,EAAE,MAAM,kCAAkC,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAEvE,MAAM,OAAO,OAAO;IACT,OAAO,CAAW;IAClB,OAAO,CAA8B;IACrC,OAAO,CAA8B;IACrC,SAAS,CAAgC;IACzC,SAAS,CAAgC;IACzC,SAAS,CAAgC;IACzC,WAAW,CAAkC;IAC7C,EAAE,CAAyB;IAC3B,cAAc,CAAqC;IACnD,UAAU,CAAiC;IAC3C,UAAU,CAAiC;IAC3C,IAAI,CAA2B;IAC/B,eAAe,CAAsC;IACrD,cAAc,CAAqC;IACnD,YAAY,CAAmC;IAC/C,KAAK,CAA4B;IACjC,IAAI,CAA2B;IAExC,iBAAiB,CAAmB;IAEpC,YAAY,MAAgB,EAAE,UAA6B,EAAE,gBAAkC;QAC7F,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAC1C,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,cAAc,CAAC;QAChD,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,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;QAClD,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,cAAc,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC;QAC5C,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,iBAAiB,GAAG,gBAAgB,CAAC;IAC5C,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CACJ,UAAgD;QAKhD,MAAM,gBAAgB,GAAG,CAAC,CAAC,UAAU,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;QACvE,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;YAC9B,OAAO,IAAI,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvE,CAAC;QACD,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,YAAY,IAAI,CAAC,EAAE,EAAE,EAAE;YAChC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC;YAC3C,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,EAAqD,CAAC;aAC3E,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,CAAC,EAAE,MAAM,KAAkB,EAAE;QACjC,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,YAAY,IAAI,CAAC,EAAE,SAAS,EAAE;YACvC,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;aAC3B;YACD,MAAM;SACP,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAoD,CAAC;aAC1E,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,OAAO;gBACL,mBAAmB,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,mBAAmB,CAAC;gBACnD,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oBAC7B,MAAM,OAAO,GACX,KAAK,CAAC,WAAW,KAAK,MAAM;wBAC1B,CAAC,CAAC,IAAI,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC;wBAClC,CAAC,CAAC,IAAI,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;oBAEvC,OAAO,IAAI,KAAK,CAAC;wBACf,OAAO;wBACP,UAAU,EAAE,IAAI,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC;wBACrC,GAAG,EAAE,KAAK,CAAC,GAAG;wBACd,IAAI,EAAE,SAAS;qBAChB,CAAC,CAAC;gBACL,CAAC,CAAC;aACM,CAAC;QACb,CAAC,CAAC,CAAC;IACP,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,KAAK,UAAU,IAAI,IAAI,CAAC,KAAK,KAAK,UAAU,CAAC;IAC3F,CAAC;IAED,MAAM,CAAC,YAAY,GAAG,mBAAmB,CAAC;;AAY5C,MAAM,0BAA0B,GAAG,CAAC,CAAC,IAAI,CAAC;IACxC,OAAO;IACP,kBAAkB;IAClB,eAAe;IACf,cAAc;IACd,QAAQ;IACR,MAAM;IACN,SAAS;IACT,gBAAgB;IAChB,QAAQ;IACR,eAAe;IACf,QAAQ;IACR,SAAS;IACT,SAAS;IACT,eAAe;IACf,QAAQ;IACR,UAAU;IACV,QAAQ;IACR,aAAa;IACb,OAAO;IACP,kBAAkB;IAClB,iBAAiB;CAClB,CAAC,CAAC;AAEH,6DAA6D;AAC7D,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,mBAAmB,EAAE,CAAC,CAAC,KAAK,CAAC,0BAA0B,CAAC;IACxD,MAAM,EAAE,CAAC,CAAC,KAAK,CACb,CAAC,CAAC,MAAM,CAAC;QACP,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;QAC/B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,0BAA0B,CAAC;QAC/C,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;KAChB,CAAC,CACH;IACD,YAAY,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;CAC1C,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AsyncResult, Err, type Result } from \"ts-results-es\";\nimport type { ProjectsResource } from \"./ProjectsResource.ts\";\nimport { projectUpdateSchema, type projectPropertiesCodec } from \"./projectPropertiesCodec.ts\";\nimport * as z from \"zod/mini\";\nimport type { V3Config } from \"../../common/Config.ts\";\nimport type { HttpError } from \"../../common/HttpError.ts\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { Grant } from \"../../enterprise/grants/Grant.ts\";\nimport { RoleGrantee, UserGrantee } from \"../../enterprise/Grantee.ts\";\n\nexport class Project implements ProjectInterface {\n  readonly #config: V3Config;\n  readonly catalog: ProjectInterface[\"catalog\"];\n  readonly cloudId: ProjectInterface[\"cloudId\"];\n  readonly cloudType: ProjectInterface[\"cloudType\"];\n  readonly createdAt: ProjectInterface[\"createdAt\"];\n  readonly createdBy: ProjectInterface[\"createdBy\"];\n  readonly credentials: ProjectInterface[\"credentials\"];\n  readonly id: ProjectInterface[\"id\"];\n  readonly lastStateError: ProjectInterface[\"lastStateError\"];\n  readonly modifiedAt: ProjectInterface[\"modifiedAt\"];\n  readonly modifiedBy: ProjectInterface[\"modifiedBy\"];\n  readonly name: ProjectInterface[\"name\"];\n  readonly numberOfEngines: ProjectInterface[\"numberOfEngines\"];\n  readonly primaryCatalog: ProjectInterface[\"primaryCatalog\"];\n  readonly projectStore: ProjectInterface[\"projectStore\"];\n  readonly state: ProjectInterface[\"state\"];\n  readonly type: ProjectInterface[\"type\"];\n\n  #projectsResource: ProjectsResource;\n\n  constructor(config: V3Config, properties: ProjectProperties, projectsResource: ProjectsResource) {\n    this.#config = config;\n    this.catalog = properties.catalog;\n    this.cloudId = properties.cloudId;\n    this.cloudType = properties.cloudType;\n    this.createdAt = properties.createdAt;\n    this.createdBy = properties.createdBy;\n    this.credentials = properties.credentials;\n    this.id = properties.id;\n    this.lastStateError = properties.lastStateError;\n    this.modifiedAt = properties.modifiedAt;\n    this.modifiedBy = properties.modifiedBy;\n    this.name = properties.name;\n    this.numberOfEngines = properties.numberOfEngines;\n    this.primaryCatalog = properties.primaryCatalog;\n    this.projectStore = properties.projectStore;\n    this.state = properties.state;\n    this.type = properties.type;\n    this.#projectsResource = projectsResource;\n  }\n\n  activate() {\n    return this.#projectsResource.activate(this.id);\n  }\n\n  delete() {\n    return this.#projectsResource.delete(this.id);\n  }\n\n  update(\n    properties: z.output<typeof projectUpdateSchema>,\n  ): AsyncResult<\n    Project,\n    HttpError | z.core.util.SafeParseResult<z.output<typeof projectUpdateSchema>>[\"error\"]\n  > {\n    const parsedProperties = z.safeEncode(projectUpdateSchema, properties);\n    if (!parsedProperties.success) {\n      return new AsyncResult(Promise.resolve(Err(parsedProperties.error)));\n    }\n    return this.#config\n      .v3Request(`projects/${this.id}`, {\n        body: JSON.stringify(parsedProperties.data),\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 projectPropertiesCodec>>)\n      .map((entity) => this.#projectsResource._createFromEntity(entity));\n  }\n\n  grants({ signal }: SignalParam = {}) {\n    return this.#config\n      .v3Request(`projects/${this.id}/grants`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n        signal,\n      })\n      .map((res) => res.json() as Promise<z.infer<typeof projectGrantsInSchema>>)\n      .map((v) => {\n        return {\n          availablePrivileges: new Set(v.availablePrivileges),\n          grants: v.grants.map((grant) => {\n            const grantee =\n              grant.granteeType === \"USER\"\n                ? new UserGrantee(grant.granteeId)\n                : new RoleGrantee(grant.granteeId);\n\n            return new Grant({\n              grantee,\n              privileges: new Set(grant.privileges),\n              tag: grant.tag,\n              type: \"PROJECT\",\n            });\n          }),\n        } as const;\n      });\n  }\n\n  get settled() {\n    return this.state === \"ACTIVE\" || this.state === \"ARCHIVED\" || this.state === \"INACTIVE\";\n  }\n\n  static updateSchema = projectUpdateSchema;\n}\n\nexport type ProjectProperties = z.output<typeof projectPropertiesCodec>;\n\nexport type ProjectMethods = {\n  delete(): Promise<Result<undefined, unknown>>;\n  get settled(): boolean;\n};\n\nexport type ProjectInterface = ProjectProperties & ProjectMethods;\n\nconst projectAvailablePrivileges = z.enum([\n  \"ALTER\",\n  \"ALTER_REFLECTION\",\n  \"CREATE_SOURCE\",\n  \"CREATE_TABLE\",\n  \"DELETE\",\n  \"DROP\",\n  \"EXECUTE\",\n  \"EXTERNAL_QUERY\",\n  \"INSERT\",\n  \"MANAGE_GRANTS\",\n  \"MODIFY\",\n  \"MONITOR\",\n  \"OPERATE\",\n  \"READ_METADATA\",\n  \"SELECT\",\n  \"TRUNCATE\",\n  \"UPDATE\",\n  \"UPLOAD_FILE\",\n  \"USAGE\",\n  \"VIEW_JOB_HISTORY\",\n  \"VIEW_REFLECTION\",\n]);\n\n// eslint-disable-next-line @typescript-eslint/no-unused-vars\nconst projectGrantsInSchema = z.object({\n  availablePrivileges: z.array(projectAvailablePrivileges),\n  grants: z.array(\n    z.object({\n      granteeId: z.string(),\n      granteeType: z.enum([\"USER\", \"ROLE\"]),\n      grantType: z.literal(\"PROJECT\"),\n      name: z.string(),\n      privileges: z.array(projectAvailablePrivileges),\n      tag: z.string(),\n    }),\n  ),\n  totalResults: z.int().check(z.minimum(0)),\n});\n"]}

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

@@ -8,18 +8,15 @@ import * as z from "zod/mini";
 import { modelProviderCreateCodec } from "./modelProviderCodec.js";
 import type { SignalParam } from "../../common/Params.ts";
 import { listModelsPropertiesSchema } from "./modelProviderConfigSchemas.js";
-import { Err, Ok } from "ts-results-es";
-import type { HttpError } from "../../common/HttpError.ts";
-import { Problem } from "../../common/Problem.ts";
 import { AgentChatExchange } from "./chat/eventWrappers/AgentChatExchange.ts";
 import { ChatSessionClient } from "./chat/ChatSessionClient.ts";
-export declare function listAvailableModels(config: V4Config): (properties: z.infer<typeof listModelsPropertiesSchema>, { signal }?: SignalParam) => import("ts-results-es").AsyncResult<Set<string>, HttpError>;
+export declare function listAvailableModels(config: V4Config): (properties: z.infer<typeof listModelsPropertiesSchema>, { signal }?: SignalParam) => import("ts-results-es").AsyncResult<Set<string>, import("../index.ts").HttpError>;
 export declare class AIResource {
     #private;
     constructor(config: Logger & V4Config & SonarV4Config);
-    createModelProvider(properties: z.output<typeof modelProviderCreateCodec>): import("ts-results-es").AsyncResult<ModelProvider, HttpError>;
+    createModelProvider(properties: z.output<typeof modelProviderCreateCodec>): import("ts-results-es").AsyncResult<ModelProvider, import("../index.ts").HttpError>;
     listModelProviders(): {
-        data({ signal }?: SignalParam): AsyncGenerator<Ok<ModelProvider> | Err<z.core.$ZodError<{
+        data({ signal }?: SignalParam): AsyncGenerator<import("ts-results-es").Ok<ModelProvider> | import("ts-results-es").Err<z.core.$ZodError<{
             name: string;
             id: string;
             tag: string;
@@ -55,6 +52,7 @@ export declare class AIResource {
             } | {
                 openAi: {
                     organizationId: string;
+                    region: "GLOBAL" | "US" | "EU";
                 };
             };
             defaultModelName: string;
@@ -64,7 +62,7 @@ export declare class AIResource {
             sqlModelNames: Set<string>;
         }>>, void, unknown>;
         getPage: ({ signal }?: SignalParam) => import("ts-results-es").AsyncResult<{
-            data: (Ok<ModelProvider> | Err<z.core.$ZodError<{
+            data: (import("ts-results-es").Ok<ModelProvider> | import("ts-results-es").Err<z.core.$ZodError<{
                 name: string;
                 id: string;
                 tag: string;
@@ -100,6 +98,7 @@ export declare class AIResource {
                 } | {
                     openAi: {
                         organizationId: string;
+                        region: "GLOBAL" | "US" | "EU";
                     };
                 };
                 defaultModelName: string;
@@ -108,10 +107,10 @@ export declare class AIResource {
                 modifiedAt: import("temporal-polyfill").Temporal.Instant;
                 sqlModelNames: Set<string>;
             }>>)[];
-        }, HttpError>;
+        }, import("../index.ts").HttpError>;
     };
-    listAvailableModels(properties: z.infer<typeof listModelsPropertiesSchema>): import("ts-results-es").AsyncResult<Set<string>, HttpError>;
-    retrieveModelProvider(id: string, { signal }?: SignalParam): import("ts-results-es").AsyncResult<ModelProvider, HttpError>;
+    listAvailableModels(properties: z.infer<typeof listModelsPropertiesSchema>): import("ts-results-es").AsyncResult<Set<string>, import("../index.ts").HttpError>;
+    retrieveModelProvider(id: string, { signal }?: SignalParam): import("ts-results-es").AsyncResult<ModelProvider, import("../index.ts").HttpError>;
     /**
      * Creates an Observable that emits each time the `AgentChatExchange` is updated
      * with events from the Agent.
@@ -124,7 +123,5 @@ export declare class AIResource {
      * Sends a chat message to the Dremio AI Agent and returns an Observable containing
      * Agent replies.
      */
-    sendChatMessage(chatSession: ChatSession): (chatMessage: UserChatMessage) => Observable<Ok<AgentChatResponse<AgentResponseContent>> | Err<HttpError | Problem<SendChatMessageProblemCodes, undefined>>>;
+    sendChatMessage(chatSession: ChatSession, userChatMessage: UserChatMessage): Observable<import("ts-results-es").Ok<AgentChatResponse<AgentResponseContent>> | import("ts-results-es").Err<import("../index.ts").HttpError | import("../index.ts").Problem<"https://api.dremio.dev/problems/ai-agent/chat/agent-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/input-too-large" | "https://api.dremio.dev/problems/ai-agent/chat/model-provider-not-found" | "https://api.dremio.dev/problems/ai-agent/chat/no-default-model-provider" | "https://api.dremio.dev/problems/ai-agent/chat/no-model-provider-privilege" | "https://api.dremio.dev/problems/ai-agent/chat/preview-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/rate-limit-exceeded", undefined>>>;
 }
-type SendChatMessageProblemCodes = "https://api.dremio.dev/problems/ai-agent/chat/agent-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/input-too-large" | "https://api.dremio.dev/problems/ai-agent/chat/model-provider-not-found" | "https://api.dremio.dev/problems/ai-agent/chat/no-default-model-provider" | "https://api.dremio.dev/problems/ai-agent/chat/no-model-provider-privilege" | "https://api.dremio.dev/problems/ai-agent/chat/preview-disabled" | "https://api.dremio.dev/problems/ai-agent/chat/rate-limit-exceeded";
-export {};