KaiChen1998 commited on
Commit
0d4bcf1
·
1 Parent(s): 9571b72

separate thinking process

Browse files
Files changed (1) hide show
  1. conversation_public.py +1 -0
conversation_public.py CHANGED
@@ -219,6 +219,7 @@ class Conversation:
219
  if type(msg) is tuple:
220
  audio_b64_str = base64.b64encode(open(msg[1], "rb").read()).decode("utf-8")
221
  msg = f'<audio src="data:audio/wav;base64,{audio_b64_str}" controls autoplay></audio>'
 
222
  ret.append({"role": "assistant", "content": msg})
223
  return ret
224
 
 
219
  if type(msg) is tuple:
220
  audio_b64_str = base64.b64encode(open(msg[1], "rb").read()).decode("utf-8")
221
  msg = f'<audio src="data:audio/wav;base64,{audio_b64_str}" controls autoplay></audio>'
222
+ msg = msg.replace('</think>', "\n# Final Response\n\n")
223
  ret.append({"role": "assistant", "content": msg})
224
  return ret
225