saves

BookWorld.py

@@ -47,7 +47,7 @@ class Server():
         map_file_path: str = config["map_file_path"] if "map_file_path" in config else ""
         role_file_dir: str = config["role_file_dir"] if "role_file_dir" in config else "./data/roles/"
         loc_file_path: str = config["loc_file_path"]
-        self.intervention: str = "" if  "intervention" in config else ""
+        self.intervention: str = config["intervention"] if  "intervention" in config else ""
         self.event = self.intervention
         self.script: str = config["script"] if "script" in config else ""
         self.language: str = config["language"] if "language" in config else "zh"
@@ -820,7 +820,7 @@ class Server():
         """
         if not self.if_save:
             return
-        save_dir = f"./experiment_saves/{self.experiment_name}/{self.role_llm_name}/{self.start_time}"
+        save_dir = f"./experiment_saves/{self.experiment_name}/{self.role_llm_name}_{self.start_time}"
         create_dir(save_dir)
         location_setted, goal_setted = False,False
         if stage in ["location","goal","action"]:
@@ -861,10 +861,10 @@ class Server():
             meta_info = load_json_file(os.path.join(save_dir, "./meta_info.json"))
             filename = os.path.join(save_dir, f"./server_info.json")
             states = load_json_file(filename)
-            self.__setstate__(states)   
-            for role_code in self.role_codes:
-                self.role_agents[role_code].load_from_file(save_dir) 
-            self.world_agent.load_from_file(save_dir)
+            # self.__setstate__(states)   
+            # for role_code in self.role_codes:
+            #     self.role_agents[role_code].load_from_file(save_dir) 
+            # self.world_agent.load_from_file(save_dir)
             self.history_manager.load_from_file(save_dir)
             
             for record in self.history_manager.detailed_history:
@@ -902,6 +902,7 @@ class BookWorld():
                         role_llm_name=role_llm_name, 
                         embedding_name=embedding_name)
         self.selected_scene = None
+        self.idx = 0
         
     def set_generator(self, 
                       rounds:int = 10, 
@@ -981,6 +982,8 @@ class BookWorld():
             "uuid": message_id,
             "scene": self.server.cur_round
         }
+        self.server.history_manager.add_message(message)
+        
         return message
         
     def get_settings_info(self):
@@ -1003,6 +1006,7 @@ class BookWorld():
             location_name,location_description = self.server.world_agent.find_location_name(location_code),self.server.world_agent.locations_info[location_code]["description"]
         status['location'] = {'name': location_name, 'description': location_description}
         status['characters'] = self.get_characters_info()
+        self.server.history_manager.add_status(status)
         return status
     
     def handle_message_edit(self,record_id,new_text):
@@ -1038,7 +1042,12 @@ class BookWorld():
         logs = self.server.history_manager.get_complete_history()
         story = self.server.world_agent.log2story(logs)
         return story
-    
+
+    def load_next_message_status(self,):
+        message = self.server.history_manager.message_history[self.idx]
+        status = self.server.history_manager.status_history[self.idx]
+        self.idx += 1
+        return message, status
 
     
 if __name__ == "__main__":

app.py

@@ -1,3 +1,4 @@
+from email import message
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect, HTTPException, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import HTMLResponse, FileResponse
@@ -7,13 +8,17 @@ import asyncio
 import os
 from pathlib import Path
 from datetime import datetime
-from bw_utils import is_image, load_json_file
+from bw_utils import get_grandchild_folders, is_image, load_json_file
 from BookWorld import BookWorld
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
 
 app = FastAPI()
 default_icon_path = './frontend/assets/images/default-icon.jpg'
 config = load_json_file('config.json')
+experiment_name = config["preset_path"].split("/")[-1].split(".")[0]
+save_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'experiment_saves', experiment_name)
+config["save_dir"] = os.path.join(save_dir, get_grandchild_folders(save_dir)[0])
+
 for key in config:
     if "API_KEY" in key and config[key]:
         os.environ[key] = config[key]
@@ -91,7 +96,7 @@ class ConnectionManager:
                         'data': status
                     })
                     # 添加延迟，控制消息发送频率
-                    await asyncio.sleep(0.2)  # 可以调整这个值
+                    await asyncio.sleep(2)  # 可以调整这个值
                 else:
                     break
         except asyncio.CancelledError:
@@ -102,20 +107,24 @@ class ConnectionManager:
 
     async def get_initial_data(self):
         """获取初始化数据"""
-        return {
+        data =  {
             'characters': self.bw.get_characters_info(),
             'map': self.bw.get_map_info(),
             'settings': self.bw.get_settings_info(),
             'status': self.bw.get_current_status(),
-            'history_messages':self.bw.get_history_messages(save_dir = config["save_dir"]),
+            # 'history_messages':self.bw.get_history_messages(save_dir = config["save_dir"]),
+            'history_messages':[],
         }
+        self.bw.get_history_messages(save_dir = config["save_dir"])
+        return data
     
     async def get_next_message(self):
         """从BookWorld获取下一条消息"""
-        message = self.bw.generate_next_message()
-        if not os.path.exists(message["icon"]) or not is_image(message["icon"]):
-            message["icon"] = default_icon_path
-        status = self.bw.get_current_status()
+        # message = self.bw.generate_next_message()
+        # if not os.path.exists(message["icon"]) or not is_image(message["icon"]):
+        #     message["icon"] = default_icon_path
+        # status = self.bw.get_current_status()
+        message,status = self.bw.load_next_message_status()
         return message,status
 
 manager = ConnectionManager()
@@ -173,6 +182,10 @@ async def load_preset(request: Request):
                 role_llm_name=config["role_llm_name"],
                 embedding_name=config["embedding_model_name"]
             )
+            config["preset_path"] = preset_path
+            experiment_name = preset_path.split("/")[-1].split(".")[0]
+            save_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'experiment_saves', experiment_name)
+            config["save_dir"] = os.path.join(save_dir, get_grandchild_folders(save_dir)[0])
             manager.bw.set_generator(
                 rounds=config["rounds"],
                 save_dir=config["save_dir"],

config.json

@@ -1,13 +1,13 @@
 {
-    "role_llm_name": "gemini-2.0-flash",
-    "world_llm_name": "gemini-2.0-flash",
+    "role_llm_name": "claude-3.5-sonnet",
+    "world_llm_name": "claude-3.5-sonnet",
     "embedding_model_name":"bge-m3",
-    "preset_path":"./experiment_presets/experiment_icefire.json",
-    "if_save": 0,
+    "preset_path":"./experiment_presets/experiment_icefire_bloody_wedding.json",
+    "if_save": 1,
     "scene_mode": 1,
-    "rounds": 10,
+    "rounds": 5,
     "save_dir": "",
-    "mode": "free",
+    "mode": "script",
 
     "OPENAI_API_KEY":"",
     "GEMINI_API_KEY":"",

data/maps/example_map.csv

@@ -1,4 +1,3 @@
-,laboratory,university,eldridge_corporation
-laboratory,0,2,5
-university,2,0,4
-eldridge_corporation,5,4,0
+,university,eldridge_corporation
+university,0,4
+eldridge_corporation,4,0

experiment_presets/example_free.json

@@ -4,7 +4,7 @@
     "map_file_path":"./data/maps/example_map.csv",
     "loc_file_path":"./data/locations/example_locations.json",
     "role_file_dir":"./data/roles/",
-    "role_agent_codes":["Lacia-en","Trek-en"],
+    "role_agent_codes":["Lacia-en","Trek-en","Falko-en"],
     "intervention":"One day, an enigmatic signal from an unknown source reached Lacia and Trek...",
     "script":"",
     "source":"example_world",

experiment_presets/example_script.json

@@ -4,7 +4,7 @@
     "map_file_path":"./data/maps/example_map.csv",
     "loc_file_path":"./data/locations/example_locations.json",
     "role_file_dir":"./data/roles/",
-    "role_agent_codes":["Lacia-en","Trek-en"],
+    "role_agent_codes":["Lacia-en","Trek-en","Falko-en"],
     "intervention":"",
     "script":"One day, an enigmatic signal from an unknown source reached Lacia and Trek. The pattern resembled Trek’s early consciousness digitization code—but far more evolved. Tracing its origin to lunar orbit, the two found themselves divided: Lacia urged caution, fearing the dangers of an unknown intelligence, while Trek saw it as proof that human evolution had already begun elsewhere. As they followed the signal, ideological tension grew—one seeking to contain it, the other longing to embrace it.",
     "source":"example_world",

modules/history_manager.py

@@ -8,6 +8,8 @@ import os
 class HistoryManager:
     def __init__(self):
         self.detailed_history = []
+        self.status_history = []
+        self.message_history = []
 
     def add_record(self, record):
         """添加一个事件记录
@@ -25,6 +27,12 @@ class HistoryManager:
         """
         self.detailed_history.append(record)
 
+    def add_status(self, status):
+        self.status_history.append(status)
+        
+    def add_message(self, message):
+        self.message_history.append(message)
+        
     def modify_record(self, record_id: str, detail: str):
         """修改特定记录"""
         for record in self.detailed_history: