@dremio/js-sdk 0.42.0 → 0.43.0

package/dist/cloud/ai/AIResource.js

@@ -22,10 +22,11 @@ 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";
+import { listConversations } from "../../enterprise/ai/conversations/methods/listConversations.js";
+import { createConversation, } from "../../enterprise/ai/conversations/methods/createConversation.js";
+import { retrieveConversation } from "../../enterprise/ai/conversations/methods/retrieveConversation.js";
+import { retrieveConversationHistory } from "../../enterprise/ai/conversations/methods/retrieveConversationHistory.js";
+import { createConversationMachine } from "../../enterprise/ai/conversations/createConversationMachine.js";
 export class AIResource {
     #config;
     constructor(config) {
@@ -46,6 +47,39 @@ export class AIResource {
             .map((res) => res.json())
             .andThen((entity) => safeDecodeModelProvider(this.#config, entity));
     }
+    get conversationMachine() {
+        return createConversationMachine(this.#config);
+    }
+    createConversation(body) {
+        return createConversation(this.#config)(body);
+    }
+    listConversations() {
+        const getPage = listConversations(this.#config);
+        return {
+            async *data({ signal } = {}) {
+                yield* (await getPage({ maxResults: 100 }, { signal }).map((response) => response.data).promise).unwrap();
+            },
+            getPage,
+        };
+    }
+    retrieveConversation(id, { signal } = {}) {
+        return retrieveConversation(this.#config)(id, { signal });
+    }
+    retrieveConversationHistory(id) {
+        // eslint-disable-next-line @typescript-eslint/no-empty-object-type
+        const getPage = (params, { signal } = {}) => retrieveConversationHistory(this.#config)(id, { signal });
+        return {
+            async *data() {
+                const history = await getPage({}).then((result) => {
+                    if (result.isErr()) {
+                        throw result.error;
+                    }
+                    return result.value;
+                });
+                yield* history.data;
+            },
+        };
+    }
     listModelProviders() {
         const getPage = ({ signal } = {}) => {
             return this.#config
@@ -88,23 +122,11 @@ export class AIResource {
     /**
      * Sends a chat message to the Dremio AI Agent and returns an Observable containing
      * Agent replies.
+     *
+     * @deprecated
      */
     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,WAAW,EAAE,MAAM,yCAAyC,CAAC;AAGtE,OAAO,EAAE,aAAa,EAAE,MAAM,oDAAoD,CAAC;AACnF,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,yDAAyD,CAAC;AACvI,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/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/modelProvider/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/eventWrappers/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"]}
+{"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,oDAAoD,CAAC;AACnF,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,yDAAyD,CAAC;AACvI,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;AAC5F,OAAO,EAAE,iBAAiB,EAAE,MAAM,gEAAgE,CAAC;AACnG,OAAO,EACL,kBAAkB,GAEnB,MAAM,iEAAiE,CAAC;AACzE,OAAO,EAAE,oBAAoB,EAAE,MAAM,mEAAmE,CAAC;AACzG,OAAO,EAAE,2BAA2B,EAAE,MAAM,0EAA0E,CAAC;AAGvH,OAAO,EAAE,yBAAyB,EAAE,MAAM,gEAAgE,CAAC;AAE3G,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,IAAI,mBAAmB;QACrB,OAAO,yBAAyB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,kBAAkB,CAAC,IAA6C;QAC9D,OAAO,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC;IAChD,CAAC;IAED,iBAAiB;QACf,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,OAAO;YACL,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,KAAkB,EAAE;gBACtC,KAAK,CAAC,CAAC,CACL,MAAM,OAAO,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,CACxF,CAAC,MAAM,EAAE,CAAC;YACb,CAAC;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,oBAAoB,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;QAC3D,OAAO,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,2BAA2B,CAAC,EAAU;QACpC,mEAAmE;QACnE,MAAM,OAAO,GAAG,CAAC,MAAU,EAAE,EAAE,MAAM,KAAkB,EAAE,EAAE,EAAE,CAC3D,2BAA2B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5D,OAAO;YACL,KAAK,CAAC,CAAC,IAAI;gBACT,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;oBAChD,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;wBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;oBACrB,CAAC;oBACD,OAAO,MAAM,CAAC,KAAK,CAAC;gBACtB,CAAC,CAAC,CAAC;gBACH,KAAK,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;YACtB,CAAC;SACF,CAAC;IACJ,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;;;;;OAKG;IACH,eAAe,CAAC,WAAwB,EAAE,eAAgC;QACxE,OAAO,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAC3E,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/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/modelProvider/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 { listConversations } from \"../../enterprise/ai/conversations/methods/listConversations.ts\";\nimport {\n  createConversation,\n  type createConversationCodec,\n} from \"../../enterprise/ai/conversations/methods/createConversation.ts\";\nimport { retrieveConversation } from \"../../enterprise/ai/conversations/methods/retrieveConversation.ts\";\nimport { retrieveConversationHistory } from \"../../enterprise/ai/conversations/methods/retrieveConversationHistory.ts\";\nimport type { ChatEvent } from \"../../enterprise/ai/chat/chatEventSchema.ts\";\nimport type { HttpError } from \"../../common/HttpError.ts\";\nimport { createConversationMachine } from \"../../enterprise/ai/conversations/createConversationMachine.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  get conversationMachine() {\n    return createConversationMachine(this.#config);\n  }\n\n  createConversation(body: z.input<typeof createConversationCodec>) {\n    return createConversation(this.#config)(body);\n  }\n\n  listConversations() {\n    const getPage = listConversations(this.#config);\n    return {\n      async *data({ signal }: SignalParam = {}) {\n        yield* (\n          await getPage({ maxResults: 100 }, { signal }).map((response) => response.data).promise\n        ).unwrap();\n      },\n      getPage,\n    };\n  }\n\n  retrieveConversation(id: string, { signal }: SignalParam = {}) {\n    return retrieveConversation(this.#config)(id, { signal });\n  }\n\n  retrieveConversationHistory(id: string) {\n    // eslint-disable-next-line @typescript-eslint/no-empty-object-type\n    const getPage = (params: {}, { signal }: SignalParam = {}) =>\n      retrieveConversationHistory(this.#config)(id, { signal });\n    return {\n      async *data(): AsyncGenerator<ChatEvent, void, HttpError> {\n        const history = await getPage({}).then((result) => {\n          if (result.isErr()) {\n            throw result.error;\n          }\n          return result.value;\n        });\n        yield* history.data;\n      },\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   * @deprecated\n   */\n  sendChatMessage(chatSession: ChatSession, userChatMessage: UserChatMessage) {\n    return createSendChatMessage(this.#config)(chatSession, userChatMessage);\n  }\n}\n"]}

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

@@ -7,7 +7,7 @@ export declare class DremioModelProvider implements DremioModelProviderPropertie
     readonly id: DremioProviderConstructorProperties["id"];
     readonly isDefault: DremioProviderConstructorProperties["isDefault"];
     constructor(config: V4Config, properties: DremioProviderConstructorProperties);
-    setDefault(): import("ts-results-es").AsyncResult<DremioModelProvider, import("../index.ts").HttpError>;
+    setDefault(): import("ts-results-es").AsyncResult<void, import("../index.ts").HttpError>;
 }
 type DremioModelProviderProperties = Pick<z.infer<typeof dremioProvidedSchema>, "id" | "isDefault">;
 export {};

package/dist/cloud/ai/DremioModelProvider.js

@@ -18,8 +18,11 @@ import { dremioProvidedSchema } from "./modelProviderCodec.js";
 export class DremioModelProvider {
     id;
     isDefault;
+    // eslint-disable-next-line no-unused-private-class-members
     #defaultModelName;
+    // eslint-disable-next-line no-unused-private-class-members
     #sqlModelNames;
+    // eslint-disable-next-line no-unused-private-class-members
     #tag;
     #config;
     constructor(config, properties) {
@@ -32,27 +35,17 @@ export class DremioModelProvider {
     }
     setDefault() {
         return this.#config
-            .v4Request(`model-provider/config/${this.id}`, {
+            .v4Request(`model-provider:setDefault`, {
             body: JSON.stringify({
-                config: {
-                    dremioProvided: {},
-                },
-                defaultModelName: this.#defaultModelName,
                 id: this.id,
-                isDefault: true,
-                name: "Dremio",
-                sqlModelNames: this.#sqlModelNames,
-                tag: this.#tag,
             }),
             headers: {
-                Accept: "application/json",
                 "Content-Type": "application/json",
             },
             keepalive: true,
-            method: "PUT",
+            method: "POST",
         })
-            .map((res) => res.json())
-            .map((entity) => new DremioModelProvider(this.#config, z.parse(dremioProvidedSchema, entity)));
+            .map(() => undefined);
     }
 }
 //# sourceMappingURL=DremioModelProvider.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"DremioModelProvider.js","sourceRoot":"","sources":["../../../src/cloud/ai/DremioModelProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAG/D,MAAM,OAAO,mBAAmB;IACrB,EAAE,CAA4C;IAC9C,SAAS,CAAmD;IAC5D,iBAAiB,CAA0D;IAC3E,cAAc,CAAuD;IACrE,IAAI,CAA6C;IAE1D,OAAO,CAAW;IAElB,YAAY,MAAgB,EAAE,UAA+C;QAC3E,IAAI,CAAC,iBAAiB,GAAG,UAAU,CAAC,gBAAgB,CAAC;QACrD,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QAEtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,aAAa,CAAC;QAC/C,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC;IAC7B,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,IAAI,CAAC,EAAE,EAAE,EAAE;YAC7C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,MAAM,EAAE;oBACN,cAAc,EAAE,EAAE;iBACnB;gBACD,gBAAgB,EAAE,IAAI,CAAC,iBAAiB;gBACxC,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,SAAS,EAAE,IAAI;gBACf,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,IAAI,CAAC,cAAc;gBAClC,GAAG,EAAE,IAAI,CAAC,IAAI;aACf,CAAC;YACF,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,EAAmD,CAAC;aACzE,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,oBAAoB,EAAE,MAAM,CAAC,CAAC,CACzF,CAAC;IACN,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\nimport type { V4Config } from \"../../common/Config.ts\";\nimport { dremioProvidedSchema } from \"./modelProviderCodec.js\";\n\ntype DremioProviderConstructorProperties = z.infer<typeof dremioProvidedSchema>;\nexport class DremioModelProvider implements DremioModelProviderProperties {\n  readonly id: DremioProviderConstructorProperties[\"id\"];\n  readonly isDefault: DremioProviderConstructorProperties[\"isDefault\"];\n  readonly #defaultModelName: DremioProviderConstructorProperties[\"defaultModelName\"];\n  readonly #sqlModelNames: DremioProviderConstructorProperties[\"sqlModelNames\"];\n  readonly #tag: DremioProviderConstructorProperties[\"tag\"];\n\n  #config: V4Config;\n\n  constructor(config: V4Config, properties: DremioProviderConstructorProperties) {\n    this.#defaultModelName = properties.defaultModelName;\n    this.id = properties.id;\n    this.isDefault = properties.isDefault;\n\n    this.#config = config;\n    this.#sqlModelNames = properties.sqlModelNames;\n    this.#tag = properties.tag;\n  }\n\n  setDefault() {\n    return this.#config\n      .v4Request(`model-provider/config/${this.id}`, {\n        body: JSON.stringify({\n          config: {\n            dremioProvided: {},\n          },\n          defaultModelName: this.#defaultModelName,\n          id: this.id,\n          isDefault: true,\n          name: \"Dremio\",\n          sqlModelNames: this.#sqlModelNames,\n          tag: this.#tag,\n        }),\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.infer<typeof dremioProvidedSchema>>)\n      .map(\n        (entity) => new DremioModelProvider(this.#config, z.parse(dremioProvidedSchema, entity)),\n      );\n  }\n}\n\ntype DremioModelProviderProperties = Pick<z.infer<typeof dremioProvidedSchema>, \"id\" | \"isDefault\">;\n"]}
+{"version":3,"file":"DremioModelProvider.js","sourceRoot":"","sources":["../../../src/cloud/ai/DremioModelProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAG/D,MAAM,OAAO,mBAAmB;IACrB,EAAE,CAA4C;IAC9C,SAAS,CAAmD;IACrE,2DAA2D;IAClD,iBAAiB,CAA0D;IACpF,2DAA2D;IAClD,cAAc,CAAuD;IAC9E,2DAA2D;IAClD,IAAI,CAA6C;IAE1D,OAAO,CAAW;IAElB,YAAY,MAAgB,EAAE,UAA+C;QAC3E,IAAI,CAAC,iBAAiB,GAAG,UAAU,CAAC,gBAAgB,CAAC;QACrD,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QAEtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,aAAa,CAAC;QAC/C,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC;IAC7B,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,2BAA2B,EAAE;YACtC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,EAAE,EAAE,IAAI,CAAC,EAAE;aACZ,CAAC;YACF,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC;aACD,GAAG,CAAC,GAAG,EAAE,CAAC,SAAiB,CAAC,CAAC;IAClC,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\nimport type { V4Config } from \"../../common/Config.ts\";\nimport { dremioProvidedSchema } from \"./modelProviderCodec.js\";\n\ntype DremioProviderConstructorProperties = z.infer<typeof dremioProvidedSchema>;\nexport class DremioModelProvider implements DremioModelProviderProperties {\n  readonly id: DremioProviderConstructorProperties[\"id\"];\n  readonly isDefault: DremioProviderConstructorProperties[\"isDefault\"];\n  // eslint-disable-next-line no-unused-private-class-members\n  readonly #defaultModelName: DremioProviderConstructorProperties[\"defaultModelName\"];\n  // eslint-disable-next-line no-unused-private-class-members\n  readonly #sqlModelNames: DremioProviderConstructorProperties[\"sqlModelNames\"];\n  // eslint-disable-next-line no-unused-private-class-members\n  readonly #tag: DremioProviderConstructorProperties[\"tag\"];\n\n  #config: V4Config;\n\n  constructor(config: V4Config, properties: DremioProviderConstructorProperties) {\n    this.#defaultModelName = properties.defaultModelName;\n    this.id = properties.id;\n    this.isDefault = properties.isDefault;\n\n    this.#config = config;\n    this.#sqlModelNames = properties.sqlModelNames;\n    this.#tag = properties.tag;\n  }\n\n  setDefault() {\n    return this.#config\n      .v4Request(`model-provider:setDefault`, {\n        body: JSON.stringify({\n          id: this.id,\n        }),\n        headers: {\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map(() => undefined as void);\n  }\n}\n\ntype DremioModelProviderProperties = Pick<z.infer<typeof dremioProvidedSchema>, \"id\" | \"isDefault\">;\n"]}

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

@@ -63,6 +63,12 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniUnion<r
                 EU: "EU";
             }>;
         }, z.core.$strip>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAiCompatibleApi: z.ZodMiniObject<{
+            completionsApiUrl: z.ZodMiniString<string>;
+            modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strip>;
     }, z.core.$strict>]>;
 }, z.core.$strip>, z.ZodMiniObject<{
     config: z.ZodMiniObject<{
@@ -116,6 +122,12 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniUnion<r
                 EU: "EU";
             }>;
         }, z.core.$strip>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAiCompatibleApi: z.ZodMiniObject<{
+            completionsApiUrl: z.ZodMiniString<string>;
+            modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strip>;
     }, z.core.$strict>]>;
     defaultModelName: z.ZodMiniString<string>;
     isDefault: z.ZodMiniBoolean<boolean>;

package/dist/cloud/replaceOriginResource.js

@@ -16,6 +16,7 @@
 const prodLikePattern = /^([^.]+)(\.(?:(?:eu\.)?dremio\.cloud|[^.]+\.dremio\.site))$/;
 const privateLinkPattern = /^([^.]+\.)([^.]+)((\.privatelink\.dremio\.cloud)|(\.privatelink\.([^.]+)\.dremio\.site))$/;
 const daasEnginePattern = /^(dcs-(?:aws|cloud)-\d+-ns\.)([^.]+)(\.drem\.io)$/;
+const awsInternalPattern = /^([^.]+)(\.([^.]+\.aws\.dremio\.site))$/;
 /**
  * @hidden
  * @internal
@@ -23,6 +24,9 @@ const daasEnginePattern = /^(dcs-(?:aws|cloud)-\d+-ns\.)([^.]+)(\.drem\.io)$/;
 export function _replaceOriginResource(origin, resource) {
     const url = new URL(origin);
     const hostname = url.hostname;
+    if (awsInternalPattern.test(hostname)) {
+        url.hostname = hostname.replace(awsInternalPattern, `${resource}$2`);
+    }
     if (privateLinkPattern.test(hostname)) {
         url.hostname = hostname.replace(privateLinkPattern, `$1${resource}$3`);
         return url.origin;

package/dist/cloud/replaceOriginResource.js.map

@@ -1 +1 @@
-{"version":3,"file":"replaceOriginResource.js","sourceRoot":"","sources":["../../src/cloud/replaceOriginResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,MAAM,eAAe,GAAG,6DAA6D,CAAC;AACtF,MAAM,kBAAkB,GACtB,2FAA2F,CAAC;AAC9F,MAAM,iBAAiB,GAAG,mDAAmD,CAAC;AAE9E;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAwB,EAAE,QAAgB;IAC/E,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;IAC5B,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;IAE9B,IAAI,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,kBAAkB,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC;QACvE,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,IAAI,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC;QACtE,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,IAAI,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,QAAQ,IAAI,CAAC,CAAC;QAClE,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,OAAO,GAAG,CAAC,MAAM,CAAC;AACpB,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 { Config } from \"../common/Config.ts\";\n\nconst prodLikePattern = /^([^.]+)(\\.(?:(?:eu\\.)?dremio\\.cloud|[^.]+\\.dremio\\.site))$/;\nconst privateLinkPattern =\n  /^([^.]+\\.)([^.]+)((\\.privatelink\\.dremio\\.cloud)|(\\.privatelink\\.([^.]+)\\.dremio\\.site))$/;\nconst daasEnginePattern = /^(dcs-(?:aws|cloud)-\\d+-ns\\.)([^.]+)(\\.drem\\.io)$/;\n\n/**\n * @hidden\n * @internal\n */\nexport function _replaceOriginResource(origin: Config[\"origin\"], resource: string): string {\n  const url = new URL(origin);\n  const hostname = url.hostname;\n\n  if (privateLinkPattern.test(hostname)) {\n    url.hostname = hostname.replace(privateLinkPattern, `$1${resource}$3`);\n    return url.origin;\n  }\n\n  if (daasEnginePattern.test(hostname)) {\n    url.hostname = hostname.replace(daasEnginePattern, `$1${resource}$3`);\n    return url.origin;\n  }\n\n  if (prodLikePattern.test(hostname)) {\n    url.hostname = hostname.replace(prodLikePattern, `${resource}$2`);\n    return url.origin;\n  }\n\n  return url.origin;\n}\n"]}
+{"version":3,"file":"replaceOriginResource.js","sourceRoot":"","sources":["../../src/cloud/replaceOriginResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,MAAM,eAAe,GAAG,6DAA6D,CAAC;AACtF,MAAM,kBAAkB,GACtB,2FAA2F,CAAC;AAC9F,MAAM,iBAAiB,GAAG,mDAAmD,CAAC;AAC9E,MAAM,kBAAkB,GAAG,yCAAyC,CAAC;AAErE;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAwB,EAAE,QAAgB;IAC/E,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;IAC5B,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;IAE9B,IAAI,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,kBAAkB,EAAE,GAAG,QAAQ,IAAI,CAAC,CAAC;IACvE,CAAC;IAED,IAAI,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,kBAAkB,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC;QACvE,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,IAAI,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC;QACtE,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,IAAI,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,QAAQ,IAAI,CAAC,CAAC;QAClE,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,OAAO,GAAG,CAAC,MAAM,CAAC;AACpB,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 { Config } from \"../common/Config.ts\";\n\nconst prodLikePattern = /^([^.]+)(\\.(?:(?:eu\\.)?dremio\\.cloud|[^.]+\\.dremio\\.site))$/;\nconst privateLinkPattern =\n  /^([^.]+\\.)([^.]+)((\\.privatelink\\.dremio\\.cloud)|(\\.privatelink\\.([^.]+)\\.dremio\\.site))$/;\nconst daasEnginePattern = /^(dcs-(?:aws|cloud)-\\d+-ns\\.)([^.]+)(\\.drem\\.io)$/;\nconst awsInternalPattern = /^([^.]+)(\\.([^.]+\\.aws\\.dremio\\.site))$/;\n\n/**\n * @hidden\n * @internal\n */\nexport function _replaceOriginResource(origin: Config[\"origin\"], resource: string): string {\n  const url = new URL(origin);\n  const hostname = url.hostname;\n\n  if (awsInternalPattern.test(hostname)) {\n    url.hostname = hostname.replace(awsInternalPattern, `${resource}$2`);\n  }\n\n  if (privateLinkPattern.test(hostname)) {\n    url.hostname = hostname.replace(privateLinkPattern, `$1${resource}$3`);\n    return url.origin;\n  }\n\n  if (daasEnginePattern.test(hostname)) {\n    url.hostname = hostname.replace(daasEnginePattern, `$1${resource}$3`);\n    return url.origin;\n  }\n\n  if (prodLikePattern.test(hostname)) {\n    url.hostname = hostname.replace(prodLikePattern, `${resource}$2`);\n    return url.origin;\n  }\n\n  return url.origin;\n}\n"]}

package/dist/common/ConflictResolver.d.ts

@@ -1 +1,9 @@
-export type ConflictResolver<T, C> = (local: T, remote: T, changes: C) => T | null | PromiseLike<T | null>;
+type ConflictResolverReturn<C> = C | null | PromiseLike<C | null>;
+export type ConflictResolver<T, C> = (local: T, remote: T, changes: C) => ConflictResolverReturn<C>;
+declare function force<C>(_local: unknown, _remote: unknown, changes: C): ConflictResolverReturn<C>;
+declare function yieldFn(_local: unknown, _remote: unknown, _changes: unknown): null;
+export declare const ConflictResolutionStrategy: {
+    Force: typeof force;
+    Yield: typeof yieldFn;
+};
+export {};

package/dist/common/ConflictResolver.js

@@ -13,5 +13,14 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-export {};
+function force(_local, _remote, changes) {
+    return changes;
+}
+function yieldFn(_local, _remote, _changes) {
+    return null;
+}
+export const ConflictResolutionStrategy = {
+    Force: force,
+    Yield: yieldFn,
+};
 //# sourceMappingURL=ConflictResolver.js.map

package/dist/common/ConflictResolver.js.map

@@ -1 +1 @@
-{"version":3,"file":"ConflictResolver.js","sourceRoot":"","sources":["../../src/common/ConflictResolver.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport type ConflictResolver<T, C> = (\n  local: T,\n  remote: T,\n  changes: C,\n) => T | null | PromiseLike<T | null>;\n"]}
+{"version":3,"file":"ConflictResolver.js","sourceRoot":"","sources":["../../src/common/ConflictResolver.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAMH,SAAS,KAAK,CAAI,MAAe,EAAE,OAAgB,EAAE,OAAU;IAC7D,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,OAAO,CAAC,MAAe,EAAE,OAAgB,EAAE,QAAiB;IACnE,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,MAAM,0BAA0B,GAAG;IACxC,KAAK,EAAE,KAAK;IACZ,KAAK,EAAE,OAAO;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\ntype ConflictResolverReturn<C> = C | null | PromiseLike<C | null>;\n\nexport type ConflictResolver<T, C> = (local: T, remote: T, changes: C) => ConflictResolverReturn<C>;\n\nfunction force<C>(_local: unknown, _remote: unknown, changes: C): ConflictResolverReturn<C> {\n  return changes;\n}\n\nfunction yieldFn(_local: unknown, _remote: unknown, _changes: unknown) {\n  return null;\n}\n\nexport const ConflictResolutionStrategy = {\n  Force: force,\n  Yield: yieldFn,\n};\n"]}

package/dist/common/refineRfc9557Schema.d.ts

@@ -0,0 +1 @@
+export declare function refineRfc9557Schema(val: string): boolean;

package/dist/common/refineRfc9557Schema.js

@@ -0,0 +1,26 @@
+/*
+ * 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 { Temporal } from "temporal-polyfill";
+export function refineRfc9557Schema(val) {
+    try {
+        Temporal.Instant.from(val);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=refineRfc9557Schema.js.map

package/dist/common/refineRfc9557Schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"refineRfc9557Schema.js","sourceRoot":"","sources":["../../src/common/refineRfc9557Schema.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,MAAM,UAAU,mBAAmB,CAAC,GAAW;IAC7C,IAAI,CAAC;QACH,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,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 { Temporal } from \"temporal-polyfill\";\n\nexport function refineRfc9557Schema(val: string) {\n  try {\n    Temporal.Instant.from(val);\n    return true;\n  } catch {\n    return false;\n  }\n}\n"]}

package/dist/common/sharedExports.d.ts

@@ -4,3 +4,4 @@ export * from "./CredentialProvider.ts";
 export * from "./fromTextEventStream.ts";
 export * from "./Problem.ts";
 export * from "./Query.ts";
+export * from "./ConflictResolver.ts";

package/dist/common/sharedExports.js

@@ -19,4 +19,5 @@ export * from "./CredentialProvider.js";
 export * from "./fromTextEventStream.js";
 export * from "./Problem.js";
 export * from "./Query.js";
+export * from "./ConflictResolver.js";
 //# sourceMappingURL=sharedExports.js.map

package/dist/common/sharedExports.js.map

@@ -1 +1 @@
-{"version":3,"file":"sharedExports.js","sourceRoot":"","sources":["../../src/common/sharedExports.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,SAAS,EAAE,CAAC;AACrB,cAAc,yBAAyB,CAAC;AACxC,cAAc,0BAA0B,CAAC;AACzC,cAAc,cAAc,CAAC;AAC7B,cAAc,YAAY,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 { HttpError } from \"../common/HttpError.ts\";\nexport { HttpError };\nexport * from \"./CredentialProvider.ts\";\nexport * from \"./fromTextEventStream.ts\";\nexport * from \"./Problem.ts\";\nexport * from \"./Query.ts\";\n"]}
+{"version":3,"file":"sharedExports.js","sourceRoot":"","sources":["../../src/common/sharedExports.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,SAAS,EAAE,CAAC;AACrB,cAAc,yBAAyB,CAAC;AACxC,cAAc,0BAA0B,CAAC;AACzC,cAAc,cAAc,CAAC;AAC7B,cAAc,YAAY,CAAC;AAC3B,cAAc,uBAAuB,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 { HttpError } from \"../common/HttpError.ts\";\nexport { HttpError };\nexport * from \"./CredentialProvider.ts\";\nexport * from \"./fromTextEventStream.ts\";\nexport * from \"./Problem.ts\";\nexport * from \"./Query.ts\";\nexport * from \"./ConflictResolver.ts\";\n"]}

package/dist/enterprise/Dremio.d.ts

@@ -10,7 +10,6 @@ import { GrantsResource } from "./grants/GrantsResource.js";
 import { AIResource } from "./ai/AIResource.js";
 export declare class Dremio {
     #private;
-    readonly ai: AIResource;
     readonly catalog: EnterpriseCatalogResource;
     readonly engines: EnginesResource;
     readonly jobs: JobsResource;
@@ -23,6 +22,7 @@ export declare class Dremio {
     readonly sql: SqlResource;
     readonly users: EnterpriseUsersResource;
     constructor(config: Config);
+    ai: (_unused?: string) => AIResource;
     environment(): import("ts-results-es").AsyncResult<{
         clusterType: string;
         buildTime: import("temporal-polyfill").Temporal.Instant;

package/dist/enterprise/Dremio.js

@@ -25,7 +25,7 @@ import { GrantsResource } from "./grants/GrantsResource.js";
 import { AIResource } from "./ai/AIResource.js";
 import { retrieveEnvironment } from "../oss/retrieveEnvironment.js";
 export class Dremio {
-    ai;
+    #ai;
     catalog;
     engines;
     jobs;
@@ -40,7 +40,7 @@ export class Dremio {
     #resourceConfig;
     constructor(config) {
         this.#resourceConfig = getResourceConfig(config);
-        this.ai = new AIResource(this.#resourceConfig);
+        this.#ai = new AIResource(this.#resourceConfig);
         this.catalog = new EnterpriseCatalogResource(this.#resourceConfig);
         this.engines = new EnginesResource(this.#resourceConfig);
         this.jobs = new JobsResource(this.#resourceConfig);
@@ -50,6 +50,7 @@ export class Dremio {
         this.sql = new SqlResource(this.#resourceConfig);
         this.users = new EnterpriseUsersResource(this.#resourceConfig);
     }
+    ai = (_unused) => this.#ai;
     environment() {
         return retrieveEnvironment(this.#resourceConfig);
     }

package/dist/enterprise/Dremio.js.map

@@ -1 +1 @@
-{"version":3,"file":"Dremio.js","sourceRoot":"","sources":["../../src/enterprise/Dremio.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,yBAAyB,EAAE,MAAM,wCAAwC,CAAC;AACnF,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,yBAAyB,EAAE,MAAM,wCAAwC,CAAC;AACnF,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAC7E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AAEpE,MAAM,OAAO,MAAM;IACR,EAAE,CAAa;IACf,OAAO,CAA4B;IACnC,OAAO,CAAkB;IACzB,IAAI,CAAe;IAE5B;;OAEG;IACM,OAAO,CAAiB;IACxB,KAAK,CAAgB;IACrB,OAAO,CAA4B;IACnC,GAAG,CAAc;IACjB,KAAK,CAA0B;IAC/B,eAAe,CAAuC;IAE/D,YAAY,MAAc;QACxB,IAAI,CAAC,eAAe,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACjD,IAAI,CAAC,EAAE,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnE,IAAI,CAAC,OAAO,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACzD,IAAI,CAAC,IAAI,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnD,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACxD,IAAI,CAAC,KAAK,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnE,IAAI,CAAC,GAAG,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK,GAAG,IAAI,uBAAuB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACjE,CAAC;IAED,WAAW;QACT,OAAO,mBAAmB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { Config } from \"../common/Config.ts\";\nimport { EnterpriseCatalogResource } from \"./catalog/EnterpriseCatalogResource.js\";\nimport { JobsResource } from \"../oss/jobs/JobsResource.js\";\nimport { EnterpriseScriptsResource } from \"./scripts/EnterpriseScriptsResource.js\";\nimport { SqlResource } from \"../oss/sql/SqlResource.js\";\nimport { EnterpriseUsersResource } from \"./users/EnterpriseUsersResource.js\";\nimport { EnginesResource } from \"./engines/EnginesResource.js\";\nimport { RolesResource } from \"./roles/RolesResource.js\";\nimport { getResourceConfig } from \"../oss/getResourceConfig.js\";\nimport { GrantsResource } from \"./grants/GrantsResource.js\";\nimport { AIResource } from \"./ai/AIResource.js\";\nimport { retrieveEnvironment } from \"../oss/retrieveEnvironment.ts\";\n\nexport class Dremio {\n  readonly ai: AIResource;\n  readonly catalog: EnterpriseCatalogResource;\n  readonly engines: EnginesResource;\n  readonly jobs: JobsResource;\n\n  /**\n   * @experimental\n   */\n  readonly _grants: GrantsResource;\n  readonly roles: RolesResource;\n  readonly scripts: EnterpriseScriptsResource;\n  readonly sql: SqlResource;\n  readonly users: EnterpriseUsersResource;\n  readonly #resourceConfig: ReturnType<typeof getResourceConfig>;\n\n  constructor(config: Config) {\n    this.#resourceConfig = getResourceConfig(config);\n    this.ai = new AIResource(this.#resourceConfig);\n    this.catalog = new EnterpriseCatalogResource(this.#resourceConfig);\n    this.engines = new EnginesResource(this.#resourceConfig);\n    this.jobs = new JobsResource(this.#resourceConfig);\n    this._grants = new GrantsResource(this.#resourceConfig);\n    this.roles = new RolesResource(this.#resourceConfig);\n    this.scripts = new EnterpriseScriptsResource(this.#resourceConfig);\n    this.sql = new SqlResource(this.#resourceConfig);\n    this.users = new EnterpriseUsersResource(this.#resourceConfig);\n  }\n\n  environment() {\n    return retrieveEnvironment(this.#resourceConfig);\n  }\n\n  get _resourceConfig() {\n    return this.#resourceConfig;\n  }\n}\n"]}
+{"version":3,"file":"Dremio.js","sourceRoot":"","sources":["../../src/enterprise/Dremio.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,yBAAyB,EAAE,MAAM,wCAAwC,CAAC;AACnF,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,yBAAyB,EAAE,MAAM,wCAAwC,CAAC;AACnF,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAC7E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AAEpE,MAAM,OAAO,MAAM;IACR,GAAG,CAAa;IAChB,OAAO,CAA4B;IACnC,OAAO,CAAkB;IACzB,IAAI,CAAe;IAE5B;;OAEG;IACM,OAAO,CAAiB;IACxB,KAAK,CAAgB;IACrB,OAAO,CAA4B;IACnC,GAAG,CAAc;IACjB,KAAK,CAA0B;IAC/B,eAAe,CAAuC;IAE/D,YAAY,MAAc;QACxB,IAAI,CAAC,eAAe,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACjD,IAAI,CAAC,GAAG,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnE,IAAI,CAAC,OAAO,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACzD,IAAI,CAAC,IAAI,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnD,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACxD,IAAI,CAAC,KAAK,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnE,IAAI,CAAC,GAAG,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK,GAAG,IAAI,uBAAuB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACjE,CAAC;IAED,EAAE,GAAG,CAAC,OAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC;IAEpC,WAAW;QACT,OAAO,mBAAmB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { Config } from \"../common/Config.ts\";\nimport { EnterpriseCatalogResource } from \"./catalog/EnterpriseCatalogResource.js\";\nimport { JobsResource } from \"../oss/jobs/JobsResource.js\";\nimport { EnterpriseScriptsResource } from \"./scripts/EnterpriseScriptsResource.js\";\nimport { SqlResource } from \"../oss/sql/SqlResource.js\";\nimport { EnterpriseUsersResource } from \"./users/EnterpriseUsersResource.js\";\nimport { EnginesResource } from \"./engines/EnginesResource.js\";\nimport { RolesResource } from \"./roles/RolesResource.js\";\nimport { getResourceConfig } from \"../oss/getResourceConfig.js\";\nimport { GrantsResource } from \"./grants/GrantsResource.js\";\nimport { AIResource } from \"./ai/AIResource.js\";\nimport { retrieveEnvironment } from \"../oss/retrieveEnvironment.ts\";\n\nexport class Dremio {\n  readonly #ai: AIResource;\n  readonly catalog: EnterpriseCatalogResource;\n  readonly engines: EnginesResource;\n  readonly jobs: JobsResource;\n\n  /**\n   * @experimental\n   */\n  readonly _grants: GrantsResource;\n  readonly roles: RolesResource;\n  readonly scripts: EnterpriseScriptsResource;\n  readonly sql: SqlResource;\n  readonly users: EnterpriseUsersResource;\n  readonly #resourceConfig: ReturnType<typeof getResourceConfig>;\n\n  constructor(config: Config) {\n    this.#resourceConfig = getResourceConfig(config);\n    this.#ai = new AIResource(this.#resourceConfig);\n    this.catalog = new EnterpriseCatalogResource(this.#resourceConfig);\n    this.engines = new EnginesResource(this.#resourceConfig);\n    this.jobs = new JobsResource(this.#resourceConfig);\n    this._grants = new GrantsResource(this.#resourceConfig);\n    this.roles = new RolesResource(this.#resourceConfig);\n    this.scripts = new EnterpriseScriptsResource(this.#resourceConfig);\n    this.sql = new SqlResource(this.#resourceConfig);\n    this.users = new EnterpriseUsersResource(this.#resourceConfig);\n  }\n\n  ai = (_unused?: string) => this.#ai;\n\n  environment() {\n    return retrieveEnvironment(this.#resourceConfig);\n  }\n\n  get _resourceConfig() {\n    return this.#resourceConfig;\n  }\n}\n"]}