jjj

my_tools.py

@@ -25,6 +25,7 @@ import google.generativeai as genai
 HEADERS = {"User-Agent": "Mozilla/5.0"}
 
 
+
 def check_required_keys() -> None:
     missing = []
     if not os.getenv("TAVILY_API_KEY"):
@@ -49,23 +50,37 @@ ChatMessage.message = property(lambda self: self)
 class GeminiLLM(LLM):
     """Wrapper mínimo para Gemini 1.5 que satisface la interfaz de Llama-Index."""
 
-    model_name: str = Field(default="models/gemini-1.5-flash-latest")
+        model_name: str = Field(default="models/gemini-1.5-flash-latest")
     temperature: float = Field(default=0.0)
 
-    # -- inicialización -----------------------------------------------------
+    _model: object = None
+    _gen_cfg: object = None
+
+    class Config:
+        extra = "allow"
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 
-        api_key = os.getenv("GEMINI_API_KEY")
-        if not api_key:
+        # --------- Normalización rápida (evita FieldInfo) -----------------
+        if not isinstance(self.model_name, str):
+            self.model_name = "models/gemini-1.5-flash-latest"
+        if not isinstance(self.temperature, (float, int)):
+            self.temperature = 0.0
+        # ------------------------------------------------------------------
+
+        # Configurar clave y modelo
+        key = os.getenv("GEMINI_API_KEY")
+        if not key:
             raise ValueError("GEMINI_API_KEY no configurada en variables de entorno")
-        genai.configure(api_key=api_key)
+        genai.configure(api_key=key)
 
+        self._gen_cfg = genai.types.GenerationConfig(
+            temperature=float(self.temperature)
+        )
         self._model = genai.GenerativeModel(
             model_name=self.model_name,
-            generation_config=genai.types.GenerationConfig(
-                temperature=float(self.temperature)
-            ),
+            generation_config=self._gen_cfg
         )
 
         # callback manager defensivo