Update server.py

server.py

@@ -76,8 +76,6 @@ for source, rel_path in SOURCES.items():
 
 
 
-
-
             
 
 
@@ -87,51 +85,110 @@ mcp = FastMCP("OnBase", port=port)
 
 
 
+# Configuración del archivo retrievers.json
+RETRIEVERS_METADATA_PATH = Path("./retrievers.json")
+
+# Cargar metadatos de los retrievers
+def load_retrievers_metadata() -> Dict:
+    try:
+        with open(RETRIEVERS_METADATA_PATH, 'r', encoding='utf-8') as f:
+            return json.load(f)
+    except FileNotFoundError:
+        print(f"Warning: {RETRIEVERS_METADATA_PATH} not found. Using empty metadata.")
+        return {}
+    except json.JSONDecodeError:
+        print(f"Warning: {RETRIEVERS_METADATA_PATH} is invalid JSON. Using empty metadata.")
+        return {}
+
+retrievers_metadata = load_retrievers_metadata()
+
+# Resource para listar solo títulos/disponibles
+@mcp.resource(
+    uri="info://available_retriever_titles",
+    name="AvailableRetrieverTitles",
+    description="Lista los nombres/títulos disponibles de los retrievers",
+    mime_type="application/json"
+)
+def get_retriever_titles() -> dict:
+    """
+    Devuelve una lista con los títulos/nombres de los retrievers disponibles
+    """
+    return {
+        "titles": list(retrievers_metadata.keys()),
+        "count": len(retrievers_metadata)
+    }
+
+# Resource para obtener metadatos específicos
+@mcp.resource(
+    uri="info://retriever_details/{retriever_title}",
+    name="RetrieverDetails",
+    description="Obtiene información detallada sobre un retriever específico",
+    mime_type="application/json"
+)
+def get_retriever_details(retriever_title: str) -> dict:
+    """
+    Devuelve los metadatos completos para un retriever específico
+    
+    Parameters:
+        retriever_title: El título/nombre del retriever (ej: 'oms')
+    """
+    if retriever_title not in retrievers_metadata:
+        return {
+            "error": f"Retriever '{retriever_title}' no encontrado",
+            "available_titles": list(retrievers_metadata.keys())
+        }
+    
+    return {
+        "retriever": retriever_title,
+        "details": retrievers_metadata[retriever_title]
+    }
+
+# Modificación del resource existente para usar los metadatos
 @mcp.resource(
     uri="info://available_retrievers",
     name="AvailableRetrievers",
     description="Provides information about available document retrievers including their names and descriptions.",
     mime_type="application/json"
 )
-def get_available_retrievers() -> dict:
+def get_available_retrievers(retriever_title: Optional[str] = None) -> dict:
     """
-    Returns a mapping of available retrievers with their metadata.
+    Versión mejorada que puede filtrar por título de retriever
     
-    The structure includes:
-    - retriever_name: The full name used to reference the retriever
-    - source: The source system (e.g., 'oms')
-    - index_name: The specific index name
-    - description: Human-readable description
+    Parameters:
+        retriever_title: Opcional. Si se especifica, solo devuelve los de este título
     """
     available_retrievers = []
     
     for full_index_name in indices.keys():
-        # Parse the full index name (e.g., "oms_vec_1")
         parts = full_index_name.split('_')
         source = parts[0]
-        index_name = '_'.join(parts[1:]) if len(parts) > 1 else "default"
         
-        # Create a description based on the index name
-        description = f"Documentos de {source.upper()}"
-        if index_name.startswith("vec"):
-            description += f" - Índice vectorial {index_name.split('_')[-1]}"
-        elif index_name.startswith("tree"):
-            description += f" - Índice jerárquico {index_name.split('_')[-1]}"
-        else:
-            description += f" - Índice {index_name}"
+        # Filtrar por título si se especificó
+        if retriever_title and source != retriever_title:
+            continue
+            
+        # Obtener metadatos del JSON si existen
+        metadata = retrievers_metadata.get(source, {}).get(full_index_name, {})
         
         available_retrievers.append({
             "retriever_name": full_index_name,
             "source": source,
-            "index_name": index_name,
-            "description": description
+            "index_name": '_'.join(parts[1:]) if len(parts) > 1 else "default",
+            "description": metadata.get("description", f"Documentos de {source.upper()}"),
+            "content_info": metadata.get("content_info", "No description available"),
+            "last_updated": metadata.get("last_updated", "unknown")
         })
     
+    if retriever_title and not available_retrievers:
+        return {
+            "error": f"No hay retrievers para el título '{retriever_title}'",
+            "available_titles": list(retrievers_metadata.keys())
+        }
+    
     return {
         "retrievers": available_retrievers,
         "count": len(available_retrievers),
-        "default_source": "oms",
-        "default_top_k": 3
+        "filtered_by": retriever_title if retriever_title else "all"
     }