cafe3310 commited on
Commit
cb25b17
·
1 Parent(s): 551e9e2

fix: 修复 Gradio 中 type="messages" 的聊天历史记录处理逻辑

Browse files
Files changed (1) hide show
  1. app.py +5 -6
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 turn in history:
18
- user_message, bot_message = turn
19
- if user_message:
20
- messages.append(HumanMessage(content=user_message))
21
- if bot_message:
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