Просмотр исходного кода

Load and save chat history in utf-8

Aleksey Smolenchuk 2 лет назад
Родитель
Сommit
3baf5fc700
1 измененных файлов с 3 добавлено и 3 удалено
  1. 3 3
      modules/chat.py

+ 3 - 3
modules/chat.py

@@ -291,7 +291,7 @@ def save_history(timestamp=True):
         fname = f"{prefix}persistent.json"
     if not Path('logs').exists():
         Path('logs').mkdir()
-    with open(Path(f'logs/{fname}'), 'w') as f:
+    with open(Path(f'logs/{fname}'), 'w', encoding='utf-8') as f:
         f.write(json.dumps({'data': shared.history['internal'], 'data_visible': shared.history['visible']}, indent=2))
     return Path(f'logs/{fname}')
 
@@ -321,7 +321,7 @@ def load_history(file, name1, name2):
 
 def load_default_history(name1, name2):
     if Path('logs/persistent.json').exists():
-        load_history(open(Path('logs/persistent.json'), 'rb').read(), name1, name2)
+        load_history(open(Path('logs/persistent.json'), 'rb', encoding='utf-8').read(), name1, name2)
     else:
         shared.history['internal'] = []
         shared.history['visible'] = []
@@ -355,7 +355,7 @@ def load_character(_character, name1, name2):
         name2 = shared.settings['name2_pygmalion']
 
     if Path(f'logs/{shared.character}_persistent.json').exists():
-        load_history(open(Path(f'logs/{shared.character}_persistent.json'), 'rb').read(), name1, name2)
+        load_history(open(Path(f'logs/{shared.character}_persistent.json'), 'rb', encoding='utf-8').read(), name1, name2)
 
     if shared.args.cai_chat:
         return name2, context, generate_chat_html(shared.history['visible'], name1, name2, shared.character)