AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-06-04 11:24:40 +08:00
Code Issues 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/python-sdk/firecrawl/firecrawl.py
View File

@ -402,7 +402,7 @@ class ExtractParams(pydantic.BaseModel):
Parameters for the extract operation.
"""
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
allow_external_links: Optional[bool] = False
enable_web_search: Optional[bool] = False
@ -1610,7 +1610,7 @@ class FirecrawlApp:
urls: Optional[List[str]] = None,
*,
prompt: Optional[str] = None,
schema_: Optional[Any] = None,
schema: Optional[Any] = None,
system_prompt: Optional[str] = None,
allow_external_links: Optional[bool] = False,
enable_web_search: Optional[bool] = False,
@ -1622,7 +1622,7 @@ class FirecrawlApp:
Args:
urls (Optional[List[str]]): URLs to extract from
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
allow_external_links (Optional[bool]): Follow external links
enable_web_search (Optional[bool]): Enable web search
@ -1640,13 +1640,12 @@ class FirecrawlApp:
"""
headers = self._prepare_headers()
if not prompt and not schema_:
if not prompt and not schema:
raise ValueError("Either prompt or schema is required")
if not urls and not prompt:
raise ValueError("Either urls or prompt is required")
schema = schema_
if schema:
if hasattr(schema, 'model_json_schema'):
# Convert Pydantic model to JSON schema
@ -1747,7 +1746,7 @@ class FirecrawlApp:
urls: List[str],
*,
prompt: Optional[str] = None,
schema_: Optional[Any] = None,
schema: Optional[Any] = None,
system_prompt: Optional[str] = None,
allow_external_links: Optional[bool] = False,
enable_web_search: Optional[bool] = False,
@ -1760,7 +1759,7 @@ class FirecrawlApp:
Args:
urls (List[str]): URLs to extract information from
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
allow_external_links (Optional[bool]): Follow external links
enable_web_search (Optional[bool]): Enable web search
@ -1779,7 +1778,7 @@ class FirecrawlApp:
"""
headers = self._prepare_headers(idempotency_key)
schema = schema_
schema = schema
if schema:
if hasattr(schema, 'model_json_schema'):
# Convert Pydantic model to JSON schema