Update app.py

app.py

@@ -182,6 +182,75 @@ async def list_retrievers(source: str = None) -> dict:
             "source_used": "none"
         }
 
+
+
+async def search_tavily(
+    query: str,
+    days: int = 7,
+    max_results: int = 1,
+    include_answer: bool = False
+) -> dict:
+    """Perform a web search using the Tavily API.
+    
+    Args:
+        query: Search query string (required)
+        days: Restrict search to last N days (default: 7)
+        max_results: Maximum results to return (default: 1)
+        include_answer: Include a direct answer only when requested by the user (default: False)
+        
+    Returns:
+        dict: Search results from Tavily
+    """
+    # Obtener la API key de las variables de entorno
+    tavily_api_key = os.environ.get('TAVILY_API_KEY')
+    if not tavily_api_key:
+        raise ValueError("TAVILY_API_KEY environment variable not set")
+    
+    headers = {
+        "Authorization": f"Bearer {tavily_api_key}",
+        "Content-Type": "application/json"
+    }
+    
+    payload = {
+        "query": query,
+        "search_depth": "basic",
+        "max_results": max_results,
+        "days": days if days else None,
+        "include_answer": include_answer
+    }
+    
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                "https://api.tavily.com/search",
+                headers=headers,
+                json=payload
+            ) as response:
+                response.raise_for_status()
+                result = await response.json()
+                return result
+                
+    except Exception as e:
+        return {
+            "error": str(e),
+            "status": "failed",
+            "query": query
+        }
+
+
+
+
+
+
+
+
+
+
+
+
+
+        
+
 # Configuración de la interfaz Gradio
 demo = gr.TabbedInterface(
     [