eliott/browser-use
1
0
Fork 0
mirror of https://github.com/browser-use/browser-use synced 2026-05-06 17:52:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

Refactor perplexity search to use httpx for asynchronous requests

Browse Source 
This update replaces the synchronous requests library with httpx for making asynchronous API calls in the perplexity search function. The change improves performance and error handling by utilizing async capabilities, ensuring better integration with the overall asynchronous architecture.
This commit is contained in:
Magnus Müller
2025-05-25 16:46:08 +02:00
parent 18d1c0d895
commit 8a5f7a5ead
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
examples/custom-functions/perplexity_search.py
View File

@@ -37,9 +37,7 @@ controller = Controller(exclude_actions=['search_google'], output_model=PersonLi
@controller.registry.action('Search the web for a specific query with perplexity')
async def search_web(query: str):
import json
import requests
import httpx
url = 'https://api.perplexity.ai/chat/completions'
@@ -52,14 +50,15 @@ async def search_web(query: str):
}
headers = {'Authorization': f'Bearer {PERPLEXITY_API_KEY}', 'Content-Type': 'application/json'}
response = requests.request('POST', url, json=payload, headers=headers)
response_json = json.loads(response.text)
content = response_json['choices'][0]['message']['content']
citations = response_json['citations']
output = f'{content}\n\nCitations:\n' + '\n'.join(citations)
logger.info(output)
return ActionResult(extracted_content=output, include_in_memory=True)
async with httpx.AsyncClient() as client:
response = await client.post(url, json=payload, headers=headers)
response.raise_for_status()
response_json = response.json()
content = response_json['choices'][0]['message']['content']
citations = response_json['citations']
output = f'{content}\n\nCitations:\n' + '\n'.join(citations)
logger.info(output)
return ActionResult(extracted_content=output, include_in_memory=True)
names = [