vvvvvvvv

my_tools.py

@@ -1,35 +1,38 @@
 import os, math, wikipedia
 import google.generativeai as genai
-from llama_index.core.llms import LLM
-from llama_index.llms.generic_utils import llm_completion_to_chat_response
+from llama_index.core.llms import LLM, ChatMessage, ChatResponse
 from llama_index.core.agent import ReActAgent
 from llama_index.core.tools import FunctionTool
 from duckduckgo_search import DDGS
 
-# ── Configurar clave de Google Gemini (asegúrate de tener el secreto GEMINI_API_KEY) ──
+# ── Configurar clave de Google Gemini ──
 genai.configure(api_key=os.getenv("GEMINI_API_KEY", ""))
 
 class GeminiLLM(LLM):
-    """Wrapper minimal para usar Gemini 1.5 Flash con LlamaIndex."""
-
+    """Wrapper minimal para usar Gemini 1.5 Flash con LlamaIndex."""
     def __init__(self, model: str = "models/gemini-1.5-flash-latest"):
         self.model = genai.GenerativeModel(model)
 
     def chat(self, messages, **kwargs):
         prompt = "\n".join(
-            [("User: " if m.role == "user" else "Assistant: ") + m.content for m in messages]
+            [("User: " if m.role == "user" else "Assistant: ") + m.content
+             for m in messages]
         ) + "\nAssistant:"
         resp = self.model.generate_content(prompt)
-        return llm_completion_to_chat_response(resp.text)
+        return ChatResponse(
+            message=ChatMessage(role="assistant", content=resp.text),
+            raw=resp
+        )
 
 # ── Instanciar el LLM ──
 llm = GeminiLLM()
 
 # ── Herramientas auxiliares ──
-
 def buscar_web(query: str) -> str:
     with DDGS() as ddgs:
-        results = list(ddgs.text(query, region="es-es", safesearch="moderate", timelimit="y", max_results=3))
+        results = list(ddgs.text(query, region="es-es",
+                                 safesearch="moderate", timelimit="y",
+                                 max_results=3))
         if results:
             return "\n".join([f"{r['title']}: {r['body']}" for r in results])
     return "No se encontraron resultados."