This commit is contained in:
Timothy Jaeryang Baek
2026-03-17 17:58:01 -05:00
parent fcf7208352
commit de3317e26b
220 changed files with 17200 additions and 22836 deletions
+8 -8
View File
@@ -24,26 +24,26 @@ def search_tavily(
Returns:
list[SearchResult]: A list of search results
"""
url = "https://api.tavily.com/search"
url = 'https://api.tavily.com/search'
headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {api_key}",
'Content-Type': 'application/json',
'Authorization': f'Bearer {api_key}',
}
data = {"query": query, "max_results": count}
data = {'query': query, 'max_results': count}
response = requests.post(url, headers=headers, json=data)
response.raise_for_status()
json_response = response.json()
results = json_response.get("results", [])
results = json_response.get('results', [])
if filter_list:
results = get_filtered_results(results, filter_list)
return [
SearchResult(
link=result["url"],
title=result.get("title", ""),
snippet=result.get("content"),
link=result['url'],
title=result.get('title', ''),
snippet=result.get('content'),
)
for result in results
]