3fi<ddlmZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZmZmZddlmZmZmZddlmZmZddlmZmZmZdd lmZe r dd lm Z dd l!m"Z"d Z#dd Z$ddZ%GddeZ&GddeZ'GddeZ(y)) annotationsN)StringIO) version_info)IO TYPE_CHECKINGAnyCallableListOptionalTypeUnion)AsyncCallbackManagerForToolRunCallbackManagerCallbackManagerForToolRun)BaseToolTool) BaseModelField PrivateAttr)Unparser)EnvVars)ArtifactaEvaluates python code in a sandbox environment. The environment is long running and exists across multiple executions. You must send the whole script every time and print your outputs. Script should be pure python code that can be evaluated. It should be in python format NOT markdown. The code should NOT be wrapped in backticks. All python packages including requests, matplotlib, scipy, numpy, pandas, etc are available. Create and display chart using `plt.show()`.ctjdkr9t}t|||j }|j |St j|}|S)zUnparse the AST. )file)rminorrrgetvaluecloseastunparse)trees source_codes n/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/tools/e2b_data_analysis/tool.py_unparser%"sSA JAjjl    kk$' c tj|}|jd}t|tjrt|j tj r\t|j jtjr.|j jjdk(r t|St|tjrktjtj tjdtj|j gg|jd<t|S)aAdd print statement to the last line if it's missing. Sometimes, the LLM-generated code doesn't have `print(variable_name)`, instead the LLM tries to print the variable only by writing `variable_name` (as you would in REPL, for example). This methods checks the AST of the generated Python code and adds the print statement to the last line if it's missing. print)idctx)funcargskeywords)value) rparsebody isinstanceExprr/Callr,Namer*r%Load)coder!nodes r$add_last_line_printr9.s 99T?D 99R=D$!jSXX&F djjoosxx 0TZZ__5G5G75RD> !$!((XXchhj9jj\  "  D>r&c0eZdZUdZded<ded<ded<y) UploadedFilez6Description of the uploaded path with its remote path.strname remote_path descriptionN)__name__ __module__ __qualname____doc____annotations__r&r$r;r;Js@ Ir&r;c4eZdZUdZeddgdZded<y) E2BDataAnalysisToolArgumentsz&Arguments for the E2BDataAnalysisTool..zprint('Hello World')zhThe python script to be evaluated. The contents will be in main.py. It should not be in markdown format.)examplesr?r< python_codeN)r@rArBrCrrIrDrEr&r$rGrGRs$0 () 3 Kr&rGc,eZdZUdZdZded<eZded<ded<ded <ee Z d ed < d dfd Z ddZ e ddZ d ddZ d d dZ d!dZd"dZd"dZd#dZd$dZd%dZd&dZxZS)'E2BDataAnalysisToolzJTool for running python code in a sandboxed environment for data analysis.e2b_data_analysisr<r=zType[BaseModel] args_schemarsessionr?)default_factoryzList[UploadedFile]_uploaded_filesc  ddlm} t |ddt i|| ||||||||_y#t$r} td| d} ~ wwxYw)Nr) DataAnalysiszjoin)r_linesfs r$uploaded_files_descriptionz.E2BDataAnalysisTool.uploaded_files_descriptions t## $ )@A%% A}}" yq9:  .A!--PQR  yyr&ct|}|t|jdd}nd}|jj ||\}}}||t t d|d}tj|S)NrY)rYc|jSN)r=)artifacts r$z*E2BDataAnalysisTool._run..s 8==r&)stdoutstderr artifacts) r9getattrmetadatarN run_pythonlistmapjsondumps) r_rI run_manager callbacksrYrtrurvouts r$_runzE2BDataAnalysisTool._runs *+6  !)"4"4mTJKK$(LL$;$; [%<% ! c"@)LM  zz#r&c Ktdw)Nz(e2b_data_analysis does not support async)NotImplementedError)r_rIr~s r$_arunzE2BDataAnalysisTool._aruns ""LMMs c|jjj|}|j}|j|j |j dS)z!Run shell command in the sandbox.)rtru exit_code)rNprocessstartwaitrtrur)r_cmdprocoutputs r$ run_commandzE2BDataAnalysisTool.run_commandsJ ||##))#.mmmm))  r&c:|jj|y)z'Install python packages in the sandbox.N)rNinstall_python_packagesr_ package_namess r$rz+E2BDataAnalysisTool.install_python_packages ,,];r&c:|jj|y)z1Install system packages (via apt) in the sandbox.N)rNinstall_system_packagesrs r$rz+E2BDataAnalysisTool.install_system_packagesrr&c8|jj|S)zDownload file from the sandbox.)rN download_file)r_r>s r$rz!E2BDataAnalysisTool.download_files||))+66r&c|jj|}ttjj |j ||}|jj||jdz|jz|_ |S)z^Upload file to the sandbox. The file is uploaded to the '/home/user/' path.)r=r>r?rh) rN upload_filer;ospathbasenamer=rPrjr?rn)r_rr?r>rms r$rzE2BDataAnalysisTool.upload_fileswll..t4 !!$)),##  ##A&++d2T5T5TTr&c|jjj|j|jDcgc]}|j|jk7r|c}|_|j dz|j z|_ycc}w)z&Remove uploaded file from the sandbox.rhN)rN filesystemremover>rPr?rn)r_ uploaded_filerms r$remove_uploaded_filez(E2BDataAnalysisTool.remove_uploaded_filesz &&}'@'@A)) }} 9 99    ++d2T5T5TT  s"B ctj|j|j|j|j S)N)r,r=r?rM)r from_functionrr=r?rMrds r$as_toolzE2BDataAnalysisTool.as_tools5!!((((   r&)NNNNNNN)rS Optional[str]rTrrUzOptional[EnvVars]rVOptional[Callable[[str], Any]]rWrrYz#Optional[Callable[[Artifact], Any]]rXzOptional[Callable[[int], Any]]r`r)returnNone)rr<)NN)rIr<r~z#Optional[CallbackManagerForToolRun]rzOptional[CallbackManager]rr<rq)rIr<r~z(Optional[AsyncCallbackManagerForToolRun]rr<)rr<rdict)rzUnion[str, List[str]]rr)r>r<rbytes)rrr?r<rr;)rr;rr)rr)r@rArBrCr=rDrGrMrrzrPr]rpropertyrnrrrrrrrrr __classcell__)rbs@r$rKrK`sNT#D###?K? L*5d*KO'K"&!&*4848;?26   $  2  2  9 0  >     "<@/3 9-  6AENN>N N       <<7 U r&rK)r!zast.ASTrr<)r7r<rr<)) __future__rrr|riorsysrtypingrrrr r r r r langchain_core.callbacksrrrlangchain_core.toolsrrpydanticrrr3langchain_community.tools.e2b_data_analysis.unparserrZre2b.templates.data_analysisrr^r%r9r;rGrKrEr&r$rsz" PPP 022H4C 89 9 S (S r&