github/v-firecrawl
github/v-firecrawl
0
Fork 0
Code Actions

Prepare headers consistently

Browse Source
This commit is contained in:
Matt Joyce 2024-06-08 11:08:26 +10:00
parent 8f9a165c2f
commit c71ea7a795
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -45,10 +45,8 @@ class FirecrawlApp:
Exception: If the scrape request fails.
"""
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {self.api_key}'
}
headers = self._prepare_headers()
# Prepare the base scrape parameters with the URL
scrape_params = {'url': url}
@ -101,10 +99,7 @@ class FirecrawlApp:
Raises:
Exception: If the search request fails.
"""
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {self.api_key}'
}
headers = self._prepare_headers()
json_data = {'query': query}
if params:
json_data.update(params)
@ -297,3 +292,4 @@ class FirecrawlApp:
raise Exception(f'Failed to {action}. Status code: {response.status_code}. Error: {error_message}')
else:
raise Exception(f'Unexpected error occurred while trying to {action}. Status code: {response.status_code}')