jarvis-ai-assistant 0.1.12__tar.gz → 0.1.14__tar.gz

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.12
+Version: 0.1.14
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -19,6 +19,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: requests>=2.25.1
 Requires-Dist: pyyaml>=5.1
 Requires-Dist: colorama>=0.4.6
+Requires-Dist: prompt_toolkit>=3.0.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -72,6 +73,42 @@ Dynamic: requires-python
 - Multi-line input support
 - Colored output with progress indicators
 
+
+## ⚙️ Environment Setup
+
+Before using Jarvis, you need to set up your environment:
+
+1. **API Key Configuration**
+
+Create a `.jarvis_env` file in your home directory (`~/.jarvis_env`):
+
+```bash
+KIMI_API_KEY=your_kimi_api_key_here
+```
+
+To get your Kimi API key:
+1. Visit [Kimi AI Platform](https://kimi.moonshot.cn) in your browser
+2. Login to your account
+3. Open browser Developer Tools (F12 or right-click -> Inspect)
+4. Go to Network tab
+5. Make any request (e.g., send a message)
+6. Find a request to the Kimi API
+7. Look for the `Authorization` header in the request headers
+8. Copy the token value (remove the "Bearer " prefix)
+9. Use this token as your `KIMI_API_KEY` in the `.jarvis_env` file
+
+2. **Task Configuration (Optional)**
+
+Create a `.jarvis` file in your working directory to define predefined tasks:
+
+```yaml
+# .jarvis
+analyze_code: Analyze the code structure and quality in the current directory
+fix_bugs: Help me find and fix potential bugs in the code
+optimize: Suggest optimizations for the code
+document: Generate documentation for the code
+```
+
 ## 🚀 Installation
 
 ```bash
@@ -80,19 +117,44 @@ pip install jarvis-ai-assistant
 
 ## 💡 Usage
 
+1. **Basic Usage**
 ```bash
-# Quick Start
+# Start Jarvis
 jarvis
 
-# Process files
+# Process specific files
 jarvis -f file1.txt file2.py
+```
 
-# Using predefined tasks
-# Create .jarvis file in your working directory:
-analyze_code: Analyze the code in the current directory
-fix_bugs: Help me find and fix bugs in the code
+2. **Using Predefined Tasks**
+
+If you have a `.jarvis` file in your working directory:
+```bash
+# Jarvis will show available tasks on startup
+# Select a task number or start a new conversation
 ```
 
+3. **Interactive Features**
+- Multi-line input support (press Enter twice to submit)
+- File understanding and analysis
+- Context-aware conversations
+- Tool integration for system operations
+
+4. **Environment Variables**
+- `KIMI_API_KEY`: Your Kimi AI API key (required)
+- Location: `~/.jarvis_env`
+- Format: `KEY=value` (one per line)
+
+5. **Task Configuration**
+- File: `.jarvis` in working directory
+- Format: `task_name: task_description`
+- Purpose: Define commonly used tasks for quick access
+- Example tasks:
+  - Code analysis
+  - Bug finding
+  - Documentation generation
+  - Performance optimization
+
 ## 🧰 Tools
 
 | Tool | Description | Example |

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/README.md

@@ -42,6 +42,42 @@
 - Multi-line input support
 - Colored output with progress indicators
 
+
+## ⚙️ Environment Setup
+
+Before using Jarvis, you need to set up your environment:
+
+1. **API Key Configuration**
+
+Create a `.jarvis_env` file in your home directory (`~/.jarvis_env`):
+
+```bash
+KIMI_API_KEY=your_kimi_api_key_here
+```
+
+To get your Kimi API key:
+1. Visit [Kimi AI Platform](https://kimi.moonshot.cn) in your browser
+2. Login to your account
+3. Open browser Developer Tools (F12 or right-click -> Inspect)
+4. Go to Network tab
+5. Make any request (e.g., send a message)
+6. Find a request to the Kimi API
+7. Look for the `Authorization` header in the request headers
+8. Copy the token value (remove the "Bearer " prefix)
+9. Use this token as your `KIMI_API_KEY` in the `.jarvis_env` file
+
+2. **Task Configuration (Optional)**
+
+Create a `.jarvis` file in your working directory to define predefined tasks:
+
+```yaml
+# .jarvis
+analyze_code: Analyze the code structure and quality in the current directory
+fix_bugs: Help me find and fix potential bugs in the code
+optimize: Suggest optimizations for the code
+document: Generate documentation for the code
+```
+
 ## 🚀 Installation
 
 ```bash
@@ -50,19 +86,44 @@ pip install jarvis-ai-assistant
 
 ## 💡 Usage
 
+1. **Basic Usage**
 ```bash
-# Quick Start
+# Start Jarvis
 jarvis
 
-# Process files
+# Process specific files
 jarvis -f file1.txt file2.py
+```
 
-# Using predefined tasks
-# Create .jarvis file in your working directory:
-analyze_code: Analyze the code in the current directory
-fix_bugs: Help me find and fix bugs in the code
+2. **Using Predefined Tasks**
+
+If you have a `.jarvis` file in your working directory:
+```bash
+# Jarvis will show available tasks on startup
+# Select a task number or start a new conversation
 ```
 
+3. **Interactive Features**
+- Multi-line input support (press Enter twice to submit)
+- File understanding and analysis
+- Context-aware conversations
+- Tool integration for system operations
+
+4. **Environment Variables**
+- `KIMI_API_KEY`: Your Kimi AI API key (required)
+- Location: `~/.jarvis_env`
+- Format: `KEY=value` (one per line)
+
+5. **Task Configuration**
+- File: `.jarvis` in working directory
+- Format: `task_name: task_description`
+- Purpose: Define commonly used tasks for quick access
+- Example tasks:
+  - Code analysis
+  - Bug finding
+  - Documentation generation
+  - Performance optimization
+
 ## 🧰 Tools
 
 | Tool | Description | Example |

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.12"
+version = "0.1.14"
 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" }]
@@ -23,6 +23,7 @@ dependencies = [
     "requests>=2.25.1",
     "pyyaml>=5.1",
     "colorama>=0.4.6",
+    "prompt_toolkit>=3.0.0",
 ]
 requires-python = ">=3.8"
 

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.12",
+    version="0.1.14",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -16,6 +16,7 @@ setup(
         "requests>=2.25.1",
         "pyyaml>=5.1",
         "colorama>=0.4.6",
+        "prompt_toolkit>=3.0.0",
     ],
     entry_points={
         "console_scripts": [

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/src/jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.12" 
+__version__ = "0.1.14" 

jarvis_ai_assistant-0.1.14/src/jarvis/main.py

@@ -0,0 +1,155 @@
+#!/usr/bin/env python3
+"""Command line interface for Jarvis."""
+
+import argparse
+import yaml
+import os
+import sys
+from pathlib import Path
+
+# 添加父目录到Python路径以支持导入
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+from jarvis.agent import Agent
+from jarvis.tools import ToolRegistry
+from jarvis.models import KimiModel
+from jarvis.utils import PrettyOutput, OutputType, get_multiline_input, load_env_from_file
+
+
+def load_tasks() -> dict:
+    """Load tasks from .jarvis files in user home and current directory."""
+    tasks = {}
+    
+    # 检查用户目录下的 .jarvis
+    user_jarvis = os.path.expanduser("~/.jarvis")
+    if os.path.exists(user_jarvis):
+        try:
+            with open(user_jarvis, "r", encoding="utf-8") as f:
+                user_tasks = yaml.safe_load(f)
+                
+            if isinstance(user_tasks, dict):
+                # 验证并添加用户目录的任务
+                for name, desc in user_tasks.items():
+                    if desc:  # 确保描述不为空
+                        tasks[str(name)] = str(desc)
+            else:
+                PrettyOutput.print("Warning: ~/.jarvis file should contain a dictionary of task_name: task_description", OutputType.ERROR)
+        except Exception as e:
+            PrettyOutput.print(f"Error loading ~/.jarvis file: {str(e)}", OutputType.ERROR)
+    
+    # 检查当前目录下的 .jarvis
+    if os.path.exists(".jarvis"):
+        try:
+            with open(".jarvis", "r", encoding="utf-8") as f:
+                local_tasks = yaml.safe_load(f)
+                
+            if isinstance(local_tasks, dict):
+                # 验证并添加当前目录的任务，如果有重名则覆盖用户目录的任务
+                for name, desc in local_tasks.items():
+                    if desc:  # 确保描述不为空
+                        tasks[str(name)] = str(desc)
+            else:
+                PrettyOutput.print("Warning: .jarvis file should contain a dictionary of task_name: task_description", OutputType.ERROR)
+        except Exception as e:
+            PrettyOutput.print(f"Error loading .jarvis file: {str(e)}", OutputType.ERROR)
+    
+    return tasks
+
+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 ""
+    
+    # Convert tasks to list for ordered display
+    task_names = list(tasks.keys())
+    
+    PrettyOutput.print("\nAvailable tasks:", OutputType.INFO)
+    for i, name in enumerate(task_names, 1):
+        PrettyOutput.print(f"[{i}] {name}", OutputType.INFO)
+    PrettyOutput.print("[0] Skip predefined tasks", OutputType.INFO)
+    
+    while True:
+        try:
+            choice = input("\nSelect a task number (0 to skip): ").strip()
+            if not choice:
+                return ""
+            
+            choice = int(choice)
+            if choice == 0:
+                return ""
+            elif 1 <= choice <= len(task_names):
+                selected_name = task_names[choice - 1]
+                return tasks[selected_name]  # Return the task description
+            else:
+                PrettyOutput.print("Invalid choice. Please try again.", OutputType.ERROR)
+        except ValueError:
+            PrettyOutput.print("Please enter a valid number.", OutputType.ERROR)
+
+def main():
+    """Main entry point for Jarvis."""
+    # Add argument parser
+    parser = argparse.ArgumentParser(description='Jarvis AI Assistant')
+    parser.add_argument('-f', '--files', nargs='*', help='List of files to process')
+    args = parser.parse_args()
+
+    load_env_from_file()
+    
+    try:
+        kimi_api_key = os.getenv("KIMI_API_KEY")
+        if not kimi_api_key:
+            PrettyOutput.section("环境配置缺失", OutputType.ERROR)
+            PrettyOutput.print("\n需要设置 KIMI_API_KEY 才能使用 Jarvis。请按以下步骤操作：", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("\n1. 获取 Kimi API Key:", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   • 访问 Kimi AI 平台: https://kimi.moonshot.cn", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   • 登录您的账号", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   • 打开浏览器开发者工具 (F12 或右键 -> 检查)", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   • 切换到 Network 标签页", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   • 发送任意消息", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   • 在请求中找到 Authorization 头部", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   • 复制 token 值（去掉 'Bearer ' 前缀）", OutputType.INFO, timestamp=False)
+            
+            PrettyOutput.print("\n2. 设置环境变量:", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   方法 1: 创建或编辑 ~/.jarvis_env 文件:", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   echo 'KIMI_API_KEY=your_key_here' > ~/.jarvis_env", OutputType.CODE, timestamp=False)
+            
+            PrettyOutput.print("\n   方法 2: 直接设置环境变量:", OutputType.INFO, timestamp=False)
+            PrettyOutput.print("   export KIMI_API_KEY=your_key_here", OutputType.CODE, timestamp=False)
+            
+            PrettyOutput.print("\n设置完成后重新运行 Jarvis。", OutputType.INFO, timestamp=False)
+            return 1
+        
+        model = KimiModel(kimi_api_key)
+
+        tool_registry = ToolRegistry(model)
+        agent = Agent(model, tool_registry)
+
+        # 欢迎信息
+        PrettyOutput.print(f"Jarvis 已初始化 - With Kimi", OutputType.SYSTEM)
+        
+        # 加载预定义任务
+        tasks = load_tasks()
+        if tasks:
+            selected_task = select_task(tasks)
+            if selected_task:
+                PrettyOutput.print(f"\n执行任务: {selected_task}", OutputType.INFO)
+                agent.run(selected_task, args.files)
+                return 0
+        
+        # 如果没有选择预定义任务，进入交互模式
+        while True:
+            try:
+                user_input = get_multiline_input("请输入您的任务(输入空行退出):")
+                if not user_input or user_input == "__interrupt__":
+                    break
+                agent.run(user_input, args.files)
+            except Exception as e:
+                PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
+
+    except Exception as e:
+        PrettyOutput.print(f"初始化错误: {str(e)}", OutputType.ERROR)
+        return 1
+
+    return 0
+
+if __name__ == "__main__":
+    exit(main()) 

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/src/jarvis/utils.py

@@ -1,12 +1,15 @@
 from pathlib import Path
 import sys
 import time
+import os
 from typing import Dict, Optional
 from enum import Enum
 from datetime import datetime
 import colorama
-from colorama import Fore, Style
-import os
+from colorama import Fore, Style as ColoramaStyle
+from prompt_toolkit import PromptSession
+from prompt_toolkit.styles import Style as PromptStyle
+from prompt_toolkit.formatted_text import FormattedText
 
 # 初始化colorama
 colorama.init()
@@ -84,10 +87,10 @@ class PrettyOutput:
         prefix = PrettyOutput.PREFIXES.get(output_type, "")
         
         # 添加时间戳 - 使用白色
-        time_str = f"{Fore.WHITE}[{datetime.now().strftime('%H:%M:%S')}]{Style.RESET_ALL} " if timestamp else ""
+        time_str = f"{Fore.WHITE}[{datetime.now().strftime('%H:%M:%S')}]{ColoramaStyle.RESET_ALL} " if timestamp else ""
         
         # 格式化输出
-        formatted_text = f"{time_str}{color}{icon} {prefix}: {text}{Style.RESET_ALL}"
+        formatted_text = f"{time_str}{color}{icon} {prefix}: {text}{ColoramaStyle.RESET_ALL}"
         
         return formatted_text
 
@@ -101,15 +104,15 @@ class PrettyOutput:
         """打印带分隔线的段落标题"""
         width = 60
         color = PrettyOutput.COLORS.get(output_type, "")
-        print(f"\n{color}" + "=" * width + f"{Style.RESET_ALL}")
+        print(f"\n{color}" + "=" * width + f"{ColoramaStyle.RESET_ALL}")
         PrettyOutput.print(title.center(width - 10), output_type, timestamp=False)
-        print(f"{color}" + "=" * width + f"{Style.RESET_ALL}\n")
+        print(f"{color}" + "=" * width + f"{ColoramaStyle.RESET_ALL}\n")
 
     @staticmethod
     def print_stream(text: str, output_type: OutputType):
         """打印流式输出，不换行"""
         color = PrettyOutput.COLORS.get(output_type, "")
-        sys.stdout.write(f"{color}{text}{Style.RESET_ALL}")
+        sys.stdout.write(f"{color}{text}{ColoramaStyle.RESET_ALL}")
         sys.stdout.flush()
 
     @staticmethod
@@ -119,29 +122,44 @@ class PrettyOutput:
         sys.stdout.flush()
 
 def get_multiline_input(tip: str) -> str:
-    """获取多行输入"""
+    """获取多行输入，支持方向键、历史记录等功能"""
     PrettyOutput.print(tip + "\n", OutputType.INFO)
-    lines = []
     
-    while True:
-        try:
-            prompt = "... " if lines else ">>> "
-            sys.stdout.write(f"{Fore.GREEN}{prompt}{Style.RESET_ALL}")
-            sys.stdout.flush()
+    # 创建输入会话，启用历史记录
+    session = PromptSession(history=None)  # 使用默认历史记录
+    
+    # 定义提示符样式
+    style = PromptStyle.from_dict({
+        'prompt': 'ansicyan',
+    })
+    
+    lines = []
+    try:
+        while True:
+            # 设置提示符
+            prompt = FormattedText([
+                ('class:prompt', '... ' if lines else '>>> ')
+            ])
             
-            line = input().strip()
+            # 获取输入
+            line = session.prompt(
+                prompt,
+                style=style,
+            ).strip()
+            
+            # 空行处理
             if not line:
-                if not lines:  # 如果是第一行就输入空行
+                if not lines:  # 第一行就输入空行
                     return ""
-                break
-            
+                break  # 结束多行输入
+                
             lines.append(line)
             
-        except KeyboardInterrupt:
-            PrettyOutput.print("\n输入已取消", OutputType.ERROR)
-            return "__interrupt__"
+    except KeyboardInterrupt:
+        PrettyOutput.print("\n输入已取消", OutputType.ERROR)
+        return "__interrupt__"
     
-    return "\n".join(lines).strip()
+    return "\n".join(lines)
 
 def load_env_from_file():
     """从~/.jarvis_env加载环境变量"""

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/src/jarvis_ai_assistant.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.12
+Version: 0.1.14
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -19,6 +19,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: requests>=2.25.1
 Requires-Dist: pyyaml>=5.1
 Requires-Dist: colorama>=0.4.6
+Requires-Dist: prompt_toolkit>=3.0.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -72,6 +73,42 @@ Dynamic: requires-python
 - Multi-line input support
 - Colored output with progress indicators
 
+
+## ⚙️ Environment Setup
+
+Before using Jarvis, you need to set up your environment:
+
+1. **API Key Configuration**
+
+Create a `.jarvis_env` file in your home directory (`~/.jarvis_env`):
+
+```bash
+KIMI_API_KEY=your_kimi_api_key_here
+```
+
+To get your Kimi API key:
+1. Visit [Kimi AI Platform](https://kimi.moonshot.cn) in your browser
+2. Login to your account
+3. Open browser Developer Tools (F12 or right-click -> Inspect)
+4. Go to Network tab
+5. Make any request (e.g., send a message)
+6. Find a request to the Kimi API
+7. Look for the `Authorization` header in the request headers
+8. Copy the token value (remove the "Bearer " prefix)
+9. Use this token as your `KIMI_API_KEY` in the `.jarvis_env` file
+
+2. **Task Configuration (Optional)**
+
+Create a `.jarvis` file in your working directory to define predefined tasks:
+
+```yaml
+# .jarvis
+analyze_code: Analyze the code structure and quality in the current directory
+fix_bugs: Help me find and fix potential bugs in the code
+optimize: Suggest optimizations for the code
+document: Generate documentation for the code
+```
+
 ## 🚀 Installation
 
 ```bash
@@ -80,19 +117,44 @@ pip install jarvis-ai-assistant
 
 ## 💡 Usage
 
+1. **Basic Usage**
 ```bash
-# Quick Start
+# Start Jarvis
 jarvis
 
-# Process files
+# Process specific files
 jarvis -f file1.txt file2.py
+```
 
-# Using predefined tasks
-# Create .jarvis file in your working directory:
-analyze_code: Analyze the code in the current directory
-fix_bugs: Help me find and fix bugs in the code
+2. **Using Predefined Tasks**
+
+If you have a `.jarvis` file in your working directory:
+```bash
+# Jarvis will show available tasks on startup
+# Select a task number or start a new conversation
 ```
 
+3. **Interactive Features**
+- Multi-line input support (press Enter twice to submit)
+- File understanding and analysis
+- Context-aware conversations
+- Tool integration for system operations
+
+4. **Environment Variables**
+- `KIMI_API_KEY`: Your Kimi AI API key (required)
+- Location: `~/.jarvis_env`
+- Format: `KEY=value` (one per line)
+
+5. **Task Configuration**
+- File: `.jarvis` in working directory
+- Format: `task_name: task_description`
+- Purpose: Define commonly used tasks for quick access
+- Example tasks:
+  - Code analysis
+  - Bug finding
+  - Documentation generation
+  - Performance optimization
+
 ## 🧰 Tools
 
 | Tool | Description | Example |

{jarvis_ai_assistant-0.1.12 → jarvis_ai_assistant-0.1.14}/src/jarvis_ai_assistant.egg-info/requires.txt

@@ -1,6 +1,7 @@
 requests>=2.25.1
 pyyaml>=5.1
 colorama>=0.4.6
+prompt_toolkit>=3.0.0
 
 [dev]
 pytest

jarvis_ai_assistant-0.1.12/src/jarvis/main.py

@@ -1,122 +0,0 @@
-#!/usr/bin/env python3
-"""Command line interface for Jarvis."""
-
-import argparse
-import yaml
-import os
-import sys
-from pathlib import Path
-
-# 添加父目录到Python路径以支持导入
-sys.path.insert(0, str(Path(__file__).parent.parent))
-
-from jarvis.agent import Agent
-from jarvis.tools import ToolRegistry
-from jarvis.models import KimiModel
-from jarvis.utils import PrettyOutput, OutputType, get_multiline_input, load_env_from_file
-
-
-def load_tasks() -> dict:
-    """Load tasks from .jarvis file if it exists."""
-    if not os.path.exists(".jarvis"):
-        return {}
-    
-    try:
-        with open(".jarvis", "r", encoding="utf-8") as f:
-            tasks = yaml.safe_load(f)
-            
-        if not isinstance(tasks, dict):
-            PrettyOutput.print("Warning: .jarvis file should contain a dictionary of task_name: task_description", OutputType.ERROR)
-            return {}
-            
-        # Validate format and convert all values to strings
-        validated_tasks = {}
-        for name, desc in tasks.items():
-            if desc:  # Ensure description is not empty
-                validated_tasks[str(name)] = str(desc)
-                
-        return validated_tasks
-    except Exception as e:
-        PrettyOutput.print(f"Error loading .jarvis file: {str(e)}", OutputType.ERROR)
-        return {}
-
-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 ""
-    
-    # Convert tasks to list for ordered display
-    task_names = list(tasks.keys())
-    
-    PrettyOutput.print("\nAvailable tasks:", OutputType.INFO)
-    for i, name in enumerate(task_names, 1):
-        PrettyOutput.print(f"[{i}] {name}", OutputType.INFO)
-    PrettyOutput.print("[0] Skip predefined tasks", OutputType.INFO)
-    
-    while True:
-        try:
-            choice = input("\nSelect a task number (0 to skip): ").strip()
-            if not choice:
-                return ""
-            
-            choice = int(choice)
-            if choice == 0:
-                return ""
-            elif 1 <= choice <= len(task_names):
-                selected_name = task_names[choice - 1]
-                return tasks[selected_name]  # Return the task description
-            else:
-                PrettyOutput.print("Invalid choice. Please try again.", OutputType.ERROR)
-        except ValueError:
-            PrettyOutput.print("Please enter a valid number.", OutputType.ERROR)
-
-def main():
-    """Main entry point for Jarvis."""
-    # Add argument parser
-    parser = argparse.ArgumentParser(description='Jarvis AI Assistant')
-    parser.add_argument('-f', '--files', nargs='*', help='List of files to process')
-    args = parser.parse_args()
-
-    load_env_from_file()
-    
-    try:
-        kimi_api_key = os.getenv("KIMI_API_KEY")
-        if not kimi_api_key:
-            PrettyOutput.print("Kimi API key 未设置", OutputType.ERROR)
-            return 1
-        
-        model = KimiModel(kimi_api_key)
-
-        tool_registry = ToolRegistry(model)
-        agent = Agent(model, tool_registry)
-
-        # 欢迎信息
-        PrettyOutput.print(f"Jarvis 已初始化 - With Kimi", OutputType.SYSTEM)
-        
-        # 加载预定义任务
-        tasks = load_tasks()
-        if tasks:
-            selected_task = select_task(tasks)
-            if selected_task:
-                PrettyOutput.print(f"\n执行任务: {selected_task}", OutputType.INFO)
-                agent.run(selected_task, args.files)
-                return 0
-        
-        # 如果没有选择预定义任务，进入交互模式
-        while True:
-            try:
-                user_input = get_multiline_input("请输入您的任务(输入空行退出):")
-                if not user_input or user_input == "__interrupt__":
-                    break
-                agent.run(user_input, args.files)
-            except Exception as e:
-                PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
-
-    except Exception as e:
-        PrettyOutput.print(f"初始化错误: {str(e)}", OutputType.ERROR)
-        return 1
-
-    return 0
-
-if __name__ == "__main__":
-    exit(main())