@databricks/sdk-uc-registeredmodels 0.34.0 → 0.36.0

package/dist/v1/client.js

@@ -1,528 +1,421 @@
-// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
-import { VERSION as AUTH_VERSION } from '@databricks/sdk-auth';
-import { createDefault } from '@databricks/sdk-core/clientinfo';
-import { NoOpLogger } from '@databricks/sdk-core/logger';
-import { resolveClientConfig } from './transport';
-import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse, } from './utils';
-import pkgJson from '../../package.json' with { type: 'json' };
-import { marshalCreateRegisteredModelRequestSchema, marshalSetRegisteredModelAliasRequestSchema, marshalUpdateModelVersionRequestSchema, marshalUpdateRegisteredModelRequestSchema, unmarshalDeleteModelVersionResponseSchema, unmarshalDeleteRegisteredModelAliasResponseSchema, unmarshalDeleteRegisteredModelResponseSchema, unmarshalListModelVersionsResponseSchema, unmarshalListRegisteredModelsResponseSchema, unmarshalModelVersionInfoSchema, unmarshalRegisteredModelAliasInfoSchema, unmarshalRegisteredModelInfoSchema, } from './model';
-// Package identity segment for this client to be used in the User-Agent header.
+import { resolveClientConfig } from "./transport.js";
+import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse } from "./utils.js";
+import { name, version } from "../package.js";
+import { marshalCreateRegisteredModelRequestSchema, marshalSetRegisteredModelAliasRequestSchema, marshalUpdateModelVersionRequestSchema, marshalUpdateRegisteredModelRequestSchema, unmarshalDeleteModelVersionResponseSchema, unmarshalDeleteRegisteredModelAliasResponseSchema, unmarshalDeleteRegisteredModelResponseSchema, unmarshalListModelVersionsResponseSchema, unmarshalListRegisteredModelsResponseSchema, unmarshalModelVersionInfoSchema, unmarshalRegisteredModelAliasInfoSchema, unmarshalRegisteredModelInfoSchema } from "./model.js";
+import { VERSION } from "@databricks/sdk-auth";
+import { createDefault } from "@databricks/sdk-core/clientinfo";
+import { NoOpLogger } from "@databricks/sdk-core/logger";
+
+//#region src/v1/client.ts
 const PACKAGE_SEGMENT = {
-    key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
-    value: pkgJson.version,
+	key: "sdk-js-" + name.replace(/^@[^/]+\/sdk-/, ""),
+	value: version
 };
-export class RegisteredModelsClient {
-    options;
-    logger;
-    // User-Agent header value. Composed once at construction from
-    // createDefault() merged with this package's identity and the active
-    // credential's name.
-    userAgent;
-    // Memoized configuration. The profile is resolved once, lazily, on the first
-    // request, then reused; host, workspaceId/accountId, and credentials are
-    // filled from it when not set explicitly on the options.
-    config;
-    constructor(options) {
-        this.options = options;
-        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();
-    }
-    resolveConfig() {
-        this.config ??= resolveClientConfig(this.options);
-        return this.config;
-    }
-    /**
-     * 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/2.1/unity-catalog/models`;
-        const body = marshalRequest(req, marshalCreateRegisteredModelRequestSchema);
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalRegisteredModelInfoSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/versions/${String(req.versionArg ?? '')}`;
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalDeleteModelVersionResponseSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}`;
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalDeleteRegisteredModelResponseSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/aliases/${req.aliasArg ?? ''}`;
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalDeleteRegisteredModelAliasResponseSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${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;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalModelVersionInfoSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${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;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalModelVersionInfoSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${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;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalRegisteredModelInfoSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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 receive 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${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;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalListModelVersionsResponseSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    async *listModelVersionsIter(req, options) {
-        const pageReq = { ...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 receive 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${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;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalListRegisteredModelsResponseSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    async *listRegisteredModelsIter(req, options) {
-        const pageReq = { ...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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/aliases/${req.aliasArg ?? ''}`;
-        const body = marshalRequest(req, marshalSetRegisteredModelAliasRequestSchema);
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('PUT', url, headers, callSignal, body);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalRegisteredModelAliasInfoSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}/versions/${String(req.versionArg ?? '')}`;
-        const body = marshalRequest(req, marshalUpdateModelVersionRequestSchema);
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalModelVersionInfoSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation 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, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ''}`;
-        const body = marshalRequest(req, marshalUpdateRegisteredModelRequestSchema);
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalRegisteredModelInfoSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-}
+var RegisteredModelsClient = class {
+	options;
+	logger;
+	userAgent;
+	config;
+	constructor(options) {
+		this.options = options;
+		this.logger = options.logger ?? new NoOpLogger();
+		this.userAgent = createDefault().with(PACKAGE_SEGMENT).with({
+			key: "sdk-js-auth",
+			value: VERSION
+		}).with({
+			key: "auth",
+			value: options.credentials?.name() ?? "default"
+		}).toString();
+	}
+	resolveConfig() {
+		this.config ??= resolveClientConfig(this.options);
+		return this.config;
+	}
+	/**
+	* 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models`;
+		const body = marshalRequest(req, marshalCreateRegisteredModelRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalRegisteredModelInfoSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}/versions/${String(req.versionArg ?? "")}`;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalDeleteModelVersionResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}`;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalDeleteRegisteredModelResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}/aliases/${req.aliasArg ?? ""}`;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalDeleteRegisteredModelAliasResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}/versions/${String(req.versionArg ?? "")}`;
+		const params = new URLSearchParams();
+		if (req.includeAliases !== void 0) params.append("include_aliases", String(req.includeAliases));
+		if (req.includeBrowse !== void 0) params.append("include_browse", String(req.includeBrowse));
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalModelVersionInfoSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}/aliases/${req.aliasArg ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.includeAliases !== void 0) params.append("include_aliases", String(req.includeAliases));
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalModelVersionInfoSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.includeAliases !== void 0) params.append("include_aliases", String(req.includeAliases));
+		if (req.includeBrowse !== void 0) params.append("include_browse", String(req.includeBrowse));
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalRegisteredModelInfoSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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 receive 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}/versions`;
+		const params = new URLSearchParams();
+		if (req.maxResults !== void 0) params.append("max_results", String(req.maxResults));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		if (req.includeBrowse !== void 0) params.append("include_browse", String(req.includeBrowse));
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListModelVersionsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listModelVersionsIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listModelVersions(pageReq, options);
+			for (const item of resp.modelVersions ?? []) yield item;
+			if (resp.nextPageToken === void 0 || 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 receive 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models`;
+		const params = new URLSearchParams();
+		if (req.catalogName !== void 0) params.append("catalog_name", req.catalogName);
+		if (req.schemaName !== void 0) params.append("schema_name", req.schemaName);
+		if (req.includeBrowse !== void 0) params.append("include_browse", String(req.includeBrowse));
+		if (req.maxResults !== void 0) params.append("max_results", String(req.maxResults));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListRegisteredModelsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listRegisteredModelsIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listRegisteredModels(pageReq, options);
+			for (const item of resp.registeredModels ?? []) yield item;
+			if (resp.nextPageToken === void 0 || 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}/aliases/${req.aliasArg ?? ""}`;
+		const body = marshalRequest(req, marshalSetRegisteredModelAliasRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PUT", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalRegisteredModelAliasInfoSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}/versions/${String(req.versionArg ?? "")}`;
+		const body = marshalRequest(req, marshalUpdateModelVersionRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalModelVersionInfoSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation 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, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.1/unity-catalog/models/${req.fullNameArg ?? ""}`;
+		const body = marshalRequest(req, marshalUpdateRegisteredModelRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalRegisteredModelInfoSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+};
+
+//#endregion
+export { RegisteredModelsClient };
 //# sourceMappingURL=client.js.map