codeshield-runtime 0.1.0__tar.gz → 0.1.1__tar.gz

{codeshield_runtime-0.1.0 → codeshield_runtime-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: codeshield-runtime
-Version: 0.1.0
+Version: 0.1.1
 Summary: A secure, isolated, and self-healing Python code execution engine powered by uv and AST analysis.
 Project-URL: Homepage, https://github.com/AlgorithmicMind/codeshield
 Project-URL: Repository, https://github.com/AlgorithmicMind/codeshield

{codeshield_runtime-0.1.0 → codeshield_runtime-0.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "codeshield-runtime"
-version = "0.1.0"
+version = "0.1.1"
 description = "A secure, isolated, and self-healing Python code execution engine powered by uv and AST analysis."
 readme = "README.md"
 license = { text = "MIT" }

{codeshield_runtime-0.1.0 → codeshield_runtime-0.1.1}/src/codeshield/environment.py

@@ -63,6 +63,11 @@ class SandboxManager:
         """Return the root sandbox workspace directory."""
         return self._workspace
 
+    @property
+    def workspace_path(self) -> Path:
+        """Alias for the root sandbox workspace directory."""
+        return self._workspace
+
     @property
     def venv_path(self) -> Path:
         """Return the virtual environment directory."""

{codeshield_runtime-0.1.0 → codeshield_runtime-0.1.1}/src/codeshield/loop.py

@@ -224,6 +224,9 @@ class SelfHealingEngine:
     ) -> tuple[ExecutionResult, ErrorDiagnosis | None]:
         """Run code through the AST -> sandbox -> heal loop.
 
+        The sandbox is created automatically if ``run`` is called outside a
+        ``with`` context manager and is cleaned up once execution finishes.
+
         Args:
             request: Either a ``CodeExecutionRequest`` or a raw source string.
             timeout: Optional execution timeout override.
@@ -238,6 +241,22 @@ class SelfHealingEngine:
         if isinstance(request, str):
             request = CodeExecutionRequest(code=request)
 
+        auto_created = not self._sandbox.python_executable.exists()
+        if auto_created:
+            self._sandbox.create()
+
+        try:
+            return self._run_loop(request, timeout)
+        finally:
+            if auto_created:
+                self._sandbox.cleanup()
+
+    def _run_loop(
+        self,
+        request: CodeExecutionRequest,
+        timeout: float | None,
+    ) -> tuple[ExecutionResult, ErrorDiagnosis | None]:
+        """Execute the AST -> sandbox -> heal loop assuming the sandbox exists."""
         diagnosis: ErrorDiagnosis | None = None
         for attempt in range(1, self._max_iterations + 1):
             logger.info("Self-healing iteration %d/%d", attempt, self._max_iterations)

{codeshield_runtime-0.1.0 → codeshield_runtime-0.1.1}/tests/test_engine.py

@@ -232,3 +232,15 @@ def test_self_healing_engine_fixes_name_error() -> None:
         assert result.exit_code == 0
         assert result.stdout.strip() == "4.0"
         assert diagnosis is None
+
+
+def test_self_healing_engine_run_without_context_manager() -> None:
+    """The engine can be used without a ``with`` block and still auto-manages the sandbox."""
+    engine = SelfHealingEngine(use_llm=False)
+
+    result, diagnosis = engine.run("print('no context manager')")
+
+    assert result.exit_code == 0
+    assert result.stdout.strip() == "no context manager"
+    assert diagnosis is None
+    assert not engine._sandbox.workspace_path.exists()