Spaces:
Runtime error
Runtime error
hhhhhhhhh
Browse files- 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 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 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)
|