Jn-Huang commited on
Commit
eaaeae1
·
1 Parent(s): 1a77428

Fix multi-turn conversation: handle dict history format

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -79,11 +79,9 @@ def chat_fn(message, history, system_prompt, max_new_tokens, temperature, top_p)
79
  if system_prompt:
80
  messages.append({"role": "system", "content": system_prompt})
81
 
82
- for user_msg, assistant_msg in (history or []):
83
- if user_msg:
84
- messages.append({"role": "user", "content": user_msg})
85
- if assistant_msg:
86
- messages.append({"role": "assistant", "content": assistant_msg})
87
 
88
  if message:
89
  messages.append({"role": "user", "content": message})
 
79
  if system_prompt:
80
  messages.append({"role": "system", "content": system_prompt})
81
 
82
+ # History is already in dict format: [{"role": "user", "content": "..."}, ...]
83
+ for msg in (history or []):
84
+ messages.append(msg)
 
 
85
 
86
  if message:
87
  messages.append({"role": "user", "content": message})