Iyeoluwads/firecrawl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update firecrawl.py

Browse Source
This commit is contained in:
Nicolas
2025-04-18 00:51:06 -07:00
parent d8792d2301
commit 390f3d44a3
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/python-sdk/firecrawl/firecrawl.py
+7 -8
View File
@@ -402,7 +402,7 @@ class ExtractParams(pydantic.BaseModel):
Parameters for the extract operation. Parameters for the extract operation.
""" """
prompt: Optional[str] = None prompt: Optional[str] = None
schema_: Optional[Any] = pydantic.Field(None, alias='schema') schema: Optional[Any] = pydantic.Field(None, alias='schema')
system_prompt: Optional[str] = None system_prompt: Optional[str] = None
allow_external_links: Optional[bool] = False allow_external_links: Optional[bool] = False
enable_web_search: Optional[bool] = False enable_web_search: Optional[bool] = False
@@ -1610,7 +1610,7 @@ class FirecrawlApp:
urls: Optional[List[str]] = None, urls: Optional[List[str]] = None,
*, *,
prompt: Optional[str] = None, prompt: Optional[str] = None,
schema_: Optional[Any] = None, schema: Optional[Any] = None,
system_prompt: Optional[str] = None, system_prompt: Optional[str] = None,
allow_external_links: Optional[bool] = False, allow_external_links: Optional[bool] = False,
enable_web_search: Optional[bool] = False, enable_web_search: Optional[bool] = False,
@@ -1622,7 +1622,7 @@ class FirecrawlApp:
Args: Args:
urls (Optional[List[str]]): URLs to extract from urls (Optional[List[str]]): URLs to extract from
prompt (Optional[str]): Custom extraction prompt prompt (Optional[str]): Custom extraction prompt
schema_ (Optional[Any]): JSON schema/Pydantic model schema (Optional[Any]): JSON schema/Pydantic model
system_prompt (Optional[str]): System context system_prompt (Optional[str]): System context
allow_external_links (Optional[bool]): Follow external links allow_external_links (Optional[bool]): Follow external links
enable_web_search (Optional[bool]): Enable web search enable_web_search (Optional[bool]): Enable web search
@@ -1640,13 +1640,12 @@ class FirecrawlApp:
""" """
headers = self._prepare_headers() headers = self._prepare_headers()
if not prompt and not schema_: if not prompt and not schema:
raise ValueError("Either prompt or schema is required") raise ValueError("Either prompt or schema is required")
if not urls and not prompt: if not urls and not prompt:
raise ValueError("Either urls or prompt is required") raise ValueError("Either urls or prompt is required")
schema = schema_
if schema: if schema:
if hasattr(schema, 'model_json_schema'): if hasattr(schema, 'model_json_schema'):
# Convert Pydantic model to JSON schema # Convert Pydantic model to JSON schema
@@ -1747,7 +1746,7 @@ class FirecrawlApp:
urls: List[str], urls: List[str],
*, *,
prompt: Optional[str] = None, prompt: Optional[str] = None,
schema_: Optional[Any] = None, schema: Optional[Any] = None,
system_prompt: Optional[str] = None, system_prompt: Optional[str] = None,
allow_external_links: Optional[bool] = False, allow_external_links: Optional[bool] = False,
enable_web_search: Optional[bool] = False, enable_web_search: Optional[bool] = False,
@@ -1760,7 +1759,7 @@ class FirecrawlApp:
Args: Args:
urls (List[str]): URLs to extract information from urls (List[str]): URLs to extract information from
prompt (Optional[str]): Custom extraction prompt prompt (Optional[str]): Custom extraction prompt
schema_ (Optional[Any]): JSON schema/Pydantic model schema (Optional[Any]): JSON schema/Pydantic model
system_prompt (Optional[str]): System context system_prompt (Optional[str]): System context
allow_external_links (Optional[bool]): Follow external links allow_external_links (Optional[bool]): Follow external links
enable_web_search (Optional[bool]): Enable web search enable_web_search (Optional[bool]): Enable web search
@@ -1779,7 +1778,7 @@ class FirecrawlApp:
""" """
headers = self._prepare_headers(idempotency_key) headers = self._prepare_headers(idempotency_key)
schema = schema_ schema = schema
if schema: if schema:
if hasattr(schema, 'model_json_schema'): if hasattr(schema, 'model_json_schema'):
# Convert Pydantic model to JSON schema # Convert Pydantic model to JSON schema