Update server.py

server.py

@@ -100,50 +100,52 @@ arxiv_tool.return_direct = True
 
 @mcp.tool()
 async def search_arxiv(
-    ctx: Context, 
     query: str,
     max_results: int = 5
-) -> dict:
+) -> Dict[str, Any]:
     """
-    Busca artículos académicos en ArXiv y devuelve los resultados en bruto.
+    Busca artículos académicos en ArXiv.
     
     Args:
-        query: Términos de búsqueda (ej. "machine learning", "ia")
+        query: Términos de búsqueda (ej. "deep learning")
         max_results: Número máximo de resultados (1-10, default 5)
     
     Returns:
-        dict: Resultados directos de la API de ArXiv en formato JSON
+        Dict: Resultados de la búsqueda con metadatos de los papers
     """
     try:
         # Configurar máximo de resultados
-        arxiv_tool.metadata.max_results = min(max(1, max_results), 10)
+        max_results = min(max(1, max_results), 10)
+        arxiv_tool.metadata.max_results = max_results
         
         # Ejecutar búsqueda y obtener resultados
-        results = arxiv_tool(query=query)
+        tool_output = arxiv_tool(query=query)
         
-        # Procesar documentos a formato simple
+        # Procesar documentos
         papers = []
-        for doc in results[0][1]:  # Lista de documentos
+        for doc in tool_output.raw_output:  # Acceder correctamente a los documentos
+            content = doc.text_resource.text.split('\n')
             papers.append({
-                "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/')
+                'title': content[0].split(': ')[1] if ': ' in content[0] else content[0],
+                'abstract': '\n'.join(content[1:]).strip(),
+                'pdf_url': content[0].split(': ')[0].replace('http://', 'https://'),
+                'arxiv_id': content[0].split(': ')[0].split('/')[-1].replace('v1', '')
             })
         
         return {
-            "papers": papers,
-            "count": len(papers),
-            "query": query,
-            "status": "success"
+            'papers': papers,
+            'count': len(papers),
+            'query': query,
+            'status': 'success'
         }
         
     except Exception as e:
         return {
-            "papers": [],
-            "count": 0,
-            "query": query,
-            "status": "error",
-            "error": str(e)
+            'papers': [],
+            'count': 0,
+            'query': query,
+            'status': 'error',
+            'error': str(e)
         }