Nancy1906 commited on
Commit
a6fd858
·
verified ·
1 Parent(s): 137d3f7

sssssssssss

Browse files
Files changed (1) hide show
  1. 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
- 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
 
 
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