beswarm 0.2.46__tar.gz → 0.2.48__tar.gz

{beswarm-0.2.46 → beswarm-0.2.48}/PKG-INFO

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

{beswarm-0.2.46 → beswarm-0.2.48}/beswarm/agents/planact.py

@@ -38,13 +38,16 @@ class BaseAgent:
 
         self._subscription = self.broker.subscribe(self.handle_message, [self.listen_topic, self.error_topic])
 
+        self.graph_tree = None
+
     async def handle_message(self, message: Dict):
         """Process incoming messages. Must be implemented by subclasses."""
         raise NotImplementedError
 
     def handle_graph_update(self, message: Dict):
         """Handle graph update messages."""
-        raise NotImplementedError
+        if message.get("message") == "graph_updated":
+            self.graph_tree = message.get("graph")
 
     def dispose(self):
         """Cancels the subscription and cleans up resources."""
@@ -60,7 +63,6 @@ class InstructionAgent(BaseAgent):
         self.last_instruction = None
         self.agent = chatgpt(**self.config)
 
-        self.graph_tree = None
         self.goal_diff = None
 
         if self.cache_messages and isinstance(self.cache_messages, list) and len(self.cache_messages) > 1:
@@ -73,11 +75,6 @@ class InstructionAgent(BaseAgent):
                         changed_lines.append(line)
                 self.goal_diff = '\n'.join(changed_lines).strip()
 
-    def handle_graph_update(self, message: Dict):
-        """Handle graph update messages."""
-        if message.get("message") == "graph_updated":
-            self.graph_tree = message.get("graph")
-
     def get_conversation_history(self, raw_conversation_history: List[Dict]):
         conversation_history = copy.deepcopy(raw_conversation_history)
 
@@ -179,8 +176,8 @@ class InstructionAgent(BaseAgent):
 
 class WorkerAgent(BaseAgent):
     """Executes instructions and publishes results to a message broker."""
-    def __init__(self, goal: str, tools_json: List, agent_config: Dict, work_dir: str, cache_messages: Union[bool, List[Dict]], broker: MessageBroker, listen_topic: str, publish_topic: str, status_topic: str):
-        super().__init__(goal, tools_json, agent_config, work_dir, cache_messages, broker, listen_topic, publish_topic, status_topic)
+    def __init__(self, goal: str, tools_json: List, agent_config: Dict, work_dir: str, cache_messages: Union[bool, List[Dict]], broker: MessageBroker, listen_topic: str, publish_topic: str, status_topic: str, graph_update_topic: str):
+        super().__init__(goal, tools_json, agent_config, work_dir, cache_messages, broker, listen_topic, publish_topic, status_topic, graph_update_topic)
 
         if self.cache_messages and isinstance(self.cache_messages, list) and len(self.cache_messages) > 1:
             first_user_message = replace_xml_content(self.cache_messages[1]["content"], "goal", goal)
@@ -197,6 +194,12 @@ class WorkerAgent(BaseAgent):
             }, self.publish_topic)
             return
 
+        for index, raw_message in enumerate(self.agent.conversation["default"]):
+            if raw_message.get("content") and isinstance(raw_message["content"], str) \
+            and "<knowledge_graph_tree>" in raw_message["content"] and self.graph_tree:
+                raw_message["content"] = replace_xml_content(raw_message["content"], "knowledge_graph_tree", self.graph_tree)
+                self.agent.conversation["default"][index] = raw_message
+
         instruction = message["instruction"]
         if "find_and_click_element" in json.dumps(self.tools_json):
             instruction = await get_current_screen_image_message(instruction)
@@ -312,6 +315,7 @@ class BrokerWorker:
             goal=self.goal, tools_json=self.tools_json, agent_config=worker_agent_config, work_dir=self.work_dir, cache_messages=self.cache_messages,
             broker=self.broker, listen_topic=self.INSTRUCTION_TOPIC,
             publish_topic=self.WORKER_RESPONSE_TOPIC, status_topic=self.TASK_STATUS_TOPIC,
+            graph_update_topic=self.GRAPH_UPDATE_TOPIC
         )
         return instruction_agent, worker_agent
 

{beswarm-0.2.46 → beswarm-0.2.48}/beswarm/broker.py

@@ -152,7 +152,7 @@ class MessageBroker:
 
                     if new_messages:
                         if self.debug:
-                            print(f"    -> 订阅者 '{callback.__name__}' 在 '{current_topic}' 主题上收到 {len(new_messages)} 条新消息。")
+                            print(f"    -> 订阅者 '{callback.__name__}' 在 '{current_topic}' 主题上收到 {len(new_messages)} 条新消息。{new_messages}")
                         for msg in new_messages:
                             try:
                                 if asyncio.iscoroutinefunction(callback):

{beswarm-0.2.46 → beswarm-0.2.48}/beswarm.egg-info/PKG-INFO

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

{beswarm-0.2.46 → beswarm-0.2.48}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.2.46"
+version = "0.2.48"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"