1

Author: Danelegend

code-interpreter/app/services/executor_kubernetes.py

@@ -533,6 +533,52 @@ def execute_python(
             duration_ms=duration_ms,
             files=workspace_snapshot,
         )
+    
+    def execute_python_streaming(
+        self,
+        *,
+        code: str,
+        stdin: str | None,
+        timeout_ms: int,
+        max_output_bytes: int,
+        cpu_time_limit_sec: int | None = None,
+        memory_limit_mb: int | None = None,
+        files: Sequence[tuple[str, bytes]] | None = None,
+        last_line_interactive: bool = True,
+    ) -> Generator[StreamEvent, None, None]:
+        """Execute Python code and yield output chunks as they arrive.
+
+        Yields StreamChunk events during execution, then a single StreamResult
+        at the end containing exit_code, timing, and workspace files.
+        """
+        with self._run_in_pod(
+            code=code,
+            cpu_time_limit_sec=cpu_time_limit_sec,
+            memory_limit_mb=memory_limit_mb,
+            files=files,
+            last_line_interactive=last_line_interactive,
+        ) as ctx:
+            if stdin:
+                logger.debug("Writing stdin to Python process")
+                ctx.exec_resp.write_stdin(stdin)
+
+            deadline = time.time() + (timeout_ms / 1000.0)
+            exit_code, timed_out = yield from _stream_kube_output(
+                ctx.exec_resp, deadline, max_output_bytes
+            )
+
+            if timed_out:
+                self._kill_python_process(ctx.pod_name)
+
+            workspace_snapshot = self._extract_workspace_snapshot(ctx.pod_name)
+
+        duration_ms = int((time.perf_counter() - ctx.start) * 1000)
+        yield StreamResult(
+            exit_code=exit_code if not timed_out else None,
+            timed_out=timed_out,
+            duration_ms=duration_ms,
+            files=workspace_snapshot,
+        )
 
     def _validate_relative_path(self, path_str: str) -> Path:
         path = Path(path_str)