AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-12 17:39:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Nick: fix extract python sdk

Browse Source
This commit is contained in:
Nicolas 2024-11-26 19:43:32 -03:00
parent dbda659e71
commit ea1d3d7025
2 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/python-sdk/firecrawl/__init__.py
View File

@ -13,7 +13,7 @@ import os
from .firecrawl import FirecrawlApp # noqa from .firecrawl import FirecrawlApp # noqa
__version__ = "1.6.0" __version__ = "1.6.1"
# Define the logger for the Firecrawl project # Define the logger for the Firecrawl project
logger: logging.Logger = logging.getLogger("firecrawl") logger: logging.Logger = logging.getLogger("firecrawl")

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

@ -39,13 +39,6 @@ class FirecrawlApp:
data: Optional[Any] = None data: Optional[Any] = None
error: Optional[str] = None error: Optional[str] = None
class ErrorResponse(pydantic.BaseModel):
"""
Error response.
"""
success: bool
error: str
def __init__(self, api_key: Optional[str] = None, api_url: Optional[str] = None) -> None: def __init__(self, api_key: Optional[str] = None, api_url: Optional[str] = None) -> None:
""" """
Initialize the FirecrawlApp instance with API key, API URL. Initialize the FirecrawlApp instance with API key, API URL.
@ -460,7 +453,7 @@ class FirecrawlApp:
self._handle_error(response, 'check batch scrape status') self._handle_error(response, 'check batch scrape status')
def extract(self, urls: List[str], params: Optional[ExtractParams] = None) -> Union[ExtractResponse, ErrorResponse]: def extract(self, urls: List[str], params: Optional[ExtractParams] = None) -> Any:
""" """
Extracts information from a URL using the Firecrawl API. Extracts information from a URL using the Firecrawl API.
@ -493,7 +486,11 @@ class FirecrawlApp:
headers headers
) )
if response.status_code == 200: if response.status_code == 200:
return response.json() data = response.json()
if data['success']:
return data
else:
raise Exception(f'Failed to extract. Error: {data["error"]}')
else: else:
self._handle_error(response, "extract") self._handle_error(response, "extract")
except Exception as e: except Exception as e: