makaveli10 commited on
Commit
5d9af51
1 Parent(s): e3f7cd8

fix: strip() on list

Browse files
Files changed (1) hide show
  1. llm_service.py +1 -1
llm_service.py CHANGED
@@ -221,7 +221,7 @@ class TensorRTLLMEngine:
221
  llm_queue.put({"uid": transcription_output["uid"], "llm_output": self.last_output, "eos": self.eos})
222
  audio_queue.put({"llm_output": self.last_output, "eos": self.eos})
223
  conversation_history[transcription_output["uid"]].append(
224
- (transcription_output['prompt'].strip(), self.last_output.strip())
225
  )
226
  print(f"History: {conversation_history}")
227
  continue
 
221
  llm_queue.put({"uid": transcription_output["uid"], "llm_output": self.last_output, "eos": self.eos})
222
  audio_queue.put({"llm_output": self.last_output, "eos": self.eos})
223
  conversation_history[transcription_output["uid"]].append(
224
+ (transcription_output['prompt'].strip(), self.last_output[0].strip())
225
  )
226
  print(f"History: {conversation_history}")
227
  continue