Allow JSON imports in coding env

Author: burtenshaw

envs/coding_env/server/python_executor.py

@@ -31,6 +31,8 @@
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.NullHandler())
 
+DEFAULT_SAFE_IMPORTS = ["json"]
+
 
 class PyExecutor:
     """Wrapper around smolagents LocalPythonExecutor.
@@ -43,7 +45,7 @@ class PyExecutor:
 
     def __init__(self, additional_imports: list[str] | None = None):
         if additional_imports is None:
-            additional_imports = []
+            additional_imports = DEFAULT_SAFE_IMPORTS.copy()
 
         self._executor = LocalPythonExecutor(
             additional_authorized_imports=additional_imports

tests/envs/test_python_codeact_reset.py

@@ -119,6 +119,20 @@ def test_reset_clears_imports():
     )
 
 
+def test_json_import_is_available_by_default():
+    """Test that the coding env permits common JSON serialization code."""
+    env = PythonCodeActEnv()
+    env.reset()
+
+    action = CodeAction(
+        code="import json\nprint(json.dumps({'status': 'ok', 'n': 3}, sort_keys=True))"
+    )
+    obs = env.step(action)
+
+    assert obs.exit_code == 0
+    assert obs.stdout.strip() == '{"n": 3, "status": "ok"}'
+
+
 def test_reset_preserves_step_count_reset():
     """Test that reset() properly resets step count."""
     env = PythonCodeActEnv()