python 3.10 or less compatibility

Author: miguelg719

stagehand/context.py

@@ -99,8 +99,8 @@ async def _handle_new_page(self, pw_page: Page):
         Uses the page switch lock to prevent race conditions with ongoing operations.
         """
         try:
-            # Use timeout to prevent indefinite blocking
-            async with asyncio.timeout(30):
+            # Use wait_for for Python 3.10 compatibility (timeout prevents indefinite blocking)
+            async def handle_with_lock():
                 async with self.stagehand._page_switch_lock:
                     self.stagehand.logger.debug(
                         f"Creating StagehandPage for new page with URL: {pw_page.url}",
@@ -111,6 +111,8 @@ async def _handle_new_page(self, pw_page: Page):
                     self.stagehand.logger.debug(
                         "New page detected and initialized", category="context"
                     )
+
+            await asyncio.wait_for(handle_with_lock(), timeout=30)
         except asyncio.TimeoutError:
             self.stagehand.logger.error(
                 f"Timeout waiting for page switch lock when handling new page: {pw_page.url}",

stagehand/main.py

@@ -47,10 +47,12 @@ async def _ensure_page_stability(self):
         """Wait for any pending page switches to complete"""
         if hasattr(self._stagehand, "_page_switch_lock"):
             try:
-                # Use timeout to prevent indefinite blocking
-                async with asyncio.timeout(30):
+                # Use wait_for for Python 3.10 compatibility (timeout prevents indefinite blocking)
+                async def acquire_lock():
                     async with self._stagehand._page_switch_lock:
                         pass  # Just wait for any ongoing switches
+
+                await asyncio.wait_for(acquire_lock(), timeout=30)
             except asyncio.TimeoutError:
                 # Log the timeout and raise to let caller handle it
                 if hasattr(self._stagehand, "logger"):