jarvis-ai-assistant 0.1.104__tar.gz → 0.1.106__tar.gz

{jarvis_ai_assistant-0.1.104/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.106}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.104
+Version: 0.1.106
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -59,6 +59,7 @@ Requires-Dist: rich>=13.3.1
 Requires-Dist: pygments>=2.15.0
 Requires-Dist: fuzzywuzzy>=0.18.0
 Requires-Dist: python-Levenshtein>=0.26.1
+Requires-Dist: jedi>=0.19.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -133,15 +134,25 @@ Jarvis supports configuration through environment variables that can be set in t
 
 | Environment Variable | Description | Default Value | Required |
 |---------|------|--------|------|
-| JARVIS_PLATFORM | AI platform to use, supports kimi/openai/ai8 etc | kimi | Yes |
-| JARVIS_MODEL | Model name to use | - | No |
-| JARVIS_CODEGEN_PLATFORM | AI platform for code generation | Same as JARVIS_PLATFORM | No |
-| JARVIS_CODEGEN_MODEL | Model name for code generation | Same as JARVIS_MODEL | No |
-| JARVIS_CHEAP_PLATFORM | AI platform for cheap operations | Same as JARVIS_PLATFORM | No |
-| JARVIS_CHEAP_MODEL | Model name for cheap operations | Same as JARVIS_MODEL | No |
-| JARVIS_THINKING_PLATFORM | AI platform for thinking | Same as JARVIS_PLATFORM | No |
-| JARVIS_THINKING_MODEL | Model name for thinking | Same as JARVIS_MODEL | No |
-| JARVIS_THREAD_COUNT | Number of threads for parallel processing | 10 | No |
+| JARVIS_PLATFORM | AI platform to use | kimi | Yes |
+| JARVIS_MODEL | Model name to use | kimi | No |
+| JARVIS_CODEGEN_PLATFORM | Platform for code generation | Same as JARVIS_PLATFORM | No |
+| JARVIS_CODEGEN_MODEL | Model for code generation | Same as JARVIS_MODEL | No |
+| JARVIS_THINKING_PLATFORM | Platform for thinking tasks | Same as JARVIS_PLATFORM | No |
+| JARVIS_THINKING_MODEL | Model for thinking tasks | Same as JARVIS_MODEL | No |
+| JARVIS_CHEAP_PLATFORM | Platform for cheap operations | Same as JARVIS_PLATFORM | No |
+| JARVIS_CHEAP_MODEL | Model for cheap operations | Same as JARVIS_MODEL | No |
+| JARVIS_THREAD_COUNT | Number of threads | 1 | No |
+| JARVIS_MAX_CONTEXT_LENGTH | Maximum context length | 131072 | No |
+| JARVIS_MIN_PARAGRAPH_LENGTH | Minimum paragraph length | 50 | No |
+| JARVIS_MAX_PARAGRAPH_LENGTH | Maximum paragraph length | 1000 | No |
+| JARVIS_CONTEXT_WINDOW | Context window size | 5 | No |
+| JARVIS_AUTO_COMPLETE | Enable auto completion | false | No |
+| JARVIS_DISABLE_CODEBASE | Disable codebase features | false | No |
+| JARVIS_USE_METHODOLOGY | Enable methodology | true | No |
+| JARVIS_RECORD_METHODOLOGY | Record methodology | true | No |
+| JARVIS_NEED_SUMMARY | Generate summaries | true | No |
+| JARVIS_DONT_USE_LOCAL_MODEL | Avoid using local models | false | No |
 | OPENAI_API_KEY | API key for OpenAI platform | - | Required for OpenAI |
 | OPENAI_API_BASE | Base URL for OpenAI API | https://api.deepseek.com | No |
 | OPENAI_MODEL_NAME | Model name for OpenAI | deepseek-chat | No |
@@ -191,11 +202,8 @@ jss "describe what you want to do"
 # Manage git commits
 jarvis-git-commit
 
-# Generate and manage ctags
-jarvis-ctags
-
 # Manage AI platforms
-jarvis-platform
+jarvis-platform-manager
 ```
 
 Each command supports `--help` flag for detailed usage information:
@@ -205,9 +213,8 @@ jarvis-code-agent --help
 jarvis-codebase --help
 jarvis-rag --help
 jarvis-smart-shell --help
-jarvis-platform --help
+jarvis-platform-manager --help
 jarvis-git-commit --help
-jarvis-ctags --help
 ```
 
 ## 🛠️ Tools

{jarvis_ai_assistant-0.1.104 → jarvis_ai_assistant-0.1.106}/README.md

@@ -63,15 +63,25 @@ Jarvis supports configuration through environment variables that can be set in t
 
 | Environment Variable | Description | Default Value | Required |
 |---------|------|--------|------|
-| JARVIS_PLATFORM | AI platform to use, supports kimi/openai/ai8 etc | kimi | Yes |
-| JARVIS_MODEL | Model name to use | - | No |
-| JARVIS_CODEGEN_PLATFORM | AI platform for code generation | Same as JARVIS_PLATFORM | No |
-| JARVIS_CODEGEN_MODEL | Model name for code generation | Same as JARVIS_MODEL | No |
-| JARVIS_CHEAP_PLATFORM | AI platform for cheap operations | Same as JARVIS_PLATFORM | No |
-| JARVIS_CHEAP_MODEL | Model name for cheap operations | Same as JARVIS_MODEL | No |
-| JARVIS_THINKING_PLATFORM | AI platform for thinking | Same as JARVIS_PLATFORM | No |
-| JARVIS_THINKING_MODEL | Model name for thinking | Same as JARVIS_MODEL | No |
-| JARVIS_THREAD_COUNT | Number of threads for parallel processing | 10 | No |
+| JARVIS_PLATFORM | AI platform to use | kimi | Yes |
+| JARVIS_MODEL | Model name to use | kimi | No |
+| JARVIS_CODEGEN_PLATFORM | Platform for code generation | Same as JARVIS_PLATFORM | No |
+| JARVIS_CODEGEN_MODEL | Model for code generation | Same as JARVIS_MODEL | No |
+| JARVIS_THINKING_PLATFORM | Platform for thinking tasks | Same as JARVIS_PLATFORM | No |
+| JARVIS_THINKING_MODEL | Model for thinking tasks | Same as JARVIS_MODEL | No |
+| JARVIS_CHEAP_PLATFORM | Platform for cheap operations | Same as JARVIS_PLATFORM | No |
+| JARVIS_CHEAP_MODEL | Model for cheap operations | Same as JARVIS_MODEL | No |
+| JARVIS_THREAD_COUNT | Number of threads | 1 | No |
+| JARVIS_MAX_CONTEXT_LENGTH | Maximum context length | 131072 | No |
+| JARVIS_MIN_PARAGRAPH_LENGTH | Minimum paragraph length | 50 | No |
+| JARVIS_MAX_PARAGRAPH_LENGTH | Maximum paragraph length | 1000 | No |
+| JARVIS_CONTEXT_WINDOW | Context window size | 5 | No |
+| JARVIS_AUTO_COMPLETE | Enable auto completion | false | No |
+| JARVIS_DISABLE_CODEBASE | Disable codebase features | false | No |
+| JARVIS_USE_METHODOLOGY | Enable methodology | true | No |
+| JARVIS_RECORD_METHODOLOGY | Record methodology | true | No |
+| JARVIS_NEED_SUMMARY | Generate summaries | true | No |
+| JARVIS_DONT_USE_LOCAL_MODEL | Avoid using local models | false | No |
 | OPENAI_API_KEY | API key for OpenAI platform | - | Required for OpenAI |
 | OPENAI_API_BASE | Base URL for OpenAI API | https://api.deepseek.com | No |
 | OPENAI_MODEL_NAME | Model name for OpenAI | deepseek-chat | No |
@@ -121,11 +131,8 @@ jss "describe what you want to do"
 # Manage git commits
 jarvis-git-commit
 
-# Generate and manage ctags
-jarvis-ctags
-
 # Manage AI platforms
-jarvis-platform
+jarvis-platform-manager
 ```
 
 Each command supports `--help` flag for detailed usage information:
@@ -135,9 +142,8 @@ jarvis-code-agent --help
 jarvis-codebase --help
 jarvis-rag --help
 jarvis-smart-shell --help
-jarvis-platform --help
+jarvis-platform-manager --help
 jarvis-git-commit --help
-jarvis-ctags --help
 ```
 
 ## 🛠️ Tools

{jarvis_ai_assistant-0.1.104 → jarvis_ai_assistant-0.1.106}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.104"
+version = "0.1.106"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "Your Name", email = "your.email@example.com" }]
@@ -41,6 +41,7 @@ dependencies = [
     "pygments>=2.15.0",
     "fuzzywuzzy>=0.18.0",
     "python-Levenshtein>=0.26.1",
+    "jedi>=0.19.0",
 ]
 requires-python = ">=3.8"
 
@@ -57,7 +58,6 @@ jarvis-codebase = "jarvis.jarvis_codebase.main:main"
 jarvis-rag = "jarvis.jarvis_rag.main:main"
 jarvis-smart-shell = "jarvis.jarvis_smart_shell.main:main"
 jss = "jarvis.jarvis_smart_shell.main:main"
-jarvis-platform = "jarvis.jarvis_platform.main:main"
-jarvis-git-commit = "jarvis.tools.git_commiter:main"
-jarvis-ctags = "jarvis.tools.create_ctags_agent:main"
-jarvis-code-review = "jarvis.tools.code_review:main"
+jarvis-platform-manager = "jarvis.jarvis_platform_manager.main:main"
+jarvis-git-commit = "jarvis.jarvis_tools.git_commiter:main"
+jarvis-code-review = "jarvis.jarvis_tools.code_review:main"

{jarvis_ai_assistant-0.1.104 → jarvis_ai_assistant-0.1.106}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.104",
+    version="0.1.106",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -33,6 +33,7 @@ setup(
         "pygments>=2.15.0",
         "fuzzywuzzy>=0.18.0",
         "python-Levenshtein>=0.26.1",
+        "jedi>=0.19.0",
     ],
     entry_points={
         "console_scripts": [
@@ -42,10 +43,9 @@ setup(
             "jarvis-rag=jarvis.jarvis_rag.main:main",
             "jarvis-smart-shell=jarvis.jarvis_smart_shell.main:main",
             "jss=jarvis.jarvis_smart_shell.main:main",
-            "jarvis-platform=jarvis.jarvis_platform.main:main",
-            "jarvis-git-commit=jarvis.tools.git_commiter:main",
-            "jarvis-ctags=jarvis.tools.create_ctags_agent:main",
-            "jarvis-code-review=jarvis.tools.code_review:main",
+            "jarvis-platform-manager=jarvis.jarvis_platform_manager.main:main",
+            "jarvis-git-commit=jarvis.jarvis_tools.git_commiter:main",
+            "jarvis-code-review=jarvis.jarvis_tools.code_review:main",
         ],
     },
     python_requires=">=3.8",

{jarvis_ai_assistant-0.1.104 → jarvis_ai_assistant-0.1.106}/src/jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.104" 
+__version__ = "0.1.106" 

{jarvis_ai_assistant-0.1.104 → jarvis_ai_assistant-0.1.106}/src/jarvis/agent.py

@@ -5,22 +5,29 @@ from typing import Callable, Dict, List, Optional
 from prompt_toolkit import prompt
 import yaml
 
-from jarvis.models.base import BasePlatform
-from jarvis.models.registry import PlatformRegistry
-from jarvis.tools.registry import ToolRegistry, tool_call_help
-from jarvis.utils import PrettyOutput, OutputType, is_auto_complete, load_methodology, add_agent, delete_current_agent, get_max_context_length, get_multiline_input, init_env
+from jarvis.jarvis_platform.base import BasePlatform
+from jarvis.jarvis_platform.registry import PlatformRegistry
+from jarvis.jarvis_tools.registry import ToolRegistry, tool_call_help
+from jarvis.utils import PrettyOutput, OutputType, is_auto_complete, is_need_summary, is_record_methodology, load_methodology, add_agent, delete_current_agent, get_max_context_length, get_multiline_input, init_env, is_use_methodology
 import os
 
 class Agent:
 
-    def __del__(self):
-        delete_current_agent()
-
     def set_summary_prompt(self, summary_prompt: str):
+        """Set the summary prompt for task completion.
+        
+        Args:
+            summary_prompt: The prompt template for generating task summaries
+        """
         self.summary_prompt = summary_prompt
 
-    def set_output_filter(self, output_filter: List[Callable]):
-        self.output_filter = output_filter
+    def set_output_handler_before_tool(self, handler: List[Callable]):
+        """Set handlers to process output before tool execution.
+        
+        Args:
+            handler: List of callable functions to process output
+        """
+        self.output_handler_before_tool = handler
         
     def __init__(self, 
                  system_prompt: str, 
@@ -29,35 +36,47 @@ class Agent:
                  tool_registry: Optional[ToolRegistry] = None, 
                  platform: Optional[BasePlatform] = None, 
                  summary_prompt: Optional[str] = None, 
-                 auto_complete: bool = False, 
-                 record_methodology: bool = True,
-                 output_filter: Optional[List[Callable]] = None,
-                 need_summary: bool = True):
-        """Initialize Agent with a model, optional tool registry and name
+                 auto_complete: Optional[bool] = None, 
+                 output_handler_before_tool: Optional[List[Callable]] = None,
+                 output_handler_after_tool: Optional[List[Callable]] = None,
+                 use_methodology: Optional[bool] = None,
+                 record_methodology: Optional[bool] = None,
+                 need_summary: Optional[bool] = None,
+                 max_context_length: Optional[int] = None):
+        """Initialize an Agent instance.
         
         Args:
-            system_prompt: System prompt
-            name: Agent name, default is "Jarvis"
-            is_sub_agent: Whether it is a sub-agent, default is False
-            tool_registry: Tool registry instance
-            platform: Optional platform instance, default uses normal platform
+            system_prompt: The system prompt defining agent behavior
+            name: Agent name, defaults to "Jarvis"
+            is_sub_agent: Whether this is a sub-agent
+            tool_registry: Registry of available tools
+            platform: AI platform to use
+            summary_prompt: Template for generating summaries
+            auto_complete: Whether to enable auto-completion
+            output_handler_before_tool: Handlers to process output before tool execution
+            output_handler_after_tool: Handlers to process output after tool execution
+            use_methodology: Whether to use methodology
+            record_methodology: Whether to record methodology
+            need_summary: Whether to generate summaries
+            max_context_length: Maximum context length
         """
-        add_agent(name)
         PrettyOutput.print(f"Welcome to Jarvis, your AI assistant, Initiating...", OutputType.SYSTEM)
         if platform is not None:
             self.model = platform
         else:
             self.model = PlatformRegistry.get_global_platform_registry().get_normal_platform()
         self.tool_registry = tool_registry if tool_registry else ToolRegistry()
-        self.record_methodology = record_methodology
+        self.record_methodology = record_methodology if record_methodology is not None else is_record_methodology()
+        self.use_methodology = use_methodology if use_methodology is not None else is_use_methodology()
         self.name = name
         self.is_sub_agent = is_sub_agent
         self.prompt = ""
         self.conversation_length = 0  # Use length counter instead
         self.system_prompt = system_prompt
-        self.need_summary = need_summary
+        self.need_summary = need_summary if need_summary is not None else is_need_summary()
         # Load configuration from environment variables
-        self.output_filter = output_filter if output_filter else []
+        self.output_handler_before_tool = output_handler_before_tool if output_handler_before_tool else []
+        self.output_handler_after_tool = output_handler_after_tool if output_handler_after_tool else []
 
         self.summary_prompt = summary_prompt if summary_prompt else f"""Please generate a concise summary report of the task execution, including:
 
@@ -70,28 +89,22 @@ class Agent:
 Please describe in concise bullet points, highlighting important information.
 """
         
-        self.max_context_length = get_max_context_length()
-
-        self.auto_complete = auto_complete
-
+        self.max_context_length = max_context_length if max_context_length is not None else get_max_context_length()
 
-        
-            
-        # Initialize methodology related attributes
-        self.methodology_data = []
+        self.auto_complete = auto_complete if auto_complete is not None else is_auto_complete()
 
         PrettyOutput.section(f"Jarvis initialized - With {self.model.name()}", OutputType.SYSTEM)
+
         tools = self.tool_registry.get_all_tools()
         if tools:
             PrettyOutput.section(f"Available tools: {', '.join([tool['name'] for tool in tools])}", OutputType.SYSTEM)
 
-
-        # Load methodology
         
         tools_prompt = self.tool_registry.load_tools()
         complete_prompt = """"""
         if self.auto_complete:
             complete_prompt = """
+            ## Task Completion
             When the task is completed, you should print the following message:
             <!!!COMPLETE!!!>
             """
@@ -106,8 +119,18 @@ Please describe in concise bullet points, highlighting important information.
         self.first = True
 
     @staticmethod
-    def extract_tool_calls(content: str) -> List[Dict]:
-        """Extract tool calls from content, if multiple tool calls are detected, raise an exception, and return the content before the tool call and the tool call"""
+    def _extract_tool_calls(content: str) -> List[Dict]:
+        """Extract tool calls from content.
+        
+        Args:
+            content: The content containing tool calls
+            
+        Returns:
+            List[Dict]: List of extracted tool calls with name and arguments
+            
+        Raises:
+            Exception: If tool call is missing necessary fields
+        """
         # Split content into lines
         lines = content.split('\n')
         tool_call_lines = []
@@ -143,7 +166,17 @@ Please describe in concise bullet points, highlighting important information.
         return []
 
     def _call_model(self, message: str) -> str:
-        """Call model to get response"""
+        """Call the AI model with retry logic.
+        
+        Args:
+            message: The input message for the model
+            
+        Returns:
+            str: Model's response
+            
+        Note:
+            Will retry with exponential backoff up to 30 seconds between retries
+        """
         sleep_time = 5
         while True:
             ret = self.model.chat_until_success(message)
@@ -159,16 +192,17 @@ Please describe in concise bullet points, highlighting important information.
 
 
     def _summarize_and_clear_history(self) -> None:
-        """
-        [System message]
-        Summarize current conversation history and clear history, only keep system message and summary
+        """Summarize current conversation and clear history.
         
         This method will:
-        1. Request the model to summarize the key information from the current conversation
+        1. Generate a summary of key information
         2. Clear the conversation history
         3. Keep the system message
-        4. Add the summary as new context
-        5. Reset the conversation round
+        4. Add summary as new context
+        5. Reset conversation length
+        
+        Note:
+            Used when context length exceeds maximum
         """
         # Create a new model instance to summarize, avoid affecting the main conversation
 
@@ -203,10 +237,14 @@ Please continue the task based on the above information.
             PrettyOutput.print(f"Failed to summarize conversation history: {str(e)}", OutputType.ERROR)
 
     def _complete_task(self) -> str:
-        """Complete task and generate summary
+        """Complete the current task and generate summary if needed.
         
         Returns:
             str: Task summary or completion status
+            
+        Note:
+            - For main agent: May generate methodology if enabled
+            - For sub-agent: May generate summary if enabled
         """
         PrettyOutput.section("Task completed", OutputType.SUCCESS)
         
@@ -227,7 +265,7 @@ Please continue the task based on the above information.
                     
                     # 检查是否包含工具调用
                     try:
-                        tool_calls = Agent.extract_tool_calls(response)
+                        tool_calls = Agent._extract_tool_calls(response)
                         if tool_calls:
                             self.tool_registry.handle_tool_calls(tool_calls)
                     except Exception as e:
@@ -246,17 +284,23 @@ Please continue the task based on the above information.
 
 
     def run(self, user_input: str, file_list: Optional[List[str]] = None) -> str:
-        """Process user input and return response, return task summary report
+        """Process user input and execute the task.
         
         Args:
-            user_input: User input task description
-            file_list: Optional file list, default is None
-        
+            user_input: User's task description or request
+            file_list: Optional list of files to process
+            
         Returns:
             str: Task summary report
+            
+        Note:
+            - Handles context management
+            - Processes tool calls
+            - Manages conversation flow
+            - Supports interactive mode
         """
 
-        
+        add_agent(self.name)
 
         try:
             PrettyOutput.section("Preparing environment", OutputType.PLANNING)
@@ -266,7 +310,7 @@ Please continue the task based on the above information.
             # 显示任务开始
             PrettyOutput.section(f"Starting new task: {self.name}", OutputType.PLANNING)
 
-            if self.first:
+            if self.first and self.use_methodology:
                 self.prompt = f"{user_input}\n\n{load_methodology(user_input)}"
                 self.first = False
             else:
@@ -289,11 +333,11 @@ Please continue the task based on the above information.
                         self.prompt = ""
                         self.conversation_length += len(current_response) 
 
-                    for filter in self.output_filter:
-                        self.prompt += filter(current_response)
+                    for handler in self.output_handler_before_tool:
+                        self.prompt += handler(current_response)
 
                     try:
-                        result = Agent.extract_tool_calls(current_response)
+                        result = Agent._extract_tool_calls(current_response)
                     except Exception as e:
                         PrettyOutput.print(f"Tool call error: {str(e)}", OutputType.ERROR)
                         self.prompt += f"Tool call error: {str(e)}"
@@ -303,6 +347,9 @@ Please continue the task based on the above information.
                         PrettyOutput.print("Executing tool call...", OutputType.PROGRESS)
                         tool_result = self.tool_registry.handle_tool_calls(result)
                         self.prompt += tool_result
+
+                    for handler in self.output_handler_after_tool:
+                        self.prompt += handler(current_response)
                     
                     if self.prompt:
                         continue
@@ -328,9 +375,17 @@ Please continue the task based on the above information.
             PrettyOutput.print(str(e), OutputType.ERROR)
             return f"Task failed: {str(e)}"
         
+        finally:
+            delete_current_agent()
 
-    def clear_history(self):
-        """Clear conversation history, only keep system prompt"""
+    def _clear_history(self):
+        """Clear conversation history while preserving system prompt.
+        
+        This will:
+        1. Clear the prompt
+        2. Reset the model
+        3. Reset conversation length counter
+        """
         self.prompt = "" 
         self.model.reset()
         self.conversation_length = 0  # Reset conversation length
@@ -338,7 +393,7 @@ Please continue the task based on the above information.
 
 
 
-def load_tasks() -> dict:
+def _load_tasks() -> dict:
     """Load tasks from .jarvis files in user home and current directory."""
     tasks = {}
     
@@ -375,18 +430,20 @@ def load_tasks() -> dict:
         except Exception as e:
             PrettyOutput.print(f"Error loading .jarvis/pre-command file: {str(e)}", OutputType.ERROR)
 
-    # Read methodology
-    method_path = os.path.expanduser("~/.jarvis/methodology")
-    if os.path.exists(method_path):
-        with open(method_path, "r", encoding="utf-8") as f:
-            methodology = yaml.safe_load(f)
-        if isinstance(methodology, dict):
-            for name, desc in methodology.items():
-                tasks[f"Run Methodology: {str(name)}\n {str(desc)}" ] = str(desc)
+    
+    if is_use_methodology():
+        # Read methodology
+        method_path = os.path.expanduser("~/.jarvis/methodology")
+        if os.path.exists(method_path):
+            with open(method_path, "r", encoding="utf-8") as f:
+                methodology = yaml.safe_load(f)
+            if isinstance(methodology, dict):
+                for name, desc in methodology.items():
+                    tasks[f"Run Methodology: {str(name)}\n {str(desc)}" ] = str(desc)
     
     return tasks
 
-def select_task(tasks: dict) -> str:
+def _select_task(tasks: dict) -> str:
     """Let user select a task from the list or skip. Returns task description if selected."""
     if not tasks:
         return ""
@@ -462,9 +519,9 @@ def main():
         agent = Agent(system_prompt=origin_agent_system_prompt, tool_registry=ToolRegistry())
 
         # 加载预定义任务
-        tasks = load_tasks()
+        tasks = _load_tasks()
         if tasks:
-            selected_task = select_task(tasks)
+            selected_task = _select_task(tasks)
             if selected_task:
                 PrettyOutput.print(f"\nExecute task: {selected_task}", OutputType.INFO)
                 agent.run(selected_task, args.files)