update

Author: filip-michalsky

stagehand/page.py

@@ -16,7 +16,7 @@
     ObserveOptions,
     ObserveResult,
 )
-from .types import DefaultExtractSchema
+from .types import DefaultExtractSchema, EmptyExtractSchema
 
 _INJECTION_SCRIPT = None
 
@@ -361,12 +361,17 @@ async def extract(
             processed_data_payload = result_dict
             if schema_to_validate_with and isinstance(processed_data_payload, dict):
                 try:
-                    validated_model = schema_to_validate_with.model_validate(
-                        processed_data_payload
-                    )
-                    processed_data_payload = (
-                        validated_model  # Payload is now the Pydantic model instance
-                    )
+                    # For extract with no params
+                    if not options_obj:
+                        validated_model = EmptyExtractSchema.model_validate(
+                            processed_data_payload
+                        )
+                        processed_data_payload = validated_model
+                    else:
+                        validated_model = schema_to_validate_with.model_validate(
+                            processed_data_payload
+                        )
+                        processed_data_payload = validated_model
                 except Exception as e:
                     self._stagehand.logger.error(
                         f"Failed to validate extracted data against schema {schema_to_validate_with.__name__}: {e}. Keeping raw data dict in .data field."