Update app.py

app.py

@@ -97,13 +97,13 @@ space_clients: Dict[str, Client] = {}
 
 
 
+##### ARXIV INSTANCE #####
+arxiv_tool = ArxivToolSpec(max_results=5).to_tool_list()[0]
+arxiv_tool.return_direct = True
 
 
 
-
-
-arxiv_tool = ArxivToolSpec(max_results=5).to_tool_list()[0]
-arxiv_tool.return_direct = True
+##### MCP TOOLS #####
 
 async def search_arxiv(
     query: str,
@@ -262,6 +262,104 @@ async def search_tavily(
 
 
 
+async def list_spaces_names() -> dict:
+    """
+    Devuelve una lista simplificada con los nombres y descripciones de todos los spaces disponibles.
+    
+    Returns:
+        dict: {
+            "status": "success"|"error",
+            "spaces": list[dict{"name": str, "description": str}],
+            "count": int
+        }
+    """
+    try:
+        spaces_list = [
+            {"name": space["name"], "description": space["description"]} 
+            for space in spaces_metadata["spaces"]
+        ]
+        
+        return {
+            "status": "success",
+            "spaces": spaces_list,
+            "count": len(spaces_list)
+        }
+    except Exception as e:
+        return {
+            "status": "error",
+            "message": f"Error al obtener la lista de spaces: {str(e)}",
+            "spaces": [],
+            "count": 0
+        }
+
+
+async def query_context( 
+    space_name: str, 
+    message_text: str,
+    api_name: str = "/get_context_only"
+) -> dict:
+    """
+    Esta herramienta obtiene SOLO el contexto relevante para una consulta desde un Space especializado.
+    
+    Parámetros:
+        space_name (str): Nombre exacto del Space a consultar (debe existir en 'list_spaces_names')
+        message_text (str): Texto de la consulta/pregunta para generar el contexto
+        api_name (str, optional): Endpoint de la API a usar (siempre "/get_context_only")
+        
+    Retorna:
+        dict: {
+            "status": "success"|"error",
+            "space": str,  # Nombre del Space consultado
+            "query": str,  # Texto de la consulta enviada
+            "context": str,  # Contexto formateado obtenido
+            "url": str  # URL del Space
+        }
+    """
+    
+    try:
+        space_info = next((s for s in spaces_metadata["spaces"] if s["name"] == space_name), None)
+        
+        if not space_info:
+            logger.warning(f"Space no encontrado: {space_name}")
+            return {
+                "status": "error",
+                "message": f"Space '{space_name}' no encontrado",
+                "available_spaces": [s["name"] for s in spaces_metadata["spaces"]]
+            }
+        
+        logger.info(f"Obteniendo contexto del space: {space_name}")
+
+        if space_name not in space_clients:
+            logger.debug(f"Creando nuevo cliente para space: {space_name}")
+            space_clients[space_name] = Client(space_info["url"], hf_token=HF_TOKEN)
+        
+        client = space_clients[space_name]
+        
+        logger.info(f"Enviando consulta para contexto a space {space_name}")
+        context = client.predict(
+            message=message_text,
+            api_name=api_name
+        )
+        
+        logger.debug(f"Contexto obtenido (longitud: {len(context) if context else 0})")
+        
+        return {
+            "status": "success",
+            "space": space_name,
+            "query": message_text,
+            "context": context,
+            "url": space_info["url"]
+        }
+        
+    except Exception as e:
+        logger.error(f"Error en query_context: {str(e)}", exc_info=True)
+        return {
+            "status": "error",
+            "space": space_name,
+            "query": message_text,
+            "error": str(e),
+            "url": space_info.get("url", "") if space_info else ""
+        }