From 150ec134b267bc8e41ba42fd5d39f03c964fc3ef Mon Sep 17 00:00:00 2001
From: 997146918 <997146918@qq.com>
Date: Sat, 9 Aug 2025 08:18:18 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E6=95=B0=E6=8D=AE=E6=A0=BC?=
 =?UTF-8?q?=E5=BC=8Fbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 AITrain/prepare_dialogue_data.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/AITrain/prepare_dialogue_data.py b/AITrain/prepare_dialogue_data.py
index 3d1f5d7..ec614b6 100644
--- a/AITrain/prepare_dialogue_data.py
+++ b/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())
-            dialogues.append(data)
+            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]: