@databricks/sdk-modelregistry 0.1.0-dev.2 → 0.1.0-dev.4

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
@@ -15,19 +15,19 @@ import { z } from 'zod';
15
15
  *
16
16
  * * `DELETE_COMMENT`: Delete the comment
17
17
  */
18
- export var ActivityAction;
19
- (function (ActivityAction) {
18
+ // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
19
+ export const ActivityAction = {
20
20
  /** Approve a transition request. Available to users with sufficient permissions. */
21
- ActivityAction["APPROVE_TRANSITION_REQUEST"] = "APPROVE_TRANSITION_REQUEST";
21
+ APPROVE_TRANSITION_REQUEST: 'APPROVE_TRANSITION_REQUEST',
22
22
  /** Reject a transition request. Available to users with sufficient permissions. */
23
- ActivityAction["REJECT_TRANSITION_REQUEST"] = "REJECT_TRANSITION_REQUEST";
23
+ REJECT_TRANSITION_REQUEST: 'REJECT_TRANSITION_REQUEST',
24
24
  /** Cancel a transition request. Available to the user who created the request. */
25
- ActivityAction["CANCEL_TRANSITION_REQUEST"] = "CANCEL_TRANSITION_REQUEST";
25
+ CANCEL_TRANSITION_REQUEST: 'CANCEL_TRANSITION_REQUEST',
26
26
  /** Edit the comment */
27
- ActivityAction["EDIT_COMMENT"] = "EDIT_COMMENT";
27
+ EDIT_COMMENT: 'EDIT_COMMENT',
28
28
  /** Delete the comment */
29
- ActivityAction["DELETE_COMMENT"] = "DELETE_COMMENT";
30
- })(ActivityAction || (ActivityAction = {}));
29
+ DELETE_COMMENT: 'DELETE_COMMENT',
30
+ };
31
31
  /**
32
32
  * Type of activity. Valid values are:
33
33
  * * `APPLIED_TRANSITION`: User applied the corresponding stage transition.
@@ -42,23 +42,23 @@ export var ActivityAction;
42
42
  *
43
43
  * * `SYSTEM_TRANSITION`: For events performed as a side effect, such as archiving existing model versions in a stage.
44
44
  */
45
- export var ActivityType;
46
- (function (ActivityType) {
45
+ // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
46
+ export const ActivityType = {
47
47
  /** Indicates that the corresponding stage transition was applied by user. */
48
- ActivityType["APPLIED_TRANSITION"] = "APPLIED_TRANSITION";
48
+ APPLIED_TRANSITION: 'APPLIED_TRANSITION',
49
49
  /** Corresponding stage transition was requested by user. */
50
- ActivityType["REQUESTED_TRANSITION"] = "REQUESTED_TRANSITION";
50
+ REQUESTED_TRANSITION: 'REQUESTED_TRANSITION',
51
51
  /** User cancelled an existing request. */
52
- ActivityType["CANCELLED_REQUEST"] = "CANCELLED_REQUEST";
52
+ CANCELLED_REQUEST: 'CANCELLED_REQUEST',
53
53
  /** Corresponding transition request was approved by user. */
54
- ActivityType["APPROVED_REQUEST"] = "APPROVED_REQUEST";
54
+ APPROVED_REQUEST: 'APPROVED_REQUEST',
55
55
  /** Corresponding transition request was rejected by user. */
56
- ActivityType["REJECTED_REQUEST"] = "REJECTED_REQUEST";
56
+ REJECTED_REQUEST: 'REJECTED_REQUEST',
57
57
  /** User posted a new comment */
58
- ActivityType["NEW_COMMENT"] = "NEW_COMMENT";
58
+ NEW_COMMENT: 'NEW_COMMENT',
59
59
  /** Corresponding transition for events such as archiving existing model versions */
60
- ActivityType["SYSTEM_TRANSITION"] = "SYSTEM_TRANSITION";
61
- })(ActivityType || (ActivityType = {}));
60
+ SYSTEM_TRANSITION: 'SYSTEM_TRANSITION',
61
+ };
62
62
  /**
63
63
  * The status of the model version. Valid values are:
64
64
  * * `PENDING_REGISTRATION`: Request to register a new model version is pending as server performs background tasks.
@@ -67,31 +67,31 @@ export var ActivityType;
67
67
  *
68
68
  * * `READY`: Model version is ready for use.
69
69
  */
70
- export var ModelVersionStatus;
71
- (function (ModelVersionStatus) {
70
+ // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
71
+ export const ModelVersionStatus = {
72
72
  /** Request to register a new model version is pending as server performs background tasks. */
73
- ModelVersionStatus["PENDING_REGISTRATION"] = "PENDING_REGISTRATION";
73
+ PENDING_REGISTRATION: 'PENDING_REGISTRATION',
74
74
  /** Request to register a new model version has failed. */
75
- ModelVersionStatus["FAILED_REGISTRATION"] = "FAILED_REGISTRATION";
75
+ FAILED_REGISTRATION: 'FAILED_REGISTRATION',
76
76
  /** Model version is ready for use. */
77
- ModelVersionStatus["READY"] = "READY";
78
- })(ModelVersionStatus || (ModelVersionStatus = {}));
77
+ READY: 'READY',
78
+ };
79
79
  /** Permission level of the requesting user on the object. For what is allowed at each level, see [MLflow Model permissions](..). */
80
- export var PermissionLevel;
81
- (function (PermissionLevel) {
82
- PermissionLevel["CAN_MANAGE"] = "CAN_MANAGE";
80
+ // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
81
+ export const PermissionLevel = {
82
+ CAN_MANAGE: 'CAN_MANAGE',
83
83
  /** reserved 1; // IS_OWNER = 1; was DEPRECATED */
84
- PermissionLevel["CAN_EDIT"] = "CAN_EDIT";
85
- PermissionLevel["CAN_READ"] = "CAN_READ";
86
- PermissionLevel["CAN_MANAGE_STAGING_VERSIONS"] = "CAN_MANAGE_STAGING_VERSIONS";
87
- PermissionLevel["CAN_MANAGE_PRODUCTION_VERSIONS"] = "CAN_MANAGE_PRODUCTION_VERSIONS";
84
+ CAN_EDIT: 'CAN_EDIT',
85
+ CAN_READ: 'CAN_READ',
86
+ CAN_MANAGE_STAGING_VERSIONS: 'CAN_MANAGE_STAGING_VERSIONS',
87
+ CAN_MANAGE_PRODUCTION_VERSIONS: 'CAN_MANAGE_PRODUCTION_VERSIONS',
88
88
  /**
89
89
  * Only applicable to the root ACL path, for which it is the default value if no permissions are
90
90
  * set explicitly for the user. It is the default set by the MLflow service and The ACL database
91
91
  * does not understand this value.
92
92
  */
93
- PermissionLevel["CAN_CREATE_REGISTERED_MODEL"] = "CAN_CREATE_REGISTERED_MODEL";
94
- })(PermissionLevel || (PermissionLevel = {}));
93
+ CAN_CREATE_REGISTERED_MODEL: 'CAN_CREATE_REGISTERED_MODEL',
94
+ };
95
95
  /**
96
96
  * .. note::
97
97
  * Experimental: This entity may change or be removed in a future release without warning.
@@ -101,28 +101,28 @@ export var PermissionLevel;
101
101
  * - `SUBSCRIBED`: Subscribed to notifications.
102
102
  * - `UNSUBSCRIBED`: Not subscribed to notifications.
103
103
  */
104
- export var RegistryEmailSubscriptionType;
105
- (function (RegistryEmailSubscriptionType) {
106
- RegistryEmailSubscriptionType["ALL_EVENTS"] = "ALL_EVENTS";
107
- RegistryEmailSubscriptionType["DEFAULT"] = "DEFAULT";
108
- RegistryEmailSubscriptionType["SUBSCRIBED"] = "SUBSCRIBED";
109
- RegistryEmailSubscriptionType["UNSUBSCRIBED"] = "UNSUBSCRIBED";
110
- })(RegistryEmailSubscriptionType || (RegistryEmailSubscriptionType = {}));
111
- export var RegistryWebhookEvent;
112
- (function (RegistryWebhookEvent) {
113
- RegistryWebhookEvent["MODEL_VERSION_CREATED"] = "MODEL_VERSION_CREATED";
114
- RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_STAGE"] = "MODEL_VERSION_TRANSITIONED_STAGE";
115
- RegistryWebhookEvent["TRANSITION_REQUEST_CREATED"] = "TRANSITION_REQUEST_CREATED";
116
- RegistryWebhookEvent["COMMENT_CREATED"] = "COMMENT_CREATED";
117
- RegistryWebhookEvent["REGISTERED_MODEL_CREATED"] = "REGISTERED_MODEL_CREATED";
118
- RegistryWebhookEvent["MODEL_VERSION_TAG_SET"] = "MODEL_VERSION_TAG_SET";
119
- RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_TO_STAGING"] = "MODEL_VERSION_TRANSITIONED_TO_STAGING";
120
- RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_TO_PRODUCTION"] = "MODEL_VERSION_TRANSITIONED_TO_PRODUCTION";
121
- RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_TO_ARCHIVED"] = "MODEL_VERSION_TRANSITIONED_TO_ARCHIVED";
122
- RegistryWebhookEvent["TRANSITION_REQUEST_TO_STAGING_CREATED"] = "TRANSITION_REQUEST_TO_STAGING_CREATED";
123
- RegistryWebhookEvent["TRANSITION_REQUEST_TO_PRODUCTION_CREATED"] = "TRANSITION_REQUEST_TO_PRODUCTION_CREATED";
124
- RegistryWebhookEvent["TRANSITION_REQUEST_TO_ARCHIVED_CREATED"] = "TRANSITION_REQUEST_TO_ARCHIVED_CREATED";
125
- })(RegistryWebhookEvent || (RegistryWebhookEvent = {}));
104
+ // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
105
+ export const RegistryEmailSubscriptionType = {
106
+ ALL_EVENTS: 'ALL_EVENTS',
107
+ DEFAULT: 'DEFAULT',
108
+ SUBSCRIBED: 'SUBSCRIBED',
109
+ UNSUBSCRIBED: 'UNSUBSCRIBED',
110
+ };
111
+ // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
112
+ export const RegistryWebhookEvent = {
113
+ MODEL_VERSION_CREATED: 'MODEL_VERSION_CREATED',
114
+ MODEL_VERSION_TRANSITIONED_STAGE: 'MODEL_VERSION_TRANSITIONED_STAGE',
115
+ TRANSITION_REQUEST_CREATED: 'TRANSITION_REQUEST_CREATED',
116
+ COMMENT_CREATED: 'COMMENT_CREATED',
117
+ REGISTERED_MODEL_CREATED: 'REGISTERED_MODEL_CREATED',
118
+ MODEL_VERSION_TAG_SET: 'MODEL_VERSION_TAG_SET',
119
+ MODEL_VERSION_TRANSITIONED_TO_STAGING: 'MODEL_VERSION_TRANSITIONED_TO_STAGING',
120
+ MODEL_VERSION_TRANSITIONED_TO_PRODUCTION: 'MODEL_VERSION_TRANSITIONED_TO_PRODUCTION',
121
+ MODEL_VERSION_TRANSITIONED_TO_ARCHIVED: 'MODEL_VERSION_TRANSITIONED_TO_ARCHIVED',
122
+ TRANSITION_REQUEST_TO_STAGING_CREATED: 'TRANSITION_REQUEST_TO_STAGING_CREATED',
123
+ TRANSITION_REQUEST_TO_PRODUCTION_CREATED: 'TRANSITION_REQUEST_TO_PRODUCTION_CREATED',
124
+ TRANSITION_REQUEST_TO_ARCHIVED_CREATED: 'TRANSITION_REQUEST_TO_ARCHIVED_CREATED',
125
+ };
126
126
  /**
127
127
  * Enable or disable triggering the webhook, or put the webhook into test mode. The default is `ACTIVE`:
128
128
  * * `ACTIVE`: Webhook is triggered when an associated event happens.
@@ -131,15 +131,15 @@ export var RegistryWebhookEvent;
131
131
  *
132
132
  * * `TEST_MODE`: Webhook can be triggered through the test endpoint, but is not triggered on a real event.
133
133
  */
134
- export var RegistryWebhookStatus;
135
- (function (RegistryWebhookStatus) {
134
+ // eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
135
+ export const RegistryWebhookStatus = {
136
136
  /** Event and test triggers will be sent. */
137
- RegistryWebhookStatus["ACTIVE"] = "ACTIVE";
137
+ ACTIVE: 'ACTIVE',
138
138
  /** No triggers will be sent. */
139
- RegistryWebhookStatus["DISABLED"] = "DISABLED";
139
+ DISABLED: 'DISABLED',
140
140
  /** Test triggers will be sent, but not actual events. */
141
- RegistryWebhookStatus["TEST_MODE"] = "TEST_MODE";
142
- })(RegistryWebhookStatus || (RegistryWebhookStatus = {}));
141
+ TEST_MODE: 'TEST_MODE',
142
+ };
143
143
  export const unmarshalActivitySchema = z
144
144
  .object({
145
145
  creation_timestamp: z
@@ -147,7 +147,7 @@ export const unmarshalActivitySchema = z
147
147
  .transform(v => BigInt(v))
148
148
  .optional(),
149
149
  user_id: z.string().optional(),
150
- activity_type: z.enum(ActivityType).optional(),
150
+ activity_type: z.string().optional(),
151
151
  comment: z.string().optional(),
152
152
  last_updated_timestamp: z
153
153
  .union([z.number(), z.bigint()])
@@ -156,7 +156,7 @@ export const unmarshalActivitySchema = z
156
156
  from_stage: z.string().optional(),
157
157
  to_stage: z.string().optional(),
158
158
  system_comment: z.string().optional(),
159
- available_actions: z.array(z.enum(ActivityAction)).optional(),
159
+ available_actions: z.array(z.string()).optional(),
160
160
  id: z.string().optional(),
161
161
  })
162
162
  .transform(d => ({
@@ -171,8 +171,7 @@ export const unmarshalActivitySchema = z
171
171
  availableActions: d.available_actions,
172
172
  id: d.id,
173
173
  }));
174
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
175
- export const unmarshalApproveTransitionRequest_ResponseSchema = z
174
+ export const unmarshalApproveTransitionResponseSchema = z
176
175
  .object({
177
176
  activity: z.lazy(() => unmarshalActivitySchema).optional(),
178
177
  })
@@ -186,7 +185,7 @@ export const unmarshalCommentObjectSchema = z
186
185
  .transform(v => BigInt(v))
187
186
  .optional(),
188
187
  user_id: z.string().optional(),
189
- activity_type: z.enum(ActivityType).optional(),
188
+ activity_type: z.string().optional(),
190
189
  comment: z.string().optional(),
191
190
  last_updated_timestamp: z
192
191
  .union([z.number(), z.bigint()])
@@ -195,7 +194,7 @@ export const unmarshalCommentObjectSchema = z
195
194
  from_stage: z.string().optional(),
196
195
  to_stage: z.string().optional(),
197
196
  system_comment: z.string().optional(),
198
- available_actions: z.array(z.enum(ActivityAction)).optional(),
197
+ available_actions: z.array(z.string()).optional(),
199
198
  id: z.string().optional(),
200
199
  })
201
200
  .transform(d => ({
@@ -210,60 +209,48 @@ export const unmarshalCommentObjectSchema = z
210
209
  availableActions: d.available_actions,
211
210
  id: d.id,
212
211
  }));
213
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
214
- export const unmarshalCreateCommentRequest_ResponseSchema = z
212
+ export const unmarshalCreateCommentResponseSchema = z
215
213
  .object({
216
214
  comment: z.lazy(() => unmarshalCommentObjectSchema).optional(),
217
215
  })
218
216
  .transform(d => ({
219
217
  comment: d.comment,
220
218
  }));
221
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
222
- export const unmarshalCreateModelVersionRequest_ResponseSchema = z
219
+ export const unmarshalCreateModelVersionResponseSchema = z
223
220
  .object({
224
221
  model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
225
222
  })
226
223
  .transform(d => ({
227
224
  modelVersion: d.model_version,
228
225
  }));
229
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
230
- export const unmarshalCreateRegisteredModelRequest_ResponseSchema = z
226
+ export const unmarshalCreateRegisteredModelResponseSchema = z
231
227
  .object({
232
228
  registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
233
229
  })
234
230
  .transform(d => ({
235
231
  registeredModel: d.registered_model,
236
232
  }));
237
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
238
- export const unmarshalCreateRegistryWebhookRequest_ResponseSchema = z
233
+ export const unmarshalCreateRegistryWebhookResponseSchema = z
239
234
  .object({
240
235
  webhook: z.lazy(() => unmarshalRegistryWebhookSchema).optional(),
241
236
  })
242
237
  .transform(d => ({
243
238
  webhook: d.webhook,
244
239
  }));
245
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
246
- export const unmarshalCreateTransitionRequest_ResponseSchema = z
240
+ export const unmarshalCreateTransitionResponseSchema = z
247
241
  .object({
248
242
  request: z.lazy(() => unmarshalTransitionRequestSchema).optional(),
249
243
  })
250
244
  .transform(d => ({
251
245
  request: d.request,
252
246
  }));
253
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
254
- export const unmarshalDeleteCommentRequest_ResponseSchema = z.object({});
255
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
256
- export const unmarshalDeleteModelVersionRequest_ResponseSchema = z.object({});
257
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
258
- export const unmarshalDeleteModelVersionTagRequest_ResponseSchema = z.object({});
259
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
260
- export const unmarshalDeleteRegisteredModelRequest_ResponseSchema = z.object({});
261
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
262
- export const unmarshalDeleteRegisteredModelTagRequest_ResponseSchema = z.object({});
263
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
264
- export const unmarshalDeleteRegistryWebhookRequest_ResponseSchema = z.object({});
265
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
266
- export const unmarshalDeleteTransitionRequest_ResponseSchema = z
247
+ export const unmarshalDeleteCommentResponseSchema = z.object({});
248
+ export const unmarshalDeleteModelVersionResponseSchema = z.object({});
249
+ export const unmarshalDeleteModelVersionTagResponseSchema = z.object({});
250
+ export const unmarshalDeleteRegisteredModelResponseSchema = z.object({});
251
+ export const unmarshalDeleteRegisteredModelTagResponseSchema = z.object({});
252
+ export const unmarshalDeleteRegistryWebhookResponseSchema = z.object({});
253
+ export const unmarshalDeleteTransitionResponseSchema = z
267
254
  .object({
268
255
  activity: z.lazy(() => unmarshalActivitySchema).optional(),
269
256
  })
@@ -277,24 +264,30 @@ export const unmarshalFeatureListSchema = z
277
264
  .transform(d => ({
278
265
  features: d.features,
279
266
  }));
280
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
281
- export const unmarshalGetModelVersionDownloadUriRequest_ResponseSchema = z
267
+ export const unmarshalGetLatestVersionsResponseSchema = z
268
+ .object({
269
+ model_versions: z
270
+ .array(z.lazy(() => unmarshalModelVersionSchema))
271
+ .optional(),
272
+ })
273
+ .transform(d => ({
274
+ modelVersions: d.model_versions,
275
+ }));
276
+ export const unmarshalGetModelVersionDownloadUriResponseSchema = z
282
277
  .object({
283
278
  artifact_uri: z.string().optional(),
284
279
  })
285
280
  .transform(d => ({
286
281
  artifactUri: d.artifact_uri,
287
282
  }));
288
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
289
- export const unmarshalGetModelVersionRequest_ResponseSchema = z
283
+ export const unmarshalGetModelVersionResponseSchema = z
290
284
  .object({
291
285
  model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
292
286
  })
293
287
  .transform(d => ({
294
288
  modelVersion: d.model_version,
295
289
  }));
296
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
297
- export const unmarshalGetRegisteredModelDatabricksRequest_ResponseSchema = z
290
+ export const unmarshalGetRegisteredModelDatabricksResponseSchema = z
298
291
  .object({
299
292
  registered_model_databricks: z
300
293
  .lazy(() => unmarshalRegisteredModelDatabricksSchema)
@@ -338,18 +331,7 @@ export const unmarshalLinkedFeatureSchema = z
338
331
  featureName: d.feature_name,
339
332
  featureTableId: d.feature_table_id,
340
333
  }));
341
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
342
- export const unmarshalListLatestVersionsRequest_ResponseSchema = z
343
- .object({
344
- model_versions: z
345
- .array(z.lazy(() => unmarshalModelVersionSchema))
346
- .optional(),
347
- })
348
- .transform(d => ({
349
- modelVersions: d.model_versions,
350
- }));
351
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
352
- export const unmarshalListRegisteredModelsRequest_ResponseSchema = z
334
+ export const unmarshalListRegisteredModelsResponseSchema = z
353
335
  .object({
354
336
  registered_models: z
355
337
  .array(z.lazy(() => unmarshalRegisteredModelSchema))
@@ -360,8 +342,7 @@ export const unmarshalListRegisteredModelsRequest_ResponseSchema = z
360
342
  registeredModels: d.registered_models,
361
343
  nextPageToken: d.next_page_token,
362
344
  }));
363
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
364
- export const unmarshalListRegistryWebhooksRequest_ResponseSchema = z
345
+ export const unmarshalListRegistryWebhooksResponseSchema = z
365
346
  .object({
366
347
  webhooks: z
367
348
  .array(z.lazy(() => unmarshalRegistryWebhookSchema))
@@ -372,8 +353,7 @@ export const unmarshalListRegistryWebhooksRequest_ResponseSchema = z
372
353
  webhooks: d.webhooks,
373
354
  nextPageToken: d.next_page_token,
374
355
  }));
375
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
376
- export const unmarshalListTransitionRequest_ResponseSchema = z
356
+ export const unmarshalListTransitionResponseSchema = z
377
357
  .object({
378
358
  requests: z.array(z.lazy(() => unmarshalActivitySchema)).optional(),
379
359
  })
@@ -397,7 +377,7 @@ export const unmarshalModelVersionSchema = z
397
377
  description: z.string().optional(),
398
378
  source: z.string().optional(),
399
379
  run_id: z.string().optional(),
400
- status: z.enum(ModelVersionStatus).optional(),
380
+ status: z.string().optional(),
401
381
  status_message: z.string().optional(),
402
382
  tags: z.array(z.lazy(() => unmarshalModelVersionTagSchema)).optional(),
403
383
  run_link: z.string().optional(),
@@ -434,15 +414,13 @@ export const unmarshalModelVersionDatabricksSchema = z
434
414
  description: z.string().optional(),
435
415
  source: z.string().optional(),
436
416
  run_id: z.string().optional(),
437
- status: z.enum(ModelVersionStatus).optional(),
417
+ status: z.string().optional(),
438
418
  status_message: z.string().optional(),
439
419
  open_requests: z.array(z.lazy(() => unmarshalActivitySchema)).optional(),
440
- permission_level: z.enum(PermissionLevel).optional(),
420
+ permission_level: z.string().optional(),
441
421
  tags: z.array(z.lazy(() => unmarshalModelVersionTagSchema)).optional(),
442
422
  run_link: z.string().optional(),
443
- email_subscription_status: z
444
- .enum(RegistryEmailSubscriptionType)
445
- .optional(),
423
+ email_subscription_status: z.string().optional(),
446
424
  feature_list: z.lazy(() => unmarshalFeatureListSchema).optional(),
447
425
  })
448
426
  .transform(d => ({
@@ -517,7 +495,7 @@ export const unmarshalRegisteredModelDatabricksSchema = z
517
495
  .array(z.lazy(() => unmarshalModelVersionSchema))
518
496
  .optional(),
519
497
  id: z.string().optional(),
520
- permission_level: z.enum(PermissionLevel).optional(),
498
+ permission_level: z.string().optional(),
521
499
  tags: z.array(z.lazy(() => unmarshalRegisteredModelTagSchema)).optional(),
522
500
  })
523
501
  .transform(d => ({
@@ -543,7 +521,7 @@ export const unmarshalRegisteredModelTagSchema = z
543
521
  export const unmarshalRegistryWebhookSchema = z
544
522
  .object({
545
523
  id: z.string().optional(),
546
- events: z.array(z.enum(RegistryWebhookEvent)).optional(),
524
+ events: z.array(z.string()).optional(),
547
525
  creation_timestamp: z
548
526
  .union([z.number(), z.bigint()])
549
527
  .transform(v => BigInt(v))
@@ -553,7 +531,7 @@ export const unmarshalRegistryWebhookSchema = z
553
531
  .transform(v => BigInt(v))
554
532
  .optional(),
555
533
  description: z.string().optional(),
556
- status: z.enum(RegistryWebhookStatus).optional(),
534
+ status: z.string().optional(),
557
535
  http_url_spec: z.lazy(() => unmarshalHttpUrlSpecSchema).optional(),
558
536
  job_spec: z.lazy(() => unmarshalJobSpecSchema).optional(),
559
537
  model_name: z.string().optional(),
@@ -569,24 +547,21 @@ export const unmarshalRegistryWebhookSchema = z
569
547
  jobSpec: d.job_spec,
570
548
  modelName: d.model_name,
571
549
  }));
572
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
573
- export const unmarshalRejectTransitionRequest_ResponseSchema = z
550
+ export const unmarshalRejectTransitionResponseSchema = z
574
551
  .object({
575
552
  activity: z.lazy(() => unmarshalActivitySchema).optional(),
576
553
  })
577
554
  .transform(d => ({
578
555
  activity: d.activity,
579
556
  }));
580
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
581
- export const unmarshalRenameRegisteredModelRequest_ResponseSchema = z
557
+ export const unmarshalRenameRegisteredModelResponseSchema = z
582
558
  .object({
583
559
  registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
584
560
  })
585
561
  .transform(d => ({
586
562
  registeredModel: d.registered_model,
587
563
  }));
588
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
589
- export const unmarshalSearchModelVersionsRequest_ResponseSchema = z
564
+ export const unmarshalSearchModelVersionsResponseSchema = z
590
565
  .object({
591
566
  model_versions: z
592
567
  .array(z.lazy(() => unmarshalModelVersionSchema))
@@ -597,8 +572,7 @@ export const unmarshalSearchModelVersionsRequest_ResponseSchema = z
597
572
  modelVersions: d.model_versions,
598
573
  nextPageToken: d.next_page_token,
599
574
  }));
600
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
601
- export const unmarshalSearchRegisteredModelsRequest_ResponseSchema = z
575
+ export const unmarshalSearchRegisteredModelsResponseSchema = z
602
576
  .object({
603
577
  registered_models: z
604
578
  .array(z.lazy(() => unmarshalRegisteredModelSchema))
@@ -609,12 +583,9 @@ export const unmarshalSearchRegisteredModelsRequest_ResponseSchema = z
609
583
  registeredModels: d.registered_models,
610
584
  nextPageToken: d.next_page_token,
611
585
  }));
612
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
613
- export const unmarshalSetModelVersionTagRequest_ResponseSchema = z.object({});
614
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
615
- export const unmarshalSetRegisteredModelTagRequest_ResponseSchema = z.object({});
616
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
617
- export const unmarshalTestRegistryWebhookRequest_ResponseSchema = z
586
+ export const unmarshalSetModelVersionTagResponseSchema = z.object({});
587
+ export const unmarshalSetRegisteredModelTagResponseSchema = z.object({});
588
+ export const unmarshalTestRegistryWebhookResponseSchema = z
618
589
  .object({
619
590
  status_code: z.number().optional(),
620
591
  body: z.string().optional(),
@@ -623,8 +594,7 @@ export const unmarshalTestRegistryWebhookRequest_ResponseSchema = z
623
594
  statusCode: d.status_code,
624
595
  body: d.body,
625
596
  }));
626
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
627
- export const unmarshalTransitionModelVersionStageDatabricksRequest_ResponseSchema = z
597
+ export const unmarshalTransitionModelVersionStageDatabricksResponseSchema = z
628
598
  .object({
629
599
  model_version_databricks: z
630
600
  .lazy(() => unmarshalModelVersionDatabricksSchema)
@@ -640,7 +610,7 @@ export const unmarshalTransitionRequestSchema = z
640
610
  .transform(v => BigInt(v))
641
611
  .optional(),
642
612
  user_id: z.string().optional(),
643
- activity_type: z.enum(ActivityType).optional(),
613
+ activity_type: z.string().optional(),
644
614
  comment: z.string().optional(),
645
615
  last_updated_timestamp: z
646
616
  .union([z.number(), z.bigint()])
@@ -649,7 +619,7 @@ export const unmarshalTransitionRequestSchema = z
649
619
  from_stage: z.string().optional(),
650
620
  to_stage: z.string().optional(),
651
621
  system_comment: z.string().optional(),
652
- available_actions: z.array(z.enum(ActivityAction)).optional(),
622
+ available_actions: z.array(z.string()).optional(),
653
623
  id: z.string().optional(),
654
624
  })
655
625
  .transform(d => ({
@@ -664,32 +634,28 @@ export const unmarshalTransitionRequestSchema = z
664
634
  availableActions: d.available_actions,
665
635
  id: d.id,
666
636
  }));
667
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
668
- export const unmarshalUpdateCommentRequest_ResponseSchema = z
637
+ export const unmarshalUpdateCommentResponseSchema = z
669
638
  .object({
670
639
  comment: z.lazy(() => unmarshalCommentObjectSchema).optional(),
671
640
  })
672
641
  .transform(d => ({
673
642
  comment: d.comment,
674
643
  }));
675
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
676
- export const unmarshalUpdateModelVersionRequest_ResponseSchema = z
644
+ export const unmarshalUpdateModelVersionResponseSchema = z
677
645
  .object({
678
646
  model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
679
647
  })
680
648
  .transform(d => ({
681
649
  modelVersion: d.model_version,
682
650
  }));
683
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
684
- export const unmarshalUpdateRegisteredModelRequest_ResponseSchema = z
651
+ export const unmarshalUpdateRegisteredModelResponseSchema = z
685
652
  .object({
686
653
  registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
687
654
  })
688
655
  .transform(d => ({
689
656
  registeredModel: d.registered_model,
690
657
  }));
691
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
692
- export const unmarshalUpdateRegistryWebhookRequest_ResponseSchema = z
658
+ export const unmarshalUpdateRegistryWebhookResponseSchema = z
693
659
  .object({
694
660
  webhook: z.lazy(() => unmarshalRegistryWebhookSchema).optional(),
695
661
  })
@@ -753,9 +719,9 @@ export const marshalCreateRegisteredModelRequestSchema = z
753
719
  export const marshalCreateRegistryWebhookRequestSchema = z
754
720
  .object({
755
721
  modelName: z.string().optional(),
756
- events: z.array(z.enum(RegistryWebhookEvent)).optional(),
722
+ events: z.array(z.string()).optional(),
757
723
  description: z.string().optional(),
758
- status: z.enum(RegistryWebhookStatus).optional(),
724
+ status: z.string().optional(),
759
725
  httpUrlSpec: z.lazy(() => marshalHttpUrlSpecSchema).optional(),
760
726
  jobSpec: z.lazy(() => marshalJobSpecSchema).optional(),
761
727
  })
@@ -880,7 +846,7 @@ export const marshalSetRegisteredModelTagRequestSchema = z
880
846
  export const marshalTestRegistryWebhookRequestSchema = z
881
847
  .object({
882
848
  id: z.string().optional(),
883
- event: z.enum(RegistryWebhookEvent).optional(),
849
+ event: z.string().optional(),
884
850
  })
885
851
  .transform(d => ({
886
852
  id: d.id,
@@ -933,9 +899,9 @@ export const marshalUpdateRegisteredModelRequestSchema = z
933
899
  export const marshalUpdateRegistryWebhookRequestSchema = z
934
900
  .object({
935
901
  id: z.string().optional(),
936
- events: z.array(z.enum(RegistryWebhookEvent)).optional(),
902
+ events: z.array(z.string()).optional(),
937
903
  description: z.string().optional(),
938
- status: z.enum(RegistryWebhookStatus).optional(),
904
+ status: z.string().optional(),
939
905
  httpUrlSpec: z.lazy(() => marshalHttpUrlSpecSchema).optional(),
940
906
  jobSpec: z.lazy(() => marshalJobSpecSchema).optional(),
941
907
  })