Prechádzať zdrojové kódy

Fix some old JSON characters not loading (#740)

Alex "mcmonkey" Goodwin 2 rokov pred
rodič
commit
8b1f20aa04
1 zmenil súbory, kde vykonal 3 pridanie a 2 odobranie
  1. 3 2
      modules/chat.py

+ 3 - 2
modules/chat.py

@@ -349,11 +349,12 @@ def load_character(_character, name1, name2):
     if _character != 'None':
         shared.character = _character
 
-        for extension in  ["yml", "yaml", "json"]:
+        for extension in ["yml", "yaml", "json"]:
             filepath = Path(f'characters/{_character}.{extension}')
             if filepath.exists():
                 break
-        data = yaml.safe_load(open(filepath, 'r', encoding='utf-8').read())
+        file_contents = open(filepath, 'r', encoding='utf-8').read()
+        data = json.loads(file_contents) if extension == "json" else yaml.safe_load(file_contents)
 
         name2 = data['name'] if 'name' in data else data['char_name']
         for field in ['context', 'greeting', 'example_dialogue', 'char_persona', 'char_greeting', 'world_scenario']: