~htZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlZddlZddlmZddlmZmZdd lmZdd lmZmZdd lmZgd Z dd l m!Z!e!GddeZ#Gdde#Z$Gdde#Z%Gdde#Z&y#e"$rY4wxYw)N)BytesIO)Path)dedent)Any)FinalAnswerTool) CodeOutputPythonExecutor)LogLevel)Toolget_tools_definition_code) AgentError) E2BExecutorDockerExecutor WasmExecutor) load_dotenvceZdZdZdeefdZdedefdZde ee ffdZ d e fd Z d edefd Z deefd ZdefdZy)RemotePythonExecutorFinalAnswerExceptionadditional_importscd||_||_|jjdg|_y)Nz!Initializing executor, hold on...)rloggerloginstalled_packages)selfrrs W/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/smolagents/remote_executors.py__init__zRemotePythonExecutor.__init__6s*"4  ;<"$codereturnct)zy Execute code, return the result and output, also determining if the result is the final answer. )NotImplementedError)rrs rrun_code_raise_errorsz*RemotePythonExecutor.run_code_raise_errors<s "!rtoolscd|vr|j|d|jDchc].}|jdD]}||jdgzvr|0}}}|r-|xj|j t |z c_t |}|r7|j|}|jj|jyycc}}w)N final_answer requirements smolagents) "_patch_final_answer_with_exceptionvaluesto_dictrinstall_packageslistr r#rrlogs)rr$toolpkgpackages_to_installr code_outputs r send_toolszRemotePythonExecutor.send_toolsCs U "  3 3E.4I J  ||~n5 $11\NBB      # #t'<': ""4(r code_actionc$|j|S)z5Run the code and determine if it is the final answer.)r#)rr=s r__call__zRemotePythonExecutor.__call__as))+66rc|rH|jddj|}|jj|j|S)Nz !pip install  )r#joinrrr.)rrr2s rr,z%RemotePythonExecutor.install_packagesesC 44}SXXN`EaDb5cdK KKOOK,, -!!rfinal_answer_toolcGdd|j}dtfd}||_|jj}||_t j |jdd|j_||_y)aPatch the FinalAnswerTool to raise an exception. This is necessary because the remote executors rely on the FinalAnswerTool to detect the final answer. It modifies the `forward` method of the FinalAnswerTool to raise a `FinalAnswerException` with the final answer as a pickled value. This allows the executor to catch this exception and return the final answer. Args: final_answer_tool (`FinalAnswerTool`): FinalAnswerTool instance to patch. c eZdZy)QRemotePythonExecutor._patch_final_answer_with_exception.._FinalAnswerToolN)__name__ __module__ __qualname__rr_FinalAnswerToolrFys rrKr c ddl}ddl}Gddt}||j|j|j |i|j )NrceZdZdZy)fRemotePythonExecutor._patch_final_answer_with_exception..forward..FinalAnswerExceptionc||_y)N)value)rrPs rrzoRemotePythonExecutor._patch_final_answer_with_exception..forward..FinalAnswerException.__init__s !&DJrN)rGrHrIrrJrrrrNs'rr)r6r8 Exceptionr7r9_forwardr:)rargskwargsr6r8rs rforwardzHRemotePythonExecutor._patch_final_answer_with_exception..forward~sU   'y '''7v'7'7  ]T]]TXEc\bEc8d'e'l'l'no orz def forward(z def _forward(N) __class__rrU__func__rRinspect getsourcereplace __source__)rrCrKrUoriginal_forward_functions rr)z7RemotePythonExecutor._patch_final_answer_with_exceptionks 0::   pc p$+ %6$=$=$F$F!$=!/6/@/@AZ/[/c/c O0 !!, '7#rN)rGrHrIFINAL_ANSWER_EXCEPTIONr-strrr r#dictr r3r<r?r,rr)rJrrrr3sz3%49% "#"*".S$Y.$ ) )7C7J7"49" +7O+7rrcDeZdZdZdeeffd ZdedefdZdZ xZ S)rz Executes Python code using E2B. Args: additional_imports (`list[str]`): Additional imports to install. logger (`Logger`): Logger to use. **kwargs: Additional arguments to pass to the E2B Sandbox. rc t||| ddlm}|di||_|j ||_|jjdtjy#t$r t dwxYw)Nr)SandboxzNPlease install 'e2b' extra to use E2BExecutor: `pip install 'smolagents[e2b]'`zE2B is runninglevelrJ) superre2b_code_interpreterrbModuleNotFoundErrorsandboxr,rrrr INFO)rrrrTrbrVs rrzE2BExecutor.__init__sz +V4  4 (( "&"7"78J"K ( > # %d  s A((A=rr c 2|jj|}dj|jjDcgc] }t |c}}|j r|j jtjk(rJtjtj|j j}t||dS|d|j jd|j jd|j j }t#||j$|j&std|dS|j&D]}|j(sdD]m}t+||d} | tj| j-d} tt.j0j3t5| |dccSdD]#}t+||d} | t| |dccStd|dScc}w) N Toutputr.is_final_answerz" Executing code yielded an error: F)jpegpngutf-8) chartdatahtml javascriptjsonlatexmarkdownpdfsvgtext)rhrun_coderBr.stdoutr^errornamerr]r8loadsr6 b64decoderPr tracebackrrresultsis_main_resultgetattrencodePILImageopenr) rr executionrexecution_logsr& error_messageresultattribute_nameimg_data decoded_bytesrss rr#z!E2BExecutor.run_code_raise_errorssLL))$/  8M8M#NCH#NO ??##';'R'RR%||F,<,4@'$*$4$4X__W5M$NM%"yy~~gm.DENlq   # _v~t<#%TX]^^ _ _8NERRg$OsHcl t|drs|jjdtj|j j |jjdtj|`yy#t$r(}|jjd|Yd}~yd}~wwxYw)z'Clean up the E2B sandbox and resources.rhzShutting down sandbox...rczSandbox cleanup completedError during cleanup: N) hasattrrrr rirhkillrQ log_errorres rcleanupzE2BExecutor.cleanups @tY'  :(--P !!#  ;8==QL (  @ KK ! !$:1#"> ? ? @sA>B B3 B..B3) rGrHrI__doc__r-r^rr r#r __classcell__rVs@rrrs5 ?49 ?5S#5S*5Sn @rrceZdZdZ ddeedededededeee fdzf fd Z d ed e fd Z d ed efdZ dZdZxZS)rzR Executes Python code using Jupyter Kernel Gateway in a Docker container. Nrhostport image_namebuild_new_imagecontainer_run_kwargsc t||| ddl}ddlm} ||_||_||_ |j|_ |sh |jjj|j|j j#d|jt$j&|r:|j j#d |jd t$j&t+t,j.d z } | j1s/t3| d5} | j5t7dddd|jjj9t;| j.t;| |j\} }|D]P}|jddj=x}s&|j j#|t$j>R|j j#d|d|d t$j&i}|r|jA|tC|jdtDsi|d<||f|dd<d |d<|jjFjH|jfi||_%d}|jJjLdk7r|dkr|j j#d|jJjLdt$j&tOjPd|jJjS|dz }|jJjLdk7r|dkrd|d||_*tWjX|jTd}|jZdk7r|jZtE|j\|j^|j`|jbjdtE|jbj\|jbjfd }|j jid!tkjl|d"#td$|jZd%|j`d|jkd&|_7d'|d|d(|jnd)}| ||_8|js||_:|j j#d*|jJjvd+|jnt$j&y#t $r t dwxYw#|jj$r} td| d} ~ wwxYw#|jj($r?|j j#d|jd t$j&d }YwxYw#1swYnxYw#tx$r$} |j{td,| | d} ~ wwxYw)-a3 Initialize the Docker-based Jupyter Kernel Gateway executor. Args: additional_imports: Additional imports to install. logger: Logger to use. host: Host to bind to. port: Port to bind to. image_name: Name of the Docker image to use. If the image doesn't exist, it will be built. build_new_image: If True, the image will be rebuilt even if it already exists. container_run_kwargs: Additional keyword arguments to pass to the Docker container run command. rN)create_connectionzWPlease install 'docker' extra to use DockerExecutor: `pip install 'smolagents[docker]'`z@Could not connect to Docker daemon: make sure Docker is running.zUsing existing Docker image: rczImage z not found, building...TzBuilding Docker image ... Dockerfilewa FROM python:3.12-slim RUN pip install jupyter_kernel_gateway jupyter_client EXPOSE 8888 CMD ["jupyter", "kernelgateway", "--KernelGatewayApp.ip='0.0.0.0'", "--KernelGatewayApp.port=8888", "--KernelGatewayApp.allow_origin='*'"] )path dockerfiletagstreamzStarting container on :portsz8888/tcpdetachrunningzContainer status: z , waiting...rzhttp://z /api/kernels) status_codeheadersurlbodyrequest_methodrequest_headers request_bodyz"Failed to create kernel. Details: )indentz Failed to create kernel: Status z Response: idzws://z /api/kernels/z /channelsz Container z is running with kernel z%Failed to initialize Jupyter kernel: )>rerdocker websocketrrgrrrfrom_envclienterrorsDockerException RuntimeErrorimagesgetrrr ri ImageNotFoundr__file__parentexistsrwriterbuildr^rstripDEBUGupdate isinstancer_ containersrun containerstatustimesleepreloadbase_urlrequestspostrrrr{requestmethodrrrvr9 kernel_idwsr,rshort_idrQr)rrrrrrrrrrrdockerfile_pathf_ build_logs log_chunk log_messagecontainer_kwargsretriesr error_detailsws_urlrVs rrzDockerExecutor.__init__s%, +V4   3   $ j //+DK U S"+KK&&**4??;KKOO&CDOOCT$U]e]j]jOk  "88I MU]UbUbc"&x."7"7,"F&--/os3 q"!$   !% 2 2 8 8_334_AU[_[j[j!9! :",KI&/mmHb&A&H&H&JJ{J  8>>JK KKOO4TF!D6EX]]O [! # ''(<=.227;TB,. )594L W %j 1)- X &7T[[3377\K[\DNG..''941 "4T^^5J5J4K< X`h`m`mn 1 %%'1  ..''941 &dV1TF3DM |<=A}}##$==#AII55FF&'ii&6&6'+AII,=,='>$%IINN!  %%(J4::VclmKnJo&pq"%Eamm_T`abagag`h#ijpttVVXd^DNTF!D6t~~6FiPF'/DG&*&;&;.>{xY[?\.]!^*.(;??;PR3S)TVZVaVabb)k:K.LPV.V'*V"'''2BTcd d  KK ! !$;A3"? @  sF F F9F44F9rc ddl}t|j}|dt|jdddii|ddidd d }|jj t j ||S) z&Send code execution request to kernel.rN anonymousexecute_requestz5.0)rusernamesessionrversionFT)rsilent store_historyuser_expressions allow_stdin)headerrmetadatar)uuidr^uuid4rsendrvr9)rrrrrs rrz$DockerExecutor._send_execute_requests{TZZ\" !'tzz|,-  !%$&$  &  TZZ01 rc t|dr|jjd|jjdt j |jj|jj|jjdt j |`yy#t$r(}|jjd|Yd}~yd}~wwxYw)z,Clean up the Docker container and resources.rz Stopping and removing container rrczContainer cleanup completedrN) rrrrrr ristopremoverQrrs rrzDockerExecutor.cleanups @t[) "B4>>CZCZB[[^ _gogtgtu##%%%'  =X]]SN *  @ KK ! !$:1#"> ? ? @sB0B44 C%=C  C%c$|jyzEnsure cleanup on deletion.Nrrs rdeletezDockerExecutor.delete  r)z 127.0.0.1i"zjupyter-kernelTN)rGrHrIrr-r^intboolr_rrr r#rrrrrs@rrrs * $6:~S I~S ~S  ~S  ~S~S#38nt3~S@###J##< @rrc eZdZdZ ddeededeedzdeffd ZdZd Z d ed e fd Z deed eefd Z dZ dZedZxZS)ruQ Remote Python code executor in a sandboxed WebAssembly environment powered by Pyodide and Deno. This executor combines Deno's secure runtime with Pyodide's WebAssembly‑compiled Python interpreter to deliver s trong isolation guarantees while enabling full Python 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. timeout (`int`, optional): Timeout in seconds for code execution. Default is 60 seconds. Nr deno_pathdeno_permissionstimeoutct||| tj|dgdd||_||_|6tjd}ddjgdzd |d d |d g}|Dcgc]}d | c}|_ |j|j||_|j j#d t$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/HOMEz allow-net=,)z 0.0.0.0:8000zcdn.jsdelivr.net:443z'pypi.org:443,files.pythonhosted.org:443z allow-read=z /.cache/denoz allow-write=z--zWasmExecutor is runningrc)rer subprocessrSubprocessErrorFileNotFoundErrorrrrosgetenvrBr_create_deno_runnerr,rrrr ri) rrrrrrhome_dirpermrVs rrzWasmExecutor.__init__s +V4  NNI{3DPT U #   #yy(H((hZ|4xj 5  :J J2dV J   "#'"7"78J"K 1G?**,=> i  0!KsC7 D%DcDtjd|_tjj |jd|_t|j d5}|j|jddd|jy#1swYxYw)zNCreate the Deno JavaScript file that will run Pyodide and execute Python code. pyodide_deno_)prefixzpyodide_runner.jsrN) tempfilemkdtemp runner_dirr rrB runner_pathrrJS_CODE_start_deno_server)rrs rr"z WasmExecutor._create_deno_runner sw"**/B77<<9LM$""C ( "A GGDLL ! " !  " "s !BBc|jdg|jz|jgz}tj|tj tj d|_tjd|j j2|j jj}td|d|_ tj|j}|j dk7r%td |j d |j"y#tj$$r}td |d}~wwxYw) z8Start the Deno server that will run our JavaScript code.rT)r}stderrr{rNzFailed to start Deno server: zhttp://localhost:8000z"Server responded with status code : z"Failed to connect to Deno server: )rrr+rPopenPIPEserver_processrrpollr/readr server_urlrrrr{RequestException)rcmdr/responsers rr-zWasmExecutor._start_deno_servers)~~u%(=(==AQAQ@RR).. ????   1     # # % 1((//446F!>vhGH H1 I||DOO4H##s*"%GH\H\G]]_`h`m`m_n#opp+(( I!CA3GH H Is AD!!E4EErr ch ||jd}tj|j||j}|j dk7r#t d|j|jd}d}|j}|jdr|jd}n|jdr|d}|jd tjk(r2d |vr.tjtj |d }d }nL|jd d d|jdd}d|vr |d|dz }t ||j|jdd} t#|t$rt|jddk(r`|jdd} tj | j'd} t(j*j-t/| | fSt1|| |S#tj2$r} t d| |jd} ~ wwxYw)a Execute Python code in the Pyodide environment and return the result. Args: code (`str`): Python code to execute. Returns: `CodeOutput`: Code output containing the result, logs, and whether it is the final answer. )rpackages)rvrr0zServer error: NFrr~pythonExceptionTypepythonExceptionValueTrErrorr1messagez Unknown errorstackrkr}rtypeimagersrqrlz(Failed to communicate with Deno server: )rrrr7rrrr{rrvrrr]r8rr6rrr_rrrrrr r8) rrpayloadr:rrn result_datar~rr image_datarrs rr#z"WasmExecutor.run_code_raise_errors3s1 Z 33G  }}T__7DLLYH##s* >(--!A4;;OOF#O#--/Kx($2)#G,II348L8c8cc.%7#\\&*:*:5AW;X*YZF&*O',yy'A&B"UYYyZiEjDk$lM%'%2eGn-=)>> $]DKK@@)__Xr:N&$'FJJv,>',I#ZZ3 & 0 01B1B71K L yy~~gm&<=~MMV.Rab b(( ZGsKT[[Y Y ZsG/H2 HH1H,,H1c|jjddj|tj|S)z Install additional Python packages in the Pyodide environment. Args: additional_imports (`list[str]`): Package names to install. Returns: list[str]: Installed packages. zAdding packages to load: z, rc)rrrBr ri)rrs rr,zWasmExecutor.install_packagesps8 3DII>P4Q3RS[c[h[hi!!rc t|drn|jrb|jjdtj |jj  |jjdt|drJtjj|jr ddl}|j!|jyyy#tj$r|jjYwxYw) z(Clean up resources used by the executor.r4zStopping Deno server...rcr)rr*rN)rr4rrr ri terminatewaitrTimeoutExpiredrr rrr*shutilrmtree)rrLs rrzWasmExecutor.cleanups 4) *t/B/B KKOO5X]]O K    ) ) + +##(((3 4 &277>>$//+J  MM$// *,K & ,, +##((* +sC-DDc$|jyr rrs rrzWasmExecutor.deleterraE // 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(); // Function to execute Python code and return the result async function executePythonCode(code) { const pyodide = await pyodidePromise; // 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 result = await pyodide.runPythonAsync(code); // Get captured stdout stdout = pyodide.runPython("sys.stdout.getvalue()"); } catch (e) { error = { name: e.constructor.name, message: e.message, stack: e.stack }; // Extract Python exception details if (e.constructor.name === "PythonError") { // Get the Python exception type from the error message: at the end of the traceback const errorMatch = e.message.match(/\n([^:]+Exception): /); if (errorMatch) { error.pythonExceptionType = errorMatch[1].split(".").pop(); } // If the error is a FinalAnswerException, extract its the encoded value if (error.pythonExceptionType === "FinalAnswerException") { // Extract the base64 encoded value from the error message const valueMatch = e.message.match(/FinalAnswerException: (.*?)(?:\n|$)/); if (valueMatch) { error.pythonExceptionValue = valueMatch[1]; } } } } return { result, stdout, 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, packages = [] } = body; // Load any requested packages if (packages && packages.length > 0) { const pyodide = await pyodidePromise; //await pyodide.loadPackagesFromImports(code); await pyodide.loadPackage("micropip"); const micropip = pyodide.pyimport("micropip"); try { await micropip.install(packages); } catch (e) { console.error(`Failed to load package ${pkg}: ${e.message}`); } } const result = await executePythonCode(code); 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" } }); }); )denoN<)rGrHrIrr-r^rrr"r-r r#r,rrrr,rrs@rrrs & -1 ,H I,H ,H s)d* ,H  ,H\ "I<;Z#;Z*;Zz"49"c" + f f Grr)'r6rXrvr r8rr(riorpathlibrtextwraprtypingr PIL.Imagerr default_toolsrlocal_python_executorr r monitoringr r$r r utilsr__all__dotenvrrgrrrrrJrrr\s"  *= 2 < "M c7>c7LV@&V@rV)Vro 'o y   s( B""B*)B*