@deepnote/runtime-core 0.1.0 → 0.2.0

package/dist/index.cjs

@@ -84,7 +84,7 @@ var KernelClient = class {
 		while (this.kernel.status !== "idle") {
 			if (Date.now() - startTime > timeoutMs) throw new Error(`Kernel failed to reach idle status within ${timeoutMs}ms. Current status: ${this.kernel.status}`);
 			if (this.kernel.status === "dead") throw new Error("Kernel is dead");
-			await new Promise((resolve) => setTimeout(resolve, 100));
+			await new Promise((resolve$1) => setTimeout(resolve$1, 100));
 		}
 	}
 	/**
@@ -92,7 +92,7 @@ var KernelClient = class {
 	*/
 	async execute(code, callbacks) {
 		if (!this.kernel) throw new Error("Kernel not connected. Call connect() first.");
-		return new Promise((resolve, reject) => {
+		return new Promise((resolve$1, reject) => {
 			const outputs = [];
 			let executionCount = null;
 			const future = this.kernel?.requestExecute({ code });
@@ -122,7 +122,7 @@ var KernelClient = class {
 					executionCount
 				};
 				callbacks?.onDone?.(result);
-				resolve(result);
+				resolve$1(result);
 			}).catch(reject).finally(() => future?.dispose());
 		});
 	}
@@ -188,8 +188,10 @@ var KernelClient = class {
 
 //#endregion
 //#region src/python-env.ts
-const PYTHON_EXECUTABLES_UNIX = ["python", "python3"];
-const PYTHON_EXECUTABLES_WIN = ["python.exe", "python3.exe"];
+const IS_WINDOWS = process.platform === "win32";
+const PYTHON_EXECUTABLES = IS_WINDOWS ? ["python.exe", "python3.exe"] : ["python", "python3"];
+const VENV_BIN_DIR = IS_WINDOWS ? "Scripts" : "bin";
+const BARE_PYTHON_COMMAND = /^python[0-9.]*$/;
 /**
 * Resolves the Python executable path using smart detection.
 *
@@ -204,7 +206,7 @@ const PYTHON_EXECUTABLES_WIN = ["python.exe", "python3.exe"];
 * @throws Error if the path doesn't exist or no Python executable is found
 */
 async function resolvePythonExecutable(pythonPath) {
-	if (pythonPath === "python" || pythonPath === "python3") return pythonPath;
+	if (isBareSystemPython(pythonPath)) return pythonPath;
 	let fileStat;
 	try {
 		fileStat = await (0, node_fs_promises.stat)(pythonPath);
@@ -218,15 +220,14 @@ async function resolvePythonExecutable(pythonPath) {
 		throw new Error(`Path is a file but doesn't appear to be a Python executable: ${pythonPath}\nExpected a file named python, python3, python.exe, or similar.`);
 	}
 	if (!fileStat.isDirectory()) throw new Error(`Python path is neither a file nor a directory: ${pythonPath}`);
-	const candidates = process.platform === "win32" ? PYTHON_EXECUTABLES_WIN : PYTHON_EXECUTABLES_UNIX;
-	const directPython = await findPythonInDirectory(pythonPath, candidates);
+	const directPython = await findPythonInDirectory(pythonPath, PYTHON_EXECUTABLES);
 	if (directPython) return directPython;
-	const binDir = process.platform === "win32" ? (0, node_path.join)(pythonPath, "Scripts") : (0, node_path.join)(pythonPath, "bin");
+	const binDir = (0, node_path.join)(pythonPath, VENV_BIN_DIR);
 	if ((await (0, node_fs_promises.stat)(binDir).catch(() => null))?.isDirectory()) {
-		const venvPython = await findPythonInDirectory(binDir, candidates);
+		const venvPython = await findPythonInDirectory(binDir, PYTHON_EXECUTABLES);
 		if (venvPython) return venvPython;
 	}
-	const searchedPaths = [...candidates.map((c) => (0, node_path.join)(pythonPath, c)), ...candidates.map((c) => (0, node_path.join)(binDir, c))];
+	const searchedPaths = [...PYTHON_EXECUTABLES.map((c) => (0, node_path.join)(pythonPath, c)), ...PYTHON_EXECUTABLES.map((c) => (0, node_path.join)(binDir, c))];
 	throw new Error(`No Python executable found at: ${pythonPath}\n\nSearched for:\n${searchedPaths.map((p) => `  - ${p}`).join("\n")}\n\nYou can pass:
   - A Python executable: --python /path/to/venv/bin/python
   - A bin directory: --python /path/to/venv/bin
@@ -255,6 +256,13 @@ function detectDefaultPython() {
 	throw new Error("No Python executable found.\n\nPlease ensure Python is installed and available in your PATH,\nor specify the path explicitly with --python <path>");
 }
 /**
+* Checks if the given string is a bare system Python command (e.g. 'python', 'python3', 'python3.11')
+* as opposed to an absolute/relative path to a Python executable.
+*/
+function isBareSystemPython(pythonPath) {
+	return BARE_PYTHON_COMMAND.test(pythonPath);
+}
+/**
 * Checks if a Python command is available on the system.
 */
 function isPythonAvailable(command) {
@@ -265,6 +273,46 @@ function isPythonAvailable(command) {
 		return false;
 	}
 }
+/**
+* Detects the virtual environment root for a given Python executable path.
+*
+* Checks for `pyvenv.cfg` which is the standard marker for Python venvs.
+* Handles both `/path/to/venv/bin/python` and `/path/to/venv/Scripts/python.exe`.
+*
+* @returns The venv root directory, or null if the Python is not in a venv
+*/
+async function detectVenvRoot(pythonExecutable) {
+	const possibleVenvRoot = (0, node_path.dirname)((0, node_path.dirname)(pythonExecutable));
+	if ((await (0, node_fs_promises.stat)((0, node_path.join)(possibleVenvRoot, "pyvenv.cfg")).catch(() => null))?.isFile()) return possibleVenvRoot;
+	return null;
+}
+/**
+* Builds environment variables appropriate for the resolved Python executable.
+*
+* When a specific Python path is provided (not just 'python' or 'python3'),
+* this ensures the spawned process environment is consistent with the specified
+* Python by:
+* - Prepending the Python's directory to PATH so subprocesses find the right Python
+* - Setting VIRTUAL_ENV if the Python is in a venv
+* - Clearing VIRTUAL_ENV if the Python is NOT in a venv (to avoid inheriting
+*   the current shell's active venv)
+*
+* @param resolvedPythonPath - The resolved path from resolvePythonExecutable()
+* @param baseEnv - The base environment to modify (defaults to process.env)
+* @returns Environment variables object for use with child_process.spawn()
+*/
+async function buildPythonEnv(resolvedPythonPath, baseEnv = process.env) {
+	const env = { ...baseEnv };
+	if (isBareSystemPython(resolvedPythonPath)) return env;
+	const pythonDir = (0, node_path.dirname)((0, node_path.resolve)(resolvedPythonPath));
+	const pathKey = Object.keys(env).find((k) => k.toLowerCase() === "path") || "PATH";
+	const currentPath = env[pathKey] || "";
+	env[pathKey] = currentPath ? `${pythonDir}${node_path.delimiter}${currentPath}` : pythonDir;
+	const venvRoot = await detectVenvRoot((0, node_path.resolve)(resolvedPythonPath));
+	if (venvRoot) env.VIRTUAL_ENV = venvRoot;
+	else delete env.VIRTUAL_ENV;
+	return env;
+}
 
 //#endregion
 //#region src/server-starter.ts
@@ -280,7 +328,10 @@ async function startServer(options) {
 	const pythonPath = await resolvePythonExecutable(pythonEnv);
 	const jupyterPort = await findConsecutiveAvailablePorts(port ?? DEFAULT_PORT);
 	const lspPort = jupyterPort + 1;
-	const env = { ...process.env };
+	const env = await buildPythonEnv(pythonPath, {
+		...process.env,
+		...options.env
+	});
 	env.DEEPNOTE_RUNTIME__RUNNING_IN_DETACHED_MODE = "true";
 	env.DEEPNOTE_ENFORCE_PIP_CONSTRAINTS = "true";
 	const serverProcess = (0, node_child_process.spawn)(pythonPath, [
@@ -335,14 +386,14 @@ async function startServer(options) {
 async function stopServer(info) {
 	if (info.process.exitCode !== null) return;
 	info.process.kill("SIGTERM");
-	await new Promise((resolve) => {
+	await new Promise((resolve$1) => {
 		const timeout = setTimeout(() => {
 			if (info.process.exitCode === null) info.process.kill("SIGKILL");
-			resolve();
+			resolve$1();
 		}, 2e3);
 		info.process.once("exit", () => {
 			clearTimeout(timeout);
-			resolve();
+			resolve$1();
 		});
 	});
 }
@@ -382,7 +433,7 @@ async function waitForServer(info, timeoutMs) {
 	throw new Error(`Server failed to start within ${timeoutMs}ms at ${info.url}`);
 }
 function sleep(ms) {
-	return new Promise((resolve) => setTimeout(resolve, ms));
+	return new Promise((resolve$1) => setTimeout(resolve$1, ms));
 }
 
 //#endregion
@@ -442,7 +493,8 @@ var ExecutionEngine = class {
 		this.server = await startServer({
 			pythonEnv: this.config.pythonEnv,
 			workingDirectory: this.config.workingDirectory,
-			port: this.config.serverPort
+			port: this.config.serverPort,
+			env: this.config.env
 		});
 		try {
 			this.kernel = new KernelClient();
@@ -611,6 +663,7 @@ var ExecutionEngine = class {
 //#endregion
 exports.ExecutionEngine = ExecutionEngine;
 exports.KernelClient = KernelClient;
+exports.buildPythonEnv = buildPythonEnv;
 exports.detectDefaultPython = detectDefaultPython;
 exports.executableBlockTypeSet = executableBlockTypeSet;
 exports.executableBlockTypes = executableBlockTypes;

package/dist/index.d.cts

@@ -2,7 +2,7 @@ import { z } from "zod";
 import { IDisplayData, IError, IExecuteResult, IOutput, IOutput as IOutput$1, IStream } from "@jupyterlab/nbformat";
 import { ChildProcess } from "node:child_process";
 
-//#region ../blocks/src/deserialize-file/deepnote-file-schema.d.ts
+//#region ../blocks/src/deepnote-file/deepnote-file-schema.d.ts
 
 declare const codeBlockSchema: z.ZodObject<{
   type: z.ZodLiteral<"code">;
@@ -14689,6 +14689,8 @@ interface RuntimeConfig {
   workingDirectory: string;
   /** Optional port for the Jupyter server (auto-assigned if not provided) */
   serverPort?: number;
+  /** Optional environment variables to pass to the server */
+  env?: Record<string, string>;
 }
 interface BlockExecutionResult {
   blockId: string;
@@ -14862,6 +14864,22 @@ declare function resolvePythonExecutable(pythonPath: string): Promise<string>;
  * @throws Error if neither python nor python3 is found
  */
 declare function detectDefaultPython(): string;
+/**
+ * Builds environment variables appropriate for the resolved Python executable.
+ *
+ * When a specific Python path is provided (not just 'python' or 'python3'),
+ * this ensures the spawned process environment is consistent with the specified
+ * Python by:
+ * - Prepending the Python's directory to PATH so subprocesses find the right Python
+ * - Setting VIRTUAL_ENV if the Python is in a venv
+ * - Clearing VIRTUAL_ENV if the Python is NOT in a venv (to avoid inheriting
+ *   the current shell's active venv)
+ *
+ * @param resolvedPythonPath - The resolved path from resolvePythonExecutable()
+ * @param baseEnv - The base environment to modify (defaults to process.env)
+ * @returns Environment variables object for use with child_process.spawn()
+ */
+declare function buildPythonEnv(resolvedPythonPath: string, baseEnv?: Record<string, string | undefined>): Promise<Record<string, string | undefined>>;
 //#endregion
 //#region src/server-starter.d.ts
 interface ServerInfo {
@@ -14879,6 +14897,8 @@ interface ServerOptions {
   port?: number;
   /** Optional timeout for server startup in milliseconds */
   startupTimeoutMs?: number;
+  /** Optional environment variables to pass to the server */
+  env?: Record<string, string>;
 }
 /**
  * Start the deepnote-toolkit Jupyter server.
@@ -14890,4 +14910,4 @@ declare function startServer(options: ServerOptions): Promise<ServerInfo>;
  */
 declare function stopServer(info: ServerInfo): Promise<void>;
 //#endregion
-export { type BlockExecutionResult, type DeepnoteBlock, type DeepnoteFile, type ExecutionCallbacks, ExecutionEngine, type ExecutionOptions, type ExecutionResult, type ExecutionSummary, type IDisplayData, type IError, type IExecuteResult, type IOutput, type IStream, KernelClient, type RuntimeConfig, type ServerInfo, type ServerOptions, detectDefaultPython, executableBlockTypeSet, executableBlockTypes, resolvePythonExecutable, startServer, stopServer };
+export { type BlockExecutionResult, type DeepnoteBlock, type DeepnoteFile, type ExecutionCallbacks, ExecutionEngine, type ExecutionOptions, type ExecutionResult, type ExecutionSummary, type IDisplayData, type IError, type IExecuteResult, type IOutput, type IStream, KernelClient, type RuntimeConfig, type ServerInfo, type ServerOptions, buildPythonEnv, detectDefaultPython, executableBlockTypeSet, executableBlockTypes, resolvePythonExecutable, startServer, stopServer };

package/dist/index.d.ts

@@ -2,7 +2,7 @@ import { ChildProcess } from "node:child_process";
 import { z } from "zod";
 import { IDisplayData, IError, IExecuteResult, IOutput, IOutput as IOutput$1, IStream } from "@jupyterlab/nbformat";
 
-//#region ../blocks/src/deserialize-file/deepnote-file-schema.d.ts
+//#region ../blocks/src/deepnote-file/deepnote-file-schema.d.ts
 
 declare const codeBlockSchema: z.ZodObject<{
   type: z.ZodLiteral<"code">;
@@ -14689,6 +14689,8 @@ interface RuntimeConfig {
   workingDirectory: string;
   /** Optional port for the Jupyter server (auto-assigned if not provided) */
   serverPort?: number;
+  /** Optional environment variables to pass to the server */
+  env?: Record<string, string>;
 }
 interface BlockExecutionResult {
   blockId: string;
@@ -14862,6 +14864,22 @@ declare function resolvePythonExecutable(pythonPath: string): Promise<string>;
  * @throws Error if neither python nor python3 is found
  */
 declare function detectDefaultPython(): string;
+/**
+ * Builds environment variables appropriate for the resolved Python executable.
+ *
+ * When a specific Python path is provided (not just 'python' or 'python3'),
+ * this ensures the spawned process environment is consistent with the specified
+ * Python by:
+ * - Prepending the Python's directory to PATH so subprocesses find the right Python
+ * - Setting VIRTUAL_ENV if the Python is in a venv
+ * - Clearing VIRTUAL_ENV if the Python is NOT in a venv (to avoid inheriting
+ *   the current shell's active venv)
+ *
+ * @param resolvedPythonPath - The resolved path from resolvePythonExecutable()
+ * @param baseEnv - The base environment to modify (defaults to process.env)
+ * @returns Environment variables object for use with child_process.spawn()
+ */
+declare function buildPythonEnv(resolvedPythonPath: string, baseEnv?: Record<string, string | undefined>): Promise<Record<string, string | undefined>>;
 //#endregion
 //#region src/server-starter.d.ts
 interface ServerInfo {
@@ -14879,6 +14897,8 @@ interface ServerOptions {
   port?: number;
   /** Optional timeout for server startup in milliseconds */
   startupTimeoutMs?: number;
+  /** Optional environment variables to pass to the server */
+  env?: Record<string, string>;
 }
 /**
  * Start the deepnote-toolkit Jupyter server.
@@ -14890,4 +14910,4 @@ declare function startServer(options: ServerOptions): Promise<ServerInfo>;
  */
 declare function stopServer(info: ServerInfo): Promise<void>;
 //#endregion
-export { type BlockExecutionResult, type DeepnoteBlock, type DeepnoteFile, type ExecutionCallbacks, ExecutionEngine, type ExecutionOptions, type ExecutionResult, type ExecutionSummary, type IDisplayData, type IError, type IExecuteResult, type IOutput, type IStream, KernelClient, type RuntimeConfig, type ServerInfo, type ServerOptions, detectDefaultPython, executableBlockTypeSet, executableBlockTypes, resolvePythonExecutable, startServer, stopServer };
+export { type BlockExecutionResult, type DeepnoteBlock, type DeepnoteFile, type ExecutionCallbacks, ExecutionEngine, type ExecutionOptions, type ExecutionResult, type ExecutionSummary, type IDisplayData, type IError, type IExecuteResult, type IOutput, type IStream, KernelClient, type RuntimeConfig, type ServerInfo, type ServerOptions, buildPythonEnv, detectDefaultPython, executableBlockTypeSet, executableBlockTypes, resolvePythonExecutable, startServer, stopServer };

package/dist/index.js

@@ -3,7 +3,7 @@ import { createPythonCode, decodeUtf8NoBom, deserializeDeepnoteFile, isExecutabl
 import { KernelManager, ServerConnection, SessionManager } from "@jupyterlab/services";
 import { execSync, spawn } from "node:child_process";
 import tcpPortUsed from "tcp-port-used";
-import { basename, join } from "node:path";
+import { basename, delimiter, dirname, join, resolve } from "node:path";
 
 //#region src/kernel-client.ts
 /**
@@ -55,7 +55,7 @@ var KernelClient = class {
 		while (this.kernel.status !== "idle") {
 			if (Date.now() - startTime > timeoutMs) throw new Error(`Kernel failed to reach idle status within ${timeoutMs}ms. Current status: ${this.kernel.status}`);
 			if (this.kernel.status === "dead") throw new Error("Kernel is dead");
-			await new Promise((resolve) => setTimeout(resolve, 100));
+			await new Promise((resolve$1) => setTimeout(resolve$1, 100));
 		}
 	}
 	/**
@@ -63,7 +63,7 @@ var KernelClient = class {
 	*/
 	async execute(code, callbacks) {
 		if (!this.kernel) throw new Error("Kernel not connected. Call connect() first.");
-		return new Promise((resolve, reject) => {
+		return new Promise((resolve$1, reject) => {
 			const outputs = [];
 			let executionCount = null;
 			const future = this.kernel?.requestExecute({ code });
@@ -93,7 +93,7 @@ var KernelClient = class {
 					executionCount
 				};
 				callbacks?.onDone?.(result);
-				resolve(result);
+				resolve$1(result);
 			}).catch(reject).finally(() => future?.dispose());
 		});
 	}
@@ -159,8 +159,10 @@ var KernelClient = class {
 
 //#endregion
 //#region src/python-env.ts
-const PYTHON_EXECUTABLES_UNIX = ["python", "python3"];
-const PYTHON_EXECUTABLES_WIN = ["python.exe", "python3.exe"];
+const IS_WINDOWS = process.platform === "win32";
+const PYTHON_EXECUTABLES = IS_WINDOWS ? ["python.exe", "python3.exe"] : ["python", "python3"];
+const VENV_BIN_DIR = IS_WINDOWS ? "Scripts" : "bin";
+const BARE_PYTHON_COMMAND = /^python[0-9.]*$/;
 /**
 * Resolves the Python executable path using smart detection.
 *
@@ -175,7 +177,7 @@ const PYTHON_EXECUTABLES_WIN = ["python.exe", "python3.exe"];
 * @throws Error if the path doesn't exist or no Python executable is found
 */
 async function resolvePythonExecutable(pythonPath) {
-	if (pythonPath === "python" || pythonPath === "python3") return pythonPath;
+	if (isBareSystemPython(pythonPath)) return pythonPath;
 	let fileStat;
 	try {
 		fileStat = await stat(pythonPath);
@@ -189,15 +191,14 @@ async function resolvePythonExecutable(pythonPath) {
 		throw new Error(`Path is a file but doesn't appear to be a Python executable: ${pythonPath}\nExpected a file named python, python3, python.exe, or similar.`);
 	}
 	if (!fileStat.isDirectory()) throw new Error(`Python path is neither a file nor a directory: ${pythonPath}`);
-	const candidates = process.platform === "win32" ? PYTHON_EXECUTABLES_WIN : PYTHON_EXECUTABLES_UNIX;
-	const directPython = await findPythonInDirectory(pythonPath, candidates);
+	const directPython = await findPythonInDirectory(pythonPath, PYTHON_EXECUTABLES);
 	if (directPython) return directPython;
-	const binDir = process.platform === "win32" ? join(pythonPath, "Scripts") : join(pythonPath, "bin");
+	const binDir = join(pythonPath, VENV_BIN_DIR);
 	if ((await stat(binDir).catch(() => null))?.isDirectory()) {
-		const venvPython = await findPythonInDirectory(binDir, candidates);
+		const venvPython = await findPythonInDirectory(binDir, PYTHON_EXECUTABLES);
 		if (venvPython) return venvPython;
 	}
-	const searchedPaths = [...candidates.map((c) => join(pythonPath, c)), ...candidates.map((c) => join(binDir, c))];
+	const searchedPaths = [...PYTHON_EXECUTABLES.map((c) => join(pythonPath, c)), ...PYTHON_EXECUTABLES.map((c) => join(binDir, c))];
 	throw new Error(`No Python executable found at: ${pythonPath}\n\nSearched for:\n${searchedPaths.map((p) => `  - ${p}`).join("\n")}\n\nYou can pass:
   - A Python executable: --python /path/to/venv/bin/python
   - A bin directory: --python /path/to/venv/bin
@@ -226,6 +227,13 @@ function detectDefaultPython() {
 	throw new Error("No Python executable found.\n\nPlease ensure Python is installed and available in your PATH,\nor specify the path explicitly with --python <path>");
 }
 /**
+* Checks if the given string is a bare system Python command (e.g. 'python', 'python3', 'python3.11')
+* as opposed to an absolute/relative path to a Python executable.
+*/
+function isBareSystemPython(pythonPath) {
+	return BARE_PYTHON_COMMAND.test(pythonPath);
+}
+/**
 * Checks if a Python command is available on the system.
 */
 function isPythonAvailable(command) {
@@ -236,6 +244,46 @@ function isPythonAvailable(command) {
 		return false;
 	}
 }
+/**
+* Detects the virtual environment root for a given Python executable path.
+*
+* Checks for `pyvenv.cfg` which is the standard marker for Python venvs.
+* Handles both `/path/to/venv/bin/python` and `/path/to/venv/Scripts/python.exe`.
+*
+* @returns The venv root directory, or null if the Python is not in a venv
+*/
+async function detectVenvRoot(pythonExecutable) {
+	const possibleVenvRoot = dirname(dirname(pythonExecutable));
+	if ((await stat(join(possibleVenvRoot, "pyvenv.cfg")).catch(() => null))?.isFile()) return possibleVenvRoot;
+	return null;
+}
+/**
+* Builds environment variables appropriate for the resolved Python executable.
+*
+* When a specific Python path is provided (not just 'python' or 'python3'),
+* this ensures the spawned process environment is consistent with the specified
+* Python by:
+* - Prepending the Python's directory to PATH so subprocesses find the right Python
+* - Setting VIRTUAL_ENV if the Python is in a venv
+* - Clearing VIRTUAL_ENV if the Python is NOT in a venv (to avoid inheriting
+*   the current shell's active venv)
+*
+* @param resolvedPythonPath - The resolved path from resolvePythonExecutable()
+* @param baseEnv - The base environment to modify (defaults to process.env)
+* @returns Environment variables object for use with child_process.spawn()
+*/
+async function buildPythonEnv(resolvedPythonPath, baseEnv = process.env) {
+	const env = { ...baseEnv };
+	if (isBareSystemPython(resolvedPythonPath)) return env;
+	const pythonDir = dirname(resolve(resolvedPythonPath));
+	const pathKey = Object.keys(env).find((k) => k.toLowerCase() === "path") || "PATH";
+	const currentPath = env[pathKey] || "";
+	env[pathKey] = currentPath ? `${pythonDir}${delimiter}${currentPath}` : pythonDir;
+	const venvRoot = await detectVenvRoot(resolve(resolvedPythonPath));
+	if (venvRoot) env.VIRTUAL_ENV = venvRoot;
+	else delete env.VIRTUAL_ENV;
+	return env;
+}
 
 //#endregion
 //#region src/server-starter.ts
@@ -251,7 +299,10 @@ async function startServer(options) {
 	const pythonPath = await resolvePythonExecutable(pythonEnv);
 	const jupyterPort = await findConsecutiveAvailablePorts(port ?? DEFAULT_PORT);
 	const lspPort = jupyterPort + 1;
-	const env = { ...process.env };
+	const env = await buildPythonEnv(pythonPath, {
+		...process.env,
+		...options.env
+	});
 	env.DEEPNOTE_RUNTIME__RUNNING_IN_DETACHED_MODE = "true";
 	env.DEEPNOTE_ENFORCE_PIP_CONSTRAINTS = "true";
 	const serverProcess = spawn(pythonPath, [
@@ -306,14 +357,14 @@ async function startServer(options) {
 async function stopServer(info) {
 	if (info.process.exitCode !== null) return;
 	info.process.kill("SIGTERM");
-	await new Promise((resolve) => {
+	await new Promise((resolve$1) => {
 		const timeout = setTimeout(() => {
 			if (info.process.exitCode === null) info.process.kill("SIGKILL");
-			resolve();
+			resolve$1();
 		}, 2e3);
 		info.process.once("exit", () => {
 			clearTimeout(timeout);
-			resolve();
+			resolve$1();
 		});
 	});
 }
@@ -353,7 +404,7 @@ async function waitForServer(info, timeoutMs) {
 	throw new Error(`Server failed to start within ${timeoutMs}ms at ${info.url}`);
 }
 function sleep(ms) {
-	return new Promise((resolve) => setTimeout(resolve, ms));
+	return new Promise((resolve$1) => setTimeout(resolve$1, ms));
 }
 
 //#endregion
@@ -413,7 +464,8 @@ var ExecutionEngine = class {
 		this.server = await startServer({
 			pythonEnv: this.config.pythonEnv,
 			workingDirectory: this.config.workingDirectory,
-			port: this.config.serverPort
+			port: this.config.serverPort,
+			env: this.config.env
 		});
 		try {
 			this.kernel = new KernelClient();
@@ -580,4 +632,4 @@ var ExecutionEngine = class {
 };
 
 //#endregion
-export { ExecutionEngine, KernelClient, detectDefaultPython, executableBlockTypeSet, executableBlockTypes, resolvePythonExecutable, startServer, stopServer };
+export { ExecutionEngine, KernelClient, buildPythonEnv, detectDefaultPython, executableBlockTypeSet, executableBlockTypes, resolvePythonExecutable, startServer, stopServer };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@deepnote/runtime-core",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Core runtime for executing Deepnote projects",
   "keywords": [
     "deepnote",
@@ -35,7 +35,7 @@
     "@jupyterlab/services": "^7.3.2",
     "tcp-port-used": "^1.0.2",
     "ws": "^8.18.0",
-    "@deepnote/blocks": "4.0.0"
+    "@deepnote/blocks": "4.3.0"
   },
   "devDependencies": {
     "@types/tcp-port-used": "^1.0.4",