Spaces:
Runtime error
Runtime error
sssssssssss
Browse files- my_tools.py +4 -4
my_tools.py
CHANGED
|
@@ -115,10 +115,10 @@ class GeminiLLM(LLM):
|
|
| 115 |
return ChatMessage(role="assistant", content=f"Error al generar respuesta: {e}")
|
| 116 |
'''
|
| 117 |
def chat(self, messages: list[ChatMessage], **kwargs) -> ChatMessage:
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
|
| 123 |
last_user_message = messages[-1].content
|
| 124 |
|
|
|
|
| 115 |
return ChatMessage(role="assistant", content=f"Error al generar respuesta: {e}")
|
| 116 |
'''
|
| 117 |
def chat(self, messages: list[ChatMessage], **kwargs) -> ChatMessage:
|
| 118 |
+
gemini_history = []
|
| 119 |
+
for msg in messages[:-1]:
|
| 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 |
|