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

package/dist/v1/client.js

@@ -5,13 +5,13 @@ import { NoOpLogger } from '@databricks/sdk-core/logger';
 import { newHttpClient } 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, marshalGetLatestVersionsRequestSchema, marshalRejectTransitionRequestSchema, marshalRenameRegisteredModelRequestSchema, marshalSetModelVersionTagRequestSchema, marshalSetRegisteredModelTagRequestSchema, marshalTestRegistryWebhookRequestSchema, marshalTransitionModelVersionStageDatabricksRequestSchema, marshalUpdateCommentRequestSchema, marshalUpdateModelVersionRequestSchema, marshalUpdateRegisteredModelRequestSchema, marshalUpdateRegistryWebhookRequestSchema, unmarshalApproveTransitionRequest_ResponseSchema, unmarshalCreateCommentRequest_ResponseSchema, unmarshalCreateModelVersionRequest_ResponseSchema, unmarshalCreateRegisteredModelRequest_ResponseSchema, unmarshalCreateRegistryWebhookRequest_ResponseSchema, unmarshalCreateTransitionRequest_ResponseSchema, unmarshalDeleteCommentRequest_ResponseSchema, unmarshalDeleteModelVersionRequest_ResponseSchema, unmarshalDeleteModelVersionTagRequest_ResponseSchema, unmarshalDeleteRegisteredModelRequest_ResponseSchema, unmarshalDeleteRegisteredModelTagRequest_ResponseSchema, unmarshalDeleteRegistryWebhookRequest_ResponseSchema, unmarshalDeleteTransitionRequest_ResponseSchema, unmarshalGetLatestVersionsRequest_ResponseSchema, unmarshalGetModelVersionDownloadUriRequest_ResponseSchema, unmarshalGetModelVersionRequest_ResponseSchema, unmarshalGetRegisteredModelDatabricksRequest_ResponseSchema, unmarshalListRegisteredModelsRequest_ResponseSchema, unmarshalListRegistryWebhooksRequest_ResponseSchema, unmarshalListTransitionRequest_ResponseSchema, unmarshalRejectTransitionRequest_ResponseSchema, unmarshalRenameRegisteredModelRequest_ResponseSchema, unmarshalSearchModelVersionsRequest_ResponseSchema, unmarshalSearchRegisteredModelsRequest_ResponseSchema, unmarshalSetModelVersionTagRequest_ResponseSchema, unmarshalSetRegisteredModelTagRequest_ResponseSchema, unmarshalTestRegistryWebhookRequest_ResponseSchema, unmarshalTransitionModelVersionStageDatabricksRequest_ResponseSchema, unmarshalUpdateCommentRequest_ResponseSchema, unmarshalUpdateModelVersionRequest_ResponseSchema, unmarshalUpdateRegisteredModelRequest_ResponseSchema, unmarshalUpdateRegistryWebhookRequest_ResponseSchema, } from './model';
+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';
 // Package identity segment for this client to be used in the User-Agent header.
 const PACKAGE_SEGMENT = {
     key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
     value: pkgJson.version,
 };
-export class ModelregistryClient {
+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
@@ -54,11 +54,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalApproveTransitionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalApproveTransitionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -82,11 +82,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalCreateCommentRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalCreateCommentResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -110,11 +110,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalCreateRegistryWebhookRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalCreateRegistryWebhookResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -135,11 +135,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalCreateTransitionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalCreateTransitionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -165,11 +165,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalDeleteCommentRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalDeleteCommentResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -198,11 +198,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalDeleteRegistryWebhookRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalDeleteRegistryWebhookResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -240,11 +240,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalDeleteTransitionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalDeleteTransitionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -274,11 +274,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalGetRegisteredModelDatabricksRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalGetRegisteredModelDatabricksResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -316,11 +316,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalListRegistryWebhooksRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalListRegistryWebhooksResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -362,11 +362,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalListTransitionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalListTransitionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -387,11 +387,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalRejectTransitionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalRejectTransitionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -415,11 +415,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalTestRegistryWebhookRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalTestRegistryWebhookResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -444,11 +444,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalTransitionModelVersionStageDatabricksRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalTransitionModelVersionStageDatabricksResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -469,11 +469,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalUpdateCommentRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalUpdateCommentResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -497,11 +497,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalUpdateRegistryWebhookRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalUpdateRegistryWebhookResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -522,11 +522,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalCreateModelVersionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalCreateModelVersionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -550,11 +550,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalCreateRegisteredModelRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalCreateRegisteredModelResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -583,11 +583,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalDeleteModelVersionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalDeleteModelVersionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -619,11 +619,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalDeleteModelVersionTagRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalDeleteModelVersionTagResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -649,11 +649,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalDeleteRegisteredModelRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalDeleteRegisteredModelResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -682,36 +682,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalDeleteRegisteredModelTagRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalDeleteRegisteredModelTagResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
-        }
-        return resp;
-    }
-    /** Gets the latest version of a registered model. */
-    async getLatestVersions(req, options) {
-        const url = `${this.host}/api/2.0/mlflow/registered-models/get-latest-versions`;
-        const body = marshalRequest(req, marshalGetLatestVersionsRequestSchema);
-        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);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient: this.httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalGetLatestVersionsRequest_ResponseSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -740,11 +715,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalGetModelVersionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalGetModelVersionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -773,11 +748,36 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalGetModelVersionDownloadUriRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalGetModelVersionDownloadUriResponseSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
+    /** 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 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);
+            }
+            headers.set('User-Agent', this.userAgent);
+            const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
+            const respBody = await executeHttpCall({
+                request: httpReq,
+                httpClient: this.httpClient,
+                logger: this.logger,
+            });
+            resp = parseResponse(respBody, unmarshalGetLatestVersionsResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -806,11 +806,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalListRegisteredModelsRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalListRegisteredModelsResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -844,11 +844,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalRenameRegisteredModelRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalRenameRegisteredModelResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -883,11 +883,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalSearchModelVersionsRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalSearchModelVersionsResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -935,11 +935,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalSearchRegisteredModelsRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalSearchRegisteredModelsResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -973,11 +973,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalSetModelVersionTagRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalSetModelVersionTagResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -998,11 +998,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalSetRegisteredModelTagRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalSetRegisteredModelTagResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -1023,11 +1023,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalUpdateModelVersionRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalUpdateModelVersionResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }
@@ -1048,11 +1048,11 @@ export class ModelregistryClient {
                 httpClient: this.httpClient,
                 logger: this.logger,
             });
-            resp = parseResponse(respBody, unmarshalUpdateRegisteredModelRequest_ResponseSchema);
+            resp = parseResponse(respBody, unmarshalUpdateRegisteredModelResponseSchema);
         };
         await executeCall(call, options);
         if (resp === undefined) {
-            throw new Error('API call completed without a result.');
+            throw new Error('operation completed without a result.');
         }
         return resp;
     }