@databricks/sdk-postgres 0.53.0 → 0.53.2

package/dist/v1/client.js

@@ -1,7 +1,7 @@
 import { resolveClientConfig } from "./transport.js";
 import { StillRunningError, buildHttpRequest, executeCall, executeHttpCall, executeWait, marshalRequest, parseResponse } from "./utils.js";
 import { name, version } from "../package.js";
-import { marshalBranchSchema, marshalCatalogSchema, marshalCdfConfigSchema, marshalDataApiSchema, marshalDatabaseSchema, marshalEndpointSchema, marshalGenerateDatabaseCredentialRequestSchema, marshalProjectSchema, marshalRevealRolePasswordRequestSchema, marshalRoleSchema, marshalSnapshotScheduleSchema, marshalSnapshotSchema, marshalSyncedTableSchema, marshalUndeleteBranchRequestSchema, marshalUndeleteProjectRequestSchema, unmarshalBranchOperationMetadataSchema, unmarshalBranchSchema, unmarshalCatalogOperationMetadataSchema, unmarshalCatalogSchema, unmarshalCdfConfigOperationMetadataSchema, unmarshalCdfConfigSchema, unmarshalCdfStatusSchema, unmarshalDataApiOperationMetadataSchema, unmarshalDataApiSchema, unmarshalDatabaseCredentialSchema, unmarshalDatabaseOperationMetadataSchema, unmarshalDatabaseSchema, unmarshalEndpointOperationMetadataSchema, unmarshalEndpointSchema, unmarshalListBranchesResponseSchema, unmarshalListCdfConfigsResponseSchema, unmarshalListCdfStatusesResponseSchema, unmarshalListDatabasesResponseSchema, unmarshalListEndpointsResponseSchema, unmarshalListProjectsResponseSchema, unmarshalListRolesResponseSchema, unmarshalListSnapshotsResponseSchema, unmarshalOperationSchema, unmarshalProjectOperationMetadataSchema, unmarshalProjectSchema, unmarshalRevealRolePasswordResponseSchema, unmarshalRoleOperationMetadataSchema, unmarshalRoleSchema, unmarshalSnapshotOperationMetadataSchema, unmarshalSnapshotScheduleOperationMetadataSchema, unmarshalSnapshotScheduleSchema, unmarshalSnapshotSchema, unmarshalSyncedTableOperationMetadataSchema, unmarshalSyncedTableSchema } from "./model.js";
+import { marshalAcknowledgeAssessmentWarningsRequestSchema, marshalAssessMigrationRequestSchema, marshalBranchSchema, marshalCatalogSchema, marshalCdfConfigSchema, marshalDataApiSchema, marshalDatabaseSchema, marshalEndpointSchema, marshalGenerateDatabaseCredentialRequestSchema, marshalMigrationSchema, marshalProjectSchema, marshalRevealRolePasswordRequestSchema, marshalRoleSchema, marshalSnapshotScheduleSchema, marshalSnapshotSchema, marshalStartMigrationRequestSchema, marshalSyncedTableSchema, marshalUndeleteBranchRequestSchema, marshalUndeleteMigrationRequestSchema, marshalUndeleteProjectRequestSchema, unmarshalAcknowledgeAssessmentWarningsResponseSchema, unmarshalAssessMigrationResponseSchema, unmarshalAssessmentSchema, unmarshalBranchOperationMetadataSchema, unmarshalBranchSchema, unmarshalCatalogOperationMetadataSchema, unmarshalCatalogSchema, unmarshalCdfConfigOperationMetadataSchema, unmarshalCdfConfigSchema, unmarshalCdfStatusSchema, unmarshalDataApiOperationMetadataSchema, unmarshalDataApiSchema, unmarshalDatabaseCredentialSchema, unmarshalDatabaseOperationMetadataSchema, unmarshalDatabaseSchema, unmarshalEndpointOperationMetadataSchema, unmarshalEndpointSchema, unmarshalListAssessmentsResponseSchema, unmarshalListBranchesResponseSchema, unmarshalListCdfConfigsResponseSchema, unmarshalListCdfStatusesResponseSchema, unmarshalListDatabasesResponseSchema, unmarshalListEndpointsResponseSchema, unmarshalListMigrationsResponseSchema, unmarshalListProjectsResponseSchema, unmarshalListRolesResponseSchema, unmarshalListSnapshotsResponseSchema, unmarshalMigrationOperationMetadataSchema, unmarshalMigrationSchema, unmarshalOperationSchema, unmarshalProjectOperationMetadataSchema, unmarshalProjectSchema, unmarshalRevealRolePasswordResponseSchema, unmarshalRoleOperationMetadataSchema, unmarshalRoleSchema, unmarshalSnapshotOperationMetadataSchema, unmarshalSnapshotScheduleOperationMetadataSchema, unmarshalSnapshotScheduleSchema, unmarshalSnapshotSchema, unmarshalStartMigrationResponseSchema, unmarshalSyncedTableOperationMetadataSchema, unmarshalSyncedTableSchema, unmarshalUndeleteMigrationResponseSchema } from "./model.js";
 import { VERSION } from "@databricks/sdk-auth";
 import { createDefault } from "@databricks/sdk-core/clientinfo";
 import { NoOpLogger } from "@databricks/sdk-core/logger";
@@ -32,6 +32,54 @@ var PostgresClient = class {
 		this.config ??= resolveClientConfig(this.options);
 		return this.config;
 	}
+	/** Acknowledges non-blocking warnings on a specific assessment. */
+	async acknowledgeAssessmentWarningsBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/${req.name ?? ""}/acknowledgeWarnings`;
+		const body = marshalRequest(req, marshalAcknowledgeAssessmentWarningsRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Acknowledges non-blocking warnings on a specific assessment. */
+	async acknowledgeAssessmentWarnings(req, options) {
+		return new AcknowledgeAssessmentWarningsOperation(await this.acknowledgeAssessmentWarningsBase(req, options), (req, options) => this.getOperation(req, options));
+	}
+	/** Creates an immutable assessment run for the migration's current source configuration. */
+	async assessMigrationBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/${req.name ?? ""}/assess`;
+		const body = marshalRequest(req, marshalAssessMigrationRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Creates an immutable assessment run for the migration's current source configuration. */
+	async assessMigration(req, options) {
+		return new AssessMigrationOperation(await this.assessMigrationBase(req, options), (req, options) => this.getOperation(req, options));
+	}
 	/** Creates a new database branch in the project. */
 	async createBranchBase(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -217,6 +265,34 @@ var PostgresClient = class {
 	async createEndpoint(req, options) {
 		return new CreateEndpointOperation(await this.createEndpointBase(req, options), (req, options) => this.getOperation(req, options));
 	}
+	/** Creates a migration. */
+	async createMigrationBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/migrations`;
+		const params = new URLSearchParams();
+		if (req.migrationId !== void 0) params.append("migration_id", req.migrationId);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.migration, marshalMigrationSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Creates a migration. */
+	async createMigration(req, options) {
+		return new CreateMigrationOperation(await this.createMigrationBase(req, options), (req, options) => this.getOperation(req, options));
+	}
 	/** Creates a new Lakebase Autoscaling Postgres database project, which contains branches and compute endpoints. */
 	async createProjectBase(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -484,6 +560,29 @@ var PostgresClient = class {
 	async deleteEndpoint(req, options) {
 		return new DeleteEndpointOperation(await this.deleteEndpointBase(req, options), (req, options) => this.getOperation(req, options));
 	}
+	/** Soft-deletes a migration. */
+	async deleteMigrationBase(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 !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Soft-deletes a migration. */
+	async deleteMigration(req, options) {
+		return new DeleteMigrationOperation(await this.deleteMigrationBase(req, options), (req, options) => this.getOperation(req, options));
+	}
 	/** Deletes the specified database project. */
 	async deleteProjectBase(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -604,6 +703,25 @@ var PostgresClient = class {
 		if (resp === void 0) throw new Error("operation completed without a result.");
 		return resp;
 	}
+	/** Gets a migration assessment run. */
+	async getAssessment(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 !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalAssessmentSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
 	/** Retrieves information about the specified database branch. */
 	async getBranch(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -743,6 +861,25 @@ var PostgresClient = class {
 		if (resp === void 0) throw new Error("operation completed without a result.");
 		return resp;
 	}
+	/** Gets a migration. */
+	async getMigration(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 !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalMigrationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
 	/** Retrieves the status of a long-running operation. */
 	async getOperation(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -860,6 +997,39 @@ var PostgresClient = class {
 		if (resp === void 0) throw new Error("operation completed without a result.");
 		return resp;
 	}
+	/** Lists assessment runs for a migration. */
+	async listAssessments(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/${req.parent ?? ""}/assessments`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListAssessmentsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listAssessmentsIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listAssessments(pageReq, options);
+			for (const item of resp.assessments ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
 	/** Returns a paginated list of database branches in the project. */
 	async listBranches(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -1034,6 +1204,40 @@ var PostgresClient = class {
 			pageReq.pageToken = resp.nextPageToken;
 		}
 	}
+	/** Lists migrations in the workspace. */
+	async listMigrations(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/migrations`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		if (req.showDeleted !== void 0) params.append("show_deleted", String(req.showDeleted));
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListMigrationsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listMigrationsIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listMigrations(pageReq, options);
+			for (const item of resp.migrations ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
 	/** Returns a paginated list of database projects in the workspace that the user has permission to access. */
 	async listProjects(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -1154,6 +1358,38 @@ var PostgresClient = class {
 		if (resp === void 0) throw new Error("operation completed without a result.");
 		return resp;
 	}
+	/**
+	* Starts the assessed migration named in the request, moving it from READY_TO_START into
+	* execution. Its first phase, SCHEMA_LOAD, creates the target databases, roles, and schema on the
+	* migration's target Lakebase branch without copying any table data, ahead of INITIAL_DATA_LOAD.
+	*/
+	async startMigrationBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/${req.name ?? ""}/start`;
+		const body = marshalRequest(req, marshalStartMigrationRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Starts the assessed migration named in the request, moving it from READY_TO_START into
+	* execution. Its first phase, SCHEMA_LOAD, creates the target databases, roles, and schema on the
+	* migration's target Lakebase branch without copying any table data, ahead of INITIAL_DATA_LOAD.
+	*/
+	async startMigration(req, options) {
+		return new StartMigrationOperation(await this.startMigrationBase(req, options), (req, options) => this.getOperation(req, options));
+	}
 	/** Undeletes the specified database branch. */
 	async undeleteBranchBase(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -1178,6 +1414,30 @@ var PostgresClient = class {
 	async undeleteBranch(req, options) {
 		return new UndeleteBranchOperation(await this.undeleteBranchBase(req, options), (req, options) => this.getOperation(req, options));
 	}
+	/** Restores a soft-deleted migration before it is permanently purged. */
+	async undeleteMigrationBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/${req.name ?? ""}/undelete`;
+		const body = marshalRequest(req, marshalUndeleteMigrationRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Restores a soft-deleted migration before it is permanently purged. */
+	async undeleteMigration(req, options) {
+		return new UndeleteMigrationOperation(await this.undeleteMigrationBase(req, options), (req, options) => this.getOperation(req, options));
+	}
 	/** Undeletes a soft-deleted project. */
 	async undeleteProjectBase(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -1314,6 +1574,34 @@ var PostgresClient = class {
 	async updateEndpoint(req, options) {
 		return new UpdateEndpointOperation(await this.updateEndpointBase(req, options), (req, options) => this.getOperation(req, options));
 	}
+	/** Updates user-owned migration configuration. */
+	async updateMigrationBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/postgres/${req.migration?.name ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.updateMask !== void 0) params.append("update_mask", req.updateMask.toString());
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.migration, marshalMigrationSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Updates user-owned migration configuration. */
+	async updateMigration(req, options) {
+		return new UpdateMigrationOperation(await this.updateMigrationBase(req, options), (req, options) => this.getOperation(req, options));
+	}
 	/** Updates the specified database project. */
 	async updateProjectBase(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -1421,7 +1709,7 @@ var PostgresClient = class {
 		return new UpdateSnapshotScheduleOperation(await this.updateSnapshotScheduleBase(req, options), (req, options) => this.getOperation(req, options));
 	}
 };
-var CreateBranchOperation = class {
+var AcknowledgeAssessmentWarningsOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1433,7 +1721,7 @@ var CreateBranchOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalBranchOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalMigrationOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -1454,7 +1742,7 @@ var CreateBranchOperation = class {
 				throw new Error(`operation failed: ${errorMsg}`, { cause: err });
 			}
 			if (op.result?.$case !== "response") throw new Error("operation completed without a response");
-			result = z.lazy(() => unmarshalBranchSchema).parse(op.result.response);
+			result = z.lazy(() => unmarshalAcknowledgeAssessmentWarningsResponseSchema).parse(op.result.response);
 		};
 		await executeWait(call, options);
 		if (result === void 0) throw new Error("operation completed without a result.");
@@ -1467,7 +1755,7 @@ var CreateBranchOperation = class {
 		return op.done;
 	}
 };
-var CreateCatalogOperation = class {
+var AssessMigrationOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1479,7 +1767,7 @@ var CreateCatalogOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalCatalogOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalMigrationOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -1500,7 +1788,7 @@ var CreateCatalogOperation = class {
 				throw new Error(`operation failed: ${errorMsg}`, { cause: err });
 			}
 			if (op.result?.$case !== "response") throw new Error("operation completed without a response");
-			result = z.lazy(() => unmarshalCatalogSchema).parse(op.result.response);
+			result = z.lazy(() => unmarshalAssessMigrationResponseSchema).parse(op.result.response);
 		};
 		await executeWait(call, options);
 		if (result === void 0) throw new Error("operation completed without a result.");
@@ -1513,7 +1801,7 @@ var CreateCatalogOperation = class {
 		return op.done;
 	}
 };
-var CreateCdfConfigOperation = class {
+var CreateBranchOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1525,7 +1813,7 @@ var CreateCdfConfigOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalCdfConfigOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalBranchOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -1546,7 +1834,7 @@ var CreateCdfConfigOperation = class {
 				throw new Error(`operation failed: ${errorMsg}`, { cause: err });
 			}
 			if (op.result?.$case !== "response") throw new Error("operation completed without a response");
-			result = z.lazy(() => unmarshalCdfConfigSchema).parse(op.result.response);
+			result = z.lazy(() => unmarshalBranchSchema).parse(op.result.response);
 		};
 		await executeWait(call, options);
 		if (result === void 0) throw new Error("operation completed without a result.");
@@ -1559,7 +1847,7 @@ var CreateCdfConfigOperation = class {
 		return op.done;
 	}
 };
-var CreateDataApiOperation = class {
+var CreateCatalogOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1571,7 +1859,7 @@ var CreateDataApiOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalDataApiOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalCatalogOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -1592,7 +1880,7 @@ var CreateDataApiOperation = class {
 				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);
+			result = z.lazy(() => unmarshalCatalogSchema).parse(op.result.response);
 		};
 		await executeWait(call, options);
 		if (result === void 0) throw new Error("operation completed without a result.");
@@ -1605,7 +1893,7 @@ var CreateDataApiOperation = class {
 		return op.done;
 	}
 };
-var CreateDatabaseOperation = class {
+var CreateCdfConfigOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1617,7 +1905,7 @@ var CreateDatabaseOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalDatabaseOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalCdfConfigOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -1638,7 +1926,7 @@ var CreateDatabaseOperation = class {
 				throw new Error(`operation failed: ${errorMsg}`, { cause: err });
 			}
 			if (op.result?.$case !== "response") throw new Error("operation completed without a response");
-			result = z.lazy(() => unmarshalDatabaseSchema).parse(op.result.response);
+			result = z.lazy(() => unmarshalCdfConfigSchema).parse(op.result.response);
 		};
 		await executeWait(call, options);
 		if (result === void 0) throw new Error("operation completed without a result.");
@@ -1651,7 +1939,7 @@ var CreateDatabaseOperation = class {
 		return op.done;
 	}
 };
-var CreateEndpointOperation = class {
+var CreateDataApiOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1663,7 +1951,7 @@ var CreateEndpointOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalEndpointOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalDataApiOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -1684,7 +1972,7 @@ var CreateEndpointOperation = class {
 				throw new Error(`operation failed: ${errorMsg}`, { cause: err });
 			}
 			if (op.result?.$case !== "response") throw new Error("operation completed without a response");
-			result = z.lazy(() => unmarshalEndpointSchema).parse(op.result.response);
+			result = z.lazy(() => unmarshalDataApiSchema).parse(op.result.response);
 		};
 		await executeWait(call, options);
 		if (result === void 0) throw new Error("operation completed without a result.");
@@ -1697,7 +1985,7 @@ var CreateEndpointOperation = class {
 		return op.done;
 	}
 };
-var CreateProjectOperation = class {
+var CreateDatabaseOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1709,7 +1997,7 @@ var CreateProjectOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalProjectOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalDatabaseOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -1730,7 +2018,7 @@ var CreateProjectOperation = class {
 				throw new Error(`operation failed: ${errorMsg}`, { cause: err });
 			}
 			if (op.result?.$case !== "response") throw new Error("operation completed without a response");
-			result = z.lazy(() => unmarshalProjectSchema).parse(op.result.response);
+			result = z.lazy(() => unmarshalDatabaseSchema).parse(op.result.response);
 		};
 		await executeWait(call, options);
 		if (result === void 0) throw new Error("operation completed without a result.");
@@ -1743,7 +2031,7 @@ var CreateProjectOperation = class {
 		return op.done;
 	}
 };
-var CreateRoleOperation = class {
+var CreateEndpointOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
 		this.getOperation = getOperation;
@@ -1755,7 +2043,145 @@ var CreateRoleOperation = class {
 	/** Returns metadata associated with the long-running operation. */
 	metadata() {
 		if (this.operation.metadata === void 0) return Promise.resolve(void 0);
-		return Promise.resolve(z.lazy(() => unmarshalRoleOperationMetadataSchema).parse(this.operation.metadata));
+		return Promise.resolve(z.lazy(() => unmarshalEndpointOperationMetadataSchema).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 !== void 0 ? { signal: callSignal } : void 0);
+			this.operation = op;
+			if (op.done === void 0) 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 !== void 0 && err.message !== "" ? err.message : "unknown error";
+				const errorMsg = err.errorCode !== void 0 ? `[${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(() => unmarshalEndpointSchema).parse(op.result.response);
+		};
+		await executeWait(call, options);
+		if (result === void 0) 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;
+	}
+};
+var CreateMigrationOperation = class {
+	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 === void 0) return Promise.resolve(void 0);
+		return Promise.resolve(z.lazy(() => unmarshalMigrationOperationMetadataSchema).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 !== void 0 ? { signal: callSignal } : void 0);
+			this.operation = op;
+			if (op.done === void 0) 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 !== void 0 && err.message !== "" ? err.message : "unknown error";
+				const errorMsg = err.errorCode !== void 0 ? `[${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(() => unmarshalMigrationSchema).parse(op.result.response);
+		};
+		await executeWait(call, options);
+		if (result === void 0) 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;
+	}
+};
+var CreateProjectOperation = class {
+	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 === void 0) return Promise.resolve(void 0);
+		return Promise.resolve(z.lazy(() => unmarshalProjectOperationMetadataSchema).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 !== void 0 ? { signal: callSignal } : void 0);
+			this.operation = op;
+			if (op.done === void 0) 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 !== void 0 && err.message !== "" ? err.message : "unknown error";
+				const errorMsg = err.errorCode !== void 0 ? `[${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(() => unmarshalProjectSchema).parse(op.result.response);
+		};
+		await executeWait(call, options);
+		if (result === void 0) 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;
+	}
+};
+var CreateRoleOperation = class {
+	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 === void 0) return Promise.resolve(void 0);
+		return Promise.resolve(z.lazy(() => unmarshalRoleOperationMetadataSchema).parse(this.operation.metadata));
 	}
 	/**
 	* Polls the operation until it completes.
@@ -2127,6 +2553,52 @@ var DeleteEndpointOperation = class {
 		return op.done;
 	}
 };
+var DeleteMigrationOperation = class {
+	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 === void 0) return Promise.resolve(void 0);
+		return Promise.resolve(z.lazy(() => unmarshalMigrationOperationMetadataSchema).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 !== void 0 ? { signal: callSignal } : void 0);
+			this.operation = op;
+			if (op.done === void 0) 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 !== void 0 && err.message !== "" ? err.message : "unknown error";
+				const errorMsg = err.errorCode !== void 0 ? `[${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(() => unmarshalMigrationSchema).parse(op.result.response);
+		};
+		await executeWait(call, options);
+		if (result === void 0) 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;
+	}
+};
 var DeleteProjectOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
@@ -2291,6 +2763,52 @@ var DeleteSyncedTableOperation = class {
 		return op.done;
 	}
 };
+var StartMigrationOperation = class {
+	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 === void 0) return Promise.resolve(void 0);
+		return Promise.resolve(z.lazy(() => unmarshalMigrationOperationMetadataSchema).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 !== void 0 ? { signal: callSignal } : void 0);
+			this.operation = op;
+			if (op.done === void 0) 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 !== void 0 && err.message !== "" ? err.message : "unknown error";
+				const errorMsg = err.errorCode !== void 0 ? `[${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(() => unmarshalStartMigrationResponseSchema).parse(op.result.response);
+		};
+		await executeWait(call, options);
+		if (result === void 0) 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;
+	}
+};
 var UndeleteBranchOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
@@ -2332,6 +2850,52 @@ var UndeleteBranchOperation = class {
 		return op.done;
 	}
 };
+var UndeleteMigrationOperation = class {
+	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 === void 0) return Promise.resolve(void 0);
+		return Promise.resolve(z.lazy(() => unmarshalMigrationOperationMetadataSchema).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 !== void 0 ? { signal: callSignal } : void 0);
+			this.operation = op;
+			if (op.done === void 0) 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 !== void 0 && err.message !== "" ? err.message : "unknown error";
+				const errorMsg = err.errorCode !== void 0 ? `[${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(() => unmarshalUndeleteMigrationResponseSchema).parse(op.result.response);
+		};
+		await executeWait(call, options);
+		if (result === void 0) 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;
+	}
+};
 var UndeleteProjectOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
@@ -2557,6 +3121,52 @@ var UpdateEndpointOperation = class {
 		return op.done;
 	}
 };
+var UpdateMigrationOperation = class {
+	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 === void 0) return Promise.resolve(void 0);
+		return Promise.resolve(z.lazy(() => unmarshalMigrationOperationMetadataSchema).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 !== void 0 ? { signal: callSignal } : void 0);
+			this.operation = op;
+			if (op.done === void 0) 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 !== void 0 && err.message !== "" ? err.message : "unknown error";
+				const errorMsg = err.errorCode !== void 0 ? `[${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(() => unmarshalMigrationSchema).parse(op.result.response);
+		};
+		await executeWait(call, options);
+		if (result === void 0) 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;
+	}
+};
 var UpdateProjectOperation = class {
 	constructor(operation, getOperation) {
 		this.operation = operation;
@@ -2697,5 +3307,5 @@ var UpdateSnapshotScheduleOperation = class {
 };
 
 //#endregion
-export { CreateBranchOperation, CreateCatalogOperation, CreateCdfConfigOperation, CreateDataApiOperation, CreateDatabaseOperation, CreateEndpointOperation, CreateProjectOperation, CreateRoleOperation, CreateSnapshotOperation, CreateSyncedTableOperation, DeleteBranchOperation, DeleteCatalogOperation, DeleteCdfConfigOperation, DeleteDataApiOperation, DeleteDatabaseOperation, DeleteEndpointOperation, DeleteProjectOperation, DeleteRoleOperation, DeleteSnapshotOperation, DeleteSyncedTableOperation, PostgresClient, UndeleteBranchOperation, UndeleteProjectOperation, UpdateBranchOperation, UpdateDataApiOperation, UpdateDatabaseOperation, UpdateEndpointOperation, UpdateProjectOperation, UpdateRoleOperation, UpdateSnapshotScheduleOperation };
+export { AcknowledgeAssessmentWarningsOperation, AssessMigrationOperation, CreateBranchOperation, CreateCatalogOperation, CreateCdfConfigOperation, CreateDataApiOperation, CreateDatabaseOperation, CreateEndpointOperation, CreateMigrationOperation, CreateProjectOperation, CreateRoleOperation, CreateSnapshotOperation, CreateSyncedTableOperation, DeleteBranchOperation, DeleteCatalogOperation, DeleteCdfConfigOperation, DeleteDataApiOperation, DeleteDatabaseOperation, DeleteEndpointOperation, DeleteMigrationOperation, DeleteProjectOperation, DeleteRoleOperation, DeleteSnapshotOperation, DeleteSyncedTableOperation, PostgresClient, StartMigrationOperation, UndeleteBranchOperation, UndeleteMigrationOperation, UndeleteProjectOperation, UpdateBranchOperation, UpdateDataApiOperation, UpdateDatabaseOperation, UpdateEndpointOperation, UpdateMigrationOperation, UpdateProjectOperation, UpdateRoleOperation, UpdateSnapshotScheduleOperation };
 //# sourceMappingURL=client.js.map