Jn-Huang
commited on
Commit
·
eaaeae1
1
Parent(s):
1a77428
Fix multi-turn conversation: handle dict history format
Browse files
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 |
-
|
| 83 |
-
|
| 84 |
-
|
| 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})
|