formatting, remove logs

Author: miguelg719

stagehand/context.py

@@ -28,10 +28,10 @@ async def create_stagehand_page(self, pw_page: Page) -> StagehandPage:
         stagehand_page = StagehandPage(pw_page, self.stagehand, self)
         await self.inject_custom_scripts(pw_page)
         self.page_map[pw_page] = stagehand_page
-        
+
         # Initialize frame tracking for this page
         await self._attach_frame_navigated_listener(pw_page, stagehand_page)
-        
+
         return stagehand_page
 
     async def inject_custom_scripts(self, pw_page: Page):
@@ -78,25 +78,18 @@ def get_active_page(self) -> StagehandPage:
     def register_frame_id(self, frame_id: str, page: StagehandPage):
         """Register a frame ID to StagehandPage mapping."""
         self.frame_id_map[frame_id] = page
-        self.stagehand.logger.debug(
-            f"Registered frame ID {frame_id} to page", category="context"
-        )
 
     def unregister_frame_id(self, frame_id: str):
         """Unregister a frame ID from the mapping."""
         if frame_id in self.frame_id_map:
             del self.frame_id_map[frame_id]
-            self.stagehand.logger.debug(
-                f"Unregistered frame ID {frame_id}", category="context"
-            )
 
     def get_stagehand_page_by_frame_id(self, frame_id: str) -> StagehandPage:
         """Get StagehandPage by frame ID."""
         return self.frame_id_map.get(frame_id)
 
     @classmethod
     async def init(cls, context: BrowserContext, stagehand):
-        stagehand.logger.debug("StagehandContext.init() called", category="context")
         instance = cls(context, stagehand)
         # Pre-initialize StagehandPages for any existing pages
         stagehand.logger.debug(
@@ -176,7 +169,9 @@ async def wrapped_pages():
             return wrapped_pages
         return attr
 
-    async def _attach_frame_navigated_listener(self, pw_page: Page, stagehand_page: StagehandPage):
+    async def _attach_frame_navigated_listener(
+        self, pw_page: Page, stagehand_page: StagehandPage
+    ):
         """
         Attach CDP listener for frame navigation events to track frame IDs.
         This mirrors the TypeScript implementation's frame tracking.
@@ -185,55 +180,55 @@ async def _attach_frame_navigated_listener(self, pw_page: Page, stagehand_page:
             # Create CDP session for the page
             cdp_session = await self._context.new_cdp_session(pw_page)
             await cdp_session.send("Page.enable")
-            
+
             # Get the current root frame ID
             frame_tree = await cdp_session.send("Page.getFrameTree")
             root_frame_id = frame_tree.get("frameTree", {}).get("frame", {}).get("id")
-            
+
             if root_frame_id:
                 # Initialize the page with its frame ID
                 stagehand_page.update_root_frame_id(root_frame_id)
                 self.register_frame_id(root_frame_id, stagehand_page)
-            
+
             # Set up event listener for frame navigation
             def on_frame_navigated(params):
                 """Handle Page.frameNavigated events"""
                 frame = params.get("frame", {})
                 frame_id = frame.get("id")
                 parent_id = frame.get("parentId")
-                
+
                 # Only track root frames (no parent)
                 if not parent_id and frame_id:
                     # Skip if it's the same frame ID
                     if frame_id == stagehand_page.frame_id:
                         return
-                    
+
                     # Unregister old frame ID if exists
                     old_id = stagehand_page.frame_id
                     if old_id:
                         self.unregister_frame_id(old_id)
-                    
+
                     # Register new frame ID
                     self.register_frame_id(frame_id, stagehand_page)
                     stagehand_page.update_root_frame_id(frame_id)
-                    
+
                     self.stagehand.logger.debug(
                         f"Frame navigated from {old_id} to {frame_id}",
-                        category="context"
+                        category="context",
                     )
-            
+
             # Register the event listener
             cdp_session.on("Page.frameNavigated", on_frame_navigated)
-            
+
             # Clean up frame ID when page closes
             def on_page_close():
                 if stagehand_page.frame_id:
                     self.unregister_frame_id(stagehand_page.frame_id)
-            
+
             pw_page.once("close", on_page_close)
-            
+
         except Exception as e:
             self.stagehand.logger.error(
                 f"Failed to attach frame navigation listener: {str(e)}",
-                category="context"
+                category="context",
             )

stagehand/page.py

@@ -46,14 +46,12 @@ def __init__(self, page: Page, stagehand_client, context=None):
     def frame_id(self) -> Optional[str]:
         """Get the current root frame ID."""
         return self._frame_id
-    
+
     def update_root_frame_id(self, new_id: str):
         """Update the root frame ID."""
         self._frame_id = new_id
-        self._stagehand.logger.debug(
-            f"Updated frame ID to {new_id}", category="page"
-        )
-    
+        self._stagehand.logger.debug(f"Updated frame ID to {new_id}", category="page")
+
     # TODO try catch here
     async def ensure_injection(self):
         """Ensure custom injection scripts are present on the page using domScripts.js."""
@@ -114,7 +112,7 @@ async def goto(
         payload = {"url": url}
         if options:
             payload["options"] = options
-        
+
         # Add frame ID if available
         if self._frame_id:
             payload["frameId"] = self._frame_id
@@ -192,7 +190,7 @@ async def act(
         # Add frame ID if available
         if self._frame_id:
             payload["frameId"] = self._frame_id
-        
+
         lock = self._stagehand._get_lock_for_session()
         async with lock:
             result = await self._stagehand._execute("act", payload)
@@ -265,7 +263,7 @@ async def observe(
         # Add frame ID if available
         if self._frame_id:
             payload["frameId"] = self._frame_id
-        
+
         lock = self._stagehand._get_lock_for_session()
         async with lock:
             result = await self._stagehand._execute("observe", payload)
@@ -393,7 +391,7 @@ async def extract(
         # Add frame ID if available
         if self._frame_id:
             payload["frameId"] = self._frame_id
-        
+
         lock = self._stagehand._get_lock_for_session()
         async with lock:
             result_dict = await self._stagehand._execute("extract", payload)
@@ -521,7 +519,6 @@ async def _wait_for_settled_dom(self, timeout_ms: int = None):
                 If None, uses the stagehand client's dom_settle_timeout_ms.
         """
 
-
         timeout = timeout_ms or getattr(self._stagehand, "dom_settle_timeout_ms", 30000)
         client = await self.get_cdp_client()