Update server.py

server.py

@@ -95,55 +95,39 @@ port = int(os.getenv("PORT", 7860))
 mcp = FastMCP("OnBase", port=port)
 
 
-# Instancia global de ArXiv
-paper_tool = ArxivToolSpec()
+arxiv_tool = ArxivToolSpec(max_results=5).to_tool_list()[0]
+arxiv_tool.return_direct = True
 
 @mcp.tool()
-async def search_arxiv_papers(
-    ctx: Context,
+async def search_arxiv(
+    ctx: Context, 
     query: str,
-    max_results: Optional[int] = 5,
-    sort_by: Optional[str] = "relevance"
+    max_results: int = 5
 ) -> dict:
     """
-    Search for academic papers on ArXiv using natural language queries.
+    Busca artículos académicos en ArXiv y devuelve los resultados en bruto.
     
     Args:
-        query: Natural language search query (e.g. "machine learning in healthcare")
-        max_results: Maximum number of results to return (default 5, max 10)
-        sort_by: Sorting method ("relevance" or "last_updated_date")
+        query: Términos de búsqueda (ej. "machine learning", "ia")
+        max_results: Número máximo de resultados (1-10, default 5)
     
     Returns:
-        dict: {
-            "papers": List of paper summaries,
-            "count": Number of results,
-            "query": Original query,
-            "status": "success" or "error"
-        }
+        dict: Resultados directos de la API de ArXiv en formato JSON
     """
     try:
-        # Validar parámetros
-        max_results = min(max(1, max_results), 10)  # Limitar entre 1 y 10
-        if sort_by not in ["relevance", "last_updated_date"]:
-            sort_by = "relevance"
+        # Configurar máximo de resultados
+        arxiv_tool.metadata.max_results = min(max(1, max_results), 10)
         
-        # Usar el tool de ArXiv
-        results = paper_tool.arxiv_search(
-            query=query,
-            max_results=max_results,
-            sort_by=sort_by
-        )
+        # Ejecutar búsqueda y obtener resultados
+        results = arxiv_tool(query=query)
         
-        # Procesar resultados
+        # Procesar documentos a formato simple
         papers = []
-        for paper in results:
+        for doc in results[0][1]:  # Lista de documentos
             papers.append({
-                "title": paper.metadata.get("Title", ""),
-                "authors": paper.metadata.get("Authors", ""),
-                "abstract": paper.metadata.get("Summary", ""),
-                "published": paper.metadata.get("Published", ""),
-                "pdf_url": paper.metadata.get("PDF url", ""),
-                "arxiv_id": paper.metadata.get("Entry ID", "").split('/')[-1]
+                "content": doc.text_resource.text,
+                "metadata": str(doc.metadata),
+                "pdf_url": doc.text_resource.text.split('\n')[0].replace('http://arxiv.org/pdf/', 'https://arxiv.org/pdf/')
             })
         
         return {
@@ -154,7 +138,6 @@ async def search_arxiv_papers(
         }
         
     except Exception as e:
-        await ctx.error(f"Error in ArXiv search: {str(e)}")
         return {
             "papers": [],
             "count": 0,