NgdZddlmZmZmZddlmZddlmZm Z ddl m Z m Z Gdde Z Gdd eZGd d e ZGd d eZdS)z Tool implementations for the Riza (https://riza.io) code interpreter API. Documentation: https://docs.riza.io API keys: https://dashboard.riza.io )AnyOptionalType)CallbackManagerForToolRun)BaseTool ToolException) BaseModelFieldc2eZdZUedZeed<dS)ExecPythonInputzthe Python code to execute descriptioncodeN__name__ __module__ __qualname__r rstr__annotations__b/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/tools/riza/command.pyr r s0">???D#?????rr ceZdZUdZdZeed<dZeed<eZ e e ed<dZ e ed<d Zeed <d ed d ffd Z ddedeed efdZxZS) ExecPythonaRiza Code tool. Setup: Install ``langchain-community`` and ``rizaio`` and set environment variable ``RIZA_API_KEY``. .. code-block:: bash pip install -U langchain-community rizaio export RIZA_API_KEY="your-api-key" Instantiation: .. code-block:: python from langchain_community.tools.riza.command import ExecPython tool = ExecPython() Invocation with args: .. code-block:: python tool.invoke("x = 5; print(x)") .. code-block:: python '5\n' Invocation with ToolCall: .. code-block:: python tool.invoke({"args": {"code":"x = 5; print(x)"}, "id": "1", "name": tool.name, "type": "tool_call"}) .. code-block:: python tool.invoke({"args": {"code":"x = 5; print(x)"}, "id": "1", "name": tool.name, "type": "tool_call"}) riza_exec_pythonnamezExecute Python code to solve problems. The Python runtime does not have filesystem access. You can use the httpx or requests library to make HTTP requests. Always print output to stdout.r args_schemaThandle_tool_errorNclientkwargsreturnc  ddlm}n"#t$r}td|d}~wwxYwtjdi|||_dSNr)RizazGCouldn't import the `rizaio` package. Try running `pip install rizaio`.rrizaior$ ImportErrorsuper__init__rselfr r$e __class__s rr)zExecPython.__init__F  # # # # # # #   4   ""6"""dff  )$)r run_managerc|jjd|}|jdkrt d|j|jS)NPYTHONlanguagerrXRiza code execution returned a non-zero exit code. The output captured from stderr was: rcommandexec exit_coderstderrstdoutr+rr0outputs r_runzExecPython._runQsb$))8$)GG  a  I9?II }rN)rrr__doc__rrrrr rrr rboolrrr)rrr> __classcell__r-s@rrrs$$L#D#"""QKQQQ$3Ki222"t"""FC        MQ   &./H&I          rrc2eZdZUedZeed<dS)ExecJavaScriptInputzthe JavaScript code to executer rNrrrrrErE]s0"BCCCD#CCCCCrrEceZdZUdZdZeed<dZeed<eZ e e ed<dZ e ed<d Zeed <d ed d ffd Z ddedeed efdZxZS)ExecJavaScriptzLA tool implementation to execute JavaScript via Riza's Code Interpreter API.riza_exec_javascriptrzExecute JavaScript code to solve problems. The JavaScript runtime does not have filesystem access, but can use fetch to make HTTP requests and does include the global JSON object. Always print output to stdout.rrTrNrr r!c  ddlm}n"#t$r}td|d}~wwxYwtjdi|||_dSr#r%r*s rr)zExecJavaScript.__init__or.r/rr0c|jjd|}|jdkrt d|j|jS)N JAVASCRIPTr3rr5r6r<s rr>zExecJavaScript._runzsb$))rBrCs@rrGrGasVV&D#&&&K $7Ki666"t"""FC        MQ   &./H&I          rrGN)r@typingrrrlangchain_core.callbacksrlangchain_core.toolsrrpydanticr r r rrErGrrrrPsJ'&&&&&&&&&98888888%%%%%%%%@@@@@i@@@EEEEEEEEPDDDDD)DDD"""""X"""""r