simplify sandbox example

Author: samuelcolvin

examples/sandbox.py

@@ -1,16 +1,26 @@
 from mcp_run_python import code_sandbox
 
 
-async def print_wait(level: str, message: str):
-    await asyncio.sleep(1)
-    print(f'[{level}] {message}')
+def print_wait(level: str, message: str):
+    print(f'{level}: {message}')
+
+
+code = """
+import numpy
+a = numpy.array([1, 2, 3])
+print(a)
+a
+"""
 
 
 async def main():
     async with code_sandbox(dependencies=['numpy'], print_handler=print_wait) as sandbox:
-        for i in range(10):
-            result = await sandbox.eval(f'import numpy as np\na = np.array([1, 2, {i}])\nprint(a)\na')
-            print(result)
+        result = await sandbox.eval(code)
+        print(f'{result["status"].title()}:')
+        if result['status'] == 'success':
+            print(result['return_value'])
+        else:
+            print(result['error'])
 
 
 if __name__ == '__main__':

mcp_run_python/code_sandbox.py

@@ -10,13 +10,13 @@
 
 from .main import deno_args_prepare
 
-JsonData: TypeAlias = 'str| bool | int | float | None | list[JsonData] | dict[str, JsonData]'
+JsonData: TypeAlias = 'str | bool | int | float | None | list[JsonData] | dict[str, JsonData]'
 
 
 class RunSuccess(TypedDict):
     status: Literal['success']
     output: list[str]
-    returnValueJson: JsonData
+    return_value: JsonData
 
 
 class RunError(TypedDict):