7/7

app.py

@@ -16,8 +16,8 @@ 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])
+# 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]:
@@ -115,16 +115,16 @@ class ConnectionManager:
             # 'history_messages':self.bw.get_history_messages(save_dir = config["save_dir"]),
             'history_messages':[],
         }
-        self.bw.get_history_messages(save_dir = config["save_dir"])
+        # 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,status = self.bw.load_next_message_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()
@@ -184,8 +184,8 @@ async def load_preset(request: Request):
             )
             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])
+            # 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"],

modules/llm/DeepSeek.py

@@ -30,10 +30,7 @@ class DeepSeek(BaseLLM):
     
         response = self.client.chat.completions.create(
         model="deepseek-chat",
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant"},
-            {"role": "user", "content": "Hello"},
-        ],
+        messages=self.messages,
         stream=False
 )
         return response.choices[0].message.content

requirements.txt

@@ -1,6 +1,6 @@
 anthropic
 beautifulsoup4
-chromadb
+chromadb==1.0.0
 fastapi
 langchain
 langchain_experimental