PythonCodeExecutionTool

def __init__(
    self,
    *,
    timeout: int = 30,
    working_directory: Optional[WorkingDirectory] = None,
    python_environment: Optional[PythonEnvironment] = None,
) -> None:
    """
    Initialize the PythonCodeExecutionTool.

    **CAUTION**: If provided a local environment, this tool will execute code in your local environment, which can be dangerous if the code is untrusted.

    Args:
        timeout: Maximum execution time allowed in seconds, will raise a TimeoutError exception if exceeded.
        working_directory: Optional WorkingDirectory context manager to use.
        python_environment: Optional PythonEnvironment to use. If None, will auto-detect or create based on other parameters.
    """
    # Store configuration parameters
    self.timeout = timeout
    self.working_directory = WorkingDirectory.get_current_working_directory(working_directory)
    tool_python_environment = PythonEnvironment.get_current_python_environment(python_environment)

    assert self.working_directory, "No Working directory found"
    assert tool_python_environment, "No Python environment found"

    self.python_environment = tool_python_environment

    # Pydantic model to contain the code and list of libraries to execute
    class CodeExecutionRequest(BaseModel):
        code: Annotated[str, Field(description="Python code to execute")]
        libraries: Annotated[list[str], Field(description="List of libraries to install before execution")]

    # The tool function, this is what goes to the LLM
    async def execute_python_code(
        code_execution_request: Annotated[CodeExecutionRequest, "Python code and the libraries required"],
    ) -> dict[str, Any]:
        """
        Executes Python code in the attached environment and returns the result.

        Args:
            code_execution_request (CodeExecutionRequest): The Python code and libraries to execute
        """
        code = code_execution_request.code

        # NOTE: Libraries are not installed (something to consider for future versions)

        # Prepare a script file path
        script_dir = self._get_script_directory()
        script_path = os.path.join(script_dir, "script.py")

        # Execute the code
        return await self.python_environment.execute_code(code=code, script_path=script_path, timeout=self.timeout)

    super().__init__(
        name="python_execute_code",
        description="Executes Python code and returns the result.",
        func_or_tool=execute_python_code,
    )