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 +20 -0
- beswarm/tools/worker.py +1 -1
- beswarm/utils.py +11 -0
- {beswarm-0.1.5.dist-info → beswarm-0.1.7.dist-info}/METADATA +1 -1
- {beswarm-0.1.5.dist-info → beswarm-0.1.7.dist-info}/RECORD +7 -6
- {beswarm-0.1.5.dist-info → beswarm-0.1.7.dist-info}/WHEEL +0 -0
- {beswarm-0.1.5.dist-info → beswarm-0.1.7.dist-info}/top_level.txt +0 -0
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,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=
|
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=
|
57
|
-
beswarm-0.1.
|
58
|
-
beswarm-0.1.
|
59
|
-
beswarm-0.1.
|
60
|
-
beswarm-0.1.
|
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,,
|
File without changes
|
File without changes
|