Update my_tools.py

my_tools.py

@@ -13,6 +13,7 @@ import google.generativeai as genai
 import asyncio
 from pydantic import Field # Para declarar campos si fuera necesario, aunque no para _generation_config
 from llama_index.core.llms import ChatMessage
+import time # Para reintentos
 
 # --- Intento mejorado para obtener la versión de LlamaIndex ---
 try:
@@ -264,43 +265,72 @@ class GeminiLLM(LLM):
 
 llm = GeminiLLM()
 
-# --- HERRAMIENTAS RESTAURADAS ---
-def buscar_web(query: str) -> str:
-    """Busca en la web utilizando DuckDuckGo y devuelve los 3 primeros resultados."""
-    try:
-        with DDGS() as ddgs:
-            # Nota: ddgs.text devuelve un generador. Convertir a lista para obtener resultados.
-            results = list(ddgs.text(query, region='es-es', safesearch='moderate', timelimit='y', max_results=3))
-            if results:
-                return "\n".join([f"Título: {r['title']}, Cuerpo: {r['body']}" for r in results])
-        return "No se encontraron resultados en la web."
-    except Exception as e:
-        return f"Error al buscar en la web: {e}"
+HEADERS = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'}
+
+def buscar_web(query: str, max_attempts: int = 2) -> str: # Añadido max_attempts
+    """Busca en la web utilizando DuckDuckGo y devuelve los 3 primeros resultados.
+    Intenta ser robusto contra errores de red o rate limits de DDG."""
+    print(f"Herramienta buscar_web: Buscando '{query}'")
+    for attempt in range(max_attempts):
+        try:
+            # Aumentar timeout y añadir headers
+            with DDGS(headers=HEADERS, timeout=25) as ddgs: # Timeout de 25 segundos
+                results = list(ddgs.text(query, region='es-es', safesearch='moderate', timelimit='y', max_results=3))
+                if results:
+                    return "\n".join([f"Título: {r['title']}, Cuerpo: {r['body']}" for r in results])
+            return "No se encontraron resultados en la web."
+        except Exception as e:
+            print(f"Error en herramienta buscar_web (intento {attempt + 1}/{max_attempts}): {type(e).__name__} - {e}")
+            if "ratelimit" in str(e).lower() or "timed out" in str(e).lower():
+                if attempt < max_attempts - 1: # Si no es el último intento
+                    wait_time = (attempt + 1) * 5 # Backoff simple: 5s, 10s
+                    print(f"Esperando {wait_time}s antes de reintentar búsqueda web...")
+                    time.sleep(wait_time)
+                else: # Último intento fallido
+                    return f"Error persistente al buscar en la web después de {max_attempts} intentos: {e}"
+            else: # Otro tipo de error, no reintentar
+                return f"Error al buscar en la web: {e}"
+    return f"La búsqueda web falló después de {max_attempts} intentos." # Si el bucle termina
 
 search_tool = FunctionTool.from_defaults(
     fn=buscar_web,
     name="web_search",
-    description="Útil para buscar información actual o general en internet. Proporciona un resumen de los resultados de búsqueda."
+    description="Busca en la web utilizando DuckDuckGo para obtener información actualizada o temas generales. Devuelve pequeños snippets de texto de los resultados. Úsalo cuando Wikipedia no sea suficiente o la información sea muy reciente."
 )
 
 def get_wikipedia_summary(query: str) -> str:
-    """Busca un resumen breve de un tema en Wikipedia (primeras 3 frases)."""
+    """Busca un resumen conciso de un tema específico en Wikipedia.
+    Intenta primero en español, y si no encuentra la página, intenta en inglés.
+    Devuelve las primeras 3 frases del artículo. Ideal para definiciones, hechos históricos."""
+    print(f"Herramienta wikipedia_lookup: Buscando '{query}'")
     try:
-        wikipedia.set_lang("es") # Asegurar el idioma
-        return wikipedia.summary(query, sentences=3, auto_suggest=False)
+        wikipedia.set_lang("es")
+        summary = wikipedia.summary(query, sentences=3, auto_suggest=False, redirect=True)
+        return summary
     except wikipedia.exceptions.PageError:
-        return f"La página '{query}' no existe en Wikipedia en español."
-    except wikipedia.exceptions.DisambiguationError as e:
-        # Devolver algunas opciones para que el LLM pueda refinar la búsqueda si es necesario
-        options_str = ", ".join(e.options[:3])
-        return f"La búsqueda '{query}' es ambigua. Posibles opciones: {options_str}. Por favor, sé más específico."
-    except Exception as e:
-        return f"Error al buscar en Wikipedia: {e}"
+        print(f"Página '{query}' no encontrada en Wikipedia en español. Intentando en inglés...")
+        try:
+            wikipedia.set_lang("en")
+            summary_en = wikipedia.summary(query, sentences=3, auto_suggest=False, redirect=True)
+            # Indicar que la info es de Wikipedia en inglés puede ser útil para el LLM
+            return f"[Información de Wikipedia en Inglés]: {summary_en}"
+        except wikipedia.exceptions.PageError:
+            return f"La página '{query}' no se encontró ni en Wikipedia en español ni en inglés."
+        except wikipedia.exceptions.DisambiguationError as e_en:
+            options_str_en = ", ".join(e_en.options[:3])
+            return f"La búsqueda '{query}' en Wikipedia en inglés es ambigua. Opciones posibles: {options_str_en}. Por favor, sé más específico."
+        except Exception as e_gen_en:
+            return f"Error al buscar '{query}' en Wikipedia en inglés: {type(e_gen_en).__name__} {e_gen_en}"
+    except wikipedia.exceptions.DisambiguationError as e_es:
+        options_str_es = ", ".join(e_es.options[:3])
+        return f"La búsqueda '{query}' en Wikipedia en español es ambigua. Opciones posibles: {options_str_es}. Por favor, sé más específico."
+    except Exception as e_gen_es:
+        return f"Error al buscar '{query}' en Wikipedia en español: {type(e_gen_es).__name__} {e_gen_es}"
 
 wikipedia_tool = FunctionTool.from_defaults(
     fn=get_wikipedia_summary,
     name="wikipedia_lookup",
-    description="Busca un resumen conciso de un tema específico en Wikipedia. Ideal para definiciones, hechos históricos, biografías, etc."
+    description="Busca un resumen conciso de un tema específico en Wikipedia (primero en español, luego en inglés si es necesario). Ideal para definiciones, hechos históricos, biografías. Ejemplo de input: {'query': 'Albert Einstein'}"
 )
 
 def calcular_expresion(expr: str) -> str:
@@ -338,10 +368,9 @@ Sigue estrictamente el formato Thought/Action/Action Input/Observation.
 Cuando respondas, hazlo en el mismo idioma que la pregunta original del usuario.
 Piensa paso a paso. Considera si la información necesaria ya está en la pregunta o en el historial de conversación antes de usar una herramienta.
 Si la pregunta contiene datos tabulares o listas directamente en el prompt, analiza esos datos cuidadosamente para responder. No necesitas herramientas para información ya proporcionada.
-Si una herramienta no te da la respuesta, considera reformular tu consulta a la herramienta o si otra herramienta podría ser más útil.
-Si después de varios intentos razonables no puedes encontrar una respuesta definitiva, indícalo claramente.
+Si una herramienta devuelve un error o no encuentra información útil después de un par de intentos, considera si puedes responder de otra manera o si debes indicar que no puedes obtener la respuesta.
 No inventes respuestas si no tienes la información.
-Para tareas de clasificación que requieren conocimiento especializado (ej. botánica), si no estás seguro de una clasificación, usa herramientas de búsqueda para verificar cada ítem antes de dar la respuesta final.
+Para tareas de clasificación que requieren conocimiento especializado (ej. botánica), si no estás seguro de una clasificación, indica que necesitas más información o que la clasificación puede ser compleja.
 """
 
 # --- AGENTE RESTAURADO ---