Spaces:
Sleeping
Sleeping
fix: 修复 Gradio 中 type="messages" 的聊天历史记录处理逻辑
Browse files
app.py
CHANGED
|
@@ -14,12 +14,11 @@ def respond(message, history, system_message, hf_token: gr.OAuthToken = None):
|
|
| 14 |
if system_message:
|
| 15 |
messages.append(SystemMessage(content=system_message))
|
| 16 |
|
| 17 |
-
for
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
messages.append(AIMessage(content=bot_message))
|
| 23 |
|
| 24 |
messages.append(HumanMessage(content=message))
|
| 25 |
|
|
|
|
| 14 |
if system_message:
|
| 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 |
|