@dremio/js-sdk 0.42.0 → 0.43.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +209 -0
- package/dist/cloud/ai/AIResource.d.ts +716 -17
- package/dist/cloud/ai/AIResource.js +40 -18
- package/dist/cloud/ai/AIResource.js.map +1 -1
- package/dist/cloud/ai/DremioModelProvider.d.ts +1 -1
- package/dist/cloud/ai/DremioModelProvider.js +6 -13
- package/dist/cloud/ai/DremioModelProvider.js.map +1 -1
- package/dist/cloud/ai/modelProviderCodec.d.ts +12 -0
- package/dist/cloud/replaceOriginResource.js +4 -0
- package/dist/cloud/replaceOriginResource.js.map +1 -1
- package/dist/common/ConflictResolver.d.ts +9 -1
- package/dist/common/ConflictResolver.js +10 -1
- package/dist/common/ConflictResolver.js.map +1 -1
- package/dist/common/refineRfc9557Schema.d.ts +1 -0
- package/dist/common/refineRfc9557Schema.js +26 -0
- package/dist/common/refineRfc9557Schema.js.map +1 -0
- package/dist/common/sharedExports.d.ts +1 -0
- package/dist/common/sharedExports.js +1 -0
- package/dist/common/sharedExports.js.map +1 -1
- package/dist/enterprise/Dremio.d.ts +1 -1
- package/dist/enterprise/Dremio.js +3 -2
- package/dist/enterprise/Dremio.js.map +1 -1
- package/dist/enterprise/ai/AIResource.d.ts +699 -18
- package/dist/enterprise/ai/AIResource.js +41 -19
- package/dist/enterprise/ai/AIResource.js.map +1 -1
- package/dist/enterprise/ai/chat/ChatSession.d.ts +4 -25
- package/dist/enterprise/ai/chat/ChatSession.js +6 -55
- package/dist/enterprise/ai/chat/ChatSession.js.map +1 -1
- package/dist/enterprise/ai/chat/UserChatMessage.d.ts +38 -22
- package/dist/enterprise/ai/chat/UserChatMessage.js +33 -41
- package/dist/enterprise/ai/chat/UserChatMessage.js.map +1 -1
- package/dist/enterprise/ai/chat/chatEventSchema.d.ts +169 -11
- package/dist/enterprise/ai/chat/chatEventSchema.js +183 -35
- package/dist/enterprise/ai/chat/chatEventSchema.js.map +1 -1
- package/dist/enterprise/ai/chat/eventWrappers/AgentChatPartialResponse.d.ts +0 -6
- package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.d.ts +6 -6
- package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.js +2 -2
- package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.js.map +1 -1
- package/dist/enterprise/ai/chat/index.d.ts +0 -3
- package/dist/enterprise/ai/chat/index.js +0 -1
- package/dist/enterprise/ai/chat/index.js.map +1 -1
- package/dist/enterprise/ai/chat/methods/sendChatMessage.js +4 -1
- package/dist/enterprise/ai/chat/methods/sendChatMessage.js.map +1 -1
- package/dist/enterprise/ai/chat/responseContent/AgentModelResponseContent.d.ts +1 -2
- package/dist/enterprise/ai/chat/responseContent/AgentModelResponseContent.js.map +1 -1
- package/dist/enterprise/ai/chat/responseContent/AgentToolRequestResponseContent.d.ts +1 -2
- package/dist/enterprise/ai/chat/responseContent/AgentToolRequestResponseContent.js +0 -1
- package/dist/enterprise/ai/chat/responseContent/AgentToolRequestResponseContent.js.map +1 -1
- package/dist/enterprise/ai/chat/responseContent/AgentToolResultResponseContent.d.ts +1 -2
- package/dist/enterprise/ai/chat/responseContent/AgentToolResultResponseContent.js +0 -1
- package/dist/enterprise/ai/chat/responseContent/AgentToolResultResponseContent.js.map +1 -1
- package/dist/enterprise/ai/conversations/AgentConversation.d.ts +215 -0
- package/dist/enterprise/ai/conversations/AgentConversation.js +180 -0
- package/dist/enterprise/ai/conversations/AgentConversation.js.map +1 -0
- package/dist/enterprise/ai/conversations/conversationPropertiesCodec.d.ts +22 -0
- package/dist/enterprise/ai/conversations/conversationPropertiesCodec.js +62 -0
- package/dist/enterprise/ai/conversations/conversationPropertiesCodec.js.map +1 -0
- package/dist/enterprise/ai/conversations/createConversationMachine.d.ts +682 -0
- package/dist/enterprise/ai/conversations/createConversationMachine.js +354 -0
- package/dist/enterprise/ai/conversations/createConversationMachine.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/createConversation.d.ts +28 -0
- package/dist/enterprise/ai/conversations/methods/createConversation.js +68 -0
- package/dist/enterprise/ai/conversations/methods/createConversation.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/createExchangeRun.d.ts +16 -0
- package/dist/enterprise/ai/conversations/methods/createExchangeRun.js +36 -0
- package/dist/enterprise/ai/conversations/methods/createExchangeRun.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/deleteConversation.d.ts +2 -0
- package/dist/enterprise/ai/conversations/methods/deleteConversation.js +24 -0
- package/dist/enterprise/ai/conversations/methods/deleteConversation.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/listConversations.d.ts +8 -0
- package/dist/enterprise/ai/conversations/methods/listConversations.js +30 -0
- package/dist/enterprise/ai/conversations/methods/listConversations.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/retrieveConversation.d.ts +12 -0
- package/dist/enterprise/ai/conversations/methods/retrieveConversation.js +29 -0
- package/dist/enterprise/ai/conversations/methods/retrieveConversation.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.d.ts +86 -0
- package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.js +32 -0
- package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/stopExchangeRun.d.ts +2 -0
- package/dist/enterprise/ai/conversations/methods/stopExchangeRun.js +24 -0
- package/dist/enterprise/ai/conversations/methods/stopExchangeRun.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/streamRunEvents.d.ts +6 -0
- package/dist/enterprise/ai/conversations/methods/streamRunEvents.js +33 -0
- package/dist/enterprise/ai/conversations/methods/streamRunEvents.js.map +1 -0
- package/dist/enterprise/ai/conversations/methods/updateConversation.d.ts +15 -0
- package/dist/enterprise/ai/conversations/methods/updateConversation.js +35 -0
- package/dist/enterprise/ai/conversations/methods/updateConversation.js.map +1 -0
- package/dist/enterprise/ai/conversations/reduceChatEvents.d.ts +17 -0
- package/dist/enterprise/ai/conversations/reduceChatEvents.js +74 -0
- package/dist/enterprise/ai/conversations/reduceChatEvents.js.map +1 -0
- package/dist/enterprise/ai/index.d.ts +5 -1
- package/dist/enterprise/ai/index.js +4 -1
- package/dist/enterprise/ai/index.js.map +1 -1
- package/dist/enterprise/ai/modelProvider/ModelProvider.d.ts +15 -1
- package/dist/enterprise/ai/modelProvider/ModelProvider.js +12 -1
- package/dist/enterprise/ai/modelProvider/ModelProvider.js.map +1 -1
- package/dist/enterprise/ai/modelProvider/modelProviderCodec.d.ts +40 -0
- package/dist/enterprise/ai/modelProvider/modelProviderCodec.js +10 -1
- package/dist/enterprise/ai/modelProvider/modelProviderCodec.js.map +1 -1
- package/dist/enterprise/ai/modelProvider/modelProviderConfigSchemas.d.ts +21 -0
- package/dist/enterprise/ai/modelProvider/modelProviderConfigSchemas.js +17 -0
- package/dist/enterprise/ai/modelProvider/modelProviderConfigSchemas.js.map +1 -1
- package/dist/oss/catalog/CatalogReferences/DatasetCatalogReference.d.ts +1 -0
- package/dist/oss/catalog/CatalogReferences/DatasetCatalogReference.js +1 -0
- package/dist/oss/catalog/CatalogReferences/DatasetCatalogReference.js.map +1 -1
- package/dist-iife/cloud.js +21620 -6202
- package/dist-iife/community.js +14 -0
- package/dist-iife/enterprise.js +19404 -4038
- package/package.json +5 -2
- package/dist/enterprise/ai/chat/ChatSessionClient.d.ts +0 -18
- package/dist/enterprise/ai/chat/ChatSessionClient.js +0 -55
- package/dist/enterprise/ai/chat/ChatSessionClient.js.map +0 -1
- package/dist/enterprise/ai/chat/UserChatMessageContent.d.ts +0 -19
- package/dist/enterprise/ai/chat/UserChatMessageContent.js +0 -44
- package/dist/enterprise/ai/chat/UserChatMessageContent.js.map +0 -1
|
@@ -47,6 +47,12 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniObject<
|
|
|
47
47
|
EU: "EU";
|
|
48
48
|
}>;
|
|
49
49
|
}, z.core.$strip>;
|
|
50
|
+
}, z.core.$strict>, z.ZodMiniObject<{
|
|
51
|
+
openAiCompatibleApi: z.ZodMiniObject<{
|
|
52
|
+
completionsApiUrl: z.ZodMiniString<string>;
|
|
53
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
54
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
55
|
+
}, z.core.$strip>;
|
|
50
56
|
}, z.core.$strict>]>;
|
|
51
57
|
}, z.core.$strip>, z.ZodMiniObject<{
|
|
52
58
|
name: z.ZodMiniString<string>;
|
|
@@ -90,6 +96,12 @@ export declare const modelProviderRetrieveCodec: z.ZodMiniCodec<z.ZodMiniObject<
|
|
|
90
96
|
EU: "EU";
|
|
91
97
|
}>;
|
|
92
98
|
}, z.core.$strip>;
|
|
99
|
+
}, z.core.$strict>, z.ZodMiniObject<{
|
|
100
|
+
openAiCompatibleApi: z.ZodMiniObject<{
|
|
101
|
+
completionsApiUrl: z.ZodMiniString<string>;
|
|
102
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
103
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
104
|
+
}, z.core.$strip>;
|
|
93
105
|
}, z.core.$strict>]>;
|
|
94
106
|
defaultModelName: z.ZodMiniString<string>;
|
|
95
107
|
isDefault: z.ZodMiniBoolean<boolean>;
|
|
@@ -145,6 +157,13 @@ export declare const modelProviderCreateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
|
|
|
145
157
|
EU: "EU";
|
|
146
158
|
}>>;
|
|
147
159
|
}, z.core.$strip>;
|
|
160
|
+
}, z.core.$strict>, z.ZodMiniObject<{
|
|
161
|
+
openAiCompatibleApi: z.ZodMiniObject<{
|
|
162
|
+
accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
163
|
+
completionsApiUrl: z.ZodMiniString<string>;
|
|
164
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
165
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
166
|
+
}, z.core.$strip>;
|
|
148
167
|
}, z.core.$strict>]>;
|
|
149
168
|
}, z.core.$strip>, z.ZodMiniObject<{
|
|
150
169
|
name: z.ZodMiniString<string>;
|
|
@@ -191,6 +210,13 @@ export declare const modelProviderCreateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
|
|
|
191
210
|
EU: "EU";
|
|
192
211
|
}>>;
|
|
193
212
|
}, z.core.$strip>;
|
|
213
|
+
}, z.core.$strict>, z.ZodMiniObject<{
|
|
214
|
+
openAiCompatibleApi: z.ZodMiniObject<{
|
|
215
|
+
accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
216
|
+
completionsApiUrl: z.ZodMiniString<string>;
|
|
217
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
218
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
219
|
+
}, z.core.$strip>;
|
|
194
220
|
}, z.core.$strict>]>;
|
|
195
221
|
defaultModelName: z.ZodMiniString<string>;
|
|
196
222
|
isDefault: z.ZodMiniBoolean<boolean>;
|
|
@@ -250,6 +276,13 @@ export declare const modelProviderUpdateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
|
|
|
250
276
|
EU: "EU";
|
|
251
277
|
}>>>;
|
|
252
278
|
}, z.core.$strip>;
|
|
279
|
+
}, z.core.$strict>, z.ZodMiniObject<{
|
|
280
|
+
openAiCompatibleApi: z.ZodMiniObject<{
|
|
281
|
+
accessKey: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
|
|
282
|
+
completionsApiUrl: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
283
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
|
|
284
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
|
|
285
|
+
}, z.core.$strip>;
|
|
253
286
|
}, z.core.$strict>]>;
|
|
254
287
|
}, z.core.$strip>, z.ZodMiniObject<{
|
|
255
288
|
name: z.ZodMiniString<string>;
|
|
@@ -302,6 +335,13 @@ export declare const modelProviderUpdateCodec: z.ZodMiniCodec<z.ZodMiniObject<{
|
|
|
302
335
|
EU: "EU";
|
|
303
336
|
}>>>;
|
|
304
337
|
}, z.core.$strip>;
|
|
338
|
+
}, z.core.$strict>, z.ZodMiniObject<{
|
|
339
|
+
openAiCompatibleApi: z.ZodMiniObject<{
|
|
340
|
+
accessKey: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
|
|
341
|
+
completionsApiUrl: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
342
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
|
|
343
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniOptional<z.ZodMiniString<string>>>;
|
|
344
|
+
}, z.core.$strip>;
|
|
305
345
|
}, z.core.$strict>]>;
|
|
306
346
|
defaultModelName: z.ZodMiniString<string>;
|
|
307
347
|
isDefault: z.ZodMiniBoolean<boolean>;
|
|
@@ -14,7 +14,7 @@
|
|
|
14
14
|
* limitations under the License.
|
|
15
15
|
*/
|
|
16
16
|
import * as z from "zod/mini";
|
|
17
|
-
import { amazonBedrockModelProviderConfigRetrieveSchema, amazonBedrockModelProviderConfigSchema, amazonBedrockModelProviderConfigUpdateSchema, anthropicModelProviderConfigRetrieveSchema, anthropicModelProviderConfigSchema, azureOpenAiModelProviderConfigRetrieveSchema, azureOpenAiModelProviderConfigSchema, azureOpenAiModelProviderConfigUpdateSchema, googleGeminiModelProviderConfigRetrieveSchema, googleGeminiModelProviderConfigSchema, openAiModelProviderConfigRetrieveSchema, openAiModelProviderConfigSchema, } from "./modelProviderConfigSchemas.js";
|
|
17
|
+
import { amazonBedrockModelProviderConfigRetrieveSchema, amazonBedrockModelProviderConfigSchema, amazonBedrockModelProviderConfigUpdateSchema, anthropicModelProviderConfigRetrieveSchema, anthropicModelProviderConfigSchema, azureOpenAiModelProviderConfigRetrieveSchema, azureOpenAiModelProviderConfigSchema, azureOpenAiModelProviderConfigUpdateSchema, googleGeminiModelProviderConfigRetrieveSchema, googleGeminiModelProviderConfigSchema, openAiCompatibleApiModelProviderConfigRetrieveSchema, openAiCompatibleApiModelProviderConfigSchema, openAiModelProviderConfigRetrieveSchema, openAiModelProviderConfigSchema, } from "./modelProviderConfigSchemas.js";
|
|
18
18
|
import { Temporal } from "temporal-polyfill";
|
|
19
19
|
const modelProviderBaseSchema = z.object({
|
|
20
20
|
createdAt: z.string().check(z.iso.datetime()),
|
|
@@ -46,6 +46,9 @@ const modelProviderRetrieveInSchema = z.extend(modelProviderBaseSchema, {
|
|
|
46
46
|
z.strictObject({
|
|
47
47
|
openAi: openAiModelProviderConfigRetrieveSchema,
|
|
48
48
|
}),
|
|
49
|
+
z.strictObject({
|
|
50
|
+
openAiCompatibleApi: openAiCompatibleApiModelProviderConfigRetrieveSchema,
|
|
51
|
+
}),
|
|
49
52
|
]),
|
|
50
53
|
});
|
|
51
54
|
const modelProviderCreateInSchema = z.extend(z.pick(modelProviderBaseSchema, {
|
|
@@ -70,6 +73,9 @@ const modelProviderCreateInSchema = z.extend(z.pick(modelProviderBaseSchema, {
|
|
|
70
73
|
z.strictObject({
|
|
71
74
|
openAi: openAiModelProviderConfigSchema,
|
|
72
75
|
}),
|
|
76
|
+
z.strictObject({
|
|
77
|
+
openAiCompatibleApi: openAiCompatibleApiModelProviderConfigSchema,
|
|
78
|
+
}),
|
|
73
79
|
]),
|
|
74
80
|
});
|
|
75
81
|
const modelProviderUpdateInSchema = z.extend(z.omit(modelProviderCreateInSchema, { config: true }), {
|
|
@@ -89,6 +95,9 @@ const modelProviderUpdateInSchema = z.extend(z.omit(modelProviderCreateInSchema,
|
|
|
89
95
|
z.strictObject({
|
|
90
96
|
openAi: z.partial(openAiModelProviderConfigSchema),
|
|
91
97
|
}),
|
|
98
|
+
z.strictObject({
|
|
99
|
+
openAiCompatibleApi: z.partial(openAiCompatibleApiModelProviderConfigSchema),
|
|
100
|
+
}),
|
|
92
101
|
]),
|
|
93
102
|
});
|
|
94
103
|
export const modelProviderRetrieveCodec = z.codec(modelProviderRetrieveInSchema, z.extend(z.omit(modelProviderRetrieveInSchema, {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"modelProviderCodec.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/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"]}
|
|
1
|
+
{"version":3,"file":"modelProviderCodec.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/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,oDAAoD,EACpD,4CAA4C,EAC5C,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;QACF,CAAC,CAAC,YAAY,CAAC;YACb,mBAAmB,EAAE,oDAAoD;SAC1E,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;QACF,CAAC,CAAC,YAAY,CAAC;YACb,mBAAmB,EAAE,4CAA4C;SAClE,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;QACF,CAAC,CAAC,YAAY,CAAC;YACb,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,4CAA4C,CAAC;SAC7E,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 openAiCompatibleApiModelProviderConfigRetrieveSchema,\n openAiCompatibleApiModelProviderConfigSchema,\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 z.strictObject({\n openAiCompatibleApi: openAiCompatibleApiModelProviderConfigRetrieveSchema,\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 z.strictObject({\n openAiCompatibleApi: openAiCompatibleApiModelProviderConfigSchema,\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 z.strictObject({\n openAiCompatibleApi: z.partial(openAiCompatibleApiModelProviderConfigSchema),\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"]}
|
|
@@ -110,6 +110,20 @@ export declare const openAiModelProviderConfigRetrieveSchema: z.ZodMiniObject<{
|
|
|
110
110
|
EU: "EU";
|
|
111
111
|
}>;
|
|
112
112
|
}, z.core.$strip>;
|
|
113
|
+
/**
|
|
114
|
+
* OpenAI Compatible
|
|
115
|
+
*/
|
|
116
|
+
export declare const openAiCompatibleApiModelProviderConfigSchema: z.ZodMiniObject<{
|
|
117
|
+
accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
118
|
+
completionsApiUrl: z.ZodMiniString<string>;
|
|
119
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
120
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
121
|
+
}, z.core.$strip>;
|
|
122
|
+
export declare const openAiCompatibleApiModelProviderConfigRetrieveSchema: z.ZodMiniObject<{
|
|
123
|
+
completionsApiUrl: z.ZodMiniString<string>;
|
|
124
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
125
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
126
|
+
}, z.core.$strip>;
|
|
113
127
|
export declare const listModelsPropertiesSchema: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
|
|
114
128
|
access: z.ZodMiniUnion<readonly [z.ZodMiniObject<{
|
|
115
129
|
amazonBedrock: z.ZodMiniObject<{
|
|
@@ -154,6 +168,13 @@ export declare const listModelsPropertiesSchema: z.ZodMiniUnion<readonly [z.ZodM
|
|
|
154
168
|
EU: "EU";
|
|
155
169
|
}>>;
|
|
156
170
|
}, z.core.$strip>;
|
|
171
|
+
}, z.core.$strict>, z.ZodMiniObject<{
|
|
172
|
+
openAiCompatibleApi: z.ZodMiniObject<{
|
|
173
|
+
accessKey: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
174
|
+
completionsApiUrl: z.ZodMiniString<string>;
|
|
175
|
+
modelParameters: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
176
|
+
organizationId: z.ZodMiniOptional<z.ZodMiniString<string>>;
|
|
177
|
+
}, z.core.$strip>;
|
|
157
178
|
}, z.core.$strict>]>;
|
|
158
179
|
}, z.core.$strip>, z.ZodMiniObject<{
|
|
159
180
|
modelProviderName: z.ZodMiniString<string>;
|
|
@@ -110,6 +110,20 @@ export const openAiModelProviderConfigRetrieveSchema = z.object({
|
|
|
110
110
|
organizationId: z.string(),
|
|
111
111
|
region: z.enum(["GLOBAL", "US", "EU"]),
|
|
112
112
|
});
|
|
113
|
+
/**
|
|
114
|
+
* OpenAI Compatible
|
|
115
|
+
*/
|
|
116
|
+
export const openAiCompatibleApiModelProviderConfigSchema = z.object({
|
|
117
|
+
accessKey: z.optional(z.string()),
|
|
118
|
+
completionsApiUrl: z.string(),
|
|
119
|
+
modelParameters: z.optional(z.string()),
|
|
120
|
+
organizationId: z.optional(z.string()),
|
|
121
|
+
});
|
|
122
|
+
export const openAiCompatibleApiModelProviderConfigRetrieveSchema = z.object({
|
|
123
|
+
completionsApiUrl: z.string(),
|
|
124
|
+
modelParameters: z.optional(z.string()),
|
|
125
|
+
organizationId: z.optional(z.string()),
|
|
126
|
+
});
|
|
113
127
|
const listModelsAccessSchema = z.union([
|
|
114
128
|
z.strictObject({
|
|
115
129
|
amazonBedrock: amazonBedrockModelProviderConfigSchema,
|
|
@@ -126,6 +140,9 @@ const listModelsAccessSchema = z.union([
|
|
|
126
140
|
z.strictObject({
|
|
127
141
|
openAi: openAiModelProviderConfigSchema,
|
|
128
142
|
}),
|
|
143
|
+
z.strictObject({
|
|
144
|
+
openAiCompatibleApi: openAiCompatibleApiModelProviderConfigSchema,
|
|
145
|
+
}),
|
|
129
146
|
]);
|
|
130
147
|
export const listModelsPropertiesSchema = z.union([
|
|
131
148
|
z.object({
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"modelProviderConfigSchemas.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/modelProviderConfigSchemas.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;AAEpD,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC;QAClB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;QACvB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;KAC5B,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACb,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7D,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,MAAM,CAAC;IACrE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC;QACZ,CAAC,CAAC,MAAM,CAAC;YACP,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,MAAM,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,CAAC,CAAC,MAAM,CAAC;YACP,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,MAAM,CAAC;IACzD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,gBAAgB,EAAE,CAAC,CAAC,MAAM,CAAC;QACzB,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,MAAM,CAAC;IAC3D,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC;IACrC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,4CAA4C,GAAG,CAAC,CAAC,MAAM,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACb,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,MAAM,CAAC;IACjE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,oCAAoC,CAAC,KAAK,CAAC,IAAI;QAC/C,CAAC,CAAC,MAAM,CAAC;YACP,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,MAAM,CAAC,EAAE,CAAC,CAAC;AAEvE;;GAEG;AACH,MAAM,CAAC,MAAM,qCAAqC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,6CAA6C,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAE1E;;GAEG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;CACnD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,uCAAuC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9D,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC;IACrC,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;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAAC;IAChD,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,sBAAsB;KAC/B,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;KAC9B,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.string();\n\nconst amazonBedrockAccessKeyAuth = z.object({\n accessKey: z.object({\n accessKeyId: z.string(),\n secretAccessKey: z.string(),\n }),\n});\n\nconst amazonBedrockRoleAuth = z.object({\n role: z.object({\n roleArn: z.string(),\n }),\n});\n\nexport const amazonBedrockModelProviderConfigSchema = z.object({\n auth: z.union([amazonBedrockAccessKeyAuth, amazonBedrockRoleAuth]),\n region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigRetrieveSchema = z.object({\n auth: z.union([\n z.object({\n accessKey: z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }),\n }),\n amazonBedrockRoleAuth,\n ]),\n region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigUpdateSchema = z.object({\n auth: z.optional(\n z.union([\n z.object({\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.object({\n accessKey: z.string(),\n});\n\n/**\n * Azure OpenAI\n */\nconst azureOpenAiAuth = z.object({\n clientSecretAuth: z.object({\n azureClientId: z.string(),\n azureClientSecret: z.string(),\n azureTenantId: z.string(),\n }),\n});\n\nexport const azureOpenAiModelProviderConfigSchema = z.object({\n auth: z.object(azureOpenAiAuth.shape),\n resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigRetrieveSchema = z.object({\n auth: z.object({\n clientSecretAuth: z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }),\n }),\n resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigUpdateSchema = z.object({\n auth: z.optional(\n z.union([\n azureOpenAiModelProviderConfigSchema.shape.auth,\n z.object({\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.object({});\n\n/**\n * Google Gemini\n */\nexport const googleGeminiModelProviderConfigSchema = z.object({\n accessKey: z.string(),\n});\n\nexport const googleGeminiModelProviderConfigRetrieveSchema = z.object({});\n\n/**\n * OpenAI\n */\nexport const openAiModelProviderConfigSchema = z.object({\n accessKey: z.string(),\n organizationId: z.string(),\n region: z.optional(z.enum([\"GLOBAL\", \"US\", \"EU\"])),\n});\n\nexport const openAiModelProviderConfigRetrieveSchema = z.object({\n organizationId: z.string(),\n region: z.enum([\"GLOBAL\", \"US\", \"EU\"]),\n});\n\nconst listModelsAccessSchema = 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\nexport const listModelsPropertiesSchema = z.union([\n z.object({\n access: listModelsAccessSchema,\n }),\n z.object({\n modelProviderName: z.string(),\n }),\n]);\n"]}
|
|
1
|
+
{"version":3,"file":"modelProviderConfigSchemas.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/modelProvider/modelProviderConfigSchemas.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;AAEpD,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC;QAClB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;QACvB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;KAC5B,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACb,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7D,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,MAAM,CAAC;IACrE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC;QACZ,CAAC,CAAC,MAAM,CAAC;YACP,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,MAAM,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,CAAC,CAAC,MAAM,CAAC;YACP,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,MAAM,CAAC;IACzD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,gBAAgB,EAAE,CAAC,CAAC,MAAM,CAAC;QACzB,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,MAAM,CAAC;IAC3D,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC;IACrC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,4CAA4C,GAAG,CAAC,CAAC,MAAM,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACb,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,MAAM,CAAC;IACjE,IAAI,EAAE,CAAC,CAAC,QAAQ,CACd,CAAC,CAAC,KAAK,CAAC;QACN,oCAAoC,CAAC,KAAK,CAAC,IAAI;QAC/C,CAAC,CAAC,MAAM,CAAC;YACP,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,MAAM,CAAC,EAAE,CAAC,CAAC;AAEvE;;GAEG;AACH,MAAM,CAAC,MAAM,qCAAqC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,6CAA6C,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAE1E;;GAEG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;CACnD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,uCAAuC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9D,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;CACvC,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,4CAA4C,GAAG,CAAC,CAAC,MAAM,CAAC;IACnE,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACjC,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;IAC7B,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACvC,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oDAAoD,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3E,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;IAC7B,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACvC,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC;IACrC,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;IACF,CAAC,CAAC,YAAY,CAAC;QACb,mBAAmB,EAAE,4CAA4C;KAClE,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAAC;IAChD,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,sBAAsB;KAC/B,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;KAC9B,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.string();\n\nconst amazonBedrockAccessKeyAuth = z.object({\n accessKey: z.object({\n accessKeyId: z.string(),\n secretAccessKey: z.string(),\n }),\n});\n\nconst amazonBedrockRoleAuth = z.object({\n role: z.object({\n roleArn: z.string(),\n }),\n});\n\nexport const amazonBedrockModelProviderConfigSchema = z.object({\n auth: z.union([amazonBedrockAccessKeyAuth, amazonBedrockRoleAuth]),\n region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigRetrieveSchema = z.object({\n auth: z.union([\n z.object({\n accessKey: z.omit(amazonBedrockAccessKeyAuth.shape.accessKey, { secretAccessKey: true }),\n }),\n amazonBedrockRoleAuth,\n ]),\n region: amazonBedrockRegionSchema,\n});\n\nexport const amazonBedrockModelProviderConfigUpdateSchema = z.object({\n auth: z.optional(\n z.union([\n z.object({\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.object({\n accessKey: z.string(),\n});\n\n/**\n * Azure OpenAI\n */\nconst azureOpenAiAuth = z.object({\n clientSecretAuth: z.object({\n azureClientId: z.string(),\n azureClientSecret: z.string(),\n azureTenantId: z.string(),\n }),\n});\n\nexport const azureOpenAiModelProviderConfigSchema = z.object({\n auth: z.object(azureOpenAiAuth.shape),\n resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigRetrieveSchema = z.object({\n auth: z.object({\n clientSecretAuth: z.omit(azureOpenAiAuth.shape.clientSecretAuth, { azureClientSecret: true }),\n }),\n resourceName: z.string(),\n});\n\nexport const azureOpenAiModelProviderConfigUpdateSchema = z.object({\n auth: z.optional(\n z.union([\n azureOpenAiModelProviderConfigSchema.shape.auth,\n z.object({\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.object({});\n\n/**\n * Google Gemini\n */\nexport const googleGeminiModelProviderConfigSchema = z.object({\n accessKey: z.string(),\n});\n\nexport const googleGeminiModelProviderConfigRetrieveSchema = z.object({});\n\n/**\n * OpenAI\n */\nexport const openAiModelProviderConfigSchema = z.object({\n accessKey: z.string(),\n organizationId: z.string(),\n region: z.optional(z.enum([\"GLOBAL\", \"US\", \"EU\"])),\n});\n\nexport const openAiModelProviderConfigRetrieveSchema = z.object({\n organizationId: z.string(),\n region: z.enum([\"GLOBAL\", \"US\", \"EU\"]),\n});\n\n/**\n * OpenAI Compatible\n */\nexport const openAiCompatibleApiModelProviderConfigSchema = z.object({\n accessKey: z.optional(z.string()),\n completionsApiUrl: z.string(),\n modelParameters: z.optional(z.string()),\n organizationId: z.optional(z.string()),\n});\n\nexport const openAiCompatibleApiModelProviderConfigRetrieveSchema = z.object({\n completionsApiUrl: z.string(),\n modelParameters: z.optional(z.string()),\n organizationId: z.optional(z.string()),\n});\n\nconst listModelsAccessSchema = 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 z.strictObject({\n openAiCompatibleApi: openAiCompatibleApiModelProviderConfigSchema,\n }),\n]);\n\nexport const listModelsPropertiesSchema = z.union([\n z.object({\n access: listModelsAccessSchema,\n }),\n z.object({\n modelProviderName: z.string(),\n }),\n]);\n"]}
|
|
@@ -12,6 +12,7 @@ export declare class DatasetCatalogReference extends BaseCatalogReference {
|
|
|
12
12
|
tags(): Promise<Result<import("../CatalogTags.ts").CatalogTags, import("../../index.ts").HttpError>>;
|
|
13
13
|
}
|
|
14
14
|
export declare const mappedType: {
|
|
15
|
+
readonly OBJECT_TABLE: "DATASET_DIRECT";
|
|
15
16
|
readonly PHYSICAL_DATASET: "DATASET_PROMOTED";
|
|
16
17
|
readonly VIRTUAL_DATASET: "DATASET_VIRTUAL";
|
|
17
18
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"DatasetCatalogReference.js","sourceRoot":"","sources":["../../../../src/oss/catalog/CatalogReferences/DatasetCatalogReference.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EACL,oBAAoB,GAGrB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAG7C,MAAM,OAAO,uBAAwB,SAAQ,oBAAoB;IACtD,IAAI,CAAiD;IAC9D,eAAe,CAAiB;IAChC,OAAO,CAAgB;IAEvB,YACE,UAEC,EACD,cAA8B,EAC9B,MAAqB;QAErB,KAAK,CAAC,UAAU,CAAC,CAAC;QAClB,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAmD,CAAC;IAC3F,CAAC;IAED,IAAI;QACF,OAAO,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC;IACvC,CAAC;CACF;AAED,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,gBAAgB,EAAE,kBAAkB;IACpC,eAAe,EAAE,iBAAiB;CAC1B,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 { Result } from \"ts-results-es\";\nimport {\n BaseCatalogReference,\n type BaseCatalogReferenceProperties,\n type RetrieveByPath,\n} from \"./BaseCatalogReference.ts\";\nimport type { DatasetCatalogObject } from \"../CatalogObjects/DatasetCatalogObject.ts\";\nimport { getTags } from \"./utils/getTags.ts\";\nimport type { SonarV3Config } from \"../../../common/Config.ts\";\n\nexport class DatasetCatalogReference extends BaseCatalogReference {\n readonly type: `DATASET_${\"DIRECT\" | \"PROMOTED\" | \"VIRTUAL\"}`;\n #retrieveByPath: RetrieveByPath;\n #config: SonarV3Config;\n\n constructor(\n properties: BaseCatalogReferenceProperties & {\n type: `DATASET_${\"DIRECT\" | \"PROMOTED\" | \"VIRTUAL\"}`;\n },\n retrieveByPath: RetrieveByPath,\n config: SonarV3Config,\n ) {\n super(properties);\n this.type = properties.type;\n this.#retrieveByPath = retrieveByPath;\n this.#config = config;\n }\n\n catalogObject() {\n return this.#retrieveByPath(this.path) as Promise<Result<DatasetCatalogObject, unknown>>;\n }\n\n tags() {\n return getTags(this.#config, this)();\n }\n}\n\nexport const mappedType = {\n PHYSICAL_DATASET: \"DATASET_PROMOTED\",\n VIRTUAL_DATASET: \"DATASET_VIRTUAL\",\n} as const;\n"]}
|
|
1
|
+
{"version":3,"file":"DatasetCatalogReference.js","sourceRoot":"","sources":["../../../../src/oss/catalog/CatalogReferences/DatasetCatalogReference.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EACL,oBAAoB,GAGrB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAG7C,MAAM,OAAO,uBAAwB,SAAQ,oBAAoB;IACtD,IAAI,CAAiD;IAC9D,eAAe,CAAiB;IAChC,OAAO,CAAgB;IAEvB,YACE,UAEC,EACD,cAA8B,EAC9B,MAAqB;QAErB,KAAK,CAAC,UAAU,CAAC,CAAC;QAClB,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAmD,CAAC;IAC3F,CAAC;IAED,IAAI;QACF,OAAO,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC;IACvC,CAAC;CACF;AAED,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,YAAY,EAAE,gBAAgB;IAC9B,gBAAgB,EAAE,kBAAkB;IACpC,eAAe,EAAE,iBAAiB;CAC1B,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 { Result } from \"ts-results-es\";\nimport {\n BaseCatalogReference,\n type BaseCatalogReferenceProperties,\n type RetrieveByPath,\n} from \"./BaseCatalogReference.ts\";\nimport type { DatasetCatalogObject } from \"../CatalogObjects/DatasetCatalogObject.ts\";\nimport { getTags } from \"./utils/getTags.ts\";\nimport type { SonarV3Config } from \"../../../common/Config.ts\";\n\nexport class DatasetCatalogReference extends BaseCatalogReference {\n readonly type: `DATASET_${\"DIRECT\" | \"PROMOTED\" | \"VIRTUAL\"}`;\n #retrieveByPath: RetrieveByPath;\n #config: SonarV3Config;\n\n constructor(\n properties: BaseCatalogReferenceProperties & {\n type: `DATASET_${\"DIRECT\" | \"PROMOTED\" | \"VIRTUAL\"}`;\n },\n retrieveByPath: RetrieveByPath,\n config: SonarV3Config,\n ) {\n super(properties);\n this.type = properties.type;\n this.#retrieveByPath = retrieveByPath;\n this.#config = config;\n }\n\n catalogObject() {\n return this.#retrieveByPath(this.path) as Promise<Result<DatasetCatalogObject, unknown>>;\n }\n\n tags() {\n return getTags(this.#config, this)();\n }\n}\n\nexport const mappedType = {\n OBJECT_TABLE: \"DATASET_DIRECT\",\n PHYSICAL_DATASET: \"DATASET_PROMOTED\",\n VIRTUAL_DATASET: \"DATASET_VIRTUAL\",\n} as const;\n"]}
|