@databricks/sdk-modelregistry 0.1.0-dev.3 → 0.1.0-dev.5

package/dist/v1/client.js

@@ -2,7 +2,7 @@
 import { VERSION as AUTH_VERSION } from '@databricks/sdk-auth';
 import { createDefault } from '@databricks/sdk-core/clientinfo';
 import { NoOpLogger } from '@databricks/sdk-core/logger';
-import { newHttpClient } from './transport';
+import { resolveClientConfig } from './transport';
 import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse, } from './utils';
 import pkgJson from '../../package.json' with { type: 'json' };
 import { marshalApproveTransitionRequestSchema, marshalCreateCommentRequestSchema, marshalCreateModelVersionRequestSchema, marshalCreateRegisteredModelRequestSchema, marshalCreateRegistryWebhookRequestSchema, marshalCreateTransitionRequestSchema, marshalListLatestVersionsRequestSchema, marshalRejectTransitionRequestSchema, marshalRenameRegisteredModelRequestSchema, marshalSetModelVersionTagRequestSchema, marshalSetRegisteredModelTagRequestSchema, marshalTestRegistryWebhookRequestSchema, marshalTransitionModelVersionStageDatabricksRequestSchema, marshalUpdateCommentRequestSchema, marshalUpdateModelVersionRequestSchema, marshalUpdateRegisteredModelRequestSchema, marshalUpdateRegistryWebhookRequestSchema, unmarshalApproveTransitionResponseSchema, unmarshalCreateCommentResponseSchema, unmarshalCreateModelVersionResponseSchema, unmarshalCreateRegisteredModelResponseSchema, unmarshalCreateRegistryWebhookResponseSchema, unmarshalCreateTransitionResponseSchema, unmarshalDeleteCommentResponseSchema, unmarshalDeleteModelVersionResponseSchema, unmarshalDeleteModelVersionTagResponseSchema, unmarshalDeleteRegisteredModelResponseSchema, unmarshalDeleteRegisteredModelTagResponseSchema, unmarshalDeleteRegistryWebhookResponseSchema, unmarshalDeleteTransitionResponseSchema, unmarshalGetLatestVersionsResponseSchema, unmarshalGetModelVersionDownloadUriResponseSchema, unmarshalGetModelVersionResponseSchema, unmarshalGetRegisteredModelDatabricksResponseSchema, unmarshalListRegisteredModelsResponseSchema, unmarshalListRegistryWebhooksResponseSchema, unmarshalListTransitionResponseSchema, unmarshalRejectTransitionResponseSchema, unmarshalRenameRegisteredModelResponseSchema, unmarshalSearchModelVersionsResponseSchema, unmarshalSearchRegisteredModelsResponseSchema, unmarshalSetModelVersionTagResponseSchema, unmarshalSetRegisteredModelTagResponseSchema, unmarshalTestRegistryWebhookResponseSchema, unmarshalTransitionModelVersionStageDatabricksResponseSchema, unmarshalUpdateCommentResponseSchema, unmarshalUpdateModelVersionResponseSchema, unmarshalUpdateRegisteredModelResponseSchema, unmarshalUpdateRegistryWebhookResponseSchema, } from './model';
@@ -12,46 +12,45 @@ const PACKAGE_SEGMENT = {
     value: pkgJson.version,
 };
 export class ModelRegistryClient {
-    host;
-    // 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.
-    workspaceId;
-    httpClient;
+    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) {
-        if (options.host === undefined) {
-            throw new Error('Host is required.');
-        }
-        this.host = options.host.replace(/\/$/, '');
-        this.workspaceId = options.workspaceId;
+        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();
-        this.httpClient = newHttpClient(options);
+    }
+    resolveConfig() {
+        this.config ??= resolveClientConfig(this.options);
+        return this.config;
     }
     /** Approves a model version stage transition request. */
     async approveTransitionRequest(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/transition-requests/approve`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/transition-requests/approve`;
         const body = marshalRequest(req, marshalApproveTransitionRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalApproveTransitionResponseSchema);
@@ -67,19 +66,20 @@ export class ModelRegistryClient {
      * relevant information about the model. For example, test results or deployment errors.
      */
     async createComment(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/comments/create`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/comments/create`;
         const body = marshalRequest(req, marshalCreateCommentRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalCreateCommentResponseSchema);
@@ -95,19 +95,20 @@ export class ModelRegistryClient {
      * Creates a registry webhook.
      */
     async createRegistryWebhook(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registry-webhooks/create`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registry-webhooks/create`;
         const body = marshalRequest(req, marshalCreateRegistryWebhookRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalCreateRegistryWebhookResponseSchema);
@@ -120,19 +121,20 @@ export class ModelRegistryClient {
     }
     /** Creates a model version stage transition request. */
     async createTransitionRequest(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/transition-requests/create`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/transition-requests/create`;
         const body = marshalRequest(req, marshalCreateTransitionRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalCreateTransitionResponseSchema);
@@ -145,7 +147,8 @@ export class ModelRegistryClient {
     }
     /** Deletes a comment on a model version. */
     async deleteComment(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/comments/delete`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/comments/delete`;
         const params = new URLSearchParams();
         if (req.id !== undefined) {
             params.append('id', req.id);
@@ -155,14 +158,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalDeleteCommentResponseSchema);
@@ -178,7 +181,8 @@ export class ModelRegistryClient {
      * Deletes a registry webhook.
      */
     async deleteRegistryWebhook(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registry-webhooks/delete`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registry-webhooks/delete`;
         const params = new URLSearchParams();
         if (req.id !== undefined) {
             params.append('id', req.id);
@@ -188,14 +192,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalDeleteRegistryWebhookResponseSchema);
@@ -208,7 +212,8 @@ export class ModelRegistryClient {
     }
     /** Cancels a model version stage transition request. */
     async deleteTransitionRequest(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/transition-requests/delete`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/transition-requests/delete`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -230,14 +235,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalDeleteTransitionResponseSchema);
@@ -254,7 +259,8 @@ export class ModelRegistryClient {
      * that also returns the model's <Databricks> workspace ID and the permission level of the requesting user on the model.
      */
     async getRegisteredModelDatabricks(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/databricks/registered-models/get`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/databricks/registered-models/get`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -264,14 +270,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalGetRegisteredModelDatabricksResponseSchema);
@@ -287,7 +293,8 @@ export class ModelRegistryClient {
      * Lists all registry webhooks.
      */
     async listRegistryWebhooks(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registry-webhooks/list`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registry-webhooks/list`;
         const params = new URLSearchParams();
         if (req.modelName !== undefined) {
             params.append('model_name', req.modelName);
@@ -306,14 +313,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalListRegistryWebhooksResponseSchema);
@@ -339,7 +346,8 @@ export class ModelRegistryClient {
     }
     /** Gets a list of all open stage transition requests for the model version. */
     async listTransitionsRequest(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/transition-requests/list`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/transition-requests/list`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -352,14 +360,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalListTransitionResponseSchema);
@@ -372,19 +380,20 @@ export class ModelRegistryClient {
     }
     /** Rejects a model version stage transition request. */
     async rejectTransitionRequest(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/transition-requests/reject`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/transition-requests/reject`;
         const body = marshalRequest(req, marshalRejectTransitionRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalRejectTransitionResponseSchema);
@@ -400,19 +409,20 @@ export class ModelRegistryClient {
      * Tests a registry webhook.
      */
     async testRegistryWebhook(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registry-webhooks/test`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registry-webhooks/test`;
         const body = marshalRequest(req, marshalTestRegistryWebhookRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalTestRegistryWebhookResponseSchema);
@@ -429,19 +439,20 @@ export class ModelRegistryClient {
      * that also accepts a comment associated with the transition to be recorded.
      */
     async transitionModelVersionStageDatabricks(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/databricks/model-versions/transition-stage`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/databricks/model-versions/transition-stage`;
         const body = marshalRequest(req, marshalTransitionModelVersionStageDatabricksRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalTransitionModelVersionStageDatabricksResponseSchema);
@@ -454,19 +465,20 @@ export class ModelRegistryClient {
     }
     /** Post an edit to a comment on a model version. */
     async updateComment(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/comments/update`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/comments/update`;
         const body = marshalRequest(req, marshalUpdateCommentRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalUpdateCommentResponseSchema);
@@ -482,19 +494,20 @@ export class ModelRegistryClient {
      * Updates a registry webhook.
      */
     async updateRegistryWebhook(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registry-webhooks/update`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registry-webhooks/update`;
         const body = marshalRequest(req, marshalUpdateRegistryWebhookRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalUpdateRegistryWebhookResponseSchema);
@@ -507,19 +520,20 @@ export class ModelRegistryClient {
     }
     /** Creates a model version. */
     async createModelVersion(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/create`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/create`;
         const body = marshalRequest(req, marshalCreateModelVersionRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalCreateModelVersionResponseSchema);
@@ -535,19 +549,20 @@ export class ModelRegistryClient {
      * Throws `RESOURCE_ALREADY_EXISTS` if a registered model with the given name exists.
      */
     async createRegisteredModel(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/create`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/create`;
         const body = marshalRequest(req, marshalCreateRegisteredModelRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalCreateRegisteredModelResponseSchema);
@@ -560,7 +575,8 @@ export class ModelRegistryClient {
     }
     /** Deletes a model version. */
     async deleteModelVersion(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/delete`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/delete`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -573,14 +589,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalDeleteModelVersionResponseSchema);
@@ -593,7 +609,8 @@ export class ModelRegistryClient {
     }
     /** Deletes a model version tag. */
     async deleteModelVersionTag(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/delete-tag`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/delete-tag`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -609,14 +626,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalDeleteModelVersionTagResponseSchema);
@@ -629,7 +646,8 @@ export class ModelRegistryClient {
     }
     /** Deletes a registered model. */
     async deleteRegisteredModel(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/delete`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/delete`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -639,14 +657,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalDeleteRegisteredModelResponseSchema);
@@ -659,7 +677,8 @@ export class ModelRegistryClient {
     }
     /** Deletes the tag for a registered model. */
     async deleteRegisteredModelTag(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/delete-tag`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/delete-tag`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -672,14 +691,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalDeleteRegisteredModelTagResponseSchema);
@@ -692,7 +711,8 @@ export class ModelRegistryClient {
     }
     /** Get a model version. */
     async getModelVersion(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/get`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/get`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -705,14 +725,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalGetModelVersionResponseSchema);
@@ -725,7 +745,8 @@ export class ModelRegistryClient {
     }
     /** Gets a URI to download the model version. */
     async getModelVersionDownloadUri(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/get-download-uri`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/get-download-uri`;
         const params = new URLSearchParams();
         if (req.name !== undefined) {
             params.append('name', req.name);
@@ -738,14 +759,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalGetModelVersionDownloadUriResponseSchema);
@@ -758,19 +779,20 @@ export class ModelRegistryClient {
     }
     /** Gets the latest version of a registered model. */
     async listLatestVersions(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/get-latest-versions`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/get-latest-versions`;
         const body = marshalRequest(req, marshalListLatestVersionsRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalGetLatestVersionsResponseSchema);
@@ -783,7 +805,8 @@ export class ModelRegistryClient {
     }
     /** Lists all available registered models, up to the limit specified in __max_results__. */
     async listRegisteredModels(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/list`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/list`;
         const params = new URLSearchParams();
         if (req.maxResults !== undefined) {
             params.append('max_results', String(req.maxResults));
@@ -796,14 +819,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalListRegisteredModelsResponseSchema);
@@ -829,19 +852,20 @@ export class ModelRegistryClient {
     }
     /** Renames a registered model. */
     async renameRegisteredModel(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/rename`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/rename`;
         const body = marshalRequest(req, marshalRenameRegisteredModelRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalRenameRegisteredModelResponseSchema);
@@ -854,7 +878,8 @@ export class ModelRegistryClient {
     }
     /** Searches for specific model versions based on the supplied __filter__. */
     async searchModelVersions(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/search`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/search`;
         const params = new URLSearchParams();
         if (req.filter !== undefined) {
             params.append('filter', req.filter);
@@ -873,14 +898,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalSearchModelVersionsResponseSchema);
@@ -906,7 +931,8 @@ export class ModelRegistryClient {
     }
     /** Search for registered models based on the specified __filter__. */
     async searchRegisteredModels(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/search`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/search`;
         const params = new URLSearchParams();
         if (req.filter !== undefined) {
             params.append('filter', req.filter);
@@ -925,14 +951,14 @@ export class ModelRegistryClient {
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalSearchRegisteredModelsResponseSchema);
@@ -958,19 +984,20 @@ export class ModelRegistryClient {
     }
     /** Sets a model version tag. */
     async setModelVersionTag(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/set-tag`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/set-tag`;
         const body = marshalRequest(req, marshalSetModelVersionTagRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalSetModelVersionTagResponseSchema);
@@ -983,19 +1010,20 @@ export class ModelRegistryClient {
     }
     /** Sets a tag on a registered model. */
     async setRegisteredModelTag(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/set-tag`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/set-tag`;
         const body = marshalRequest(req, marshalSetRegisteredModelTagRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalSetRegisteredModelTagResponseSchema);
@@ -1008,19 +1036,20 @@ export class ModelRegistryClient {
     }
     /** Updates the model version. */
     async updateModelVersion(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/model-versions/update`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/model-versions/update`;
         const body = marshalRequest(req, marshalUpdateModelVersionRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalUpdateModelVersionResponseSchema);
@@ -1033,19 +1062,20 @@ export class ModelRegistryClient {
     }
     /** Updates a registered model. */
     async updateRegisteredModel(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/update`;
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/mlflow/registered-models/update`;
         const body = marshalRequest(req, marshalUpdateRegisteredModelRequestSchema);
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (this.workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', this.workspaceId);
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Org-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
             const respBody = await executeHttpCall({
                 request: httpReq,
-                httpClient: this.httpClient,
+                httpClient,
                 logger: this.logger,
             });
             resp = parseResponse(respBody, unmarshalUpdateRegisteredModelResponseSchema);