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

package/src/v1/model.ts

@@ -498,22 +498,6 @@ export interface FeatureList {
   features?: LinkedFeature[] | undefined;
 }
 
-export interface GetLatestVersionsRequest {
-  /** Registered model unique name identifier. */
-  name?: string | undefined;
-  /** List of stages. */
-  stages?: string[] | undefined;
-}
-
-// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
-export interface GetLatestVersionsRequest_Response {
-  /**
-   * Latest version models for each requests stage. Only return models with current `READY` status.
-   * If no `stages` provided, returns the latest version for each stage, including `"None"`.
-   */
-  modelVersions?: ModelVersion[] | undefined;
-}
-
 export interface GetModelVersionDownloadUriRequest {
   /** Name of the registered model */
   name?: string | undefined;
@@ -579,6 +563,22 @@ export interface LinkedFeature {
   featureTableId?: string | undefined;
 }
 
+export interface ListLatestVersionsRequest {
+  /** Registered model unique name identifier. */
+  name?: string | undefined;
+  /** List of stages. */
+  stages?: string[] | undefined;
+}
+
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export interface ListLatestVersionsRequest_Response {
+  /**
+   * Latest version models for each requests stage. Only return models with current `READY` status.
+   * If no `stages` provided, returns the latest version for each stage, including `"None"`.
+   */
+  modelVersions?: ModelVersion[] | undefined;
+}
+
 export interface ListRegisteredModelsRequest {
   /** Maximum number of registered models desired. Max threshold is 1000. */
   maxResults?: bigint | undefined;
@@ -1304,18 +1304,6 @@ export const unmarshalFeatureListSchema: z.ZodType<FeatureList> = z
     features: d.features,
   }));
 
-// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
-export const unmarshalGetLatestVersionsRequest_ResponseSchema: z.ZodType<GetLatestVersionsRequest_Response> =
-  z
-    .object({
-      model_versions: z
-        .array(z.lazy(() => unmarshalModelVersionSchema))
-        .optional(),
-    })
-    .transform(d => ({
-      modelVersions: d.model_versions,
-    }));
-
 // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
 export const unmarshalGetModelVersionDownloadUriRequest_ResponseSchema: z.ZodType<GetModelVersionDownloadUriRequest_Response> =
   z
@@ -1386,6 +1374,18 @@ export const unmarshalLinkedFeatureSchema: z.ZodType<LinkedFeature> = z
     featureTableId: d.feature_table_id,
   }));
 
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalListLatestVersionsRequest_ResponseSchema: z.ZodType<ListLatestVersionsRequest_Response> =
+  z
+    .object({
+      model_versions: z
+        .array(z.lazy(() => unmarshalModelVersionSchema))
+        .optional(),
+    })
+    .transform(d => ({
+      modelVersions: d.model_versions,
+    }));
+
 // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
 export const unmarshalListRegisteredModelsRequest_ResponseSchema: z.ZodType<ListRegisteredModelsRequest_Response> =
   z
@@ -1865,16 +1865,6 @@ export const marshalCreateTransitionRequestSchema: z.ZodType = z
     comment: d.comment,
   }));
 
-export const marshalGetLatestVersionsRequestSchema: z.ZodType = z
-  .object({
-    name: z.string().optional(),
-    stages: z.array(z.string()).optional(),
-  })
-  .transform(d => ({
-    name: d.name,
-    stages: d.stages,
-  }));
-
 export const marshalHttpUrlSpecSchema: z.ZodType = z
   .object({
     url: z.string().optional(),
@@ -1901,6 +1891,16 @@ export const marshalJobSpecSchema: z.ZodType = z
     access_token: d.accessToken,
   }));
 
+export const marshalListLatestVersionsRequestSchema: z.ZodType = z
+  .object({
+    name: z.string().optional(),
+    stages: z.array(z.string()).optional(),
+  })
+  .transform(d => ({
+    name: d.name,
+    stages: d.stages,
+  }));
+
 export const marshalModelVersionTagSchema: z.ZodType = z
   .object({
     key: z.string().optional(),

package/src/v1/utils.ts

@@ -1,7 +1,7 @@
 // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
 
-import type {Call, Options} from '@databricks/sdk-core/api';
-import {execute} from '@databricks/sdk-core/api';
+import type {Options} from '@databricks/sdk-core/ops';
+import {execute} from '@databricks/sdk-core/ops';
 import {ApiError} from '@databricks/sdk-core/apierror';
 import type {
   HttpClient,
@@ -30,7 +30,7 @@ export interface HttpCallOptions {
  * API from the executor's internal type so they can diverge.
  */
 export async function executeCall(
-  call: Call,
+  call: (signal?: AbortSignal) => Promise<void>,
   options?: CallOptions
 ): Promise<void> {
   const opts: Options = {