Update server.py

server.py

@@ -98,22 +98,17 @@ mcp = FastMCP("OnBase", port=port)
 
 
 
-@mcp.resource("data://retrievers/list/{source}")
-async def retrievers_list_resource(ctx: Context, source: str) -> dict:
+@mcp.resource("data://retrievers/list")
+async def retrievers_list_resource(ctx: Context) -> dict:
     """
-    Resource interno que proporciona la lista de retrievers.
+    Resource que devuelve todos los retrievers disponibles.
+    El filtrado por source se manejará en el tool.
     """
     try:
-        # Cargamos los metadatos desde el contexto
         retrievers_metadata = ctx.get("retrievers_metadata", {})
-        
         available = []
-        source_exists = source in retrievers_metadata if source != "all" else False
         
         for current_source, indexes in retrievers_metadata.items():
-            if source_exists and current_source != source:
-                continue
-                
             for index_name, metadata in indexes.items():
                 available.append({
                     "name": index_name,
@@ -125,9 +120,7 @@ async def retrievers_list_resource(ctx: Context, source: str) -> dict:
         return {
             "retrievers": available,
             "count": len(available),
-            "status": "success",
-            "source_requested": source,
-            "source_used": source if source_exists else "all"
+            "status": "success"
         }
         
     except Exception as e:
@@ -135,25 +128,43 @@ async def retrievers_list_resource(ctx: Context, source: str) -> dict:
             "retrievers": [],
             "count": 0,
             "status": "error",
-            "error": str(e),
-            "source_requested": source,
-            "source_used": "none"
+            "error": str(e)
         }
 
 @mcp.tool()
 async def list_retrievers(ctx: Context, source: str = None) -> dict:
     """
-    Devuelve la lista de retrievers disponibles.
+    Devuelve la lista de retrievers disponibles, opcionalmente filtrados por source.
     
     Args:
         source (str, optional): Fuente para filtrar. Default None.
     
     Returns:
-        dict: Lista de retrievers.
+        dict: Lista de retrievers con metadatos.
     """
-    # Convertimos None a "all" para el resource
-    source_param = "all" if source is None else source
-    return await ctx.resources.read(f"data://retrievers/list/{source_param}")
+    # Obtenemos todos los retrievers del resource
+    result = await ctx.resources.read("data://retrievers/list")
+    
+    # Si hay error en el resource, lo propagamos
+    if result.get("status") == "error":
+        return result
+    
+    # Filtramos por source si se especificó
+    if source:
+        filtered = [r for r in result["retrievers"] if r["source"] == source]
+        return {
+            "retrievers": filtered,
+            "count": len(filtered),
+            "status": "success",
+            "source_requested": source,
+            "source_used": source if filtered else "none"
+        }
+    
+    return {
+        **result,
+        "source_requested": None,
+        "source_used": "all"
+    }