vvvvvvvvvvvv

my_tools.py

@@ -12,6 +12,7 @@ from llama_index.core.callbacks.llama_debug import LlamaDebugHandler
 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
 
 # --- Intento mejorado para obtener la versión de LlamaIndex ---
 try:
@@ -32,7 +33,11 @@ except ImportError:
 
 print(f"LlamaIndex version detectada: {llama_index_version}")
 
-
+class PatchedChatMessage(ChatMessage):
+    @property
+    def message(self):
+        return self
+        
 # --- Gemini LLM personalizado ---
 class GeminiLLM(LLM):
     model_name: str = Field(default="models/gemini-1.5-flash-latest", description="The Gemini model to use.")
@@ -92,6 +97,7 @@ class GeminiLLM(LLM):
     #     return self._callback_manager
 
     # --- Implementación de Chat ---
+    '''
     def chat(self, messages: list[ChatMessage], **kwargs) -> ChatMessage:
         gemini_history = []
         for msg in messages[:-1]: 
@@ -107,6 +113,22 @@ class GeminiLLM(LLM):
         except Exception as e:
             print(f"Error en Gemini chat: {e}")
             return ChatMessage(role="assistant", content=f"Error al generar respuesta: {e}")
+    '''
+    def chat(self, messages: list[ChatMessage], **kwargs) -> ChatMessage:
+    gemini_history = []
+    for msg in messages[:-1]: 
+        role = "user" if msg.role == "user" else "model"
+        gemini_history.append({'role': role, 'parts': [{'text': msg.content}]})
+    
+    last_user_message = messages[-1].content
+    
+    chat_session = self._model_instance.start_chat(history=gemini_history)
+    try:
+        response = chat_session.send_message(last_user_message)
+        return PatchedChatMessage(role="assistant", content=response.text)
+    except Exception as e:
+        print(f"Error en Gemini chat: {e}")
+        return PatchedChatMessage(role="assistant", content=f"Error al generar respuesta: {e}")
 
     async def achat(self, messages: list[ChatMessage], **kwargs) -> ChatMessage:
         return await asyncio.to_thread(self.chat, messages, **kwargs)