Fix AttributeError crash from calling nonexistent warning method on StagehandLogger (#320)

Author: tkattkat

.changeset/prehistoric-fluffy-sturgeon.md

@@ -0,0 +1,5 @@
+---
+"stagehand": patch
+---
+
+Fix AttributeError crash from calling nonexistent `warning` method on StagehandLogger

stagehand/context.py

@@ -164,7 +164,7 @@ async def handle_with_lock():
                                 category="context",
                             )
                         else:
-                            self.stagehand.logger.warning(
+                            self.stagehand.logger.debug(
                                 "Could not find StagehandPage wrapper for remaining page",
                                 category="context",
                             )

stagehand/main.py

@@ -563,7 +563,7 @@ async def init(self):
             )
             self.logger.debug("Set up CDP download behavior")
         except Exception as e:
-            self.logger.warning(f"Failed to set up CDP download behavior: {str(e)}")
+            self.logger.debug(f"Failed to set up CDP download behavior: {str(e)}")
 
         self._initialized = True
 
@@ -618,7 +618,7 @@ async def close(self):
                         f"Error ending server session {self.session_id}: {str(e)}"
                     )
             elif self.session_id:
-                self.logger.warning(
+                self.logger.debug(
                     "Cannot end server session: HTTP client not available."
                 )
 

stagehand/page.py

@@ -77,7 +77,7 @@ async def ensure_injection(self):
                 # Ensure that the script is injected on future navigations
                 await self._page.add_init_script(_INJECTION_SCRIPT)
         except TargetClosedError as e:
-            self._stagehand.logger.warning(
+            self._stagehand.logger.debug(
                 f"ensure_injection failed (page may be navigating): {e}",
                 category="page",
             )