Nancy1906 commited on
Commit
8228da5
·
verified ·
1 Parent(s): a6fd858

hhhhhhhhh

Browse files
Files changed (1) hide show
  1. my_tools.py +9 -9
my_tools.py CHANGED
@@ -120,15 +120,15 @@ class GeminiLLM(LLM):
120
  role = "user" if msg.role == "user" else "model"
121
  gemini_history.append({'role': role, 'parts': [{'text': msg.content}]})
122
 
123
- last_user_message = messages[-1].content
124
-
125
- chat_session = self._model_instance.start_chat(history=gemini_history)
126
- try:
127
- response = chat_session.send_message(last_user_message)
128
- return PatchedChatMessage(role="assistant", content=response.text)
129
- except Exception as e:
130
- print(f"Error en Gemini chat: {e}")
131
- return PatchedChatMessage(role="assistant", content=f"Error al generar respuesta: {e}")
132
 
133
  async def achat(self, messages: list[ChatMessage], **kwargs) -> ChatMessage:
134
  return await asyncio.to_thread(self.chat, messages, **kwargs)
 
120
  role = "user" if msg.role == "user" else "model"
121
  gemini_history.append({'role': role, 'parts': [{'text': msg.content}]})
122
 
123
+ last_user_message = messages[-1].content
124
+
125
+ chat_session = self._model_instance.start_chat(history=gemini_history)
126
+ try:
127
+ response = chat_session.send_message(last_user_message)
128
+ return PatchedChatMessage(role="assistant", content=response.text)
129
+ except Exception as e:
130
+ print(f"Error en Gemini chat: {e}")
131
+ return PatchedChatMessage(role="assistant", content=f"Error al generar respuesta: {e}")
132
 
133
  async def achat(self, messages: list[ChatMessage], **kwargs) -> ChatMessage:
134
  return await asyncio.to_thread(self.chat, messages, **kwargs)