beswarm 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

beswarm/tools/edit_file.py

@@ -0,0 +1,162 @@
+import os
+import re
+from aient.plugins import register_tool
+
+@register_tool()
+def edit_file(file_path, diff_content, match_precision=0.8):
+    """
+使用diff模式编辑代码文件，支持Git的diff格式和多块编辑。
+你应该清楚地表明编辑内容，同时最小化你写的未更改代码。
+你仍应倾向于尽可能少地重复原始文件的行来传达更改。
+但是，每个编辑都应包含足够的未更改行上下文，以消除你正在编辑的代码的歧义。
+
+参数:
+    file_path: 要编辑的文件路径
+    diff_content: diff内容，格式为：
+<<<<<<< SEARCH
+原始代码
+=======
+替换代码
+>>>>>>> REPLACE
+
+可以包含多个这样的块进行多处编辑
+    match_precision: 匹配精度 (0.0-1.0)，值越高要求匹配越精确，默认0.8
+
+返回:
+    编辑结果的状态信息，包括成功或错误信息
+    """
+    try:
+        # 检查文件是否存在
+        if not os.path.exists(file_path):
+            return f"错误: 文件 '{file_path}' 不存在"
+
+        # 检查是否为文件
+        if not os.path.isfile(file_path):
+            return f"错误: '{file_path}' 不是一个文件"
+
+        # 尝试读取文件的前几个字节来检测是否为二进制文件
+        with open(file_path, 'rb') as file:
+            data = file.read(1024)
+            # 快速检查是否可能是二进制文件
+            if b'\x00' in data:
+                return f"错误: 文件 '{file_path}' 可能是二进制文件，编码无法解析"
+
+        # 读取原文件内容
+        with open(file_path, 'r', encoding='utf-8') as file:
+            content = file.read()
+
+        # 解析diff内容中的搜索和替换块
+        diff_blocks = re.findall(r'<<<<<<< SEARCH\n(.*?)\n=======\n(.*?)\n>>>>>>> REPLACE', diff_content, re.DOTALL)
+
+        if not diff_blocks:
+            return f"错误: 无效的diff格式，未找到搜索和替换块"
+
+        # 记录修改次数和行数变化
+        edits_count = 0
+        total_original_lines = 0
+        total_new_lines = 0
+
+        # 应用每个diff块
+        new_content = content
+        for search_block, replace_block in diff_blocks:
+            # 检查搜索块是否为空
+            if not search_block.strip():
+                return f"错误: 搜索块不能为空"
+
+            if search_block in new_content:
+                # 直接替换完全匹配的块
+                new_content = new_content.replace(search_block, replace_block, 1)
+                edits_count += 1
+
+                # 计算行数
+                original_lines = search_block.count('\n') + 1
+                new_lines = replace_block.count('\n') + 1
+                total_original_lines += original_lines
+                total_new_lines += new_lines
+            else:
+                # 尝试进行更宽松的匹配
+                search_lines = search_block.splitlines()
+                content_lines = new_content.splitlines()
+
+                # 避免空搜索块
+                if len(search_lines) == 0:
+                    return f"错误: 搜索块不能为空"
+
+                # 尝试找到最佳匹配位置
+                best_match_index = -1
+                best_match_score = 0
+
+                for i in range(len(content_lines) - len(search_lines) + 1):
+                    # 计算当前位置的匹配分数
+                    match_score = 0
+                    total_chars = 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
+
+                    # 计算整体匹配分数
+                    if total_chars > 0:
+                        match_score = match_score / total_chars
+                    else:
+                        match_score = 0
+
+                    # 如果使用高精度匹配 (>0.95)，我们需要更严格地检查匹配结果
+                    if match_precision > 0.95:
+                        # 字符串长度差异也要考虑
+                        search_text = '\n'.join(search_lines)
+                        potential_match_text = '\n'.join(content_lines[i:i + len(search_lines)])
+                        length_diff_ratio = abs(len(search_text) - len(potential_match_text)) / max(len(search_text), 1)
+
+                        # 如果长度差异太大，降低匹配分数
+                        if length_diff_ratio > 0.1:  # 允许10%的长度差异
+                            match_score *= (1 - length_diff_ratio)
+
+                    if match_score > best_match_score:
+                        best_match_score = match_score
+                        best_match_index = i
+
+                # 使用匹配精度作为阈值
+                if best_match_score >= match_precision and best_match_index >= 0:
+                    # 提取找到的实际文本
+                    matched_lines = content_lines[best_match_index:best_match_index + len(search_lines)]
+                    actual_search_block = '\n'.join(matched_lines)
+
+                    # 替换找到的匹配项
+                    new_content = new_content.replace(actual_search_block, replace_block, 1)
+                    edits_count += 1
+
+                    # 计算行数
+                    original_lines = len(matched_lines)
+                    new_lines = replace_block.count('\n') + 1
+                    total_original_lines += original_lines
+                    total_new_lines += new_lines
+                else:
+                    return f"错误: 在文件中未找到足够匹配的代码块，最佳匹配分数为 {best_match_score:.2f}，但要求为 {match_precision:.2f}"
+
+        # 如果没有进行任何编辑，返回错误
+        if edits_count == 0:
+            return f"错误: 未能应用任何diff编辑"
+
+        # 写入修改后的内容
+        with open(file_path, 'w', encoding='utf-8') as file:
+            file.write(new_content)
+
+        return f"成功: 文件 '{file_path}' 已更新。应用了 {edits_count} 处编辑，替换了 {total_original_lines} 行代码为 {total_new_lines} 行。"
+
+    except PermissionError:
+        return f"错误: 没有权限修改文件 '{file_path}'"
+    except UnicodeDecodeError:
+        return f"错误: 文件 '{file_path}' 不是文本文件或编码不是UTF-8，无法进行编码解析"
+    except Exception as e:
+        return f"编辑文件时发生错误: {e}"

beswarm/tools/planner.py

@@ -0,0 +1,33 @@
+import os
+from aient.plugins import register_tool, get_function_call_list
+
+from aient.models import chatgpt
+from aient.prompt import planner_system_prompt
+
+@register_tool()
+async def planner(goal, tools, work_dir):
+    tools_json = [value for _, value in get_function_call_list(tools).items()]
+    instruction_agent_config = {
+        "api_key": os.getenv("API_KEY"),
+        "api_url": os.getenv("BASE_URL"),
+        "engine": os.getenv("MODEL"),
+        "system_prompt": planner_system_prompt.format(worker_tool_use_rules=tools_json, workspace_path=work_dir),
+        "print_log": False,
+        "max_tokens": 4000,
+        "temperature": 0.7,
+        "use_plugins": True,
+    }
+
+    # 指令agent初始化
+    instruction_agent = chatgpt(**instruction_agent_config)
+
+    instruction_prompt = f"""
+任务目标: {goal}
+
+请生成一个详细的任务计划，包括每个步骤的详细描述、所需的工具和预期结果。然后调用worker工具来完成每个步骤。
+    """
+    # 让指令agent分析对话历史并生成新指令
+    next_instruction = await instruction_agent.ask_async(instruction_prompt)
+    print("\n🤖 指令智能体生成的下一步指令:", next_instruction)
+
+    return "任务已完成"