beswarm 0.1.75__tar.gz → 0.1.76__tar.gz

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

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

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

@@ -1,9 +1,10 @@
 import os
 import re
+import difflib
 from ..aient.src.aient.plugins import register_tool
 
 @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 +23,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。当完全精确匹配失败时，此参数生效。
 
 返回:
     编辑结果的状态信息，包括成功或错误信息。
@@ -93,27 +94,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:
@@ -173,7 +166,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.76}/beswarm.egg-info/PKG-INFO

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

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

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