llllllllll

my_tools.py

@@ -1,6 +1,14 @@
-import os, math, wikipedia
+import os, math, wikipedia, asyncio
 import google.generativeai as genai
-from llama_index.core.llms import LLM, ChatMessage, ChatResponse
+from typing import List, Generator
+
+from llama_index.core.llms import (
+    LLM,
+    ChatMessage,
+    ChatResponse,
+    CompletionResponse,
+    CompletionResponseGen,
+)
 from llama_index.core.agent import ReActAgent
 from llama_index.core.tools import FunctionTool
 from duckduckgo_search import DDGS
@@ -9,30 +17,71 @@ from duckduckgo_search import DDGS
 genai.configure(api_key=os.getenv("GEMINI_API_KEY", ""))
 
 class GeminiLLM(LLM):
-    """Wrapper minimal para usar Gemini 1.5 Flash con LlamaIndex."""
+    """Implementación mínima de LLM que satisface todos los métodos abstractos."""
+
     def __init__(self, model: str = "models/gemini-1.5-flash-latest"):
-        self.model = genai.GenerativeModel(model)
+        self._model = genai.GenerativeModel(model)
+        self._model_name = model
 
-    def chat(self, messages, **kwargs):
+    # ---------- metadata ----------
+    @property
+    def metadata(self):
+        return {
+            "context_window": 32768,
+            "num_output": 2048,
+            "is_chat_model": True,
+            "model_name": self._model_name,
+        }
+
+    # ---------- chat ----------
+    def chat(self, messages: List[ChatMessage], **kwargs) -> ChatResponse:
         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)
+        resp = self._model.generate_content(prompt)
         return ChatResponse(
             message=ChatMessage(role="assistant", content=resp.text),
             raw=resp
         )
 
+    def stream_chat(self, messages: List[ChatMessage], **kwargs) -> Generator[ChatResponse, None, None]:
+        yield self.chat(messages, **kwargs)
+
+    async def achat(self, messages: List[ChatMessage], **kwargs) -> ChatResponse:
+        return self.chat(messages, **kwargs)
+
+    async def astream_chat(self, messages: List[ChatMessage], **kwargs):
+        yield self.chat(messages, **kwargs)
+
+    # ---------- complete ----------
+    def complete(self, prompt: str, **kwargs) -> CompletionResponse:
+        resp = self._model.generate_content(prompt)
+        return CompletionResponse(text=resp.text, raw=resp)
+
+    def stream_complete(self, prompt: str, **kwargs) -> CompletionResponseGen:
+        yield self.complete(prompt, **kwargs)
+
+    async def acomplete(self, prompt: str, **kwargs) -> CompletionResponse:
+        return self.complete(prompt, **kwargs)
+
+    async def astream_complete(self, prompt: str, **kwargs):
+        yield self.complete(prompt, **kwargs)
+
 # ── 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."
@@ -40,7 +89,7 @@ def buscar_web(query: str) -> str:
 search_tool = FunctionTool.from_defaults(
     fn=buscar_web,
     name="web_search",
-    description="Busca en la web utilizando DuckDuckGo."
+    description="Busca en la web utilizando DuckDuckGo.",
 )
 
 def get_wikipedia_summary(query: str) -> str:
@@ -52,7 +101,7 @@ def get_wikipedia_summary(query: str) -> str:
 wikipedia_tool = FunctionTool.from_defaults(
     fn=get_wikipedia_summary,
     name="wikipedia_lookup",
-    description="Obtiene un resumen breve de un tema en Wikipedia."
+    description="Obtiene un resumen breve de un tema en Wikipedia.",
 )
 
 def calcular_expresion(expr: str) -> str:
@@ -65,7 +114,7 @@ def calcular_expresion(expr: str) -> str:
 calculator_tool = FunctionTool.from_defaults(
     fn=calcular_expresion,
     name="calculadora",
-    description="Resuelve expresiones matemáticas (trigonometría, log, potencias, etc.)."
+    description="Resuelve expresiones matemáticas (trigonometría, log, potencias, etc.).",
 )
 
 # ── Agente ReAct con herramientas ──