geronimo-pericoli commited on
Commit
7047942
·
verified ·
1 Parent(s): b52b4d6

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +7 -6
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
- api_key = os.environ.get('TAVILY_API_KEY')
66
- if not api_key:
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
- return await response.json()
 
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 {