Update server.py

server.py

@@ -1,17 +1,62 @@
-# server.py
 from mcp.server.fastmcp import FastMCP
 import os
+import aiohttp  # Necesario para las peticiones HTTP asíncronas
 
 port = int(os.getenv("PORT", 7860))
 mcp = FastMCP("OnBase", port=port)
 
 @mcp.tool()
-def get_document_info(id: int):
-    return {
-        "doc_id": id,
-        "doc_name": "Document Name",
-        # ... resto de tus campos
+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 direct answer (default: False)
+        
+    Returns:
+        dict: Search results from Tavily
+    """
+    # Obtener la API key de las variables de entorno
+    api_key = os.environ.get('TAVILY_API_KEY')
+    if not api_key:
+        raise ValueError("TAVILY_API_KEY environment variable not set")
+    
+    headers = {
+        "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,
+                params={"api_key": api_key}  # Alternativa para enviar la key
+            ) as response:
+                response.raise_for_status()
+                return await response.json()
+                
+    except Exception as e:
+        return {
+            "error": str(e),
+            "status": "failed",
+            "query": query
+        }
 
 if __name__ == "__main__":
     mcp.run("sse")