justest commited on
Commit
f49b24a
1 Parent(s): ffa1a72

feat: 支持history

Browse files
Files changed (1) hide show
  1. app.py +14 -4
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([[None, None]], label='AI')
81
  msg = gr.Textbox(value="", label='请输入:')
82
  with gr.Row():
83
  clear = gr.Button("清空对话", scale=2)
@@ -135,12 +135,22 @@ with gr.Blocks() as demo:
135
  ```
136
  请回答:
137
  """
 
 
 
 
 
 
 
 
 
 
 
 
138
  bot_msg = g4f.ChatCompletion.create(
139
  model=model_name,
140
  provider=provider_dict[provider_name],
141
- messages=[{"role": "user",
142
- "content": prompt.format(system_msg,
143
- history[-1][0])}],
144
  stream=True)
145
  for c in bot_msg:
146
  history[-1][1] += c
 
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)
 
135
  ```
136
  请回答:
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
144
+ assistant_dict = {"role": "assistant", "content": assistant_message}
145
+
146
+ # 把用户和助理的消息字典依次添加到 chatgpt_history 中
147
+ messages.append(user_dict)
148
+ messages.append(assistant_dict)
149
+
150
  bot_msg = g4f.ChatCompletion.create(
151
  model=model_name,
152
  provider=provider_dict[provider_name],
153
+ messages=messages,
 
 
154
  stream=True)
155
  for c in bot_msg:
156
  history[-1][1] += c