@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/client.d.ts +33 -33
- package/dist/v1/client.d.ts.map +1 -1
- package/dist/v1/client.js +33 -33
- package/dist/v1/client.js.map +1 -1
- package/dist/v1/index.d.ts +1 -1
- package/dist/v1/index.d.ts.map +1 -1
- package/dist/v1/model.d.ts +131 -124
- package/dist/v1/model.d.ts.map +1 -1
- package/dist/v1/model.js +119 -153
- package/dist/v1/model.js.map +1 -1
- package/dist/v1/utils.d.ts.map +1 -1
- package/dist/v1/utils.js +2 -1
- package/dist/v1/utils.js.map +1 -1
- package/package.json +6 -6
- package/src/v1/client.ts +0 -1435
- package/src/v1/index.ts +0 -94
- package/src/v1/model.ts +0 -2049
- package/src/v1/transport.ts +0 -73
- package/src/v1/utils.ts +0 -156
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
|
-
|
|
19
|
-
|
|
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
|
-
|
|
21
|
+
APPROVE_TRANSITION_REQUEST: 'APPROVE_TRANSITION_REQUEST',
|
|
22
22
|
/** Reject a transition request. Available to users with sufficient permissions. */
|
|
23
|
-
|
|
23
|
+
REJECT_TRANSITION_REQUEST: 'REJECT_TRANSITION_REQUEST',
|
|
24
24
|
/** Cancel a transition request. Available to the user who created the request. */
|
|
25
|
-
|
|
25
|
+
CANCEL_TRANSITION_REQUEST: 'CANCEL_TRANSITION_REQUEST',
|
|
26
26
|
/** Edit the comment */
|
|
27
|
-
|
|
27
|
+
EDIT_COMMENT: 'EDIT_COMMENT',
|
|
28
28
|
/** Delete the comment */
|
|
29
|
-
|
|
30
|
-
}
|
|
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
|
-
|
|
46
|
-
|
|
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
|
-
|
|
48
|
+
APPLIED_TRANSITION: 'APPLIED_TRANSITION',
|
|
49
49
|
/** Corresponding stage transition was requested by user. */
|
|
50
|
-
|
|
50
|
+
REQUESTED_TRANSITION: 'REQUESTED_TRANSITION',
|
|
51
51
|
/** User cancelled an existing request. */
|
|
52
|
-
|
|
52
|
+
CANCELLED_REQUEST: 'CANCELLED_REQUEST',
|
|
53
53
|
/** Corresponding transition request was approved by user. */
|
|
54
|
-
|
|
54
|
+
APPROVED_REQUEST: 'APPROVED_REQUEST',
|
|
55
55
|
/** Corresponding transition request was rejected by user. */
|
|
56
|
-
|
|
56
|
+
REJECTED_REQUEST: 'REJECTED_REQUEST',
|
|
57
57
|
/** User posted a new comment */
|
|
58
|
-
|
|
58
|
+
NEW_COMMENT: 'NEW_COMMENT',
|
|
59
59
|
/** Corresponding transition for events such as archiving existing model versions */
|
|
60
|
-
|
|
61
|
-
}
|
|
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
|
-
|
|
71
|
-
|
|
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
|
-
|
|
73
|
+
PENDING_REGISTRATION: 'PENDING_REGISTRATION',
|
|
74
74
|
/** Request to register a new model version has failed. */
|
|
75
|
-
|
|
75
|
+
FAILED_REGISTRATION: 'FAILED_REGISTRATION',
|
|
76
76
|
/** Model version is ready for use. */
|
|
77
|
-
|
|
78
|
-
}
|
|
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
|
-
|
|
81
|
-
|
|
82
|
-
|
|
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
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
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
|
-
|
|
94
|
-
}
|
|
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
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
}
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
}
|
|
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
|
-
|
|
135
|
-
|
|
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
|
-
|
|
137
|
+
ACTIVE: 'ACTIVE',
|
|
138
138
|
/** No triggers will be sent. */
|
|
139
|
-
|
|
139
|
+
DISABLED: 'DISABLED',
|
|
140
140
|
/** Test triggers will be sent, but not actual events. */
|
|
141
|
-
|
|
142
|
-
}
|
|
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.
|
|
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.
|
|
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
|
-
|
|
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.
|
|
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.
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
254
|
-
export const
|
|
255
|
-
|
|
256
|
-
export const
|
|
257
|
-
|
|
258
|
-
export const
|
|
259
|
-
|
|
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
|
-
|
|
281
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
613
|
-
export const
|
|
614
|
-
|
|
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
|
-
|
|
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.
|
|
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.
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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.
|
|
722
|
+
events: z.array(z.string()).optional(),
|
|
757
723
|
description: z.string().optional(),
|
|
758
|
-
status: z.
|
|
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.
|
|
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.
|
|
902
|
+
events: z.array(z.string()).optional(),
|
|
937
903
|
description: z.string().optional(),
|
|
938
|
-
status: z.
|
|
904
|
+
status: z.string().optional(),
|
|
939
905
|
httpUrlSpec: z.lazy(() => marshalHttpUrlSpecSchema).optional(),
|
|
940
906
|
jobSpec: z.lazy(() => marshalJobSpecSchema).optional(),
|
|
941
907
|
})
|