cafe3310 commited on
Commit
ce5f3e0
·
1 Parent(s): a074dc6

fix: 使用正确的字典键访问处理 Gradio 历史记录

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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.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
 
 
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