Fix python sdk for extract

mendableai · Jan 18, 2025 · 6383bf2 · 6383bf2
1 parent 3363b2d
commit 6383bf2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: