diff --git a/apps/python-sdk/firecrawl/firecrawl.py b/apps/python-sdk/firecrawl/firecrawl.py index ea74d4e69d..eecc15afc3 100644 --- a/apps/python-sdk/firecrawl/firecrawl.py +++ b/apps/python-sdk/firecrawl/firecrawl.py @@ -599,8 +599,8 @@ def extract(self, urls: List[str], params: Optional[ExtractParams] = None) -> An """ headers = self._prepare_headers() - #if not params or not params.get('prompt'): - # raise ValueError("Prompt is required") + if not params or (not params.get('prompt') and not params.get('schema')): + raise ValueError("Either prompt or schema is required") schema = params.get('schema') if schema: