@databricks/sdk-modelregistry 0.1.0-dev.1 → 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/model.ts CHANGED
@@ -213,8 +213,7 @@ export interface ApproveTransitionRequest {
213
213
  comment?: string | undefined;
214
214
  }
215
215
 
216
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
217
- export interface ApproveTransitionRequest_Response {
216
+ export interface ApproveTransitionResponse {
218
217
  /** New activity generated as a result of this operation. */
219
218
  activity?: Activity | undefined;
220
219
  }
@@ -276,8 +275,7 @@ export interface CreateCommentRequest {
276
275
  comment?: string | undefined;
277
276
  }
278
277
 
279
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
280
- export interface CreateCommentRequest_Response {
278
+ export interface CreateCommentResponse {
281
279
  /** New comment object */
282
280
  comment?: CommentObject | undefined;
283
281
  }
@@ -303,8 +301,7 @@ export interface CreateModelVersionRequest {
303
301
  description?: string | undefined;
304
302
  }
305
303
 
306
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
307
- export interface CreateModelVersionRequest_Response {
304
+ export interface CreateModelVersionResponse {
308
305
  /** Return new version number generated for this model in registry. */
309
306
  modelVersion?: ModelVersion | undefined;
310
307
  }
@@ -318,8 +315,7 @@ export interface CreateRegisteredModelRequest {
318
315
  description?: string | undefined;
319
316
  }
320
317
 
321
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
322
- export interface CreateRegisteredModelRequest_Response {
318
+ export interface CreateRegisteredModelResponse {
323
319
  registeredModel?: RegisteredModel | undefined;
324
320
  }
325
321
 
@@ -371,8 +367,7 @@ export interface CreateRegistryWebhookRequest {
371
367
  jobSpec?: JobSpec | undefined;
372
368
  }
373
369
 
374
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
375
- export interface CreateRegistryWebhookRequest_Response {
370
+ export interface CreateRegistryWebhookResponse {
376
371
  webhook?: RegistryWebhook | undefined;
377
372
  }
378
373
 
@@ -398,8 +393,7 @@ export interface CreateTransitionRequest {
398
393
  comment?: string | undefined;
399
394
  }
400
395
 
401
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
402
- export interface CreateTransitionRequest_Response {
396
+ export interface CreateTransitionResponse {
403
397
  /** New activity generated for stage transition request. */
404
398
  request?: TransitionRequest | undefined;
405
399
  }
@@ -409,8 +403,8 @@ export interface DeleteCommentRequest {
409
403
  id?: string | undefined;
410
404
  }
411
405
 
412
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
413
- export interface DeleteCommentRequest_Response {}
406
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
407
+ export interface DeleteCommentResponse {}
414
408
 
415
409
  export interface DeleteModelVersionRequest {
416
410
  /** Name of the registered model */
@@ -419,8 +413,8 @@ export interface DeleteModelVersionRequest {
419
413
  version?: string | undefined;
420
414
  }
421
415
 
422
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
423
- export interface DeleteModelVersionRequest_Response {}
416
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
417
+ export interface DeleteModelVersionResponse {}
424
418
 
425
419
  export interface DeleteModelVersionTagRequest {
426
420
  /** Name of the registered model that the tag was logged under. */
@@ -431,16 +425,16 @@ export interface DeleteModelVersionTagRequest {
431
425
  key?: string | undefined;
432
426
  }
433
427
 
434
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
435
- export interface DeleteModelVersionTagRequest_Response {}
428
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
429
+ export interface DeleteModelVersionTagResponse {}
436
430
 
437
431
  export interface DeleteRegisteredModelRequest {
438
432
  /** Registered model unique name identifier. */
439
433
  name?: string | undefined;
440
434
  }
441
435
 
442
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
443
- export interface DeleteRegisteredModelRequest_Response {}
436
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
437
+ export interface DeleteRegisteredModelResponse {}
444
438
 
445
439
  export interface DeleteRegisteredModelTagRequest {
446
440
  /** Name of the registered model that the tag was logged under. */
@@ -449,8 +443,8 @@ export interface DeleteRegisteredModelTagRequest {
449
443
  key?: string | undefined;
450
444
  }
451
445
 
452
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
453
- export interface DeleteRegisteredModelTagRequest_Response {}
446
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
447
+ export interface DeleteRegisteredModelTagResponse {}
454
448
 
455
449
  /**
456
450
  * .. note::
@@ -461,8 +455,8 @@ export interface DeleteRegistryWebhookRequest {
461
455
  id?: string | undefined;
462
456
  }
463
457
 
464
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
465
- export interface DeleteRegistryWebhookRequest_Response {}
458
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
459
+ export interface DeleteRegistryWebhookResponse {}
466
460
 
467
461
  export interface DeleteTransitionRequest {
468
462
  /** Name of the model. */
@@ -487,8 +481,7 @@ export interface DeleteTransitionRequest {
487
481
  comment?: string | undefined;
488
482
  }
489
483
 
490
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
491
- export interface DeleteTransitionRequest_Response {
484
+ export interface DeleteTransitionResponse {
492
485
  /** New activity generated as a result of this operation. */
493
486
  activity?: Activity | undefined;
494
487
  }
@@ -498,15 +491,7 @@ export interface FeatureList {
498
491
  features?: LinkedFeature[] | undefined;
499
492
  }
500
493
 
501
- export interface GetLatestVersionsRequest {
502
- /** Registered model unique name identifier. */
503
- name?: string | undefined;
504
- /** List of stages. */
505
- stages?: string[] | undefined;
506
- }
507
-
508
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
509
- export interface GetLatestVersionsRequest_Response {
494
+ export interface GetLatestVersionsResponse {
510
495
  /**
511
496
  * Latest version models for each requests stage. Only return models with current `READY` status.
512
497
  * If no `stages` provided, returns the latest version for each stage, including `"None"`.
@@ -521,8 +506,7 @@ export interface GetModelVersionDownloadUriRequest {
521
506
  version?: string | undefined;
522
507
  }
523
508
 
524
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
525
- export interface GetModelVersionDownloadUriRequest_Response {
509
+ export interface GetModelVersionDownloadUriResponse {
526
510
  /** URI corresponding to where artifacts for this model version are stored. */
527
511
  artifactUri?: string | undefined;
528
512
  }
@@ -534,8 +518,7 @@ export interface GetModelVersionRequest {
534
518
  version?: string | undefined;
535
519
  }
536
520
 
537
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
538
- export interface GetModelVersionRequest_Response {
521
+ export interface GetModelVersionResponse {
539
522
  modelVersion?: ModelVersion | undefined;
540
523
  }
541
524
 
@@ -544,8 +527,7 @@ export interface GetRegisteredModelDatabricksRequest {
544
527
  name?: string | undefined;
545
528
  }
546
529
 
547
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
548
- export interface GetRegisteredModelDatabricksRequest_Response {
530
+ export interface GetRegisteredModelDatabricksResponse {
549
531
  registeredModelDatabricks?: RegisteredModelDatabricks | undefined;
550
532
  }
551
533
 
@@ -579,6 +561,13 @@ export interface LinkedFeature {
579
561
  featureTableId?: string | undefined;
580
562
  }
581
563
 
564
+ export interface ListLatestVersionsRequest {
565
+ /** Registered model unique name identifier. */
566
+ name?: string | undefined;
567
+ /** List of stages. */
568
+ stages?: string[] | undefined;
569
+ }
570
+
582
571
  export interface ListRegisteredModelsRequest {
583
572
  /** Maximum number of registered models desired. Max threshold is 1000. */
584
573
  maxResults?: bigint | undefined;
@@ -586,8 +575,7 @@ export interface ListRegisteredModelsRequest {
586
575
  pageToken?: string | undefined;
587
576
  }
588
577
 
589
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
590
- export interface ListRegisteredModelsRequest_Response {
578
+ export interface ListRegisteredModelsResponse {
591
579
  registeredModels?: RegisteredModel[] | undefined;
592
580
  /** Pagination token to request next page of models for the same query. */
593
581
  nextPageToken?: string | undefined;
@@ -634,8 +622,7 @@ export interface ListRegistryWebhooksRequest {
634
622
  maxResults?: bigint | undefined;
635
623
  }
636
624
 
637
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
638
- export interface ListRegistryWebhooksRequest_Response {
625
+ export interface ListRegistryWebhooksResponse {
639
626
  /** Array of registry webhooks. */
640
627
  webhooks?: RegistryWebhook[] | undefined;
641
628
  /** Token that can be used to retrieve the next page of artifact results */
@@ -649,8 +636,7 @@ export interface ListTransitionRequest {
649
636
  version?: string | undefined;
650
637
  }
651
638
 
652
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
653
- export interface ListTransitionRequest_Response {
639
+ export interface ListTransitionResponse {
654
640
  /** Array of open transition requests. */
655
641
  requests?: Activity[] | undefined;
656
642
  }
@@ -849,8 +835,7 @@ export interface RejectTransitionRequest {
849
835
  comment?: string | undefined;
850
836
  }
851
837
 
852
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
853
- export interface RejectTransitionRequest_Response {
838
+ export interface RejectTransitionResponse {
854
839
  /** New activity generated as a result of this operation. */
855
840
  activity?: Activity | undefined;
856
841
  }
@@ -862,8 +847,7 @@ export interface RenameRegisteredModelRequest {
862
847
  newName?: string | undefined;
863
848
  }
864
849
 
865
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
866
- export interface RenameRegisteredModelRequest_Response {
850
+ export interface RenameRegisteredModelResponse {
867
851
  registeredModel?: RegisteredModel | undefined;
868
852
  }
869
853
 
@@ -886,8 +870,7 @@ export interface SearchModelVersionsRequest {
886
870
  pageToken?: string | undefined;
887
871
  }
888
872
 
889
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
890
- export interface SearchModelVersionsRequest_Response {
873
+ export interface SearchModelVersionsResponse {
891
874
  /** Models that match the search criteria */
892
875
  modelVersions?: ModelVersion[] | undefined;
893
876
  /** Pagination token to request next page of models for the same search query. */
@@ -913,8 +896,7 @@ export interface SearchRegisteredModelsRequest {
913
896
  pageToken?: string | undefined;
914
897
  }
915
898
 
916
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
917
- export interface SearchRegisteredModelsRequest_Response {
899
+ export interface SearchRegisteredModelsResponse {
918
900
  /** Registered Models that match the search criteria. */
919
901
  registeredModels?: RegisteredModel[] | undefined;
920
902
  /** Pagination token to request the next page of models. */
@@ -939,8 +921,8 @@ export interface SetModelVersionTagRequest {
939
921
  value?: string | undefined;
940
922
  }
941
923
 
942
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
943
- export interface SetModelVersionTagRequest_Response {}
924
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
925
+ export interface SetModelVersionTagResponse {}
944
926
 
945
927
  export interface SetRegisteredModelTagRequest {
946
928
  /** Unique name of the model. */
@@ -958,8 +940,8 @@ export interface SetRegisteredModelTagRequest {
958
940
  value?: string | undefined;
959
941
  }
960
942
 
961
- // eslint-disable-next-line @typescript-eslint/naming-convention, @typescript-eslint/no-empty-object-type -- Proto-style nested message name.
962
- export interface SetRegisteredModelTagRequest_Response {}
943
+ // eslint-disable-next-line @typescript-eslint/no-empty-object-type
944
+ export interface SetRegisteredModelTagResponse {}
963
945
 
964
946
  /** Details required to test a registry webhook. */
965
947
  export interface TestRegistryWebhookRequest {
@@ -969,8 +951,7 @@ export interface TestRegistryWebhookRequest {
969
951
  event?: RegistryWebhookEvent | undefined;
970
952
  }
971
953
 
972
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
973
- export interface TestRegistryWebhookRequest_Response {
954
+ export interface TestRegistryWebhookResponse {
974
955
  /** Status code returned by the webhook URL */
975
956
  statusCode?: number | undefined;
976
957
  /** Body of the response from the webhook URL */
@@ -1001,8 +982,7 @@ export interface TransitionModelVersionStageDatabricksRequest {
1001
982
  comment?: string | undefined;
1002
983
  }
1003
984
 
1004
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1005
- export interface TransitionModelVersionStageDatabricksRequest_Response {
985
+ export interface TransitionModelVersionStageDatabricksResponse {
1006
986
  /** Updated model version */
1007
987
  modelVersionDatabricks?: ModelVersionDatabricks | undefined;
1008
988
  }
@@ -1062,8 +1042,7 @@ export interface UpdateCommentRequest {
1062
1042
  comment?: string | undefined;
1063
1043
  }
1064
1044
 
1065
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1066
- export interface UpdateCommentRequest_Response {
1045
+ export interface UpdateCommentResponse {
1067
1046
  /** Updated comment object */
1068
1047
  comment?: CommentObject | undefined;
1069
1048
  }
@@ -1077,8 +1056,7 @@ export interface UpdateModelVersionRequest {
1077
1056
  description?: string | undefined;
1078
1057
  }
1079
1058
 
1080
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1081
- export interface UpdateModelVersionRequest_Response {
1059
+ export interface UpdateModelVersionResponse {
1082
1060
  /** Return new version number generated for this model in registry. */
1083
1061
  modelVersion?: ModelVersion | undefined;
1084
1062
  }
@@ -1090,8 +1068,7 @@ export interface UpdateRegisteredModelRequest {
1090
1068
  description?: string | undefined;
1091
1069
  }
1092
1070
 
1093
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1094
- export interface UpdateRegisteredModelRequest_Response {
1071
+ export interface UpdateRegisteredModelResponse {
1095
1072
  registeredModel?: RegisteredModel | undefined;
1096
1073
  }
1097
1074
 
@@ -1133,8 +1110,7 @@ export interface UpdateRegistryWebhookRequest {
1133
1110
  jobSpec?: JobSpec | undefined;
1134
1111
  }
1135
1112
 
1136
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1137
- export interface UpdateRegistryWebhookRequest_Response {
1113
+ export interface UpdateRegistryWebhookResponse {
1138
1114
  webhook?: RegistryWebhook | undefined;
1139
1115
  }
1140
1116
 
@@ -1170,8 +1146,7 @@ export const unmarshalActivitySchema: z.ZodType<Activity> = z
1170
1146
  id: d.id,
1171
1147
  }));
1172
1148
 
1173
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1174
- export const unmarshalApproveTransitionRequest_ResponseSchema: z.ZodType<ApproveTransitionRequest_Response> =
1149
+ export const unmarshalApproveTransitionResponseSchema: z.ZodType<ApproveTransitionResponse> =
1175
1150
  z
1176
1151
  .object({
1177
1152
  activity: z.lazy(() => unmarshalActivitySchema).optional(),
@@ -1212,8 +1187,7 @@ export const unmarshalCommentObjectSchema: z.ZodType<CommentObject> = z
1212
1187
  id: d.id,
1213
1188
  }));
1214
1189
 
1215
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1216
- export const unmarshalCreateCommentRequest_ResponseSchema: z.ZodType<CreateCommentRequest_Response> =
1190
+ export const unmarshalCreateCommentResponseSchema: z.ZodType<CreateCommentResponse> =
1217
1191
  z
1218
1192
  .object({
1219
1193
  comment: z.lazy(() => unmarshalCommentObjectSchema).optional(),
@@ -1222,8 +1196,7 @@ export const unmarshalCreateCommentRequest_ResponseSchema: z.ZodType<CreateComme
1222
1196
  comment: d.comment,
1223
1197
  }));
1224
1198
 
1225
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1226
- export const unmarshalCreateModelVersionRequest_ResponseSchema: z.ZodType<CreateModelVersionRequest_Response> =
1199
+ export const unmarshalCreateModelVersionResponseSchema: z.ZodType<CreateModelVersionResponse> =
1227
1200
  z
1228
1201
  .object({
1229
1202
  model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
@@ -1232,8 +1205,7 @@ export const unmarshalCreateModelVersionRequest_ResponseSchema: z.ZodType<Create
1232
1205
  modelVersion: d.model_version,
1233
1206
  }));
1234
1207
 
1235
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1236
- export const unmarshalCreateRegisteredModelRequest_ResponseSchema: z.ZodType<CreateRegisteredModelRequest_Response> =
1208
+ export const unmarshalCreateRegisteredModelResponseSchema: z.ZodType<CreateRegisteredModelResponse> =
1237
1209
  z
1238
1210
  .object({
1239
1211
  registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
@@ -1242,8 +1214,7 @@ export const unmarshalCreateRegisteredModelRequest_ResponseSchema: z.ZodType<Cre
1242
1214
  registeredModel: d.registered_model,
1243
1215
  }));
1244
1216
 
1245
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1246
- export const unmarshalCreateRegistryWebhookRequest_ResponseSchema: z.ZodType<CreateRegistryWebhookRequest_Response> =
1217
+ export const unmarshalCreateRegistryWebhookResponseSchema: z.ZodType<CreateRegistryWebhookResponse> =
1247
1218
  z
1248
1219
  .object({
1249
1220
  webhook: z.lazy(() => unmarshalRegistryWebhookSchema).optional(),
@@ -1252,8 +1223,7 @@ export const unmarshalCreateRegistryWebhookRequest_ResponseSchema: z.ZodType<Cre
1252
1223
  webhook: d.webhook,
1253
1224
  }));
1254
1225
 
1255
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1256
- export const unmarshalCreateTransitionRequest_ResponseSchema: z.ZodType<CreateTransitionRequest_Response> =
1226
+ export const unmarshalCreateTransitionResponseSchema: z.ZodType<CreateTransitionResponse> =
1257
1227
  z
1258
1228
  .object({
1259
1229
  request: z.lazy(() => unmarshalTransitionRequestSchema).optional(),
@@ -1262,32 +1232,25 @@ export const unmarshalCreateTransitionRequest_ResponseSchema: z.ZodType<CreateTr
1262
1232
  request: d.request,
1263
1233
  }));
1264
1234
 
1265
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1266
- export const unmarshalDeleteCommentRequest_ResponseSchema: z.ZodType<DeleteCommentRequest_Response> =
1235
+ export const unmarshalDeleteCommentResponseSchema: z.ZodType<DeleteCommentResponse> =
1267
1236
  z.object({});
1268
1237
 
1269
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1270
- export const unmarshalDeleteModelVersionRequest_ResponseSchema: z.ZodType<DeleteModelVersionRequest_Response> =
1238
+ export const unmarshalDeleteModelVersionResponseSchema: z.ZodType<DeleteModelVersionResponse> =
1271
1239
  z.object({});
1272
1240
 
1273
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1274
- export const unmarshalDeleteModelVersionTagRequest_ResponseSchema: z.ZodType<DeleteModelVersionTagRequest_Response> =
1241
+ export const unmarshalDeleteModelVersionTagResponseSchema: z.ZodType<DeleteModelVersionTagResponse> =
1275
1242
  z.object({});
1276
1243
 
1277
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1278
- export const unmarshalDeleteRegisteredModelRequest_ResponseSchema: z.ZodType<DeleteRegisteredModelRequest_Response> =
1244
+ export const unmarshalDeleteRegisteredModelResponseSchema: z.ZodType<DeleteRegisteredModelResponse> =
1279
1245
  z.object({});
1280
1246
 
1281
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1282
- export const unmarshalDeleteRegisteredModelTagRequest_ResponseSchema: z.ZodType<DeleteRegisteredModelTagRequest_Response> =
1247
+ export const unmarshalDeleteRegisteredModelTagResponseSchema: z.ZodType<DeleteRegisteredModelTagResponse> =
1283
1248
  z.object({});
1284
1249
 
1285
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1286
- export const unmarshalDeleteRegistryWebhookRequest_ResponseSchema: z.ZodType<DeleteRegistryWebhookRequest_Response> =
1250
+ export const unmarshalDeleteRegistryWebhookResponseSchema: z.ZodType<DeleteRegistryWebhookResponse> =
1287
1251
  z.object({});
1288
1252
 
1289
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1290
- export const unmarshalDeleteTransitionRequest_ResponseSchema: z.ZodType<DeleteTransitionRequest_Response> =
1253
+ export const unmarshalDeleteTransitionResponseSchema: z.ZodType<DeleteTransitionResponse> =
1291
1254
  z
1292
1255
  .object({
1293
1256
  activity: z.lazy(() => unmarshalActivitySchema).optional(),
@@ -1304,8 +1267,7 @@ export const unmarshalFeatureListSchema: z.ZodType<FeatureList> = z
1304
1267
  features: d.features,
1305
1268
  }));
1306
1269
 
1307
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1308
- export const unmarshalGetLatestVersionsRequest_ResponseSchema: z.ZodType<GetLatestVersionsRequest_Response> =
1270
+ export const unmarshalGetLatestVersionsResponseSchema: z.ZodType<GetLatestVersionsResponse> =
1309
1271
  z
1310
1272
  .object({
1311
1273
  model_versions: z
@@ -1316,8 +1278,7 @@ export const unmarshalGetLatestVersionsRequest_ResponseSchema: z.ZodType<GetLate
1316
1278
  modelVersions: d.model_versions,
1317
1279
  }));
1318
1280
 
1319
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1320
- export const unmarshalGetModelVersionDownloadUriRequest_ResponseSchema: z.ZodType<GetModelVersionDownloadUriRequest_Response> =
1281
+ export const unmarshalGetModelVersionDownloadUriResponseSchema: z.ZodType<GetModelVersionDownloadUriResponse> =
1321
1282
  z
1322
1283
  .object({
1323
1284
  artifact_uri: z.string().optional(),
@@ -1326,8 +1287,7 @@ export const unmarshalGetModelVersionDownloadUriRequest_ResponseSchema: z.ZodTyp
1326
1287
  artifactUri: d.artifact_uri,
1327
1288
  }));
1328
1289
 
1329
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1330
- export const unmarshalGetModelVersionRequest_ResponseSchema: z.ZodType<GetModelVersionRequest_Response> =
1290
+ export const unmarshalGetModelVersionResponseSchema: z.ZodType<GetModelVersionResponse> =
1331
1291
  z
1332
1292
  .object({
1333
1293
  model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
@@ -1336,8 +1296,7 @@ export const unmarshalGetModelVersionRequest_ResponseSchema: z.ZodType<GetModelV
1336
1296
  modelVersion: d.model_version,
1337
1297
  }));
1338
1298
 
1339
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1340
- export const unmarshalGetRegisteredModelDatabricksRequest_ResponseSchema: z.ZodType<GetRegisteredModelDatabricksRequest_Response> =
1299
+ export const unmarshalGetRegisteredModelDatabricksResponseSchema: z.ZodType<GetRegisteredModelDatabricksResponse> =
1341
1300
  z
1342
1301
  .object({
1343
1302
  registered_model_databricks: z
@@ -1386,8 +1345,7 @@ export const unmarshalLinkedFeatureSchema: z.ZodType<LinkedFeature> = z
1386
1345
  featureTableId: d.feature_table_id,
1387
1346
  }));
1388
1347
 
1389
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1390
- export const unmarshalListRegisteredModelsRequest_ResponseSchema: z.ZodType<ListRegisteredModelsRequest_Response> =
1348
+ export const unmarshalListRegisteredModelsResponseSchema: z.ZodType<ListRegisteredModelsResponse> =
1391
1349
  z
1392
1350
  .object({
1393
1351
  registered_models: z
@@ -1400,8 +1358,7 @@ export const unmarshalListRegisteredModelsRequest_ResponseSchema: z.ZodType<List
1400
1358
  nextPageToken: d.next_page_token,
1401
1359
  }));
1402
1360
 
1403
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1404
- export const unmarshalListRegistryWebhooksRequest_ResponseSchema: z.ZodType<ListRegistryWebhooksRequest_Response> =
1361
+ export const unmarshalListRegistryWebhooksResponseSchema: z.ZodType<ListRegistryWebhooksResponse> =
1405
1362
  z
1406
1363
  .object({
1407
1364
  webhooks: z
@@ -1414,8 +1371,7 @@ export const unmarshalListRegistryWebhooksRequest_ResponseSchema: z.ZodType<List
1414
1371
  nextPageToken: d.next_page_token,
1415
1372
  }));
1416
1373
 
1417
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1418
- export const unmarshalListTransitionRequest_ResponseSchema: z.ZodType<ListTransitionRequest_Response> =
1374
+ export const unmarshalListTransitionResponseSchema: z.ZodType<ListTransitionResponse> =
1419
1375
  z
1420
1376
  .object({
1421
1377
  requests: z.array(z.lazy(() => unmarshalActivitySchema)).optional(),
@@ -1623,8 +1579,7 @@ export const unmarshalRegistryWebhookSchema: z.ZodType<RegistryWebhook> = z
1623
1579
  modelName: d.model_name,
1624
1580
  }));
1625
1581
 
1626
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1627
- export const unmarshalRejectTransitionRequest_ResponseSchema: z.ZodType<RejectTransitionRequest_Response> =
1582
+ export const unmarshalRejectTransitionResponseSchema: z.ZodType<RejectTransitionResponse> =
1628
1583
  z
1629
1584
  .object({
1630
1585
  activity: z.lazy(() => unmarshalActivitySchema).optional(),
@@ -1633,8 +1588,7 @@ export const unmarshalRejectTransitionRequest_ResponseSchema: z.ZodType<RejectTr
1633
1588
  activity: d.activity,
1634
1589
  }));
1635
1590
 
1636
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1637
- export const unmarshalRenameRegisteredModelRequest_ResponseSchema: z.ZodType<RenameRegisteredModelRequest_Response> =
1591
+ export const unmarshalRenameRegisteredModelResponseSchema: z.ZodType<RenameRegisteredModelResponse> =
1638
1592
  z
1639
1593
  .object({
1640
1594
  registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
@@ -1643,8 +1597,7 @@ export const unmarshalRenameRegisteredModelRequest_ResponseSchema: z.ZodType<Ren
1643
1597
  registeredModel: d.registered_model,
1644
1598
  }));
1645
1599
 
1646
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1647
- export const unmarshalSearchModelVersionsRequest_ResponseSchema: z.ZodType<SearchModelVersionsRequest_Response> =
1600
+ export const unmarshalSearchModelVersionsResponseSchema: z.ZodType<SearchModelVersionsResponse> =
1648
1601
  z
1649
1602
  .object({
1650
1603
  model_versions: z
@@ -1657,8 +1610,7 @@ export const unmarshalSearchModelVersionsRequest_ResponseSchema: z.ZodType<Searc
1657
1610
  nextPageToken: d.next_page_token,
1658
1611
  }));
1659
1612
 
1660
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1661
- export const unmarshalSearchRegisteredModelsRequest_ResponseSchema: z.ZodType<SearchRegisteredModelsRequest_Response> =
1613
+ export const unmarshalSearchRegisteredModelsResponseSchema: z.ZodType<SearchRegisteredModelsResponse> =
1662
1614
  z
1663
1615
  .object({
1664
1616
  registered_models: z
@@ -1671,16 +1623,13 @@ export const unmarshalSearchRegisteredModelsRequest_ResponseSchema: z.ZodType<Se
1671
1623
  nextPageToken: d.next_page_token,
1672
1624
  }));
1673
1625
 
1674
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1675
- export const unmarshalSetModelVersionTagRequest_ResponseSchema: z.ZodType<SetModelVersionTagRequest_Response> =
1626
+ export const unmarshalSetModelVersionTagResponseSchema: z.ZodType<SetModelVersionTagResponse> =
1676
1627
  z.object({});
1677
1628
 
1678
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1679
- export const unmarshalSetRegisteredModelTagRequest_ResponseSchema: z.ZodType<SetRegisteredModelTagRequest_Response> =
1629
+ export const unmarshalSetRegisteredModelTagResponseSchema: z.ZodType<SetRegisteredModelTagResponse> =
1680
1630
  z.object({});
1681
1631
 
1682
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1683
- export const unmarshalTestRegistryWebhookRequest_ResponseSchema: z.ZodType<TestRegistryWebhookRequest_Response> =
1632
+ export const unmarshalTestRegistryWebhookResponseSchema: z.ZodType<TestRegistryWebhookResponse> =
1684
1633
  z
1685
1634
  .object({
1686
1635
  status_code: z.number().optional(),
@@ -1691,8 +1640,7 @@ export const unmarshalTestRegistryWebhookRequest_ResponseSchema: z.ZodType<TestR
1691
1640
  body: d.body,
1692
1641
  }));
1693
1642
 
1694
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1695
- export const unmarshalTransitionModelVersionStageDatabricksRequest_ResponseSchema: z.ZodType<TransitionModelVersionStageDatabricksRequest_Response> =
1643
+ export const unmarshalTransitionModelVersionStageDatabricksResponseSchema: z.ZodType<TransitionModelVersionStageDatabricksResponse> =
1696
1644
  z
1697
1645
  .object({
1698
1646
  model_version_databricks: z
@@ -1735,8 +1683,7 @@ export const unmarshalTransitionRequestSchema: z.ZodType<TransitionRequest> = z
1735
1683
  id: d.id,
1736
1684
  }));
1737
1685
 
1738
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1739
- export const unmarshalUpdateCommentRequest_ResponseSchema: z.ZodType<UpdateCommentRequest_Response> =
1686
+ export const unmarshalUpdateCommentResponseSchema: z.ZodType<UpdateCommentResponse> =
1740
1687
  z
1741
1688
  .object({
1742
1689
  comment: z.lazy(() => unmarshalCommentObjectSchema).optional(),
@@ -1745,8 +1692,7 @@ export const unmarshalUpdateCommentRequest_ResponseSchema: z.ZodType<UpdateComme
1745
1692
  comment: d.comment,
1746
1693
  }));
1747
1694
 
1748
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1749
- export const unmarshalUpdateModelVersionRequest_ResponseSchema: z.ZodType<UpdateModelVersionRequest_Response> =
1695
+ export const unmarshalUpdateModelVersionResponseSchema: z.ZodType<UpdateModelVersionResponse> =
1750
1696
  z
1751
1697
  .object({
1752
1698
  model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
@@ -1755,8 +1701,7 @@ export const unmarshalUpdateModelVersionRequest_ResponseSchema: z.ZodType<Update
1755
1701
  modelVersion: d.model_version,
1756
1702
  }));
1757
1703
 
1758
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1759
- export const unmarshalUpdateRegisteredModelRequest_ResponseSchema: z.ZodType<UpdateRegisteredModelRequest_Response> =
1704
+ export const unmarshalUpdateRegisteredModelResponseSchema: z.ZodType<UpdateRegisteredModelResponse> =
1760
1705
  z
1761
1706
  .object({
1762
1707
  registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
@@ -1765,8 +1710,7 @@ export const unmarshalUpdateRegisteredModelRequest_ResponseSchema: z.ZodType<Upd
1765
1710
  registeredModel: d.registered_model,
1766
1711
  }));
1767
1712
 
1768
- // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
1769
- export const unmarshalUpdateRegistryWebhookRequest_ResponseSchema: z.ZodType<UpdateRegistryWebhookRequest_Response> =
1713
+ export const unmarshalUpdateRegistryWebhookResponseSchema: z.ZodType<UpdateRegistryWebhookResponse> =
1770
1714
  z
1771
1715
  .object({
1772
1716
  webhook: z.lazy(() => unmarshalRegistryWebhookSchema).optional(),
@@ -1865,16 +1809,6 @@ export const marshalCreateTransitionRequestSchema: z.ZodType = z
1865
1809
  comment: d.comment,
1866
1810
  }));
1867
1811
 
1868
- export const marshalGetLatestVersionsRequestSchema: z.ZodType = z
1869
- .object({
1870
- name: z.string().optional(),
1871
- stages: z.array(z.string()).optional(),
1872
- })
1873
- .transform(d => ({
1874
- name: d.name,
1875
- stages: d.stages,
1876
- }));
1877
-
1878
1812
  export const marshalHttpUrlSpecSchema: z.ZodType = z
1879
1813
  .object({
1880
1814
  url: z.string().optional(),
@@ -1901,6 +1835,16 @@ export const marshalJobSpecSchema: z.ZodType = z
1901
1835
  access_token: d.accessToken,
1902
1836
  }));
1903
1837
 
1838
+ export const marshalListLatestVersionsRequestSchema: z.ZodType = z
1839
+ .object({
1840
+ name: z.string().optional(),
1841
+ stages: z.array(z.string()).optional(),
1842
+ })
1843
+ .transform(d => ({
1844
+ name: d.name,
1845
+ stages: d.stages,
1846
+ }));
1847
+
1904
1848
  export const marshalModelVersionTagSchema: z.ZodType = z
1905
1849
  .object({
1906
1850
  key: z.string().optional(),
package/src/v1/utils.ts CHANGED
@@ -1,7 +1,7 @@
1
1
  // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
2
 
3
- import type {Call, Options} from '@databricks/sdk-core/api';
4
- import {execute} from '@databricks/sdk-core/api';
3
+ import type {Options} from '@databricks/sdk-core/ops';
4
+ import {execute} from '@databricks/sdk-core/ops';
5
5
  import {ApiError} from '@databricks/sdk-core/apierror';
6
6
  import type {
7
7
  HttpClient,
@@ -30,7 +30,7 @@ export interface HttpCallOptions {
30
30
  * API from the executor's internal type so they can diverge.
31
31
  */
32
32
  export async function executeCall(
33
- call: Call,
33
+ call: (signal?: AbortSignal) => Promise<void>,
34
34
  options?: CallOptions
35
35
  ): Promise<void> {
36
36
  const opts: Options = {