jarvis-ai-assistant 0.1.3__py3-none-any.whl → 0.1.6__py3-none-any.whl

jarvis/utils.py

@@ -1,86 +1,122 @@
+from pathlib import Path
 import sys
 import time
-import threading
-from typing import Optional
+from typing import Dict, Optional
 from enum import Enum
 from datetime import datetime
 import colorama
 from colorama import Fore, Style
+import os
 
 # 初始化colorama
 colorama.init()
 
-class Spinner:
-    """加载动画类"""
-    def __init__(self, message: str = "思考中"):
-        self.spinner_chars = "⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏"
-        self.message = message
-        self.running = False
-        self.spinner_thread = None
-
-    def _spin(self):
-        i = 0
-        while self.running:
-            sys.stdout.write(f"\r{Fore.BLUE}{self.spinner_chars[i]} {self.message}...{Style.RESET_ALL}")
-            sys.stdout.flush()
-            time.sleep(0.1)
-            i = (i + 1) % len(self.spinner_chars)
-        sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
-        sys.stdout.flush()
-
-    def start(self):
-        self.running = True
-        self.spinner_thread = threading.Thread(target=self._spin)
-        self.spinner_thread.start()
-
-    def stop(self):
-        self.running = False
-        if self.spinner_thread:
-            self.spinner_thread.join()
-
 class OutputType(Enum):
-    SYSTEM = "system"
-    CODE = "code"
-    RESULT = "result"
-    ERROR = "error"
-    INFO = "info"
+    SYSTEM = "system"      # AI助手消息
+    CODE = "code"         # 代码相关
+    RESULT = "result"     # 工具执行结果
+    ERROR = "error"       # 错误信息
+    INFO = "info"         # 系统提示
+    PLANNING = "planning" # 任务规划
+    PROGRESS = "progress" # 执行进度
+    SUCCESS = "success"   # 成功信息
+    WARNING = "warning"   # 警告信息
+    DEBUG = "debug"       # 调试信息
+    USER = "user"         # 用户输入
+    TOOL = "tool"         # 工具调用
 
 class PrettyOutput:
     """美化输出类"""
+    
+    # 颜色方案 - 只使用前景色
+    COLORS = {
+        OutputType.SYSTEM: Fore.CYAN,      # 青色 - AI助手
+        OutputType.CODE: Fore.GREEN,       # 绿色 - 代码
+        OutputType.RESULT: Fore.BLUE,      # 蓝色 - 结果
+        OutputType.ERROR: Fore.RED,        # 红色 - 错误
+        OutputType.INFO: Fore.YELLOW,      # 黄色 - 提示
+        OutputType.PLANNING: Fore.MAGENTA, # 紫色 - 规划
+        OutputType.PROGRESS: Fore.WHITE,   # 白色 - 进度
+        OutputType.SUCCESS: Fore.GREEN,    # 绿色 - 成功
+        OutputType.WARNING: Fore.YELLOW,   # 黄色 - 警告
+        OutputType.DEBUG: Fore.BLUE,       # 蓝色 - 调试
+        OutputType.USER: Fore.GREEN,       # 绿色 - 用户
+        OutputType.TOOL: Fore.YELLOW,      # 黄色 - 工具
+    }
+    
+    # 图标方案
+    ICONS = {
+        OutputType.SYSTEM: "🤖",    # 机器人 - AI助手
+        OutputType.CODE: "📝",      # 记事本 - 代码
+        OutputType.RESULT: "✨",    # 闪光 - 结果
+        OutputType.ERROR: "❌",     # 错误 - 错误
+        OutputType.INFO: "ℹ️",      # 信息 - 提示
+        OutputType.PLANNING: "📋",  # 剪贴板 - 规划
+        OutputType.PROGRESS: "⏳",  # 沙漏 - 进度
+        OutputType.SUCCESS: "✅",   # 勾选 - 成功
+        OutputType.WARNING: "⚠️",   # 警告 - 警告
+        OutputType.DEBUG: "🔍",     # 放大镜 - 调试
+        OutputType.USER: "👤",      # 用户 - 用户
+        OutputType.TOOL: "🔧",      # 扳手 - 工具
+    }
+    
+    # 前缀方案
+    PREFIXES = {
+        OutputType.SYSTEM: "Assistant",
+        OutputType.CODE: "Code",
+        OutputType.RESULT: "Result",
+        OutputType.ERROR: "Error",
+        OutputType.INFO: "Info",
+        OutputType.PLANNING: "Plan",
+        OutputType.PROGRESS: "Progress",
+        OutputType.SUCCESS: "Success",
+        OutputType.WARNING: "Warning",
+        OutputType.DEBUG: "Debug",
+        OutputType.USER: "User",
+        OutputType.TOOL: "Tool",
+    }
+
     @staticmethod
     def format(text: str, output_type: OutputType, timestamp: bool = True) -> str:
-        # 颜色映射
-        colors = {
-            OutputType.SYSTEM: Fore.CYAN,
-            OutputType.CODE: Fore.GREEN,
-            OutputType.RESULT: Fore.BLUE,
-            OutputType.ERROR: Fore.RED,
-            OutputType.INFO: Fore.YELLOW
-        }
-
-        # 图标映射
-        icons = {
-            OutputType.SYSTEM: "🤖",
-            OutputType.CODE: "📝",
-            OutputType.RESULT: "✨",
-            OutputType.ERROR: "❌",
-            OutputType.INFO: "ℹ️"
-        }
-
-        color = colors.get(output_type, "")
-        icon = icons.get(output_type, "")
+        """格式化输出文本"""
+        color = PrettyOutput.COLORS.get(output_type, "")
+        icon = PrettyOutput.ICONS.get(output_type, "")
+        prefix = PrettyOutput.PREFIXES.get(output_type, "")
         
-        # 添加时间戳
-        time_str = f"[{datetime.now().strftime('%H:%M:%S')}] " if timestamp else ""
+        # 添加时间戳 - 使用白色
+        time_str = f"{Fore.WHITE}[{datetime.now().strftime('%H:%M:%S')}]{Style.RESET_ALL} " if timestamp else ""
         
         # 格式化输出
-        formatted_text = f"{color}{time_str}{icon} {text}{Style.RESET_ALL}"
+        formatted_text = f"{time_str}{color}{icon} {prefix}: {text}{Style.RESET_ALL}"
         
         return formatted_text
 
     @staticmethod
     def print(text: str, output_type: OutputType, timestamp: bool = True):
-        print(PrettyOutput.format(text, output_type, timestamp)) 
+        """打印格式化的输出"""
+        print(PrettyOutput.format(text, output_type, timestamp))
+
+    @staticmethod
+    def section(title: str, output_type: OutputType = OutputType.INFO):
+        """打印带分隔线的段落标题"""
+        width = 60
+        color = PrettyOutput.COLORS.get(output_type, "")
+        print(f"\n{color}" + "=" * width + f"{Style.RESET_ALL}")
+        PrettyOutput.print(title.center(width - 10), output_type, timestamp=False)
+        print(f"{color}" + "=" * width + f"{Style.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.flush()
+
+    @staticmethod
+    def print_stream_end():
+        """流式输出结束，打印换行"""
+        sys.stdout.write("\n")
+        sys.stdout.flush()
 
 def get_multiline_input(tip: str) -> str:
     """获取多行输入"""
@@ -89,10 +125,13 @@ def get_multiline_input(tip: str) -> str:
     
     while True:
         try:
-            line = input("... " if lines else ">>> ").strip()
-            # 检查是否结束输入
+            prompt = "... " if lines else ">>> "
+            sys.stdout.write(f"{Fore.GREEN}{prompt}{Style.RESET_ALL}")
+            sys.stdout.flush()
+            
+            line = input().strip()
             if not line:
-                if not lines:  # 如果是第一行就输入空行或finish
+                if not lines:  # 如果是第一行就输入空行
                     return ""
                 break
             
@@ -100,6 +139,26 @@ def get_multiline_input(tip: str) -> str:
             
         except KeyboardInterrupt:
             PrettyOutput.print("\n输入已取消", OutputType.ERROR)
-            return ""
+            return "__interrupt__"
+    
+    return "\n".join(lines).strip()
+
+def load_env_from_file():
+    """从~/.jarvis_env加载环境变量"""
+    env_file = Path.home() / ".jarvis_env"
+    
+    if env_file.exists():
+        try:
+            with open(env_file, "r", encoding="utf-8") as f:
+                for line in f:
+                    line = line.strip()
+                    if line and not line.startswith("#"):
+                        try:
+                            key, value = line.split("=", 1)
+                            os.environ[key.strip()] = value.strip().strip("'").strip('"')
+                        except ValueError:
+                            continue
+        except Exception as e:
+            PrettyOutput.print(f"Warning: Failed to read ~/.jarvis_env: {e}", OutputType.WARNING)
     
-    return "\n".join(lines).strip()
+    

jarvis/zte_llm.py

@@ -0,0 +1,136 @@
+import requests
+import json
+import os
+from pathlib import Path
+from typing import Dict, Any, List, Optional
+
+from jarvis.utils import OutputType, PrettyOutput
+from .models import BaseModel
+
+class ZteLLM(BaseModel):
+    """ZTE Nebula LLM implementation"""
+    
+    def __init__(self, 
+                 app_id: str,
+                 app_key: str,
+                 emp_no: str,
+                 auth_value: str,
+                 model: str = "nebulacoder",
+                 ):
+        """Initialize ZTE LLM with required credentials"""
+        self.app_id = str(app_id)
+        self.app_key = str(app_key)
+        self.emp_no = str(emp_no)
+        self.auth_value = str(auth_value)
+        self.model = model
+        self.base_url = "https://studio.zte.com.cn/zte-studio-ai-platform/openapi/v1"
+        
+    def _make_request(self, endpoint: str, data: Dict[str, Any]) -> Dict[str, Any]:
+        """Make request to ZTE API"""
+        headers = {
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {self.app_id}-{self.app_key}',
+            'X-Emp-No': self.emp_no,
+            'X-Auth-Value': self.auth_value
+        }
+        
+        response = requests.post(
+            f"{self.base_url}/{endpoint}",
+            headers=headers,
+            json=data
+        )
+        
+        response.raise_for_status()
+        result = response.json()
+        
+        if result["code"]["code"] != "0000":
+            raise Exception(f"API Error: {result['code']['msg']}")
+            
+        ret = result["bo"]
+        PrettyOutput.print_stream(ret, OutputType.SYSTEM)
+        return ret
+
+    def chat(self, messages: List[Dict[str, Any]], tools: Optional[List[Dict]] = None) -> Dict[str, Any]:
+        """Chat with ZTE LLM"""
+        # Convert messages to prompt
+        prompt = self._convert_messages_to_prompt(messages)
+        
+        # Prepare data for API call
+        data = {
+            "chatUuid": "",
+            "chatName": "",
+            "stream": False,
+            "keep": False,
+            "text": prompt,
+            "model": self.model
+        }
+        
+        # If tools are provided, add them to the prompt
+        if tools:
+            tools_desc = "Available tools:\n\n" + json.dumps(tools, indent=2, ensure_ascii=False)
+            data["text"] = tools_desc + "\n\n" + data["text"]
+        
+        try:
+            result = self._make_request("chat", data)
+            
+            # Parse the response to extract potential tool calls
+            response_text = result["result"]
+            tool_calls = BaseModel.extract_tool_calls(response_text)
+            
+            return {
+                "message": {
+                    "content": response_text,
+                    "tool_calls": tool_calls
+                }
+            }
+            
+        except Exception as e:
+            raise Exception(f"ZTE LLM chat failed: {str(e)}")
+
+    def _convert_messages_to_prompt(self, messages: List[Dict[str, Any]]) -> str:
+        """Convert message list to a single prompt string"""
+        prompt_parts = []
+        
+        for message in messages:
+            role = message["role"]
+            content = message.get("content", "")
+            
+            if role == "system":
+                prompt_parts.append(f"System: {content}")
+            elif role == "user":
+                prompt_parts.append(f"User: {content}")
+            elif role == "assistant":
+                prompt_parts.append(f"Assistant: {content}")
+            elif role == "tool":
+                prompt_parts.append(f"Tool Result: {content}")
+                
+        return "\n\n".join(prompt_parts)
+
+
+def create_zte_llm(model_name: str = "NebulaBiz") -> ZteLLM:
+    """Create ZTE LLM instance with provided parameters"""
+    # Load environment variables from file
+    
+    # Get credentials from parameters, env file, or system environment variables
+    app_id = os.getenv('ZTE_APP_ID') 
+    app_key = os.getenv('ZTE_APP_KEY') 
+    emp_no = os.getenv('ZTE_EMP_NO') 
+    auth_value = os.getenv('ZTE_AUTH_VALUE') 
+    
+    # Validate required credentials
+    if not all([app_id, app_key, emp_no, auth_value]):
+        raise ValueError(
+            "Missing required credentials. Please provide through either:\n"
+            "1. Function parameters\n"
+            "2. ~/.jarvis_env file\n"
+            "3. System environment variables\n\n"
+            "Required variables: ZTE_APP_ID, ZTE_APP_KEY, ZTE_EMP_NO, ZTE_AUTH_VALUE"
+        )
+    
+    return ZteLLM(
+        app_id=app_id,
+        app_key=app_key,
+        emp_no=emp_no,
+        auth_value=auth_value,
+        model=model_name
+    ) 

{jarvis_ai_assistant-0.1.3.dist-info → jarvis_ai_assistant-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.3
+Version: 0.1.6
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -21,13 +21,15 @@ Requires-Dist: beautifulsoup4>=4.9.3
 Requires-Dist: duckduckgo-search>=3.0.0
 Requires-Dist: pyyaml>=5.1
 Requires-Dist: ollama>=0.1.6
-Requires-Dist: sentence-transformers>=2.5.1
-Requires-Dist: chromadb>=0.4.24
+Requires-Dist: colorama>=0.4.6
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
 Requires-Dist: isort; extra == "dev"
 Requires-Dist: mypy; extra == "dev"
+Dynamic: author
+Dynamic: home-page
+Dynamic: requires-python
 
 <div align="center">
 

jarvis_ai_assistant-0.1.6.dist-info/RECORD

@@ -0,0 +1,38 @@
+jarvis/__init__.py,sha256=IB2q6sPGefOsNYLmp5aaOnkS9qWc9ZMY1HQ6V-l-xaY,49
+jarvis/agent.py,sha256=KLeC4oWUYMRXg4QgjyVbfSaJwaIghfJ6KE4JpIS2blI,7225
+jarvis/main.py,sha256=pRwtXWRnbBC3J-xbtuYTxQ0e9JuwU5XxCzbS5SsQESU,6118
+jarvis/models.py,sha256=ZSdUEdVb3aFgvvMtNZuDh3kVXAlA_0CPWv-BzTsLdv8,4508
+jarvis/utils.py,sha256=3hLtv-HcBL8Ngw69cowhARuIFrjcQ6yRP3Y1o9CvtsI,5992
+jarvis/zte_llm.py,sha256=fACFyhEX2ssGC-VOJucBDF_khCFZtyQEC9r9FChae-Q,4631
+jarvis/__pycache__/__init__.cpython-313.pyc,sha256=---ccXWtkwpa6rJNXmjP8vCpz9eCZLk6M1yN3clIrbk,208
+jarvis/__pycache__/agent.cpython-313.pyc,sha256=_i8Bj7p2cBi8G6y8s-7LF-cLO05V3c8AVDoEGbmZbgg,8362
+jarvis/__pycache__/main.cpython-313.pyc,sha256=gJ8eJXVTDuNplc_yjMOxuM17xHwLi6jMtYlB9yiABBc,7963
+jarvis/__pycache__/models.cpython-313.pyc,sha256=7ofJ31hdS9H_VHVpAi_gQGxIcl25i4VWZUpXOinRAnQ,5548
+jarvis/__pycache__/tools.cpython-313.pyc,sha256=lAD4LrnnWzNZQmHXGfZ_2l7oskOpr2_2OC-gdFhxQY8,33933
+jarvis/__pycache__/utils.cpython-313.pyc,sha256=k4jyAlx4tlW0MKLMLzV7OOH9zsKrK0H955kY6M2SuFU,8772
+jarvis/__pycache__/zte_llm.cpython-313.pyc,sha256=OHFPd1MiNkqoncDIFlCj592Y9bQZ8c0Gq6JKJ0enyPA,5553
+jarvis/tools/__init__.py,sha256=Bv3Djt_To60p248ECO2H_LqsTmUriLsq9MDb1Q-uuzA,356
+jarvis/tools/base.py,sha256=rYLrNEpIdmrDm1WavOTyldXZ5dZWVRAmukiMcX40UZU,4436
+jarvis/tools/file_ops.py,sha256=05Vc4u-NGMjLD15WI52eL_nt_RyYt-Z_cXJnnBepf-c,3781
+jarvis/tools/search.py,sha256=dyJmeP_s2tWv5KQejOl-TuVF12B6SXViiXEyTemD1Wo,1412
+jarvis/tools/shell.py,sha256=P0rSaXt-GOXX_-XwkpiLpOXt8Thou4835xGAedxGoGA,2926
+jarvis/tools/sub_agent.py,sha256=t4Lb61_rLyjrbpL4p17Fd1-Or_K12zsNGkAXbXmZPic,3756
+jarvis/tools/user_input.py,sha256=ux3-FE4m5LJIQaoR4qYWeabBK2t0Dk0Irt6_96qUMIc,2313
+jarvis/tools/webpage.py,sha256=UTEomu5j7jbOw8c5az2jsjv5E7LeokWKj1QahvZO7xc,3077
+jarvis/tools/__pycache__/__init__.cpython-313.pyc,sha256=aptY-PF6DpP4xXXHRVfjPV1jQGXAhqcBaTv_ZNei2FA,497
+jarvis/tools/__pycache__/base.cpython-313.pyc,sha256=fEj3a31NMJpaZug4nohtE9OnSCfM3YtzclYYTJvNYJg,6632
+jarvis/tools/__pycache__/file_ops.cpython-313.pyc,sha256=bawv11xhWSj5wCtW0QeJLI-InhUBUXaSkogq6rZzvTQ,3636
+jarvis/tools/__pycache__/python_script.cpython-313.pyc,sha256=8JpryqTovEiTvBlWAK1KjZmPvHUuPc9GT9rTXBEQoJc,6693
+jarvis/tools/__pycache__/rag.cpython-313.pyc,sha256=JH6-PSZRMKAvTZqCwlRXJGClxYXNMs-vetU0q7hBLz0,6064
+jarvis/tools/__pycache__/search.cpython-313.pyc,sha256=VX9zztOwIsPCkYwev0A0XJGyu4Tr0PQcQkbbqx8vfB0,1870
+jarvis/tools/__pycache__/shell.cpython-313.pyc,sha256=0B-mfYwkkZCCCfBQV6Mut0S2ZtW2W4Cykh4rP2W6LJM,3652
+jarvis/tools/__pycache__/sub_agent.cpython-313.pyc,sha256=vkTS0oWAU-We7j7tPs7IbvDcsLh2QNJVQ9Kv_LiMnUs,4021
+jarvis/tools/__pycache__/user_confirmation.cpython-313.pyc,sha256=wK3Ev10lHSUSRvoYmi7A0GzxYkzU-C4Wfhs5qW_HBqs,2271
+jarvis/tools/__pycache__/user_input.cpython-313.pyc,sha256=JjTFOhObKsKF4Pn8KBRuKfV1_Ssj083fjU7Mfc_5z7c,2531
+jarvis/tools/__pycache__/user_interaction.cpython-313.pyc,sha256=RuVZ-pmiPBDywY3efgXSfohMAciC1avMGPmBK5qlnew,3305
+jarvis/tools/__pycache__/webpage.cpython-313.pyc,sha256=VcpkaV8IyOOtebedXjn8ybjr8AglU-3-Cs80yzE4FYo,3628
+jarvis_ai_assistant-0.1.6.dist-info/METADATA,sha256=3ACjmVA3LVSxIHOWf5hHlm5kEvy9mSOqeSgVgIcYJ7s,3690
+jarvis_ai_assistant-0.1.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jarvis_ai_assistant-0.1.6.dist-info/entry_points.txt,sha256=iKu7OMfew9dtfGhW71gIMTg4wvafuPqKb4wyQOnMAGU,44
+jarvis_ai_assistant-0.1.6.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.6.dist-info/RECORD,,

{jarvis_ai_assistant-0.1.3.dist-info → jarvis_ai_assistant-0.1.6.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.7.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

jarvis/.jarvis

@@ -1 +0,0 @@
-- 查看当前git仓库状态，如果有未提交的变更，自动根据当前仓库的文件修改内容总结生成一个git commit，然后执行git push推送到远端仓库，全流程不要询问用户

jarvis/tools/python_script.py

@@ -1,150 +0,0 @@
-import os
-import time
-import subprocess
-import sys
-import pkgutil
-from pathlib import Path
-from typing import Dict, Any, Optional
-from ..utils import PrettyOutput, OutputType
-
-class PythonScript:
-    """Python脚本管理类"""
-    SCRIPTS_DIR = "/tmp/ai_scripts"
-
-    @classmethod
-    def init_scripts_dir(cls):
-        """初始化脚本目录"""
-        Path(cls.SCRIPTS_DIR).mkdir(parents=True, exist_ok=True)
-
-    @classmethod
-    def generate_script_path(cls, name: Optional[str] = None) -> str:
-        """生成脚本文件路径"""
-        if name:
-            safe_name = "".join(c for c in name if c.isalnum() or c in "._- ")
-            filename = f"{int(time.time())}_{safe_name}.py"
-        else:
-            filename = f"{int(time.time())}_script.py"
-        return str(Path(cls.SCRIPTS_DIR) / filename)
-
-class PythonScriptTool:
-    name = "execute_python"
-    description = """Execute Python code and return the results.
-    Notes:
-    1. Use print() to output results
-    2. Automatic dependency management
-    3. Code saved to temporary file
-    """
-    parameters = {
-        "type": "object",
-        "properties": {
-            "code": {
-                "type": "string",
-                "description": "Python code to execute"
-            },
-            "dependencies": {
-                "type": "array",
-                "items": {"type": "string"},
-                "description": "Python package dependencies",
-                "default": []
-            },
-            "name": {
-                "type": "string",
-                "description": "Script name",
-                "default": ""
-            }
-        },
-        "required": ["code"]
-    }
-
-    @staticmethod
-    def _is_builtin_package(package_name: str) -> bool:
-        package_name = package_name.split("==")[0].strip()
-        if hasattr(sys.modules, package_name) or package_name in sys.stdlib_module_names:
-            return True
-        try:
-            return pkgutil.find_spec(package_name) is not None
-        except Exception:
-            return False
-
-    def execute(self, args: Dict) -> Dict[str, Any]:
-        """执行Python代码"""
-        try:
-            code = args["code"]
-            dependencies = args.get("dependencies", [])
-            script_name = args.get("name", "")
-            
-            # 初始化脚本目录
-            PythonScript.init_scripts_dir()
-            
-            # 生成脚本路径
-            script_path = PythonScript.generate_script_path(script_name)
-            
-            # 安装依赖
-            missing_deps = []
-            for dep in dependencies:
-                if not self._is_builtin_package(dep):
-                    missing_deps.append(dep)
-            
-            if missing_deps:
-                PrettyOutput.print("正在安装依赖...", OutputType.INFO)
-                try:
-                    subprocess.run(
-                        [sys.executable, "-m", "pip", "install"] + missing_deps,
-                        check=True,
-                        capture_output=True,
-                        text=True
-                    )
-                    PrettyOutput.print("依赖安装完成", OutputType.INFO)
-                except subprocess.CalledProcessError as e:
-                    return {
-                        "success": False,
-                        "error": f"依赖安装失败: {e.stderr}"
-                    }
-            
-            # 写入脚本文件
-            with open(script_path, 'w', encoding='utf-8') as f:
-                f.write(code)
-            
-            # 执行脚本
-            PrettyOutput.print(f"执行脚本: {script_path}", OutputType.INFO)
-            result = subprocess.run(
-                [sys.executable, script_path],
-                capture_output=True,
-                text=True
-            )
-            
-            # 构建输出
-            output = []
-            
-            # 添加脚本信息
-            output.append(f"脚本路径: {script_path}")
-            if dependencies:
-                output.append(f"依赖项: {', '.join(dependencies)}")
-            output.append("")
-            
-            # 添加执行结果
-            if result.stdout:
-                output.append("输出:")
-                output.append(result.stdout)
-            
-            # 添加错误信息（如果有）
-            if result.stderr:
-                output.append("错误:")
-                output.append(result.stderr)
-            
-            # 添加返回码
-            output.append(f"返回码: {result.returncode}")
-            
-            return {
-                "success": result.returncode == 0,
-                "stdout": "\n".join(output),
-                "stderr": result.stderr,
-                "return_code": result.returncode,
-                "script_path": script_path
-            }
-            
-        except Exception as e:
-            return {
-                "success": False,
-                "error": f"执行Python代码失败: {str(e)}"
-            }