tenki_tool.py

"""A LangChain tool that runs Python in a disposable Tenki sandbox.

`make_code_tool(sandbox)` closes over a live Tenki sandbox and returns a
LangChain tool the agent can call. `exec(command, *args)` passes args straight
through (no shell splitting), so model-generated, multi-line code goes across as
a single `-c` argument with no escaping surprises.

Uses the official `tenki-sandbox` Python SDK (pip install tenki-sandbox; Python 3.10+).
"""
from langchain_core.tools import tool
from tenki_sandbox import Sandbox


def make_code_tool(sandbox: Sandbox):
    """Return a `run_python` LangChain tool bound to a live Tenki sandbox."""

    @tool
    def run_python(code: str) -> str:
        """Execute Python code in a secure, disposable sandbox and return its stdout/stderr.

        Use it for calculations, data work, or anything that needs real execution.
        """
        result = sandbox.exec("python3", "-c", code)
        out = f"{result.stdout_text}{result.stderr_text}".strip()
        return out or f"(no output; exit code {result.exit_code})"

    return run_python