beswarm 0.2.78__py3-none-any.whl → 0.2.79__py3-none-any.whl

beswarm/core.py

@@ -12,6 +12,7 @@ broker = MessageBroker()
 mcp_manager = MCPManager()
 kgm = KnowledgeGraphManager(broker=broker)
 current_task_manager = contextvars.ContextVar('current_task_manager')
+current_work_dir = contextvars.ContextVar('current_work_dir', default=None)
 
 def get_task_manager():
     """Creates a new, isolated TaskManager instance."""

beswarm/tools/search_web.py

@@ -1,10 +1,14 @@
 import re
 import os
+import csv
+import time
 import json
 import httpx
 import threading
+from pathlib import Path
 
 from ..aient.aient.plugins import register_tool, get_url_content # Assuming a similar plugin structure
+from ..core import current_work_dir
 
 class ThreadWithReturnValue(threading.Thread):
     def run(self):
@@ -19,6 +23,7 @@ class ThreadWithReturnValue(threading.Thread):
 async def search_web(query: str):
     """
 获取 Google 搜索结果。
+搜索结果将保存在 csv 文件到 .beswarm/cache 目录下
 
 参数:
     query (str): 要在 Google 上搜索的查询字符串。
@@ -289,13 +294,45 @@ async def search_web(query: str):
     # with open(output_filename, "w", encoding="utf-8") as f:
     #     json.dump(final_web_content, f, indent=2, ensure_ascii=False)
     # print(f"Filtered web content saved to {output_filename}")
-    final_result = ""
-    for item in final_web_content:
-        final_result += item["content"]
-        final_result += "\n\n"
-    if not final_result:
-        return "No web content"
-    return final_result
+
+    if final_web_content:
+        work_dir = current_work_dir.get()
+        if work_dir:
+            output_path = Path(work_dir) / ".beswarm" / "cache"
+        else:
+            output_path = Path(".")
+
+        output_path.mkdir(parents=True, exist_ok=True)
+
+        # 使用时间戳生成唯一的文件名
+        timestamp = time.strftime("%Y%m%d-%H%M%S")
+        csv_filename = f"web_content_{timestamp}.csv"
+        csv_filepath = output_path / csv_filename
+
+        with open(csv_filepath, "w", newline="", encoding="utf-8") as csvfile:
+            # 根据字典中的键定义字段名
+            fieldnames = ["query", "url", "content"]
+            writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
+
+            # 写入标题行
+            writer.writeheader()
+
+            # 写入数据行
+            for item in final_web_content:
+                item['query'] = query
+                if 'content' in item and isinstance(item['content'], str):
+                    item['content'] = item['content'].encode('unicode_escape').decode('utf-8')
+                writer.writerow(item)
+        return f"已将筛选后的网页内容保存到 {csv_filepath.absolute()}"
+    return "No web content"
+
+
+    # for item in final_web_content:
+    #     final_result += item["content"]
+    #     final_result += "\n\n"
+    # if not final_result:
+    #     return "No web content"
+    # return final_result
 
 import difflib
 

beswarm/tools/worker.py

@@ -1,7 +1,7 @@
 from datetime import datetime
 from typing import List, Dict, Union
 
-from ..core import mcp_manager, broker, kgm, get_task_manager, current_task_manager
+from ..core import mcp_manager, broker, kgm, get_task_manager, current_task_manager, current_work_dir
 from ..agents.planact import BrokerWorker
 from ..agents.chatgroup import ChatGroupWorker
 from ..aient.aient.plugins import register_tool
@@ -12,6 +12,7 @@ async def worker(goal: str, tools: List[Union[str, Dict]], work_dir: str, cache_
     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)
     result = await worker_instance.run()
     end_time = datetime.now()

{beswarm-0.2.78.dist-info → beswarm-0.2.79.dist-info}/METADATA

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

{beswarm-0.2.78.dist-info → beswarm-0.2.79.dist-info}/RECORD

@@ -1,6 +1,6 @@
 beswarm/__init__.py,sha256=HZjUOJtZR5QhMuDbq-wukQQn1VrBusNWai_ysGo-VVI,20
 beswarm/broker.py,sha256=64Y-djrKYaZfBQ8obwHOmr921QgZeu9BtScZWaYLfDo,9887
-beswarm/core.py,sha256=pB1vKVmmw6pPRiLpp8MoU411ahPfDpxI6il0FJP63g4,477
+beswarm/core.py,sha256=jKStpTTOu6Ojond_i-okTZLrvSAJ4yUoTZwtDfFTiRs,553
 beswarm/knowledge_graph.py,sha256=oiOMknAJzGrOHc2AyQgvrCcZAkGLhFnsnvSBdfFBWMw,14831
 beswarm/prompt.py,sha256=qlH-yYCGv9j9-5grnMXl0B53CDZgfTpe4ncwBwgWg7o,32816
 beswarm/taskmanager.py,sha256=vMmcoZ4FlNvjEliRkv3AniPji50NcY4Q1_2HETzR0DU,12226
@@ -112,11 +112,11 @@ beswarm/tools/repomap.py,sha256=w98aHmjNjtvcUVc5maWORqzKqDy2KVGLooOe__uJVCU,4523
 beswarm/tools/request_input.py,sha256=3n2UW8m8Q7dxGhd7L7hzSJ1kk4ekMbtdtNZZT3dJf20,938
 beswarm/tools/screenshot.py,sha256=hyL6F8_k9Y03Nb_X18cY-klCpWWdkqyC-iGXfKX-7jc,1007
 beswarm/tools/search_arxiv.py,sha256=NLiJV1B7Um6EuZXLxnL950d837_he2LGG7qaGACSgwg,10750
-beswarm/tools/search_web.py,sha256=NYrb5KL_WUGPm-fOKT8Cyjon04lxBU-gaLdrVjeYgGo,16143
+beswarm/tools/search_web.py,sha256=juFbife765vRZXpTHOV_6ki4h6NpSmZbBcWJ8Fp-uiE,17497
 beswarm/tools/subtasks.py,sha256=UI9QznALmZNbNOSoR_2AdlnDAkZBuKMq0GifnBCgpLo,10458
-beswarm/tools/worker.py,sha256=o7IR6dXUic_GVDsI6b9ilY0DfsUSDTMxLrPNXbSLPC0,2278
+beswarm/tools/worker.py,sha256=mQ1qdrQ8MgL99byAbTvxfEByFFGN9mty3UHqHjARMQ8,2331
 beswarm/tools/write_csv.py,sha256=u0Hq18Ksfheb52MVtyLNCnSDHibITpsYBPs2ub7USYA,1466
-beswarm-0.2.78.dist-info/METADATA,sha256=PmFd0gPieB7lqZZaWCqvG8QeiCCUMD0Osly8cnWtxvI,3878
-beswarm-0.2.78.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.2.78.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.2.78.dist-info/RECORD,,
+beswarm-0.2.79.dist-info/METADATA,sha256=owpFd3kO5L7n5-r1RpUs25emR-2kpRJ4SIasXxExmbQ,3878
+beswarm-0.2.79.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.2.79.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.2.79.dist-info/RECORD,,