Update server.py

server.py

@@ -80,7 +80,7 @@ for source, rel_path in SOURCES.items():
                 storage_path = os.path.join(root, "storage_nodes")
                 storage_context = StorageContext.from_defaults(persist_dir=storage_path)
                 index_name = os.path.basename(root)
-                indices[index_name] = load_index_from_storage(storage_context)
+                indices[index_name] = load_index_from_storage(storage_context) #, index_id="vector_index"
                 logger.info(f"Índice cargado correctamente: {index_name}")
             except Exception as e:
                 logger.error(f"Error cargando índice {index_name}: {str(e)}", exc_info=True)
@@ -241,7 +241,7 @@ async def list_retrievers(ctx: Context, source: str = None) -> dict:
 
 
 
-# Función principal de búsqueda
+# Función de retrievers
 @mcp.tool()
 def retrieve_docs(
     query: str,
@@ -255,26 +255,27 @@ def retrieve_docs(
         query (str): Texto de búsqueda (requerido)
         retrievers (List[str]): Nombres de retrievers a consultar (requerido)
         top_k (int): Número de resultados por retriever (opcional, default=3)
-    
-    Ejemplo:
-        retrieve_docs(
-            query="estándares farmacéuticos",
-            retrievers=["vec_1", "vec_2"],
-            top_k=2
-        )
     """
+    logger.info(f"Iniciando búsqueda para query: '{query}'")
+    logger.debug(f"Parámetros - retrievers: {retrievers}, top_k: {top_k}")
+    
     results = {}
     invalid = []
     
     for name in retrievers:
         if name not in indices:
+            logger.warning(f"Retriever no encontrado: {name}")
             invalid.append(name)
             continue
             
         try:
+            logger.info(f"Procesando retriever: {name}")
+            
             # 1. Obtener el índice y realizar la búsqueda
+            logger.debug(f"Creando retriever para {name} con top_k={top_k}")
             retriever = indices[name].as_retriever(similarity_top_k=top_k)
             nodes = retriever.retrieve(query)
+            logger.info(f"Retrieved {len(nodes)} documentos de {name}")
             
             # 2. Buscar metadatos COMPLETOS
             metadata = {}
@@ -284,8 +285,9 @@ def retrieve_docs(
                     metadata = indexes[name]
                     source = src
                     break
+            logger.debug(f"Metadatos encontrados para {name}: {metadata.keys()}")
             
-            # 3. Construir respuesta para ESTE retriever
+            # 3. Construir respuesta
             results[name] = {
                 "title": metadata.get("title", name),
                 "documents": [
@@ -300,8 +302,10 @@ def retrieve_docs(
                 "source": source,
                 "last_updated": metadata.get("last_updated", "")
             }
+            logger.info(f"Retriever {name} procesado exitosamente")
             
         except Exception as e:
+            logger.error(f"Error procesando retriever {name}: {str(e)}", exc_info=True)
             results[name] = {
                 "error": str(e),
                 "retriever": name
@@ -315,11 +319,13 @@ def retrieve_docs(
     }
     
     if invalid:
+        logger.warning(f"Retrievers inválidos: {invalid}. Opciones válidas: {list(indices.keys())}")
         response["warnings"] = {
             "invalid_retrievers": invalid,
             "valid_options": list(indices.keys())
         }
     
+    logger.info(f"Búsqueda completada. Total resultados: {len(results)}")
     return response