beswarm 0.2.16__py3-none-any.whl → 0.2.17__py3-none-any.whl

beswarm/tools/worker.py

@@ -46,9 +46,15 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     Returns:
         str: 当任务成功完成时，返回字符串 "任务已完成"。
     """
+    cache_dir = Path(work_dir) / ".beswarm"
+    cache_dir.mkdir(exist_ok=True)
+    cache_file = cache_dir / "work_agent_conversation_history.json"
+    if not cache_file.exists():
+        cache_file.write_text("[]", encoding="utf-8")
+
     DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
     if DEBUG:
-        log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a", encoding="utf-8")
+        log_file = open(cache_dir / "history.log", "a", encoding="utf-8")
         log_file.write(f"========== {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ==========\n")
         original_stdout = sys.stdout
         original_stderr = sys.stderr
@@ -94,11 +100,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     }
     if cache_messages:
         if isinstance(cache_messages, bool) and cache_messages == True:
-            cache_messages = None
-            cache_file_path = Path(work_dir) / ".beswarm" / "work_agent_conversation_history.json"
-            if cache_file_path.exists():
-                with cache_file_path.open("r", encoding="utf-8") as f:
-                    cache_messages = json.load(f)
+            cache_messages = json.loads(cache_file.read_text(encoding="utf-8"))
         if cache_messages and isinstance(cache_messages, list) and len(cache_messages) > 1:
             old_goal = extract_xml_content(cache_messages[1]["content"], "goal")
             if old_goal.strip() != goal.strip():
@@ -144,11 +146,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
             instruction_agent = chatgpt(**instruction_agent_config)
             conversation_history = copy.deepcopy(work_agent.conversation["default"])
 
-            cache_dir = os.path.join(work_dir, ".beswarm")
-            os.makedirs(cache_dir, exist_ok=True)
-            cache_file = os.path.join(cache_dir, "work_agent_conversation_history.json")
-            with open(cache_file, "w", encoding="utf-8") as f:
-                f.write(json.dumps(conversation_history, ensure_ascii=False, indent=4))
+            cache_file.write_text(json.dumps(conversation_history, ensure_ascii=False, indent=4), encoding="utf-8")
 
             work_agent_system_prompt = conversation_history.pop(0)
             if conversation_history:
@@ -248,9 +246,15 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     return "任务已完成"
 
 async def worker_gen(goal, tools, work_dir, cache_messages=None):
+    cache_dir = Path(work_dir) / ".beswarm"
+    cache_dir.mkdir(exist_ok=True)
+    cache_file = cache_dir / "work_agent_conversation_history.json"
+    if not cache_file.exists():
+        cache_file.write_text("[]", encoding="utf-8")
+
     DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
     if DEBUG:
-        log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a", encoding="utf-8")
+        log_file = open(cache_dir / "history.log", "a", encoding="utf-8")
         log_file.write(f"========== {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ==========\n")
         original_stdout = sys.stdout
         original_stderr = sys.stderr
@@ -296,11 +300,7 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
     }
     if cache_messages:
         if isinstance(cache_messages, bool) and cache_messages == True:
-            cache_messages = None
-            cache_file_path = Path(work_dir) / ".beswarm" / "work_agent_conversation_history.json"
-            if cache_file_path.exists():
-                with cache_file_path.open("r", encoding="utf-8") as f:
-                    cache_messages = json.load(f)
+            cache_messages = json.loads(cache_file.read_text(encoding="utf-8"))
         if cache_messages and isinstance(cache_messages, list) and len(cache_messages) > 1:
             old_goal = extract_xml_content(cache_messages[1]["content"], "goal")
             if old_goal.strip() != goal.strip():
@@ -346,11 +346,7 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
             instruction_agent = chatgpt(**instruction_agent_config)
             conversation_history = copy.deepcopy(work_agent.conversation["default"])
 
-            cache_dir = os.path.join(work_dir, ".beswarm")
-            os.makedirs(cache_dir, exist_ok=True)
-            cache_file = os.path.join(cache_dir, "work_agent_conversation_history.json")
-            with open(cache_file, "w", encoding="utf-8") as f:
-                f.write(json.dumps(conversation_history, ensure_ascii=False, indent=4))
+            cache_file.write_text(json.dumps(conversation_history, ensure_ascii=False, indent=4), encoding="utf-8")
 
             work_agent_system_prompt = conversation_history.pop(0)
             if conversation_history:

{beswarm-0.2.16.dist-info → beswarm-0.2.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.2.16
+Version: 0.2.17
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.2.16.dist-info → beswarm-0.2.17.dist-info}/RECORD

@@ -134,8 +134,8 @@ beswarm/tools/request_input.py,sha256=gXNAJPOJektMqxJVyzNTFOeMQ7xUkO-wWMYH-r2Rdw
 beswarm/tools/screenshot.py,sha256=u6t8FCgW5YHJ_Oc4coo8e0F3wTusWE_-H8dFh1rBq9Q,1011
 beswarm/tools/search_arxiv.py,sha256=caVIUOzMhFu-r_gVgJZrH2EO9xI5iV_qLAg0b3Ie9Xg,8095
 beswarm/tools/search_web.py,sha256=w0T0aCqOVlb6Of5hn_TtpnrGXo6bMtw2aKZdkrYjycI,12069
-beswarm/tools/worker.py,sha256=ZFgSe17Je9NJuITvojZ89uEk2p-pXWJIy9GHNvMDAVo,23491
-beswarm-0.2.16.dist-info/METADATA,sha256=7dkfw1GTU0uQQ5_DxHdk956ZuvAWHfI4m6jWThucZ_4,3847
-beswarm-0.2.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.2.16.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.2.16.dist-info/RECORD,,
+beswarm/tools/worker.py,sha256=ywjD0TUVLzcrVQbVhQuNwtagca0sAuZwSjnKCVb3R6g,23027
+beswarm-0.2.17.dist-info/METADATA,sha256=PcWMP5H6J2h5o-EjQLL2ENjS4VCldPzrSvdTyghV_Ko,3847
+beswarm-0.2.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.2.17.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.2.17.dist-info/RECORD,,