@dremio/js-sdk 0.31.8 → 0.33.0

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

@@ -0,0 +1,465 @@
+import * as z from "zod/mini";
+import { Temporal } from "temporal-polyfill";
+export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniObject<{
+    createdAt: z.ZodMiniString<string>;
+    createdBy: z.ZodMiniString<string>;
+    defaultModelName: z.ZodMiniString<string>;
+    id: z.ZodMiniString<string>;
+    isDefault: z.ZodMiniBoolean<boolean>;
+    modifiedAt: z.ZodMiniString<string>;
+    modifiedBy: z.ZodMiniString<string>;
+    name: z.ZodMiniString<string>;
+    sqlModelNames: z.ZodMiniArray<z.ZodMiniString<string>>;
+    tag: z.ZodMiniString<string>;
+    config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        amazonBedrock: z.ZodMiniObject<{
+            auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                accessKey: z.ZodMiniObject<{
+                    accessKeyId: z.ZodMiniString<string>;
+                }, 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<{}, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        azureOpenAi: z.ZodMiniObject<{
+            auth: z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>;
+            resourceName: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        googleGemini: z.ZodMiniObject<{}, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAi: z.ZodMiniObject<{
+            organizationId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+}, z.core.$strip>, z.ZodMiniObject<{
+    name: z.ZodMiniString<string>;
+    id: z.ZodMiniString<string>;
+    tag: z.ZodMiniString<string>;
+    createdBy: z.ZodMiniString<string>;
+    modifiedBy: z.ZodMiniString<string>;
+    config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        amazonBedrock: z.ZodMiniObject<{
+            auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                accessKey: z.ZodMiniObject<{
+                    accessKeyId: z.ZodMiniString<string>;
+                }, 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<{}, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        azureOpenAi: z.ZodMiniObject<{
+            auth: z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>;
+            resourceName: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        googleGemini: z.ZodMiniObject<{}, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAi: z.ZodMiniObject<{
+            organizationId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+    defaultModelName: z.ZodMiniString<string>;
+    isDefault: z.ZodMiniBoolean<boolean>;
+    createdAt: z.ZodMiniCustom<Temporal.Instant, Temporal.Instant>;
+    modifiedAt: z.ZodMiniCustom<Temporal.Instant, Temporal.Instant>;
+    sqlModelNames: z.ZodMiniSet<z.ZodMiniString<string>>;
+}, z.core.$strip>>;
+export declare const modelProviderCreateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
+    name: z.ZodMiniString<string>;
+    defaultModelName: z.ZodMiniString<string>;
+    isDefault: z.ZodMiniBoolean<boolean>;
+    sqlModelNames: z.ZodMiniArray<z.ZodMiniString<string>>;
+    config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        amazonBedrock: z.ZodMiniObject<{
+            auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                accessKey: z.ZodMiniObject<{
+                    accessKeyId: z.ZodMiniString<string>;
+                    secretAccessKey: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>, z.ZodMiniObject<{
+                role: z.ZodMiniObject<{
+                    roleArn: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>]>;
+            region: z.ZodMiniEnum<{
+                "ap-east-2": "ap-east-2";
+                "ap-northeast-1": "ap-northeast-1";
+                "ap-northeast-2": "ap-northeast-2";
+                "ap-northeast-3": "ap-northeast-3";
+                "ap-south-1": "ap-south-1";
+                "ap-south-2": "ap-south-2";
+                "ap-southeast-1": "ap-southeast-1";
+                "ap-southeast-2": "ap-southeast-2";
+                "ap-southeast-3": "ap-southeast-3";
+                "ap-southeast-4": "ap-southeast-4";
+                "ap-southeast-5": "ap-southeast-5";
+                "ap-southeast-7": "ap-southeast-7";
+                "ca-central-1": "ca-central-1";
+                "eu-central-1": "eu-central-1";
+                "eu-central-2": "eu-central-2";
+                "eu-north-1": "eu-north-1";
+                "eu-south-1": "eu-south-1";
+                "eu-south-2": "eu-south-2";
+                "eu-west-1": "eu-west-1";
+                "eu-west-2": "eu-west-2";
+                "eu-west-3": "eu-west-3";
+                "il-central-1": "il-central-1";
+                "me-central-1": "me-central-1";
+                "sa-east-1": "sa-east-1";
+                "us-east-1": "us-east-1";
+                "us-east-2": "us-east-2";
+                "us-gov-east-1": "us-gov-east-1";
+                "us-gov-west-1": "us-gov-west-1";
+                "us-west-1": "us-west-1";
+                "us-west-2": "us-west-2";
+            }>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        anthropic: z.ZodMiniObject<{
+            accessKey: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        azureOpenAi: z.ZodMiniObject<{
+            auth: z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureClientSecret: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>;
+            resourceName: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        googleGemini: z.ZodMiniObject<{
+            accessKey: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAi: z.ZodMiniObject<{
+            accessKey: z.ZodMiniString<string>;
+            organizationId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+}, z.core.$strip>, z.ZodMiniObject<{
+    name: z.ZodMiniString<string>;
+    config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        amazonBedrock: z.ZodMiniObject<{
+            auth: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                accessKey: z.ZodMiniObject<{
+                    accessKeyId: z.ZodMiniString<string>;
+                    secretAccessKey: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>, z.ZodMiniObject<{
+                role: z.ZodMiniObject<{
+                    roleArn: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>]>;
+            region: z.ZodMiniEnum<{
+                "ap-east-2": "ap-east-2";
+                "ap-northeast-1": "ap-northeast-1";
+                "ap-northeast-2": "ap-northeast-2";
+                "ap-northeast-3": "ap-northeast-3";
+                "ap-south-1": "ap-south-1";
+                "ap-south-2": "ap-south-2";
+                "ap-southeast-1": "ap-southeast-1";
+                "ap-southeast-2": "ap-southeast-2";
+                "ap-southeast-3": "ap-southeast-3";
+                "ap-southeast-4": "ap-southeast-4";
+                "ap-southeast-5": "ap-southeast-5";
+                "ap-southeast-7": "ap-southeast-7";
+                "ca-central-1": "ca-central-1";
+                "eu-central-1": "eu-central-1";
+                "eu-central-2": "eu-central-2";
+                "eu-north-1": "eu-north-1";
+                "eu-south-1": "eu-south-1";
+                "eu-south-2": "eu-south-2";
+                "eu-west-1": "eu-west-1";
+                "eu-west-2": "eu-west-2";
+                "eu-west-3": "eu-west-3";
+                "il-central-1": "il-central-1";
+                "me-central-1": "me-central-1";
+                "sa-east-1": "sa-east-1";
+                "us-east-1": "us-east-1";
+                "us-east-2": "us-east-2";
+                "us-gov-east-1": "us-gov-east-1";
+                "us-gov-west-1": "us-gov-west-1";
+                "us-west-1": "us-west-1";
+                "us-west-2": "us-west-2";
+            }>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        anthropic: z.ZodMiniObject<{
+            accessKey: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        azureOpenAi: z.ZodMiniObject<{
+            auth: z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureClientSecret: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>;
+            resourceName: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        googleGemini: z.ZodMiniObject<{
+            accessKey: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAi: z.ZodMiniObject<{
+            accessKey: z.ZodMiniString<string>;
+            organizationId: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+    defaultModelName: z.ZodMiniString<string>;
+    isDefault: z.ZodMiniBoolean<boolean>;
+    sqlModelNames: z.ZodMiniSet<z.ZodMiniString<string>>;
+}, z.core.$strip>>;
+export declare const modelProviderUpdateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
+    name: z.ZodMiniString<string>;
+    defaultModelName: z.ZodMiniString<string>;
+    isDefault: z.ZodMiniBoolean<boolean>;
+    sqlModelNames: z.ZodMiniArray<z.ZodMiniString<string>>;
+    config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        amazonBedrock: z.ZodMiniObject<{
+            auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                accessKey: z.ZodMiniObject<{
+                    accessKeyId: z.ZodMiniString<string>;
+                    secretAccessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                }, z.core.$strict>;
+            }, z.core.$strict>, z.ZodMiniObject<{
+                role: z.ZodMiniObject<{
+                    roleArn: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>]>>;
+            region: z.ZodMiniOptional<z.ZodMiniEnum<{
+                "ap-east-2": "ap-east-2";
+                "ap-northeast-1": "ap-northeast-1";
+                "ap-northeast-2": "ap-northeast-2";
+                "ap-northeast-3": "ap-northeast-3";
+                "ap-south-1": "ap-south-1";
+                "ap-south-2": "ap-south-2";
+                "ap-southeast-1": "ap-southeast-1";
+                "ap-southeast-2": "ap-southeast-2";
+                "ap-southeast-3": "ap-southeast-3";
+                "ap-southeast-4": "ap-southeast-4";
+                "ap-southeast-5": "ap-southeast-5";
+                "ap-southeast-7": "ap-southeast-7";
+                "ca-central-1": "ca-central-1";
+                "eu-central-1": "eu-central-1";
+                "eu-central-2": "eu-central-2";
+                "eu-north-1": "eu-north-1";
+                "eu-south-1": "eu-south-1";
+                "eu-south-2": "eu-south-2";
+                "eu-west-1": "eu-west-1";
+                "eu-west-2": "eu-west-2";
+                "eu-west-3": "eu-west-3";
+                "il-central-1": "il-central-1";
+                "me-central-1": "me-central-1";
+                "sa-east-1": "sa-east-1";
+                "us-east-1": "us-east-1";
+                "us-east-2": "us-east-2";
+                "us-gov-east-1": "us-gov-east-1";
+                "us-gov-west-1": "us-gov-west-1";
+                "us-west-1": "us-west-1";
+                "us-west-2": "us-west-2";
+            }>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        anthropic: z.ZodMiniObject<{
+            accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        azureOpenAi: z.ZodMiniObject<{
+            auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureClientSecret: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>, z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                    azureClientSecret: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                }, z.core.$strict>;
+            }, z.core.$strict>]>>;
+            resourceName: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        googleGemini: z.ZodMiniObject<{
+            accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAi: z.ZodMiniObject<{
+            accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+}, z.core.$strip>, z.ZodMiniObject<{
+    name: z.ZodMiniString<string>;
+    config: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+        amazonBedrock: z.ZodMiniObject<{
+            auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                accessKey: z.ZodMiniObject<{
+                    accessKeyId: z.ZodMiniString<string>;
+                    secretAccessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                }, z.core.$strict>;
+            }, z.core.$strict>, z.ZodMiniObject<{
+                role: z.ZodMiniObject<{
+                    roleArn: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>]>>;
+            region: z.ZodMiniOptional<z.ZodMiniEnum<{
+                "ap-east-2": "ap-east-2";
+                "ap-northeast-1": "ap-northeast-1";
+                "ap-northeast-2": "ap-northeast-2";
+                "ap-northeast-3": "ap-northeast-3";
+                "ap-south-1": "ap-south-1";
+                "ap-south-2": "ap-south-2";
+                "ap-southeast-1": "ap-southeast-1";
+                "ap-southeast-2": "ap-southeast-2";
+                "ap-southeast-3": "ap-southeast-3";
+                "ap-southeast-4": "ap-southeast-4";
+                "ap-southeast-5": "ap-southeast-5";
+                "ap-southeast-7": "ap-southeast-7";
+                "ca-central-1": "ca-central-1";
+                "eu-central-1": "eu-central-1";
+                "eu-central-2": "eu-central-2";
+                "eu-north-1": "eu-north-1";
+                "eu-south-1": "eu-south-1";
+                "eu-south-2": "eu-south-2";
+                "eu-west-1": "eu-west-1";
+                "eu-west-2": "eu-west-2";
+                "eu-west-3": "eu-west-3";
+                "il-central-1": "il-central-1";
+                "me-central-1": "me-central-1";
+                "sa-east-1": "sa-east-1";
+                "us-east-1": "us-east-1";
+                "us-east-2": "us-east-2";
+                "us-gov-east-1": "us-gov-east-1";
+                "us-gov-west-1": "us-gov-west-1";
+                "us-west-1": "us-west-1";
+                "us-west-2": "us-west-2";
+            }>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        anthropic: z.ZodMiniObject<{
+            accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        azureOpenAi: z.ZodMiniObject<{
+            auth: z.ZodMiniOptional<z.ZodMiniUnion<readonly [z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureClientSecret: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                }, z.core.$strict>;
+            }, z.core.$strict>, z.ZodMiniObject<{
+                clientSecretAuth: z.ZodMiniObject<{
+                    azureClientId: z.ZodMiniString<string>;
+                    azureTenantId: z.ZodMiniString<string>;
+                    azureClientSecret: z.ZodMiniOptional<z.ZodMiniString<string>>;
+                }, z.core.$strict>;
+            }, z.core.$strict>]>>;
+            resourceName: z.ZodMiniString<string>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        googleGemini: z.ZodMiniObject<{
+            accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>, z.ZodMiniObject<{
+        openAi: z.ZodMiniObject<{
+            accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
+            organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
+        }, z.core.$strict>;
+    }, z.core.$strict>]>;
+    defaultModelName: z.ZodMiniString<string>;
+    isDefault: z.ZodMiniBoolean<boolean>;
+    sqlModelNames: z.ZodMiniSet<z.ZodMiniString<string>>;
+}, z.core.$strip>>;

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"]}