beswarm 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
beswarm/tools/__init__.py CHANGED
@@ -3,6 +3,17 @@ from .edit_file import edit_file
3
3
  from .worker import worker
4
4
  from .search_arxiv import search_arxiv
5
5
  from .repomap import get_code_repo_map
6
+ #显式导入 aient.plugins 中的所需内容
7
+ from aient.plugins import (
8
+ excute_command,
9
+ get_time,
10
+ generate_image,
11
+ list_directory,
12
+ read_file,
13
+ run_python_script,
14
+ get_search_results,
15
+ write_to_file,
16
+ )
6
17
 
7
18
  __all__ = [
8
19
  "think",
@@ -10,4 +21,13 @@ __all__ = [
10
21
  "worker",
11
22
  "search_arxiv",
12
23
  "get_code_repo_map",
24
+ # aient.plugins
25
+ "excute_command",
26
+ "get_time",
27
+ "generate_image",
28
+ "list_directory",
29
+ "read_file",
30
+ "run_python_script",
31
+ "get_search_results",
32
+ "write_to_file",
13
33
  ]
beswarm/tools/worker.py CHANGED
@@ -6,7 +6,7 @@ from aient.models import chatgpt
6
6
  from aient.plugins import get_function_call_list
7
7
  from aient.prompt import system_prompt, instruction_system_prompt
8
8
 
9
- from utils import extract_xml_content
9
+ from ..utils import extract_xml_content
10
10
 
11
11
  @register_tool()
12
12
  async def worker(goal, tools, work_dir):
beswarm/utils.py ADDED
@@ -0,0 +1,11 @@
1
+ import re
2
+
3
+ def extract_xml_content(text, xml_tag):
4
+ result = ''
5
+ pattern = rf'^<{xml_tag}>$\n*([\D\d\s]+?)\n*^<\/{xml_tag}>$'
6
+ match = re.search(pattern, text, re.MULTILINE)
7
+ if match:
8
+ result = match.group(1)
9
+ if not result:
10
+ return ''
11
+ return result
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: beswarm
3
- Version: 0.1.5
3
+ Version: 0.1.7
4
4
  Summary: MAS
5
5
  Requires-Python: >=3.11
6
6
  Description-Content-Type: text/markdown
@@ -1,4 +1,5 @@
1
1
  beswarm/__init__.py,sha256=HZjUOJtZR5QhMuDbq-wukQQn1VrBusNWai_ysGo-VVI,20
2
+ beswarm/utils.py,sha256=AdDCcqAIIKQEMl7PfryVgeT9G5sHe7QNsZnrvmTGA8E,283
2
3
  beswarm/queries/tree-sitter-language-pack/README.md,sha256=ivZSEuWqYfUVLZl2AZZGRlm0bQsaG-VTBKBwACyM07k,291
3
4
  beswarm/queries/tree-sitter-language-pack/arduino-tags.scm,sha256=HbgdothT9Jjk56COXTtUkVAdZ14rZNnqzLbWVLeRs5U,177
4
5
  beswarm/queries/tree-sitter-language-pack/c-tags.scm,sha256=EIz45o5hBh8yEuck5ZR_4IpcGyWSeNrzxFmtkKZGt2k,461
@@ -47,14 +48,14 @@ beswarm/queries/tree-sitter-languages/ruby-tags.scm,sha256=vIidsCeE2A0vdFN18yXKq
47
48
  beswarm/queries/tree-sitter-languages/rust-tags.scm,sha256=9ljM1nzhfPs_ZTRw7cr2P9ToOyhGcKkCoN4_HPXSWi4,1451
48
49
  beswarm/queries/tree-sitter-languages/scala-tags.scm,sha256=UxQjz80JIrrJ7Pm56uUnQyThfmQNvwk7aQzPNypB-Ao,1761
49
50
  beswarm/queries/tree-sitter-languages/typescript-tags.scm,sha256=OMdCeedPiA24ky82DpgTMKXK_l2ySTuF2zrQ2fJAi9E,1253
50
- beswarm/tools/__init__.py,sha256=4B4VowEmOhXEA14SUbcmijs7oM01NFKbLgiidCHQ8J4,266
51
+ beswarm/tools/__init__.py,sha256=ZDM8V2QLJeiZiwZTD3hqAMU4mpY71rg0WHWuSOiSbgY,689
51
52
  beswarm/tools/edit_file.py,sha256=liGc7S3siGTcpj5Ny2HsHVkp6V1kb4sVSvNO2VrrToE,6953
52
53
  beswarm/tools/planner.py,sha256=4uWZHuXj37-YIuz8MgHAlg_TO7EuHrfJFMn5eSK_v9M,1229
53
54
  beswarm/tools/repomap.py,sha256=g-DNw4loTKau_F2cFlH6CEmua8JZomOtvaFgqdC4FA4,40680
54
55
  beswarm/tools/search_arxiv.py,sha256=NF21a1VB7u6cWFv7RviLN-HEEe-zzuOVJob0d9nP5w4,7925
55
56
  beswarm/tools/think.py,sha256=E1ey-cwMtOdOiNVJqKpcK1zdloAR3fByVAbDDmtDxYU,1748
56
- beswarm/tools/worker.py,sha256=Psev2BYQNORDxQ00X546eqIYnjHuSg7ypDVk4vKuAtM,4738
57
- beswarm-0.1.5.dist-info/METADATA,sha256=-JJ4Owv8KllxnYpTFlxWVFBNUyWNtiT3B_g03RpN8Ds,1650
58
- beswarm-0.1.5.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
59
- beswarm-0.1.5.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
60
- beswarm-0.1.5.dist-info/RECORD,,
57
+ beswarm/tools/worker.py,sha256=kwxpwm-CkAYki-jH532YE80OrV1fer_N-xL3fgj8m20,4740
58
+ beswarm-0.1.7.dist-info/METADATA,sha256=Kb5pN3FVsOkpO4qyL1M1gb2Nk38D3CLjqX2ZQpbdRjw,1650
59
+ beswarm-0.1.7.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
60
+ beswarm-0.1.7.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
61
+ beswarm-0.1.7.dist-info/RECORD,,