@databricks/sdk-uc-registeredmodels 0.0.0-dev → 0.1.0-dev.1

package/dist/v1/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,OAAO,EAAC,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAC,QAAQ,EAAC,MAAM,+BAA+B,CAAC;AAQvD,OAAO,OAAO,MAAM,aAAa,CAAC;AAGlC,0EAA0E;AAC1E,6EAA6E;AAC7E,gEAAgE;AAChE,MAAM,UAAU,GAAG,OAAO,CAAC,EAAC,eAAe,EAAE,IAAI,EAAC,CAAC,CAAC;AAQpD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAU,EACV,OAAqB;IAErB,MAAM,IAAI,GAAY;QACpB,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;QACjE,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;QACF,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;KAClE,CAAC;IACF,OAAO,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,IAAuC;IAEvC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAC3B,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,SAAS,CAAC;QACR,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,MAAM;QACR,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;QAChC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;QAC3B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;KACtB,CAAC,CAAC;IAEH,IAAI,IAAkB,CAAC;IACvB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;IACV,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE;QACjC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;KACrC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC3E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,MAAM,MAAM,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,GAAW,EACX,OAAgB,EAChB,MAAoB,EACpB,IAA0C;IAE1C,MAAM,GAAG,GAAgB,EAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC;IAChD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAI,IAAgB,EAAE,MAAoB;IACrE,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAY,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAa,EAAE,MAAiB;IAC7D,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAc,EACd,KAAc,EACd,MAAuB;IAEvB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC1E,kBAAkB,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;SAAM,IACL,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS;QAC1B,OAAO,KAAK,KAAK,QAAQ,EACzB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,qCAAqC,OAAO,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;AACH,CAAC"}

package/package.json

@@ -1,7 +1,41 @@
 {
   "name": "@databricks/sdk-uc-registeredmodels",
-  "version": "0.0.0-dev",
-  "description": "Bootstrap placeholder; real contents in a later release.",
-  "main": "index.js",
-  "license": "Apache-2.0"
+  "version": "0.1.0-dev.1",
+  "description": "",
+  "type": "module",
+  "exports": {
+    "./v1": {
+      "types": "./dist/v1/index.d.ts",
+      "import": "./dist/v1/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "src",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc -b",
+    "lint": "eslint src --ext .ts",
+    "lint:fix": "eslint src --ext .ts --fix",
+    "format": "prettier --write \"src/**/*.ts\"",
+    "format:check": "prettier --check \"src/**/*.ts\"",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist tsconfig.tsbuildinfo",
+    "test": "echo 'no tests'",
+    "test:browser": "echo 'no tests'"
+  },
+  "author": "Databricks",
+  "license": "Apache-2.0",
+  "dependencies": {
+    "@databricks/sdk-auth": ">=0.1.0-dev.2 <1.0.0",
+    "@databricks/sdk-core": ">=0.1.0-dev.3 <1.0.0",
+    "@databricks/sdk-options": ">=0.1.0-dev.2 <1.0.0",
+    "@js-temporal/polyfill": "^0.5.0",
+    "json-bigint": "^1.0.0",
+    "zod": "^4.3.6"
+  },
+  "engines": {
+    "node": ">=22.0.0"
+  }
 }

package/src/v1/client.ts

@@ -0,0 +1,641 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+
+import {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';
+import type {Call} from '@databricks/sdk-core/api';
+import {createDefault} from '@databricks/sdk-core/clientinfo';
+import type {Logger} from '@databricks/sdk-core/logger';
+import {NoOpLogger} from '@databricks/sdk-core/logger';
+import type {CallOptions} from '@databricks/sdk-options/call';
+import type {ClientOptions} from '@databricks/sdk-options/client';
+import type {HttpClient} from '@databricks/sdk-core/http';
+import {newHttpClient} from './transport';
+import {
+  buildHttpRequest,
+  executeCall,
+  executeHttpCall,
+  marshalRequest,
+  parseResponse,
+} from './utils';
+import pkgJson from '../../package.json' with {type: 'json'};
+import type {
+  CreateRegisteredModelRequest,
+  DeleteModelVersionRequest,
+  DeleteModelVersionRequest_Response,
+  DeleteRegisteredModelAliasRequest,
+  DeleteRegisteredModelAliasRequest_Response,
+  DeleteRegisteredModelRequest,
+  DeleteRegisteredModelRequest_Response,
+  GetModelVersionByAliasRequest,
+  GetModelVersionRequest,
+  GetRegisteredModelRequest,
+  ListModelVersionsRequest,
+  ListModelVersionsRequest_Response,
+  ListRegisteredModelsRequest,
+  ListRegisteredModelsRequest_Response,
+  ModelVersionInfo,
+  RegisteredModelAliasInfo,
+  RegisteredModelInfo,
+  SetRegisteredModelAliasRequest,
+  UpdateModelVersionRequest,
+  UpdateRegisteredModelRequest,
+} from './model';
+import {
+  marshalCreateRegisteredModelRequestSchema,
+  marshalSetRegisteredModelAliasRequestSchema,
+  marshalUpdateModelVersionRequestSchema,
+  marshalUpdateRegisteredModelRequestSchema,
+  unmarshalDeleteModelVersionRequest_ResponseSchema,
+  unmarshalDeleteRegisteredModelAliasRequest_ResponseSchema,
+  unmarshalDeleteRegisteredModelRequest_ResponseSchema,
+  unmarshalListModelVersionsRequest_ResponseSchema,
+  unmarshalListRegisteredModelsRequest_ResponseSchema,
+  unmarshalModelVersionInfoSchema,
+  unmarshalRegisteredModelAliasInfoSchema,
+  unmarshalRegisteredModelInfoSchema,
+} from './model';
+
+// Package identity segment for this client to be used in the User-Agent header.
+const PACKAGE_SEGMENT = {
+  key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
+  value: pkgJson.version,
+};
+
+export class RegisteredmodelsClient {
+  private readonly host: string;
+  // Workspace ID used to route workspace-level calls on unified hosts (SPOG).
+  // When set, workspace-level methods send X-Databricks-Org-Id on every
+  // request.
+  private readonly workspaceId: string | undefined;
+  private readonly httpClient: HttpClient;
+  private readonly logger: Logger;
+  // User-Agent header value. Composed once at construction from
+  // createDefault() merged with this package's identity and the active
+  // credential's name.
+  private readonly userAgent: string;
+
+  constructor(options: ClientOptions) {
+    if (options.host === undefined) {
+      throw new Error('Host is required.');
+    }
+    this.host = options.host.replace(/\/$/, '');
+    this.workspaceId = options.workspaceId;
+    this.logger = options.logger ?? new NoOpLogger();
+    const info = createDefault()
+      .with(PACKAGE_SEGMENT)
+      .with({key: 'sdk-js-auth', value: AUTH_VERSION})
+      .with({key: 'auth', value: options.credentials?.name() ?? 'default'});
+    this.userAgent = info.toString();
+    this.httpClient = newHttpClient(options);
+  }
+
+  /**
+   * Creates a new registered model in Unity Catalog.
+   *
+   * File storage for model versions in the registered model
+   * will be located in the default location which is specified by the parent schema,
+   * or the parent catalog, or the Metastore.
+   *
+   * For registered model creation to succeed, the user must satisfy the following conditions:
+   * - The caller must be a metastore admin, or be the owner of the parent catalog and schema,
+   * or have the **USE_CATALOG** privilege on the parent catalog
+   * and the **USE_SCHEMA** privilege on the parent schema.
+   * - The caller must have the **CREATE MODEL** or **CREATE FUNCTION** privilege on the parent schema.
+   */
+  async createRegisteredModel(
+    req: CreateRegisteredModelRequest,
+    options?: CallOptions
+  ): Promise<RegisteredModelInfo> {
+    const url = `${this.host}/api/2.1/unity-catalog/models`;
+    const body = marshalRequest(req, marshalCreateRegisteredModelRequestSchema);
+    let resp: RegisteredModelInfo | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalRegisteredModelInfoSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Deletes a model version from the specified registered model. Any aliases assigned to the
+   * model version will also be deleted.
+   *
+   * The caller must be a metastore admin or an owner of the parent registered model.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   */
+  async deleteModelVersion(
+    req: DeleteModelVersionRequest,
+    options?: CallOptions
+  ): Promise<DeleteModelVersionRequest_Response> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/versions/${String(req.versionArg ?? '')}`;
+    let resp: DeleteModelVersionRequest_Response | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalDeleteModelVersionRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Deletes a registered model and all its model versions from the specified parent catalog and schema.
+   *
+   * The caller must be a metastore admin or an owner of the registered model.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   */
+  async deleteRegisteredModel(
+    req: DeleteRegisteredModelRequest,
+    options?: CallOptions
+  ): Promise<DeleteRegisteredModelRequest_Response> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}`;
+    let resp: DeleteRegisteredModelRequest_Response | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalDeleteRegisteredModelRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Deletes a registered model alias.
+   *
+   * The caller must be a metastore admin or an owner of the registered model.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   */
+  async deleteRegisteredModelAlias(
+    req: DeleteRegisteredModelAliasRequest,
+    options?: CallOptions
+  ): Promise<DeleteRegisteredModelAliasRequest_Response> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/aliases/${req.aliasArg ?? ''}`;
+    let resp: DeleteRegisteredModelAliasRequest_Response | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalDeleteRegisteredModelAliasRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Get a model version.
+   *
+   * The caller must be a metastore admin or an owner of (or have the **EXECUTE**
+   * privilege on) the parent registered model. For the latter case, the caller must also be the owner
+   * or have the **USE_CATALOG** privilege on the parent catalog and the **USE_SCHEMA** privilege
+   * on the parent schema.
+   */
+  async getModelVersion(
+    req: GetModelVersionRequest,
+    options?: CallOptions
+  ): Promise<ModelVersionInfo> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/versions/${String(req.versionArg ?? '')}`;
+    const params = new URLSearchParams();
+    if (req.includeAliases !== undefined) {
+      params.append('include_aliases', String(req.includeAliases));
+    }
+    if (req.includeBrowse !== undefined) {
+      params.append('include_browse', String(req.includeBrowse));
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: ModelVersionInfo | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalModelVersionInfoSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Get a model version by alias.
+   *
+   * The caller must be a metastore admin or an owner of (or have the **EXECUTE**
+   * privilege on) the registered model. For the latter case, the caller must also be the owner
+   * or have the **USE_CATALOG** privilege on the parent catalog and the **USE_SCHEMA** privilege
+   * on the parent schema.
+   */
+  async getModelVersionByAlias(
+    req: GetModelVersionByAliasRequest,
+    options?: CallOptions
+  ): Promise<ModelVersionInfo> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/aliases/${req.aliasArg ?? ''}`;
+    const params = new URLSearchParams();
+    if (req.includeAliases !== undefined) {
+      params.append('include_aliases', String(req.includeAliases));
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: ModelVersionInfo | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalModelVersionInfoSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Get a registered model.
+   *
+   * The caller must be a metastore admin or an owner of (or have the **EXECUTE**
+   * privilege on) the registered model. For the latter case, the caller must also be the owner
+   * or have the **USE_CATALOG** privilege on the parent catalog and the **USE_SCHEMA** privilege
+   * on the parent schema.
+   */
+  async getRegisteredModel(
+    req: GetRegisteredModelRequest,
+    options?: CallOptions
+  ): Promise<RegisteredModelInfo> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}`;
+    const params = new URLSearchParams();
+    if (req.includeAliases !== undefined) {
+      params.append('include_aliases', String(req.includeAliases));
+    }
+    if (req.includeBrowse !== undefined) {
+      params.append('include_browse', String(req.includeBrowse));
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: RegisteredModelInfo | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalRegisteredModelInfoSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * List model versions. You can list model versions under a particular schema,
+   * or list all model versions in the current metastore.
+   *
+   * The returned models are filtered based on the privileges of the calling user.
+   * For example, the metastore admin is able to list all the model versions.
+   * A regular user needs to be the owner or have the **EXECUTE** privilege
+   * on the parent registered model to recieve the model versions in the response.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   *
+   * There is no guarantee of a specific ordering of the elements in the response. The
+   * elements in the response will not contain any aliases or tags.
+   *
+   * PAGINATION BEHAVIOR: The API is by default paginated, a page may contain zero results while still providing a next_page_token.
+   * Clients must continue reading pages until next_page_token is absent, which is the only indication that the end of results has been reached.
+   */
+  async listModelVersions(
+    req: ListModelVersionsRequest,
+    options?: CallOptions
+  ): Promise<ListModelVersionsRequest_Response> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/versions`;
+    const params = new URLSearchParams();
+    if (req.maxResults !== undefined) {
+      params.append('max_results', String(req.maxResults));
+    }
+    if (req.pageToken !== undefined) {
+      params.append('page_token', req.pageToken);
+    }
+    if (req.includeBrowse !== undefined) {
+      params.append('include_browse', String(req.includeBrowse));
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: ListModelVersionsRequest_Response | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalListModelVersionsRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  async *listModelVersionsIter(
+    req: ListModelVersionsRequest,
+    options?: CallOptions
+  ): AsyncGenerator<ModelVersionInfo> {
+    const pageReq: ListModelVersionsRequest = {...req};
+    for (;;) {
+      const resp = await this.listModelVersions(pageReq, options);
+      for (const item of resp.modelVersions ?? []) {
+        yield item;
+      }
+      if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
+        return;
+      }
+      pageReq.pageToken = resp.nextPageToken;
+    }
+  }
+
+  /**
+   * List registered models. You can list registered models under a particular schema,
+   * or list all registered models in the current metastore.
+   *
+   * The returned models are filtered based on the privileges of the calling user.
+   * For example, the metastore admin is able to list all the registered models.
+   * A regular user needs to be the owner or have the **EXECUTE** privilege
+   * on the registered model to recieve the registered models in the response.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   *
+   * There is no guarantee of a specific ordering of the elements in the response.
+   *
+   * PAGINATION BEHAVIOR: The API is by default paginated, a page may contain zero results while still providing a next_page_token.
+   * Clients must continue reading pages until next_page_token is absent, which is the only indication that the end of results has been reached.
+   */
+  async listRegisteredModels(
+    req: ListRegisteredModelsRequest,
+    options?: CallOptions
+  ): Promise<ListRegisteredModelsRequest_Response> {
+    const url = `${this.host}/api/2.1/unity-catalog/models`;
+    const params = new URLSearchParams();
+    if (req.catalogName !== undefined) {
+      params.append('catalog_name', req.catalogName);
+    }
+    if (req.schemaName !== undefined) {
+      params.append('schema_name', req.schemaName);
+    }
+    if (req.includeBrowse !== undefined) {
+      params.append('include_browse', String(req.includeBrowse));
+    }
+    if (req.maxResults !== undefined) {
+      params.append('max_results', String(req.maxResults));
+    }
+    if (req.pageToken !== undefined) {
+      params.append('page_token', req.pageToken);
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: ListRegisteredModelsRequest_Response | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalListRegisteredModelsRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  async *listRegisteredModelsIter(
+    req: ListRegisteredModelsRequest,
+    options?: CallOptions
+  ): AsyncGenerator<RegisteredModelInfo> {
+    const pageReq: ListRegisteredModelsRequest = {...req};
+    for (;;) {
+      const resp = await this.listRegisteredModels(pageReq, options);
+      for (const item of resp.registeredModels ?? []) {
+        yield item;
+      }
+      if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
+        return;
+      }
+      pageReq.pageToken = resp.nextPageToken;
+    }
+  }
+
+  /**
+   * Set an alias on the specified registered model.
+   *
+   * The caller must be a metastore admin or an owner of the registered model.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   */
+  async setRegisteredModelAlias(
+    req: SetRegisteredModelAliasRequest,
+    options?: CallOptions
+  ): Promise<RegisteredModelAliasInfo> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/aliases/${req.aliasArg ?? ''}`;
+    const body = marshalRequest(
+      req,
+      marshalSetRegisteredModelAliasRequestSchema
+    );
+    let resp: RegisteredModelAliasInfo | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('PUT', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalRegisteredModelAliasInfoSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Updates the specified model version.
+   *
+   * The caller must be a metastore admin or an owner of the parent registered model.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   *
+   * Currently only the comment of the model version can be updated.
+   */
+  async updateModelVersion(
+    req: UpdateModelVersionRequest,
+    options?: CallOptions
+  ): Promise<ModelVersionInfo> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/versions/${String(req.versionArg ?? '')}`;
+    const body = marshalRequest(req, marshalUpdateModelVersionRequestSchema);
+    let resp: ModelVersionInfo | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalModelVersionInfoSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Updates the specified registered model.
+   *
+   * The caller must be a metastore admin or an owner of the registered model.
+   * For the latter case, the caller must also be the owner or have the **USE_CATALOG**
+   * privilege on the parent catalog and the **USE_SCHEMA** privilege on the parent schema.
+   *
+   * Currently only the name, the owner or the comment of the registered model can be updated.
+   */
+  async updateRegisteredModel(
+    req: UpdateRegisteredModelRequest,
+    options?: CallOptions
+  ): Promise<RegisteredModelInfo> {
+    const url = `${this.host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}`;
+    const body = marshalRequest(req, marshalUpdateRegisteredModelRequestSchema);
+    let resp: RegisteredModelInfo | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalRegisteredModelInfoSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+}