beswarm 0.3.16__py3-none-any.whl → 0.3.18__py3-none-any.whl

beswarm/agents/planact.py

@@ -12,9 +12,9 @@ from ..broker import MessageBroker
 from ..aient.aient.models import chatgpt
 from ..aient.aient.plugins import get_function_call_list, registry
 from ..prompt import worker_system_prompt, instruction_system_prompt, Goal
-from ..utils import extract_xml_content, get_current_screen_image_message, replace_xml_content, register_mcp_tools, setup_logger
+from ..utils import extract_xml_content, get_current_screen_image_message, register_mcp_tools, setup_logger
 from ..aient.aient.models.chatgpt import ModelNotFoundError, TaskComplete, RetryFailedError, InputTokenCountExceededError, BadRequestError
-from ..aient.aient.architext.architext import Messages, SystemMessage, UserMessage, AssistantMessage, ToolCalls, ToolResults, Texts, RoleMessage, Images, Files
+from ..aient.aient.architext.architext import Messages, UserMessage, Texts
 
 try:
     from importlib import metadata
@@ -39,7 +39,7 @@ class BaseAgent:
         self.config = agent_config
         self.logger = agent_config.get("logger", None)
         self.cache_messages = cache_messages
-        if cache_messages and isinstance(cache_messages, bool) and cache_messages == True:
+        if cache_messages and isinstance(cache_messages, bool) and cache_messages:
             self.cache_messages = Messages.load(self.pkl_file)
         self.broker = broker
         self.listen_topic = listen_topic
@@ -119,14 +119,14 @@ class InstructionAgent(BaseAgent):
             raw_response = await self.agent.ask_async(instruction_prompt)
         except ModelNotFoundError as e:
             raise Exception(str(e))
-        except RetryFailedError as e:
+        except RetryFailedError:
             self.logger.error("❌ Commander retry failed, retrying...")
             self.broker.publish(message, self.error_topic)
             return
-        except InputTokenCountExceededError as e:
+        except InputTokenCountExceededError:
             self.broker.publish({"status": "error", "result": "The request body is too long, please try again."}, self.status_topic)
             return
-        except BadRequestError as e:
+        except BadRequestError:
             self.broker.publish({"status": "error", "result": "Bad request error!"}, self.status_topic)
             return
 
@@ -192,13 +192,13 @@ class WorkerAgent(BaseAgent):
         except TaskComplete as e:
             self.broker.publish({"status": "finished", "result": e.completion_message}, self.status_topic)
             return
-        except InputTokenCountExceededError as e:
+        except InputTokenCountExceededError:
             self.broker.publish({"status": "error", "result": "The request body is too long, please try again."}, self.status_topic)
             return
-        except BadRequestError as e:
+        except BadRequestError:
             self.broker.publish({"status": "error", "result": "Bad request error!"}, self.status_topic)
             return
-        except RetryFailedError as e:
+        except RetryFailedError:
             self.logger.error("❌ Worker retry failed, retrying...")
             self.broker.publish(message, self.error_topic)
             return
@@ -261,7 +261,8 @@ class BrokerWorker:
             all_mcp_tools = await self.mcp_manager.get_all_tools()
             self.tools.extend([tool.name for tool in sum(all_mcp_tools.values(), [])])
         self.tools = [item for item in self.tools if not isinstance(item, dict)]
-        if "task_complete" not in self.tools: self.tools.append("task_complete")
+        if "task_complete" not in self.tools:
+            self.tools.append("task_complete")
         self.tools_json = [value for _, value in get_function_call_list(self.tools).items()]
 
     def _task_status_subscriber(self, message: Dict):

beswarm/tools/worker.py

@@ -26,6 +26,7 @@ async def worker_gen(goal: str, tools: List[Union[str, Dict]], work_dir: str, ca
     start_time = datetime.now()
     task_manager = get_task_manager()
     current_task_manager.set(task_manager)
+    current_work_dir.set(work_dir)
     worker_instance = BrokerWorker(goal, tools, work_dir, cache_messages, broker, mcp_manager, task_manager, kgm)
     async for result in worker_instance.stream_run():
         yield result

{beswarm-0.3.16.dist-info → beswarm-0.3.18.dist-info}/METADATA

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

{beswarm-0.3.16.dist-info → beswarm-0.3.18.dist-info}/RECORD

@@ -7,7 +7,7 @@ beswarm/prompt.py,sha256=45onnyoY9plKM86KQefbPw5z9QJMn-mVnjlFQZcrjz0,34373
 beswarm/taskmanager.py,sha256=2Uz_cthW9rWkQMJdzgsXAMlfN8Ni2Qj_DOq_L-p6XZc,12662
 beswarm/utils.py,sha256=mtWzksvdar7dR8oRGvFBEAFXjigk4m1MD1arrx9wEuU,9157
 beswarm/agents/chatgroup.py,sha256=PzrmRcDKAbB7cxL16nMod_CzPosDV6bfTmXxQVuv-AQ,12012
-beswarm/agents/planact.py,sha256=FwflJBgGOvfbnaMYvXng2JDqX5hBQrvNEa4liwlP-CI,18787
+beswarm/agents/planact.py,sha256=_47F8GMyqswwE6AiHnPK1CTI_GifCyTkjIEWnXapYfQ,18655
 beswarm/aient/aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
 beswarm/aient/aient/architext/architext/__init__.py,sha256=79Ih1151rfcqZdr7F8HSZSTs_iT2SKd1xCkehMsXeXs,19
 beswarm/aient/aient/architext/architext/core.py,sha256=E5QTM0NNoHPxRawH_Lmb01azAAlmLlm0bPy5c78zG38,37972
@@ -118,11 +118,11 @@ beswarm/tools/screenshot.py,sha256=hyL6F8_k9Y03Nb_X18cY-klCpWWdkqyC-iGXfKX-7jc,1
 beswarm/tools/search_arxiv.py,sha256=NLiJV1B7Um6EuZXLxnL950d837_he2LGG7qaGACSgwg,10750
 beswarm/tools/search_web.py,sha256=yARa0fLRXGDm3Wdrxb_ISmWe99b-fPH-BrP37qsdMYY,17804
 beswarm/tools/subtasks.py,sha256=Fsf_542CkECcwFwxD-F38_IUsmk06xe1P7e6pBPTy4Y,12894
-beswarm/tools/worker.py,sha256=mQ1qdrQ8MgL99byAbTvxfEByFFGN9mty3UHqHjARMQ8,2331
+beswarm/tools/worker.py,sha256=lVBnuyGyUR5G0NqeDMx6pR9VP4DGXeFKx1R3P6dOySM,2366
 beswarm/tools/write_csv.py,sha256=u0Hq18Ksfheb52MVtyLNCnSDHibITpsYBPs2ub7USYA,1466
 beswarm/tools/write_file.py,sha256=L2coBqz-aRFxPBvJBrbWbUJhu7p3oKAAGb9R144bFtk,4926
-beswarm-0.3.16.dist-info/METADATA,sha256=kfgEfg8-PLGeQID5lf520jNtUkuJlba_jsOOt7moEJo,3878
-beswarm-0.3.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.3.16.dist-info/entry_points.txt,sha256=URK7Y4PDzBgxIecQnxsWTu4O-eaFa1CoAcNTWh5R7LM,45
-beswarm-0.3.16.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.3.16.dist-info/RECORD,,
+beswarm-0.3.18.dist-info/METADATA,sha256=U4J6elSXjb8Ijm8gKFDAsFbTqqCDUWecm0VMgwUx04k,3878
+beswarm-0.3.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.3.18.dist-info/entry_points.txt,sha256=URK7Y4PDzBgxIecQnxsWTu4O-eaFa1CoAcNTWh5R7LM,45
+beswarm-0.3.18.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.3.18.dist-info/RECORD,,