Propagate kernel restarts to users

e2b-dev · Sep 9, 2024 · d8831bd · d8831bd
1 parent b536b32
commit d8831bd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 16 deletions.
diff --git a/template/e2b.toml b/template/e2b.toml
diff --git a/template/server/messaging.py b/template/server/messaging.py
@@ -198,6 +198,12 @@ async def _process_message(self, data: dict):
 
         :param data: The message data
         """
+        if data["msg_type"] == "status" and data['content']['execution_state'] == 'restarting':
+            logger.error("Kernel is restarting")
+            for execution in self._executions.values():
+                await execution.queue.put(Error(name="KernelRestarting", value="Kernel was restarted", traceback=""))
+                await execution.queue.put(EndOfExecution())
+            return
 
         parent_msg_ig = data["parent_header"].get("msg_id", None)
         if parent_msg_ig is None: