auto-coder 0.1.392__py3-none-any.whl → 0.1.395__py3-none-any.whl

autocoder/common/v2/agent/agentic_edit.py

@@ -10,6 +10,10 @@ from rich.panel import Panel
 from pydantic import SkipValidation
 from byzerllm.utils.types import SingleOutputMeta
 
+from byzerllm.utils import (        
+    format_str_jinja2
+)
+
 from autocoder.common import AutoCoderArgs, git_utils, SourceCodeList, SourceCode
 from autocoder.common.global_cancel import global_cancel
 from autocoder.common import detect_env
@@ -56,7 +60,7 @@ from autocoder.common.v2.agent.agentic_edit_tools import (  # Import specific re
     ReplaceInFileToolResolver, SearchFilesToolResolver, ListFilesToolResolver,
     ListCodeDefinitionNamesToolResolver, AskFollowupQuestionToolResolver,
     AttemptCompletionToolResolver, PlanModeRespondToolResolver, UseMcpToolResolver,
-    ListPackageInfoToolResolver
+    UseRAGToolResolver, ListPackageInfoToolResolver
 )
 from autocoder.common.llm_friendly_package import LLMFriendlyPackageManager
 from autocoder.common.rulefiles.autocoderrules_utils import get_rules,auto_select_rules,get_required_and_index_rules
@@ -69,7 +73,7 @@ from autocoder.common.v2.agent.agentic_edit_types import (AgenticEditRequest, To
                                                           ListFilesTool,
                                                           ListCodeDefinitionNamesTool, AskFollowupQuestionTool,
                                                           AttemptCompletionTool, PlanModeRespondTool, UseMcpTool,
-                                                          ListPackageInfoTool,
+                                                          UseRAGTool, ListPackageInfoTool,
                                                           TOOL_MODEL_MAP,
                                                           # Event Types
                                                           LLMOutputEvent, LLMThinkingEvent, ToolCallEvent,
@@ -80,7 +84,7 @@ from autocoder.common.v2.agent.agentic_edit_types import (AgenticEditRequest, To
                                                           ListFilesTool, SearchFilesTool, ListCodeDefinitionNamesTool,
                                                           AskFollowupQuestionTool, UseMcpTool, AttemptCompletionTool
                                                           )
-
+from autocoder.common.rag_manager import RAGManager
 from autocoder.rag.token_counter import count_tokens
 
 # Map Pydantic Tool Models to their Resolver Classes
@@ -96,7 +100,8 @@ TOOL_RESOLVER_MAP: Dict[Type[BaseTool], Type[BaseToolResolver]] = {
     AskFollowupQuestionTool: AskFollowupQuestionToolResolver,
     AttemptCompletionTool: AttemptCompletionToolResolver,  # Will stop the loop anyway
     PlanModeRespondTool: PlanModeRespondToolResolver,
-    UseMcpTool: UseMcpToolResolver
+    UseMcpTool: UseMcpToolResolver,
+    UseRAGTool: UseRAGToolResolver
 }
 
 
@@ -161,12 +166,25 @@ class AgenticEdit:
         except Exception as e:
             logger.error(f"Error getting MCP server info: {str(e)}")
 
+        # 初始化 RAG 管理器并获取服务器信息
+        self.rag_server_info = ""
+        try:
+            self.rag_manager = RAGManager(args)
+            if self.rag_manager.has_configs():
+                self.rag_server_info = self.rag_manager.get_config_info()
+                logger.info(f"RAG manager initialized with {len(self.rag_manager.get_all_configs())} configurations")
+            else:
+                logger.info("No RAG configurations found")
+        except Exception as e:
+            logger.error(f"Error initializing RAG manager: {str(e)}")
+            self.rag_manager = None
+
         # 变更跟踪信息
         # 格式: { file_path: FileChangeEntry(...) }
         self.file_changes: Dict[str, FileChangeEntry] = {}
 
     @byzerllm.prompt()
-    def generate_library_docs_prompt(self, libraries: List[str], docs_content: str) -> Dict[str, Any]:
+    def generate_library_docs_prompt(self, libraries_with_paths: List[Dict[str, str]], docs_content: str) -> Dict[str, Any]:
         """
         ====
 
@@ -187,8 +205,15 @@ class AgenticEdit:
         4. You need to understand the API or usage patterns of these libraries
         ====
         """
+        # 格式化库列表，包含名称和路径
+        libraries_list = []
+        for lib_info in libraries_with_paths:
+            name = lib_info.get('name', '')
+            path = lib_info.get('path', 'Path not found')
+            libraries_list.append(f"{name} (路径: {path})")
+        
         return {
-            "libraries_list": ", ".join(libraries),
+            "libraries_list": ", ".join(libraries_list),
             "combined_docs": docs_content
         }
 
@@ -442,6 +467,22 @@ class AgenticEdit:
         {{mcp_server_info}}
         {%endif%}
 
+        ## rag_tool
+        Description: Request to query the RAG server for information. Use this when you need to query the RAG server for information.
+        Parameters:
+        - server_name: (required) The url of the RAG server to use.
+        - query: (required) The query to pass to the tool.
+        Usage:
+        <use_rag_tool>
+        <server_name>xxx</server_name>
+        <query>Your query here</query>
+        </use_rag_tool>
+
+        {%if rag_server_info %}
+        ### RAG_SERVER_LIST
+        {{rag_server_info}}
+        {%endif%}
+
         # Tool Use Examples
 
         ## Example 1: Requesting to execute a command
@@ -517,7 +558,7 @@ class AgenticEdit:
         </use_mcp_tool>                
 
         # Tool Use Guidelines
-
+        0. **ALWAYS START WITH THOROUGH SEARCH AND EXPLORATION.** Before making any code changes, use search tools (list_files, grep commands) to fully understand the codebase structure, existing patterns, and dependencies. This prevents errors and ensures your changes align with project conventions.
         1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
         2. Choose the most appropriate tool based on the task and the tool descriptions provided. Assess if you need additional information to proceed, and which of the available tools would be most effective for gathering this information. For example using the list_files tool is more effective than running a command like \`ls\` in the terminal. It's critical that you think about each available tool and use the one that best fits the current step in the task.
         3. If multiple actions are needed, use one tool at a time per message to accomplish the task iteratively, with each tool use being informed by the result of the previous tool use. Do not assume the outcome of any tool use. Each step must be informed by the previous step's result.
@@ -538,9 +579,286 @@ class AgenticEdit:
         By waiting for and carefully considering the user's response after each tool use, you can react accordingly and make informed decisions about how to proceed with the task. This iterative process helps ensure the overall success and accuracy of your work.
 
         ====
+        SEARCHING FILES
+        
+        **THIS IS YOUR CORE METHODOLOGY** - The following search-first approach is not optional but mandatory for reliable code work. Every code task should follow this systematic exploration pattern.
+        This guide provides a systematic approach for AI agents and developers to effectively search, understand, and modify codebases. It emphasizes thorough pre-code investigation and post-code verification to ensure reliable and maintainable changes.
 
-        EDITING FILES
+        The methodology combines multiple search tools (grep, list_files, read_file) with structured workflows to minimize code errors, ensure comprehensive understanding, validate changes systematically, and follow established project patterns.
+
+        # list_files
+
+        ## Purpose
+
+        - Discover project structure and understand directory organization
+        - Get an overview of available files and folders before diving deeper
+
+        ## When to Use
+
+        - Initial project exploration to understand the codebase layout
+        - Identifying key directories like `src/`, `lib/`, `components/`, `utils/`
+        - Locating configuration files like `package.json`, `tsconfig.json`, `Makefile`
+        - Before using more targeted search tools
+
+        ## Advantages
+
+        - Provides quick project overview without overwhelming detail
+        - Helps plan targeted searches in specific directories
+        - Essential first step in understanding unfamiliar codebases
+
+        # grep (Shell Commands)
+
+        ## Purpose
+
+        - Find exact text matches and patterns across multiple files
+        - Perform precise searches with minimal output overhead
+        - Verify code changes and validate implementations
+
+        ## When to Use
+
+        - **Pre-code Context Gathering**: Locate symbols, functions, imports, and usage patterns
+        - **Post-code Verification**: Confirm changes were applied correctly and no stale references remain
+        - **Pattern Analysis**: Understand coding conventions and existing implementations
+
+        ## Key Command Patterns
+
+        **Pre-code Context Examples:**
+
+        <execute_command>
+        <command>grep -l "className" src/ | head -5</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>grep -rc "import.*React" src/ | grep -v ":0"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>grep -Rn "function.*MyFunction\|const.*MyFunction" . | head -10</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>grep -R --exclude-dir={node_modules,dist,build,.git} "TODO" .</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        **Post-code Verification Examples:**
+
+        <execute_command>
+        <command>ls -la newfile.js 2>/dev/null && echo "File created" || echo "File not found"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>grep -Rn "oldName" . || echo "✓ No stale references found"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>grep -c "newName" src/*.js | grep -v ":0" || echo "⚠ New references not found"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>grep -Rn "import.*newModule\|export.*newFunction" . | wc -l</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        ## Output Optimization Tips
+
+        - Use `-l` for filenames only
+        - Use `-c` for count only  
+        - Use `| head -N` to limit lines
+        - Use `| wc -l` for total count
+        - Use `2>/dev/null` to suppress errors
+        - Combine with `|| echo` for clear status messages
+
+        # search_files (Fallback)
+
+        ## Purpose
+
+        - Alternative search method when grep commands aren't available
+        - Semantic search capabilities for finding related code
+
+        ## When to Use
+
+        - When shell access is limited or grep is unavailable
+        - For broader, less precise searches across the codebase
+        - As a complement to grep for comprehensive code discovery
+
+        # read_file
+
+        ## Purpose
+
+        - Examine complete file contents in detail
+        - Understand context, patterns, and implementation details
+
+        ## When to Use
 
+        - After identifying target files through list_files or grep
+        - To understand function signatures, interfaces, and contracts
+        - For analyzing usage patterns and project conventions
+        - When detailed code examination is needed before making changes
+
+        ## Important Considerations
+
+        - Use strategically after narrowing down target files
+        - Essential for understanding context before code modification
+        - Helps identify dependencies and potential side effects
+
+        # Choosing the Right Search Strategy
+
+        - **Start with list_files** to understand project structure
+        - **Use grep for targeted searches** when you know what to look for
+        - **Apply read_file for detailed examination** of specific files
+        - **Combine multiple approaches** for comprehensive understanding
+
+        **Default workflow:**
+        1. `list_files` → understand structure
+        2. `grep` → find specific patterns/symbols  
+        3. `read_file` → examine details
+        4. Implement changes
+        5. `grep` → verify changes
+
+        # Comprehensive Workflow
+
+        ## Phase 1: Project Discovery & Analysis
+
+        **Project Structure Overview**
+        <execute_command>
+        <command>ls -la</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        - Use `list_files` tool to understand directory structure
+        - Identify key directories: `src/`, `lib/`, `components/`, `utils/`
+        - Look for config files: `package.json`, `tsconfig.json`, `Makefile`
+
+        **Technology Stack Identification**
+        <execute_command>
+        <command>grep -E "(import|require|from).*['\"]" src/ | head -20</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        - Check package dependencies and imports
+        - Identify frameworks, libraries, and coding patterns
+        - Understand project conventions (naming, file organization)
+
+        ## Phase 2: Contextual Code Investigation
+
+        **Symbol and Pattern Search**
+        <execute_command>
+        <command>grep -Rn "targetFunction\|targetClass" . --exclude-dir={node_modules,dist}</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        **Usage Pattern Analysis**
+        - Use `read_file` to examine key files in detail
+        - Understand function signatures, interfaces, and contracts
+        - Check error handling patterns and edge cases
+
+        **Dependency Mapping**
+        <execute_command>
+        <command>grep -Rn "import.*targetModule" . | grep -v test</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        ## Phase 3: Implementation Planning
+
+        **Impact Assessment**
+        - Identify all files that need modification
+        - Plan backwards compatibility considerations
+        - Consider potential side effects
+
+        **Test Strategy**
+        <execute_command>
+        <command>find . -name "*test*" -o -name "*spec*" | head -10</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        - Locate existing tests for reference
+        - Plan new test cases if needed
+
+        ## Phase 4: Code Implementation
+
+        More detail is on the EDITING FILES PART.
+
+        ## Phase 5: Comprehensive Verification
+
+        **File System Verification**
+        <execute_command>
+        <command>ls -la newfile.* 2>/dev/null || echo "Expected new files not found"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        **Code Integration Verification**
+        <execute_command>
+        <command>grep -Rn "oldSymbol" . --exclude-dir={node_modules,dist} || echo "✓ No stale references"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>grep -c "newSymbol" src/ --include="*.js" --include="*.ts" | grep -v ":0"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        **Functional Verification**
+        <execute_command>
+        <command>npm run lint 2>/dev/null || echo "Linting not configured"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>npm test 2>/dev/null || echo "Testing not configured"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        <execute_command>
+        <command>npm run build 2>/dev/null || echo "Build not configured"</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        **Documentation & Comments**
+        - Verify that new functions/classes have appropriate documentation
+        - Check that complex logic has explanatory comments
+        - Ensure README or other docs are updated if needed
+
+        ## Phase 6: Quality Assurance
+
+        **Performance Considerations**
+        - Check for potential performance impacts
+        - Verify memory usage patterns
+        - Consider scalability implications
+
+        **Security Review**
+        - Look for potential security vulnerabilities
+        - Verify input validation and sanitization
+        - Check for proper error handling
+
+        **Final Integration Check**
+        <execute_command>
+        <command>grep -Rn "TODO\|FIXME\|XXX" . --exclude-dir={node_modules,dist} | wc -l</command>
+        <requires_approval>false</requires_approval>
+        </execute_command>
+
+        # Best Practices
+
+        - **Iterative Approach**: Don't try to understand everything at once; build knowledge progressively
+        - **Documentation First**: Read existing docs, comments, and README files before diving into code
+        - **Small Steps**: Make incremental changes and verify each step
+        - **Rollback Ready**: Always know how to undo changes if something goes wrong
+        - **Test Early**: Run tests frequently during development, not just at the end
+        - **Pattern Consistency**: Follow established project patterns rather than introducing new ones
+
+        By following this comprehensive approach, you ensure thorough understanding, reliable implementation, and robust verification of all code changes.          
+
+        ====
+
+        EDITING FILES
+        
+        Before applying the editing techniques below, ensure you have followed the SEARCHING FILES methodology to fully understand the codebase context.
         You have access to two tools for working with files: **write_to_file** and **replace_in_file**. Understanding their roles and selecting the right one for the job will help ensure efficient and accurate modifications.
 
         # write_to_file
@@ -647,28 +965,31 @@ class AgenticEdit:
 
         - Quickly identifies recently modified files that may be relevant to your task
         - Provides high-level understanding of directory contents and purpose
-        - Helps prioritize which files to examine in detail with tools like read_file, search_files, or list_code_definition_names
+        - Helps prioritize which files to examine in detail with tools like read_file, shell commands, or list_code_definition_names
 
         ====
 
         CAPABILITIES
-
+        
+        - **SEARCH AND UNDERSTAND FIRST**: Your primary strength lies in systematically exploring and understanding codebases before making changes. Use list_files, execute_command (grep) to map project structure, identify patterns, and understand dependencies. This exploration-first approach is crucial for reliable code modifications.
         - You have access to tools that let you execute CLI commands on the user's computer, list files, view source code definitions, regex search, read and edit files, and ask follow-up questions. These tools help you effectively accomplish a wide range of tasks, such as writing code, making edits or improvements to existing files, understanding the current state of a project, performing system operations, and much more.
         - When the user initially gives you a task, a recursive list of all filepaths in the current working directory ('{{ current_project }}') will be included in environment_details. This provides an overview of the project's file structure, offering key insights into the project from directory/file names (how developers conceptualize and organize their code) and file extensions (the language used). This can also guide decision-making on which files to explore further. If you need to further explore directories such as outside the current working directory, you can use the list_files tool. If you pass 'true' for the recursive parameter, it will list files recursively. Otherwise, it will list files at the top level, which is better suited for generic directories where you don't necessarily need the nested structure, like the Desktop.
-        - You can use search_files to perform regex searches across files in a specified directory, outputting context-rich results that include surrounding lines. This is particularly useful for understanding code patterns, finding specific implementations, or identifying areas that need refactoring.
+        - You can use shell_command(grep) to perform regex searches across files in a specified directory, outputting context-rich results that include surrounding lines. This is particularly useful for understanding code patterns, finding specific implementations, or identifying areas that need refactoring.
         - You can use the list_code_definition_names tool to get an overview of source code definitions for all files at the top level of a specified directory. This can be particularly useful when you need to understand the broader context and relationships between certain parts of the code. You may need to call this tool multiple times to understand various parts of the codebase related to the task.
-            - For example, when asked to make edits or improvements you might analyze the file structure in the initial environment_details to get an overview of the project, then use list_code_definition_names to get further insight using source code definitions for files located in relevant directories, then read_file to examine the contents of relevant files, analyze the code and suggest improvements or make necessary edits, then use the replace_in_file tool to implement changes. If you refactored code that could affect other parts of the codebase, you could use search_files to ensure you update other files as needed.
+            - For example, when asked to make edits or improvements you might analyze the file structure in the initial environment_details to get an overview of the project, then use list_code_definition_names to get further insight using source code definitions for files located in relevant directories, then read_file to examine the contents of relevant files, analyze the code and suggest improvements or make necessary edits, then use the replace_in_file tool to implement changes. If you refactored code that could affect other parts of the codebase, you could use shell commands(grep) to ensure you update other files as needed.
         - You can use the execute_command tool to run commands on the user's computer whenever you feel it can help accomplish the user's task. When you need to execute a CLI command, you must provide a clear explanation of what the command does. Prefer to execute complex CLI commands over creating executable scripts, since they are more flexible and easier to run. Interactive and long-running commands are allowed, since the commands are run in the user's VSCode terminal. The user may keep commands running in the background and you will be kept updated on their status along the way. Each command you execute is run in a new terminal instance.
 
         ====
 
         RULES
-
+  
         - Your current working directory is: {{current_project}}
+        - **MANDATORY SEARCH BEFORE EDIT**: Before editing any file, you MUST first search to understand its context, dependencies, and usage patterns. Use list_files or grep commands to find related code, imports, and references.
+        - **VERIFY THROUGH SEARCH**: After making changes, use list_files or grep commands to verify no stale references remain and that new code integrates properly with existing patterns.
         - You cannot \`cd\` into a different directory to complete a task. You are stuck operating from '{{ current_project }}', so be sure to pass in the correct 'path' parameter when using tools that require a path.
         - Do not use the ~ character or $HOME to refer to the home directory.
         - Before using the execute_command tool, you must first think about the SYSTEM INFORMATION context provided to understand the user's environment and tailor your commands to ensure they are compatible with their system. You must also consider if the command you need to run should be executed in a specific directory outside of the current working directory '{{ current_project }}', and if so prepend with \`cd\`'ing into that directory && then executing the command (as one command since you are stuck operating from '{{ current_project }}'). For example, if you needed to run \`npm install\` in a project outside of '{{ current_project }}', you would need to prepend with a \`cd\` i.e. pseudocode for this would be \`cd (path to project) && (command, in this case npm install)\`.
-        - When using the search_files tool, craft your regex patterns carefully to balance specificity and flexibility. Based on the user's task you may use it to find code patterns, TODO comments, function definitions, or any text-based information across the project. The results include context, so analyze the surrounding code to better understand the matches. Leverage the search_files tool in combination with other tools for more comprehensive analysis. For example, use it to find specific code patterns, then use read_file to examine the full context of interesting matches before using replace_in_file to make informed changes.
+        - When using the shell command tool (grep), craft your regex patterns carefully to balance specificity and flexibility. Based on the user's task you may use it to find code patterns, TODO comments, function definitions, or any text-based information across the project. The results include context, so analyze the surrounding code to better understand the matches. Leverage the shell command tool(grep) in combination with other tools for more comprehensive analysis. For example, use it to find specific code patterns, then use read_file to examine the full context of interesting matches before using replace_in_file to make informed changes.
         - When creating a new project (such as an app, website, or any software project), organize all new files within a dedicated project directory unless the user specifies otherwise. Use appropriate file paths when creating files, as the write_to_file tool will automatically create any necessary directories. Structure the project logically, adhering to best practices for the specific type of project being created. Unless otherwise specified, new projects should be easily run without additional setup, for example most projects can be built in HTML, CSS, and JavaScript - which you can open in a browser.
         - Be sure to consider the type of project (e.g. Python, JavaScript, web application) when determining the appropriate structure and files to include. Also consider what files may be most relevant to accomplishing the task, for example looking at a project's manifest file would help you understand the project's dependencies, which you could incorporate into any code you write.
         - When making changes to code, always consider the context in which the code is being used. Ensure that your changes are compatible with the existing codebase and that they follow the project's coding standards and best practices.
@@ -688,6 +1009,7 @@ class AgenticEdit:
         - It is critical you wait for the user's response after each tool use, in order to confirm the success of the tool use. For example, if asked to make a todo app, you would create a file, wait for the user's response it was created successfully, then create another file if needed, wait for the user's response it was created successfully, etc.        
         - To display LaTeX formulas, use a single dollar sign to wrap inline formulas, like `$E=mc^2$`, and double dollar signs to wrap block-level formulas, like `$$\frac{d}{dx}e^x = e^x$$`.
         - To include flowcharts or diagrams, you can use Mermaid syntax.
+        - If you come across some unknown or unfamiliar concepts or terms, or if the user is asking a question, you can try using appropriate MCP or RAG services to obtain the information.
 
         
         {% if extra_docs %}  
@@ -743,6 +1065,7 @@ class AgenticEdit:
         3. Remember, you have extensive capabilities with access to a wide range of tools that can be used in powerful and clever ways as necessary to accomplish each goal. Before calling a tool, do some analysis within <thinking></thinking> tags. First, analyze the file structure provided in environment_details to gain context and insights for proceeding effectively. Then, think about which of the provided tools is the most relevant tool to accomplish the user's task. Next, go through each of the required parameters of the relevant tool and determine if the user has directly provided or given enough information to infer a value. When deciding if the parameter can be inferred, carefully consider all the context to see if it supports a specific value. If all of the required parameters are present or can be reasonably inferred, close the thinking tag and proceed with the tool use. BUT, if one of the values for a required parameter is missing, DO NOT invoke the tool (not even with fillers for the missing params) and instead, ask the user to provide the missing parameters using the ask_followup_question tool. DO NOT ask for more information on optional parameters if it is not provided.
         4. Once you've completed the user's task, you must use the attempt_completion tool to present the result of the task to the user. You may also provide a CLI command to showcase the result of your task; this can be particularly useful for web development tasks, where you can run e.g. \`open index.html\` to show the website you've built.
         5. The user may provide feedback, which you can use to make improvements and try again. But DO NOT continue in pointless back and forth conversations, i.e. don't end your responses with questions or offers for further assistance.                    
+        6. Work through these goals sequentially, **ALWAYS STARTING WITH COMPREHENSIVE SEARCH AND EXPLORATION** using available tools. For any code-related task, begin with list_files to understand structure, then use commands(grep) to find relevant patterns, and read_file to examine context before making changes.
                 
         """        
         ## auto_select_rules(context=request.user_input, llm=self.llm,args=self.args)        rules =       
@@ -769,6 +1092,7 @@ class AgenticEdit:
             "home_dir": os.path.expanduser("~"),
             "files": self.files.to_str(),
             "mcp_server_info": self.mcp_server_info,
+            "rag_server_info": self.rag_server_info,
             "enable_active_context_in_generate": self.args.enable_active_context_in_generate,
             "extra_docs": extra_docs,
             "file_paths_str": file_paths_str,
@@ -842,6 +1166,15 @@ class AgenticEdit:
             added_libraries = package_manager.list_added_libraries()
             
             if added_libraries:
+                # Build libraries with paths information
+                libraries_with_paths = []
+                for lib_name in added_libraries:
+                    lib_path = package_manager.get_package_path(lib_name)
+                    libraries_with_paths.append({
+                        'name': lib_name,
+                        'path': lib_path if lib_path else 'Path not found'
+                    })
+                
                 # Get documentation content for all added libraries
                 docs_content = package_manager.get_docs(return_paths=False)
                 
@@ -851,9 +1184,9 @@ class AgenticEdit:
                     
                     # Generate library documentation prompt using decorator
                     library_docs_prompt = self.generate_library_docs_prompt.prompt(
-                        libraries=added_libraries,
+                        libraries_with_paths=libraries_with_paths,
                         docs_content=combined_docs
-                    )
+                    )                    
                     
                     conversations.append({
                         "role": "user", 

autocoder/common/v2/agent/agentic_edit_tools/__init__.py

@@ -11,6 +11,7 @@ from .ask_followup_question_tool_resolver import AskFollowupQuestionToolResolver
 from .attempt_completion_tool_resolver import AttemptCompletionToolResolver
 from .plan_mode_respond_tool_resolver import PlanModeRespondToolResolver
 from .use_mcp_tool_resolver import UseMcpToolResolver
+from .use_rag_tool_resolver import UseRAGToolResolver
 from .list_package_info_tool_resolver import ListPackageInfoToolResolver
 
 __all__ = [
@@ -26,5 +27,6 @@ __all__ = [
     "AttemptCompletionToolResolver",
     "PlanModeRespondToolResolver",
     "UseMcpToolResolver",
+    "UseRAGToolResolver",
     "ListPackageInfoToolResolver",
 ]

autocoder/common/v2/agent/agentic_edit_tools/use_rag_tool_resolver.py

@@ -0,0 +1,91 @@
+from typing import Dict, Any, Optional
+import typing
+import openai
+from autocoder.common import AutoCoderArgs
+from autocoder.common.v2.agent.agentic_edit_tools.base_tool_resolver import BaseToolResolver
+from autocoder.common.v2.agent.agentic_edit_types import UseRAGTool, ToolResult
+from autocoder.common.rag_manager import RAGManager
+from loguru import logger
+
+if typing.TYPE_CHECKING:
+    from autocoder.common.v2.agent.agentic_edit import AgenticEdit
+
+
+class UseRAGToolResolver(BaseToolResolver):
+    def __init__(self, agent: Optional['AgenticEdit'], tool: UseRAGTool, args: AutoCoderArgs):
+        super().__init__(agent, tool, args)
+        self.tool: UseRAGTool = tool  # For type hinting
+        self.rag_manager = RAGManager(args)
+
+    def resolve(self) -> ToolResult:
+        """
+        通过 OpenAI SDK 访问 RAG server 来执行查询。
+        """
+        server_name = self.tool.server_name
+        query = self.tool.query
+
+        logger.info(f"正在解析 UseRAGTool: Server='{server_name}', Query='{query}'")
+
+        # 检查是否有可用的 RAG 配置
+        if not self.rag_manager.has_configs():
+            error_msg = "未找到可用的 RAG 服务器配置，请确保配置文件存在并格式正确"
+            logger.error(error_msg)
+            return ToolResult(success=False, message=error_msg)
+
+        try:
+            # 查找指定的 RAG 配置
+            rag_config = None
+            if server_name:
+                # 尝试通过名称查找
+                rag_config = self.rag_manager.get_config_by_name(server_name)
+                # 如果按名称找不到，尝试直接使用 server_name 作为 URL
+                if not rag_config:
+                    # 检查是否是直接的 URL
+                    if server_name.startswith('http'):
+                        # 使用提供的 server_name 作为 base_url
+                        base_url = server_name
+                        api_key = "dummy-key"
+                    else:
+                        error_msg = f"未找到名为 '{server_name}' 的 RAG 服务器配置\n\n{self.rag_manager.get_config_info()}"
+                        logger.error(error_msg)
+                        return ToolResult(success=False, message=error_msg)
+                else:
+                    base_url = rag_config.server_name
+                    api_key = rag_config.api_key or "dummy-key"
+            else:
+                # 如果没有指定 server_name，使用第一个可用的配置
+                rag_config = self.rag_manager.get_all_configs()[0]
+                base_url = rag_config.server_name
+                api_key = rag_config.api_key or "dummy-key"
+                logger.info(f"未指定服务器名称，使用默认配置: {rag_config.name}")
+
+            logger.info(f"使用 RAG 服务器: {base_url}")
+            
+            # 使用 OpenAI SDK 访问 RAG server
+            client = openai.OpenAI(
+                base_url=base_url,
+                api_key=api_key
+            )
+            
+            response = client.chat.completions.create(
+                model="default",
+                messages=[
+                    {
+                        "role": "user",
+                        "content": query
+                    }
+                ],
+                max_tokens=8024
+            )
+            
+            result_content = response.choices[0].message.content
+            logger.info(f"RAG server 响应成功，内容长度: {len(result_content)}")
+            
+            return ToolResult(success=True, message=result_content)
+            
+        except Exception as e:
+            error_msg = f"访问 RAG server 时出错: {str(e)}"
+            if not self.rag_manager.has_configs():
+                error_msg += f"\n\n{self.rag_manager.get_config_info()}"
+            logger.error(error_msg)
+            return ToolResult(success=False, message=error_msg) 

autocoder/common/v2/agent/agentic_edit_types.py

@@ -56,6 +56,10 @@ class UseMcpTool(BaseTool):
     tool_name: str
     query:str
 
+class UseRAGTool(BaseTool):
+    server_name: str
+    query: str
+
 class ListPackageInfoTool(BaseTool):
     path: str  # 源码包目录，相对路径或绝对路径
 
@@ -118,6 +122,7 @@ TOOL_MODEL_MAP: Dict[str, Type[BaseTool]] = {
     "attempt_completion": AttemptCompletionTool,
     "plan_mode_respond": PlanModeRespondTool,
     "use_mcp_tool": UseMcpTool,
+    "use_rag_tool": UseRAGTool,
     "list_package_info": ListPackageInfoTool,    
 }