Update server.py

server.py

@@ -41,18 +41,7 @@ Settings.embed_model = embed_model
 
 
 
-
-
-
-port = int(os.getenv("PORT", 7860))
-mcp = FastMCP("OnBase", port=port)
-
-
-
-
-
-
-
+# Configuración inicial (esto probablemente estaría en otro módulo)
 DOCUMENTS_BASE_PATH = "./"
 SOURCES = {
     "oms": "oms/",  # Esta será la carpeta base que contiene todos los subíndices
@@ -85,20 +74,31 @@ for source, rel_path in SOURCES.items():
                 print(f"Error cargando índice en {root}: {str(e)}")
                 continue
 
-@mcp.resource(
-    uri="retriever://documentos",
-    name="DocumentRetriever",
-    description="Retrieve documents from different regulations using semantic search.",
-    mime_type="application/json",
-    query_params=["query", "fuente", "top_k"]  # <- Agregar esta línea
-)
+
+
+
+
+            
+
+
+port = int(os.getenv("PORT", 7860))
+mcp = FastMCP("OnBase", port=port)
+
+
+
+@mcp.tool()
 def retrieve_docs(query: str, fuente: str = 'oms', top_k: int = 3) -> dict:
     """
+    Retrieve documents from different regulations using semantic search.
+    
     Parameters:
         query: Search query (required).
         fuente: Source name (default: 'oms').
         top_k: Number of results to return (default: 3).
     """
+    if not query:
+        return {"error": "Query parameter is required"}
+    
     if fuente not in indices:
         available = [k for k in indices.keys() if k.startswith("oms_")] if fuente.startswith("oms") else list(indices.keys())
         return {"error": f"Fuente '{fuente}' no disponible. Opciones: {available}"}