Update server.py

server.py

@@ -1,5 +1,6 @@
 from mcp.server.fastmcp import FastMCP
 from datetime import datetime
+from llama_index.core import VectorStoreIndex
 from llama_index.core import (
     StorageContext,
     load_index_from_storage,
@@ -18,21 +19,21 @@ mcp = FastMCP("OnBase", port=port)
 
 
 
-# Rutas a los documentos por fuente
 DOCUMENTS_BASE_PATH = "./"
 SOURCES = {
     "oms": "oms/",
-    #"other": "other/",
-    #"other2": "other2/"
+    # "other": "other/",
+    # "other2": "other2/"
 }
 
-# Cargar índices (se ejecuta una vez al iniciar el servidor)
+# Cargar índices
 indices: Dict[str, VectorStoreIndex] = {}
 for source, rel_path in SOURCES.items():
     full_path = os.path.join(DOCUMENTS_BASE_PATH, rel_path)
     if os.path.exists(full_path):
         storage_context = StorageContext.from_defaults(persist_dir=full_path)
-        indices[source] = load_index_from_storage(storage_context, index_id=source)
+        index = load_index_from_storage(storage_context, index_id="vector_index")
+        indices[source] = index
 
 
 # Recurso dinámico para búsqueda