a bg@sddlZddlZddlZddlZddlmZddlmZmZm Z ddl Z ddl m Z ddl mZmZeedddZeeeed d d Zeed d dZGdddeZdZGdddeZGdddZdS)N)Path)DictListType)Tool) BaseModelField) md_stringreturncCs&tjdd|tjd}tdd|}|S)z"Strip markdown code from a string.z ^`{1,3}.*?\n)flagsz`{1,3}$)resubDOTALL)r Zstripped_stringrs/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/bearly/tool.pystrip_markdown_code sr)pathnr cCsbzFt|d(}ddt||DWdWS1s:0YWnty\gYS0dS)z Get the first n lines of a file.rcSsg|] }t|qSr)str).0linerrr zhead_file..N)open itertoolsislice Exception)rrfrrr head_files  : r )rr cCs@t|d"}t|WdS1s20YdS)zConvert a file to base64.rbN)rbase64 b64encodereaddecode)rrrrrfile_to_base64s r&c@s*eZdZUdZeddgddZeed<dS)BearlyInterpreterToolArgumentsz(Arguments for the BearlyInterpreterTool..zprint('Hello World')zmThe pure python script to be evaluated. The contents will be in main.py. It should not be in markdown format.)Zexamples description python_codeN)__name__ __module__ __qualname____doc__rr)r__annotations__rrrrr'#s r'aEvaluates python code in a sandbox environment. The environment resets on every execution. 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. If you have any files outputted write them to "output/" relative to the execution path. Output can only be read from the directory, stdout, and stdin. Do not use things like plot.show() as it will not work instead write them out `output/` and a link to the file will be returned. print() any output and results so you can capture the output.c@s*eZdZUdZeed<eed<eed<dS)FileInfoz(Information about a file to be uploaded. source_pathr( target_pathN)r*r+r,r-rr.rrrrr/@s r/c@seZdZUdZeed<dZeed<dZeed<eZ e e ed<iZ e eefed<ed d d Zeed d dZeed ddZeed ddZeedddZeedddZeeeddddZdd ddZed dd ZdS)!BearlyInterpreterToolz9Tool for evaluating python code in a sandbox environment.api_key%https://exec.bearly.ai/v1/interpreterendpointZbearly_interpretername args_schemafilesr3cCs ||_dSNr9)selfr3rrr__init__QszBearlyInterpreterTool.__init__)r c Csbt|jdkrdSdg}|jD]4\}}t|jd}|d|d|d|jdq"d |S) Nrr z||jvrtdt|s&tdt|||d|j|<dS)Nztarget_path already existszsource_path does not exist)r1r0r()r8 ValueErrorrexistsr/)r;r0r1r(rrradd_files  zBearlyInterpreterTool.add_filecCs i|_dSr:)r8rIrrr clear_filessz!BearlyInterpreterTool.clear_filescCstj|j|j|j|jdS)N)funcr6r(r7)r from_functionrZr6r(r7rIrrras_tools zBearlyInterpreterTool.as_tool)r*r+r,r-rr.r5r6r'r7rrr8rr/r<propertyrFr(rdictrKrZr]r`rarrdrrrrr2Hs"      r2)r"rrUr pathlibrtypingrrrrSZlangchain_core.toolsrZpydanticrrrrintr r&r'rGr/r2rrrrs