clean up

Author: rcholic

sentience/cloud_tracing.py

@@ -594,7 +594,9 @@ def upload_one(seq: int, url: str) -> bool:
 
                 if response.status_code == 200:
                     if self.logger:
-                        self.logger.info(f"Screenshot {seq} uploaded successfully ({image_size / 1024:.1f} KB)")
+                        self.logger.info(
+                            f"Screenshot {seq} uploaded successfully ({image_size / 1024:.1f} KB)"
+                        )
                     return True
                 else:
                     error_msg = f"Screenshot {seq} upload failed: HTTP {response.status_code}"

sentience/snapshot.py

@@ -410,7 +410,7 @@ async def _snapshot_via_extension_async(
             format_match = screenshot_data_url.split(";")[0].split("/")[-1]
             if format_match in ["jpeg", "jpg", "png"]:
                 result["screenshot_format"] = "jpeg" if format_match in ["jpeg", "jpg"] else "png"
-    
+
 
     # Validate and parse with Pydantic
     snapshot_obj = Snapshot(**result)
@@ -460,7 +460,7 @@ async def _snapshot_via_api_async(
         """,
         raw_options,
     )
-    
+
     # Extract screenshot from raw result (extension captures it, but API doesn't return it)
     screenshot_data_url = raw_result.get("screenshot")
     screenshot_format = None
@@ -521,7 +521,7 @@ async def _snapshot_via_api_async(
                 format_match = screenshot_data_url.split(";")[0].split("/")[-1]
                 if format_match in ["jpeg", "jpg", "png"]:
                     screenshot_format = "jpeg" if format_match in ["jpeg", "jpg"] else "png"
-        
+
         # Merge API result with local data
         snapshot_data = {
             "status": api_result.get("status", "success"),

sentience/tracer_factory.py

@@ -77,7 +77,7 @@ def create_tracer(
             print(f"🔗 [Sentience] Attempting to initialize cloud tracing...")
             print(f"   API URL: {api_url}/v1/traces/init")
             print(f"   Run ID: {run_id}")
-            
+
             response = requests.post(
                 f"{api_url}/v1/traces/init",
                 headers={"Authorization": f"Bearer {api_key}"},
@@ -134,7 +134,9 @@ def create_tracer(
                 print(f"⚠️  [Sentience] Cloud init failed: HTTP {response.status_code}")
                 try:
                     error_data = response.json()
-                    error_msg = error_data.get("error") or error_data.get("message", "Unknown error")
+                    error_msg = error_data.get("error") or error_data.get(
+                        "message", "Unknown error"
+                    )
                     print(f"   Error: {error_msg}")
                     if "tier" in error_msg.lower() or "subscription" in error_msg.lower():
                         print(f"   💡 This may be a tier/subscription issue")

tests/test_cloud_tracing.py

@@ -332,15 +332,21 @@ def post_side_effect(*args, **kwargs):
             compressed_data = trace_upload_call[1]["data"]
             decompressed_data = gzip.decompress(compressed_data)
             trace_content = decompressed_data.decode("utf-8")
-            events = [json.loads(line) for line in trace_content.strip().split("\n") if line.strip()]
+            events = [
+                json.loads(line) for line in trace_content.strip().split("\n") if line.strip()
+            ]
 
             snapshot_events = [e for e in events if e.get("type") == "snapshot"]
             assert len(snapshot_events) > 0, "Should have snapshot event"
 
             for event in snapshot_events:
                 data = event.get("data", {})
-                assert "screenshot_base64" not in data, "screenshot_base64 should be removed from uploaded trace"
-                assert "screenshot_format" not in data, "screenshot_format should be removed from uploaded trace"
+                assert (
+                    "screenshot_base64" not in data
+                ), "screenshot_base64 should be removed from uploaded trace"
+                assert (
+                    "screenshot_format" not in data
+                ), "screenshot_format should be removed from uploaded trace"
 
         # Cleanup
         cache_dir = Path.home() / ".sentience" / "traces" / "pending"