a dg@sdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z ddl mZmZddlmZddlmZdd lmZmZmZdd lmZed d d ZeedddZGdddeZGdddeZGdddeZdS)z)A tool for running python code in a REPL.N)redirect_stdout)StringIO)AnyDictOptionalType)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)run_in_executor)BaseTool) BaseModelFieldmodel_validator) PythonREPL)returncCsttddS)N)Z_globalsZ_locals)rglobalsrrv/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_experimental/tools/python/tool.py_get_default_python_replsr)queryrcCs tdd|}tdd|}|S)zSanitize input to the python REPL. Remove whitespace, backtick & python (if llm mistakes python console as terminal) Args: query: The query to sanitize Returns: str: The sanitized query z^(\s|`)*(?i:python)?\s*z(\s|`)*$)resub)rrrrsanitize_inputs rc@sxeZdZUdZdZeed<dZeed<ee dZ e ed<dZ e ed <deeeed d d Zdeeeed ddZd S)PythonREPLTool'Tool for running python code in a REPL.Z Python_REPLnamezA Python shell. Use this to execute python commands. Input should be a valid python command. If you want to see the output of a value, you should print it out with `print(...)`. descriptiondefault_factory python_replTrNr run_managerrcCs|jrt|}|j|S) Use the tool.)rr runselfrr"rrr_run9szPythonREPLTool._runcs"|jrt|}td|j|IdHSzUse the tool asynchronously.N)rr r$r%rrr_arunCszPythonREPLTool._arun)N)N)__name__ __module__ __qualname____doc__rstr__annotations__rr rr rrboolrr rr'rr)rrrrr,s$     rc@s$eZdZUdZeddZeed<dS) PythonInputszPython inputs.zcode snippet to run)rrN)r*r+r,r-r rr.r/rrrrr1Os r1c@seZdZUdZdZeed<dZeed<ee dZ e e ed<ee dZ e e ed<d Zeed <eZeeed <ed d ee edddZdee eedddZdee eedddZdS)PythonAstREPLToolrZpython_repl_astrzA Python shell. Use this to execute python commands. Input should be a valid python command. When using this tool, sometimes output is abbreviated - make sure it does not look abbreviated before using it in your answer.rrrlocalsTr args_schemabefore)mode)valuesrcCstjdkrtdtj|S)zValidate valid python version.) zuThis tool relies on Python 3.9 or higher (as it uses new functionality in the `ast` module, you have Python version: )sys version_info ValueErrorversion)clsr7rrrvalidate_python_versionds z)PythonAstREPLTool.validate_python_versionNr!c Csz@|jrt|}t|}tj|jddgd}tt||j|jtj|jddgd}t|}t }ztt |Xt ||j|j}|dur| WdWWS|WdWWSWdn1s0YWnXt y>t | t||j|jWdn1s&0Y| YWS0Wn<t y~} z"dt| jt| WYd} ~ Sd} ~ 00dS)r#N) type_ignoresz{}: {})rastparseModulebodyexecunparserr3rrevalgetvalue Exceptionformattyper*r.) r&rr"treemoduleZ module_endZmodule_end_strZ io_bufferreterrrr'ps*   8 0zPythonAstREPLTool._runcstd|j|IdHSr()r r'r%rrrr)szPythonAstREPLTool._arun)N)N)r*r+r,r-rr.r/rr dictrrrr3rr0r1r4rr r classmethodrr?r r'rr)rrrrr2Us.      r2) r-rBrr: contextlibriortypingrrrrZ langchain_core.callbacks.managerrr Zlangchain_core.runnables.configr Zlangchain_core.toolsr Zpydanticr r rZ'langchain_experimental.utilities.pythonrrr.rrr1r2rrrrs      #