format code

Author: tkattkat

stagehand/agent/anthropic_cua.py

@@ -167,7 +167,7 @@ async def run_task(
             try:
                 if self.experimental:
                     compress_conversation_images(current_messages)
-                    
+
                 response = self.anthropic_sdk_client.beta.messages.create(
                     model=self.model,
                     max_tokens=self.max_tokens,

stagehand/agent/image_compression_utils.py

@@ -4,18 +4,18 @@
 def find_items_with_images(items: List[Dict[str, Any]]) -> List[int]:
     """
     Finds all items in the conversation history that contain images
-    
+
     Args:
         items: Array of conversation items to check
-        
+
     Returns:
         Array of indices where images were found
     """
     items_with_images = []
-    
+
     for index, item in enumerate(items):
         has_image = False
-        
+
         if isinstance(item.get("content"), list):
             has_image = any(
                 content_item.get("type") == "tool_result"
@@ -29,40 +29,39 @@ def find_items_with_images(items: List[Dict[str, Any]]) -> List[int]:
                 for content_item in item["content"]
                 if isinstance(content_item, dict)
             )
-        
+
         if has_image:
             items_with_images.append(index)
-    
+
     return items_with_images
 
 
 def compress_conversation_images(
-    items: List[Dict[str, Any]], 
-    keep_most_recent_count: int = 2
+    items: List[Dict[str, Any]], keep_most_recent_count: int = 2
 ) -> Dict[str, List[Dict[str, Any]]]:
     """
     Compresses conversation history by removing images from older items
     while keeping the most recent images intact
-    
+
     Args:
         items: Array of conversation items to process
         keep_most_recent_count: Number of most recent image-containing items to preserve (default: 2)
-        
+
     Returns:
         Dictionary with processed items
     """
     items_with_images = find_items_with_images(items)
-    
+
     for index, item in enumerate(items):
         image_index = -1
         if index in items_with_images:
             image_index = items_with_images.index(index)
-        
+
         should_compress = (
-            image_index >= 0 
+            image_index >= 0
             and image_index < len(items_with_images) - keep_most_recent_count
         )
-        
+
         if should_compress:
             if isinstance(item.get("content"), list):
                 new_content = []
@@ -79,15 +78,14 @@ def compress_conversation_images(
                             )
                         ):
                             # Replace the content with a text placeholder
-                            new_content.append({
-                                **content_item,
-                                "content": "screenshot taken"
-                            })
+                            new_content.append(
+                                {**content_item, "content": "screenshot taken"}
+                            )
                         else:
                             new_content.append(content_item)
                     else:
                         new_content.append(content_item)
-                
+
                 item["content"] = new_content
-    
-    return {"items": items}
+
+    return {"items": items}