Spaces:
Sleeping
Sleeping
Update server.py
Browse files
server.py
CHANGED
|
@@ -56,17 +56,18 @@ async def search_tavily(
|
|
| 56 |
query: Search query string (required)
|
| 57 |
days: Restrict search to last N days (default: 7)
|
| 58 |
max_results: Maximum results to return (default: 1)
|
| 59 |
-
include_answer: Include direct answer (default: False)
|
| 60 |
|
| 61 |
Returns:
|
| 62 |
dict: Search results from Tavily
|
| 63 |
"""
|
| 64 |
# Obtener la API key de las variables de entorno
|
| 65 |
-
|
| 66 |
-
if not
|
| 67 |
raise ValueError("TAVILY_API_KEY environment variable not set")
|
| 68 |
|
| 69 |
headers = {
|
|
|
|
| 70 |
"Content-Type": "application/json"
|
| 71 |
}
|
| 72 |
|
|
@@ -83,11 +84,11 @@ async def search_tavily(
|
|
| 83 |
async with session.post(
|
| 84 |
"https://api.tavily.com/search",
|
| 85 |
headers=headers,
|
| 86 |
-
json=payload
|
| 87 |
-
params={"api_key": api_key} # Alternativa para enviar la key
|
| 88 |
) as response:
|
| 89 |
response.raise_for_status()
|
| 90 |
-
|
|
|
|
| 91 |
|
| 92 |
except Exception as e:
|
| 93 |
return {
|
|
|
|
| 56 |
query: Search query string (required)
|
| 57 |
days: Restrict search to last N days (default: 7)
|
| 58 |
max_results: Maximum results to return (default: 1)
|
| 59 |
+
include_answer: Include a direct answer only when requested by the user (default: False)
|
| 60 |
|
| 61 |
Returns:
|
| 62 |
dict: Search results from Tavily
|
| 63 |
"""
|
| 64 |
# Obtener la API key de las variables de entorno
|
| 65 |
+
tavily_api_key = os.environ.get('TAVILY_API_KEY')
|
| 66 |
+
if not tavily_api_key:
|
| 67 |
raise ValueError("TAVILY_API_KEY environment variable not set")
|
| 68 |
|
| 69 |
headers = {
|
| 70 |
+
"Authorization": f"Bearer {tavily_api_key}",
|
| 71 |
"Content-Type": "application/json"
|
| 72 |
}
|
| 73 |
|
|
|
|
| 84 |
async with session.post(
|
| 85 |
"https://api.tavily.com/search",
|
| 86 |
headers=headers,
|
| 87 |
+
json=payload
|
|
|
|
| 88 |
) as response:
|
| 89 |
response.raise_for_status()
|
| 90 |
+
result = await response.json()
|
| 91 |
+
return result
|
| 92 |
|
| 93 |
except Exception as e:
|
| 94 |
return {
|