beswarm 0.2.43__py3-none-any.whl → 0.2.45__py3-none-any.whl

beswarm/agents/chatgroup.py

@@ -100,7 +100,7 @@ class Tee:
 
 class ChatGroupWorker:
     """The 'glue' class that orchestrates agents via a MessageBroker."""
-    def __init__(self, tools: List[Union[str, Dict]], work_dir: str, cache_messages: Union[bool, List[Dict]] = None, broker: MessageBroker = None, mcp_manager: MCPManager = None, task_manager = None):
+    def __init__(self, tools: List[Union[str, Dict]], work_dir: str, cache_messages: Union[bool, List[Dict]] = None, broker: MessageBroker = None, mcp_manager: MCPManager = None, task_manager = None, kgm = None):
         self.tools = tools
         self.work_dir = Path(work_dir)
         self.cache_messages = cache_messages
@@ -108,6 +108,7 @@ class ChatGroupWorker:
         self.broker = broker
         self.mcp_manager = mcp_manager
         self.task_manager = task_manager
+        self.kgm = kgm
         self.task_completion_event = asyncio.Event()
         self.final_result = None
         self._status_subscription = None
@@ -122,6 +123,7 @@ class ChatGroupWorker:
         cache_dir = self.work_dir / ".beswarm"
         cache_dir.mkdir(parents=True, exist_ok=True)
         self.task_manager.set_root_path(self.work_dir)
+        self.kgm.set_root_path(self.work_dir)
         self.cache_file = cache_dir / "work_agent_conversation_history.json"
         if not self.cache_file.exists():
             self.cache_file.write_text("[]", encoding="utf-8")

beswarm/agents/planact.py

@@ -17,7 +17,7 @@ from ..utils import extract_xml_content, get_current_screen_image_message, repla
 
 class BaseAgent:
     """Base class for agents, handling common initialization and disposal."""
-    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):
+    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 = None):
         self.goal = goal
         self.tools_json = tools_json
         self.work_dir = work_dir
@@ -32,12 +32,20 @@ class BaseAgent:
         self.error_topic = listen_topic + ".error"
         self.publish_topic = publish_topic
         self.status_topic = status_topic
+        self.graph_update_topic = graph_update_topic
+        if self.graph_update_topic:
+            self.graph_update_subscription = self.broker.subscribe(self.handle_graph_update, self.graph_update_topic)
+
         self._subscription = self.broker.subscribe(self.handle_message, [self.listen_topic, self.error_topic])
 
     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
+
     def dispose(self):
         """Cancels the subscription and cleans up resources."""
         if self._subscription:
@@ -46,12 +54,13 @@ class BaseAgent:
 
 class InstructionAgent(BaseAgent):
     """Generates instructions and publishes them 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)
 
         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:
@@ -64,8 +73,13 @@ class InstructionAgent(BaseAgent):
                         changed_lines.append(line)
                 self.goal_diff = '\n'.join(changed_lines).strip()
 
-    def get_conversation_history(self, conversation_history: List[Dict]):
-        conversation_history = copy.deepcopy(conversation_history)
+    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)
 
         self.cache_file.write_text(json.dumps(conversation_history, ensure_ascii=False, indent=4), encoding="utf-8")
 
@@ -83,6 +97,17 @@ class InstructionAgent(BaseAgent):
             elif isinstance(conversation_history[0]["content"], list) and extracted_content:
                 conversation_history[0]["content"].append({"type": "text", "text": extracted_content})
 
+            for message in conversation_history:
+                if message.get("content") and isinstance(message["content"], str) \
+                and "<knowledge_graph_tree>" in message["content"] and self.graph_tree:
+                    message["content"] = replace_xml_content(message["content"], "knowledge_graph_tree", self.graph_tree)
+
+        for index, message in enumerate(raw_conversation_history):
+            if message.get("content") and isinstance(message["content"], str) \
+            and "<knowledge_graph_tree>" in message["content"] and self.graph_tree:
+                message["content"] = replace_xml_content(message["content"], "knowledge_graph_tree", self.graph_tree)
+                raw_conversation_history[index] = message
+
         return conversation_history
 
     async def handle_message(self, message: Dict):
@@ -188,7 +213,7 @@ class WorkerAgent(BaseAgent):
 
 class BrokerWorker:
     """The 'glue' class that orchestrates agents via a MessageBroker."""
-    def __init__(self, goal: str, tools: List[Union[str, Dict]], work_dir: str, cache_messages: Union[bool, List[Dict]] = None, broker = None, mcp_manager = None, task_manager = None):
+    def __init__(self, goal: str, tools: List[Union[str, Dict]], work_dir: str, cache_messages: Union[bool, List[Dict]] = None, broker = None, mcp_manager = None, task_manager = None, kgm = None):
         self.goal = goal
         self.tools = tools
         self.work_dir = Path(work_dir)
@@ -197,20 +222,25 @@ class BrokerWorker:
         self.broker = broker
         self.mcp_manager = mcp_manager
         self.task_manager = task_manager
+        self.kgm = kgm
         self.task_completion_event = asyncio.Event()
         self.final_result = None
         self._status_subscription = None
-        self.setup()
 
         self.channel = self.broker.request_channel()
         self.INSTRUCTION_TOPIC = self.channel + ".instructions"
         self.WORKER_RESPONSE_TOPIC = self.channel + ".worker_responses"
         self.TASK_STATUS_TOPIC =self.channel +  ".task_status"
+        self.GRAPH_UPDATE_TOPIC = self.channel + ".knowledge_graph"
+
+        self.setup()
 
     def setup(self):
         cache_dir = self.work_dir / ".beswarm"
         cache_dir.mkdir(parents=True, exist_ok=True)
         self.task_manager.set_root_path(self.work_dir)
+        self.kgm.set_root_path(self.work_dir)
+        self.kgm.set_publish_topic(self.GRAPH_UPDATE_TOPIC)
         self.cache_file = cache_dir / "work_agent_conversation_history.json"
         if not self.cache_file.exists():
             self.cache_file.write_text("[]", encoding="utf-8")
@@ -274,13 +304,14 @@ class BrokerWorker:
         instruction_agent = InstructionAgent(
             goal=self.goal, tools_json=self.tools_json, agent_config=instruction_agent_config, work_dir=self.work_dir, cache_messages=self.cache_messages,
             broker=self.broker, listen_topic=self.WORKER_RESPONSE_TOPIC,
-            publish_topic=self.INSTRUCTION_TOPIC, status_topic=self.TASK_STATUS_TOPIC
+            publish_topic=self.INSTRUCTION_TOPIC, status_topic=self.TASK_STATUS_TOPIC,
+            graph_update_topic=self.GRAPH_UPDATE_TOPIC
         )
 
         worker_agent = WorkerAgent(
             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
+            publish_topic=self.WORKER_RESPONSE_TOPIC, status_topic=self.TASK_STATUS_TOPIC,
         )
         return instruction_agent, worker_agent
 

beswarm/core.py

@@ -1,11 +1,13 @@
-from .taskmanager import TaskManager
 from .broker import MessageBroker
 from .bemcp.bemcp import MCPManager
+from .taskmanager import TaskManager
+from .knowledge_graph import KnowledgeGraphManager
 
 """
 全局共享实例
 """
 
-task_manager = TaskManager()
 broker = MessageBroker()
-mcp_manager = MCPManager()
+mcp_manager = MCPManager()
+task_manager = TaskManager()
+kgm = KnowledgeGraphManager(broker=broker)

beswarm/knowledge_graph.py

@@ -0,0 +1,209 @@
+import uuid
+import networkx as nx
+from pathlib import Path
+
+class KnowledgeGraphManager:
+    """
+    一个使用 NetworkX 管理知识图谱的管理器。
+    - 图数据存储在 networkx.DiGraph 对象中。
+    - 数据持久化为 GraphML 文件。
+    - 提供添加、删除、重命名、移动节点和渲染树状图的功能。
+    """
+    def __init__(self, storage_path="knowledge_graph.graphml", broker=None, publish_topic=None):
+        """
+        初始化知识图谱管理器。
+
+        Args:
+            storage_path (str, optional): GraphML文件的存储路径。
+                                          默认为 'knowledge_graph.graphml'。
+        """
+        self.storage_path = Path(storage_path)
+        self.graph = nx.DiGraph()
+        self.root_path = None
+        self.broker = broker
+        self.publish_topic = publish_topic
+        # self._load_graph()
+
+    def set_root_path(self, root_path):
+        """设置工作根目录并加载持久化的任务状态。"""
+        if self.root_path is not None:
+            return
+        self.root_path = Path(root_path)
+        self.cache_dir = self.root_path / ".beswarm"
+        self.cache_dir.mkdir(parents=True, exist_ok=True)
+        self.storage_path = self.cache_dir / "knowledge_graph.graphml"
+
+        self._load_graph()
+
+    def _load_graph(self):
+        """从文件加载图，如果文件不存在或加载失败，则创建一个新的。"""
+        if self.storage_path.exists():
+            try:
+                self.graph = nx.read_graphml(self.storage_path, node_type=str)
+            except Exception:
+                self._create_new_graph()
+        else:
+            self._create_new_graph()
+
+    def _create_new_graph(self):
+        """创建一个带有根节点的新图谱并保存。"""
+        self.graph = nx.DiGraph()
+        self.graph.add_node("root", name=".", description="知识图谱根节点")
+        self._save_graph()
+
+    def set_publish_topic(self, publish_topic):
+        if not publish_topic:
+            return
+        self.publish_topic = publish_topic
+
+    def _save_graph(self):
+        """将当前图的状态保存到文件。"""
+        nx.write_graphml(self.graph, self.storage_path)
+
+        if self.publish_topic and self.broker:
+            self.broker.publish({"message": "graph_updated", "graph": self.render_tree()}, self.publish_topic)
+
+    def _get_node_id_by_path(self, path: str):
+        """通过'/'分隔的路径查找节点的唯一ID，支持前缀匹配。"""
+        if path is None or not path.strip() or path.strip() in ['.', '/']:
+            return "root"
+
+        segments = path.strip('/').split('/')
+        current_node_id = "root"
+
+        for segment in segments:
+            # 首先尝试完全匹配
+            found_child = False
+            for child_id in self.graph.successors(current_node_id):
+                if self.graph.nodes[child_id].get('name') == segment:
+                    current_node_id = child_id
+                    found_child = True
+                    break
+
+            if found_child:
+                continue
+
+            # 如果完全匹配失败，尝试前缀匹配
+            prefix_matches = []
+            for child_id in self.graph.successors(current_node_id):
+                child_name = self.graph.nodes[child_id].get('name', '')
+                if child_name.startswith(segment):
+                    prefix_matches.append(child_id)
+
+            if len(prefix_matches) == 1:
+                # 只有一个前缀匹配，使用这个节点
+                current_node_id = prefix_matches[0]
+            elif len(prefix_matches) > 1:
+                # 多个前缀匹配，存在歧义，返回None
+                return None
+            else:
+                # 没有前缀匹配，节点不存在
+                return None
+
+        return current_node_id
+
+    def add_node(self, parent_path: str, node_name: str, description: str = "") -> str:
+        """在指定父节点下添加一个新节点。"""
+        if not node_name.strip():
+            return "❌ 错误：节点名称不能为空。"
+        if '/' in node_name:
+            return f"❌ 错误：节点名称 '{node_name}' 不能包含'/'。"
+
+        parent_id = self._get_node_id_by_path(parent_path)
+        if parent_id is None:
+            return f"❌ 错误：父路径 '{parent_path}' 不存在。"
+
+        for child_id in self.graph.successors(parent_id):
+            if self.graph.nodes[child_id].get('name') == node_name:
+                return f"❌ 错误：在 '{parent_path}' 下已存在名为 '{node_name}' 的节点。"
+
+        new_node_id = str(uuid.uuid4())
+        self.graph.add_node(new_node_id, name=node_name, description=description)
+        self.graph.add_edge(parent_id, new_node_id)
+        self._save_graph()
+        return f"✅ 成功在 '{parent_path}' 下添加节点 '{node_name}'。"
+
+    def delete_node(self, node_path: str) -> str:
+        """删除一个节点及其所有子孙节点。"""
+        if node_path is None or node_path.strip() in ['.', '/']:
+            return "❌ 错误：不能删除根节点。"
+
+        node_id = self._get_node_id_by_path(node_path)
+        if node_id is None:
+            return f"❌ 错误：路径 '{node_path}' 不存在。"
+        if node_id == "root":
+            return "❌ 错误：不能删除根节点。"
+
+        descendants = nx.descendants(self.graph, node_id)
+        self.graph.remove_nodes_from(descendants.union({node_id}))
+        self._save_graph()
+        return f"✅ 成功删除节点 '{node_path}' 及其所有子节点。"
+
+    def rename_node(self, node_path: str, new_name: str) -> str:
+        """重命名一个节点。"""
+        if not new_name.strip():
+            return "❌ 错误：新名称不能为空。"
+        if '/' in new_name:
+            return f"❌ 错误：新名称 '{new_name}' 不能包含'/'。"
+
+        node_id = self._get_node_id_by_path(node_path)
+        if node_id is None:
+            return f"❌ 错误：路径 '{node_path}' 不存在。"
+        if node_id == "root":
+            return "❌ 错误：不能重命名根节点。"
+
+        parent_id = list(self.graph.predecessors(node_id))[0]
+        for sibling_id in self.graph.successors(parent_id):
+            if sibling_id != node_id and self.graph.nodes[sibling_id].get('name') == new_name:
+                return f"❌ 错误：同级目录下已存在名为 '{new_name}' 的节点。"
+
+        self.graph.nodes[node_id]['name'] = new_name
+        self._save_graph()
+        return f"✅ 成功将节点 '{node_path}' 重命名为 '{new_name}'。"
+
+    def move_node(self, source_path: str, target_parent_path: str) -> str:
+        """将一个节点移动到另一个父节点下。"""
+        source_id = self._get_node_id_by_path(source_path)
+        if source_id is None:
+            return f"❌ 错误：源路径 '{source_path}' 不存在。"
+        if source_id == "root":
+            return "❌ 错误：不能移动根节点。"
+
+        target_parent_id = self._get_node_id_by_path(target_parent_path)
+        if target_parent_id is None:
+            return f"❌ 错误：目标父路径 '{target_parent_path}' 不存在。"
+
+        if source_id == target_parent_id or target_parent_id in nx.descendants(self.graph, source_id):
+            return "❌ 错误：不能将节点移动到其自身或其子孙节点下。"
+
+        source_name = self.graph.nodes[source_id]['name']
+        for child_id in self.graph.successors(target_parent_id):
+            if self.graph.nodes[child_id].get('name') == source_name:
+                return f"❌ 错误：目标目录 '{target_parent_path}' 下已存在同名节点 '{source_name}'。"
+
+        old_parent_id = list(self.graph.predecessors(source_id))[0]
+        self.graph.remove_edge(old_parent_id, source_id)
+        self.graph.add_edge(target_parent_id, source_id)
+        self._save_graph()
+        return f"✅ 成功将节点 '{source_path}' 移动到 '{target_parent_path}' 下。"
+
+    def render_tree(self) -> str:
+        """渲染整个知识图谱为树状结构的文本。"""
+        if not self.graph or "root" not in self.graph:
+            return "图谱为空或未正确初始化。"
+
+        root_name = self.graph.nodes["root"].get("name", ".")
+        tree_lines = [root_name]
+        self._build_tree_string_recursive("root", "", tree_lines)
+        return "\n".join(tree_lines)
+
+    def _build_tree_string_recursive(self, parent_id, prefix, tree_lines):
+        """递归辅助函数，用于构建树状图字符串。"""
+        children = sorted(list(self.graph.successors(parent_id)), key=lambda n: self.graph.nodes[n].get('name', ''))
+        for i, child_id in enumerate(children):
+            is_last = (i == len(children) - 1)
+            connector = "└── " if is_last else "├── "
+            node_name = self.graph.nodes[child_id].get('name', '[Unnamed Node]')
+            tree_lines.append(f"{prefix}{connector}{node_name}")
+            new_prefix = prefix + "    " if is_last else prefix + "│   "
+            self._build_tree_string_recursive(child_id, new_prefix, tree_lines)

beswarm/taskmanager.py

@@ -1,3 +1,4 @@
+import os
 import json
 import uuid
 import asyncio
@@ -21,11 +22,12 @@ class TaskManager:
     一个带并发控制的异步任务管理器。
     它管理任务的生命周期，并通过一个固定大小的工作者池来控制并发执行的任务数量。
     """
-    def __init__(self, concurrency_limit=3):
-        if concurrency_limit <= 0:
+    def __init__(self, concurrency_limit=None):
+        self.concurrency_limit = concurrency_limit or int(os.getenv("BESWARM_CONCURRENCY_LIMIT", "3"))
+
+        if self.concurrency_limit <= 0:
             raise ValueError("并发限制必须大于0")
 
-        self.concurrency_limit = concurrency_limit
         self.tasks_cache = {}          # 存储所有任务的状态和元数据, key: task_id
 
         self._pending_queue = asyncio.Queue() # 内部待办任务队列
@@ -101,7 +103,7 @@ class TaskManager:
                     self._pending_queue.task_done()
 
             except asyncio.CancelledError:
-                print(f"[{worker_name}] 被取消，正在退出...")
+                # print(f"[{worker_name}] 被取消，正在退出...")
                 break
             except Exception as e:
                 print(f"[{worker_name}] 循环中遇到严重错误: {e}")

beswarm/tools/__init__.py

@@ -4,6 +4,7 @@ from .completion import task_complete
 from .search_arxiv import search_arxiv
 from .repomap import get_code_repo_map
 from .write_csv import append_row_to_csv
+from .graph import add_knowledge_node, delete_knowledge_node, rename_knowledge_node, move_knowledge_node, get_knowledge_graph_tree
 from .request_input import request_admin_input
 from .screenshot import save_screenshot_to_file
 from .worker import worker, worker_gen, chatgroup
@@ -48,6 +49,11 @@ __all__ = [
     "list_directory",
     "get_task_result",
     "get_url_content",
+    "add_knowledge_node",
+    "move_knowledge_node",
+    "delete_knowledge_node",
+    "rename_knowledge_node",
+    "get_knowledge_graph_tree",
     "append_row_to_csv",
     "set_readonly_path",
     "get_code_repo_map",

beswarm/tools/graph.py

@@ -0,0 +1,73 @@
+from ..aient.src.aient.plugins import register_tool
+from ..core import kgm
+
+@register_tool()
+def add_knowledge_node(parent_path: str, node_name: str, description: str = "") -> str:
+    """
+    在知识图谱的指定父路径下添加一个新节点。
+
+    Args:
+        parent_path (str): 父节点的路径。路径由'/'分隔，例如 'a/b'。根节点路径为 '.' 或 '/'。
+        node_name (str): 新节点的名称。名称中不能包含'/'字符。
+        description (str, optional): 节点的可选描述信息。默认为空字符串。
+
+    Returns:
+        str: 操作结果的描述信息，例如成功或失败的原因。
+    """
+    return kgm.add_node(parent_path, node_name, description)
+
+@register_tool()
+def delete_knowledge_node(node_path: str) -> str:
+    """
+    从知识图谱中删除一个节点及其所有后代节点。
+
+    Args:
+        node_path (str): 要删除的节点的完整路径，例如 'a/b/c'。不允许删除根节点。
+
+    Returns:
+        str: 操作结果的描述信息，例如成功或失败的原因。
+    """
+    return kgm.delete_node(node_path)
+
+@register_tool()
+def rename_knowledge_node(node_path: str, new_name: str) -> str:
+    """
+    重命名知识图谱中的一个现有节点。
+
+    Args:
+        node_path (str): 要重命名的节点的当前完整路径。
+        new_name (str): 节点的新名称。名称中不能包含'/'字符。
+
+    Returns:
+        str: 操作结果的描述信息，例如成功或失败的原因。
+    """
+    return kgm.rename_node(node_path, new_name)
+
+@register_tool()
+def move_knowledge_node(source_path: str, target_parent_path: str) -> str:
+    """
+    将一个节点（及其整个子树）移动到知识图谱中的另一个父节点下。
+
+    Args:
+        source_path (str): 要移动的节点的当前完整路径。
+        target_parent_path (str): 目标父节点的完整路径。节点将被移动到这个新父节点之下。
+
+    Returns:
+        str: 操作结果的描述信息，例如成功或失败的原因。
+    """
+    return kgm.move_node(source_path, target_parent_path)
+
+@register_tool()
+def get_knowledge_graph_tree() -> str:
+    """
+    渲染并返回整个知识图谱的文本树状图。
+
+    此工具不需要任何参数，它会读取当前的图状态并生成一个易于阅读的、
+    表示层级结构的字符串。
+
+    注意：此函数返回的知识图谱状态是实时更新的，永远是最新的。只需要调用一次，不必重复调用。
+
+    Returns:
+        str: 表示整个知识图谱的、格式化的树状结构字符串。
+    """
+    return "<knowledge_graph_tree>" + kgm.render_tree() + "</knowledge_graph_tree>"

beswarm/tools/subtasks.py

@@ -138,6 +138,7 @@ def create_tasks_from_csv(goal_template: str, csv_file_path: str, tools_json_str
     """
     从一个CSV文件批量创建子任务。
     此工具读取CSV文件的每一行，使用行数据填充goal模板，然后为每一行创建一个新的子任务。
+    可以使用 get_task_result 工具获取每个子任务的运行结果。
 
     Args:
         goal_template (str): 一个包含占位符的字符串模板。占位符的格式应为 `{column_name}`，
@@ -168,6 +169,8 @@ def create_tasks_from_csv(goal_template: str, csv_file_path: str, tools_json_str
 
     try:
         # 2. 读取并处理CSV文件
+        # 增加CSV字段大小限制以处理大字段
+        csv.field_size_limit(min(2**31-1, 2**20))  # 设置为1MB或系统最大值
         with open(csv_file_path, mode='r', encoding='utf-8') as csvfile:
             # 使用 DictReader 可以方便地通过列名访问数据
             reader = csv.DictReader(csvfile)

beswarm/tools/worker.py

@@ -1,7 +1,7 @@
 from datetime import datetime
 from typing import List, Dict, Union
 
-from ..core import mcp_manager, broker, task_manager
+from ..core import mcp_manager, broker, task_manager, kgm
 from ..agents.planact import BrokerWorker
 from ..agents.chatgroup import ChatGroupWorker
 from ..aient.src.aient.plugins import register_tool
@@ -10,7 +10,7 @@ from ..aient.src.aient.plugins import register_tool
 @register_tool()
 async def worker(goal: str, tools: List[Union[str, Dict]], work_dir: str, cache_messages: Union[bool, List[Dict]] = None):
     start_time = datetime.now()
-    worker_instance = BrokerWorker(goal, tools, work_dir, cache_messages, broker, mcp_manager, task_manager)
+    worker_instance = BrokerWorker(goal, tools, work_dir, cache_messages, broker, mcp_manager, task_manager, kgm)
     result = await worker_instance.run()
     end_time = datetime.now()
     print(f"\n任务开始时间: {start_time.strftime('%Y-%m-%d %H:%M:%S')}")
@@ -21,7 +21,7 @@ async def worker(goal: str, tools: List[Union[str, Dict]], work_dir: str, cache_
 @register_tool()
 async def worker_gen(goal: str, tools: List[Union[str, Dict]], work_dir: str, cache_messages: Union[bool, List[Dict]] = None):
     start_time = datetime.now()
-    worker_instance = BrokerWorker(goal, tools, work_dir, cache_messages, broker, mcp_manager, task_manager)
+    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
     end_time = datetime.now()
@@ -32,7 +32,7 @@ async def worker_gen(goal: str, tools: List[Union[str, Dict]], work_dir: str, ca
 @register_tool()
 async def chatgroup(tools: List[Union[str, Dict]], work_dir: str, cache_messages: Union[bool, List[Dict]] = None):
     start_time = datetime.now()
-    worker_instance = ChatGroupWorker(tools, work_dir, cache_messages, broker, mcp_manager, task_manager)
+    worker_instance = ChatGroupWorker(tools, work_dir, cache_messages, broker, mcp_manager, task_manager, kgm)
     result = await worker_instance.run()
     end_time = datetime.now()
     print(f"\n任务开始时间: {start_time.strftime('%Y-%m-%d %H:%M:%S')}")

{beswarm-0.2.43.dist-info → beswarm-0.2.45.dist-info}/METADATA

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

{beswarm-0.2.43.dist-info → beswarm-0.2.45.dist-info}/RECORD

@@ -1,11 +1,12 @@
 beswarm/__init__.py,sha256=HZjUOJtZR5QhMuDbq-wukQQn1VrBusNWai_ysGo-VVI,20
 beswarm/broker.py,sha256=RtnQZVbhf25acUHahNBiaS5FGxcrj0rhBhkon9gFY_M,9873
-beswarm/core.py,sha256=_J_LPc0HZFQTLGHy021OMEh2WbsjL_hc6jxd2engpyY,216
+beswarm/core.py,sha256=htssaaeIBZ_yOqvX9VtANoVWaZHt_7oWcxyDI1z0paQ,310
+beswarm/knowledge_graph.py,sha256=xbSl-GPPWNP8ZVQC95frn_AoDvRL0eR56mYwUVHXw2g,9174
 beswarm/prompt.py,sha256=n0a1a6NThIxAYSkisg1sEKjvz2w0tozpKL4BIplaAkI,32286
-beswarm/taskmanager.py,sha256=_v1lNHu9WpKK2ial22hvEVPzMJBUiaKTSUbXOxpEYlU,12119
+beswarm/taskmanager.py,sha256=xczsnehCaOAI3SRE8wUxhzmt6i6G7ynndSWS_bAZn3k,12192
 beswarm/utils.py,sha256=S9jEtht0hTZbjZ2Hk24p4Ip41R69BogOkYS6fgPKY2Y,8219
-beswarm/agents/chatgroup.py,sha256=4QTjTjssFDvS2kI5A58TdN92BWjL-XaNzrcd36pTQGQ,11939
-beswarm/agents/planact.py,sha256=NyZankMtcivZBvIj_EYCTlQky_E1J8q6D0B2WK9RMlc,17468
+beswarm/agents/chatgroup.py,sha256=YHofra9kE0x7UrhqZjlP7PbWvinEew1BkjJ0XilzudM,12020
+beswarm/agents/planact.py,sha256=PGj69lhy6r3q07R8w16aMoWY1OjXVXothu7lDG-UC4Y,19140
 beswarm/aient/main.py,sha256=SiYAIgQlLJqYusnTVEJOx1WNkSJKMImhgn5aWjfroxg,3814
 beswarm/aient/setup.py,sha256=Mq1M05mT9_UYBK2jk5mP_sLxKQAolcuh8PYXexfj-XU,487
 beswarm/aient/src/aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
@@ -130,20 +131,21 @@ beswarm/queries/tree-sitter-languages/ruby-tags.scm,sha256=vIidsCeE2A0vdFN18yXKq
 beswarm/queries/tree-sitter-languages/rust-tags.scm,sha256=9ljM1nzhfPs_ZTRw7cr2P9ToOyhGcKkCoN4_HPXSWi4,1451
 beswarm/queries/tree-sitter-languages/scala-tags.scm,sha256=UxQjz80JIrrJ7Pm56uUnQyThfmQNvwk7aQzPNypB-Ao,1761
 beswarm/queries/tree-sitter-languages/typescript-tags.scm,sha256=OMdCeedPiA24ky82DpgTMKXK_l2ySTuF2zrQ2fJAi9E,1253
-beswarm/tools/__init__.py,sha256=frEg4AuVdO1m-xf9bdN-mgxh5f8O9UCGXUADahU5CcI,1594
+beswarm/tools/__init__.py,sha256=fhJSEaYcfgMTW_nzUqQiyX5RLBhE-gxs8u4PpFquOdg,1868
 beswarm/tools/click.py,sha256=7g6x1X7ffTInGWp7112KS-MAQ5-8wa1Ze2sIipUIbjc,20884
 beswarm/tools/completion.py,sha256=wHEJrdzjuTKQNQZhelSuPnK2YRsGbeUqZ0P6IgT3c10,605
 beswarm/tools/edit_file.py,sha256=iwWl7a8sTVq4vj0e1ny3H6UGcHfYnxALRGcLuk5hZS8,9155
+beswarm/tools/graph.py,sha256=-Lyhi-AsJNKQ5p9BJk6mzARd1-IDhdLyKmNAVN5PdUE,2703
 beswarm/tools/planner.py,sha256=lguBCS6kpwNPoXQvqH-WySabVubT82iyWOkJnjt6dXw,1265
 beswarm/tools/repomap.py,sha256=YsTPq5MXfn_Ds5begcvHDnY_Xp2d4jH-xmWqNMHnNHY,45239
 beswarm/tools/request_input.py,sha256=gXNAJPOJektMqxJVyzNTFOeMQ7xUkO-wWMYH-r2Rdwk,942
 beswarm/tools/screenshot.py,sha256=u6t8FCgW5YHJ_Oc4coo8e0F3wTusWE_-H8dFh1rBq9Q,1011
 beswarm/tools/search_arxiv.py,sha256=8naYRvmELQryMIcCCBHebIEzNJ8_YivXamM4fzGG3Dk,10754
 beswarm/tools/search_web.py,sha256=LhgXOSHL9fwxg5T2AAOV4TTJkcJVLogsfRJW2faPvDE,16147
-beswarm/tools/subtasks.py,sha256=KCXFzwvxiyX48TF8EAyzucO29PanSAaT5D14TD3GXqQ,9670
-beswarm/tools/worker.py,sha256=auoAEeg8U_Q59dlyM2nv9cAJIWG3PuR8oy7Bp2cM2S0,1993
+beswarm/tools/subtasks.py,sha256=zZmRCjDocbvrU8MvZvpDWpQCKNntDJ123ByBiS8pRko,9889
+beswarm/tools/worker.py,sha256=-tcLGrdLPZYxc1z66iQIB1VH7RWcR-5CgVWuDIQJBCQ,2013
 beswarm/tools/write_csv.py,sha256=-r5OghcvjCg00hY0YQbp6u31VIJLrgaqDIvczAFoqDE,1470
-beswarm-0.2.43.dist-info/METADATA,sha256=qoj6R5aqa7sqxtwaRwDtAvpU0qV453VQ56yis9GO4m4,3878
-beswarm-0.2.43.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.2.43.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.2.43.dist-info/RECORD,,
+beswarm-0.2.45.dist-info/METADATA,sha256=4d7hy--9PFuYXlvmyfQlug8yurPyOx9otmIPJBBXd-U,3878
+beswarm-0.2.45.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.2.45.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.2.45.dist-info/RECORD,,