调整数据格式bug

AITrain/prepare_dialogue_data.py

@@ -15,7 +15,10 @@ def load_dialogue_data(file_path: str) -> List[Dict]:
     with open(file_path, 'r', encoding='utf-8') as f:
         for line in f:
             data = json.loads(line.strip())
+            if data and isinstance(data, dict) and 'role' in data:
                 dialogues.append(data)
+            else:
+                pass
     return dialogues
 
 def get_dialogue_characters(dialogues: List[Dict]) -> List[str]: