justest commited on
Commit
2c8f8ad
1 Parent(s): f49b24a

feat: 支持history

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,7 +77,7 @@ with gr.Blocks() as demo:
77
 
78
  memory = ConversationBufferWindowMemory(k=10, memory_key="chat_history")
79
 
80
- chatbot = gr.Chatbot([], label='AI')
81
  msg = gr.Textbox(value="", label='请输入:')
82
  with gr.Row():
83
  clear = gr.Button("清空对话", scale=2)
@@ -137,7 +137,7 @@ with gr.Blocks() as demo:
137
  """
138
 
139
  messages = []
140
- for user_message, assistant_message in history:
141
  # 创建一个字典来存储用户的消息,角色为 "user",内容为 user_message
142
  user_dict = {"role": "user", "content": user_message}
143
  # 创建一个字典来存储助理的消息,角色为 "assistant",内容为 assistant_message
 
77
 
78
  memory = ConversationBufferWindowMemory(k=10, memory_key="chat_history")
79
 
80
+ chatbot = gr.Chatbot([[None, None]], label='AI')
81
  msg = gr.Textbox(value="", label='请输入:')
82
  with gr.Row():
83
  clear = gr.Button("清空对话", scale=2)
 
137
  """
138
 
139
  messages = []
140
+ for user_message, assistant_message in history[1:]:
141
  # 创建一个字典来存储用户的消息,角色为 "user",内容为 user_message
142
  user_dict = {"role": "user", "content": user_message}
143
  # 创建一个字典来存储助理的消息,角色为 "assistant",内容为 assistant_message