Update server.py

server.py

@@ -99,20 +99,16 @@ mcp = FastMCP("OnBase", port=port)
 
 
 @mcp.resource("data://retrievers/list/{source}")
-async def retrievers_list_resource(ctx: Context) -> dict:
+async def retrievers_list_resource(ctx: Context, source: str) -> dict:
     """
     Resource interno que proporciona la lista de retrievers.
-    El parámetro 'source' viene en el URI.
     """
     try:
-        # Obtenemos el parámetro source del contexto de ruta
-        source = ctx.path_params.get("source")
-        
         # Cargamos los metadatos desde el contexto
         retrievers_metadata = ctx.get("retrievers_metadata", {})
         
         available = []
-        source_exists = source in retrievers_metadata if source else False
+        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:
@@ -144,7 +140,6 @@ async def retrievers_list_resource(ctx: Context) -> dict:
             "source_used": "none"
         }
 
-
 @mcp.tool()
 async def list_retrievers(ctx: Context, source: str = None) -> dict:
     """
@@ -156,9 +151,9 @@ async def list_retrievers(ctx: Context, source: str = None) -> dict:
     Returns:
         dict: Lista de retrievers.
     """
-    # Construimos la URI basada en si hay source o no
-    uri = "data://retrievers/list/all" if source is None else f"data://retrievers/list/{source}"
-    return await ctx.resources.read(uri)
+    # 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}")