@databricks/sdk-uc-registeredmodels 0.27.0 → 0.35.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/dist/v1/model.js CHANGED
@@ -1,424 +1,299 @@
1
- // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
- import { z } from 'zod';
3
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
4
- export const ModelVersionStatus = {
5
- MODEL_VERSION_STATUS_UNKNOWN: 'MODEL_VERSION_STATUS_UNKNOWN',
6
- /** Request to register a new model version is pending as client uploads model files. */
7
- PENDING_REGISTRATION: 'PENDING_REGISTRATION',
8
- /** Request to register a new model version has failed. */
9
- FAILED_REGISTRATION: 'FAILED_REGISTRATION',
10
- /** Model version is ready for use. */
11
- READY: 'READY',
1
+ import { z } from "zod";
2
+
3
+ //#region src/v1/model.ts
4
+ const ModelVersionStatus = {
5
+ MODEL_VERSION_STATUS_UNKNOWN: "MODEL_VERSION_STATUS_UNKNOWN",
6
+ PENDING_REGISTRATION: "PENDING_REGISTRATION",
7
+ FAILED_REGISTRATION: "FAILED_REGISTRATION",
8
+ READY: "READY"
12
9
  };
13
- export const unmarshalConnectionDependencySchema = z
14
- .object({
15
- connection_name: z.string().optional(),
16
- })
17
- .transform(d => ({
18
- connectionName: d.connection_name,
10
+ const unmarshalConnectionDependencySchema = z.object({ connection_name: z.string().optional() }).transform((d) => ({ connectionName: d.connection_name }));
11
+ const unmarshalCredentialDependencySchema = z.object({ credential_name: z.string().optional() }).transform((d) => ({ credentialName: d.credential_name }));
12
+ const unmarshalDeleteModelVersionResponseSchema = z.object({});
13
+ const unmarshalDeleteRegisteredModelAliasResponseSchema = z.object({});
14
+ const unmarshalDeleteRegisteredModelResponseSchema = z.object({});
15
+ const unmarshalDependencySchema = z.object({
16
+ table: z.lazy(() => unmarshalTableDependencySchema).optional(),
17
+ function: z.lazy(() => unmarshalFunctionDependencySchema).optional(),
18
+ connection: z.lazy(() => unmarshalConnectionDependencySchema).optional(),
19
+ credential: z.lazy(() => unmarshalCredentialDependencySchema).optional()
20
+ }).transform((d) => ({ value: d.table !== void 0 ? {
21
+ $case: "table",
22
+ table: d.table
23
+ } : d.function !== void 0 ? {
24
+ $case: "function",
25
+ function: d.function
26
+ } : d.connection !== void 0 ? {
27
+ $case: "connection",
28
+ connection: d.connection
29
+ } : d.credential !== void 0 ? {
30
+ $case: "credential",
31
+ credential: d.credential
32
+ } : void 0 }));
33
+ const unmarshalDependencyListSchema = z.object({ dependencies: z.array(z.lazy(() => unmarshalDependencySchema)).optional() }).transform((d) => ({ dependencies: d.dependencies }));
34
+ const unmarshalFunctionDependencySchema = z.object({ function_full_name: z.string().optional() }).transform((d) => ({ functionFullName: d.function_full_name }));
35
+ const unmarshalListModelVersionsResponseSchema = z.object({
36
+ model_versions: z.array(z.lazy(() => unmarshalModelVersionInfoSchema)).optional(),
37
+ next_page_token: z.string().optional()
38
+ }).transform((d) => ({
39
+ modelVersions: d.model_versions,
40
+ nextPageToken: d.next_page_token
19
41
  }));
20
- export const unmarshalCredentialDependencySchema = z
21
- .object({
22
- credential_name: z.string().optional(),
23
- })
24
- .transform(d => ({
25
- credentialName: d.credential_name,
42
+ const unmarshalListRegisteredModelsResponseSchema = z.object({
43
+ registered_models: z.array(z.lazy(() => unmarshalRegisteredModelInfoSchema)).optional(),
44
+ next_page_token: z.string().optional()
45
+ }).transform((d) => ({
46
+ registeredModels: d.registered_models,
47
+ nextPageToken: d.next_page_token
26
48
  }));
27
- export const unmarshalDeleteModelVersionResponseSchema = z.object({});
28
- export const unmarshalDeleteRegisteredModelAliasResponseSchema = z.object({});
29
- export const unmarshalDeleteRegisteredModelResponseSchema = z.object({});
30
- export const unmarshalDependencySchema = z
31
- .object({
32
- table: z.lazy(() => unmarshalTableDependencySchema).optional(),
33
- function: z.lazy(() => unmarshalFunctionDependencySchema).optional(),
34
- connection: z.lazy(() => unmarshalConnectionDependencySchema).optional(),
35
- credential: z.lazy(() => unmarshalCredentialDependencySchema).optional(),
36
- })
37
- .transform(d => ({
38
- value: d.table !== undefined
39
- ? { $case: 'table', table: d.table }
40
- : d.function !== undefined
41
- ? { $case: 'function', function: d.function }
42
- : d.connection !== undefined
43
- ? { $case: 'connection', connection: d.connection }
44
- : d.credential !== undefined
45
- ? { $case: 'credential', credential: d.credential }
46
- : undefined,
49
+ const unmarshalModelVersionInfoSchema = z.object({
50
+ model_name: z.string().optional(),
51
+ catalog_name: z.string().optional(),
52
+ schema_name: z.string().optional(),
53
+ source: z.string().optional(),
54
+ comment: z.string().optional(),
55
+ run_id: z.string().optional(),
56
+ run_workspace_id: z.union([z.number(), z.bigint()]).transform((v) => BigInt(v)).optional(),
57
+ model_version_dependencies: z.lazy(() => unmarshalDependencyListSchema).optional(),
58
+ status: z.string().optional(),
59
+ version: z.union([z.number(), z.bigint()]).transform((v) => BigInt(v)).optional(),
60
+ storage_location: z.string().optional(),
61
+ metastore_id: z.string().optional(),
62
+ created_at: z.union([z.number(), z.bigint()]).transform((v) => BigInt(v)).optional(),
63
+ created_by: z.string().optional(),
64
+ updated_at: z.union([z.number(), z.bigint()]).transform((v) => BigInt(v)).optional(),
65
+ updated_by: z.string().optional(),
66
+ id: z.string().optional(),
67
+ aliases: z.array(z.lazy(() => unmarshalRegisteredModelAliasInfoSchema)).optional()
68
+ }).transform((d) => ({
69
+ modelName: d.model_name,
70
+ catalogName: d.catalog_name,
71
+ schemaName: d.schema_name,
72
+ source: d.source,
73
+ comment: d.comment,
74
+ runId: d.run_id,
75
+ runWorkspaceId: d.run_workspace_id,
76
+ modelVersionDependencies: d.model_version_dependencies,
77
+ status: d.status,
78
+ version: d.version,
79
+ storageLocation: d.storage_location,
80
+ metastoreId: d.metastore_id,
81
+ createdAt: d.created_at,
82
+ createdBy: d.created_by,
83
+ updatedAt: d.updated_at,
84
+ updatedBy: d.updated_by,
85
+ id: d.id,
86
+ aliases: d.aliases
47
87
  }));
48
- export const unmarshalDependencyListSchema = z
49
- .object({
50
- dependencies: z.array(z.lazy(() => unmarshalDependencySchema)).optional(),
51
- })
52
- .transform(d => ({
53
- dependencies: d.dependencies,
88
+ const unmarshalRegisteredModelAliasInfoSchema = z.object({
89
+ alias_name: z.string().optional(),
90
+ version_num: z.union([z.number(), z.bigint()]).transform((v) => BigInt(v)).optional(),
91
+ id: z.string().optional(),
92
+ model_name: z.string().optional(),
93
+ catalog_name: z.string().optional(),
94
+ schema_name: z.string().optional()
95
+ }).transform((d) => ({
96
+ aliasName: d.alias_name,
97
+ versionNum: d.version_num,
98
+ id: d.id,
99
+ modelName: d.model_name,
100
+ catalogName: d.catalog_name,
101
+ schemaName: d.schema_name
54
102
  }));
55
- export const unmarshalFunctionDependencySchema = z
56
- .object({
57
- function_full_name: z.string().optional(),
58
- })
59
- .transform(d => ({
60
- functionFullName: d.function_full_name,
103
+ const unmarshalRegisteredModelInfoSchema = z.object({
104
+ name: z.string().optional(),
105
+ catalog_name: z.string().optional(),
106
+ schema_name: z.string().optional(),
107
+ owner: z.string().optional(),
108
+ comment: z.string().optional(),
109
+ storage_location: z.string().optional(),
110
+ metastore_id: z.string().optional(),
111
+ full_name: z.string().optional(),
112
+ created_at: z.union([z.number(), z.bigint()]).transform((v) => BigInt(v)).optional(),
113
+ created_by: z.string().optional(),
114
+ updated_at: z.union([z.number(), z.bigint()]).transform((v) => BigInt(v)).optional(),
115
+ updated_by: z.string().optional(),
116
+ aliases: z.array(z.lazy(() => unmarshalRegisteredModelAliasInfoSchema)).optional(),
117
+ browse_only: z.boolean().optional()
118
+ }).transform((d) => ({
119
+ name: d.name,
120
+ catalogName: d.catalog_name,
121
+ schemaName: d.schema_name,
122
+ owner: d.owner,
123
+ comment: d.comment,
124
+ storageLocation: d.storage_location,
125
+ metastoreId: d.metastore_id,
126
+ fullName: d.full_name,
127
+ createdAt: d.created_at,
128
+ createdBy: d.created_by,
129
+ updatedAt: d.updated_at,
130
+ updatedBy: d.updated_by,
131
+ aliases: d.aliases,
132
+ browseOnly: d.browse_only
61
133
  }));
62
- export const unmarshalListModelVersionsResponseSchema = z
63
- .object({
64
- model_versions: z
65
- .array(z.lazy(() => unmarshalModelVersionInfoSchema))
66
- .optional(),
67
- next_page_token: z.string().optional(),
68
- })
69
- .transform(d => ({
70
- modelVersions: d.model_versions,
71
- nextPageToken: d.next_page_token,
134
+ const unmarshalTableDependencySchema = z.object({ table_full_name: z.string().optional() }).transform((d) => ({ tableFullName: d.table_full_name }));
135
+ const marshalConnectionDependencySchema = z.object({ connectionName: z.string().optional() }).transform((d) => ({ connection_name: d.connectionName }));
136
+ const marshalCreateRegisteredModelRequestSchema = z.object({
137
+ name: z.string().optional(),
138
+ catalogName: z.string().optional(),
139
+ schemaName: z.string().optional(),
140
+ owner: z.string().optional(),
141
+ comment: z.string().optional(),
142
+ storageLocation: z.string().optional(),
143
+ metastoreId: z.string().optional(),
144
+ fullName: z.string().optional(),
145
+ createdAt: z.bigint().optional(),
146
+ createdBy: z.string().optional(),
147
+ updatedAt: z.bigint().optional(),
148
+ updatedBy: z.string().optional(),
149
+ aliases: z.array(z.lazy(() => marshalRegisteredModelAliasInfoSchema)).optional(),
150
+ browseOnly: z.boolean().optional()
151
+ }).transform((d) => ({
152
+ name: d.name,
153
+ catalog_name: d.catalogName,
154
+ schema_name: d.schemaName,
155
+ owner: d.owner,
156
+ comment: d.comment,
157
+ storage_location: d.storageLocation,
158
+ metastore_id: d.metastoreId,
159
+ full_name: d.fullName,
160
+ created_at: d.createdAt,
161
+ created_by: d.createdBy,
162
+ updated_at: d.updatedAt,
163
+ updated_by: d.updatedBy,
164
+ aliases: d.aliases,
165
+ browse_only: d.browseOnly
72
166
  }));
73
- export const unmarshalListRegisteredModelsResponseSchema = z
74
- .object({
75
- registered_models: z
76
- .array(z.lazy(() => unmarshalRegisteredModelInfoSchema))
77
- .optional(),
78
- next_page_token: z.string().optional(),
79
- })
80
- .transform(d => ({
81
- registeredModels: d.registered_models,
82
- nextPageToken: d.next_page_token,
167
+ const marshalCredentialDependencySchema = z.object({ credentialName: z.string().optional() }).transform((d) => ({ credential_name: d.credentialName }));
168
+ const marshalDependencySchema = z.object({ value: z.discriminatedUnion("$case", [
169
+ z.object({
170
+ $case: z.literal("table"),
171
+ table: z.lazy(() => marshalTableDependencySchema)
172
+ }),
173
+ z.object({
174
+ $case: z.literal("function"),
175
+ function: z.lazy(() => marshalFunctionDependencySchema)
176
+ }),
177
+ z.object({
178
+ $case: z.literal("connection"),
179
+ connection: z.lazy(() => marshalConnectionDependencySchema)
180
+ }),
181
+ z.object({
182
+ $case: z.literal("credential"),
183
+ credential: z.lazy(() => marshalCredentialDependencySchema)
184
+ })
185
+ ]).optional() }).transform((d) => ({
186
+ ...d.value?.$case === "table" && { table: d.value.table },
187
+ ...d.value?.$case === "function" && { function: d.value.function },
188
+ ...d.value?.$case === "connection" && { connection: d.value.connection },
189
+ ...d.value?.$case === "credential" && { credential: d.value.credential }
83
190
  }));
84
- export const unmarshalModelVersionInfoSchema = z
85
- .object({
86
- model_name: z.string().optional(),
87
- catalog_name: z.string().optional(),
88
- schema_name: z.string().optional(),
89
- source: z.string().optional(),
90
- comment: z.string().optional(),
91
- run_id: z.string().optional(),
92
- run_workspace_id: z
93
- .union([z.number(), z.bigint()])
94
- .transform(v => BigInt(v))
95
- .optional(),
96
- model_version_dependencies: z
97
- .lazy(() => unmarshalDependencyListSchema)
98
- .optional(),
99
- status: z.string().optional(),
100
- version: z
101
- .union([z.number(), z.bigint()])
102
- .transform(v => BigInt(v))
103
- .optional(),
104
- storage_location: z.string().optional(),
105
- metastore_id: z.string().optional(),
106
- created_at: z
107
- .union([z.number(), z.bigint()])
108
- .transform(v => BigInt(v))
109
- .optional(),
110
- created_by: z.string().optional(),
111
- updated_at: z
112
- .union([z.number(), z.bigint()])
113
- .transform(v => BigInt(v))
114
- .optional(),
115
- updated_by: z.string().optional(),
116
- id: z.string().optional(),
117
- aliases: z
118
- .array(z.lazy(() => unmarshalRegisteredModelAliasInfoSchema))
119
- .optional(),
120
- })
121
- .transform(d => ({
122
- modelName: d.model_name,
123
- catalogName: d.catalog_name,
124
- schemaName: d.schema_name,
125
- source: d.source,
126
- comment: d.comment,
127
- runId: d.run_id,
128
- runWorkspaceId: d.run_workspace_id,
129
- modelVersionDependencies: d.model_version_dependencies,
130
- status: d.status,
131
- version: d.version,
132
- storageLocation: d.storage_location,
133
- metastoreId: d.metastore_id,
134
- createdAt: d.created_at,
135
- createdBy: d.created_by,
136
- updatedAt: d.updated_at,
137
- updatedBy: d.updated_by,
138
- id: d.id,
139
- aliases: d.aliases,
191
+ const marshalDependencyListSchema = z.object({ dependencies: z.array(z.lazy(() => marshalDependencySchema)).optional() }).transform((d) => ({ dependencies: d.dependencies }));
192
+ const marshalFunctionDependencySchema = z.object({ functionFullName: z.string().optional() }).transform((d) => ({ function_full_name: d.functionFullName }));
193
+ const marshalRegisteredModelAliasInfoSchema = z.object({
194
+ aliasName: z.string().optional(),
195
+ versionNum: z.bigint().optional(),
196
+ id: z.string().optional(),
197
+ modelName: z.string().optional(),
198
+ catalogName: z.string().optional(),
199
+ schemaName: z.string().optional()
200
+ }).transform((d) => ({
201
+ alias_name: d.aliasName,
202
+ version_num: d.versionNum,
203
+ id: d.id,
204
+ model_name: d.modelName,
205
+ catalog_name: d.catalogName,
206
+ schema_name: d.schemaName
140
207
  }));
141
- export const unmarshalRegisteredModelAliasInfoSchema = z
142
- .object({
143
- alias_name: z.string().optional(),
144
- version_num: z
145
- .union([z.number(), z.bigint()])
146
- .transform(v => BigInt(v))
147
- .optional(),
148
- id: z.string().optional(),
149
- model_name: z.string().optional(),
150
- catalog_name: z.string().optional(),
151
- schema_name: z.string().optional(),
152
- })
153
- .transform(d => ({
154
- aliasName: d.alias_name,
155
- versionNum: d.version_num,
156
- id: d.id,
157
- modelName: d.model_name,
158
- catalogName: d.catalog_name,
159
- schemaName: d.schema_name,
208
+ const marshalSetRegisteredModelAliasRequestSchema = z.object({
209
+ fullNameArg: z.string().optional(),
210
+ aliasArg: z.string().optional(),
211
+ versionNum: z.bigint().optional()
212
+ }).transform((d) => ({
213
+ full_name_arg: d.fullNameArg,
214
+ alias_arg: d.aliasArg,
215
+ version_num: d.versionNum
160
216
  }));
161
- export const unmarshalRegisteredModelInfoSchema = z
162
- .object({
163
- name: z.string().optional(),
164
- catalog_name: z.string().optional(),
165
- schema_name: z.string().optional(),
166
- owner: z.string().optional(),
167
- comment: z.string().optional(),
168
- storage_location: z.string().optional(),
169
- metastore_id: z.string().optional(),
170
- full_name: z.string().optional(),
171
- created_at: z
172
- .union([z.number(), z.bigint()])
173
- .transform(v => BigInt(v))
174
- .optional(),
175
- created_by: z.string().optional(),
176
- updated_at: z
177
- .union([z.number(), z.bigint()])
178
- .transform(v => BigInt(v))
179
- .optional(),
180
- updated_by: z.string().optional(),
181
- aliases: z
182
- .array(z.lazy(() => unmarshalRegisteredModelAliasInfoSchema))
183
- .optional(),
184
- browse_only: z.boolean().optional(),
185
- })
186
- .transform(d => ({
187
- name: d.name,
188
- catalogName: d.catalog_name,
189
- schemaName: d.schema_name,
190
- owner: d.owner,
191
- comment: d.comment,
192
- storageLocation: d.storage_location,
193
- metastoreId: d.metastore_id,
194
- fullName: d.full_name,
195
- createdAt: d.created_at,
196
- createdBy: d.created_by,
197
- updatedAt: d.updated_at,
198
- updatedBy: d.updated_by,
199
- aliases: d.aliases,
200
- browseOnly: d.browse_only,
217
+ const marshalTableDependencySchema = z.object({ tableFullName: z.string().optional() }).transform((d) => ({ table_full_name: d.tableFullName }));
218
+ const marshalUpdateModelVersionRequestSchema = z.object({
219
+ fullNameArg: z.string().optional(),
220
+ versionArg: z.bigint().optional(),
221
+ modelName: z.string().optional(),
222
+ catalogName: z.string().optional(),
223
+ schemaName: z.string().optional(),
224
+ source: z.string().optional(),
225
+ comment: z.string().optional(),
226
+ runId: z.string().optional(),
227
+ runWorkspaceId: z.bigint().optional(),
228
+ modelVersionDependencies: z.lazy(() => marshalDependencyListSchema).optional(),
229
+ status: z.string().optional(),
230
+ version: z.bigint().optional(),
231
+ storageLocation: z.string().optional(),
232
+ metastoreId: z.string().optional(),
233
+ createdAt: z.bigint().optional(),
234
+ createdBy: z.string().optional(),
235
+ updatedAt: z.bigint().optional(),
236
+ updatedBy: z.string().optional(),
237
+ id: z.string().optional(),
238
+ aliases: z.array(z.lazy(() => marshalRegisteredModelAliasInfoSchema)).optional()
239
+ }).transform((d) => ({
240
+ full_name_arg: d.fullNameArg,
241
+ version_arg: d.versionArg,
242
+ model_name: d.modelName,
243
+ catalog_name: d.catalogName,
244
+ schema_name: d.schemaName,
245
+ source: d.source,
246
+ comment: d.comment,
247
+ run_id: d.runId,
248
+ run_workspace_id: d.runWorkspaceId,
249
+ model_version_dependencies: d.modelVersionDependencies,
250
+ status: d.status,
251
+ version: d.version,
252
+ storage_location: d.storageLocation,
253
+ metastore_id: d.metastoreId,
254
+ created_at: d.createdAt,
255
+ created_by: d.createdBy,
256
+ updated_at: d.updatedAt,
257
+ updated_by: d.updatedBy,
258
+ id: d.id,
259
+ aliases: d.aliases
201
260
  }));
202
- export const unmarshalTableDependencySchema = z
203
- .object({
204
- table_full_name: z.string().optional(),
205
- })
206
- .transform(d => ({
207
- tableFullName: d.table_full_name,
208
- }));
209
- export const marshalConnectionDependencySchema = z
210
- .object({
211
- connectionName: z.string().optional(),
212
- })
213
- .transform(d => ({
214
- connection_name: d.connectionName,
215
- }));
216
- export const marshalCreateRegisteredModelRequestSchema = z
217
- .object({
218
- name: z.string().optional(),
219
- catalogName: z.string().optional(),
220
- schemaName: z.string().optional(),
221
- owner: z.string().optional(),
222
- comment: z.string().optional(),
223
- storageLocation: z.string().optional(),
224
- metastoreId: z.string().optional(),
225
- fullName: z.string().optional(),
226
- createdAt: z.bigint().optional(),
227
- createdBy: z.string().optional(),
228
- updatedAt: z.bigint().optional(),
229
- updatedBy: z.string().optional(),
230
- aliases: z
231
- .array(z.lazy(() => marshalRegisteredModelAliasInfoSchema))
232
- .optional(),
233
- browseOnly: z.boolean().optional(),
234
- })
235
- .transform(d => ({
236
- name: d.name,
237
- catalog_name: d.catalogName,
238
- schema_name: d.schemaName,
239
- owner: d.owner,
240
- comment: d.comment,
241
- storage_location: d.storageLocation,
242
- metastore_id: d.metastoreId,
243
- full_name: d.fullName,
244
- created_at: d.createdAt,
245
- created_by: d.createdBy,
246
- updated_at: d.updatedAt,
247
- updated_by: d.updatedBy,
248
- aliases: d.aliases,
249
- browse_only: d.browseOnly,
250
- }));
251
- export const marshalCredentialDependencySchema = z
252
- .object({
253
- credentialName: z.string().optional(),
254
- })
255
- .transform(d => ({
256
- credential_name: d.credentialName,
257
- }));
258
- export const marshalDependencySchema = z
259
- .object({
260
- value: z
261
- .discriminatedUnion('$case', [
262
- z.object({
263
- $case: z.literal('table'),
264
- table: z.lazy(() => marshalTableDependencySchema),
265
- }),
266
- z.object({
267
- $case: z.literal('function'),
268
- function: z.lazy(() => marshalFunctionDependencySchema),
269
- }),
270
- z.object({
271
- $case: z.literal('connection'),
272
- connection: z.lazy(() => marshalConnectionDependencySchema),
273
- }),
274
- z.object({
275
- $case: z.literal('credential'),
276
- credential: z.lazy(() => marshalCredentialDependencySchema),
277
- }),
278
- ])
279
- .optional(),
280
- })
281
- .transform(d => ({
282
- ...(d.value?.$case === 'table' && { table: d.value.table }),
283
- ...(d.value?.$case === 'function' && { function: d.value.function }),
284
- ...(d.value?.$case === 'connection' && { connection: d.value.connection }),
285
- ...(d.value?.$case === 'credential' && { credential: d.value.credential }),
286
- }));
287
- export const marshalDependencyListSchema = z
288
- .object({
289
- dependencies: z.array(z.lazy(() => marshalDependencySchema)).optional(),
290
- })
291
- .transform(d => ({
292
- dependencies: d.dependencies,
293
- }));
294
- export const marshalFunctionDependencySchema = z
295
- .object({
296
- functionFullName: z.string().optional(),
297
- })
298
- .transform(d => ({
299
- function_full_name: d.functionFullName,
300
- }));
301
- export const marshalRegisteredModelAliasInfoSchema = z
302
- .object({
303
- aliasName: z.string().optional(),
304
- versionNum: z.bigint().optional(),
305
- id: z.string().optional(),
306
- modelName: z.string().optional(),
307
- catalogName: z.string().optional(),
308
- schemaName: z.string().optional(),
309
- })
310
- .transform(d => ({
311
- alias_name: d.aliasName,
312
- version_num: d.versionNum,
313
- id: d.id,
314
- model_name: d.modelName,
315
- catalog_name: d.catalogName,
316
- schema_name: d.schemaName,
317
- }));
318
- export const marshalSetRegisteredModelAliasRequestSchema = z
319
- .object({
320
- fullNameArg: z.string().optional(),
321
- aliasArg: z.string().optional(),
322
- versionNum: z.bigint().optional(),
323
- })
324
- .transform(d => ({
325
- full_name_arg: d.fullNameArg,
326
- alias_arg: d.aliasArg,
327
- version_num: d.versionNum,
328
- }));
329
- export const marshalTableDependencySchema = z
330
- .object({
331
- tableFullName: z.string().optional(),
332
- })
333
- .transform(d => ({
334
- table_full_name: d.tableFullName,
335
- }));
336
- export const marshalUpdateModelVersionRequestSchema = z
337
- .object({
338
- fullNameArg: z.string().optional(),
339
- versionArg: z.bigint().optional(),
340
- modelName: z.string().optional(),
341
- catalogName: z.string().optional(),
342
- schemaName: z.string().optional(),
343
- source: z.string().optional(),
344
- comment: z.string().optional(),
345
- runId: z.string().optional(),
346
- runWorkspaceId: z.bigint().optional(),
347
- modelVersionDependencies: z
348
- .lazy(() => marshalDependencyListSchema)
349
- .optional(),
350
- status: z.string().optional(),
351
- version: z.bigint().optional(),
352
- storageLocation: z.string().optional(),
353
- metastoreId: z.string().optional(),
354
- createdAt: z.bigint().optional(),
355
- createdBy: z.string().optional(),
356
- updatedAt: z.bigint().optional(),
357
- updatedBy: z.string().optional(),
358
- id: z.string().optional(),
359
- aliases: z
360
- .array(z.lazy(() => marshalRegisteredModelAliasInfoSchema))
361
- .optional(),
362
- })
363
- .transform(d => ({
364
- full_name_arg: d.fullNameArg,
365
- version_arg: d.versionArg,
366
- model_name: d.modelName,
367
- catalog_name: d.catalogName,
368
- schema_name: d.schemaName,
369
- source: d.source,
370
- comment: d.comment,
371
- run_id: d.runId,
372
- run_workspace_id: d.runWorkspaceId,
373
- model_version_dependencies: d.modelVersionDependencies,
374
- status: d.status,
375
- version: d.version,
376
- storage_location: d.storageLocation,
377
- metastore_id: d.metastoreId,
378
- created_at: d.createdAt,
379
- created_by: d.createdBy,
380
- updated_at: d.updatedAt,
381
- updated_by: d.updatedBy,
382
- id: d.id,
383
- aliases: d.aliases,
384
- }));
385
- export const marshalUpdateRegisteredModelRequestSchema = z
386
- .object({
387
- fullNameArg: z.string().optional(),
388
- newName: z.string().optional(),
389
- name: z.string().optional(),
390
- catalogName: z.string().optional(),
391
- schemaName: z.string().optional(),
392
- owner: z.string().optional(),
393
- comment: z.string().optional(),
394
- storageLocation: z.string().optional(),
395
- metastoreId: z.string().optional(),
396
- fullName: z.string().optional(),
397
- createdAt: z.bigint().optional(),
398
- createdBy: z.string().optional(),
399
- updatedAt: z.bigint().optional(),
400
- updatedBy: z.string().optional(),
401
- aliases: z
402
- .array(z.lazy(() => marshalRegisteredModelAliasInfoSchema))
403
- .optional(),
404
- browseOnly: z.boolean().optional(),
405
- })
406
- .transform(d => ({
407
- full_name_arg: d.fullNameArg,
408
- new_name: d.newName,
409
- name: d.name,
410
- catalog_name: d.catalogName,
411
- schema_name: d.schemaName,
412
- owner: d.owner,
413
- comment: d.comment,
414
- storage_location: d.storageLocation,
415
- metastore_id: d.metastoreId,
416
- full_name: d.fullName,
417
- created_at: d.createdAt,
418
- created_by: d.createdBy,
419
- updated_at: d.updatedAt,
420
- updated_by: d.updatedBy,
421
- aliases: d.aliases,
422
- browse_only: d.browseOnly,
261
+ const marshalUpdateRegisteredModelRequestSchema = z.object({
262
+ fullNameArg: z.string().optional(),
263
+ newName: z.string().optional(),
264
+ name: z.string().optional(),
265
+ catalogName: z.string().optional(),
266
+ schemaName: z.string().optional(),
267
+ owner: z.string().optional(),
268
+ comment: z.string().optional(),
269
+ storageLocation: z.string().optional(),
270
+ metastoreId: z.string().optional(),
271
+ fullName: z.string().optional(),
272
+ createdAt: z.bigint().optional(),
273
+ createdBy: z.string().optional(),
274
+ updatedAt: z.bigint().optional(),
275
+ updatedBy: z.string().optional(),
276
+ aliases: z.array(z.lazy(() => marshalRegisteredModelAliasInfoSchema)).optional(),
277
+ browseOnly: z.boolean().optional()
278
+ }).transform((d) => ({
279
+ full_name_arg: d.fullNameArg,
280
+ new_name: d.newName,
281
+ name: d.name,
282
+ catalog_name: d.catalogName,
283
+ schema_name: d.schemaName,
284
+ owner: d.owner,
285
+ comment: d.comment,
286
+ storage_location: d.storageLocation,
287
+ metastore_id: d.metastoreId,
288
+ full_name: d.fullName,
289
+ created_at: d.createdAt,
290
+ created_by: d.createdBy,
291
+ updated_at: d.updatedAt,
292
+ updated_by: d.updatedBy,
293
+ aliases: d.aliases,
294
+ browse_only: d.browseOnly
423
295
  }));
296
+
297
+ //#endregion
298
+ export { ModelVersionStatus, marshalConnectionDependencySchema, marshalCreateRegisteredModelRequestSchema, marshalCredentialDependencySchema, marshalDependencyListSchema, marshalDependencySchema, marshalFunctionDependencySchema, marshalRegisteredModelAliasInfoSchema, marshalSetRegisteredModelAliasRequestSchema, marshalTableDependencySchema, marshalUpdateModelVersionRequestSchema, marshalUpdateRegisteredModelRequestSchema, unmarshalConnectionDependencySchema, unmarshalCredentialDependencySchema, unmarshalDeleteModelVersionResponseSchema, unmarshalDeleteRegisteredModelAliasResponseSchema, unmarshalDeleteRegisteredModelResponseSchema, unmarshalDependencyListSchema, unmarshalDependencySchema, unmarshalFunctionDependencySchema, unmarshalListModelVersionsResponseSchema, unmarshalListRegisteredModelsResponseSchema, unmarshalModelVersionInfoSchema, unmarshalRegisteredModelAliasInfoSchema, unmarshalRegisteredModelInfoSchema, unmarshalTableDependencySchema };
424
299
  //# sourceMappingURL=model.js.map