Handle exception when iterating over process output

e2b-dev · Jan 24, 2025 · fd45f0e · fd45f0e
1 parent 5b3f58b
commit fd45f0e
Showing 1 changed file with 22 additions and 19 deletions.
diff --git a/packages/python-sdk/e2b/sandbox_sync/commands/command_handle.py b/packages/python-sdk/e2b/sandbox_sync/commands/command_handle.py
@@ -62,25 +62,28 @@ def _handle_events(
         None,
         None,
     ]:
-        for event in self._events:
-            if event.event.HasField("data"):
-                if event.event.data.stdout:
-                    out = event.event.data.stdout.decode('utf-8', 'replace')
-                    self._stdout += out
-                    yield out, None, None
-                if event.event.data.stderr:
-                    out = event.event.data.stderr.decode('utf-8', 'replace')
-                    self._stderr += out
-                    yield None, out, None
-                if event.event.data.pty:
-                    yield None, None, event.event.data.pty
-            if event.event.HasField("end"):
-                self._result = CommandResult(
-                    stdout=self._stdout,
-                    stderr=self._stderr,
-                    exit_code=event.event.end.exit_code,
-                    error=event.event.end.error,
-                )
+        try:
+            for event in self._events:
+                if event.event.HasField("data"):
+                    if event.event.data.stdout:
+                        out = event.event.data.stdout.decode("utf-8", "replace")
+                        self._stdout += out
+                        yield out, None, None
+                    if event.event.data.stderr:
+                        out = event.event.data.stderr.decode("utf-8", "replace")
+                        self._stderr += out
+                        yield None, out, None
+                    if event.event.data.pty:
+                        yield None, None, event.event.data.pty
+                if event.event.HasField("end"):
+                    self._result = CommandResult(
+                        stdout=self._stdout,
+                        stderr=self._stderr,
+                        exit_code=event.event.end.exit_code,
+                        error=event.event.end.error,
+                    )
+        except Exception as e:
+            raise handle_rpc_exception(e)
 
     def disconnect(self) -> None:
         """