Update tools/python_runner.py

tools/python_runner.py

@@ -4,6 +4,8 @@ import traceback
 import contextlib
 import signal
 
+# ---- your safe executor ----
+
 class TimeoutException(Exception):
     pass
 
@@ -11,10 +13,6 @@ def timeout_handler(signum, frame):
     raise TimeoutException("Execution timed out.")
 
 def run_python(code: str, timeout: int = 5) -> Dict[str, Any]:
-    """
-    Safely execute Python code and capture output/errors.
-    """
-
     safe_globals = {
         "__builtins__": {
             "print": print,
@@ -42,7 +40,6 @@ def run_python(code: str, timeout: int = 5) -> Dict[str, Any]:
             exec(code, safe_globals, {})
 
         signal.alarm(0)
-
         return {"status": "success", "output": stdout_buffer.getvalue()}
 
     except TimeoutException as e:
@@ -50,3 +47,11 @@ def run_python(code: str, timeout: int = 5) -> Dict[str, Any]:
 
     except Exception:
         return {"status": "error", "output": traceback.format_exc()}
+
+
+# ---- THIS PART REGISTERS THE TOOL ----
+def python_runner(code: str) -> Dict[str, Any]:
+    """
+    Executes safe Python code and returns stdout or errors.
+    """
+    return run_python(code)