Update server.py

server.py

@@ -95,53 +95,22 @@ mcp = FastMCP("OnBase", port=port)
 
 
 
-@mcp.prompt()
-def improve_oms_topic_prompt(topic: str, context: str = None) -> str:
-    """Generates an optimized prompt to request detailed WHO-related information about a health topic.
-    Args:
-        topic: The health topic to inquire about (e.g., 'malaria vaccination', 'pandemic preparedness')
-        context: Optional context to make the query more specific (e.g., 'in developing countries')
-    Returns:
-        A refined prompt asking for WHO-specific information on the topic.
-    """
-    base_prompt = f"Could you provide a comprehensive explanation of '{topic}'"
-    
-    if context:
-        base_prompt += f" {context}"
-    
-    return (
-        base_prompt + " based on WHO guidelines and latest recommendations? "
-        "Please include: key facts, recommended interventions, "
-        "target populations, and relevant WHO program references."
-    )
-
 
 
 
-@mcp.tool()
-async def list_retrievers(source: str = None) -> dict:
+@mcp.resource("data://retrievers/list")
+async def retrievers_list_resource(ctx: Context, source: str = None) -> dict:
     """
-    Devuelve la lista de retrievers disponibles. 
-    Si se especifica una source y existe, filtra por ella; si no existe, devuelve todas.
-    
-    Args:
-        source (str, optional): Fuente para filtrar. Si no existe, se ignorará. Defaults to None.
-    
-    Returns:
-        dict: {
-            "retrievers": Lista de retrievers (filtrados o completos),
-            "count": Número total,
-            "status": "success"|"error",
-            "source_requested": source,  # Muestra lo que se solicitó
-            "source_used": "all"|source  # Muestra lo que realmente se usó
-        }
+    Resource interno que proporciona la lista de retrievers.
     """
     try:
+        # Cargamos los metadatos desde el contexto
+        retrievers_metadata = ctx.get("retrievers_metadata", {})
+        
         available = []
         source_exists = source in retrievers_metadata if source else False
         
         for current_source, indexes in retrievers_metadata.items():
-            # Solo filtrar si el source existe, sino mostrar todo
             if source_exists and current_source != source:
                 continue
                 
@@ -160,6 +129,7 @@ async def list_retrievers(source: str = None) -> dict:
             "source_requested": source,
             "source_used": source if source_exists else "all"
         }
+        
     except Exception as e:
         return {
             "retrievers": [],
@@ -170,6 +140,22 @@ async def list_retrievers(source: str = None) -> dict:
             "source_used": "none"
         }
 
+# 2. Luego definimos el tool que usará el cliente
+@mcp.tool()
+async def list_retrievers(ctx: Context, source: str = None) -> dict:
+    """
+    Devuelve la lista de retrievers disponibles. 
+    Si se especifica una source y existe, filtra por ella; si no existe, devuelve todas.
+    
+    Args:
+        source (str, optional): Fuente para filtrar. Si no existe, se ignorará.
+    
+    Returns:
+        dict: Misma estructura que el resource pero accesible via tool.
+    """
+    # Simplemente delegamos al resource
+    return await ctx.resources.read("data://retrievers/list", params={"source": source})
+