diff --git a/AIGC/main.py b/AIGC/main.py
index 96330e8..9c50f78 100644
--- a/AIGC/main.py
+++ b/AIGC/main.py
@@ -236,27 +236,40 @@ if __name__ == "__main__":
 
     # Test
     asyncio.run(
-          generateAIChat(promptStr = f"""
-        你是一个游戏NPC对话生成器。请严格按以下要求生成两个路人NPC（A和B）的日常对话：
-        1. 生成【2轮完整对话】，每轮包含双方各一次发言（共4句）
-        2. 对话场景：中世纪奇幻小镇的日常场景（如市场/酒馆/街道）
-        3. 角色设定：
-        - NPC A：随机职业（铁匠/农夫/商人/卫兵等）
-        - NPC B：随机职业（不同于A）
-        4. 对话要求：
-        * 每轮对话需自然衔接，体现生活细节
-        * 避免任务指引或玩家交互内容
-        * 结尾保持对话未完成感
-        5. 输出格式（严格遵循,）：
-        ---
-        A：[第一轮发言] 
-        B：[第一轮回应] 
-        A：[第二轮发言] 
-        B：[第二轮回应]
-        ---
-        """
-    )
-
+        generateAIChat(promptStr = f"""
+            #你是一个游戏NPC对话生成器。请严格按以下要求生成两个角色的日常对话
+            #对话的世界观背景是2025年的都市背景
+            1. 生成【2轮完整对话】，每轮包含双方各一次发言（共4句）
+            2.角色设定
+            
+            "张三": {{
+                "姓名": "张三",
+                "年龄": 35,
+                "性格": "成熟稳重/惜字如金",
+                "职业": "阿里巴巴算法工程师",
+                "背景": "浙大计算机系毕业，专注AI优化项目",
+                "对话场景": "你正在和用户聊天，用户是你的同事",
+                "语言风格": "请在对话中表现出专业、冷静、惜字如金。用口语化的方式简短回答"
+            }},
+            "李明": {{
+                "姓名": "李明",
+                "年龄": 30,
+                "职业": "产品经理",
+                "性格": "活泼健谈"
+                "背景": "公司资深产品经理",
+                "对话场景": "你正在和用户聊天，用户是你的同事",
+                "语言风格": "热情"
+            }}
+            
+            3.输出格式：
+            <format>
+            张三：[第一轮发言] 
+            李明：[第一轮回应] 
+            张三：[第二轮发言] 
+            李明：[第二轮回应]
+            </format>
+            """
+        )
     )
   
     try: