@dremio/js-sdk 0.42.0 → 0.43.0

package/dist/enterprise/ai/conversations/methods/listConversations.js

@@ -0,0 +1,30 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import * as z from "zod/mini";
+import { AgentConversation } from "../AgentConversation.js";
+import { conversationPropertiesCodec, } from "../conversationPropertiesCodec.js";
+export const listConversations = (config) => function listConversations(params, { signal } = {}) {
+    return config
+        .sonarV4Request(`agent/conversations?${new URLSearchParams({ maxResults: String(params.maxResults) }).toString()}`, { signal })
+        .map((res) => res.json())
+        .map((response) => {
+        return {
+            ...response,
+            data: response.data.map((entity) => new AgentConversation(config, z.decode(conversationPropertiesCodec, entity))),
+        };
+    });
+};
+//# sourceMappingURL=listConversations.js.map

package/dist/enterprise/ai/conversations/methods/listConversations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"listConversations.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/listConversations.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAG9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EACL,2BAA2B,GAE5B,MAAM,mCAAmC,CAAC;AAE3C,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,MAAqB,EAAE,EAAE,CACzD,SAAS,iBAAiB,CAAC,MAA8B,EAAE,EAAE,MAAM,KAAkB,EAAE;IACrF,OAAO,MAAM;SACV,cAAc,CACb,uBAAuB,IAAI,eAAe,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,EAAE,EAClG,EAAE,MAAM,EAAE,CACX;SACA,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA6C,CAAC;SACnE,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QAChB,OAAO;YACL,GAAG,QAAQ;YACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CACrB,CAAC,MAAM,EAAE,EAAE,CACT,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC,CAC/E;SACF,CAAC;IACJ,CAAC,CAAC,CAAC;AACP,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 * as z from \"zod/mini\";\nimport type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport type { SignalParam } from \"../../../../common/Params.ts\";\nimport { AgentConversation } from \"../AgentConversation.ts\";\nimport {\n  conversationPropertiesCodec,\n  type ConversationEntity,\n} from \"../conversationPropertiesCodec.ts\";\n\nexport const listConversations = (config: SonarV4Config) =>\n  function listConversations(params: { maxResults: number }, { signal }: SignalParam = {}) {\n    return config\n      .sonarV4Request(\n        `agent/conversations?${new URLSearchParams({ maxResults: String(params.maxResults) }).toString()}`,\n        { signal },\n      )\n      .map((res) => res.json() as Promise<{ data: ConversationEntity[] }>)\n      .map((response) => {\n        return {\n          ...response,\n          data: response.data.map(\n            (entity) =>\n              new AgentConversation(config, z.decode(conversationPropertiesCodec, entity)),\n          ),\n        };\n      });\n  };\n"]}

package/dist/enterprise/ai/conversations/methods/retrieveConversation.d.ts

@@ -0,0 +1,12 @@
+import type { SonarV4Config } from "../../../../common/Config.ts";
+import type { SignalParam } from "../../../../common/Params.ts";
+export declare const retrieveConversation: (config: SonarV4Config) => (id: string, { signal }?: SignalParam) => import("ts-results-es").AsyncResult<{
+    createdAt: import("temporal-polyfill").Temporal.Instant;
+    currentRunId: string | null;
+    id: string;
+    modelName: string | null;
+    modelProviderId: string | null;
+    modifiedAt: import("temporal-polyfill").Temporal.Instant;
+    tag: string;
+    title: string;
+}, import("../../../index.ts").HttpError>;

package/dist/enterprise/ai/conversations/methods/retrieveConversation.js

@@ -0,0 +1,29 @@
+/*
+ * 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 z from "zod";
+import { conversationPropertiesCodec } from "../conversationPropertiesCodec.js";
+export const retrieveConversation = (config) => function retrieveConversation(id, { signal } = {}) {
+    return config
+        .sonarV4Request(`agent/conversations/${id}`, {
+        headers: {
+            Accept: "application/json",
+        },
+        signal,
+    })
+        .map((res) => res.json())
+        .map((entity) => z.decode(conversationPropertiesCodec, entity));
+};
+//# sourceMappingURL=retrieveConversation.js.map

package/dist/enterprise/ai/conversations/methods/retrieveConversation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveConversation.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/retrieveConversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,CAAC,MAAM,KAAK,CAAC;AAGpB,OAAO,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AAEhF,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,MAAqB,EAAE,EAAE,CAC5D,SAAS,oBAAoB,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;IACpE,OAAO,MAAM;SACV,cAAc,CAAC,uBAAuB,EAAE,EAAE,EAAE;QAC3C,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;SAC3B;QACD,MAAM;KACP,CAAC;SACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA0D,CAAC;SAChF,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC,CAAC;AACpE,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 z from \"zod\";\nimport type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport type { SignalParam } from \"../../../../common/Params.ts\";\nimport { conversationPropertiesCodec } from \"../conversationPropertiesCodec.ts\";\n\nexport const retrieveConversation = (config: SonarV4Config) =>\n  function retrieveConversation(id: string, { signal }: SignalParam = {}) {\n    return config\n      .sonarV4Request(`agent/conversations/${id}`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n        signal,\n      })\n      .map((res) => res.json() as Promise<z.input<typeof conversationPropertiesCodec>>)\n      .map((entity) => z.decode(conversationPropertiesCodec, entity));\n  };\n"]}

package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.d.ts

@@ -0,0 +1,86 @@
+import type { SonarV4Config } from "../../../../common/Config.ts";
+import type { SignalParam } from "../../../../common/Params.ts";
+export declare const retrieveConversationHistory: (config: SonarV4Config) => (id: string, { signal }?: SignalParam) => import("ts-results-es").AsyncResult<{
+    data: ({
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "conversationUpdate";
+            summary?: string | undefined;
+            title?: string | undefined;
+        };
+        role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "error";
+            message: string;
+        };
+        role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "model";
+            name: string;
+            result: Record<string, unknown>;
+        };
+        role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            arguments: Record<string, unknown>;
+            callId: string;
+            chunkType: "toolRequest";
+            name: string;
+            commentary?: string | undefined;
+            summarizedTitle?: string | undefined;
+        };
+        role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            callId: string;
+            chunkType: "toolResponse";
+            name: string;
+            result: Record<string, unknown>;
+            commentary?: string | undefined;
+        };
+        role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "userMessage";
+            text: string;
+        };
+        role: "user";
+    })[];
+}, import("../../../index.ts").HttpError>;

package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.js

@@ -0,0 +1,32 @@
+/*
+ * 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 { chatEventCodec } from "../../chat/chatEventSchema.js";
+import * as z from "zod/mini";
+export const retrieveConversationHistory = (config) => function retrieveConversationHistory(id, { signal } = {}) {
+    return config
+        .sonarV4Request(`agent/conversations/${id}/messages`, {
+        headers: {
+            Accept: "application/json",
+        },
+        signal,
+    })
+        .map((res) => res.json())
+        .map((response) => ({
+        ...response,
+        data: response.data.map((chatEvent) => z.decode(chatEventCodec, chatEvent)),
+    }));
+};
+//# sourceMappingURL=retrieveConversationHistory.js.map

package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveConversationHistory.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/retrieveConversationHistory.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,MAAqB,EAAE,EAAE,CACnE,SAAS,2BAA2B,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;IAC3E,OAAO,MAAM;SACV,cAAc,CAAC,uBAAuB,EAAE,WAAW,EAAE;QACpD,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;SAC3B;QACD,MAAM;KACP,CAAC;SACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;SAC/E,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAClB,GAAG,QAAQ;QACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;KAC5E,CAAC,CAAC,CAAC;AACR,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 type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport type { SignalParam } from \"../../../../common/Params.ts\";\nimport { chatEventCodec } from \"../../chat/chatEventSchema.js\";\nimport * as z from \"zod/mini\";\n\nexport const retrieveConversationHistory = (config: SonarV4Config) =>\n  function retrieveConversationHistory(id: string, { signal }: SignalParam = {}) {\n    return config\n      .sonarV4Request(`agent/conversations/${id}/messages`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n        signal,\n      })\n      .map((res) => res.json() as Promise<{ data: z.input<typeof chatEventCodec>[] }>)\n      .map((response) => ({\n        ...response,\n        data: response.data.map((chatEvent) => z.decode(chatEventCodec, chatEvent)),\n      }));\n  };\n"]}

package/dist/enterprise/ai/conversations/methods/stopExchangeRun.d.ts

@@ -0,0 +1,2 @@
+import type { SonarV4Config } from "../../../../common/Config.ts";
+export declare const stopExchangeRun: (config: SonarV4Config) => (conversationId: string, id: string) => import("ts-results-es").AsyncResult<void, import("../../../index.ts").HttpError>;

package/dist/enterprise/ai/conversations/methods/stopExchangeRun.js

@@ -0,0 +1,24 @@
+/*
+ * 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.
+ */
+export const stopExchangeRun = (config) => function stopExchangeRun(conversationId, id) {
+    return config
+        .sonarV4Request(`agent/conversations/${conversationId}/runs/${id}:cancel`, {
+        keepalive: true,
+        method: "POST",
+    })
+        .map(() => undefined);
+};
+//# sourceMappingURL=stopExchangeRun.js.map

package/dist/enterprise/ai/conversations/methods/stopExchangeRun.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stopExchangeRun.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/stopExchangeRun.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,MAAqB,EAAE,EAAE,CACvD,SAAS,eAAe,CAAC,cAAsB,EAAE,EAAU;IACzD,OAAO,MAAM;SACV,cAAc,CAAC,uBAAuB,cAAc,SAAS,EAAE,SAAS,EAAE;QACzE,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,MAAM;KACf,CAAC;SACD,GAAG,CAAC,GAAG,EAAE,CAAC,SAAiB,CAAC,CAAC;AAClC,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 type { SonarV4Config } from \"../../../../common/Config.ts\";\n\nexport const stopExchangeRun = (config: SonarV4Config) =>\n  function stopExchangeRun(conversationId: string, id: string) {\n    return config\n      .sonarV4Request(`agent/conversations/${conversationId}/runs/${id}:cancel`, {\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map(() => undefined as void);\n  };\n"]}

package/dist/enterprise/ai/conversations/methods/streamRunEvents.d.ts

@@ -0,0 +1,6 @@
+import type { SonarV4Config } from "../../../../common/Config.ts";
+import type { SignalParam } from "../../../../common/Params.ts";
+export declare const streamRunEvents: (config: SonarV4Config) => (params: {
+    conversationId: string;
+    runId: string;
+}, { signal }?: SignalParam) => Promise<Response>;

package/dist/enterprise/ai/conversations/methods/streamRunEvents.js

@@ -0,0 +1,33 @@
+/*
+ * 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.
+ */
+export const streamRunEvents = (config) => function streamRunEvents(params, { signal } = {}) {
+    return config
+        .sonarV4Request(`agent/conversations/${params.conversationId}/runs/${params.runId}`, {
+        headers: {
+            Accept: "text/event-stream",
+            "Accept-Encoding": "identity",
+            Connection: "keep-alive",
+        },
+        signal,
+    })
+        .promise.then((result) => {
+        if (result.isErr()) {
+            throw result.error;
+        }
+        return result.value;
+    });
+};
+//# sourceMappingURL=streamRunEvents.js.map

package/dist/enterprise/ai/conversations/methods/streamRunEvents.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamRunEvents.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/streamRunEvents.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,MAAqB,EAAE,EAAE,CACvD,SAAS,eAAe,CACtB,MAAiD,EACjD,EAAE,MAAM,KAAkB,EAAE;IAE5B,OAAO,MAAM;SACV,cAAc,CAAC,uBAAuB,MAAM,CAAC,cAAc,SAAS,MAAM,CAAC,KAAK,EAAE,EAAE;QACnF,OAAO,EAAE;YACP,MAAM,EAAE,mBAAmB;YAC3B,iBAAiB,EAAE,UAAU;YAC7B,UAAU,EAAE,YAAY;SACzB;QACD,MAAM;KACP,CAAC;SACD,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;QACvB,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;YACnB,MAAM,MAAM,CAAC,KAAK,CAAC;QACrB,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC;IACtB,CAAC,CAAC,CAAC;AACP,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 type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport type { SignalParam } from \"../../../../common/Params.ts\";\n\nexport const streamRunEvents = (config: SonarV4Config) =>\n  function streamRunEvents(\n    params: { conversationId: string; runId: string },\n    { signal }: SignalParam = {},\n  ) {\n    return config\n      .sonarV4Request(`agent/conversations/${params.conversationId}/runs/${params.runId}`, {\n        headers: {\n          Accept: \"text/event-stream\",\n          \"Accept-Encoding\": \"identity\",\n          Connection: \"keep-alive\",\n        },\n        signal,\n      })\n      .promise.then((result) => {\n        if (result.isErr()) {\n          throw result.error;\n        }\n        return result.value;\n      });\n  };\n"]}

package/dist/enterprise/ai/conversations/methods/updateConversation.d.ts

@@ -0,0 +1,15 @@
+import * as z from "zod/mini";
+import type { SonarV4Config } from "../../../../common/Config.ts";
+export declare const updateConversation: (config: SonarV4Config) => (id: string, tag: string, properties: z.infer<typeof conversationUpdateSchema>) => import("ts-results-es").AsyncResult<{
+    createdAt: import("temporal-polyfill").Temporal.Instant;
+    currentRunId: string | null;
+    id: string;
+    modelName: string | null;
+    modelProviderId: string | null;
+    modifiedAt: import("temporal-polyfill").Temporal.Instant;
+    tag: string;
+    title: string;
+}, import("../../../index.ts").HttpError>;
+export declare const conversationUpdateSchema: z.ZodMiniObject<{
+    title: z.ZodMiniOptional<z.ZodMiniString<string>>;
+}, z.core.$strict>;

package/dist/enterprise/ai/conversations/methods/updateConversation.js

@@ -0,0 +1,35 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import * as z from "zod/mini";
+import { conversationPropertiesCodec, } from "../conversationPropertiesCodec.js";
+export const updateConversation = (config) => function updateConversation(id, tag, properties) {
+    return config
+        .sonarV4Request(`agent/conversations/${id}`, {
+        body: JSON.stringify({
+            ...z.parse(conversationUpdateSchema, properties),
+            tag,
+        }),
+        headers: { Accept: "application/json", "Content-Type": "application/json" },
+        keepalive: true,
+        method: "PATCH",
+    })
+        .map((res) => res.json())
+        .map((entity) => z.decode(conversationPropertiesCodec, entity));
+};
+export const conversationUpdateSchema = z.strictObject({
+    title: z.optional(z.string().check(z.trim(), z.minLength(1))),
+});
+//# sourceMappingURL=updateConversation.js.map

package/dist/enterprise/ai/conversations/methods/updateConversation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"updateConversation.js","sourceRoot":"","sources":["../../../../../src/enterprise/ai/conversations/methods/updateConversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,OAAO,EACL,2BAA2B,GAE5B,MAAM,mCAAmC,CAAC;AAE3C,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,MAAqB,EAAE,EAAE,CAC1D,SAAS,kBAAkB,CACzB,EAAU,EACV,GAAW,EACX,UAAoD;IAEpD,OAAO,MAAM;SACV,cAAc,CAAC,uBAAuB,EAAE,EAAE,EAAE;QAC3C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,GAAG,CAAC,CAAC,KAAK,CAAC,wBAAwB,EAAE,UAAU,CAAC;YAChD,GAAG;SACJ,CAAC;QACF,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC3E,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,OAAO;KAChB,CAAC;SACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAiC,CAAC;SACvD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC,CAAC;AACpE,CAAC,CAAC;AAEJ,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,YAAY,CAAC;IACrD,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;CAC9D,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 * as z from \"zod/mini\";\nimport type { SonarV4Config } from \"../../../../common/Config.ts\";\nimport {\n  conversationPropertiesCodec,\n  type ConversationEntity,\n} from \"../conversationPropertiesCodec.ts\";\n\nexport const updateConversation = (config: SonarV4Config) =>\n  function updateConversation(\n    id: string,\n    tag: string,\n    properties: z.infer<typeof conversationUpdateSchema>,\n  ) {\n    return config\n      .sonarV4Request(`agent/conversations/${id}`, {\n        body: JSON.stringify({\n          ...z.parse(conversationUpdateSchema, properties),\n          tag,\n        }),\n        headers: { Accept: \"application/json\", \"Content-Type\": \"application/json\" },\n        keepalive: true,\n        method: \"PATCH\",\n      })\n      .map((res) => res.json() as Promise<ConversationEntity>)\n      .map((entity) => z.decode(conversationPropertiesCodec, entity));\n  };\n\nexport const conversationUpdateSchema = z.strictObject({\n  title: z.optional(z.string().check(z.trim(), z.minLength(1))),\n});\n"]}

package/dist/enterprise/ai/conversations/reduceChatEvents.d.ts

@@ -0,0 +1,17 @@
+import type { ChatEvent, ChatEventWithChunkType } from "../chat/chatEventSchema.ts";
+import type { UserChatMessage } from "../chat/UserChatMessage.ts";
+export type AgentToolCall = {
+    id: string;
+    state: "error" | "pending" | "success";
+    request: ChatEventWithChunkType<"toolRequest"> | undefined;
+    result: ChatEventWithChunkType<"toolResponse"> | undefined;
+};
+export type ConversationExchangeMessage = ChatEventWithChunkType<"error"> | ChatEventWithChunkType<"model"> | ChatEventWithChunkType<"userMessage">;
+export type ConversationExchange = {
+    id: string;
+    messages: Map<string, ConversationExchangeMessage>;
+    submittedUserMessage?: UserChatMessage;
+    toolCalls: Map<string, AgentToolCall>;
+};
+export type ChatEventReducerState = ConversationExchange[];
+export declare function reduceChatEvents(state: ChatEventReducerState | undefined, chatEvents: ChatEvent[]): ChatEventReducerState;

package/dist/enterprise/ai/conversations/reduceChatEvents.js

@@ -0,0 +1,74 @@
+/*
+ * 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 { create } from "mutative";
+function applyChatEventToConversation(draft, chatEvent) {
+    let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);
+    if (!conversationExchange) {
+        conversationExchange = {
+            id: chatEvent.runId,
+            messages: new Map(),
+            toolCalls: new Map(),
+        };
+        draft.push(conversationExchange);
+        conversationExchange = draft[draft.length - 1];
+    }
+    applyChatEventToConversationExchange(conversationExchange, chatEvent);
+}
+function applyChatEventToConversationExchange(conversationExchange, chatEvent) {
+    switch (chatEvent.content.chunkType) {
+        case "error":
+        case "model":
+        case "userMessage": {
+            conversationExchange.messages.set(chatEvent.id, chatEvent);
+            break;
+        }
+        case "toolRequest":
+        case "toolResponse": {
+            const callId = chatEvent.content.callId;
+            if (!conversationExchange.toolCalls.has(callId)) {
+                conversationExchange.toolCalls.set(callId, {
+                    id: callId,
+                    request: undefined,
+                    result: undefined,
+                    state: "pending",
+                });
+            }
+            const toolCall = conversationExchange.toolCalls.get(callId);
+            switch (chatEvent.content.chunkType) {
+                case "toolRequest":
+                    toolCall.request = chatEvent;
+                    break;
+                case "toolResponse":
+                    toolCall.result = chatEvent;
+                    if ("errorMessage" in toolCall.result.content.result) {
+                        toolCall.state = "error";
+                    }
+                    else {
+                        toolCall.state = "success";
+                    }
+                    break;
+            }
+        }
+    }
+}
+export function reduceChatEvents(state = [], chatEvents) {
+    return create(state, (draft) => {
+        for (const chatEvent of chatEvents) {
+            applyChatEventToConversation(draft, Object.freeze(chatEvent));
+        }
+    });
+}
+//# sourceMappingURL=reduceChatEvents.js.map

package/dist/enterprise/ai/conversations/reduceChatEvents.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reduceChatEvents.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/reduceChatEvents.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAc,MAAM,UAAU,CAAC;AAyB9C,SAAS,4BAA4B,CACnC,KAAmC,EACnC,SAAoB;IAEpB,IAAI,oBAAoB,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,SAAS,CAAC,KAAK,CAAC,CAAC;IAEzF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,oBAAoB,GAAG;YACrB,EAAE,EAAE,SAAS,CAAC,KAAK;YACnB,QAAQ,EAAE,IAAI,GAAG,EAAE;YACnB,SAAS,EAAE,IAAI,GAAG,EAAE;SACrB,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,oBAAoB,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,oCAAoC,CAAC,oBAAqB,EAAE,SAAS,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,oCAAoC,CAC3C,oBAAiD,EACjD,SAAoB;IAEpB,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;QACpC,KAAK,OAAO,CAAC;QACb,KAAK,OAAO,CAAC;QACb,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,oBAAoB,CAAC,QAAQ,CAAC,GAAG,CAC/B,SAAS,CAAC,EAAE,EACZ,SAA+C,CAChD,CAAC;YACF,MAAM;QACR,CAAC;QACD,KAAK,aAAa,CAAC;QACnB,KAAK,cAAc,CAAC,CAAC,CAAC;YACpB,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;YAExC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE;oBACzC,EAAE,EAAE,MAAM;oBACV,OAAO,EAAE,SAAS;oBAClB,MAAM,EAAE,SAAS;oBACjB,KAAK,EAAE,SAAS;iBACjB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,QAAQ,GAAG,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;YAE7D,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBACpC,KAAK,aAAa;oBAChB,QAAQ,CAAC,OAAO,GAAG,SAAoE,CAAC;oBACxF,MAAM;gBACR,KAAK,cAAc;oBACjB,QAAQ,CAAC,MAAM,GAAG,SAAqE,CAAC;oBACxF,IAAI,cAAc,IAAI,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;wBACrD,QAAQ,CAAC,KAAK,GAAG,OAAO,CAAC;oBAC3B,CAAC;yBAAM,CAAC;wBACN,QAAQ,CAAC,KAAK,GAAG,SAAS,CAAC;oBAC7B,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAA+B,EAAE,EAAE,UAAuB;IACzF,OAAO,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QAC7B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,4BAA4B,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,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 { create, type Draft } from \"mutative\";\nimport type { ChatEvent, ChatEventWithChunkType } from \"../chat/chatEventSchema.ts\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\n\nexport type AgentToolCall = {\n  id: string;\n  state: \"error\" | \"pending\" | \"success\";\n  request: ChatEventWithChunkType<\"toolRequest\"> | undefined;\n  result: ChatEventWithChunkType<\"toolResponse\"> | undefined;\n};\n\nexport type ConversationExchangeMessage =\n  | ChatEventWithChunkType<\"error\">\n  | ChatEventWithChunkType<\"model\">\n  | ChatEventWithChunkType<\"userMessage\">;\n\nexport type ConversationExchange = {\n  id: string;\n  messages: Map<string, ConversationExchangeMessage>;\n  submittedUserMessage?: UserChatMessage;\n  toolCalls: Map<string, AgentToolCall>;\n};\n\nexport type ChatEventReducerState = ConversationExchange[];\n\nfunction applyChatEventToConversation(\n  draft: Draft<ChatEventReducerState>,\n  chatEvent: ChatEvent,\n): void {\n  let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);\n\n  if (!conversationExchange) {\n    conversationExchange = {\n      id: chatEvent.runId,\n      messages: new Map(),\n      toolCalls: new Map(),\n    };\n    draft.push(conversationExchange);\n    conversationExchange = draft[draft.length - 1];\n  }\n\n  applyChatEventToConversationExchange(conversationExchange!, chatEvent);\n}\n\nfunction applyChatEventToConversationExchange(\n  conversationExchange: Draft<ConversationExchange>,\n  chatEvent: ChatEvent,\n): void {\n  switch (chatEvent.content.chunkType) {\n    case \"error\":\n    case \"model\":\n    case \"userMessage\": {\n      conversationExchange.messages.set(\n        chatEvent.id,\n        chatEvent as Draft<ConversationExchangeMessage>,\n      );\n      break;\n    }\n    case \"toolRequest\":\n    case \"toolResponse\": {\n      const callId = chatEvent.content.callId;\n\n      if (!conversationExchange.toolCalls.has(callId)) {\n        conversationExchange.toolCalls.set(callId, {\n          id: callId,\n          request: undefined,\n          result: undefined,\n          state: \"pending\",\n        });\n      }\n\n      const toolCall = conversationExchange.toolCalls.get(callId)!;\n\n      switch (chatEvent.content.chunkType) {\n        case \"toolRequest\":\n          toolCall.request = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolRequest\">>;\n          break;\n        case \"toolResponse\":\n          toolCall.result = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolResponse\">>;\n          if (\"errorMessage\" in toolCall.result.content.result) {\n            toolCall.state = \"error\";\n          } else {\n            toolCall.state = \"success\";\n          }\n          break;\n      }\n    }\n  }\n}\n\nexport function reduceChatEvents(state: ChatEventReducerState = [], chatEvents: ChatEvent[]) {\n  return create(state, (draft) => {\n    for (const chatEvent of chatEvents) {\n      applyChatEventToConversation(draft, Object.freeze(chatEvent));\n    }\n  });\n}\n"]}

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

@@ -1,3 +1,7 @@
+import { AgentConversation } from "./conversations/AgentConversation.ts";
 import { ModelProvider } from "./modelProvider/ModelProvider.ts";
+import { type ChatEvent, type ChatEventWithChunkType } from "./chat/chatEventSchema.ts";
+import type { ConversationExchange, ConversationExchangeMessage, AgentToolCall } from "./conversations/reduceChatEvents.ts";
+import { type AgentConversationMachine } from "./conversations/createConversationMachine.ts";
 export * from "./chat/index.ts";
-export { ModelProvider };
+export { AgentConversation, ModelProvider, type ChatEvent, type ChatEventWithChunkType, type ConversationExchange, type ConversationExchangeMessage, type AgentToolCall, type AgentConversationMachine, };

package/dist/enterprise/ai/index.js

@@ -13,7 +13,10 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+import { AgentConversation } from "./conversations/AgentConversation.js";
 import { ModelProvider } from "./modelProvider/ModelProvider.js";
+import {} from "./chat/chatEventSchema.js";
+import {} from "./conversations/createConversationMachine.js";
 export * from "./chat/index.js";
-export { ModelProvider };
+export { AgentConversation, ModelProvider, };
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/enterprise/ai/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,cAAc,iBAAiB,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,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 { ModelProvider } from \"./modelProvider/ModelProvider.ts\";\nexport * from \"./chat/index.ts\";\nexport { ModelProvider };\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/enterprise/ai/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAA+C,MAAM,2BAA2B,CAAC;AAMxF,OAAO,EAAiC,MAAM,8CAA8C,CAAC;AAC7F,cAAc,iBAAiB,CAAC;AAChC,OAAO,EACL,iBAAiB,EACjB,aAAa,GAOd,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 { AgentConversation } from \"./conversations/AgentConversation.ts\";\nimport { ModelProvider } from \"./modelProvider/ModelProvider.ts\";\nimport { type ChatEvent, type ChatEventWithChunkType } from \"./chat/chatEventSchema.ts\";\nimport type {\n  ConversationExchange,\n  ConversationExchangeMessage,\n  AgentToolCall,\n} from \"./conversations/reduceChatEvents.ts\";\nimport { type AgentConversationMachine } from \"./conversations/createConversationMachine.ts\";\nexport * from \"./chat/index.ts\";\nexport {\n  AgentConversation,\n  ModelProvider,\n  type ChatEvent,\n  type ChatEventWithChunkType,\n  type ConversationExchange,\n  type ConversationExchangeMessage,\n  type AgentToolCall,\n  type AgentConversationMachine,\n};\n"]}

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

@@ -16,7 +16,7 @@ export declare class ModelProvider implements Omit<ModelProviderProperties, "tag
     readonly sqlModelNames: ModelProviderProperties["sqlModelNames"];
     constructor(config: V4Config, properties: z.output<typeof modelProviderRetrieveCodec>);
     delete(): import("ts-results-es").AsyncResult<void, import("../../index.ts").HttpError>;
-    setDefault(): import("ts-results-es").AsyncResult<ModelProvider, VersionConflictError | import("../../index.ts").HttpError>;
+    setDefault(): import("ts-results-es").AsyncResult<void, import("../../index.ts").HttpError>;
     update(fields: Partial<z.output<typeof modelProviderUpdateCodec>>): import("ts-results-es").AsyncResult<ModelProvider, VersionConflictError | import("../../index.ts").HttpError>;
     static createSchema: z.ZodMiniObject<{
         name: z.ZodMiniString<string>;
@@ -63,6 +63,13 @@ export declare class ModelProvider implements Omit<ModelProviderProperties, "tag
                     EU: "EU";
                 }>>;
             }, z.core.$strip>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            openAiCompatibleApi: z.ZodMiniObject<{
+                accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                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>;
@@ -119,6 +126,13 @@ export declare class ModelProvider implements Omit<ModelProviderProperties, "tag
                     EU: "EU";
                 }>>>;
             }, z.core.$strip>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            openAiCompatibleApi: z.ZodMiniObject<{
+                accessKey: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
+                completionsApiUrl: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                modelParameters: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
+                organizationId: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
+            }, z.core.$strip>;
         }, z.core.$strict>]>;
         defaultModelName: z.ZodMiniString<string>;
         isDefault: z.ZodMiniBoolean<boolean>;

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

@@ -52,7 +52,18 @@ export class ModelProvider {
             .map(() => undefined);
     }
     setDefault() {
-        return this.update({ isDefault: true });
+        return this.#config
+            .v4Request(`model-provider:setDefault`, {
+            body: JSON.stringify({
+                id: this.id,
+            }),
+            headers: {
+                "Content-Type": "application/json",
+            },
+            keepalive: true,
+            method: "POST",
+        })
+            .map(() => undefined);
     }
     update(fields) {
         const body = z.encode(modelProviderUpdateCodec, {

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

@@ -1 +1 @@
-{"version":3,"file":"ModelProvider.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/ModelProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,EAC1B,wBAAwB,GACzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAEnE,MAAM,OAAO,aAAa;IACf,MAAM,CAAoC;IAC1C,SAAS,CAAuC;IAChD,SAAS,CAAuC;IAChD,gBAAgB,CAA8C;IAC9D,EAAE,CAAgC;IAClC,SAAS,CAAuC;IAChD,UAAU,CAAwC;IAClD,UAAU,CAAwC;IAClD,IAAI,CAAkC;IACtC,aAAa,CAA2C;IAExD,OAAO,CAAW;IAClB,IAAI,CAAS;IAEtB,YAAY,MAAgB,EAAE,UAAuD;QACnF,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,gBAAgB,CAAC;QACpD,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;QACxC,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,aAAa,GAAG,UAAU,CAAC,aAAa,CAAC;QAE9C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC;IAC7B,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,IAAI,CAAC,EAAE,EAAE,EAAE;YAC7C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,QAAQ;SACjB,CAAC;aACD,GAAG,CAAC,GAAG,EAAE,CAAC,SAAiB,CAAC,CAAC;IAClC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,MAA0D;QAC/D,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE;YAC9C,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM;YACpC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB;YAClE,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS;YAC7C,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI;YAC9B,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa;SAC1D,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,IAAI,CAAC,EAAE,EAAE,EAAE;YAC7C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACjD,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SACd,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAC1F;aACA,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;YACd,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO,IAAI,oBAAoB,EAAE,CAAC;YACpC,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,YAAY,GAAG,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAC;IACvD,MAAM,CAAC,YAAY,GAAG,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { V4Config } from \"../../../common/Config.ts\";\nimport * as z from \"zod/mini\";\nimport {\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec,\n  modelProviderUpdateCodec,\n} from \"./modelProviderCodec.ts\";\nimport { VersionConflictError } from \"../../../common/problems.ts\";\n\nexport class ModelProvider implements Omit<ModelProviderProperties, \"tag\"> {\n  readonly config: ModelProviderProperties[\"config\"];\n  readonly createdAt: ModelProviderProperties[\"createdAt\"];\n  readonly createdBy: ModelProviderProperties[\"createdBy\"];\n  readonly defaultModelName: ModelProviderProperties[\"defaultModelName\"];\n  readonly id: ModelProviderProperties[\"id\"];\n  readonly isDefault: ModelProviderProperties[\"isDefault\"];\n  readonly modifiedAt: ModelProviderProperties[\"modifiedAt\"];\n  readonly modifiedBy: ModelProviderProperties[\"modifiedBy\"];\n  readonly name: ModelProviderProperties[\"name\"];\n  readonly sqlModelNames: ModelProviderProperties[\"sqlModelNames\"];\n\n  readonly #config: V4Config;\n  readonly #tag: string;\n\n  constructor(config: V4Config, properties: z.output<typeof modelProviderRetrieveCodec>) {\n    this.config = properties.config;\n    this.createdAt = properties.createdAt;\n    this.createdBy = properties.createdBy;\n    this.defaultModelName = properties.defaultModelName;\n    this.id = properties.id;\n    this.isDefault = properties.isDefault;\n    this.modifiedAt = properties.modifiedAt;\n    this.modifiedBy = properties.modifiedBy;\n    this.name = properties.name;\n    this.sqlModelNames = properties.sqlModelNames;\n\n    this.#config = config;\n    this.#tag = properties.tag;\n  }\n\n  delete() {\n    return this.#config\n      .v4Request(`model-provider/config/${this.id}`, {\n        keepalive: true,\n        method: \"DELETE\",\n      })\n      .map(() => undefined as void);\n  }\n\n  setDefault() {\n    return this.update({ isDefault: true });\n  }\n\n  update(fields: Partial<z.output<typeof modelProviderUpdateCodec>>) {\n    const body = z.encode(modelProviderUpdateCodec, {\n      config: fields.config ?? this.config,\n      defaultModelName: fields.defaultModelName ?? this.defaultModelName,\n      isDefault: fields.isDefault ?? this.isDefault,\n      name: fields.name ?? this.name,\n      sqlModelNames: fields.sqlModelNames ?? this.sqlModelNames,\n    });\n\n    return this.#config\n      .v4Request(`model-provider/config/${this.id}`, {\n        body: JSON.stringify({ ...body, tag: this.#tag }),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"PUT\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      )\n      .mapErr((val) => {\n        if (val.status === 409) {\n          return new VersionConflictError();\n        }\n        return val;\n      });\n  }\n\n  static createSchema = modelProviderCreateCodec.def.out;\n  static updateSchema = modelProviderUpdateCodec.def.out;\n}\n\ntype ModelProviderProperties = z.output<typeof modelProviderRetrieveCodec>;\n"]}
+{"version":3,"file":"ModelProvider.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/ModelProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,EAC1B,wBAAwB,GACzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAEnE,MAAM,OAAO,aAAa;IACf,MAAM,CAAoC;IAC1C,SAAS,CAAuC;IAChD,SAAS,CAAuC;IAChD,gBAAgB,CAA8C;IAC9D,EAAE,CAAgC;IAClC,SAAS,CAAuC;IAChD,UAAU,CAAwC;IAClD,UAAU,CAAwC;IAClD,IAAI,CAAkC;IACtC,aAAa,CAA2C;IAExD,OAAO,CAAW;IAClB,IAAI,CAAS;IAEtB,YAAY,MAAgB,EAAE,UAAuD;QACnF,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,gBAAgB,CAAC;QACpD,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;QACxC,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,aAAa,GAAG,UAAU,CAAC,aAAa,CAAC;QAE9C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC;IAC7B,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,IAAI,CAAC,EAAE,EAAE,EAAE;YAC7C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,QAAQ;SACjB,CAAC;aACD,GAAG,CAAC,GAAG,EAAE,CAAC,SAAiB,CAAC,CAAC;IAClC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,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;IAED,MAAM,CAAC,MAA0D;QAC/D,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE;YAC9C,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM;YACpC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB;YAClE,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS;YAC7C,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI;YAC9B,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa;SAC1D,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,IAAI,CAAC,EAAE,EAAE,EAAE;YAC7C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACjD,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SACd,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAC1F;aACA,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;YACd,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO,IAAI,oBAAoB,EAAE,CAAC;YACpC,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,YAAY,GAAG,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAC;IACvD,MAAM,CAAC,YAAY,GAAG,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { V4Config } from \"../../../common/Config.ts\";\nimport * as z from \"zod/mini\";\nimport {\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec,\n  modelProviderUpdateCodec,\n} from \"./modelProviderCodec.ts\";\nimport { VersionConflictError } from \"../../../common/problems.ts\";\n\nexport class ModelProvider implements Omit<ModelProviderProperties, \"tag\"> {\n  readonly config: ModelProviderProperties[\"config\"];\n  readonly createdAt: ModelProviderProperties[\"createdAt\"];\n  readonly createdBy: ModelProviderProperties[\"createdBy\"];\n  readonly defaultModelName: ModelProviderProperties[\"defaultModelName\"];\n  readonly id: ModelProviderProperties[\"id\"];\n  readonly isDefault: ModelProviderProperties[\"isDefault\"];\n  readonly modifiedAt: ModelProviderProperties[\"modifiedAt\"];\n  readonly modifiedBy: ModelProviderProperties[\"modifiedBy\"];\n  readonly name: ModelProviderProperties[\"name\"];\n  readonly sqlModelNames: ModelProviderProperties[\"sqlModelNames\"];\n\n  readonly #config: V4Config;\n  readonly #tag: string;\n\n  constructor(config: V4Config, properties: z.output<typeof modelProviderRetrieveCodec>) {\n    this.config = properties.config;\n    this.createdAt = properties.createdAt;\n    this.createdBy = properties.createdBy;\n    this.defaultModelName = properties.defaultModelName;\n    this.id = properties.id;\n    this.isDefault = properties.isDefault;\n    this.modifiedAt = properties.modifiedAt;\n    this.modifiedBy = properties.modifiedBy;\n    this.name = properties.name;\n    this.sqlModelNames = properties.sqlModelNames;\n\n    this.#config = config;\n    this.#tag = properties.tag;\n  }\n\n  delete() {\n    return this.#config\n      .v4Request(`model-provider/config/${this.id}`, {\n        keepalive: true,\n        method: \"DELETE\",\n      })\n      .map(() => undefined as void);\n  }\n\n  setDefault() {\n    return this.#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  update(fields: Partial<z.output<typeof modelProviderUpdateCodec>>) {\n    const body = z.encode(modelProviderUpdateCodec, {\n      config: fields.config ?? this.config,\n      defaultModelName: fields.defaultModelName ?? this.defaultModelName,\n      isDefault: fields.isDefault ?? this.isDefault,\n      name: fields.name ?? this.name,\n      sqlModelNames: fields.sqlModelNames ?? this.sqlModelNames,\n    });\n\n    return this.#config\n      .v4Request(`model-provider/config/${this.id}`, {\n        body: JSON.stringify({ ...body, tag: this.#tag }),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"PUT\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      )\n      .mapErr((val) => {\n        if (val.status === 409) {\n          return new VersionConflictError();\n        }\n        return val;\n      });\n  }\n\n  static createSchema = modelProviderCreateCodec.def.out;\n  static updateSchema = modelProviderUpdateCodec.def.out;\n}\n\ntype ModelProviderProperties = z.output<typeof modelProviderRetrieveCodec>;\n"]}