~h2ddlZddlZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl Z ddlmZddlmZddlmZGdd eZd Zy) N)BytesIO)AnyListOptionalTuple)LogLevel)RemotePythonExecutor) AgentErrorc eZdZdZ ddeededeeedeeedef fd ZdZ d Z dd ed e d e e effd Zdeed eefdZdZdZxZS)PyodideDenoExecutora Executes Python code securely in a sandboxed JavaScript environment using Pyodide and Deno. This executor leverages Deno's secure runtime and Pyodide to run Python code in a sandboxed environment within the browser's JavaScript engine. It provides strong isolation guarantees while still allowing Python code execution. Args: additional_imports (`list[str]`): Additional Python packages to install in the Pyodide environment. logger (`Logger`): Logger to use for output and errors. deno_path (`str`, optional): Path to the Deno executable. If not provided, will use "deno" from PATH. deno_permissions (`list[str]`, optional): List of permissions to grant to the Deno runtime. Default is minimal permissions needed for execution. pyodide_packages (`list[str]`, optional): Additional Pyodide packages to load. timeout (`int`, optional): Timeout in seconds for code execution. Default is 60 seconds. additional_imports deno_pathdeno_permissionspyodide_packagestimeoutct||| tj|dgdd||_||_| gd|_ n|Dcgc]}d| c}|_ | gd|_ n||_ |j|j||_ |jjdt j" y#tjt f$r t dwxYwcc}w) Nz --versionT)capture_outputcheckzWDeno is not installed or not found in PATH. Please install Deno from https://deno.land/)z)--allow-net=cdn.jsdelivr.net,0.0.0.0:8000z --allow-readz --allow-writez--)numpypandas matplotlibpillowzPyodideDenoExecutor is runninglevel)super__init__ subprocessrunSubprocessErrorFileNotFoundError RuntimeErrorrrrr_create_deno_runnerinstall_packagesinstalled_packagesloggerlogr INFO) selfrr&rrrrperm __class__s `/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/smolagents/pyodide_deno_executor.bak.pyrzPyodideDenoExecutor.__init__2s +V4  NNI{3DPT U #   #%D !>N$NTr$[$ND !  #$OD !$4D !   "#'"7"78J"K 8 NA**,=> i  &%OsC C,%C)c8tjd|_tjj |jd|_t|j d5}|jtddd|jy#1swYxYw)zNCreate the Deno JavaScript file that will run Pyodide and execute Python code. pyodide_deno_)prefixzpyodide_runner.jswN) tempfilemkdtemp runner_dirospathjoin runner_pathopenwriteJS_CODE_start_deno_server)r)fs r,r#z'PyodideDenoExecutor._create_deno_runnerbss"**/B77<<9LM$""C ( A GGG   !   s !BBc|jdg|jz|jgz}tj|tj tj d|_ddl}|jd|j j2|j jj}td|d|_ d |_ tj|j}|j d k7r%td |j d |j"y#tj$$r}td |d}~wwxYw)z8Start the Deno server that will run our JavaScript code.rT)stdoutstderrtextrNzFailed to start Deno server: zhttp://localhost:8765zhttp://localhost:8000z"Server responded with status code : z"Failed to connect to Deno server: )rrr7rPopenPIPEserver_processtimesleeppollr?readr" server_urlrequestsget status_coder@RequestException)r)cmdrGr?responsees r,r;z&PyodideDenoExecutor._start_deno_serverns3~~u%(=(==AQAQ@RR).. ????    1     # # % 1((//446F!>vhGH H11 I||DOO4H##s*"%GH\H\G]]_`h`m`m_n#opp+(( I!CA3GH H IsAD((E;E  Ecodereturn_final_answerreturnc |||j|jzd}tj|j||j }|j dk7r#td|j|j|j}|jdrQ|d}|jddd|jd d }d |vr |d |d z }t||j|jd d}|jd} t| trt| jddk(r`| jdd} tj| j!d} t"j$j't)| |fS| |fS#tj*$r} td| |jd} ~ wwxYw)aO Execute Python code in the Pyodide environment and return the result. Args: code (str): Python code to execute. return_final_answer (bool): Whether to extract and return the final answer. Returns: Tuple[Any, str]: A tuple containing the result and execution logs. )rSreturnFinalAnswerpackages)jsonrrBzServer error: errornameErrorrCmessagez Unknown errorstack r>resulttypeimagedatazutf-8z(Failed to communicate with Deno server: N)rr%rLpostrKrrNr r@r&rYrM isinstancedictbase64 b64decodeencodePILImager8rrO) r)rSrTpayloadrQ result_datarZ error_messageexecution_logsra image_data decoded_bytesrRs r,run_code_raise_errorsz)PyodideDenoExecutor.run_code_raise_errorss( Z%8 11D4K4KKG }}T__7DLLYH##s* >(--!A4;;OO#--/Kw'#G,#(99VW#=">b9VeAf@g h e#!r%.)9%::M  <<)__Xr:N!__X.F&$'FJJv,>',I#ZZ3 & 0 01B1B71K L yy~~gm&<=~MM>) )(( ZGsKT[[Y Y ZsFFFG -GG c|jjddj|tj|S)z Install additional Python packages in the Pyodide environment. Args: additional_imports (List[str]): List of package names to install. Returns: List[str]: List of installed packages. zAdding packages to load: z, r)r&r'r6r r()r)rs r,r$z$PyodideDenoExecutor.install_packagess8 3DII>P4Q3RS[c[h[hi!!c t|drn|jrb|jjdtj |jj  |jjdt|drJtjj|jr ddl}|j!|jyyy#tj$r|jjYwxYw) z(Clean up resources used by the executor.rFzStopping Deno server...r)rr3rN)hasattrrFr&r'r r( terminatewaitrTimeoutExpiredkillr4r5existsr3shutilrmtree)r)r~s r,cleanupzPyodideDenoExecutor.cleanups 4) *t/B/B KKOO5X]]O K    ) ) + +##(((3 4 &277>>$//+J  MM$// *,K & ,, +##((* +sC-DDc$|jy)zEnsure cleanup on deletion.N)r)r)s r,deletezPyodideDenoExecutor.deletes  ru)denoNN<)F)__name__ __module__ __qualname____doc__rstrrintrr#r;boolrrrsr$rr __classcell__)r+s@r,r r s* 0404.O I.O .O #49- .O #49- .O.O` "IB3Z#3ZD3ZUZ[^`c[cUd3Zj "49 "c "+ rur a// pyodide_runner.js - Runs Python code in Pyodide within Deno import { serve } from "https://deno.land/std/http/server.ts"; import { loadPyodide } from "npm:pyodide"; // Initialize Pyodide instance const pyodidePromise = loadPyodide(); // Initialize Pyodide instance and load numpy //const pyodidePromise = (async () => { // const pyodide = await loadPyodide(); // await pyodide.loadPackage('numpy'); // return pyodide; //})(); // Function to execute Python code and return the result async function executePythonCode(code, returnFinalAnswer = false) { //TODO:AVM: const pyodide = await pyodidePromise; //const pyodide = await pyodideReadyPromise; //let pyodide = await loadPyodide(); //await pyodide.loadPackage('numpy'); // Create a capture for stdout pyodide.runPython(` import sys import io sys.stdout = io.StringIO() `); // Execute the code and capture any errors let result = null; let error = null; let stdout = ""; try { // Execute the code if (returnFinalAnswer) { // Extract the final_answer call if present const finalAnswerMatch = code.match(/final_answer\s*\((.*)\)/); if (finalAnswerMatch) { // Execute the code up to the final_answer call const preCode = code.replace(/final_answer\s*\(.*\)/, ""); pyodide.runPython(preCode); // Execute the final_answer expression and get the result const finalAnswerExpr = finalAnswerMatch[1]; result = pyodide.runPython(`${finalAnswerExpr}`); // Handle image results if (result && result.constructor.name === "Image") { // Convert PIL Image to base64 const pngBytes = pyodide.runPython(` import io import base64 buf = io.BytesIO() _result.save(buf, format='PNG') base64.b64encode(buf.getvalue()).decode('utf-8') `); result = { type: "image", data: pngBytes }; } } } else { // Just run the code without expecting a final answer result = pyodide.runPython(code); } // Get captured stdout stdout = pyodide.runPython("sys.stdout.getvalue()"); } catch (e) { error = { name: e.constructor.name, message: e.message, stack: e.stack }; } return { result: result, stdout: stdout, error: error }; } // Start a simple HTTP server to receive code execution requests //const port = 8765; //console.log(`Starting Pyodide server on port ${port}`); serve(async (req) => { if (req.method === "POST") { try { const body = await req.json(); const { code, returnFinalAnswer = false, packages = [] } = body; // Load any requested packages if (packages && packages.length > 0) { const pyodide = await pyodidePromise; //await pyodide.loadPackagesFromImports(code); for (const pkg of packages) { try { await pyodide.loadPackage(pkg); } catch (e) { console.error(`Failed to load package ${pkg}: ${e.message}`); } } } else { console.log("Skipping package loading block. Packages:", packages); // Added log for else case } //const pyodide = await pyodidePromise; //await pyodide.loadPackage("numpy"); const result = await executePythonCode(code, returnFinalAnswer); return new Response(JSON.stringify(result), { headers: { "Content-Type": "application/json" } }); } catch (e) { return new Response(JSON.stringify({ error: e.message }), { status: 500, headers: { "Content-Type": "application/json" } }); } } return new Response("Pyodide-Deno Executor is running. Send POST requests with code to execute.", { headers: { "Content-Type": "text/plain" } }); }); )rhr4rr1iortypingrrrr PIL.ImagerkrL monitoringr remote_executorsr utilsr r r:rur,rsC" -- 2E.EP@ ru