@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/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
- ApproveTransitionRequest_Response,
21
+ ApproveTransitionResponse,
22
22
  CreateCommentRequest,
23
- CreateCommentRequest_Response,
23
+ CreateCommentResponse,
24
24
  CreateModelVersionRequest,
25
- CreateModelVersionRequest_Response,
25
+ CreateModelVersionResponse,
26
26
  CreateRegisteredModelRequest,
27
- CreateRegisteredModelRequest_Response,
27
+ CreateRegisteredModelResponse,
28
28
  CreateRegistryWebhookRequest,
29
- CreateRegistryWebhookRequest_Response,
29
+ CreateRegistryWebhookResponse,
30
30
  CreateTransitionRequest,
31
- CreateTransitionRequest_Response,
31
+ CreateTransitionResponse,
32
32
  DeleteCommentRequest,
33
- DeleteCommentRequest_Response,
33
+ DeleteCommentResponse,
34
34
  DeleteModelVersionRequest,
35
- DeleteModelVersionRequest_Response,
35
+ DeleteModelVersionResponse,
36
36
  DeleteModelVersionTagRequest,
37
- DeleteModelVersionTagRequest_Response,
37
+ DeleteModelVersionTagResponse,
38
38
  DeleteRegisteredModelRequest,
39
- DeleteRegisteredModelRequest_Response,
39
+ DeleteRegisteredModelResponse,
40
40
  DeleteRegisteredModelTagRequest,
41
- DeleteRegisteredModelTagRequest_Response,
41
+ DeleteRegisteredModelTagResponse,
42
42
  DeleteRegistryWebhookRequest,
43
- DeleteRegistryWebhookRequest_Response,
43
+ DeleteRegistryWebhookResponse,
44
44
  DeleteTransitionRequest,
45
- DeleteTransitionRequest_Response,
45
+ DeleteTransitionResponse,
46
+ GetLatestVersionsResponse,
46
47
  GetModelVersionDownloadUriRequest,
47
- GetModelVersionDownloadUriRequest_Response,
48
+ GetModelVersionDownloadUriResponse,
48
49
  GetModelVersionRequest,
49
- GetModelVersionRequest_Response,
50
+ GetModelVersionResponse,
50
51
  GetRegisteredModelDatabricksRequest,
51
- GetRegisteredModelDatabricksRequest_Response,
52
+ GetRegisteredModelDatabricksResponse,
52
53
  ListLatestVersionsRequest,
53
- ListLatestVersionsRequest_Response,
54
54
  ListRegisteredModelsRequest,
55
- ListRegisteredModelsRequest_Response,
55
+ ListRegisteredModelsResponse,
56
56
  ListRegistryWebhooksRequest,
57
- ListRegistryWebhooksRequest_Response,
57
+ ListRegistryWebhooksResponse,
58
58
  ListTransitionRequest,
59
- ListTransitionRequest_Response,
59
+ ListTransitionResponse,
60
60
  ModelVersion,
61
61
  RegisteredModel,
62
62
  RegistryWebhook,
63
63
  RejectTransitionRequest,
64
- RejectTransitionRequest_Response,
64
+ RejectTransitionResponse,
65
65
  RenameRegisteredModelRequest,
66
- RenameRegisteredModelRequest_Response,
66
+ RenameRegisteredModelResponse,
67
67
  SearchModelVersionsRequest,
68
- SearchModelVersionsRequest_Response,
68
+ SearchModelVersionsResponse,
69
69
  SearchRegisteredModelsRequest,
70
- SearchRegisteredModelsRequest_Response,
70
+ SearchRegisteredModelsResponse,
71
71
  SetModelVersionTagRequest,
72
- SetModelVersionTagRequest_Response,
72
+ SetModelVersionTagResponse,
73
73
  SetRegisteredModelTagRequest,
74
- SetRegisteredModelTagRequest_Response,
74
+ SetRegisteredModelTagResponse,
75
75
  TestRegistryWebhookRequest,
76
- TestRegistryWebhookRequest_Response,
76
+ TestRegistryWebhookResponse,
77
77
  TransitionModelVersionStageDatabricksRequest,
78
- TransitionModelVersionStageDatabricksRequest_Response,
78
+ TransitionModelVersionStageDatabricksResponse,
79
79
  UpdateCommentRequest,
80
- UpdateCommentRequest_Response,
80
+ UpdateCommentResponse,
81
81
  UpdateModelVersionRequest,
82
- UpdateModelVersionRequest_Response,
82
+ UpdateModelVersionResponse,
83
83
  UpdateRegisteredModelRequest,
84
- UpdateRegisteredModelRequest_Response,
84
+ UpdateRegisteredModelResponse,
85
85
  UpdateRegistryWebhookRequest,
86
- UpdateRegistryWebhookRequest_Response,
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
- unmarshalApproveTransitionRequest_ResponseSchema,
107
- unmarshalCreateCommentRequest_ResponseSchema,
108
- unmarshalCreateModelVersionRequest_ResponseSchema,
109
- unmarshalCreateRegisteredModelRequest_ResponseSchema,
110
- unmarshalCreateRegistryWebhookRequest_ResponseSchema,
111
- unmarshalCreateTransitionRequest_ResponseSchema,
112
- unmarshalDeleteCommentRequest_ResponseSchema,
113
- unmarshalDeleteModelVersionRequest_ResponseSchema,
114
- unmarshalDeleteModelVersionTagRequest_ResponseSchema,
115
- unmarshalDeleteRegisteredModelRequest_ResponseSchema,
116
- unmarshalDeleteRegisteredModelTagRequest_ResponseSchema,
117
- unmarshalDeleteRegistryWebhookRequest_ResponseSchema,
118
- unmarshalDeleteTransitionRequest_ResponseSchema,
119
- unmarshalGetModelVersionDownloadUriRequest_ResponseSchema,
120
- unmarshalGetModelVersionRequest_ResponseSchema,
121
- unmarshalGetRegisteredModelDatabricksRequest_ResponseSchema,
122
- unmarshalListLatestVersionsRequest_ResponseSchema,
123
- unmarshalListRegisteredModelsRequest_ResponseSchema,
124
- unmarshalListRegistryWebhooksRequest_ResponseSchema,
125
- unmarshalListTransitionRequest_ResponseSchema,
126
- unmarshalRejectTransitionRequest_ResponseSchema,
127
- unmarshalRenameRegisteredModelRequest_ResponseSchema,
128
- unmarshalSearchModelVersionsRequest_ResponseSchema,
129
- unmarshalSearchRegisteredModelsRequest_ResponseSchema,
130
- unmarshalSetModelVersionTagRequest_ResponseSchema,
131
- unmarshalSetRegisteredModelTagRequest_ResponseSchema,
132
- unmarshalTestRegistryWebhookRequest_ResponseSchema,
133
- unmarshalTransitionModelVersionStageDatabricksRequest_ResponseSchema,
134
- unmarshalUpdateCommentRequest_ResponseSchema,
135
- unmarshalUpdateModelVersionRequest_ResponseSchema,
136
- unmarshalUpdateRegisteredModelRequest_ResponseSchema,
137
- unmarshalUpdateRegistryWebhookRequest_ResponseSchema,
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<ApproveTransitionRequest_Response> {
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: ApproveTransitionRequest_Response | undefined;
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<CreateCommentRequest_Response> {
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: CreateCommentRequest_Response | undefined;
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<CreateRegistryWebhookRequest_Response> {
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: CreateRegistryWebhookRequest_Response | undefined;
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
- unmarshalCreateRegistryWebhookRequest_ResponseSchema
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<CreateTransitionRequest_Response> {
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: CreateTransitionRequest_Response | undefined;
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<DeleteCommentRequest_Response> {
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: DeleteCommentRequest_Response | undefined;
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<DeleteRegistryWebhookRequest_Response> {
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: DeleteRegistryWebhookRequest_Response | undefined;
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
- unmarshalDeleteRegistryWebhookRequest_ResponseSchema
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<DeleteTransitionRequest_Response> {
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: DeleteTransitionRequest_Response | undefined;
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<GetRegisteredModelDatabricksRequest_Response> {
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: GetRegisteredModelDatabricksRequest_Response | undefined;
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
- unmarshalGetRegisteredModelDatabricksRequest_ResponseSchema
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<ListRegistryWebhooksRequest_Response> {
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: ListRegistryWebhooksRequest_Response | undefined;
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
- unmarshalListRegistryWebhooksRequest_ResponseSchema
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<ListTransitionRequest_Response> {
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: ListTransitionRequest_Response | undefined;
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<RejectTransitionRequest_Response> {
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: RejectTransitionRequest_Response | undefined;
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<TestRegistryWebhookRequest_Response> {
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: TestRegistryWebhookRequest_Response | undefined;
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
- unmarshalTestRegistryWebhookRequest_ResponseSchema
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<TransitionModelVersionStageDatabricksRequest_Response> {
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: TransitionModelVersionStageDatabricksRequest_Response | undefined;
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
- unmarshalTransitionModelVersionStageDatabricksRequest_ResponseSchema
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<UpdateCommentRequest_Response> {
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: UpdateCommentRequest_Response | undefined;
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<UpdateRegistryWebhookRequest_Response> {
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: UpdateRegistryWebhookRequest_Response | undefined;
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
- unmarshalUpdateRegistryWebhookRequest_ResponseSchema
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<CreateModelVersionRequest_Response> {
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: CreateModelVersionRequest_Response | undefined;
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<CreateRegisteredModelRequest_Response> {
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: CreateRegisteredModelRequest_Response | undefined;
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
- unmarshalCreateRegisteredModelRequest_ResponseSchema
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<DeleteModelVersionRequest_Response> {
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: DeleteModelVersionRequest_Response | undefined;
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<DeleteModelVersionTagRequest_Response> {
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: DeleteModelVersionTagRequest_Response | undefined;
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
- unmarshalDeleteModelVersionTagRequest_ResponseSchema
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<DeleteRegisteredModelRequest_Response> {
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: DeleteRegisteredModelRequest_Response | undefined;
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
- unmarshalDeleteRegisteredModelRequest_ResponseSchema
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<DeleteRegisteredModelTagRequest_Response> {
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: DeleteRegisteredModelTagRequest_Response | undefined;
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
- unmarshalDeleteRegisteredModelTagRequest_ResponseSchema
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<GetModelVersionRequest_Response> {
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: GetModelVersionRequest_Response | undefined;
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<GetModelVersionDownloadUriRequest_Response> {
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: GetModelVersionDownloadUriRequest_Response | undefined;
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
- unmarshalGetModelVersionDownloadUriRequest_ResponseSchema
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<ListLatestVersionsRequest_Response> {
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: ListLatestVersionsRequest_Response | undefined;
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<ListRegisteredModelsRequest_Response> {
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: ListRegisteredModelsRequest_Response | undefined;
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
- unmarshalListRegisteredModelsRequest_ResponseSchema
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<RenameRegisteredModelRequest_Response> {
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: RenameRegisteredModelRequest_Response | undefined;
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
- unmarshalRenameRegisteredModelRequest_ResponseSchema
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<SearchModelVersionsRequest_Response> {
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: SearchModelVersionsRequest_Response | undefined;
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
- unmarshalSearchModelVersionsRequest_ResponseSchema
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<SearchRegisteredModelsRequest_Response> {
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: SearchRegisteredModelsRequest_Response | undefined;
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
- unmarshalSearchRegisteredModelsRequest_ResponseSchema
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<SetModelVersionTagRequest_Response> {
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: SetModelVersionTagRequest_Response | undefined;
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<SetRegisteredModelTagRequest_Response> {
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: SetRegisteredModelTagRequest_Response | undefined;
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
- unmarshalSetRegisteredModelTagRequest_ResponseSchema
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<UpdateModelVersionRequest_Response> {
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: UpdateModelVersionRequest_Response | undefined;
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<UpdateRegisteredModelRequest_Response> {
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: UpdateRegisteredModelRequest_Response | undefined;
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
- unmarshalUpdateRegisteredModelRequest_ResponseSchema
1384
+ unmarshalUpdateRegisteredModelResponseSchema
1427
1385
  );
1428
1386
  };
1429
1387
  await executeCall(call, options);