auto-coder 0.1.398__py3-none-any.whl → 0.1.399__py3-none-any.whl

autocoder/common/v2/agent/agentic_edit.py

@@ -79,10 +79,12 @@ from autocoder.common.v2.agent.agentic_edit_types import (AgenticEditRequest, To
                                                           LLMOutputEvent, LLMThinkingEvent, ToolCallEvent,
                                                           ToolResultEvent, CompletionEvent, PlanModeRespondEvent, ErrorEvent, TokenUsageEvent,
                                                           WindowLengthChangeEvent,
+                                                          ConversationIdEvent,
                                                           # Import specific tool types for display mapping
                                                           ReadFileTool, WriteToFileTool, ReplaceInFileTool, ExecuteCommandTool,
                                                           ListFilesTool, SearchFilesTool, ListCodeDefinitionNamesTool,
-                                                          AskFollowupQuestionTool, UseMcpTool, AttemptCompletionTool
+                                                          AskFollowupQuestionTool, UseMcpTool, AttemptCompletionTool,
+                                                          AgenticEditConversationConfig
                                                           )
 from autocoder.common.rag_manager import RAGManager
 from autocoder.rag.token_counter import count_tokens
@@ -103,6 +105,12 @@ TOOL_RESOLVER_MAP: Dict[Type[BaseTool], Type[BaseToolResolver]] = {
     UseMcpTool: UseMcpToolResolver,
     UseRAGTool: UseRAGToolResolver
 }
+from autocoder.common.conversations.get_conversation_manager import (
+    get_conversation_manager,
+    get_conversation_manager_config,
+    reset_conversation_manager
+)
+from autocoder.common.conversations import ConversationManagerConfig
 
 
 # --- Tool Display Customization is now handled by agentic_tool_display.py ---
@@ -117,7 +125,8 @@ class AgenticEdit:
         args: AutoCoderArgs,
         memory_config: MemoryConfig,
         command_config: Optional[CommandConfig] = None,
-        conversation_name:Optional[str] = "current"        
+        conversation_name:Optional[str] = "current",
+        conversation_config:Optional[AgenticEditConversationConfig] = None
     ):
         self.llm = llm
         self.context_prune_llm = get_single_llm(args.context_prune_model or args.model,product_mode=args.product_mode) 
@@ -127,7 +136,7 @@ class AgenticEdit:
         self.files = files
         # Removed self.max_iterations
         # Note: This might need updating based on the new flow
-        self.conversation_history = conversation_history
+        self.conversation_history = conversation_history                    
 
         self.current_conversations = []
         self.memory_config = memory_config
@@ -183,6 +192,19 @@ class AgenticEdit:
         # 格式: { file_path: FileChangeEntry(...) }
         self.file_changes: Dict[str, FileChangeEntry] = {}
 
+        # 对话管理器
+        self.conversation_config =conversation_config
+        self.conversation_manager = get_conversation_manager()
+
+        if self.conversation_config.action == "new":
+            conversation_id = self.conversation_manager.create_conversation(name=self.conversation_config.query or "New Conversation",
+                                                                            description=self.conversation_config.query or "New Conversation")
+            self.conversation_manager.set_current_conversation(conversation_id)
+
+        if self.conversation_config.action == "resume" and self.conversation_config.conversation_id:
+            self.conversation_manager.set_current_conversation(self.conversation_config.conversation_id)
+            
+
     @byzerllm.prompt()
     def generate_library_docs_prompt(self, libraries_with_paths: List[Dict[str, str]], docs_content: str) -> Dict[str, Any]:
         """
@@ -419,7 +441,7 @@ class AgenticEdit:
         <list_code_definition_names>
         <path>Directory path here</path>
         </list_code_definition_names>
-
+                
         ## ask_followup_question
         Description: Ask the user a question to gather additional information needed to complete the task. This tool should be used when you encounter ambiguities, need clarification, or require more details to proceed effectively. It allows for interactive problem-solving by enabling direct communication with the user. Use this tool judiciously to maintain a balance between gathering necessary information and avoiding excessive back-and-forth.
         Parameters:
@@ -431,7 +453,7 @@ class AgenticEdit:
         <options>
         Array of options here (optional), e.g. ["Option 1", "Option 2", "Option 3"]
         </options>
-        </ask_followup_question>
+        </ask_followup_question>        
 
         ## attempt_completion
         Description: After each tool use, the user will respond with the result of that tool use, i.e. if it succeeded or failed, along with any reasons for failure. Once you've received the results of tool uses and can confirm that the task is complete, use this tool to present the result of your work to the user. Optionally you may provide a CLI command to showcase the result of your work. The user may respond with feedback if they are not satisfied with the result, which you can use to make improvements and try again.
@@ -1096,6 +1118,7 @@ class AgenticEdit:
             "enable_active_context_in_generate": self.args.enable_active_context_in_generate,
             "extra_docs": extra_docs,
             "file_paths_str": file_paths_str,
+            "agentic_auto_approve": self.args.enable_agentic_auto_approve,
         }
 
     # Removed _execute_command_result and execute_auto_command methods
@@ -1144,7 +1167,7 @@ class AgenticEdit:
         """
         Analyzes the user request, interacts with the LLM, parses responses,
         executes tools, and yields structured events for visualization until completion or error.
-        """
+        """       
         logger.info(f"Starting analyze method with user input: {request.user_input[:50]}...")
         system_prompt = self._analyze.prompt(request)
         logger.info(f"Generated system prompt with length: {len(system_prompt)}")
@@ -1154,7 +1177,7 @@ class AgenticEdit:
         conversations = [
             {"role": "system", "content": system_prompt},
         ] 
-        
+                                
         # Add third-party library documentation information
         try:
             package_manager = LLMFriendlyPackageManager(
@@ -1199,12 +1222,36 @@ class AgenticEdit:
                     })
                     
         except Exception as e:
-            logger.warning(f"Failed to load library documentation: {str(e)}")
-                                
+            logger.warning(f"Failed to load library documentation: {str(e)}")                                        
+
+        if self.conversation_config.action == "resume":
+            current_conversation = self.conversation_manager.get_current_conversation()
+            # 如果继续的是当前的对话，将其消息加入到 conversations 中
+            if current_conversation and current_conversation.get('messages'):
+                for message in current_conversation['messages']:
+                    # 确保消息格式正确（包含 role 和 content 字段）
+                    if isinstance(message, dict) and 'role' in message and 'content' in message:
+                        conversations.append({
+                            "role": message['role'],
+                            "content": message['content']
+                        })
+                logger.info(f"Resumed conversation with {len(current_conversation['messages'])} existing messages")     
+        
+        if self.conversation_manager.get_current_conversation_id() is None:
+            conv_id = self.conversation_manager.create_conversation(name=self.conversation_config.query,description=self.conversation_config.query)
+            self.conversation_manager.set_current_conversation(conv_id)            
+        
+        self.conversation_manager.set_current_conversation(self.conversation_manager.get_current_conversation_id())
+        yield ConversationIdEvent(conversation_id=self.conversation_manager.get_current_conversation_id())
+        
         conversations.append({
             "role": "user", "content": request.user_input
-        })        
+        })  
         
+        self.conversation_manager.append_message_to_current(
+                    role="user", 
+                    content=request.user_input,
+                    metadata={})
         
         self.current_conversations = conversations
         
@@ -1292,7 +1339,12 @@ class AgenticEdit:
                     conversations.append({
                         "role": "assistant",
                         "content": assistant_buffer + tool_xml
-                    })                    
+                    }) 
+                    self.conversation_manager.append_message_to_current(
+                        role="assistant", 
+                        content=assistant_buffer + tool_xml,
+                        metadata={})
+                    
                     assistant_buffer = ""  # Reset buffer after tool call
                     
                     # 计算当前对话的总 token 数量并触发事件
@@ -1311,7 +1363,7 @@ class AgenticEdit:
                         completion_event = CompletionEvent(completion=tool_obj, completion_xml=tool_xml)
                         logger.info(
                             "AgenticEdit analyze loop finished due to AttemptCompletion.")
-                        save_formatted_log(self.args.source_dir, json.dumps(conversations, ensure_ascii=False), "agentic_conversation")        
+                        save_formatted_log(self.args.source_dir, json.dumps(conversations, ensure_ascii=False), "agentic_conversation")                                
                         mark_event_should_finish = True
                         should_yield_completion_event = True
                         continue
@@ -1387,6 +1439,10 @@ class AgenticEdit:
                         "role": "user",  # Simulating the user providing the tool result
                         "content": error_xml
                     })
+                    self.conversation_manager.append_message_to_current(
+                        role="user", 
+                        content=error_xml,
+                        metadata={})
                     
                     # 计算当前对话的总 token 数量并触发事件
                     current_conversation_str = json.dumps(conversations, ensure_ascii=False)
@@ -1425,6 +1481,9 @@ class AgenticEdit:
                         logger.info("Adding new assistant message")
                         conversations.append(
                             {"role": "assistant", "content": assistant_buffer})
+                        self.conversation_manager.append_message_to_current(
+                            role="assistant", content=assistant_buffer,metadata={})
+                        
                     elif last_message["role"] == "assistant":
                         logger.info("Appending to existing assistant message")
                         last_message["content"] += assistant_buffer
@@ -1441,6 +1500,11 @@ class AgenticEdit:
                     "role": "user",
                     "content": "NOTE: You must use an appropriate tool (such as read_file, write_to_file, execute_command, etc.) or explicitly complete the task (using attempt_completion). Do not provide text responses without taking concrete actions. Please select a suitable tool to continue based on the user's task."
                 })
+
+                self.conversation_manager.append_message_to_current(
+                    role="user", 
+                    content="NOTE: You must use an appropriate tool (such as read_file, write_to_file, execute_command, etc.) or explicitly complete the task (using attempt_completion). Do not provide text responses without taking concrete actions. Please select a suitable tool to continue based on the user's task.",
+                    metadata={})
                 
                 # 计算当前对话的总 token 数量并触发事件
                 current_conversation_str = json.dumps(conversations, ensure_ascii=False)
@@ -1580,9 +1644,9 @@ class AgenticEdit:
                             else:
                                 yield ToolCallEvent(tool=tool_obj, tool_xml=reconstructed_xml)
                         else:
-                            yield ErrorEvent(message=f"Failed to parse tool: <{current_tool_tag}>")
+                            # yield ErrorEvent(message=f"Failed to parse tool: <{current_tool_tag}>")
                             # Optionally yield the raw XML as plain text?
-                            # yield LLMOutputEvent(text=tool_xml)
+                            yield LLMOutputEvent(text=f"Failed to parse tool: <{current_tool_tag}> {tool_xml}")
 
                         buffer = buffer[tool_block_end_index:]
                         in_tool_block = False
@@ -1924,17 +1988,41 @@ class AgenticEdit:
                         self.git_require_msg(
                             source_dir=self.args.source_dir, error=str(e)),
                         style="red"
-                    )
-        else:
-            self.printer.print_in_terminal("no_changes_made")
+                    )        
+            
 
     def run_in_terminal(self, request: AgenticEditRequest):
         """
         Runs the agentic edit process based on the request and displays
         the interaction streamingly in the terminal using Rich.
         """
+        import json
         console = Console()
         project_name = os.path.basename(os.path.abspath(self.args.source_dir))
+
+        if self.conversation_config.action == "list":
+            conversations = self.conversation_manager.list_conversations()
+            # 只保留 conversation_id 和 name 字段
+            filtered_conversations = []
+            for conv in conversations:
+                filtered_conv = {
+                    "conversation_id": conv.get("conversation_id"),
+                    "name": conv.get("name")
+                }
+                filtered_conversations.append(filtered_conv)
+            
+            # 格式化 JSON 输出，使用 JSON 格式渲染而不是 Markdown
+            json_str = json.dumps(filtered_conversations, ensure_ascii=False, indent=4)
+            console.print(Panel(json_str,
+                                  title="🏁 Task Completion", border_style="green", title_align="left"))
+            return
+        
+
+        if self.conversation_config.action == "new" and not request.user_input.strip():
+            console.print(Panel(Markdown(f"New conversation created: {self.conversation_manager.get_current_conversation_id()}"),
+                                  title="🏁 Task Completion", border_style="green", title_align="left"))
+            return
+
         console.rule(f"[bold cyan]Starting Agentic Edit: {project_name}[/]")
         console.print(Panel(
             f"[bold]{get_message('/agent/edit/user_query')}:[/bold]\n{request.user_input}", title=get_message("/agent/edit/objective"), border_style="blue"))
@@ -1952,6 +2040,9 @@ class AgenticEdit:
             self.apply_pre_changes()
             event_stream = self.analyze(request)
             for event in event_stream:
+                if isinstance(event, ConversationIdEvent):
+                    console.print(f"[dim]Conversation ID: {event.conversation_id}[/dim]")
+                    continue
                 if isinstance(event, TokenUsageEvent):
                     last_meta: SingleOutputMeta = event.usage
                     # Get model info for pricing
@@ -2144,6 +2235,18 @@ class AgenticEdit:
         finally:
             console.rule("[bold cyan]Agentic Edit Finished[/]")
 
+    def run(self, request: AgenticEditRequest):        
+        try:
+            event_stream = self.analyze(request)
+            for agent_event in event_stream:
+                yield agent_event
+                
+        except Exception as e:
+            logger.exception(
+                "An unexpected error occurred during agent execution: {e}")           
+            raise e
+
+
     def run_with_events(self, request: AgenticEditRequest):
         """
         Runs the agentic edit process, converting internal events to the
@@ -2275,6 +2378,16 @@ class AgenticEdit:
                     
                     # 记录日志
                     logger.info(f"当前会话总 tokens: {agent_event.tokens_used}")
+
+                elif isinstance(agent_event, ConversationIdEvent):
+                    metadata.path = "/agent/edit/conversation_id"
+                    content = EventContentCreator.create_result(
+                        content={
+                            "conversation_id": agent_event.conversation_id
+                        },
+                        metadata={}
+                    )
+                    event_manager.write_result(content=content.to_dict(), metadata=metadata.to_dict())    
                     
                 elif isinstance(agent_event, ErrorEvent):                                        
                     metadata.path = "/agent/edit/error"
@@ -2284,7 +2397,7 @@ class AgenticEdit:
                         details={"agent_event_type": "ErrorEvent"}
                     )
                     event_manager.write_error(
-                        content=content.to_dict(), metadata=metadata.to_dict())
+                        content=content.to_dict(), metadata=metadata.to_dict())                
                 else:
                     metadata.path = "/agent/edit/error"
                     logger.warning(

autocoder/common/v2/agent/agentic_edit_types.py

@@ -86,6 +86,11 @@ class TokenUsageEvent(BaseModel):
     """Represents the result of executing a tool."""
     usage: Any
 
+
+class ConversationIdEvent(BaseModel):
+    """Represents the conversation id."""
+    conversation_id: str
+
 class PlanModeRespondEvent(BaseModel):
     """Represents the LLM attempting to complete the task."""
     completion: SkipValidation[PlanModeRespondTool] # Skip validation
@@ -177,3 +182,8 @@ class CommandConfig(BaseModel):
     index_import: SkipValidation[Callable]
     exclude_files: SkipValidation[Callable]
 
+class AgenticEditConversationConfig(BaseModel):     
+    conversation_name: Optional[str] = "current"
+    conversation_id: Optional[str] = None 
+    action: Optional[str] = None
+    query: Optional[str] = None

autocoder/common/v2/code_auto_generate_editblock.py

@@ -52,7 +52,7 @@ class CodeAutoGenerateEditBlock:
                                  package_context: str = ""
                                  ) -> str:
         """
-        如果你需要生成代码，对于每个需要更改的文件,你需要按 *SEARCH/REPLACE block* 的格式进行生成。
+        如果你需要生成代码，对于每个需要更改的文件,你需要按 *SEARCH/REPLACE block* 的格式进行生成。        
 
         # *SEARCH/REPLACE block* Rules:
 
@@ -196,7 +196,15 @@ class CodeAutoGenerateEditBlock:
         <extra_context>
         {{ context }}
         </extra_context>
-        {%- endif %}     
+        {%- endif %}   
+
+        ====
+
+        RULES PROVIDED BY SYSTEM
+
+        Before outputting the *SEARCH/REPLACE block* format, you must carefully think through the user's question and make a complete design. These thoughts and designs should be placed within the <thinking></thinking> tags.
+
+        ====  
 
         {%- if extra_docs %}
         ====

autocoder/dispacher/__init__.py

@@ -8,6 +8,11 @@ from autocoder.dispacher.actions.action import (
 from autocoder.dispacher.actions.plugins.action_regex_project import ActionRegexProject
 from typing import Optional
 import byzerllm
+# from autocoder.common.conversations.get_conversation_manager import (
+#     get_conversation_manager,
+#     get_conversation_manager_config,
+#     reset_conversation_manager
+# )
 
 
 class Dispacher:
@@ -16,6 +21,11 @@ class Dispacher:
         self.llm = llm
 
     def dispach(self):
+        # manager = get_conversation_manager()
+        # if not manager.get_current_conversation():
+        #     manager.create_conversation(name="New Conversation", description="New Conversation")
+        #     manager.set_current_conversation(manager.get_current_conversation_id())
+
         args = self.args
         actions = [            
             ActionTSProject(args=args, llm=self.llm),

autocoder/rags.py

@@ -369,30 +369,3 @@ def update_rag_status(name: str, status: str, **kwargs) -> bool:
     return rag_manager.update_status(name, status, **kwargs)
 
 
-# 使用示例
-if __name__ == "__main__":
-    # 创建配置
-    config = {
-        "model": "deepseek_chat",
-        "tokenizer_path": "/Users/allwefantasy/Downloads/tokenizer.json",
-        "doc_dir": "/path/to/docs",
-        "rag_doc_filter_relevance": 2.0,
-        "host": "0.0.0.0",
-        "port": 8000,
-        "enable_hybrid_index": False,
-        "required_exts": ".md,.rst",
-        "disable_inference_enhance": True
-    }
-    
-    # 创建
-    create_rag_config("test_rag", config)
-    
-    # 读取
-    all_configs = get_rag_config()
-    specific_config = get_rag_config("test_rag")
-    
-    # 更新
-    update_rag_config("test_rag", {"status": "running", "port": 8001})
-    
-    # 删除
-    delete_rag_config("test_rag")

autocoder/run_context.py

@@ -9,6 +9,7 @@ class RunMode(Enum):
     """Enum representing different run modes for Auto-Coder."""
     TERMINAL = auto()
     WEB = auto()
+    API = auto()
 
 
 class RunContext:

autocoder/sdk/__init__.py

@@ -0,0 +1,188 @@
+"""
+Auto-Coder SDK
+
+为第三方开发者提供的 Python SDK，允许通过命令行工具和 Python API 两种方式使用 Auto-Coder 的核心功能。
+"""
+
+from typing import AsyncIterator, Optional, Dict, Any
+
+from .core.auto_coder_core import AutoCoderCore
+from .models.options import AutoCodeOptions
+from .models.messages import Message
+from .models.responses import StreamEvent, CodeModificationResult
+from .session.session import Session
+from .session.session_manager import SessionManager
+from .exceptions import (
+    AutoCoderSDKError,
+    SessionNotFoundError,
+    InvalidOptionsError,
+    BridgeError,
+    ValidationError
+)
+
+__version__ = "1.0.0"
+__all__ = [
+    # 核心功能
+    "query",
+    "query_sync",
+    "modify_code",
+    "modify_code_stream",
+    
+    # 数据模型
+    "AutoCodeOptions",
+    "Message",
+    "StreamEvent",
+    "CodeModificationResult",
+    
+    # 会话管理
+    "Session",
+    "SessionManager",
+    
+    # 异常
+    "AutoCoderSDKError",
+    "SessionNotFoundError",
+    "InvalidOptionsError",
+    "BridgeError",
+    "ValidationError",
+]
+
+
+async def query(
+    prompt: str, 
+    options: Optional[AutoCodeOptions] = None,
+    show_terminal: bool = True
+) -> AsyncIterator[Message]:
+    """
+    异步流式查询接口
+    
+    Args:
+        prompt: 查询提示
+        options: 配置选项
+        show_terminal: 是否在终端显示友好的渲染输出
+        
+    Yields:
+        Message: 响应消息流
+        
+    Example:
+        >>> import asyncio
+        >>> from autocoder.sdk import query, AutoCodeOptions
+        >>> 
+        >>> async def main():
+        ...     options = AutoCodeOptions(max_turns=3)
+        ...     async for message in query("Write a hello world function", options):
+        ...         print(f"[{message.role}] {message.content}")
+        >>> 
+        >>> asyncio.run(main())
+    """
+    if options is None:
+        options = AutoCodeOptions()
+    core = AutoCoderCore(options)
+    async for message in core.query_stream(f"/new {prompt}", show_terminal):
+        yield message
+
+
+def query_sync(
+    prompt: str, 
+    options: Optional[AutoCodeOptions] = None,
+    show_terminal: bool = True
+) -> str:
+    """
+    同步查询接口
+    
+    Args:
+        prompt: 查询提示
+        options: 配置选项
+        show_terminal: 是否在终端显示友好的渲染输出
+        
+    Returns:
+        str: 响应内容
+        
+    Example:
+        >>> from autocoder.sdk import query_sync, AutoCodeOptions
+        >>> 
+        >>> options = AutoCodeOptions(max_turns=1)
+        >>> response = query_sync("Write a simple calculator function", options)
+        >>> print(response)
+    """
+    if options is None:
+        options = AutoCodeOptions()
+    core = AutoCoderCore(options)
+    return core.query_sync(f"/new {prompt}", show_terminal)
+
+
+def modify_code(
+    prompt: str,
+    pre_commit: bool = False,
+    extra_args: Optional[Dict[str, Any]] = None,
+    options: Optional[AutoCodeOptions] = None,
+    show_terminal: bool = True
+) -> CodeModificationResult:
+    """
+    代码修改接口
+    
+    Args:
+        prompt: 修改提示
+        pre_commit: 是否预提交
+        extra_args: 额外参数
+        options: 配置选项
+        show_terminal: 是否在终端显示友好的渲染输出
+        
+    Returns:
+        CodeModificationResult: 修改结果
+        
+    Example:
+        >>> from autocoder.sdk import modify_code, AutoCodeOptions
+        >>> 
+        >>> options = AutoCodeOptions(cwd="/path/to/project")
+        >>> result = modify_code(
+        ...     "Add error handling to the main function",
+        ...     pre_commit=False,
+        ...     options=options
+        ... )
+        >>> 
+        >>> if result.success:
+        ...     print(f"Modified files: {result.modified_files}")
+        ... else:
+        ...     print(f"Error: {result.error_details}")
+    """
+    core = AutoCoderCore(options or AutoCodeOptions())
+    return core.modify_code(prompt, pre_commit, extra_args, show_terminal)
+
+
+async def modify_code_stream(
+    prompt: str,
+    pre_commit: bool = False,
+    extra_args: Optional[Dict[str, Any]] = None,
+    options: Optional[AutoCodeOptions] = None,
+    show_terminal: bool = True
+) -> AsyncIterator[StreamEvent]:
+    """
+    异步流式代码修改接口
+    
+    Args:
+        prompt: 修改提示
+        pre_commit: 是否预提交
+        extra_args: 额外参数
+        options: 配置选项
+        show_terminal: 是否在终端显示友好的渲染输出
+        
+    Yields:
+        StreamEvent: 修改事件流
+        
+    Example:
+        >>> import asyncio
+        >>> from autocoder.sdk import modify_code_stream, AutoCodeOptions
+        >>> 
+        >>> async def main():
+        ...     options = AutoCodeOptions(cwd="/path/to/project")
+        ...     async for event in modify_code_stream(
+        ...         "Refactor the user authentication module",
+        ...         options=options
+        ...     ):
+        ...         print(f"[{event.event_type}] {event.data}")
+        >>> 
+        >>> asyncio.run(main())
+    """
+    core = AutoCoderCore(options or AutoCodeOptions())
+    async for event in core.modify_code_stream(prompt, pre_commit, extra_args, show_terminal):
+        yield event

autocoder/sdk/cli/__init__.py

@@ -0,0 +1,15 @@
+"""
+Auto-Coder CLI 模块
+
+提供命令行接口，允许用户通过终端使用 Auto-Coder 的核心功能。
+"""
+
+from .main import AutoCoderCLI
+from .options import CLIOptions, CLIResult
+
+# 导出main函数作为入口点
+def main():
+    """CLI主入口点函数"""
+    return AutoCoderCLI.main()
+
+__all__ = ["AutoCoderCLI", "CLIOptions", "CLIResult", "main"]

autocoder/sdk/cli/__main__.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+"""
+CLI 模块主入口点
+"""
+
+import sys
+import argparse
+from pathlib import Path
+
+def main():
+    """主入口点，路由到不同的 CLI 功能"""
+    
+    # 检查是否是自动补全相关的命令
+    if len(sys.argv) > 1 and sys.argv[1] in ["install", "uninstall", "test"]:
+        # 路由到自动补全安装工具
+        from .install_completion import main as install_main
+        install_main()
+        return        
+    
+    # 默认显示自动补全工具的帮助
+    from .install_completion import main as install_main
+    install_main()
+
+if __name__ == "__main__":
+    main()
+