lint

Author: filip-michalsky

stagehand/handlers/extract_handler.py

@@ -14,9 +14,9 @@
     ExtractResult,
 )
 from stagehand.utils import (
+    convert_dict_keys_to_snake_case,
     inject_urls,
     transform_url_strings_to_ids,
-    convert_dict_keys_to_snake_case,
 )
 
 T = TypeVar("T", bound=BaseModel)
@@ -151,7 +151,9 @@ async def extract(
 
         processed_data_payload = raw_data_dict  # Default to the raw dictionary
 
-        if schema and isinstance(raw_data_dict, dict):  # schema is the Pydantic model type
+        if schema and isinstance(
+            raw_data_dict, dict
+        ):  # schema is the Pydantic model type
             # Try direct validation first
             try:
                 validated_model_instance = schema.model_validate(raw_data_dict)

stagehand/page.py

@@ -416,11 +416,17 @@ async def extract(
                 except Exception as first_error:
                     # Fallback: normalize keys to snake_case and try once more
                     try:
-                        normalized = convert_dict_keys_to_snake_case(processed_data_payload)
+                        normalized = convert_dict_keys_to_snake_case(
+                            processed_data_payload
+                        )
                         if not options_obj:
-                            validated_model = EmptyExtractSchema.model_validate(normalized)
+                            validated_model = EmptyExtractSchema.model_validate(
+                                normalized
+                            )
                         else:
-                            validated_model = schema_to_validate_with.model_validate(normalized)
+                            validated_model = schema_to_validate_with.model_validate(
+                                normalized
+                            )
                         processed_data_payload = validated_model
                     except Exception as second_error:
                         self._stagehand.logger.error(