From 9d081893aae509a9eea695ae246587204f5e16b0 Mon Sep 17 00:00:00 2001
From: 997146918 <997146918@qq.com>
Date: Fri, 8 Aug 2025 10:16:38 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E6=95=B0=E6=8D=AEbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 AITrain/prepare_dialogue_data.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/AITrain/prepare_dialogue_data.py b/AITrain/prepare_dialogue_data.py
index 5e282e6..3d1f5d7 100644
--- a/AITrain/prepare_dialogue_data.py
+++ b/AITrain/prepare_dialogue_data.py
@@ -50,7 +50,8 @@ def create_character_dialogue_samples(character:str, dialogues: List[Dict]) ->Li
             tempDialogue = dialogue['dialogue']
         elif tempDialogue != '':
             #确定是提问对话
-            character_samples.append({"instruction": tempDialogue,
+            character_samples.append({"character": character,
+                                      "instruction": tempDialogue,
                                       "input": "",
                                       "output": dialogue['dialogue']})
             tempDialogue = ''
@@ -65,9 +66,9 @@ def main():
     for char, convs in character_dialogues.items():
         print(f"  {char}: {len(convs)}条对话")
     #获得最终训练数据
-    final_samples = {}
+    final_samples = []
     for character in characters:
-        final_samples[character] = create_character_dialogue_samples(character, dialogues)
+        final_samples += create_character_dialogue_samples(character, dialogues)