beswarm 0.1.75__tar.gz → 0.1.77__tar.gz

{beswarm-0.1.75 → beswarm-0.1.77}/PKG-INFO

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

{beswarm-0.1.75 → beswarm-0.1.77}/beswarm/aient/setup.py

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

{beswarm-0.1.75 → beswarm-0.1.77}/beswarm/aient/src/aient/plugins/excute_command.py

@@ -1,10 +1,9 @@
 import subprocess
 from .registry import register_tool
-from ..utils.scripts import sandbox
+from ..utils.scripts import sandbox, unescape_html
 
 import re
 import os
-import html
 import select
 
 # 检查是否在 Unix-like 系统上 (pty 模块主要用于 Unix)
@@ -39,18 +38,6 @@ def compare_line(line: str) -> bool:
     # print(f"similarity: {similarity}")
     return similarity > 0.89
 
-def unescape_html(input_string: str) -> str:
-  """
-  将字符串中的 HTML 实体（例如 &）转换回其原始字符（例如 &）。
-
-  Args:
-    input_string: 包含 HTML 实体的输入字符串。
-
-  Returns:
-    转换后的字符串。
-  """
-  return html.unescape(input_string)
-
 def get_python_executable(command: str) -> str:
     """
     获取 Python 可执行文件的路径。

{beswarm-0.1.75 → beswarm-0.1.77}/beswarm/aient/src/aient/plugins/write_file.py

@@ -1,19 +1,7 @@
 from .registry import register_tool
+from ..utils.scripts import unescape_html
 
 import os
-import html
-
-def unescape_html(input_string: str) -> str:
-  """
-  将字符串中的 HTML 实体（例如 &）转换回其原始字符（例如 &）。
-
-  Args:
-    input_string: 包含 HTML 实体的输入字符串。
-
-  Returns:
-    转换后的字符串。
-  """
-  return html.unescape(input_string)
 
 @register_tool()
 def write_to_file(path, content, mode='w'):

{beswarm-0.1.75 → beswarm-0.1.77}/beswarm/aient/src/aient/utils/scripts.py

@@ -202,6 +202,19 @@ def parse_tools_from_cursor_prompt(text):
             print(f"解析 JSON 时出错: {e}")
     return []
 
+import html
+def unescape_html(input_string: str) -> str:
+  """
+  将字符串中的 HTML 实体（例如 &）转换回其原始字符（例如 &）。
+
+  Args:
+    input_string: 包含 HTML 实体的输入字符串。
+
+  Returns:
+    转换后的字符串。
+  """
+  return html.unescape(input_string)
+
 import sys
 import shlex
 import builtins

{beswarm-0.1.75 → beswarm-0.1.77}/beswarm/tools/edit_file.py

@@ -1,9 +1,11 @@
 import os
 import re
+import difflib
 from ..aient.src.aient.plugins import register_tool
+from ..aient.src.aient.utils.scripts import unescape_html
 
 @register_tool()
-def edit_file(file_path, diff_content, match_precision=0.8):
+def edit_file(file_path, diff_content, match_precision=0.9):
     """
 使用diff模式编辑代码文件。**重要：此函数每次调用只能对文件进行一处修改。**
 
@@ -22,7 +24,7 @@ def edit_file(file_path, diff_content, match_precision=0.8):
 参数:
     file_path: 要编辑的文件路径。
     diff_content: 包含单次修改信息的diff字符串，格式必须符合上述要求。
-    match_precision: 匹配精度 (0.0-1.0)，值越高要求匹配越精确，默认0.8。当完全精确匹配失败时，此参数生效。
+    match_precision: 匹配精度 (0.0-1.0)，值越高要求匹配越精确，默认为0.9。当完全精确匹配失败时，此参数生效。
 
 返回:
     编辑结果的状态信息，包括成功或错误信息。
@@ -63,6 +65,8 @@ def edit_file(file_path, diff_content, match_precision=0.8):
         # 应用每个diff块
         new_content = content
         for search_block, replace_block in diff_blocks:
+            search_block = unescape_html(search_block)
+            replace_block = unescape_html(replace_block)
             # 检查搜索块是否为空
             if not search_block.strip():
                 return f"错误: 搜索块不能为空"
@@ -93,27 +97,19 @@ def edit_file(file_path, diff_content, match_precision=0.8):
                 for i in range(len(content_lines) - len(search_lines) + 1):
                     # 计算当前位置的匹配分数
                     match_score = 0
-                    total_chars = 0
+                    lines_compared = 0
 
                     for j in range(len(search_lines)):
                         search_line = search_lines[j]
                         content_line = content_lines[i + j]
 
-                        # 计算两行之间的相似度（使用更精确的算法）
-                        # 1. 字符级别的比较
-                        common_chars = sum(c1 == c2 for c1, c2 in zip(search_line, content_line))
-                        total_line_chars = max(len(search_line), len(content_line))
-
-                        if total_line_chars > 0:
-                            line_match = common_chars / total_line_chars
-                            match_score += line_match
-                            total_chars += 1
+                        # 使用 difflib 计算行相似度，该算法对插入和删除更具鲁棒性
+                        line_match = difflib.SequenceMatcher(None, search_line, content_line).ratio()
+                        match_score += line_match
+                        lines_compared += 1
 
                     # 计算整体匹配分数
-                    if total_chars > 0:
-                        match_score = match_score / total_chars
-                    else:
-                        match_score = 0
+                    match_score = match_score / lines_compared
 
                     # 如果使用高精度匹配 (>0.95)，我们需要更严格地检查匹配结果
                     if match_precision > 0.95:
@@ -146,7 +142,14 @@ def edit_file(file_path, diff_content, match_precision=0.8):
                     total_original_lines += original_lines
                     total_new_lines += new_lines
                 else:
-                    return f"错误: 在文件中未找到足够匹配的代码块，最佳匹配分数为 {best_match_score:.2f}，但要求为 {match_precision:.2f}"
+                    error_message = f"错误: 在文件中未找到足够匹配的代码块，最佳匹配分数为 {best_match_score:.2f}，但要求为 {match_precision:.2f}。"
+                    if best_match_index != -1:
+                        start_index = max(0, best_match_index - len(search_lines))
+                        end_index = min(len(content_lines), best_match_index + 2 * len(search_lines))
+                        context_lines = content_lines[start_index:end_index]
+                        context_block = '\n'.join(context_lines)
+                        error_message += f"\n\n文件的最佳匹配范围原文如下 (上下文已扩展):\n<real_file_content>\n{context_block}\n</real_file_content>"
+                    return error_message
 
         # 如果没有进行任何编辑，返回错误
         if edits_count == 0:
@@ -173,7 +176,7 @@ def edit_file(file_path, diff_content, match_precision=0.8):
 if __name__ == "__main__":
     edit_str = """
 <<<<<<< SEARCH
-    parser.add_argument('--dataset', type=str, default='cifar10', choices=['cifar10', 'cifar100', 'imagenet'], help="Dataset to use.")
+    parser.add_argument('--dataset',type=str,default='cifar10', choices=['cifar10', 'cifar100', 'imagenet'],help="Dataset to use.")
 =======
     parser.add_argument('--dataset', type=str, default='cifar10', choices=['cifar10', 'cifar100', 'imagenet', 'tinyimagenet'], help="Dataset to use.")
 >>>>>>> REPLACE

{beswarm-0.1.75 → beswarm-0.1.77}/beswarm.egg-info/PKG-INFO

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

{beswarm-0.1.75 → beswarm-0.1.77}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.1.75"
+version = "0.1.77"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"