Spaces:
Sleeping
Sleeping
fix: 使用正确的字典键访问处理 Gradio 历史记录
Browse files
app.py
CHANGED
|
@@ -15,10 +15,10 @@ def respond(message, history, system_message, hf_token: gr.OAuthToken = None):
|
|
| 15 |
messages.append(SystemMessage(content=system_message))
|
| 16 |
|
| 17 |
for chat_message in history:
|
| 18 |
-
if chat_message
|
| 19 |
-
messages.append(HumanMessage(content=chat_message
|
| 20 |
-
elif chat_message
|
| 21 |
-
messages.append(AIMessage(content=chat_message
|
| 22 |
|
| 23 |
messages.append(HumanMessage(content=message))
|
| 24 |
|
|
|
|
| 15 |
messages.append(SystemMessage(content=system_message))
|
| 16 |
|
| 17 |
for chat_message in history:
|
| 18 |
+
if chat_message['role'] == "user":
|
| 19 |
+
messages.append(HumanMessage(content=chat_message['content']))
|
| 20 |
+
elif chat_message['role'] == "assistant":
|
| 21 |
+
messages.append(AIMessage(content=chat_message['content']))
|
| 22 |
|
| 23 |
messages.append(HumanMessage(content=message))
|
| 24 |
|