Update agentrun/integration/builtin/sandbox.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Signed-off-by: OhYee <oyohyee@oyohyee.com>

Author: OhYee

agentrun/integration/builtin/sandbox.py

@@ -741,17 +741,24 @@ def _get_playwright(self, sb: BrowserSandbox) -> "BrowserPlaywrightSync":
         Automatically recreates the connection when the thread that created it has exited,
         because Playwright's internal greenlet is bound to the thread that created it.
         """
-        if (
-            self._playwright_sync is not None
-            and self._playwright_thread is not None
-            and not self._playwright_thread.is_alive()
-        ):
-            logger.debug(
-                "Playwright creating thread (id=%s) has exited, recreating"
-                " connection",
-                self._playwright_thread.ident,
-            )
-            self._reset_playwright()
+        if self._playwright_sync is not None and self._playwright_thread is not None:
+            current_thread = threading.current_thread()
+            creator_thread = self._playwright_thread
+            if not creator_thread.is_alive() or current_thread is not creator_thread:
+                if not creator_thread.is_alive():
+                    logger.debug(
+                        "Playwright creating thread (id=%s) has exited, recreating"
+                        " connection",
+                        creator_thread.ident,
+                    )
+                else:
+                    logger.debug(
+                        "Playwright creating thread (id=%s) differs from current"
+                        " thread (id=%s), recreating connection",
+                        creator_thread.ident,
+                        current_thread.ident,
+                    )
+                self._reset_playwright()
 
         if self._playwright_sync is None:
             with self.lock: