完成大量美化

2026-05-01 15:44:14 +08:00
parent 6b65b24b0f
commit f0e7e75ffb
36 changed files with 4393 additions and 1506 deletions
--- a/generate_test_chats.py
+++ b/generate_test_chats.py
@@ -0,0 +1,109 @@
+"""
+为测试角色生成测试聊天记录
+"""
+import json
+import uuid
+from datetime import datetime
+from pathlib import Path
+
+def create_test_chat(character_name, first_mes, chat_name="默认聊天"):
+    """
+    为角色创建测试聊天记录
+    
+    Args:
+        character_name: 角色名称
+        first_mes: 开场白
+        chat_name: 聊天名称
+    """
+    # 聊天文件路径
+    chat_dir = Path(f"data/characters/{character_name}/chats")
+    chat_dir.mkdir(parents=True, exist_ok=True)
+    chat_file = chat_dir / f"{chat_name}.jsonl"
+    
+    # 如果文件已存在，跳过
+    if chat_file.exists():
+        print(f"跳过已存在的聊天: {character_name}/{chat_name}")
+        return
+    
+    # 生成header
+    header = {
+        "user_name": "User",
+        "character_name": character_name,
+        "integrity": str(uuid.uuid4()),
+        "chat_id_hash": str(uuid.uuid4()),
+        "note_prompt": "",
+        "note_interval": 0,
+        "note_position": 0,
+        "note_depth": 0,
+        "note_role": 0,
+        "extensions": {},
+        "timedWorldInfo": {},
+        "variables": {},
+        "tainted": False,
+        "lastInContextMessageId": -1
+    }
+    
+    # 生成AI开场白消息（floor 1）
+    ai_message = {
+        "name": character_name,
+        "is_user": False,
+        "is_system": False,
+        "floor": 1,
+        "send_date": str(int(datetime.now().timestamp() * 1000)),
+        "mes": first_mes,
+        "extra": {},
+        "swipes": [],
+        "swipe_id": 0
+    }
+    
+    # 写入文件
+    with open(chat_file, 'w', encoding='utf-8') as f:
+        f.write(json.dumps(header, ensure_ascii=False) + '\n')
+        f.write(json.dumps(ai_message, ensure_ascii=False) + '\n')
+    
+    print(f"✓ 已创建聊天: {character_name}/{chat_name}")
+
+
+def main():
+    """主函数"""
+    characters_dir = Path("data/characters")
+    
+    if not characters_dir.exists():
+        print("错误: characters目录不存在")
+        return
+    
+    print("开始为测试角色生成聊天记录...\n")
+    
+    for char_folder in characters_dir.iterdir():
+        if not char_folder.is_dir():
+            continue
+        
+        char_name = char_folder.name
+        char_json = char_folder / "character.json"
+        
+        if not char_json.exists():
+            print(f"跳过 {char_name}: 没有character.json文件")
+            continue
+        
+        try:
+            # 读取角色信息
+            with open(char_json, 'r', encoding='utf-8') as f:
+                char_data = json.load(f)
+            
+            first_mes = char_data.get('first_mes', '')
+            
+            if not first_mes:
+                print(f"跳过 {char_name}: 没有开场白")
+                continue
+            
+            # 创建测试聊天
+            create_test_chat(char_name, first_mes)
+            
+        except Exception as e:
+            print(f"处理 {char_name} 时出错: {e}")
+    
+    print("\n完成！")
+
+
+if __name__ == '__main__':
+    main()