@dremio/js-sdk 0.31.8 → 0.32.0

package/dist/enterprise/ai/modelProviderCodec.js

@@ -0,0 +1,154 @@
+/*
+ * 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 { amazonBedrockModelProviderConfigRetrieveSchema, amazonBedrockModelProviderConfigSchema, amazonBedrockModelProviderConfigUpdateSchema, anthropicModelProviderConfigRetrieveSchema, anthropicModelProviderConfigSchema, azureOpenAiModelProviderConfigRetrieveSchema, azureOpenAiModelProviderConfigSchema, azureOpenAiModelProviderConfigUpdateSchema, googleGeminiModelProviderConfigRetrieveSchema, googleGeminiModelProviderConfigSchema, openAiModelProviderConfigRetrieveSchema, openAiModelProviderConfigSchema, } from "./modelProviderConfigSchemas.js";
+import { Temporal } from "temporal-polyfill";
+const modelProviderBaseSchema = z.object({
+    createdAt: z.string().check(z.iso.datetime()),
+    createdBy: z.string(),
+    defaultModelName: z.string(),
+    id: z.string(),
+    isDefault: z.boolean(),
+    modifiedAt: z.string().check(z.iso.datetime()),
+    modifiedBy: z.string(),
+    // eslint-disable-next-line regexp/prefer-w, regexp/use-ignore-case
+    name: z.string().check(z.regex(/^[a-zA-Z0-9_-]+$/)),
+    sqlModelNames: z.array(z.string()),
+    tag: z.string(),
+});
+const modelProviderRetrieveInSchema = z.extend(modelProviderBaseSchema, {
+    config: z.union([
+        z.strictObject({
+            amazonBedrock: amazonBedrockModelProviderConfigRetrieveSchema,
+        }),
+        z.strictObject({
+            anthropic: anthropicModelProviderConfigRetrieveSchema,
+        }),
+        z.strictObject({
+            azureOpenAi: azureOpenAiModelProviderConfigRetrieveSchema,
+        }),
+        z.strictObject({
+            googleGemini: googleGeminiModelProviderConfigRetrieveSchema,
+        }),
+        z.strictObject({
+            openAi: openAiModelProviderConfigRetrieveSchema,
+        }),
+    ]),
+});
+const modelProviderCreateInSchema = z.extend(z.pick(modelProviderBaseSchema, {
+    defaultModelName: true,
+    isDefault: true,
+    name: true,
+    sqlModelNames: true,
+}), {
+    config: z.union([
+        z.strictObject({
+            amazonBedrock: amazonBedrockModelProviderConfigSchema,
+        }),
+        z.strictObject({
+            anthropic: anthropicModelProviderConfigSchema,
+        }),
+        z.strictObject({
+            azureOpenAi: azureOpenAiModelProviderConfigSchema,
+        }),
+        z.strictObject({
+            googleGemini: googleGeminiModelProviderConfigSchema,
+        }),
+        z.strictObject({
+            openAi: openAiModelProviderConfigSchema,
+        }),
+    ]),
+});
+const modelProviderUpdateInSchema = z.extend(z.omit(modelProviderCreateInSchema, { config: true }), {
+    config: z.union([
+        z.strictObject({
+            amazonBedrock: amazonBedrockModelProviderConfigUpdateSchema,
+        }),
+        z.strictObject({
+            anthropic: z.partial(anthropicModelProviderConfigSchema),
+        }),
+        z.strictObject({
+            azureOpenAi: azureOpenAiModelProviderConfigUpdateSchema,
+        }),
+        z.strictObject({
+            googleGemini: z.partial(googleGeminiModelProviderConfigSchema),
+        }),
+        z.strictObject({
+            openAi: z.partial(openAiModelProviderConfigSchema),
+        }),
+    ]),
+});
+export const modelProviderRetrieveCodec = z.codec(modelProviderRetrieveInSchema, z.extend(z.omit(modelProviderRetrieveInSchema, {
+    createdAt: true,
+    modifiedAt: true,
+    sqlModelNames: true,
+}), {
+    createdAt: z.instanceof(Temporal.Instant),
+    modifiedAt: z.instanceof(Temporal.Instant),
+    sqlModelNames: z.set(z.string()),
+}), {
+    decode(v) {
+        return {
+            ...v,
+            createdAt: Temporal.Instant.from(v.createdAt),
+            modifiedAt: Temporal.Instant.from(v.modifiedAt),
+            sqlModelNames: new Set(v.sqlModelNames),
+        };
+    },
+    encode(v) {
+        return {
+            ...v,
+            createdAt: v.createdAt.toString(),
+            modifiedAt: v.modifiedAt.toString(),
+            sqlModelNames: Array.from(v.sqlModelNames),
+        };
+    },
+});
+export const modelProviderCreateCodec = z.codec(modelProviderCreateInSchema, z.extend(z.omit(modelProviderCreateInSchema, {
+    sqlModelNames: true,
+}), {
+    sqlModelNames: z.set(z.string()),
+}), {
+    decode(v) {
+        return {
+            ...v,
+            sqlModelNames: new Set(v.sqlModelNames),
+        };
+    },
+    encode(v) {
+        return {
+            ...v,
+            sqlModelNames: Array.from(v.sqlModelNames),
+        };
+    },
+});
+export const modelProviderUpdateCodec = z.codec(modelProviderUpdateInSchema, z.extend(z.omit(modelProviderUpdateInSchema, { sqlModelNames: true }), {
+    sqlModelNames: z.set(z.string()),
+}), {
+    decode(v) {
+        return {
+            ...v,
+            sqlModelNames: new Set(v.sqlModelNames),
+        };
+    },
+    encode(v) {
+        return {
+            ...v,
+            sqlModelNames: Array.from(v.sqlModelNames),
+        };
+    },
+});
+//# sourceMappingURL=modelProviderCodec.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"modelProviderCodec.js","sourceRoot":"","sources":["../../../src/enterprise/ai/modelProviderCodec.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,8CAA8C,EAC9C,sCAAsC,EACtC,4CAA4C,EAC5C,0CAA0C,EAC1C,kCAAkC,EAClC,4CAA4C,EAC5C,oCAAoC,EACpC,0CAA0C,EAC1C,6CAA6C,EAC7C,qCAAqC,EACrC,uCAAuC,EACvC,+BAA+B,GAChC,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE;IAC5B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE;IACtB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC9C,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;IACtB,mEAAmE;IACnE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACnD,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;CAChB,CAAC,CAAC;AAEH,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC,uBAAuB,EAAE;IACtE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC;QACd,CAAC,CAAC,YAAY,CAAC;YACb,aAAa,EAAE,8CAA8C;SAC9D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,0CAA0C;SACtD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,WAAW,EAAE,4CAA4C;SAC1D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,YAAY,EAAE,6CAA6C;SAC5D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,MAAM,EAAE,uCAAuC;SAChD,CAAC;KACH,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAC1C,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE;IAC9B,gBAAgB,EAAE,IAAI;IACtB,SAAS,EAAE,IAAI;IACf,IAAI,EAAE,IAAI;IACV,aAAa,EAAE,IAAI;CACpB,CAAC,EACF;IACE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC;QACd,CAAC,CAAC,YAAY,CAAC;YACb,aAAa,EAAE,sCAAsC;SACtD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,kCAAkC;SAC9C,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,WAAW,EAAE,oCAAoC;SAClD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,YAAY,EAAE,qCAAqC;SACpD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,MAAM,EAAE,+BAA+B;SACxC,CAAC;KACH,CAAC;CACH,CACF,CAAC;AAEF,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAC1C,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EACrD;IACE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC;QACd,CAAC,CAAC,YAAY,CAAC;YACb,aAAa,EAAE,4CAA4C;SAC5D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,kCAAkC,CAAC;SACzD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,WAAW,EAAE,0CAA0C;SACxD,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,YAAY,EAAE,CAAC,CAAC,OAAO,CAAC,qCAAqC,CAAC;SAC/D,CAAC;QACF,CAAC,CAAC,YAAY,CAAC;YACb,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,+BAA+B,CAAC;SACnD,CAAC;KACH,CAAC;CACH,CACF,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAC/C,6BAA6B,EAC7B,CAAC,CAAC,MAAM,CACN,CAAC,CAAC,IAAI,CAAC,6BAA6B,EAAE;IACpC,SAAS,EAAE,IAAI;IACf,UAAU,EAAE,IAAI;IAChB,aAAa,EAAE,IAAI;CACpB,CAAC,EACF;IACE,SAAS,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC;IACzC,UAAU,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC;IAC1C,aAAa,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACjC,CACF,EACD;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAC7C,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC;YAC/C,aAAa,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC;SACxC,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;YACjC,UAAU,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,EAAE;YACnC,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF,CACF,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAC7C,2BAA2B,EAC3B,CAAC,CAAC,MAAM,CACN,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE;IAClC,aAAa,EAAE,IAAI;CACpB,CAAC,EACF;IACE,aAAa,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACjC,CACF,EACD;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC;SACxC,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF,CACF,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAC7C,2BAA2B,EAC3B,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE;IACrE,aAAa,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACjC,CAAC,EACF;IACE,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC;SACxC,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,CAAC;QACN,OAAO;YACL,GAAG,CAAC;YACJ,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF,CACF,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\nimport {\n  amazonBedrockModelProviderConfigRetrieveSchema,\n  amazonBedrockModelProviderConfigSchema,\n  amazonBedrockModelProviderConfigUpdateSchema,\n  anthropicModelProviderConfigRetrieveSchema,\n  anthropicModelProviderConfigSchema,\n  azureOpenAiModelProviderConfigRetrieveSchema,\n  azureOpenAiModelProviderConfigSchema,\n  azureOpenAiModelProviderConfigUpdateSchema,\n  googleGeminiModelProviderConfigRetrieveSchema,\n  googleGeminiModelProviderConfigSchema,\n  openAiModelProviderConfigRetrieveSchema,\n  openAiModelProviderConfigSchema,\n} from \"./modelProviderConfigSchemas.ts\";\nimport { Temporal } from \"temporal-polyfill\";\n\nconst modelProviderBaseSchema = z.object({\n  createdAt: z.string().check(z.iso.datetime()),\n  createdBy: z.string(),\n  defaultModelName: z.string(),\n  id: z.string(),\n  isDefault: z.boolean(),\n  modifiedAt: z.string().check(z.iso.datetime()),\n  modifiedBy: z.string(),\n  // eslint-disable-next-line regexp/prefer-w, regexp/use-ignore-case\n  name: z.string().check(z.regex(/^[a-zA-Z0-9_-]+$/)),\n  sqlModelNames: z.array(z.string()),\n  tag: z.string(),\n});\n\nconst modelProviderRetrieveInSchema = z.extend(modelProviderBaseSchema, {\n  config: z.union([\n    z.strictObject({\n      amazonBedrock: amazonBedrockModelProviderConfigRetrieveSchema,\n    }),\n    z.strictObject({\n      anthropic: anthropicModelProviderConfigRetrieveSchema,\n    }),\n    z.strictObject({\n      azureOpenAi: azureOpenAiModelProviderConfigRetrieveSchema,\n    }),\n    z.strictObject({\n      googleGemini: googleGeminiModelProviderConfigRetrieveSchema,\n    }),\n    z.strictObject({\n      openAi: openAiModelProviderConfigRetrieveSchema,\n    }),\n  ]),\n});\n\nconst modelProviderCreateInSchema = z.extend(\n  z.pick(modelProviderBaseSchema, {\n    defaultModelName: true,\n    isDefault: true,\n    name: true,\n    sqlModelNames: true,\n  }),\n  {\n    config: z.union([\n      z.strictObject({\n        amazonBedrock: amazonBedrockModelProviderConfigSchema,\n      }),\n      z.strictObject({\n        anthropic: anthropicModelProviderConfigSchema,\n      }),\n      z.strictObject({\n        azureOpenAi: azureOpenAiModelProviderConfigSchema,\n      }),\n      z.strictObject({\n        googleGemini: googleGeminiModelProviderConfigSchema,\n      }),\n      z.strictObject({\n        openAi: openAiModelProviderConfigSchema,\n      }),\n    ]),\n  },\n);\n\nconst modelProviderUpdateInSchema = z.extend(\n  z.omit(modelProviderCreateInSchema, { config: true }),\n  {\n    config: z.union([\n      z.strictObject({\n        amazonBedrock: amazonBedrockModelProviderConfigUpdateSchema,\n      }),\n      z.strictObject({\n        anthropic: z.partial(anthropicModelProviderConfigSchema),\n      }),\n      z.strictObject({\n        azureOpenAi: azureOpenAiModelProviderConfigUpdateSchema,\n      }),\n      z.strictObject({\n        googleGemini: z.partial(googleGeminiModelProviderConfigSchema),\n      }),\n      z.strictObject({\n        openAi: z.partial(openAiModelProviderConfigSchema),\n      }),\n    ]),\n  },\n);\n\nexport const modelProviderRetrieveCodec = z.codec(\n  modelProviderRetrieveInSchema,\n  z.extend(\n    z.omit(modelProviderRetrieveInSchema, {\n      createdAt: true,\n      modifiedAt: true,\n      sqlModelNames: true,\n    }),\n    {\n      createdAt: z.instanceof(Temporal.Instant),\n      modifiedAt: z.instanceof(Temporal.Instant),\n      sqlModelNames: z.set(z.string()),\n    },\n  ),\n  {\n    decode(v) {\n      return {\n        ...v,\n        createdAt: Temporal.Instant.from(v.createdAt),\n        modifiedAt: Temporal.Instant.from(v.modifiedAt),\n        sqlModelNames: new Set(v.sqlModelNames),\n      };\n    },\n    encode(v) {\n      return {\n        ...v,\n        createdAt: v.createdAt.toString(),\n        modifiedAt: v.modifiedAt.toString(),\n        sqlModelNames: Array.from(v.sqlModelNames),\n      };\n    },\n  },\n);\n\nexport const modelProviderCreateCodec = z.codec(\n  modelProviderCreateInSchema,\n  z.extend(\n    z.omit(modelProviderCreateInSchema, {\n      sqlModelNames: true,\n    }),\n    {\n      sqlModelNames: z.set(z.string()),\n    },\n  ),\n  {\n    decode(v) {\n      return {\n        ...v,\n        sqlModelNames: new Set(v.sqlModelNames),\n      };\n    },\n    encode(v) {\n      return {\n        ...v,\n        sqlModelNames: Array.from(v.sqlModelNames),\n      };\n    },\n  },\n);\n\nexport const modelProviderUpdateCodec = z.codec(\n  modelProviderUpdateInSchema,\n  z.extend(z.omit(modelProviderUpdateInSchema, { sqlModelNames: true }), {\n    sqlModelNames: z.set(z.string()),\n  }),\n  {\n    decode(v) {\n      return {\n        ...v,\n        sqlModelNames: new Set(v.sqlModelNames),\n      };\n    },\n    encode(v) {\n      return {\n        ...v,\n        sqlModelNames: Array.from(v.sqlModelNames),\n      };\n    },\n  },\n);\n"]}

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

@@ -0,0 +1,296 @@
+import * as z from "zod/mini";
+/**
+ * Amazon Bedrock
+ */
+export declare const amazonBedrockRegionSchema: z.ZodMiniEnum<{
+    "ap-east-2": "ap-east-2";
+    "ap-northeast-1": "ap-northeast-1";
+    "ap-northeast-2": "ap-northeast-2";
+    "ap-northeast-3": "ap-northeast-3";
+    "ap-south-1": "ap-south-1";
+    "ap-south-2": "ap-south-2";
+    "ap-southeast-1": "ap-southeast-1";
+    "ap-southeast-2": "ap-southeast-2";
+    "ap-southeast-3": "ap-southeast-3";
+    "ap-southeast-4": "ap-southeast-4";
+    "ap-southeast-5": "ap-southeast-5";
+    "ap-southeast-7": "ap-southeast-7";
+    "ca-central-1": "ca-central-1";
+    "eu-central-1": "eu-central-1";
+    "eu-central-2": "eu-central-2";
+    "eu-north-1": "eu-north-1";
+    "eu-south-1": "eu-south-1";
+    "eu-south-2": "eu-south-2";
+    "eu-west-1": "eu-west-1";
+    "eu-west-2": "eu-west-2";
+    "eu-west-3": "eu-west-3";
+    "il-central-1": "il-central-1";
+    "me-central-1": "me-central-1";
+    "sa-east-1": "sa-east-1";
+    "us-east-1": "us-east-1";
+    "us-east-2": "us-east-2";
+    "us-gov-east-1": "us-gov-east-1";
+    "us-gov-west-1": "us-gov-west-1";
+    "us-west-1": "us-west-1";
+    "us-west-2": "us-west-2";
+}>;
+export declare const amazonBedrockModelProviderConfigSchema: z.ZodMiniObject<{
+    auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        accessKey: z.ZodMiniObject<{
+            accessKeyId: z.ZodMiniString<string>;
+            secretAccessKey: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        role: z.ZodMiniObject<{
+            roleArn: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+    region: z.ZodMiniEnum<{
+        "ap-east-2": "ap-east-2";
+        "ap-northeast-1": "ap-northeast-1";
+        "ap-northeast-2": "ap-northeast-2";
+        "ap-northeast-3": "ap-northeast-3";
+        "ap-south-1": "ap-south-1";
+        "ap-south-2": "ap-south-2";
+        "ap-southeast-1": "ap-southeast-1";
+        "ap-southeast-2": "ap-southeast-2";
+        "ap-southeast-3": "ap-southeast-3";
+        "ap-southeast-4": "ap-southeast-4";
+        "ap-southeast-5": "ap-southeast-5";
+        "ap-southeast-7": "ap-southeast-7";
+        "ca-central-1": "ca-central-1";
+        "eu-central-1": "eu-central-1";
+        "eu-central-2": "eu-central-2";
+        "eu-north-1": "eu-north-1";
+        "eu-south-1": "eu-south-1";
+        "eu-south-2": "eu-south-2";
+        "eu-west-1": "eu-west-1";
+        "eu-west-2": "eu-west-2";
+        "eu-west-3": "eu-west-3";
+        "il-central-1": "il-central-1";
+        "me-central-1": "me-central-1";
+        "sa-east-1": "sa-east-1";
+        "us-east-1": "us-east-1";
+        "us-east-2": "us-east-2";
+        "us-gov-east-1": "us-gov-east-1";
+        "us-gov-west-1": "us-gov-west-1";
+        "us-west-1": "us-west-1";
+        "us-west-2": "us-west-2";
+    }>;
+}, z.core.$strict>;
+export declare const amazonBedrockModelProviderConfigRetrieveSchema: z.ZodMiniObject<{
+    auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        accessKey: z.ZodMiniObject<{
+            accessKeyId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        role: z.ZodMiniObject<{
+            roleArn: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+    region: z.ZodMiniEnum<{
+        "ap-east-2": "ap-east-2";
+        "ap-northeast-1": "ap-northeast-1";
+        "ap-northeast-2": "ap-northeast-2";
+        "ap-northeast-3": "ap-northeast-3";
+        "ap-south-1": "ap-south-1";
+        "ap-south-2": "ap-south-2";
+        "ap-southeast-1": "ap-southeast-1";
+        "ap-southeast-2": "ap-southeast-2";
+        "ap-southeast-3": "ap-southeast-3";
+        "ap-southeast-4": "ap-southeast-4";
+        "ap-southeast-5": "ap-southeast-5";
+        "ap-southeast-7": "ap-southeast-7";
+        "ca-central-1": "ca-central-1";
+        "eu-central-1": "eu-central-1";
+        "eu-central-2": "eu-central-2";
+        "eu-north-1": "eu-north-1";
+        "eu-south-1": "eu-south-1";
+        "eu-south-2": "eu-south-2";
+        "eu-west-1": "eu-west-1";
+        "eu-west-2": "eu-west-2";
+        "eu-west-3": "eu-west-3";
+        "il-central-1": "il-central-1";
+        "me-central-1": "me-central-1";
+        "sa-east-1": "sa-east-1";
+        "us-east-1": "us-east-1";
+        "us-east-2": "us-east-2";
+        "us-gov-east-1": "us-gov-east-1";
+        "us-gov-west-1": "us-gov-west-1";
+        "us-west-1": "us-west-1";
+        "us-west-2": "us-west-2";
+    }>;
+}, z.core.$strict>;
+export declare const amazonBedrockModelProviderConfigUpdateSchema: z.ZodMiniObject<{
+    auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        accessKey: z.ZodMiniObject<{
+            accessKeyId: z.ZodMiniString<string>;
+            secretAccessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        role: z.ZodMiniObject<{
+            roleArn: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>>;
+    region: z.ZodMiniOptional<z.ZodMiniEnum<{
+        "ap-east-2": "ap-east-2";
+        "ap-northeast-1": "ap-northeast-1";
+        "ap-northeast-2": "ap-northeast-2";
+        "ap-northeast-3": "ap-northeast-3";
+        "ap-south-1": "ap-south-1";
+        "ap-south-2": "ap-south-2";
+        "ap-southeast-1": "ap-southeast-1";
+        "ap-southeast-2": "ap-southeast-2";
+        "ap-southeast-3": "ap-southeast-3";
+        "ap-southeast-4": "ap-southeast-4";
+        "ap-southeast-5": "ap-southeast-5";
+        "ap-southeast-7": "ap-southeast-7";
+        "ca-central-1": "ca-central-1";
+        "eu-central-1": "eu-central-1";
+        "eu-central-2": "eu-central-2";
+        "eu-north-1": "eu-north-1";
+        "eu-south-1": "eu-south-1";
+        "eu-south-2": "eu-south-2";
+        "eu-west-1": "eu-west-1";
+        "eu-west-2": "eu-west-2";
+        "eu-west-3": "eu-west-3";
+        "il-central-1": "il-central-1";
+        "me-central-1": "me-central-1";
+        "sa-east-1": "sa-east-1";
+        "us-east-1": "us-east-1";
+        "us-east-2": "us-east-2";
+        "us-gov-east-1": "us-gov-east-1";
+        "us-gov-west-1": "us-gov-west-1";
+        "us-west-1": "us-west-1";
+        "us-west-2": "us-west-2";
+    }>>;
+}, z.core.$strict>;
+/**
+ * Anthropic
+ */
+export declare const anthropicModelProviderConfigSchema: z.ZodMiniObject<{
+    accessKey: z.ZodMiniString<string>;
+}, z.core.$strict>;
+export declare const azureOpenAiModelProviderConfigSchema: z.ZodMiniObject<{
+    auth: z.ZodMiniObject<{
+        clientSecretAuth: z.ZodMiniObject<{
+            azureClientId: z.ZodMiniString<string>;
+            azureClientSecret: z.ZodMiniString<string>;
+            azureTenantId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>;
+    resourceName: z.ZodMiniString<string>;
+}, z.core.$strict>;
+export declare const azureOpenAiModelProviderConfigRetrieveSchema: z.ZodMiniObject<{
+    auth: z.ZodMiniObject<{
+        clientSecretAuth: z.ZodMiniObject<{
+            azureClientId: z.ZodMiniString<string>;
+            azureTenantId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>;
+    resourceName: z.ZodMiniString<string>;
+}, z.core.$strict>;
+export declare const azureOpenAiModelProviderConfigUpdateSchema: z.ZodMiniObject<{
+    auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        clientSecretAuth: z.ZodMiniObject<{
+            azureClientId: z.ZodMiniString<string>;
+            azureClientSecret: z.ZodMiniString<string>;
+            azureTenantId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        clientSecretAuth: z.ZodMiniObject<{
+            azureClientId: z.ZodMiniString<string>;
+            azureTenantId: z.ZodMiniString<string>;
+            azureClientSecret: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>>;
+    resourceName: z.ZodMiniString<string>;
+}, z.core.$strict>;
+export declare const anthropicModelProviderConfigRetrieveSchema: z.ZodMiniObject<{}, z.core.$strict>;
+/**
+ * Google Gemini
+ */
+export declare const googleGeminiModelProviderConfigSchema: z.ZodMiniObject<{
+    accessKey: z.ZodMiniString<string>;
+}, z.core.$strict>;
+export declare const googleGeminiModelProviderConfigRetrieveSchema: z.ZodMiniObject<{}, z.core.$strict>;
+/**
+ * OpenAI
+ */
+export declare const openAiModelProviderConfigSchema: z.ZodMiniObject<{
+    accessKey: z.ZodMiniString<string>;
+    organizationId: z.ZodMiniString<string>;
+}, z.core.$strict>;
+export declare const openAiModelProviderConfigRetrieveSchema: z.ZodMiniObject<{
+    organizationId: z.ZodMiniString<string>;
+}, z.core.$strict>;
+export declare const listModelsPropertiesSchema: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+    amazonBedrock: z.ZodMiniObject<{
+        auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+            accessKey: z.ZodMiniObject<{
+                accessKeyId: z.ZodMiniString<string>;
+                secretAccessKey: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>, z.ZodMiniObject<{
+            role: z.ZodMiniObject<{
+                roleArn: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>]>;
+        region: z.ZodMiniEnum<{
+            "ap-east-2": "ap-east-2";
+            "ap-northeast-1": "ap-northeast-1";
+            "ap-northeast-2": "ap-northeast-2";
+            "ap-northeast-3": "ap-northeast-3";
+            "ap-south-1": "ap-south-1";
+            "ap-south-2": "ap-south-2";
+            "ap-southeast-1": "ap-southeast-1";
+            "ap-southeast-2": "ap-southeast-2";
+            "ap-southeast-3": "ap-southeast-3";
+            "ap-southeast-4": "ap-southeast-4";
+            "ap-southeast-5": "ap-southeast-5";
+            "ap-southeast-7": "ap-southeast-7";
+            "ca-central-1": "ca-central-1";
+            "eu-central-1": "eu-central-1";
+            "eu-central-2": "eu-central-2";
+            "eu-north-1": "eu-north-1";
+            "eu-south-1": "eu-south-1";
+            "eu-south-2": "eu-south-2";
+            "eu-west-1": "eu-west-1";
+            "eu-west-2": "eu-west-2";
+            "eu-west-3": "eu-west-3";
+            "il-central-1": "il-central-1";
+            "me-central-1": "me-central-1";
+            "sa-east-1": "sa-east-1";
+            "us-east-1": "us-east-1";
+            "us-east-2": "us-east-2";
+            "us-gov-east-1": "us-gov-east-1";
+            "us-gov-west-1": "us-gov-west-1";
+            "us-west-1": "us-west-1";
+            "us-west-2": "us-west-2";
+        }>;
+    }, z.core.$strict>;
+}, z.core.$strict>, z.ZodMiniObject<{
+    anthropic: z.ZodMiniObject<{
+        accessKey: z.ZodMiniString<string>;
+    }, z.core.$strict>;
+}, z.core.$strict>, z.ZodMiniObject<{
+    azureOpenAi: z.ZodMiniObject<{
+        auth: z.ZodMiniObject<{
+            clientSecretAuth: z.ZodMiniObject<{
+                azureClientId: z.ZodMiniString<string>;
+                azureClientSecret: z.ZodMiniString<string>;
+                azureTenantId: z.ZodMiniString<string>;
+            }, z.core.$strict>;
+        }, z.core.$strict>;
+        resourceName: z.ZodMiniString<string>;
+    }, z.core.$strict>;
+}, z.core.$strict>, z.ZodMiniObject<{
+    googleGemini: z.ZodMiniObject<{
+        accessKey: z.ZodMiniString<string>;
+    }, z.core.$strict>;
+}, z.core.$strict>, z.ZodMiniObject<{
+    openAi: z.ZodMiniObject<{
+        accessKey: z.ZodMiniString<string>;
+        organizationId: z.ZodMiniString<string>;
+    }, z.core.$strict>;
+}, z.core.$strict>]>;

package/dist/enterprise/ai/modelProviderConfigSchemas.js

@@ -0,0 +1,159 @@
+/*
+ * 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";
+/**
+ * Amazon Bedrock
+ */
+export const amazonBedrockRegionSchema = z.enum([
+    "ap-east-2",
+    "ap-northeast-1",
+    "ap-northeast-2",
+    "ap-northeast-3",
+    "ap-south-1",
+    "ap-south-2",
+    "ap-southeast-1",
+    "ap-southeast-2",
+    "ap-southeast-3",
+    "ap-southeast-4",
+    "ap-southeast-5",
+    "ap-southeast-7",
+    "ca-central-1",
+    "eu-central-1",
+    "eu-central-2",
+    "eu-north-1",
+    "eu-south-1",
+    "eu-south-2",
+    "eu-west-1",
+    "eu-west-2",
+    "eu-west-3",
+    "il-central-1",
+    "me-central-1",
+    "sa-east-1",
+    "us-east-1",
+    "us-east-2",
+    "us-gov-east-1",
+    "us-gov-west-1",
+    "us-west-1",
+    "us-west-2",
+]);
+const amazonBedrockAccessKeyAuth = z.strictObject({
+    accessKey: z.strictObject({
+        accessKeyId: z.string(),
+        secretAccessKey: z.string(),
+    }),
+});
+const amazonBedrockRoleAuth = z.strictObject({
+    role: z.strictObject({
+        roleArn: z.string(),
+    }),
+});
+export const amazonBedrockModelProviderConfigSchema = z.strictObject({
+    auth: z.union([amazonBedrockAccessKeyAuth, amazonBedrockRoleAuth]),
+    region: amazonBedrockRegionSchema,
+});
+export const amazonBedrockModelProviderConfigRetrieveSchema = z.strictObject({
+    auth: z.union([
+        z.strictObject({
+            accessKey: z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }),
+        }),
+        amazonBedrockRoleAuth,
+    ]),
+    region: amazonBedrockRegionSchema,
+});
+export const amazonBedrockModelProviderConfigUpdateSchema = z.strictObject({
+    auth: z.optional(z.union([
+        z.strictObject({
+            accessKey: z.extend(z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }), {
+                secretAccessKey: z.optional(amazonBedrockAccessKeyAuth.shape.accessKey.shape.secretAccessKey),
+            }),
+        }),
+        amazonBedrockRoleAuth,
+    ])),
+    region: z.optional(amazonBedrockRegionSchema),
+});
+/**
+ * Anthropic
+ */
+export const anthropicModelProviderConfigSchema = z.strictObject({
+    accessKey: z.string(),
+});
+/**
+ * Azure OpenAI
+ */
+const azureOpenAiAuth = z.strictObject({
+    clientSecretAuth: z.strictObject({
+        azureClientId: z.string(),
+        azureClientSecret: z.string(),
+        azureTenantId: z.string(),
+    }),
+});
+export const azureOpenAiModelProviderConfigSchema = z.strictObject({
+    auth: z.strictObject(azureOpenAiAuth.shape),
+    resourceName: z.string(),
+});
+export const azureOpenAiModelProviderConfigRetrieveSchema = z.strictObject({
+    auth: z.strictObject({
+        clientSecretAuth: z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }),
+    }),
+    resourceName: z.string(),
+});
+export const azureOpenAiModelProviderConfigUpdateSchema = z.strictObject({
+    auth: z.optional(z.union([
+        azureOpenAiModelProviderConfigSchema.shape.auth,
+        z.strictObject({
+            clientSecretAuth: z.extend(z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }), {
+                azureClientSecret: z.optional(azureOpenAiAuth.shape.clientSecretAuth.shape.azureClientSecret),
+            }),
+        }),
+    ])),
+    resourceName: z.string(),
+});
+export const anthropicModelProviderConfigRetrieveSchema = z.strictObject({});
+/**
+ * Google Gemini
+ */
+export const googleGeminiModelProviderConfigSchema = z.strictObject({
+    accessKey: z.string(),
+});
+export const googleGeminiModelProviderConfigRetrieveSchema = z.strictObject({});
+/**
+ * OpenAI
+ */
+export const openAiModelProviderConfigSchema = z.strictObject({
+    accessKey: z.string(),
+    organizationId: z.string(),
+});
+export const openAiModelProviderConfigRetrieveSchema = z.strictObject({
+    organizationId: z.string(),
+});
+export const listModelsPropertiesSchema = z.union([
+    z.strictObject({
+        amazonBedrock: amazonBedrockModelProviderConfigSchema,
+    }),
+    z.strictObject({
+        anthropic: anthropicModelProviderConfigSchema,
+    }),
+    z.strictObject({
+        azureOpenAi: azureOpenAiModelProviderConfigSchema,
+    }),
+    z.strictObject({
+        googleGemini: googleGeminiModelProviderConfigSchema,
+    }),
+    z.strictObject({
+        openAi: openAiModelProviderConfigSchema,
+    }),
+]);
+//# sourceMappingURL=modelProviderConfigSchemas.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"modelProviderConfigSchemas.js","sourceRoot":"","sources":["../../../src/enterprise/ai/modelProviderConfigSchemas.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,IAAI,CAAC;IAC9C,WAAW;IACX,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,YAAY;IACZ,YAAY;IACZ,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,cAAc;IACd,cAAc;IACd,cAAc;IACd,YAAY;IACZ,YAAY;IACZ,YAAY;IACZ,WAAW;IACX,WAAW;IACX,WAAW;IACX,cAAc;IACd,cAAc;IACd,WAAW;IACX,WAAW;IACX,WAAW;IACX,eAAe;IACf,eAAe;IACf,WAAW;IACX,WAAW;CACZ,CAAC,CAAC;AAEH,MAAM,0BAA0B,GAAG,CAAC,CAAC,YAAY,CAAC;IAChD,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC;QACxB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;QACvB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;KAC5B,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAAC,CAAC,YAAY,CAAC;IAC3C,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC;QACnB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,YAAY,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,0BAA0B,EAAE,qBAAqB,CAAC,CAAC;IAClE,MAAM,EAAE,yBAAyB;CAClC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,8CAA8C,GAAG,CAAC,CAAC,YAAY,CAAC;IAC3E,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC;QACZ,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;SACzF,CAAC;QACF,qBAAqB;KACtB,CAAC;IACF,MAAM,EAAE,yBAAyB;CAClC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,4CAA4C,GAAG,CAAC,CAAC,YAAY,CAAC;IACzE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,CAAC,CAAC,YAAY,CAAC;YACb,SAAS,EAAE,CAAC,CAAC,MAAM,CACjB,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,EAC7E;gBACE,eAAe,EAAE,CAAC,CAAC,QAAQ,CACzB,0BAA0B,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CACjE;aACF,CACF;SACF,CAAC;QACF,qBAAqB;KACtB,CAAC,CACH;IACD,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,yBAAyB,CAAC;CAC9C,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,kCAAkC,GAAG,CAAC,CAAC,YAAY,CAAC;IAC/D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,eAAe,GAAG,CAAC,CAAC,YAAY,CAAC;IACrC,gBAAgB,EAAE,CAAC,CAAC,YAAY,CAAC;QAC/B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;QACzB,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;QAC7B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;KAC1B,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oCAAoC,GAAG,CAAC,CAAC,YAAY,CAAC;IACjE,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC;IAC3C,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,4CAA4C,GAAG,CAAC,CAAC,YAAY,CAAC;IACzE,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC;QACnB,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;KAC9F,CAAC;IACF,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0CAA0C,GAAG,CAAC,CAAC,YAAY,CAAC;IACvE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,oCAAoC,CAAC,KAAK,CAAC,IAAI;QAC/C,CAAC,CAAC,YAAY,CAAC;YACb,gBAAgB,EAAE,CAAC,CAAC,MAAM,CACxB,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,EAC3E;gBACE,iBAAiB,EAAE,CAAC,CAAC,QAAQ,CAC3B,eAAe,CAAC,KAAK,CAAC,gBAAgB,CAAC,KAAK,CAAC,iBAAiB,CAC/D;aACF,CACF;SACF,CAAC;KACH,CAAC,CACH;IACD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0CAA0C,GAAG,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;AAE7E;;GAEG;AACH,MAAM,CAAC,MAAM,qCAAqC,GAAG,CAAC,CAAC,YAAY,CAAC;IAClE,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,6CAA6C,GAAG,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;AAEhF;;GAEG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,YAAY,CAAC;IAC5D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;CAC3B,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,uCAAuC,GAAG,CAAC,CAAC,YAAY,CAAC;IACpE,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;CAC3B,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAAC;IAChD,CAAC,CAAC,YAAY,CAAC;QACb,aAAa,EAAE,sCAAsC;KACtD,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,SAAS,EAAE,kCAAkC;KAC9C,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,WAAW,EAAE,oCAAoC;KAClD,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,YAAY,EAAE,qCAAqC;KACpD,CAAC;IACF,CAAC,CAAC,YAAY,CAAC;QACb,MAAM,EAAE,+BAA+B;KACxC,CAAC;CACH,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\";\n\n/**\n * Amazon Bedrock\n */\nexport const amazonBedrockRegionSchema = z.enum([\n  \"ap-east-2\",\n  \"ap-northeast-1\",\n  \"ap-northeast-2\",\n  \"ap-northeast-3\",\n  \"ap-south-1\",\n  \"ap-south-2\",\n  \"ap-southeast-1\",\n  \"ap-southeast-2\",\n  \"ap-southeast-3\",\n  \"ap-southeast-4\",\n  \"ap-southeast-5\",\n  \"ap-southeast-7\",\n  \"ca-central-1\",\n  \"eu-central-1\",\n  \"eu-central-2\",\n  \"eu-north-1\",\n  \"eu-south-1\",\n  \"eu-south-2\",\n  \"eu-west-1\",\n  \"eu-west-2\",\n  \"eu-west-3\",\n  \"il-central-1\",\n  \"me-central-1\",\n  \"sa-east-1\",\n  \"us-east-1\",\n  \"us-east-2\",\n  \"us-gov-east-1\",\n  \"us-gov-west-1\",\n  \"us-west-1\",\n  \"us-west-2\",\n]);\n\nconst amazonBedrockAccessKeyAuth = z.strictObject({\n  accessKey: z.strictObject({\n    accessKeyId: z.string(),\n    secretAccessKey: z.string(),\n  }),\n});\n\nconst amazonBedrockRoleAuth = z.strictObject({\n  role: z.strictObject({\n    roleArn: z.string(),\n  }),\n});\n\nexport const amazonBedrockModelProviderConfigSchema = z.strictObject({\n  auth: z.union([amazonBedrockAccessKeyAuth, amazonBedrockRoleAuth]),\n  region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigRetrieveSchema = z.strictObject({\n  auth: z.union([\n    z.strictObject({\n      accessKey: z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }),\n    }),\n    amazonBedrockRoleAuth,\n  ]),\n  region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigUpdateSchema = z.strictObject({\n  auth: z.optional(\n    z.union([\n      z.strictObject({\n        accessKey: z.extend(\n          z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }),\n          {\n            secretAccessKey: z.optional(\n              amazonBedrockAccessKeyAuth.shape.accessKey.shape.secretAccessKey,\n            ),\n          },\n        ),\n      }),\n      amazonBedrockRoleAuth,\n    ]),\n  ),\n  region: z.optional(amazonBedrockRegionSchema),\n});\n\n/**\n * Anthropic\n */\nexport const anthropicModelProviderConfigSchema = z.strictObject({\n  accessKey: z.string(),\n});\n\n/**\n * Azure OpenAI\n */\nconst azureOpenAiAuth = z.strictObject({\n  clientSecretAuth: z.strictObject({\n    azureClientId: z.string(),\n    azureClientSecret: z.string(),\n    azureTenantId: z.string(),\n  }),\n});\n\nexport const azureOpenAiModelProviderConfigSchema = z.strictObject({\n  auth: z.strictObject(azureOpenAiAuth.shape),\n  resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigRetrieveSchema = z.strictObject({\n  auth: z.strictObject({\n    clientSecretAuth: z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }),\n  }),\n  resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigUpdateSchema = z.strictObject({\n  auth: z.optional(\n    z.union([\n      azureOpenAiModelProviderConfigSchema.shape.auth,\n      z.strictObject({\n        clientSecretAuth: z.extend(\n          z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }),\n          {\n            azureClientSecret: z.optional(\n              azureOpenAiAuth.shape.clientSecretAuth.shape.azureClientSecret,\n            ),\n          },\n        ),\n      }),\n    ]),\n  ),\n  resourceName: z.string(),\n});\n\nexport const anthropicModelProviderConfigRetrieveSchema = z.strictObject({});\n\n/**\n * Google Gemini\n */\nexport const googleGeminiModelProviderConfigSchema = z.strictObject({\n  accessKey: z.string(),\n});\n\nexport const googleGeminiModelProviderConfigRetrieveSchema = z.strictObject({});\n\n/**\n * OpenAI\n */\nexport const openAiModelProviderConfigSchema = z.strictObject({\n  accessKey: z.string(),\n  organizationId: z.string(),\n});\n\nexport const openAiModelProviderConfigRetrieveSchema = z.strictObject({\n  organizationId: z.string(),\n});\n\nexport const listModelsPropertiesSchema = z.union([\n  z.strictObject({\n    amazonBedrock: amazonBedrockModelProviderConfigSchema,\n  }),\n  z.strictObject({\n    anthropic: anthropicModelProviderConfigSchema,\n  }),\n  z.strictObject({\n    azureOpenAi: azureOpenAiModelProviderConfigSchema,\n  }),\n  z.strictObject({\n    googleGemini: googleGeminiModelProviderConfigSchema,\n  }),\n  z.strictObject({\n    openAi: openAiModelProviderConfigSchema,\n  }),\n]);\n"]}

package/dist/enterprise/interfaces.d.ts

@@ -15,3 +15,4 @@ export * from "./Grantee.ts";
 export type { Engine, EnterpriseCatalogObject, EnterpriseCatalogReference, EnterpriseScript, EnterpriseUser, Job, JobResultsSchema, Reflection, ReflectionSummary, Role, BranchHeadVersionReference, BareCommitVersionReference, TagVersionReference, VersionReference, };
 export * from "./catalog/catalogSearch/CatalogSearchResult.ts";
 export * from "./grants/Grant.ts";
+export * from "./ai/index.ts";

package/dist/enterprise/interfaces.js

@@ -19,4 +19,5 @@ export * from "./Ownable.js";
 export * from "./Grantee.js";
 export * from "./catalog/catalogSearch/CatalogSearchResult.js";
 export * from "./grants/Grant.js";
+export * from "./ai/index.js";
 //# sourceMappingURL=interfaces.js.map