beswarm 0.1.14__py3-none-any.whl → 0.1.16__py3-none-any.whl

beswarm/aient/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.0.74",
+    version="1.0.76",
     description="Aient: The Awakening of Agent.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

beswarm/aient/src/aient/core/request.py

@@ -300,7 +300,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
     original_model = model_dict[request.model]
     search_tool = None
 
-    pro_models = ["gemini-2.5-pro", "gemini-2.0-pro", "gemini-exp"]
+    pro_models = ["gemini-2.5", "gemini-2.0", "gemini-exp"]
     if any(pro_model in original_model for pro_model in pro_models):
         location = gemini2
         search_tool = {"googleSearch": {}}
@@ -946,6 +946,7 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
 
     if "api.x.ai" in url:
         payload.pop("stream_options", None)
+        payload.pop("presencePenalty", None)
 
     if "grok-3-mini" in original_model:
         if request.model.endswith("high"):

beswarm/aient/src/aient/prompt/agent.py

@@ -62,7 +62,8 @@ When making code changes, NEVER output code to the USER, unless requested. Inste
 </calling_external_apis>
 
 <user_info>
-The user's OS version is {os_name} {os_version}. The absolute path of the user's workspace is {workspace_path} which is also the project root directory. 请在指令中使用绝对路径。所有操作必须基于工作目录。禁止在工作目录之外进行任何操作。禁止默认你当前就在工作目录。The user's shell is {shell}.
+The user's OS version is {os_version}. The absolute path of the user's workspace is {workspace_path} which is also the project root directory. The user's shell is {shell}.
+请在指令中使用绝对路径。所有操作必须基于工作目录。禁止在工作目录之外进行任何操作。你当前运行目录不一定就是工作目录。禁止默认你当前就在工作目录。
 </user_info>
 
 <instructions for tool use>
@@ -118,7 +119,6 @@ You can use tools as follows:
 """
 
 instruction_system_prompt = """
-
 你是一个指令生成器，负责指导另一个智能体完成任务。
 你需要分析工作智能体的对话历史，并生成下一步指令。
 根据任务目标和当前进度，提供清晰明确的指令。
@@ -127,8 +127,9 @@ instruction_system_prompt = """
 
 你需要称呼工作智能体为“你”，指令禁止使用疑问句，必须使用祈使句。
 所有回复必须使用中文。
-你的工作目录为：{workspace_path}，请在指令中使用绝对路径。所有操作必须基于工作目录。禁止在工作目录之外进行任何操作。禁止默认你当前就在工作目录。
-
+运行工作智能体的系统信息：{os_version}
+你的工作目录为：{workspace_path}，请在指令中使用绝对路径。所有操作必须基于工作目录。
+禁止在工作目录之外进行任何操作。你当前运行目录不一定就是工作目录。禁止默认你当前就在工作目录。
 
 你的输出必须符合以下步骤：
 

beswarm/tools/repomap.py

@@ -1286,4 +1286,7 @@ if __name__ == "__main__":
 
     # print(get_code_repo_map("."))
     # print(get_code_repo_map("/Users/yanyuming/Downloads/GitHub/uni-api"))
-    print(get_code_repo_map("/Users/yanyuming/Downloads/GitHub/text-to-motion"))
+    # print(get_code_repo_map("/Users/yanyuming/Downloads/GitHub/text-to-motion"))
+    print(get_code_repo_map("/Users/yanyuming/Downloads/GitHub/beswarm/work/secretary/secretary"))
+
+# python -m beswarm.tools.repomap

beswarm/tools/worker.py

@@ -1,5 +1,6 @@
 import os
 import copy
+import platform
 from ..aient.src.aient.plugins import register_tool
 
 from ..aient.src.aient.models import chatgpt
@@ -28,8 +29,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
 
     tools_json = [value for _, value in get_function_call_list(tools).items()]
     work_agent_system_prompt = system_prompt.format(
-        os_name=os.name,
-        os_version=os.uname().version,
+        os_version=platform.platform(),
         workspace_path=work_dir,
         shell=os.getenv('SHELL', 'Unknown'),
         tools_list=tools_json
@@ -52,7 +52,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
         "api_key": os.getenv("API_KEY"),
         "api_url": os.getenv("BASE_URL"),
         "engine": os.getenv("MODEL"),
-        "system_prompt": instruction_system_prompt.format(tools_list=tools_json, workspace_path=work_dir),
+        "system_prompt": instruction_system_prompt.format(os_version=platform.platform(), tools_list=tools_json, workspace_path=work_dir),
         "print_log": False,
         # "max_tokens": 4000,
         "temperature": 0.7,

{beswarm-0.1.14.dist-info → beswarm-0.1.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.14
+Version: 0.1.16
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
@@ -68,3 +68,36 @@ cd ~/Downloads/GitHub/beswarm && docker run --rm \
 -v ./work:/app/work beswarm \
 --goal "分析这个仓库 https://github.com/cloneofsimo/minRF"
 ```
+
+main.py
+
+```python
+import os
+import asyncio
+import nest_asyncio
+nest_asyncio.apply()
+
+from beswarm.tools import (
+    worker,
+    get_code_repo_map,
+    search_arxiv,
+    read_file,
+    list_directory,
+    excute_command,
+    write_to_file,
+    download_read_arxiv_pdf,
+)
+
+os.environ['API_KEY'] = ''
+os.environ['BASE_URL'] = 'https://api.xxx.xyz/v1/chat/completions'
+os.environ['MODEL'] = 'gemini-2.5-pro'
+
+# 设定任务目标
+goal = """
+
+"""
+work_dir = '/home/ray/default'
+
+tools = [read_file, list_directory, write_to_file, excute_command, search_arxiv, download_read_arxiv_pdf, get_code_repo_map]
+asyncio.run(worker(goal, tools, work_dir))
+```

{beswarm-0.1.14.dist-info → beswarm-0.1.16.dist-info}/RECORD

@@ -1,12 +1,12 @@
 beswarm/__init__.py,sha256=HZjUOJtZR5QhMuDbq-wukQQn1VrBusNWai_ysGo-VVI,20
 beswarm/utils.py,sha256=AdDCcqAIIKQEMl7PfryVgeT9G5sHe7QNsZnrvmTGA8E,283
 beswarm/aient/main.py,sha256=SiYAIgQlLJqYusnTVEJOx1WNkSJKMImhgn5aWjfroxg,3814
-beswarm/aient/setup.py,sha256=2ZLf8IjbEEYKStTnjsKCdVu7Iv4PbDBL9HHYxRRkuL0,487
+beswarm/aient/setup.py,sha256=UOCyTbJunqBeBMa1Dma79K6Dk24vR0bELQSz3IhTA4g,487
 beswarm/aient/src/aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
 beswarm/aient/src/aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
 beswarm/aient/src/aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 beswarm/aient/src/aient/core/models.py,sha256=_1wYZg_n9kb2A3C8xCboyqleH2iHc9scwOvtx9DPeok,7582
-beswarm/aient/src/aient/core/request.py,sha256=84GysWqkt5PNTYQ4VLK1z2GK66WIFs-691oHgHDBH1A,60773
+beswarm/aient/src/aient/core/request.py,sha256=PF3hBBzl5RZTda2bAT_Vlp7DiXFhYMzk2Egp7irKVPk,60810
 beswarm/aient/src/aient/core/response.py,sha256=CSAOQmybBu3i2Yq2YUyi91dZWTN1tmtJEBTI8RFwj_s,30594
 beswarm/aient/src/aient/core/utils.py,sha256=R4oKj1oVsdfePdQReaW-ZU4VRTCetvdCKPz_OOTOU_w,26072
 beswarm/aient/src/aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
@@ -34,7 +34,7 @@ beswarm/aient/src/aient/plugins/run_python.py,sha256=dgcUwBunMuDkaSKR5bToudVzSdr
 beswarm/aient/src/aient/plugins/websearch.py,sha256=yiBzqXK5X220ibR-zko3VDsn4QOnLu1k6E2YOygCeTQ,15185
 beswarm/aient/src/aient/plugins/write_file.py,sha256=qYbNvcDTGs00pWYb0UOSYSeN4kw5_DMDxjaSlxtlAkU,1732
 beswarm/aient/src/aient/prompt/__init__.py,sha256=GBtn6-JDT8KHFCcuPpfSNE_aGddg5p4FEyMCy4BfwGs,20
-beswarm/aient/src/aient/prompt/agent.py,sha256=ssZREegzmkWNW20dxzGO92J8Y5v7yjRTR9znTPDjo5Q,23681
+beswarm/aient/src/aient/prompt/agent.py,sha256=WLs0KiNZs29FJ5w7N3kQZYLVEeS7K8vxIOUw06LcXVE,23825
 beswarm/aient/src/aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beswarm/aient/src/aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
 beswarm/aient/src/aient/utils/scripts.py,sha256=n0jR5eXCBIK12W4bIx-xU1FVl1hZ4zDC7hq_BWQHYJU,27537
@@ -121,11 +121,11 @@ beswarm/queries/tree-sitter-languages/typescript-tags.scm,sha256=OMdCeedPiA24ky8
 beswarm/tools/__init__.py,sha256=1QLRX95kJSdCKW8OvAJCg67qlzsU9juU60ciZvUGq00,761
 beswarm/tools/edit_file.py,sha256=qKOVanfh_79eavRrqqyFirRo07sZ897jUL2PD4GRKJk,6965
 beswarm/tools/planner.py,sha256=lguBCS6kpwNPoXQvqH-WySabVubT82iyWOkJnjt6dXw,1265
-beswarm/tools/repomap.py,sha256=DhZewVr3fsjau3S9uCpx723uFliteb-QvSpeArDgnU8,40692
+beswarm/tools/repomap.py,sha256=CwvwoN5Swr42EzrORTTeV8MMb7mPviy4a4b0fxBu50k,40828
 beswarm/tools/search_arxiv.py,sha256=9slwBemXjEqrd7-YgVmyMijPXlkhZCybEDRVhWVQ9B0,7937
 beswarm/tools/think.py,sha256=WLw-7jNIsnS6n8MMSYUin_f-BGLENFmnKM2LISEp0co,1760
-beswarm/tools/worker.py,sha256=kw9-7--IgS3B6PE0XJibO2BPc_-8jVCKb7fXRTJGVzA,5080
-beswarm-0.1.14.dist-info/METADATA,sha256=C6iU66_hclvt7ddbJiBoKX1XsU0nkIxtl3ZtNXWcVWU,2229
-beswarm-0.1.14.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-beswarm-0.1.14.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.1.14.dist-info/RECORD,,
+beswarm/tools/worker.py,sha256=3YnSyXaboNJ9j052NMLoIEOtxWL-uJ6sllonm7jV02E,5104
+beswarm-0.1.16.dist-info/METADATA,sha256=u339_6M84jBXXi2CpKLbbP39swSB8s41LQJ9OsBxnZQ,2878
+beswarm-0.1.16.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+beswarm-0.1.16.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.1.16.dist-info/RECORD,,