Tuchuanhuhuhu commited on
Commit
f90157b
1 Parent(s): bb60438

修复历史记录加载功能

Browse files
Files changed (1) hide show
  1. utils.py +8 -2
utils.py CHANGED
@@ -251,10 +251,16 @@ def save_chat_history(filename, system, history, chatbot):
251
 
252
  def load_chat_history(filename, system, history, chatbot):
253
  try:
254
- print("Loading from history...")
255
  with open(os.path.join(HISTORY_DIR, filename), "r") as f:
256
  json_s = json.load(f)
257
- print(json_s)
 
 
 
 
 
 
 
258
  return filename, json_s["system"], json_s["history"], json_s["chatbot"]
259
  except FileNotFoundError:
260
  print("File not found.")
 
251
 
252
  def load_chat_history(filename, system, history, chatbot):
253
  try:
 
254
  with open(os.path.join(HISTORY_DIR, filename), "r") as f:
255
  json_s = json.load(f)
256
+ if type(json_s["history"]) == list:
257
+ new_history = []
258
+ for index, item in enumerate(json_s["history"]):
259
+ if index % 2 == 0:
260
+ new_history.append(construct_user(item))
261
+ else:
262
+ new_history.append(construct_assistant(item))
263
+ json_s["history"] = new_history
264
  return filename, json_s["system"], json_s["history"], json_s["chatbot"]
265
  except FileNotFoundError:
266
  print("File not found.")