Update server.py

server.py

@@ -90,15 +90,28 @@ mcp = FastMCP("OnBase", port=port)
 
 
 
-@mcp.resource(uri="info://available_retrievers")
-async def _get_available_retrievers() -> dict:
+@mcp.tool()
+async def list_retrievers(source: str = None) -> dict:
+    """
+    Devuelve la lista de retrievers disponibles, opcionalmente filtrados por source
+    
+    Args:
+        source (str, optional): Si se especifica, solo devuelve los retrievers de esta fuente. Defaults to None.
+    
+    Returns:
+        dict: Diccionario con la lista de retrievers y metadatos
+    """
     try:
         available = []
-        for source, indexes in retrievers_metadata.items():
+        for current_source, indexes in retrievers_metadata.items():
+            # Si se especificó un source y no coincide, saltar esta fuente
+            if source and current_source != source:
+                continue
+                
             for index_name, metadata in indexes.items():
                 available.append({
                     "name": index_name,
-                    "source": source,
+                    "source": current_source,
                     "title": metadata.get("title", ""),
                     "description": metadata.get("description", "")
                 })
@@ -106,67 +119,16 @@ async def _get_available_retrievers() -> dict:
         return {
             "retrievers": available,
             "count": len(available),
-            "status": "success"
-        }
-    except Exception as e:
-        return {
-            "retrievers": [],
-            "count": 0,
-            "status": "error",
-            "error": str(e)
-        }
-
-
-        
-
-@mcp.tool()
-async def list_retrievers(ctx: Context) -> dict:
-    """
-    Devuelve la lista de retrievers disponibles
-    
-    Returns:
-        dict: {
-            "retrievers": [{
-                "name": str,
-                "title": str,
-                "description": str,
-                "source": str
-            }],
-            "count": int,
-            "status": str,
-            "error": str (opcional)
+            "status": "success",
+            "source_filter": source if source else "all"
         }
-    """
-    try:
-        # Lectura del resource
-        resource_contents = await ctx.read_resource("info://available_retrievers")
-        
-        if not resource_contents:
-            await ctx.warning("No resource contents found for info://available_retrievers")
-            return {
-                "retrievers": [],
-                "count": 0,
-                "status": "error",
-                "error": "Resource not found"
-            }
-            
-        resource = resource_contents[0].content  # Acceder al contenido real
-        
-        await ctx.debug(f"Retrieved {len(resource.get('retrievers', []))} retrievers")
-        
-        return {
-            "retrievers": resource.get("retrievers", []),
-            "count": resource.get("count", 0),
-            "status": "success"
-        }
-        
     except Exception as e:
-        await ctx.error(f"Error en list_retrievers: {str(e)}")
         return {
             "retrievers": [],
             "count": 0,
             "status": "error",
-            "error": str(e)
+            "error": str(e),
+            "source_filter": source if source else "all"
         }