@dremio/js-sdk 0.42.0 → 0.43.1

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.
Files changed (115) hide show
  1. package/README.md +274 -0
  2. package/dist/cloud/ai/AIResource.d.ts +716 -17
  3. package/dist/cloud/ai/AIResource.js +40 -18
  4. package/dist/cloud/ai/AIResource.js.map +1 -1
  5. package/dist/cloud/ai/DremioModelProvider.d.ts +1 -1
  6. package/dist/cloud/ai/DremioModelProvider.js +6 -13
  7. package/dist/cloud/ai/DremioModelProvider.js.map +1 -1
  8. package/dist/cloud/ai/modelProviderCodec.d.ts +12 -0
  9. package/dist/cloud/replaceOriginResource.js +4 -0
  10. package/dist/cloud/replaceOriginResource.js.map +1 -1
  11. package/dist/common/ConflictResolver.d.ts +9 -1
  12. package/dist/common/ConflictResolver.js +10 -1
  13. package/dist/common/ConflictResolver.js.map +1 -1
  14. package/dist/common/refineRfc9557Schema.d.ts +1 -0
  15. package/dist/common/refineRfc9557Schema.js +26 -0
  16. package/dist/common/refineRfc9557Schema.js.map +1 -0
  17. package/dist/common/sharedExports.d.ts +1 -0
  18. package/dist/common/sharedExports.js +1 -0
  19. package/dist/common/sharedExports.js.map +1 -1
  20. package/dist/enterprise/Dremio.d.ts +1 -1
  21. package/dist/enterprise/Dremio.js +3 -2
  22. package/dist/enterprise/Dremio.js.map +1 -1
  23. package/dist/enterprise/ai/AIResource.d.ts +699 -18
  24. package/dist/enterprise/ai/AIResource.js +41 -19
  25. package/dist/enterprise/ai/AIResource.js.map +1 -1
  26. package/dist/enterprise/ai/chat/ChatSession.d.ts +4 -25
  27. package/dist/enterprise/ai/chat/ChatSession.js +6 -55
  28. package/dist/enterprise/ai/chat/ChatSession.js.map +1 -1
  29. package/dist/enterprise/ai/chat/UserChatMessage.d.ts +38 -22
  30. package/dist/enterprise/ai/chat/UserChatMessage.js +33 -41
  31. package/dist/enterprise/ai/chat/UserChatMessage.js.map +1 -1
  32. package/dist/enterprise/ai/chat/chatEventSchema.d.ts +169 -11
  33. package/dist/enterprise/ai/chat/chatEventSchema.js +183 -35
  34. package/dist/enterprise/ai/chat/chatEventSchema.js.map +1 -1
  35. package/dist/enterprise/ai/chat/eventWrappers/AgentChatPartialResponse.d.ts +0 -6
  36. package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.d.ts +6 -6
  37. package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.js +2 -2
  38. package/dist/enterprise/ai/chat/eventWrappers/AgentChatResponse.js.map +1 -1
  39. package/dist/enterprise/ai/chat/index.d.ts +0 -3
  40. package/dist/enterprise/ai/chat/index.js +0 -1
  41. package/dist/enterprise/ai/chat/index.js.map +1 -1
  42. package/dist/enterprise/ai/chat/methods/sendChatMessage.js +4 -1
  43. package/dist/enterprise/ai/chat/methods/sendChatMessage.js.map +1 -1
  44. package/dist/enterprise/ai/chat/responseContent/AgentModelResponseContent.d.ts +1 -2
  45. package/dist/enterprise/ai/chat/responseContent/AgentModelResponseContent.js.map +1 -1
  46. package/dist/enterprise/ai/chat/responseContent/AgentToolRequestResponseContent.d.ts +1 -2
  47. package/dist/enterprise/ai/chat/responseContent/AgentToolRequestResponseContent.js +0 -1
  48. package/dist/enterprise/ai/chat/responseContent/AgentToolRequestResponseContent.js.map +1 -1
  49. package/dist/enterprise/ai/chat/responseContent/AgentToolResultResponseContent.d.ts +1 -2
  50. package/dist/enterprise/ai/chat/responseContent/AgentToolResultResponseContent.js +0 -1
  51. package/dist/enterprise/ai/chat/responseContent/AgentToolResultResponseContent.js.map +1 -1
  52. package/dist/enterprise/ai/conversations/AgentConversation.d.ts +215 -0
  53. package/dist/enterprise/ai/conversations/AgentConversation.js +180 -0
  54. package/dist/enterprise/ai/conversations/AgentConversation.js.map +1 -0
  55. package/dist/enterprise/ai/conversations/conversationPropertiesCodec.d.ts +22 -0
  56. package/dist/enterprise/ai/conversations/conversationPropertiesCodec.js +62 -0
  57. package/dist/enterprise/ai/conversations/conversationPropertiesCodec.js.map +1 -0
  58. package/dist/enterprise/ai/conversations/createConversationMachine.d.ts +682 -0
  59. package/dist/enterprise/ai/conversations/createConversationMachine.js +354 -0
  60. package/dist/enterprise/ai/conversations/createConversationMachine.js.map +1 -0
  61. package/dist/enterprise/ai/conversations/methods/createConversation.d.ts +28 -0
  62. package/dist/enterprise/ai/conversations/methods/createConversation.js +68 -0
  63. package/dist/enterprise/ai/conversations/methods/createConversation.js.map +1 -0
  64. package/dist/enterprise/ai/conversations/methods/createExchangeRun.d.ts +16 -0
  65. package/dist/enterprise/ai/conversations/methods/createExchangeRun.js +36 -0
  66. package/dist/enterprise/ai/conversations/methods/createExchangeRun.js.map +1 -0
  67. package/dist/enterprise/ai/conversations/methods/deleteConversation.d.ts +2 -0
  68. package/dist/enterprise/ai/conversations/methods/deleteConversation.js +24 -0
  69. package/dist/enterprise/ai/conversations/methods/deleteConversation.js.map +1 -0
  70. package/dist/enterprise/ai/conversations/methods/listConversations.d.ts +8 -0
  71. package/dist/enterprise/ai/conversations/methods/listConversations.js +30 -0
  72. package/dist/enterprise/ai/conversations/methods/listConversations.js.map +1 -0
  73. package/dist/enterprise/ai/conversations/methods/retrieveConversation.d.ts +12 -0
  74. package/dist/enterprise/ai/conversations/methods/retrieveConversation.js +29 -0
  75. package/dist/enterprise/ai/conversations/methods/retrieveConversation.js.map +1 -0
  76. package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.d.ts +86 -0
  77. package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.js +32 -0
  78. package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.js.map +1 -0
  79. package/dist/enterprise/ai/conversations/methods/stopExchangeRun.d.ts +2 -0
  80. package/dist/enterprise/ai/conversations/methods/stopExchangeRun.js +24 -0
  81. package/dist/enterprise/ai/conversations/methods/stopExchangeRun.js.map +1 -0
  82. package/dist/enterprise/ai/conversations/methods/streamRunEvents.d.ts +6 -0
  83. package/dist/enterprise/ai/conversations/methods/streamRunEvents.js +33 -0
  84. package/dist/enterprise/ai/conversations/methods/streamRunEvents.js.map +1 -0
  85. package/dist/enterprise/ai/conversations/methods/updateConversation.d.ts +15 -0
  86. package/dist/enterprise/ai/conversations/methods/updateConversation.js +35 -0
  87. package/dist/enterprise/ai/conversations/methods/updateConversation.js.map +1 -0
  88. package/dist/enterprise/ai/conversations/reduceChatEvents.d.ts +17 -0
  89. package/dist/enterprise/ai/conversations/reduceChatEvents.js +74 -0
  90. package/dist/enterprise/ai/conversations/reduceChatEvents.js.map +1 -0
  91. package/dist/enterprise/ai/index.d.ts +5 -1
  92. package/dist/enterprise/ai/index.js +4 -1
  93. package/dist/enterprise/ai/index.js.map +1 -1
  94. package/dist/enterprise/ai/modelProvider/ModelProvider.d.ts +15 -1
  95. package/dist/enterprise/ai/modelProvider/ModelProvider.js +12 -1
  96. package/dist/enterprise/ai/modelProvider/ModelProvider.js.map +1 -1
  97. package/dist/enterprise/ai/modelProvider/modelProviderCodec.d.ts +40 -0
  98. package/dist/enterprise/ai/modelProvider/modelProviderCodec.js +10 -1
  99. package/dist/enterprise/ai/modelProvider/modelProviderCodec.js.map +1 -1
  100. package/dist/enterprise/ai/modelProvider/modelProviderConfigSchemas.d.ts +21 -0
  101. package/dist/enterprise/ai/modelProvider/modelProviderConfigSchemas.js +17 -0
  102. package/dist/enterprise/ai/modelProvider/modelProviderConfigSchemas.js.map +1 -1
  103. package/dist/oss/catalog/CatalogReferences/DatasetCatalogReference.d.ts +1 -0
  104. package/dist/oss/catalog/CatalogReferences/DatasetCatalogReference.js +1 -0
  105. package/dist/oss/catalog/CatalogReferences/DatasetCatalogReference.js.map +1 -1
  106. package/dist-iife/cloud.js +21620 -6202
  107. package/dist-iife/community.js +14 -0
  108. package/dist-iife/enterprise.js +19404 -4038
  109. package/package.json +5 -2
  110. package/dist/enterprise/ai/chat/ChatSessionClient.d.ts +0 -18
  111. package/dist/enterprise/ai/chat/ChatSessionClient.js +0 -55
  112. package/dist/enterprise/ai/chat/ChatSessionClient.js.map +0 -1
  113. package/dist/enterprise/ai/chat/UserChatMessageContent.d.ts +0 -19
  114. package/dist/enterprise/ai/chat/UserChatMessageContent.js +0 -44
  115. 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
  };
@@ -33,6 +33,7 @@ export class DatasetCatalogReference extends BaseCatalogReference {
33
33
  }
34
34
  }
35
35
  export const mappedType = {
36
+ OBJECT_TABLE: "DATASET_DIRECT",
36
37
  PHYSICAL_DATASET: "DATASET_PROMOTED",
37
38
  VIRTUAL_DATASET: "DATASET_VIRTUAL",
38
39
  };
@@ -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"]}