@databricks/sdk-postgres 0.1.0-dev.6 → 0.3.0

package/dist/v1/client.js

@@ -6,7 +6,7 @@ import { resolveClientConfig } from './transport';
 import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse, executeWait, StillRunningError, } from './utils';
 import pkgJson from '../../package.json' with { type: 'json' };
 import { z } from 'zod';
-import { marshalBranchSchema, marshalCatalogSchema, marshalDatabaseSchema, marshalEndpointSchema, marshalGenerateDatabaseCredentialRequestSchema, marshalProjectSchema, marshalRoleSchema, marshalSyncedTableSchema, marshalUndeleteBranchRequestSchema, marshalUndeleteProjectRequestSchema, unmarshalBranchOperationMetadataSchema, unmarshalBranchSchema, unmarshalCatalogOperationMetadataSchema, unmarshalCatalogSchema, unmarshalDatabaseCredentialSchema, unmarshalDatabaseOperationMetadataSchema, unmarshalDatabaseSchema, unmarshalEndpointOperationMetadataSchema, unmarshalEndpointSchema, unmarshalListBranchesResponseSchema, unmarshalListDatabasesResponseSchema, unmarshalListEndpointsResponseSchema, unmarshalListProjectsResponseSchema, unmarshalListRolesResponseSchema, unmarshalOperationSchema, unmarshalProjectOperationMetadataSchema, unmarshalProjectSchema, unmarshalRoleOperationMetadataSchema, unmarshalRoleSchema, unmarshalSyncedTableOperationMetadataSchema, unmarshalSyncedTableSchema, } from './model';
+import { marshalBranchSchema, marshalCatalogSchema, marshalDataApiSchema, marshalDatabaseSchema, marshalEndpointSchema, marshalGenerateDatabaseCredentialRequestSchema, marshalProjectSchema, marshalRoleSchema, marshalSyncedTableSchema, marshalUndeleteBranchRequestSchema, marshalUndeleteProjectRequestSchema, unmarshalBranchOperationMetadataSchema, unmarshalBranchSchema, unmarshalCatalogOperationMetadataSchema, unmarshalCatalogSchema, unmarshalDataApiOperationMetadataSchema, unmarshalDataApiSchema, unmarshalDatabaseCredentialSchema, unmarshalDatabaseOperationMetadataSchema, unmarshalDatabaseSchema, unmarshalEndpointOperationMetadataSchema, unmarshalEndpointSchema, unmarshalListBranchesResponseSchema, unmarshalListDatabasesResponseSchema, unmarshalListEndpointsResponseSchema, unmarshalListProjectsResponseSchema, unmarshalListRolesResponseSchema, unmarshalOperationSchema, unmarshalProjectOperationMetadataSchema, unmarshalProjectSchema, unmarshalRoleOperationMetadataSchema, unmarshalRoleSchema, unmarshalSyncedTableOperationMetadataSchema, unmarshalSyncedTableSchema, } 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-/, ''),
@@ -37,7 +37,7 @@ export class PostgresClient {
         return this.config;
     }
     /** Creates a new database branch in the project. */
-    async createBranch(req, options) {
+    async createBranchBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.parent ?? ''}/branches`;
         const params = new URLSearchParams();
@@ -54,7 +54,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -71,12 +71,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async createBranchOperation(req, options) {
-        const op = await this.createBranch(req, options);
+    /** Creates a new database branch in the project. */
+    async createBranch(req, options) {
+        const op = await this.createBranchBase(req, options);
         return new CreateBranchOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Register a Postgres database in the Unity Catalog. */
-    async createCatalog(req, options) {
+    async createCatalogBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/catalogs`;
         const params = new URLSearchParams();
@@ -90,7 +91,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -107,16 +108,48 @@ export class PostgresClient {
         }
         return resp;
     }
-    async createCatalogOperation(req, options) {
-        const op = await this.createCatalog(req, options);
+    /** Register a Postgres database in the Unity Catalog. */
+    async createCatalog(req, options) {
+        const op = await this.createCatalogBase(req, options);
         return new CreateCatalogOperation(op, (req, options) => this.getOperation(req, options));
     }
+    /** Enable Data API for a database. */
+    async createDataApiBase(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.parent ?? ''}/data-api`;
+        const body = marshalRequest(req.dataApi, marshalDataApiSchema);
+        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, unmarshalOperationSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
+    /** Enable Data API for a database. */
+    async createDataApi(req, options) {
+        const op = await this.createDataApiBase(req, options);
+        return new CreateDataApiOperation(op, (req, options) => this.getOperation(req, options));
+    }
     /**
      * Create a Database.
      *
      * Creates a database in the specified branch. A branch can have multiple databases.
      */
-    async createDatabase(req, options) {
+    async createDatabaseBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.parent ?? ''}/databases`;
         const params = new URLSearchParams();
@@ -130,7 +163,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -147,12 +180,17 @@ export class PostgresClient {
         }
         return resp;
     }
-    async createDatabaseOperation(req, options) {
-        const op = await this.createDatabase(req, options);
+    /**
+     * Create a Database.
+     *
+     * Creates a database in the specified branch. A branch can have multiple databases.
+     */
+    async createDatabase(req, options) {
+        const op = await this.createDatabaseBase(req, options);
         return new CreateDatabaseOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Creates a new compute endpoint in the branch. */
-    async createEndpoint(req, options) {
+    async createEndpointBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.parent ?? ''}/endpoints`;
         const params = new URLSearchParams();
@@ -169,7 +207,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -186,12 +224,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async createEndpointOperation(req, options) {
-        const op = await this.createEndpoint(req, options);
+    /** Creates a new compute endpoint in the branch. */
+    async createEndpoint(req, options) {
+        const op = await this.createEndpointBase(req, options);
         return new CreateEndpointOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Creates a new Lakebase Autoscaling Postgres database project, which contains branches and compute endpoints. */
-    async createProject(req, options) {
+    async createProjectBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/projects`;
         const params = new URLSearchParams();
@@ -205,7 +244,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -222,12 +261,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async createProjectOperation(req, options) {
-        const op = await this.createProject(req, options);
+    /** Creates a new Lakebase Autoscaling Postgres database project, which contains branches and compute endpoints. */
+    async createProject(req, options) {
+        const op = await this.createProjectBase(req, options);
         return new CreateProjectOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Creates a new Postgres role in the branch. */
-    async createRole(req, options) {
+    async createRoleBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.parent ?? ''}/roles`;
         const params = new URLSearchParams();
@@ -241,7 +281,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -258,12 +298,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async createRoleOperation(req, options) {
-        const op = await this.createRole(req, options);
+    /** Creates a new Postgres role in the branch. */
+    async createRole(req, options) {
+        const op = await this.createRoleBase(req, options);
         return new CreateRoleOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Create a Synced Table. */
-    async createSyncedTable(req, options) {
+    async createSyncedTableBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/synced_tables`;
         const params = new URLSearchParams();
@@ -277,7 +318,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -294,12 +335,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async createSyncedTableOperation(req, options) {
-        const op = await this.createSyncedTable(req, options);
+    /** Create a Synced Table. */
+    async createSyncedTable(req, options) {
+        const op = await this.createSyncedTableBase(req, options);
         return new CreateSyncedTableOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Deletes the specified database branch. */
-    async deleteBranch(req, options) {
+    async deleteBranchBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
         const params = new URLSearchParams();
@@ -312,7 +354,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
@@ -329,19 +371,20 @@ export class PostgresClient {
         }
         return resp;
     }
-    async deleteBranchOperation(req, options) {
-        const op = await this.deleteBranch(req, options);
+    /** Deletes the specified database branch. */
+    async deleteBranch(req, options) {
+        const op = await this.deleteBranchBase(req, options);
         return new DeleteBranchOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Delete a Database Catalog. */
-    async deleteCatalog(req, options) {
+    async deleteCatalogBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -358,19 +401,50 @@ export class PostgresClient {
         }
         return resp;
     }
-    async deleteCatalogOperation(req, options) {
-        const op = await this.deleteCatalog(req, options);
+    /** Delete a Database Catalog. */
+    async deleteCatalog(req, options) {
+        const op = await this.deleteCatalogBase(req, options);
         return new DeleteCatalogOperation(op, (req, options) => this.getOperation(req, options));
     }
+    /** Disable Data API for a database. */
+    async deleteDataApiBase(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
+        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, unmarshalOperationSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
+    /** Disable Data API for a database. */
+    async deleteDataApi(req, options) {
+        const op = await this.deleteDataApiBase(req, options);
+        return new DeleteDataApiOperation(op, (req, options) => this.getOperation(req, options));
+    }
     /** Delete a Database. */
-    async deleteDatabase(req, options) {
+    async deleteDatabaseBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -387,19 +461,20 @@ export class PostgresClient {
         }
         return resp;
     }
-    async deleteDatabaseOperation(req, options) {
-        const op = await this.deleteDatabase(req, options);
+    /** Delete a Database. */
+    async deleteDatabase(req, options) {
+        const op = await this.deleteDatabaseBase(req, options);
         return new DeleteDatabaseOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Deletes the specified compute endpoint. */
-    async deleteEndpoint(req, options) {
+    async deleteEndpointBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -416,12 +491,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async deleteEndpointOperation(req, options) {
-        const op = await this.deleteEndpoint(req, options);
+    /** Deletes the specified compute endpoint. */
+    async deleteEndpoint(req, options) {
+        const op = await this.deleteEndpointBase(req, options);
         return new DeleteEndpointOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Deletes the specified database project. */
-    async deleteProject(req, options) {
+    async deleteProjectBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
         const params = new URLSearchParams();
@@ -434,7 +510,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
@@ -451,12 +527,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async deleteProjectOperation(req, options) {
-        const op = await this.deleteProject(req, options);
+    /** Deletes the specified database project. */
+    async deleteProject(req, options) {
+        const op = await this.deleteProjectBase(req, options);
         return new DeleteProjectOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Deletes the specified Postgres role. */
-    async deleteRole(req, options) {
+    async deleteRoleBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
         const params = new URLSearchParams();
@@ -469,7 +546,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
@@ -486,19 +563,20 @@ export class PostgresClient {
         }
         return resp;
     }
-    async deleteRoleOperation(req, options) {
-        const op = await this.deleteRole(req, options);
+    /** Deletes the specified Postgres role. */
+    async deleteRole(req, options) {
+        const op = await this.deleteRoleBase(req, options);
         return new DeleteRoleOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Delete a Synced Table. */
-    async deleteSyncedTable(req, options) {
+    async deleteSyncedTableBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
         let resp;
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -515,8 +593,9 @@ export class PostgresClient {
         }
         return resp;
     }
-    async deleteSyncedTableOperation(req, options) {
-        const op = await this.deleteSyncedTable(req, options);
+    /** Delete a Synced Table. */
+    async deleteSyncedTable(req, options) {
+        const op = await this.deleteSyncedTableBase(req, options);
         return new DeleteSyncedTableOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Generate OAuth credentials for a Postgres database. */
@@ -528,7 +607,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
@@ -553,7 +632,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -578,7 +657,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -595,6 +674,31 @@ export class PostgresClient {
         }
         return resp;
     }
+    /** Get Data API configuration for a database. */
+    async getDataApi(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
+        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', url, headers, callSignal);
+            const respBody = await executeHttpCall({
+                request: httpReq,
+                httpClient,
+                logger: this.logger,
+            });
+            resp = parseResponse(respBody, unmarshalDataApiSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
     /** Get a Database. */
     async getDatabase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -603,7 +707,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -628,7 +732,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -653,7 +757,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -678,7 +782,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -703,7 +807,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -728,7 +832,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -765,7 +869,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -812,7 +916,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -859,7 +963,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -909,7 +1013,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -956,7 +1060,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -987,7 +1091,7 @@ export class PostgresClient {
         }
     }
     /** Undeletes the specified database branch. */
-    async undeleteBranch(req, options) {
+    async undeleteBranchBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}/undelete`;
         const body = marshalRequest(req, marshalUndeleteBranchRequestSchema);
@@ -995,7 +1099,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
@@ -1012,12 +1116,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async undeleteBranchOperation(req, options) {
-        const op = await this.undeleteBranch(req, options);
+    /** Undeletes the specified database branch. */
+    async undeleteBranch(req, options) {
+        const op = await this.undeleteBranchBase(req, options);
         return new UndeleteBranchOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Undeletes a soft-deleted project. */
-    async undeleteProject(req, options) {
+    async undeleteProjectBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.name ?? ''}/undelete`;
         const body = marshalRequest(req, marshalUndeleteProjectRequestSchema);
@@ -1025,7 +1130,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
@@ -1042,12 +1147,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async undeleteProjectOperation(req, options) {
-        const op = await this.undeleteProject(req, options);
+    /** Undeletes a soft-deleted project. */
+    async undeleteProject(req, options) {
+        const op = await this.undeleteProjectBase(req, options);
         return new UndeleteProjectOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Updates the specified database branch. You can set this branch as the project's default branch, or protect/unprotect it. */
-    async updateBranch(req, options) {
+    async updateBranchBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.branch?.name ?? ''}`;
         const params = new URLSearchParams();
@@ -1061,7 +1167,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1078,12 +1184,50 @@ export class PostgresClient {
         }
         return resp;
     }
-    async updateBranchOperation(req, options) {
-        const op = await this.updateBranch(req, options);
+    /** Updates the specified database branch. You can set this branch as the project's default branch, or protect/unprotect it. */
+    async updateBranch(req, options) {
+        const op = await this.updateBranchBase(req, options);
         return new UpdateBranchOperation(op, (req, options) => this.getOperation(req, options));
     }
+    /** Update Data API configuration for a database. */
+    async updateDataApiBase(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.dataApi?.name ?? ''}`;
+        const params = new URLSearchParams();
+        if (req.updateMask !== undefined) {
+            params.append('update_mask', req.updateMask.toString());
+        }
+        const query = params.toString();
+        const fullUrl = query !== '' ? `${url}?${query}` : url;
+        const body = marshalRequest(req.dataApi, marshalDataApiSchema);
+        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', fullUrl, headers, callSignal, body);
+            const respBody = await executeHttpCall({
+                request: httpReq,
+                httpClient,
+                logger: this.logger,
+            });
+            resp = parseResponse(respBody, unmarshalOperationSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
+    /** Update Data API configuration for a database. */
+    async updateDataApi(req, options) {
+        const op = await this.updateDataApiBase(req, options);
+        return new UpdateDataApiOperation(op, (req, options) => this.getOperation(req, options));
+    }
     /** Update a Database. */
-    async updateDatabase(req, options) {
+    async updateDatabaseBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.database?.name ?? ''}`;
         const params = new URLSearchParams();
@@ -1097,7 +1241,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1114,12 +1258,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async updateDatabaseOperation(req, options) {
-        const op = await this.updateDatabase(req, options);
+    /** Update a Database. */
+    async updateDatabase(req, options) {
+        const op = await this.updateDatabaseBase(req, options);
         return new UpdateDatabaseOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Updates the specified compute endpoint. You can update autoscaling limits, suspend timeout, or enable/disable the compute endpoint. */
-    async updateEndpoint(req, options) {
+    async updateEndpointBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.endpoint?.name ?? ''}`;
         const params = new URLSearchParams();
@@ -1133,7 +1278,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1150,12 +1295,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async updateEndpointOperation(req, options) {
-        const op = await this.updateEndpoint(req, options);
+    /** Updates the specified compute endpoint. You can update autoscaling limits, suspend timeout, or enable/disable the compute endpoint. */
+    async updateEndpoint(req, options) {
+        const op = await this.updateEndpointBase(req, options);
         return new UpdateEndpointOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Updates the specified database project. */
-    async updateProject(req, options) {
+    async updateProjectBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.project?.name ?? ''}`;
         const params = new URLSearchParams();
@@ -1169,7 +1315,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1186,12 +1332,13 @@ export class PostgresClient {
         }
         return resp;
     }
-    async updateProjectOperation(req, options) {
-        const op = await this.updateProject(req, options);
+    /** Updates the specified database project. */
+    async updateProject(req, options) {
+        const op = await this.updateProjectBase(req, options);
         return new UpdateProjectOperation(op, (req, options) => this.getOperation(req, options));
     }
     /** Update a role for a branch. */
-    async updateRole(req, options) {
+    async updateRoleBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
         const url = `${host}/api/2.0/postgres/${req.role?.name ?? ''}`;
         const params = new URLSearchParams();
@@ -1205,7 +1352,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1222,8 +1369,9 @@ export class PostgresClient {
         }
         return resp;
     }
-    async updateRoleOperation(req, options) {
-        const op = await this.updateRole(req, options);
+    /** Update a role for a branch. */
+    async updateRole(req, options) {
+        const op = await this.updateRoleBase(req, options);
         return new UpdateRoleOperation(op, (req, options) => this.getOperation(req, options));
     }
 }
@@ -1359,6 +1507,72 @@ export class CreateCatalogOperation {
         return op.done;
     }
 }
+export class CreateDataApiOperation {
+    operation;
+    getOperation;
+    constructor(operation, getOperation) {
+        this.operation = operation;
+        this.getOperation = getOperation;
+    }
+    /** Returns the server-assigned name of the long-running operation. */
+    name() {
+        return Promise.resolve(this.operation.name);
+    }
+    /** Returns metadata associated with the long-running operation. */
+    metadata() {
+        if (this.operation.metadata === undefined) {
+            return Promise.resolve(undefined);
+        }
+        return Promise.resolve(z
+            .lazy(() => unmarshalDataApiOperationMetadataSchema)
+            .parse(this.operation.metadata));
+    }
+    /**
+     * Polls the operation until it completes.
+     *
+     * Throws if the operation failed.
+     */
+    async wait(options) {
+        let result;
+        const call = async (callSignal) => {
+            const op = await this.getOperation({
+                name: this.operation.name,
+            }, callSignal !== undefined ? { signal: callSignal } : undefined);
+            this.operation = op;
+            if (op.done === undefined) {
+                throw new Error('operation is missing the done field');
+            }
+            if (!op.done) {
+                throw new StillRunningError();
+            }
+            if (op.result?.$case === 'error') {
+                const err = op.result.error;
+                const msg = err.message !== undefined && err.message !== ''
+                    ? err.message
+                    : 'unknown error';
+                const errorMsg = err.errorCode !== undefined ? `[${err.errorCode}] ${msg}` : msg;
+                throw new Error(`operation failed: ${errorMsg}`, {
+                    cause: err,
+                });
+            }
+            if (op.result?.$case !== 'response') {
+                throw new Error('operation completed without a response');
+            }
+            result = z.lazy(() => unmarshalDataApiSchema).parse(op.result.response);
+        };
+        await executeWait(call, options);
+        if (result === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return result;
+    }
+    /** Checks whether the operation has completed */
+    async done(options) {
+        const op = await this.getOperation({ name: this.operation.name }, options);
+        this.operation = op;
+        return op.done;
+    }
+}
 export class CreateDatabaseOperation {
     operation;
     getOperation;
@@ -1805,6 +2019,63 @@ export class DeleteCatalogOperation {
         return op.done;
     }
 }
+export class DeleteDataApiOperation {
+    operation;
+    getOperation;
+    constructor(operation, getOperation) {
+        this.operation = operation;
+        this.getOperation = getOperation;
+    }
+    /** Returns the server-assigned name of the long-running operation. */
+    name() {
+        return Promise.resolve(this.operation.name);
+    }
+    /** Returns metadata associated with the long-running operation. */
+    metadata() {
+        if (this.operation.metadata === undefined) {
+            return Promise.resolve(undefined);
+        }
+        return Promise.resolve(z
+            .lazy(() => unmarshalDataApiOperationMetadataSchema)
+            .parse(this.operation.metadata));
+    }
+    /**
+     * Polls the operation until it completes.
+     *
+     * Throws if the operation failed.
+     */
+    async wait(options) {
+        const call = async (callSignal) => {
+            const op = await this.getOperation({
+                name: this.operation.name,
+            }, callSignal !== undefined ? { signal: callSignal } : undefined);
+            this.operation = op;
+            if (op.done === undefined) {
+                throw new Error('operation is missing the done field');
+            }
+            if (!op.done) {
+                throw new StillRunningError();
+            }
+            if (op.result?.$case === 'error') {
+                const err = op.result.error;
+                const msg = err.message !== undefined && err.message !== ''
+                    ? err.message
+                    : 'unknown error';
+                const errorMsg = err.errorCode !== undefined ? `[${err.errorCode}] ${msg}` : msg;
+                throw new Error(`operation failed: ${errorMsg}`, {
+                    cause: err,
+                });
+            }
+        };
+        await executeWait(call, options);
+    }
+    /** Checks whether the operation has completed */
+    async done(options) {
+        const op = await this.getOperation({ name: this.operation.name }, options);
+        this.operation = op;
+        return op.done;
+    }
+}
 export class DeleteDatabaseOperation {
     operation;
     getOperation;
@@ -2270,6 +2541,72 @@ export class UpdateBranchOperation {
         return op.done;
     }
 }
+export class UpdateDataApiOperation {
+    operation;
+    getOperation;
+    constructor(operation, getOperation) {
+        this.operation = operation;
+        this.getOperation = getOperation;
+    }
+    /** Returns the server-assigned name of the long-running operation. */
+    name() {
+        return Promise.resolve(this.operation.name);
+    }
+    /** Returns metadata associated with the long-running operation. */
+    metadata() {
+        if (this.operation.metadata === undefined) {
+            return Promise.resolve(undefined);
+        }
+        return Promise.resolve(z
+            .lazy(() => unmarshalDataApiOperationMetadataSchema)
+            .parse(this.operation.metadata));
+    }
+    /**
+     * Polls the operation until it completes.
+     *
+     * Throws if the operation failed.
+     */
+    async wait(options) {
+        let result;
+        const call = async (callSignal) => {
+            const op = await this.getOperation({
+                name: this.operation.name,
+            }, callSignal !== undefined ? { signal: callSignal } : undefined);
+            this.operation = op;
+            if (op.done === undefined) {
+                throw new Error('operation is missing the done field');
+            }
+            if (!op.done) {
+                throw new StillRunningError();
+            }
+            if (op.result?.$case === 'error') {
+                const err = op.result.error;
+                const msg = err.message !== undefined && err.message !== ''
+                    ? err.message
+                    : 'unknown error';
+                const errorMsg = err.errorCode !== undefined ? `[${err.errorCode}] ${msg}` : msg;
+                throw new Error(`operation failed: ${errorMsg}`, {
+                    cause: err,
+                });
+            }
+            if (op.result?.$case !== 'response') {
+                throw new Error('operation completed without a response');
+            }
+            result = z.lazy(() => unmarshalDataApiSchema).parse(op.result.response);
+        };
+        await executeWait(call, options);
+        if (result === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return result;
+    }
+    /** Checks whether the operation has completed */
+    async done(options) {
+        const op = await this.getOperation({ name: this.operation.name }, options);
+        this.operation = op;
+        return op.done;
+    }
+}
 export class UpdateDatabaseOperation {
     operation;
     getOperation;