@databricks/sdk-modelregistry 0.1.0-dev.2 → 0.1.0-dev.3
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 +70 -70
- package/dist/v1/model.d.ts.map +1 -1
- package/dist/v1/model.js +40 -72
- package/dist/v1/model.js.map +1 -1
- package/package.json +6 -5
- package/src/v1/client.ts +160 -202
- package/src/v1/index.ts +32 -32
- package/src/v1/model.ts +89 -145
package/src/v1/client.ts
CHANGED
|
@@ -18,72 +18,72 @@ import {
|
|
|
18
18
|
import pkgJson from '../../package.json' with {type: 'json'};
|
|
19
19
|
import type {
|
|
20
20
|
ApproveTransitionRequest,
|
|
21
|
-
|
|
21
|
+
ApproveTransitionResponse,
|
|
22
22
|
CreateCommentRequest,
|
|
23
|
-
|
|
23
|
+
CreateCommentResponse,
|
|
24
24
|
CreateModelVersionRequest,
|
|
25
|
-
|
|
25
|
+
CreateModelVersionResponse,
|
|
26
26
|
CreateRegisteredModelRequest,
|
|
27
|
-
|
|
27
|
+
CreateRegisteredModelResponse,
|
|
28
28
|
CreateRegistryWebhookRequest,
|
|
29
|
-
|
|
29
|
+
CreateRegistryWebhookResponse,
|
|
30
30
|
CreateTransitionRequest,
|
|
31
|
-
|
|
31
|
+
CreateTransitionResponse,
|
|
32
32
|
DeleteCommentRequest,
|
|
33
|
-
|
|
33
|
+
DeleteCommentResponse,
|
|
34
34
|
DeleteModelVersionRequest,
|
|
35
|
-
|
|
35
|
+
DeleteModelVersionResponse,
|
|
36
36
|
DeleteModelVersionTagRequest,
|
|
37
|
-
|
|
37
|
+
DeleteModelVersionTagResponse,
|
|
38
38
|
DeleteRegisteredModelRequest,
|
|
39
|
-
|
|
39
|
+
DeleteRegisteredModelResponse,
|
|
40
40
|
DeleteRegisteredModelTagRequest,
|
|
41
|
-
|
|
41
|
+
DeleteRegisteredModelTagResponse,
|
|
42
42
|
DeleteRegistryWebhookRequest,
|
|
43
|
-
|
|
43
|
+
DeleteRegistryWebhookResponse,
|
|
44
44
|
DeleteTransitionRequest,
|
|
45
|
-
|
|
45
|
+
DeleteTransitionResponse,
|
|
46
|
+
GetLatestVersionsResponse,
|
|
46
47
|
GetModelVersionDownloadUriRequest,
|
|
47
|
-
|
|
48
|
+
GetModelVersionDownloadUriResponse,
|
|
48
49
|
GetModelVersionRequest,
|
|
49
|
-
|
|
50
|
+
GetModelVersionResponse,
|
|
50
51
|
GetRegisteredModelDatabricksRequest,
|
|
51
|
-
|
|
52
|
+
GetRegisteredModelDatabricksResponse,
|
|
52
53
|
ListLatestVersionsRequest,
|
|
53
|
-
ListLatestVersionsRequest_Response,
|
|
54
54
|
ListRegisteredModelsRequest,
|
|
55
|
-
|
|
55
|
+
ListRegisteredModelsResponse,
|
|
56
56
|
ListRegistryWebhooksRequest,
|
|
57
|
-
|
|
57
|
+
ListRegistryWebhooksResponse,
|
|
58
58
|
ListTransitionRequest,
|
|
59
|
-
|
|
59
|
+
ListTransitionResponse,
|
|
60
60
|
ModelVersion,
|
|
61
61
|
RegisteredModel,
|
|
62
62
|
RegistryWebhook,
|
|
63
63
|
RejectTransitionRequest,
|
|
64
|
-
|
|
64
|
+
RejectTransitionResponse,
|
|
65
65
|
RenameRegisteredModelRequest,
|
|
66
|
-
|
|
66
|
+
RenameRegisteredModelResponse,
|
|
67
67
|
SearchModelVersionsRequest,
|
|
68
|
-
|
|
68
|
+
SearchModelVersionsResponse,
|
|
69
69
|
SearchRegisteredModelsRequest,
|
|
70
|
-
|
|
70
|
+
SearchRegisteredModelsResponse,
|
|
71
71
|
SetModelVersionTagRequest,
|
|
72
|
-
|
|
72
|
+
SetModelVersionTagResponse,
|
|
73
73
|
SetRegisteredModelTagRequest,
|
|
74
|
-
|
|
74
|
+
SetRegisteredModelTagResponse,
|
|
75
75
|
TestRegistryWebhookRequest,
|
|
76
|
-
|
|
76
|
+
TestRegistryWebhookResponse,
|
|
77
77
|
TransitionModelVersionStageDatabricksRequest,
|
|
78
|
-
|
|
78
|
+
TransitionModelVersionStageDatabricksResponse,
|
|
79
79
|
UpdateCommentRequest,
|
|
80
|
-
|
|
80
|
+
UpdateCommentResponse,
|
|
81
81
|
UpdateModelVersionRequest,
|
|
82
|
-
|
|
82
|
+
UpdateModelVersionResponse,
|
|
83
83
|
UpdateRegisteredModelRequest,
|
|
84
|
-
|
|
84
|
+
UpdateRegisteredModelResponse,
|
|
85
85
|
UpdateRegistryWebhookRequest,
|
|
86
|
-
|
|
86
|
+
UpdateRegistryWebhookResponse,
|
|
87
87
|
} from './model';
|
|
88
88
|
import {
|
|
89
89
|
marshalApproveTransitionRequestSchema,
|
|
@@ -103,38 +103,38 @@ import {
|
|
|
103
103
|
marshalUpdateModelVersionRequestSchema,
|
|
104
104
|
marshalUpdateRegisteredModelRequestSchema,
|
|
105
105
|
marshalUpdateRegistryWebhookRequestSchema,
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
106
|
+
unmarshalApproveTransitionResponseSchema,
|
|
107
|
+
unmarshalCreateCommentResponseSchema,
|
|
108
|
+
unmarshalCreateModelVersionResponseSchema,
|
|
109
|
+
unmarshalCreateRegisteredModelResponseSchema,
|
|
110
|
+
unmarshalCreateRegistryWebhookResponseSchema,
|
|
111
|
+
unmarshalCreateTransitionResponseSchema,
|
|
112
|
+
unmarshalDeleteCommentResponseSchema,
|
|
113
|
+
unmarshalDeleteModelVersionResponseSchema,
|
|
114
|
+
unmarshalDeleteModelVersionTagResponseSchema,
|
|
115
|
+
unmarshalDeleteRegisteredModelResponseSchema,
|
|
116
|
+
unmarshalDeleteRegisteredModelTagResponseSchema,
|
|
117
|
+
unmarshalDeleteRegistryWebhookResponseSchema,
|
|
118
|
+
unmarshalDeleteTransitionResponseSchema,
|
|
119
|
+
unmarshalGetLatestVersionsResponseSchema,
|
|
120
|
+
unmarshalGetModelVersionDownloadUriResponseSchema,
|
|
121
|
+
unmarshalGetModelVersionResponseSchema,
|
|
122
|
+
unmarshalGetRegisteredModelDatabricksResponseSchema,
|
|
123
|
+
unmarshalListRegisteredModelsResponseSchema,
|
|
124
|
+
unmarshalListRegistryWebhooksResponseSchema,
|
|
125
|
+
unmarshalListTransitionResponseSchema,
|
|
126
|
+
unmarshalRejectTransitionResponseSchema,
|
|
127
|
+
unmarshalRenameRegisteredModelResponseSchema,
|
|
128
|
+
unmarshalSearchModelVersionsResponseSchema,
|
|
129
|
+
unmarshalSearchRegisteredModelsResponseSchema,
|
|
130
|
+
unmarshalSetModelVersionTagResponseSchema,
|
|
131
|
+
unmarshalSetRegisteredModelTagResponseSchema,
|
|
132
|
+
unmarshalTestRegistryWebhookResponseSchema,
|
|
133
|
+
unmarshalTransitionModelVersionStageDatabricksResponseSchema,
|
|
134
|
+
unmarshalUpdateCommentResponseSchema,
|
|
135
|
+
unmarshalUpdateModelVersionResponseSchema,
|
|
136
|
+
unmarshalUpdateRegisteredModelResponseSchema,
|
|
137
|
+
unmarshalUpdateRegistryWebhookResponseSchema,
|
|
138
138
|
} from './model';
|
|
139
139
|
|
|
140
140
|
// Package identity segment for this client to be used in the User-Agent header.
|
|
@@ -175,10 +175,10 @@ export class ModelRegistryClient {
|
|
|
175
175
|
async approveTransitionRequest(
|
|
176
176
|
req: ApproveTransitionRequest,
|
|
177
177
|
options?: CallOptions
|
|
178
|
-
): Promise<
|
|
178
|
+
): Promise<ApproveTransitionResponse> {
|
|
179
179
|
const url = `${this.host}/api/2.0/mlflow/transition-requests/approve`;
|
|
180
180
|
const body = marshalRequest(req, marshalApproveTransitionRequestSchema);
|
|
181
|
-
let resp:
|
|
181
|
+
let resp: ApproveTransitionResponse | undefined;
|
|
182
182
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
183
183
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
184
184
|
if (this.workspaceId !== undefined) {
|
|
@@ -191,10 +191,7 @@ export class ModelRegistryClient {
|
|
|
191
191
|
httpClient: this.httpClient,
|
|
192
192
|
logger: this.logger,
|
|
193
193
|
});
|
|
194
|
-
resp = parseResponse(
|
|
195
|
-
respBody,
|
|
196
|
-
unmarshalApproveTransitionRequest_ResponseSchema
|
|
197
|
-
);
|
|
194
|
+
resp = parseResponse(respBody, unmarshalApproveTransitionResponseSchema);
|
|
198
195
|
};
|
|
199
196
|
await executeCall(call, options);
|
|
200
197
|
if (resp === undefined) {
|
|
@@ -210,10 +207,10 @@ export class ModelRegistryClient {
|
|
|
210
207
|
async createComment(
|
|
211
208
|
req: CreateCommentRequest,
|
|
212
209
|
options?: CallOptions
|
|
213
|
-
): Promise<
|
|
210
|
+
): Promise<CreateCommentResponse> {
|
|
214
211
|
const url = `${this.host}/api/2.0/mlflow/comments/create`;
|
|
215
212
|
const body = marshalRequest(req, marshalCreateCommentRequestSchema);
|
|
216
|
-
let resp:
|
|
213
|
+
let resp: CreateCommentResponse | undefined;
|
|
217
214
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
218
215
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
219
216
|
if (this.workspaceId !== undefined) {
|
|
@@ -226,10 +223,7 @@ export class ModelRegistryClient {
|
|
|
226
223
|
httpClient: this.httpClient,
|
|
227
224
|
logger: this.logger,
|
|
228
225
|
});
|
|
229
|
-
resp = parseResponse(
|
|
230
|
-
respBody,
|
|
231
|
-
unmarshalCreateCommentRequest_ResponseSchema
|
|
232
|
-
);
|
|
226
|
+
resp = parseResponse(respBody, unmarshalCreateCommentResponseSchema);
|
|
233
227
|
};
|
|
234
228
|
await executeCall(call, options);
|
|
235
229
|
if (resp === undefined) {
|
|
@@ -245,10 +239,10 @@ export class ModelRegistryClient {
|
|
|
245
239
|
async createRegistryWebhook(
|
|
246
240
|
req: CreateRegistryWebhookRequest,
|
|
247
241
|
options?: CallOptions
|
|
248
|
-
): Promise<
|
|
242
|
+
): Promise<CreateRegistryWebhookResponse> {
|
|
249
243
|
const url = `${this.host}/api/2.0/mlflow/registry-webhooks/create`;
|
|
250
244
|
const body = marshalRequest(req, marshalCreateRegistryWebhookRequestSchema);
|
|
251
|
-
let resp:
|
|
245
|
+
let resp: CreateRegistryWebhookResponse | undefined;
|
|
252
246
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
253
247
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
254
248
|
if (this.workspaceId !== undefined) {
|
|
@@ -263,7 +257,7 @@ export class ModelRegistryClient {
|
|
|
263
257
|
});
|
|
264
258
|
resp = parseResponse(
|
|
265
259
|
respBody,
|
|
266
|
-
|
|
260
|
+
unmarshalCreateRegistryWebhookResponseSchema
|
|
267
261
|
);
|
|
268
262
|
};
|
|
269
263
|
await executeCall(call, options);
|
|
@@ -277,10 +271,10 @@ export class ModelRegistryClient {
|
|
|
277
271
|
async createTransitionRequest(
|
|
278
272
|
req: CreateTransitionRequest,
|
|
279
273
|
options?: CallOptions
|
|
280
|
-
): Promise<
|
|
274
|
+
): Promise<CreateTransitionResponse> {
|
|
281
275
|
const url = `${this.host}/api/2.0/mlflow/transition-requests/create`;
|
|
282
276
|
const body = marshalRequest(req, marshalCreateTransitionRequestSchema);
|
|
283
|
-
let resp:
|
|
277
|
+
let resp: CreateTransitionResponse | undefined;
|
|
284
278
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
285
279
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
286
280
|
if (this.workspaceId !== undefined) {
|
|
@@ -293,10 +287,7 @@ export class ModelRegistryClient {
|
|
|
293
287
|
httpClient: this.httpClient,
|
|
294
288
|
logger: this.logger,
|
|
295
289
|
});
|
|
296
|
-
resp = parseResponse(
|
|
297
|
-
respBody,
|
|
298
|
-
unmarshalCreateTransitionRequest_ResponseSchema
|
|
299
|
-
);
|
|
290
|
+
resp = parseResponse(respBody, unmarshalCreateTransitionResponseSchema);
|
|
300
291
|
};
|
|
301
292
|
await executeCall(call, options);
|
|
302
293
|
if (resp === undefined) {
|
|
@@ -309,7 +300,7 @@ export class ModelRegistryClient {
|
|
|
309
300
|
async deleteComment(
|
|
310
301
|
req: DeleteCommentRequest,
|
|
311
302
|
options?: CallOptions
|
|
312
|
-
): Promise<
|
|
303
|
+
): Promise<DeleteCommentResponse> {
|
|
313
304
|
const url = `${this.host}/api/2.0/mlflow/comments/delete`;
|
|
314
305
|
const params = new URLSearchParams();
|
|
315
306
|
if (req.id !== undefined) {
|
|
@@ -317,7 +308,7 @@ export class ModelRegistryClient {
|
|
|
317
308
|
}
|
|
318
309
|
const query = params.toString();
|
|
319
310
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
320
|
-
let resp:
|
|
311
|
+
let resp: DeleteCommentResponse | undefined;
|
|
321
312
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
322
313
|
const headers = new Headers();
|
|
323
314
|
if (this.workspaceId !== undefined) {
|
|
@@ -330,10 +321,7 @@ export class ModelRegistryClient {
|
|
|
330
321
|
httpClient: this.httpClient,
|
|
331
322
|
logger: this.logger,
|
|
332
323
|
});
|
|
333
|
-
resp = parseResponse(
|
|
334
|
-
respBody,
|
|
335
|
-
unmarshalDeleteCommentRequest_ResponseSchema
|
|
336
|
-
);
|
|
324
|
+
resp = parseResponse(respBody, unmarshalDeleteCommentResponseSchema);
|
|
337
325
|
};
|
|
338
326
|
await executeCall(call, options);
|
|
339
327
|
if (resp === undefined) {
|
|
@@ -349,7 +337,7 @@ export class ModelRegistryClient {
|
|
|
349
337
|
async deleteRegistryWebhook(
|
|
350
338
|
req: DeleteRegistryWebhookRequest,
|
|
351
339
|
options?: CallOptions
|
|
352
|
-
): Promise<
|
|
340
|
+
): Promise<DeleteRegistryWebhookResponse> {
|
|
353
341
|
const url = `${this.host}/api/2.0/mlflow/registry-webhooks/delete`;
|
|
354
342
|
const params = new URLSearchParams();
|
|
355
343
|
if (req.id !== undefined) {
|
|
@@ -357,7 +345,7 @@ export class ModelRegistryClient {
|
|
|
357
345
|
}
|
|
358
346
|
const query = params.toString();
|
|
359
347
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
360
|
-
let resp:
|
|
348
|
+
let resp: DeleteRegistryWebhookResponse | undefined;
|
|
361
349
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
362
350
|
const headers = new Headers();
|
|
363
351
|
if (this.workspaceId !== undefined) {
|
|
@@ -372,7 +360,7 @@ export class ModelRegistryClient {
|
|
|
372
360
|
});
|
|
373
361
|
resp = parseResponse(
|
|
374
362
|
respBody,
|
|
375
|
-
|
|
363
|
+
unmarshalDeleteRegistryWebhookResponseSchema
|
|
376
364
|
);
|
|
377
365
|
};
|
|
378
366
|
await executeCall(call, options);
|
|
@@ -386,7 +374,7 @@ export class ModelRegistryClient {
|
|
|
386
374
|
async deleteTransitionRequest(
|
|
387
375
|
req: DeleteTransitionRequest,
|
|
388
376
|
options?: CallOptions
|
|
389
|
-
): Promise<
|
|
377
|
+
): Promise<DeleteTransitionResponse> {
|
|
390
378
|
const url = `${this.host}/api/2.0/mlflow/transition-requests/delete`;
|
|
391
379
|
const params = new URLSearchParams();
|
|
392
380
|
if (req.name !== undefined) {
|
|
@@ -406,7 +394,7 @@ export class ModelRegistryClient {
|
|
|
406
394
|
}
|
|
407
395
|
const query = params.toString();
|
|
408
396
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
409
|
-
let resp:
|
|
397
|
+
let resp: DeleteTransitionResponse | undefined;
|
|
410
398
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
411
399
|
const headers = new Headers();
|
|
412
400
|
if (this.workspaceId !== undefined) {
|
|
@@ -419,10 +407,7 @@ export class ModelRegistryClient {
|
|
|
419
407
|
httpClient: this.httpClient,
|
|
420
408
|
logger: this.logger,
|
|
421
409
|
});
|
|
422
|
-
resp = parseResponse(
|
|
423
|
-
respBody,
|
|
424
|
-
unmarshalDeleteTransitionRequest_ResponseSchema
|
|
425
|
-
);
|
|
410
|
+
resp = parseResponse(respBody, unmarshalDeleteTransitionResponseSchema);
|
|
426
411
|
};
|
|
427
412
|
await executeCall(call, options);
|
|
428
413
|
if (resp === undefined) {
|
|
@@ -439,7 +424,7 @@ export class ModelRegistryClient {
|
|
|
439
424
|
async getRegisteredModelDatabricks(
|
|
440
425
|
req: GetRegisteredModelDatabricksRequest,
|
|
441
426
|
options?: CallOptions
|
|
442
|
-
): Promise<
|
|
427
|
+
): Promise<GetRegisteredModelDatabricksResponse> {
|
|
443
428
|
const url = `${this.host}/api/2.0/mlflow/databricks/registered-models/get`;
|
|
444
429
|
const params = new URLSearchParams();
|
|
445
430
|
if (req.name !== undefined) {
|
|
@@ -447,7 +432,7 @@ export class ModelRegistryClient {
|
|
|
447
432
|
}
|
|
448
433
|
const query = params.toString();
|
|
449
434
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
450
|
-
let resp:
|
|
435
|
+
let resp: GetRegisteredModelDatabricksResponse | undefined;
|
|
451
436
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
452
437
|
const headers = new Headers();
|
|
453
438
|
if (this.workspaceId !== undefined) {
|
|
@@ -462,7 +447,7 @@ export class ModelRegistryClient {
|
|
|
462
447
|
});
|
|
463
448
|
resp = parseResponse(
|
|
464
449
|
respBody,
|
|
465
|
-
|
|
450
|
+
unmarshalGetRegisteredModelDatabricksResponseSchema
|
|
466
451
|
);
|
|
467
452
|
};
|
|
468
453
|
await executeCall(call, options);
|
|
@@ -479,7 +464,7 @@ export class ModelRegistryClient {
|
|
|
479
464
|
async listRegistryWebhooks(
|
|
480
465
|
req: ListRegistryWebhooksRequest,
|
|
481
466
|
options?: CallOptions
|
|
482
|
-
): Promise<
|
|
467
|
+
): Promise<ListRegistryWebhooksResponse> {
|
|
483
468
|
const url = `${this.host}/api/2.0/mlflow/registry-webhooks/list`;
|
|
484
469
|
const params = new URLSearchParams();
|
|
485
470
|
if (req.modelName !== undefined) {
|
|
@@ -496,7 +481,7 @@ export class ModelRegistryClient {
|
|
|
496
481
|
}
|
|
497
482
|
const query = params.toString();
|
|
498
483
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
499
|
-
let resp:
|
|
484
|
+
let resp: ListRegistryWebhooksResponse | undefined;
|
|
500
485
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
501
486
|
const headers = new Headers();
|
|
502
487
|
if (this.workspaceId !== undefined) {
|
|
@@ -511,7 +496,7 @@ export class ModelRegistryClient {
|
|
|
511
496
|
});
|
|
512
497
|
resp = parseResponse(
|
|
513
498
|
respBody,
|
|
514
|
-
|
|
499
|
+
unmarshalListRegistryWebhooksResponseSchema
|
|
515
500
|
);
|
|
516
501
|
};
|
|
517
502
|
await executeCall(call, options);
|
|
@@ -542,7 +527,7 @@ export class ModelRegistryClient {
|
|
|
542
527
|
async listTransitionsRequest(
|
|
543
528
|
req: ListTransitionRequest,
|
|
544
529
|
options?: CallOptions
|
|
545
|
-
): Promise<
|
|
530
|
+
): Promise<ListTransitionResponse> {
|
|
546
531
|
const url = `${this.host}/api/2.0/mlflow/transition-requests/list`;
|
|
547
532
|
const params = new URLSearchParams();
|
|
548
533
|
if (req.name !== undefined) {
|
|
@@ -553,7 +538,7 @@ export class ModelRegistryClient {
|
|
|
553
538
|
}
|
|
554
539
|
const query = params.toString();
|
|
555
540
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
556
|
-
let resp:
|
|
541
|
+
let resp: ListTransitionResponse | undefined;
|
|
557
542
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
558
543
|
const headers = new Headers();
|
|
559
544
|
if (this.workspaceId !== undefined) {
|
|
@@ -566,10 +551,7 @@ export class ModelRegistryClient {
|
|
|
566
551
|
httpClient: this.httpClient,
|
|
567
552
|
logger: this.logger,
|
|
568
553
|
});
|
|
569
|
-
resp = parseResponse(
|
|
570
|
-
respBody,
|
|
571
|
-
unmarshalListTransitionRequest_ResponseSchema
|
|
572
|
-
);
|
|
554
|
+
resp = parseResponse(respBody, unmarshalListTransitionResponseSchema);
|
|
573
555
|
};
|
|
574
556
|
await executeCall(call, options);
|
|
575
557
|
if (resp === undefined) {
|
|
@@ -582,10 +564,10 @@ export class ModelRegistryClient {
|
|
|
582
564
|
async rejectTransitionRequest(
|
|
583
565
|
req: RejectTransitionRequest,
|
|
584
566
|
options?: CallOptions
|
|
585
|
-
): Promise<
|
|
567
|
+
): Promise<RejectTransitionResponse> {
|
|
586
568
|
const url = `${this.host}/api/2.0/mlflow/transition-requests/reject`;
|
|
587
569
|
const body = marshalRequest(req, marshalRejectTransitionRequestSchema);
|
|
588
|
-
let resp:
|
|
570
|
+
let resp: RejectTransitionResponse | undefined;
|
|
589
571
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
590
572
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
591
573
|
if (this.workspaceId !== undefined) {
|
|
@@ -598,10 +580,7 @@ export class ModelRegistryClient {
|
|
|
598
580
|
httpClient: this.httpClient,
|
|
599
581
|
logger: this.logger,
|
|
600
582
|
});
|
|
601
|
-
resp = parseResponse(
|
|
602
|
-
respBody,
|
|
603
|
-
unmarshalRejectTransitionRequest_ResponseSchema
|
|
604
|
-
);
|
|
583
|
+
resp = parseResponse(respBody, unmarshalRejectTransitionResponseSchema);
|
|
605
584
|
};
|
|
606
585
|
await executeCall(call, options);
|
|
607
586
|
if (resp === undefined) {
|
|
@@ -617,10 +596,10 @@ export class ModelRegistryClient {
|
|
|
617
596
|
async testRegistryWebhook(
|
|
618
597
|
req: TestRegistryWebhookRequest,
|
|
619
598
|
options?: CallOptions
|
|
620
|
-
): Promise<
|
|
599
|
+
): Promise<TestRegistryWebhookResponse> {
|
|
621
600
|
const url = `${this.host}/api/2.0/mlflow/registry-webhooks/test`;
|
|
622
601
|
const body = marshalRequest(req, marshalTestRegistryWebhookRequestSchema);
|
|
623
|
-
let resp:
|
|
602
|
+
let resp: TestRegistryWebhookResponse | undefined;
|
|
624
603
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
625
604
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
626
605
|
if (this.workspaceId !== undefined) {
|
|
@@ -635,7 +614,7 @@ export class ModelRegistryClient {
|
|
|
635
614
|
});
|
|
636
615
|
resp = parseResponse(
|
|
637
616
|
respBody,
|
|
638
|
-
|
|
617
|
+
unmarshalTestRegistryWebhookResponseSchema
|
|
639
618
|
);
|
|
640
619
|
};
|
|
641
620
|
await executeCall(call, options);
|
|
@@ -653,13 +632,13 @@ export class ModelRegistryClient {
|
|
|
653
632
|
async transitionModelVersionStageDatabricks(
|
|
654
633
|
req: TransitionModelVersionStageDatabricksRequest,
|
|
655
634
|
options?: CallOptions
|
|
656
|
-
): Promise<
|
|
635
|
+
): Promise<TransitionModelVersionStageDatabricksResponse> {
|
|
657
636
|
const url = `${this.host}/api/2.0/mlflow/databricks/model-versions/transition-stage`;
|
|
658
637
|
const body = marshalRequest(
|
|
659
638
|
req,
|
|
660
639
|
marshalTransitionModelVersionStageDatabricksRequestSchema
|
|
661
640
|
);
|
|
662
|
-
let resp:
|
|
641
|
+
let resp: TransitionModelVersionStageDatabricksResponse | undefined;
|
|
663
642
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
664
643
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
665
644
|
if (this.workspaceId !== undefined) {
|
|
@@ -674,7 +653,7 @@ export class ModelRegistryClient {
|
|
|
674
653
|
});
|
|
675
654
|
resp = parseResponse(
|
|
676
655
|
respBody,
|
|
677
|
-
|
|
656
|
+
unmarshalTransitionModelVersionStageDatabricksResponseSchema
|
|
678
657
|
);
|
|
679
658
|
};
|
|
680
659
|
await executeCall(call, options);
|
|
@@ -688,10 +667,10 @@ export class ModelRegistryClient {
|
|
|
688
667
|
async updateComment(
|
|
689
668
|
req: UpdateCommentRequest,
|
|
690
669
|
options?: CallOptions
|
|
691
|
-
): Promise<
|
|
670
|
+
): Promise<UpdateCommentResponse> {
|
|
692
671
|
const url = `${this.host}/api/2.0/mlflow/comments/update`;
|
|
693
672
|
const body = marshalRequest(req, marshalUpdateCommentRequestSchema);
|
|
694
|
-
let resp:
|
|
673
|
+
let resp: UpdateCommentResponse | undefined;
|
|
695
674
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
696
675
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
697
676
|
if (this.workspaceId !== undefined) {
|
|
@@ -704,10 +683,7 @@ export class ModelRegistryClient {
|
|
|
704
683
|
httpClient: this.httpClient,
|
|
705
684
|
logger: this.logger,
|
|
706
685
|
});
|
|
707
|
-
resp = parseResponse(
|
|
708
|
-
respBody,
|
|
709
|
-
unmarshalUpdateCommentRequest_ResponseSchema
|
|
710
|
-
);
|
|
686
|
+
resp = parseResponse(respBody, unmarshalUpdateCommentResponseSchema);
|
|
711
687
|
};
|
|
712
688
|
await executeCall(call, options);
|
|
713
689
|
if (resp === undefined) {
|
|
@@ -723,10 +699,10 @@ export class ModelRegistryClient {
|
|
|
723
699
|
async updateRegistryWebhook(
|
|
724
700
|
req: UpdateRegistryWebhookRequest,
|
|
725
701
|
options?: CallOptions
|
|
726
|
-
): Promise<
|
|
702
|
+
): Promise<UpdateRegistryWebhookResponse> {
|
|
727
703
|
const url = `${this.host}/api/2.0/mlflow/registry-webhooks/update`;
|
|
728
704
|
const body = marshalRequest(req, marshalUpdateRegistryWebhookRequestSchema);
|
|
729
|
-
let resp:
|
|
705
|
+
let resp: UpdateRegistryWebhookResponse | undefined;
|
|
730
706
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
731
707
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
732
708
|
if (this.workspaceId !== undefined) {
|
|
@@ -741,7 +717,7 @@ export class ModelRegistryClient {
|
|
|
741
717
|
});
|
|
742
718
|
resp = parseResponse(
|
|
743
719
|
respBody,
|
|
744
|
-
|
|
720
|
+
unmarshalUpdateRegistryWebhookResponseSchema
|
|
745
721
|
);
|
|
746
722
|
};
|
|
747
723
|
await executeCall(call, options);
|
|
@@ -755,10 +731,10 @@ export class ModelRegistryClient {
|
|
|
755
731
|
async createModelVersion(
|
|
756
732
|
req: CreateModelVersionRequest,
|
|
757
733
|
options?: CallOptions
|
|
758
|
-
): Promise<
|
|
734
|
+
): Promise<CreateModelVersionResponse> {
|
|
759
735
|
const url = `${this.host}/api/2.0/mlflow/model-versions/create`;
|
|
760
736
|
const body = marshalRequest(req, marshalCreateModelVersionRequestSchema);
|
|
761
|
-
let resp:
|
|
737
|
+
let resp: CreateModelVersionResponse | undefined;
|
|
762
738
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
763
739
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
764
740
|
if (this.workspaceId !== undefined) {
|
|
@@ -771,10 +747,7 @@ export class ModelRegistryClient {
|
|
|
771
747
|
httpClient: this.httpClient,
|
|
772
748
|
logger: this.logger,
|
|
773
749
|
});
|
|
774
|
-
resp = parseResponse(
|
|
775
|
-
respBody,
|
|
776
|
-
unmarshalCreateModelVersionRequest_ResponseSchema
|
|
777
|
-
);
|
|
750
|
+
resp = parseResponse(respBody, unmarshalCreateModelVersionResponseSchema);
|
|
778
751
|
};
|
|
779
752
|
await executeCall(call, options);
|
|
780
753
|
if (resp === undefined) {
|
|
@@ -790,10 +763,10 @@ export class ModelRegistryClient {
|
|
|
790
763
|
async createRegisteredModel(
|
|
791
764
|
req: CreateRegisteredModelRequest,
|
|
792
765
|
options?: CallOptions
|
|
793
|
-
): Promise<
|
|
766
|
+
): Promise<CreateRegisteredModelResponse> {
|
|
794
767
|
const url = `${this.host}/api/2.0/mlflow/registered-models/create`;
|
|
795
768
|
const body = marshalRequest(req, marshalCreateRegisteredModelRequestSchema);
|
|
796
|
-
let resp:
|
|
769
|
+
let resp: CreateRegisteredModelResponse | undefined;
|
|
797
770
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
798
771
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
799
772
|
if (this.workspaceId !== undefined) {
|
|
@@ -808,7 +781,7 @@ export class ModelRegistryClient {
|
|
|
808
781
|
});
|
|
809
782
|
resp = parseResponse(
|
|
810
783
|
respBody,
|
|
811
|
-
|
|
784
|
+
unmarshalCreateRegisteredModelResponseSchema
|
|
812
785
|
);
|
|
813
786
|
};
|
|
814
787
|
await executeCall(call, options);
|
|
@@ -822,7 +795,7 @@ export class ModelRegistryClient {
|
|
|
822
795
|
async deleteModelVersion(
|
|
823
796
|
req: DeleteModelVersionRequest,
|
|
824
797
|
options?: CallOptions
|
|
825
|
-
): Promise<
|
|
798
|
+
): Promise<DeleteModelVersionResponse> {
|
|
826
799
|
const url = `${this.host}/api/2.0/mlflow/model-versions/delete`;
|
|
827
800
|
const params = new URLSearchParams();
|
|
828
801
|
if (req.name !== undefined) {
|
|
@@ -833,7 +806,7 @@ export class ModelRegistryClient {
|
|
|
833
806
|
}
|
|
834
807
|
const query = params.toString();
|
|
835
808
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
836
|
-
let resp:
|
|
809
|
+
let resp: DeleteModelVersionResponse | undefined;
|
|
837
810
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
838
811
|
const headers = new Headers();
|
|
839
812
|
if (this.workspaceId !== undefined) {
|
|
@@ -846,10 +819,7 @@ export class ModelRegistryClient {
|
|
|
846
819
|
httpClient: this.httpClient,
|
|
847
820
|
logger: this.logger,
|
|
848
821
|
});
|
|
849
|
-
resp = parseResponse(
|
|
850
|
-
respBody,
|
|
851
|
-
unmarshalDeleteModelVersionRequest_ResponseSchema
|
|
852
|
-
);
|
|
822
|
+
resp = parseResponse(respBody, unmarshalDeleteModelVersionResponseSchema);
|
|
853
823
|
};
|
|
854
824
|
await executeCall(call, options);
|
|
855
825
|
if (resp === undefined) {
|
|
@@ -862,7 +832,7 @@ export class ModelRegistryClient {
|
|
|
862
832
|
async deleteModelVersionTag(
|
|
863
833
|
req: DeleteModelVersionTagRequest,
|
|
864
834
|
options?: CallOptions
|
|
865
|
-
): Promise<
|
|
835
|
+
): Promise<DeleteModelVersionTagResponse> {
|
|
866
836
|
const url = `${this.host}/api/2.0/mlflow/model-versions/delete-tag`;
|
|
867
837
|
const params = new URLSearchParams();
|
|
868
838
|
if (req.name !== undefined) {
|
|
@@ -876,7 +846,7 @@ export class ModelRegistryClient {
|
|
|
876
846
|
}
|
|
877
847
|
const query = params.toString();
|
|
878
848
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
879
|
-
let resp:
|
|
849
|
+
let resp: DeleteModelVersionTagResponse | undefined;
|
|
880
850
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
881
851
|
const headers = new Headers();
|
|
882
852
|
if (this.workspaceId !== undefined) {
|
|
@@ -891,7 +861,7 @@ export class ModelRegistryClient {
|
|
|
891
861
|
});
|
|
892
862
|
resp = parseResponse(
|
|
893
863
|
respBody,
|
|
894
|
-
|
|
864
|
+
unmarshalDeleteModelVersionTagResponseSchema
|
|
895
865
|
);
|
|
896
866
|
};
|
|
897
867
|
await executeCall(call, options);
|
|
@@ -905,7 +875,7 @@ export class ModelRegistryClient {
|
|
|
905
875
|
async deleteRegisteredModel(
|
|
906
876
|
req: DeleteRegisteredModelRequest,
|
|
907
877
|
options?: CallOptions
|
|
908
|
-
): Promise<
|
|
878
|
+
): Promise<DeleteRegisteredModelResponse> {
|
|
909
879
|
const url = `${this.host}/api/2.0/mlflow/registered-models/delete`;
|
|
910
880
|
const params = new URLSearchParams();
|
|
911
881
|
if (req.name !== undefined) {
|
|
@@ -913,7 +883,7 @@ export class ModelRegistryClient {
|
|
|
913
883
|
}
|
|
914
884
|
const query = params.toString();
|
|
915
885
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
916
|
-
let resp:
|
|
886
|
+
let resp: DeleteRegisteredModelResponse | undefined;
|
|
917
887
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
918
888
|
const headers = new Headers();
|
|
919
889
|
if (this.workspaceId !== undefined) {
|
|
@@ -928,7 +898,7 @@ export class ModelRegistryClient {
|
|
|
928
898
|
});
|
|
929
899
|
resp = parseResponse(
|
|
930
900
|
respBody,
|
|
931
|
-
|
|
901
|
+
unmarshalDeleteRegisteredModelResponseSchema
|
|
932
902
|
);
|
|
933
903
|
};
|
|
934
904
|
await executeCall(call, options);
|
|
@@ -942,7 +912,7 @@ export class ModelRegistryClient {
|
|
|
942
912
|
async deleteRegisteredModelTag(
|
|
943
913
|
req: DeleteRegisteredModelTagRequest,
|
|
944
914
|
options?: CallOptions
|
|
945
|
-
): Promise<
|
|
915
|
+
): Promise<DeleteRegisteredModelTagResponse> {
|
|
946
916
|
const url = `${this.host}/api/2.0/mlflow/registered-models/delete-tag`;
|
|
947
917
|
const params = new URLSearchParams();
|
|
948
918
|
if (req.name !== undefined) {
|
|
@@ -953,7 +923,7 @@ export class ModelRegistryClient {
|
|
|
953
923
|
}
|
|
954
924
|
const query = params.toString();
|
|
955
925
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
956
|
-
let resp:
|
|
926
|
+
let resp: DeleteRegisteredModelTagResponse | undefined;
|
|
957
927
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
958
928
|
const headers = new Headers();
|
|
959
929
|
if (this.workspaceId !== undefined) {
|
|
@@ -968,7 +938,7 @@ export class ModelRegistryClient {
|
|
|
968
938
|
});
|
|
969
939
|
resp = parseResponse(
|
|
970
940
|
respBody,
|
|
971
|
-
|
|
941
|
+
unmarshalDeleteRegisteredModelTagResponseSchema
|
|
972
942
|
);
|
|
973
943
|
};
|
|
974
944
|
await executeCall(call, options);
|
|
@@ -982,7 +952,7 @@ export class ModelRegistryClient {
|
|
|
982
952
|
async getModelVersion(
|
|
983
953
|
req: GetModelVersionRequest,
|
|
984
954
|
options?: CallOptions
|
|
985
|
-
): Promise<
|
|
955
|
+
): Promise<GetModelVersionResponse> {
|
|
986
956
|
const url = `${this.host}/api/2.0/mlflow/model-versions/get`;
|
|
987
957
|
const params = new URLSearchParams();
|
|
988
958
|
if (req.name !== undefined) {
|
|
@@ -993,7 +963,7 @@ export class ModelRegistryClient {
|
|
|
993
963
|
}
|
|
994
964
|
const query = params.toString();
|
|
995
965
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
996
|
-
let resp:
|
|
966
|
+
let resp: GetModelVersionResponse | undefined;
|
|
997
967
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
998
968
|
const headers = new Headers();
|
|
999
969
|
if (this.workspaceId !== undefined) {
|
|
@@ -1006,10 +976,7 @@ export class ModelRegistryClient {
|
|
|
1006
976
|
httpClient: this.httpClient,
|
|
1007
977
|
logger: this.logger,
|
|
1008
978
|
});
|
|
1009
|
-
resp = parseResponse(
|
|
1010
|
-
respBody,
|
|
1011
|
-
unmarshalGetModelVersionRequest_ResponseSchema
|
|
1012
|
-
);
|
|
979
|
+
resp = parseResponse(respBody, unmarshalGetModelVersionResponseSchema);
|
|
1013
980
|
};
|
|
1014
981
|
await executeCall(call, options);
|
|
1015
982
|
if (resp === undefined) {
|
|
@@ -1022,7 +989,7 @@ export class ModelRegistryClient {
|
|
|
1022
989
|
async getModelVersionDownloadUri(
|
|
1023
990
|
req: GetModelVersionDownloadUriRequest,
|
|
1024
991
|
options?: CallOptions
|
|
1025
|
-
): Promise<
|
|
992
|
+
): Promise<GetModelVersionDownloadUriResponse> {
|
|
1026
993
|
const url = `${this.host}/api/2.0/mlflow/model-versions/get-download-uri`;
|
|
1027
994
|
const params = new URLSearchParams();
|
|
1028
995
|
if (req.name !== undefined) {
|
|
@@ -1033,7 +1000,7 @@ export class ModelRegistryClient {
|
|
|
1033
1000
|
}
|
|
1034
1001
|
const query = params.toString();
|
|
1035
1002
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
1036
|
-
let resp:
|
|
1003
|
+
let resp: GetModelVersionDownloadUriResponse | undefined;
|
|
1037
1004
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1038
1005
|
const headers = new Headers();
|
|
1039
1006
|
if (this.workspaceId !== undefined) {
|
|
@@ -1048,7 +1015,7 @@ export class ModelRegistryClient {
|
|
|
1048
1015
|
});
|
|
1049
1016
|
resp = parseResponse(
|
|
1050
1017
|
respBody,
|
|
1051
|
-
|
|
1018
|
+
unmarshalGetModelVersionDownloadUriResponseSchema
|
|
1052
1019
|
);
|
|
1053
1020
|
};
|
|
1054
1021
|
await executeCall(call, options);
|
|
@@ -1062,10 +1029,10 @@ export class ModelRegistryClient {
|
|
|
1062
1029
|
async listLatestVersions(
|
|
1063
1030
|
req: ListLatestVersionsRequest,
|
|
1064
1031
|
options?: CallOptions
|
|
1065
|
-
): Promise<
|
|
1032
|
+
): Promise<GetLatestVersionsResponse> {
|
|
1066
1033
|
const url = `${this.host}/api/2.0/mlflow/registered-models/get-latest-versions`;
|
|
1067
1034
|
const body = marshalRequest(req, marshalListLatestVersionsRequestSchema);
|
|
1068
|
-
let resp:
|
|
1035
|
+
let resp: GetLatestVersionsResponse | undefined;
|
|
1069
1036
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1070
1037
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
1071
1038
|
if (this.workspaceId !== undefined) {
|
|
@@ -1078,10 +1045,7 @@ export class ModelRegistryClient {
|
|
|
1078
1045
|
httpClient: this.httpClient,
|
|
1079
1046
|
logger: this.logger,
|
|
1080
1047
|
});
|
|
1081
|
-
resp = parseResponse(
|
|
1082
|
-
respBody,
|
|
1083
|
-
unmarshalListLatestVersionsRequest_ResponseSchema
|
|
1084
|
-
);
|
|
1048
|
+
resp = parseResponse(respBody, unmarshalGetLatestVersionsResponseSchema);
|
|
1085
1049
|
};
|
|
1086
1050
|
await executeCall(call, options);
|
|
1087
1051
|
if (resp === undefined) {
|
|
@@ -1094,7 +1058,7 @@ export class ModelRegistryClient {
|
|
|
1094
1058
|
async listRegisteredModels(
|
|
1095
1059
|
req: ListRegisteredModelsRequest,
|
|
1096
1060
|
options?: CallOptions
|
|
1097
|
-
): Promise<
|
|
1061
|
+
): Promise<ListRegisteredModelsResponse> {
|
|
1098
1062
|
const url = `${this.host}/api/2.0/mlflow/registered-models/list`;
|
|
1099
1063
|
const params = new URLSearchParams();
|
|
1100
1064
|
if (req.maxResults !== undefined) {
|
|
@@ -1105,7 +1069,7 @@ export class ModelRegistryClient {
|
|
|
1105
1069
|
}
|
|
1106
1070
|
const query = params.toString();
|
|
1107
1071
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
1108
|
-
let resp:
|
|
1072
|
+
let resp: ListRegisteredModelsResponse | undefined;
|
|
1109
1073
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1110
1074
|
const headers = new Headers();
|
|
1111
1075
|
if (this.workspaceId !== undefined) {
|
|
@@ -1120,7 +1084,7 @@ export class ModelRegistryClient {
|
|
|
1120
1084
|
});
|
|
1121
1085
|
resp = parseResponse(
|
|
1122
1086
|
respBody,
|
|
1123
|
-
|
|
1087
|
+
unmarshalListRegisteredModelsResponseSchema
|
|
1124
1088
|
);
|
|
1125
1089
|
};
|
|
1126
1090
|
await executeCall(call, options);
|
|
@@ -1151,10 +1115,10 @@ export class ModelRegistryClient {
|
|
|
1151
1115
|
async renameRegisteredModel(
|
|
1152
1116
|
req: RenameRegisteredModelRequest,
|
|
1153
1117
|
options?: CallOptions
|
|
1154
|
-
): Promise<
|
|
1118
|
+
): Promise<RenameRegisteredModelResponse> {
|
|
1155
1119
|
const url = `${this.host}/api/2.0/mlflow/registered-models/rename`;
|
|
1156
1120
|
const body = marshalRequest(req, marshalRenameRegisteredModelRequestSchema);
|
|
1157
|
-
let resp:
|
|
1121
|
+
let resp: RenameRegisteredModelResponse | undefined;
|
|
1158
1122
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1159
1123
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
1160
1124
|
if (this.workspaceId !== undefined) {
|
|
@@ -1169,7 +1133,7 @@ export class ModelRegistryClient {
|
|
|
1169
1133
|
});
|
|
1170
1134
|
resp = parseResponse(
|
|
1171
1135
|
respBody,
|
|
1172
|
-
|
|
1136
|
+
unmarshalRenameRegisteredModelResponseSchema
|
|
1173
1137
|
);
|
|
1174
1138
|
};
|
|
1175
1139
|
await executeCall(call, options);
|
|
@@ -1183,7 +1147,7 @@ export class ModelRegistryClient {
|
|
|
1183
1147
|
async searchModelVersions(
|
|
1184
1148
|
req: SearchModelVersionsRequest,
|
|
1185
1149
|
options?: CallOptions
|
|
1186
|
-
): Promise<
|
|
1150
|
+
): Promise<SearchModelVersionsResponse> {
|
|
1187
1151
|
const url = `${this.host}/api/2.0/mlflow/model-versions/search`;
|
|
1188
1152
|
const params = new URLSearchParams();
|
|
1189
1153
|
if (req.filter !== undefined) {
|
|
@@ -1200,7 +1164,7 @@ export class ModelRegistryClient {
|
|
|
1200
1164
|
}
|
|
1201
1165
|
const query = params.toString();
|
|
1202
1166
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
1203
|
-
let resp:
|
|
1167
|
+
let resp: SearchModelVersionsResponse | undefined;
|
|
1204
1168
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1205
1169
|
const headers = new Headers();
|
|
1206
1170
|
if (this.workspaceId !== undefined) {
|
|
@@ -1215,7 +1179,7 @@ export class ModelRegistryClient {
|
|
|
1215
1179
|
});
|
|
1216
1180
|
resp = parseResponse(
|
|
1217
1181
|
respBody,
|
|
1218
|
-
|
|
1182
|
+
unmarshalSearchModelVersionsResponseSchema
|
|
1219
1183
|
);
|
|
1220
1184
|
};
|
|
1221
1185
|
await executeCall(call, options);
|
|
@@ -1246,7 +1210,7 @@ export class ModelRegistryClient {
|
|
|
1246
1210
|
async searchRegisteredModels(
|
|
1247
1211
|
req: SearchRegisteredModelsRequest,
|
|
1248
1212
|
options?: CallOptions
|
|
1249
|
-
): Promise<
|
|
1213
|
+
): Promise<SearchRegisteredModelsResponse> {
|
|
1250
1214
|
const url = `${this.host}/api/2.0/mlflow/registered-models/search`;
|
|
1251
1215
|
const params = new URLSearchParams();
|
|
1252
1216
|
if (req.filter !== undefined) {
|
|
@@ -1263,7 +1227,7 @@ export class ModelRegistryClient {
|
|
|
1263
1227
|
}
|
|
1264
1228
|
const query = params.toString();
|
|
1265
1229
|
const fullUrl = query !== '' ? `${url}?${query}` : url;
|
|
1266
|
-
let resp:
|
|
1230
|
+
let resp: SearchRegisteredModelsResponse | undefined;
|
|
1267
1231
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1268
1232
|
const headers = new Headers();
|
|
1269
1233
|
if (this.workspaceId !== undefined) {
|
|
@@ -1278,7 +1242,7 @@ export class ModelRegistryClient {
|
|
|
1278
1242
|
});
|
|
1279
1243
|
resp = parseResponse(
|
|
1280
1244
|
respBody,
|
|
1281
|
-
|
|
1245
|
+
unmarshalSearchRegisteredModelsResponseSchema
|
|
1282
1246
|
);
|
|
1283
1247
|
};
|
|
1284
1248
|
await executeCall(call, options);
|
|
@@ -1309,10 +1273,10 @@ export class ModelRegistryClient {
|
|
|
1309
1273
|
async setModelVersionTag(
|
|
1310
1274
|
req: SetModelVersionTagRequest,
|
|
1311
1275
|
options?: CallOptions
|
|
1312
|
-
): Promise<
|
|
1276
|
+
): Promise<SetModelVersionTagResponse> {
|
|
1313
1277
|
const url = `${this.host}/api/2.0/mlflow/model-versions/set-tag`;
|
|
1314
1278
|
const body = marshalRequest(req, marshalSetModelVersionTagRequestSchema);
|
|
1315
|
-
let resp:
|
|
1279
|
+
let resp: SetModelVersionTagResponse | undefined;
|
|
1316
1280
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1317
1281
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
1318
1282
|
if (this.workspaceId !== undefined) {
|
|
@@ -1325,10 +1289,7 @@ export class ModelRegistryClient {
|
|
|
1325
1289
|
httpClient: this.httpClient,
|
|
1326
1290
|
logger: this.logger,
|
|
1327
1291
|
});
|
|
1328
|
-
resp = parseResponse(
|
|
1329
|
-
respBody,
|
|
1330
|
-
unmarshalSetModelVersionTagRequest_ResponseSchema
|
|
1331
|
-
);
|
|
1292
|
+
resp = parseResponse(respBody, unmarshalSetModelVersionTagResponseSchema);
|
|
1332
1293
|
};
|
|
1333
1294
|
await executeCall(call, options);
|
|
1334
1295
|
if (resp === undefined) {
|
|
@@ -1341,10 +1302,10 @@ export class ModelRegistryClient {
|
|
|
1341
1302
|
async setRegisteredModelTag(
|
|
1342
1303
|
req: SetRegisteredModelTagRequest,
|
|
1343
1304
|
options?: CallOptions
|
|
1344
|
-
): Promise<
|
|
1305
|
+
): Promise<SetRegisteredModelTagResponse> {
|
|
1345
1306
|
const url = `${this.host}/api/2.0/mlflow/registered-models/set-tag`;
|
|
1346
1307
|
const body = marshalRequest(req, marshalSetRegisteredModelTagRequestSchema);
|
|
1347
|
-
let resp:
|
|
1308
|
+
let resp: SetRegisteredModelTagResponse | undefined;
|
|
1348
1309
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1349
1310
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
1350
1311
|
if (this.workspaceId !== undefined) {
|
|
@@ -1359,7 +1320,7 @@ export class ModelRegistryClient {
|
|
|
1359
1320
|
});
|
|
1360
1321
|
resp = parseResponse(
|
|
1361
1322
|
respBody,
|
|
1362
|
-
|
|
1323
|
+
unmarshalSetRegisteredModelTagResponseSchema
|
|
1363
1324
|
);
|
|
1364
1325
|
};
|
|
1365
1326
|
await executeCall(call, options);
|
|
@@ -1373,10 +1334,10 @@ export class ModelRegistryClient {
|
|
|
1373
1334
|
async updateModelVersion(
|
|
1374
1335
|
req: UpdateModelVersionRequest,
|
|
1375
1336
|
options?: CallOptions
|
|
1376
|
-
): Promise<
|
|
1337
|
+
): Promise<UpdateModelVersionResponse> {
|
|
1377
1338
|
const url = `${this.host}/api/2.0/mlflow/model-versions/update`;
|
|
1378
1339
|
const body = marshalRequest(req, marshalUpdateModelVersionRequestSchema);
|
|
1379
|
-
let resp:
|
|
1340
|
+
let resp: UpdateModelVersionResponse | undefined;
|
|
1380
1341
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1381
1342
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
1382
1343
|
if (this.workspaceId !== undefined) {
|
|
@@ -1389,10 +1350,7 @@ export class ModelRegistryClient {
|
|
|
1389
1350
|
httpClient: this.httpClient,
|
|
1390
1351
|
logger: this.logger,
|
|
1391
1352
|
});
|
|
1392
|
-
resp = parseResponse(
|
|
1393
|
-
respBody,
|
|
1394
|
-
unmarshalUpdateModelVersionRequest_ResponseSchema
|
|
1395
|
-
);
|
|
1353
|
+
resp = parseResponse(respBody, unmarshalUpdateModelVersionResponseSchema);
|
|
1396
1354
|
};
|
|
1397
1355
|
await executeCall(call, options);
|
|
1398
1356
|
if (resp === undefined) {
|
|
@@ -1405,10 +1363,10 @@ export class ModelRegistryClient {
|
|
|
1405
1363
|
async updateRegisteredModel(
|
|
1406
1364
|
req: UpdateRegisteredModelRequest,
|
|
1407
1365
|
options?: CallOptions
|
|
1408
|
-
): Promise<
|
|
1366
|
+
): Promise<UpdateRegisteredModelResponse> {
|
|
1409
1367
|
const url = `${this.host}/api/2.0/mlflow/registered-models/update`;
|
|
1410
1368
|
const body = marshalRequest(req, marshalUpdateRegisteredModelRequestSchema);
|
|
1411
|
-
let resp:
|
|
1369
|
+
let resp: UpdateRegisteredModelResponse | undefined;
|
|
1412
1370
|
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
1413
1371
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
1414
1372
|
if (this.workspaceId !== undefined) {
|
|
@@ -1423,7 +1381,7 @@ export class ModelRegistryClient {
|
|
|
1423
1381
|
});
|
|
1424
1382
|
resp = parseResponse(
|
|
1425
1383
|
respBody,
|
|
1426
|
-
|
|
1384
|
+
unmarshalUpdateRegisteredModelResponseSchema
|
|
1427
1385
|
);
|
|
1428
1386
|
};
|
|
1429
1387
|
await executeCall(call, options);
|