@databricks/sdk-environments 0.33.0 → 0.35.0

package/dist/v1/client.js

@@ -1,577 +1,442 @@
-// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
-import { VERSION as AUTH_VERSION } from '@databricks/sdk-auth';
-import { createDefault } from '@databricks/sdk-core/clientinfo';
-import { NoOpLogger } from '@databricks/sdk-core/logger';
-import { resolveClientConfig } from './transport';
-import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse, executeWait, StillRunningError, } from './utils';
-import pkgJson from '../../package.json' with { type: 'json' };
-import { z } from 'zod';
-import { marshalDefaultWorkspaceBaseEnvironmentSchema, marshalRefreshWorkspaceBaseEnvironmentRequestSchema, marshalWorkspaceBaseEnvironmentSchema, unmarshalDefaultWorkspaceBaseEnvironmentSchema, unmarshalListWorkspaceBaseEnvironmentsResponseSchema, unmarshalOperationSchema, unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema, unmarshalWorkspaceBaseEnvironmentSchema, } from './model';
-// Package identity segment for this client to be used in the User-Agent header.
+import { resolveClientConfig } from "./transport.js";
+import { StillRunningError, buildHttpRequest, executeCall, executeHttpCall, executeWait, marshalRequest, parseResponse } from "./utils.js";
+import { name, version } from "../package.js";
+import { marshalDefaultWorkspaceBaseEnvironmentSchema, marshalRefreshWorkspaceBaseEnvironmentRequestSchema, marshalWorkspaceBaseEnvironmentSchema, unmarshalDefaultWorkspaceBaseEnvironmentSchema, unmarshalListWorkspaceBaseEnvironmentsResponseSchema, unmarshalOperationSchema, unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema, unmarshalWorkspaceBaseEnvironmentSchema } from "./model.js";
+import { VERSION } from "@databricks/sdk-auth";
+import { createDefault } from "@databricks/sdk-core/clientinfo";
+import { NoOpLogger } from "@databricks/sdk-core/logger";
+import { z } from "zod";
+
+//#region src/v1/client.ts
 const PACKAGE_SEGMENT = {
-    key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
-    value: pkgJson.version,
+	key: "sdk-js-" + name.replace(/^@[^/]+\/sdk-/, ""),
+	value: version
 };
-export class EnvironmentsClient {
-    options;
-    logger;
-    // User-Agent header value. Composed once at construction from
-    // createDefault() merged with this package's identity and the active
-    // credential's name.
-    userAgent;
-    // Memoized configuration. The profile is resolved once, lazily, on the first
-    // request, then reused; host, workspaceId/accountId, and credentials are
-    // filled from it when not set explicitly on the options.
-    config;
-    constructor(options) {
-        this.options = options;
-        this.logger = options.logger ?? new NoOpLogger();
-        const info = createDefault()
-            .with(PACKAGE_SEGMENT)
-            .with({ key: 'sdk-js-auth', value: AUTH_VERSION })
-            .with({ key: 'auth', value: options.credentials?.name() ?? 'default' });
-        this.userAgent = info.toString();
-    }
-    resolveConfig() {
-        this.config ??= resolveClientConfig(this.options);
-        return this.config;
-    }
-    /**
-     * Creates a new WorkspaceBaseEnvironment.
-     * This is a long-running operation. The operation will asynchronously generate a materialized environment
-     * to optimize dependency resolution and is only marked as done when the materialized environment has been
-     * successfully generated or has failed.
-     */
-    async createWorkspaceBaseEnvironmentBase(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/workspace-base-environments`;
-        const params = new URLSearchParams();
-        if (req.workspaceBaseEnvironmentId !== undefined) {
-            params.append('workspace_base_environment_id', req.workspaceBaseEnvironmentId);
-        }
-        if (req.requestId !== undefined) {
-            params.append('request_id', req.requestId);
-        }
-        const query = params.toString();
-        const fullUrl = query !== '' ? `${url}?${query}` : url;
-        const body = marshalRequest(req.workspaceBaseEnvironment, marshalWorkspaceBaseEnvironmentSchema);
-        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', 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;
-    }
-    /**
-     * Creates a new WorkspaceBaseEnvironment.
-     * This is a long-running operation. The operation will asynchronously generate a materialized environment
-     * to optimize dependency resolution and is only marked as done when the materialized environment has been
-     * successfully generated or has failed.
-     */
-    async createWorkspaceBaseEnvironment(req, options) {
-        const op = await this.createWorkspaceBaseEnvironmentBase(req, options);
-        return new CreateWorkspaceBaseEnvironmentOperation(op, (req, options) => this.getOperation(req, options));
-    }
-    /**
-     * Deletes a WorkspaceBaseEnvironment.
-     * Deleting a base environment may impact linked notebooks and jobs.
-     * This operation is irreversible and should be performed only when you are certain the environment is no longer needed.
-     */
-    async deleteWorkspaceBaseEnvironment(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/${req.name ?? ''}`;
-        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);
-            await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-        };
-        await executeCall(call, options);
-    }
-    /**
-     * Gets the default WorkspaceBaseEnvironment configuration for the workspace.
-     * Returns the current default base environment settings for both CPU and GPU compute.
-     */
-    async getDefaultWorkspaceBaseEnvironment(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/${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, unmarshalDefaultWorkspaceBaseEnvironmentSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    /**
-     * Gets the status of a long-running operation.
-     * Clients can use this method to poll the operation result.
-     */
-    async getOperation(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/${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, unmarshalOperationSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    /** Retrieves a WorkspaceBaseEnvironment by its name. */
-    async getWorkspaceBaseEnvironment(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/${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, unmarshalWorkspaceBaseEnvironmentSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    /**
-     * Lists all WorkspaceBaseEnvironments in the workspace.
-     *
-     * <Databricks> provides the following base environments:
-     *
-     * - `workspace-base-environments/databricks_ai_...`: includes popular AI and deep learning packages for serverless GPU compute.
-     *
-     *
-     *
-     * - `workspace-base-environments/databricks_ml_...`: includes popular ML packages for serverless compute.
-     *
-     *
-     *
-     *
-     * Databricks-provided base environments are versioned. For example, `workspace-base-environments/databricks_ml_v5` corresponds to the ML environment built on environment version 5.
-     */
-    async listWorkspaceBaseEnvironments(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/workspace-base-environments`;
-        const params = new URLSearchParams();
-        if (req.pageSize !== undefined) {
-            params.append('page_size', String(req.pageSize));
-        }
-        if (req.pageToken !== undefined) {
-            params.append('page_token', req.pageToken);
-        }
-        const query = params.toString();
-        const fullUrl = query !== '' ? `${url}?${query}` : url;
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers();
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalListWorkspaceBaseEnvironmentsResponseSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    async *listWorkspaceBaseEnvironmentsIter(req, options) {
-        const pageReq = { ...req };
-        for (;;) {
-            const resp = await this.listWorkspaceBaseEnvironments(pageReq, options);
-            for (const item of resp.workspaceBaseEnvironments ?? []) {
-                yield item;
-            }
-            if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
-                return;
-            }
-            pageReq.pageToken = resp.nextPageToken;
-        }
-    }
-    /**
-     * Refreshes the materialized environment for a WorkspaceBaseEnvironment.
-     * This is a long-running operation. The operation will asynchronously regenerate the materialized environment
-     * and is only marked as done when the materialized environment has been successfully generated or has failed.
-     * The existing materialized environment remains available until it expires.
-     */
-    async refreshWorkspaceBaseEnvironmentBase(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/${req.name ?? ''}/refresh`;
-        const body = marshalRequest(req, marshalRefreshWorkspaceBaseEnvironmentRequestSchema);
-        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;
-    }
-    /**
-     * Refreshes the materialized environment for a WorkspaceBaseEnvironment.
-     * This is a long-running operation. The operation will asynchronously regenerate the materialized environment
-     * and is only marked as done when the materialized environment has been successfully generated or has failed.
-     * The existing materialized environment remains available until it expires.
-     */
-    async refreshWorkspaceBaseEnvironment(req, options) {
-        const op = await this.refreshWorkspaceBaseEnvironmentBase(req, options);
-        return new RefreshWorkspaceBaseEnvironmentOperation(op, (req, options) => this.getOperation(req, options));
-    }
-    /**
-     * Updates the default WorkspaceBaseEnvironment configuration for the workspace.
-     * Sets the specified base environments as the workspace defaults for CPU and/or GPU compute.
-     */
-    async updateDefaultWorkspaceBaseEnvironment(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/${req.defaultWorkspaceBaseEnvironment?.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.defaultWorkspaceBaseEnvironment, marshalDefaultWorkspaceBaseEnvironmentSchema);
-        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, unmarshalDefaultWorkspaceBaseEnvironmentSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    /**
-     * Updates an existing WorkspaceBaseEnvironment.
-     * This is a long-running operation. The operation will asynchronously regenerate the materialized environment
-     * and is only marked as done when the materialized environment has been successfully generated or has failed.
-     * The existing materialized environment remains available until it expires.
-     */
-    async updateWorkspaceBaseEnvironmentBase(req, options) {
-        const { host, workspaceId, httpClient } = await this.resolveConfig();
-        const url = `${host}/api/environments/v1/${req.name ?? ''}`;
-        const body = marshalRequest(req.workspaceBaseEnvironment, marshalWorkspaceBaseEnvironmentSchema);
-        let resp;
-        const call = async (callSignal) => {
-            const headers = new Headers({ 'Content-Type': 'application/json' });
-            if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Workspace-Id', workspaceId);
-            }
-            headers.set('User-Agent', this.userAgent);
-            const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
-            const respBody = await executeHttpCall({
-                request: httpReq,
-                httpClient,
-                logger: this.logger,
-            });
-            resp = parseResponse(respBody, unmarshalOperationSchema);
-        };
-        await executeCall(call, options);
-        if (resp === undefined) {
-            throw new Error('operation completed without a result.');
-        }
-        return resp;
-    }
-    /**
-     * Updates an existing WorkspaceBaseEnvironment.
-     * This is a long-running operation. The operation will asynchronously regenerate the materialized environment
-     * and is only marked as done when the materialized environment has been successfully generated or has failed.
-     * The existing materialized environment remains available until it expires.
-     */
-    async updateWorkspaceBaseEnvironment(req, options) {
-        const op = await this.updateWorkspaceBaseEnvironmentBase(req, options);
-        return new UpdateWorkspaceBaseEnvironmentOperation(op, (req, options) => this.getOperation(req, options));
-    }
-}
-export class CreateWorkspaceBaseEnvironmentOperation {
-    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(() => unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema)
-            .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(() => unmarshalWorkspaceBaseEnvironmentSchema)
-                .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 RefreshWorkspaceBaseEnvironmentOperation {
-    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(() => unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema)
-            .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(() => unmarshalWorkspaceBaseEnvironmentSchema)
-                .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 UpdateWorkspaceBaseEnvironmentOperation {
-    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(() => unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema)
-            .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(() => unmarshalWorkspaceBaseEnvironmentSchema)
-                .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;
-    }
-}
+var EnvironmentsClient = class {
+	options;
+	logger;
+	userAgent;
+	config;
+	constructor(options) {
+		this.options = options;
+		this.logger = options.logger ?? new NoOpLogger();
+		this.userAgent = createDefault().with(PACKAGE_SEGMENT).with({
+			key: "sdk-js-auth",
+			value: VERSION
+		}).with({
+			key: "auth",
+			value: options.credentials?.name() ?? "default"
+		}).toString();
+	}
+	resolveConfig() {
+		this.config ??= resolveClientConfig(this.options);
+		return this.config;
+	}
+	/**
+	* Creates a new WorkspaceBaseEnvironment.
+	* This is a long-running operation. The operation will asynchronously generate a materialized environment
+	* to optimize dependency resolution and is only marked as done when the materialized environment has been
+	* successfully generated or has failed.
+	*/
+	async createWorkspaceBaseEnvironmentBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/workspace-base-environments`;
+		const params = new URLSearchParams();
+		if (req.workspaceBaseEnvironmentId !== void 0) params.append("workspace_base_environment_id", req.workspaceBaseEnvironmentId);
+		if (req.requestId !== void 0) params.append("request_id", req.requestId);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.workspaceBaseEnvironment, marshalWorkspaceBaseEnvironmentSchema);
+		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 new WorkspaceBaseEnvironment.
+	* This is a long-running operation. The operation will asynchronously generate a materialized environment
+	* to optimize dependency resolution and is only marked as done when the materialized environment has been
+	* successfully generated or has failed.
+	*/
+	async createWorkspaceBaseEnvironment(req, options) {
+		return new CreateWorkspaceBaseEnvironmentOperation(await this.createWorkspaceBaseEnvironmentBase(req, options), (req, options) => this.getOperation(req, options));
+	}
+	/**
+	* Deletes a WorkspaceBaseEnvironment.
+	* Deleting a base environment may impact linked notebooks and jobs.
+	* This operation is irreversible and should be performed only when you are certain the environment is no longer needed.
+	*/
+	async deleteWorkspaceBaseEnvironment(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/${req.name ?? ""}`;
+		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);
+			await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			});
+		};
+		await executeCall(call, options);
+	}
+	/**
+	* Gets the default WorkspaceBaseEnvironment configuration for the workspace.
+	* Returns the current default base environment settings for both CPU and GPU compute.
+	*/
+	async getDefaultWorkspaceBaseEnvironment(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/${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
+			}), unmarshalDefaultWorkspaceBaseEnvironmentSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Gets the status of a long-running operation.
+	* Clients can use this method to poll the operation result.
+	*/
+	async getOperation(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/${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
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Retrieves a WorkspaceBaseEnvironment by its name. */
+	async getWorkspaceBaseEnvironment(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/${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
+			}), unmarshalWorkspaceBaseEnvironmentSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Lists all WorkspaceBaseEnvironments in the workspace.
+	*
+	* <Databricks> provides the following base environments:
+	*
+	* - `workspace-base-environments/databricks_ai_...`: includes popular AI and deep learning packages for serverless GPU compute.
+	*
+	*
+	*
+	* - `workspace-base-environments/databricks_ml_...`: includes popular ML packages for serverless compute.
+	*
+	*
+	*
+	*
+	* Databricks-provided base environments are versioned. For example, `workspace-base-environments/databricks_ml_v5` corresponds to the ML environment built on environment version 5.
+	*/
+	async listWorkspaceBaseEnvironments(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/workspace-base-environments`;
+		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
+			}), unmarshalListWorkspaceBaseEnvironmentsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listWorkspaceBaseEnvironmentsIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listWorkspaceBaseEnvironments(pageReq, options);
+			for (const item of resp.workspaceBaseEnvironments ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
+	/**
+	* Refreshes the materialized environment for a WorkspaceBaseEnvironment.
+	* This is a long-running operation. The operation will asynchronously regenerate the materialized environment
+	* and is only marked as done when the materialized environment has been successfully generated or has failed.
+	* The existing materialized environment remains available until it expires.
+	*/
+	async refreshWorkspaceBaseEnvironmentBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/${req.name ?? ""}/refresh`;
+		const body = marshalRequest(req, marshalRefreshWorkspaceBaseEnvironmentRequestSchema);
+		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;
+	}
+	/**
+	* Refreshes the materialized environment for a WorkspaceBaseEnvironment.
+	* This is a long-running operation. The operation will asynchronously regenerate the materialized environment
+	* and is only marked as done when the materialized environment has been successfully generated or has failed.
+	* The existing materialized environment remains available until it expires.
+	*/
+	async refreshWorkspaceBaseEnvironment(req, options) {
+		return new RefreshWorkspaceBaseEnvironmentOperation(await this.refreshWorkspaceBaseEnvironmentBase(req, options), (req, options) => this.getOperation(req, options));
+	}
+	/**
+	* Updates the default WorkspaceBaseEnvironment configuration for the workspace.
+	* Sets the specified base environments as the workspace defaults for CPU and/or GPU compute.
+	*/
+	async updateDefaultWorkspaceBaseEnvironment(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/${req.defaultWorkspaceBaseEnvironment?.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.defaultWorkspaceBaseEnvironment, marshalDefaultWorkspaceBaseEnvironmentSchema);
+		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
+			}), unmarshalDefaultWorkspaceBaseEnvironmentSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Updates an existing WorkspaceBaseEnvironment.
+	* This is a long-running operation. The operation will asynchronously regenerate the materialized environment
+	* and is only marked as done when the materialized environment has been successfully generated or has failed.
+	* The existing materialized environment remains available until it expires.
+	*/
+	async updateWorkspaceBaseEnvironmentBase(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/environments/v1/${req.name ?? ""}`;
+		const body = marshalRequest(req.workspaceBaseEnvironment, marshalWorkspaceBaseEnvironmentSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalOperationSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Updates an existing WorkspaceBaseEnvironment.
+	* This is a long-running operation. The operation will asynchronously regenerate the materialized environment
+	* and is only marked as done when the materialized environment has been successfully generated or has failed.
+	* The existing materialized environment remains available until it expires.
+	*/
+	async updateWorkspaceBaseEnvironment(req, options) {
+		return new UpdateWorkspaceBaseEnvironmentOperation(await this.updateWorkspaceBaseEnvironmentBase(req, options), (req, options) => this.getOperation(req, options));
+	}
+};
+var CreateWorkspaceBaseEnvironmentOperation = 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(() => unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema).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(() => unmarshalWorkspaceBaseEnvironmentSchema).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 RefreshWorkspaceBaseEnvironmentOperation = 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(() => unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema).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(() => unmarshalWorkspaceBaseEnvironmentSchema).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 UpdateWorkspaceBaseEnvironmentOperation = 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(() => unmarshalWorkspaceBaseEnvironmentOperationMetadataSchema).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(() => unmarshalWorkspaceBaseEnvironmentSchema).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;
+	}
+};
+
+//#endregion
+export { CreateWorkspaceBaseEnvironmentOperation, EnvironmentsClient, RefreshWorkspaceBaseEnvironmentOperation, UpdateWorkspaceBaseEnvironmentOperation };
 //# sourceMappingURL=client.js.map