From 1e1a36be7987c07771f27bbc196ff95a1cf08a53 Mon Sep 17 00:00:00 2001 From: xzx3344521 Date: Mon, 3 Nov 2025 18:15:46 +0800 Subject: [PATCH] Update ai --- ai | 425 +++++++++++++++++-------------------------------------------- 1 file changed, 119 insertions(+), 306 deletions(-) diff --git a/ai b/ai index 8ddedee..0cba152 100644 --- a/ai +++ b/ai @@ -1,342 +1,155 @@ #!/bin/bash -# 交互式AI驱动问题修复系统 -# 让用户清楚知道运行状态和位置 - -set -e - -# 配置 -DEEPSEEK_API_KEY="sk-your-api-key-here" # 请替换为你的真实API密钥 -SCRIPT_NAME="interactive_ai_fixer.sh" -INSTALL_DIR="/usr/local/bin" -LOG_FILE="/var/log/ai_interactive_fixer.log" +# 简单的AI连接测试脚本 +# 让你确认是否已经成功接入AI # 颜色定义 RED='\033[0;31m' GREEN='\033[0;32m' YELLOW='\033[1;33m' BLUE='\033[0;34m' -PURPLE='\033[0;35m' CYAN='\033[0;36m' NC='\033[0m' -# 显示运行位置和状态 -show_status() { - echo -e "${PURPLE}" - echo "╔══════════════════════════════════════════════════╗" - echo "║ AI驱动交互式修复系统 ║" - echo "╠══════════════════════════════════════════════════╣" - echo "║ 运行位置: $(pwd) ║" - echo "║ 脚本名称: $0 ║" - echo "║ 系统时间: $(date) ║" - echo "║ 当前用户: $(whoami) ║" - echo "║ 系统信息: $(uname -srm) ║" - echo "╚══════════════════════════════════════════════════╝" - echo -e "${NC}" - echo "" -} +# 显示标题 +echo -e "${CYAN}" +echo "╔══════════════════════════════════════╗" +echo "║ AI连接测试工具 ║" +echo "║ 确认你是否已接入AI助手 ║" +echo "╚══════════════════════════════════════╝" +echo -e "${NC}" -# 日志函数 -log() { - local message="$1" - echo "$(date '+%Y-%m-%d %H:%M:%S') - $message" | sudo tee -a "$LOG_FILE" - echo -e "${BLUE}[AI系统]${NC} $message" -} +# 配置API密钥（请先修改这里！） +API_KEY="sk-61d6716fe6b2452d94a0cee3bc5c4e2e" -# 进度显示函数 -show_progress() { - local step="$1" - local total="$2" - local message="$3" +# 测试函数：简单的AI对话 +test_ai_connection() { + echo -e "${BLUE}🔄 尝试连接AI助手...${NC}" + echo -e "${YELLOW}📡 发送测试请求到DeepSeek API...${NC}" - local percent=$((step * 100 / total)) - echo -e "${CYAN}📊 进度: [$step/$total] $message...${NC}" - echo -e "${YELLOW} ▰▰▰▰▰▰▰▰▰▰ $percent% 完成${NC}" - echo "" -} - -# 调用AI分析问题 -ask_ai() { - local problem="$1" - local context="$2" - - log "🤖 咨询AI解决方案: $problem" - echo -e "${CYAN}🔄 正在连接AI助手分析问题...${NC}" - - # 准备API请求 - local request_data=$(cat << EOF -{ - "model": "deepseek-coder", - "messages": [ - { - "role": "system", - "content": "你是一个Linux系统和脚本问题专家。用户遇到了脚本执行问题，请直接给出具体的修复命令。用中文回答，要具体可执行。" - }, - { - "role": "user", - "content": "问题：$problem\n\n上下文：$context\n\n请直接给出需要执行的Linux命令：" - } - ], - "temperature": 0.3, - "max_tokens": 1000 -} -EOF - ) - - # 调用API - echo -e "${YELLOW}📡 发送请求到AI服务器...${NC}" - local response=$(curl -s -X POST "https://api.deepseek.com/v1/chat/completions" \ + # 简单的测试请求 + response=$(curl -s -X POST "https://api.deepseek.com/v1/chat/completions" \ -H "Content-Type: application/json" \ - -H "Authorization: Bearer $DEEPSEEK_API_KEY" \ - -d "$request_data" 2>/dev/null || echo "AI请求失败") - - # 提取回复 - local answer=$(echo "$response" | grep -o '"content":"[^"]*"' | cut -d'"' -f4) + -H "Authorization: Bearer $API_KEY" \ + -d '{ + "model": "deepseek-coder", + "messages": [ + { + "role": "system", + "content": "你是一个AI助手。请用中文简短回复，确认连接成功。" + }, + { + "role": "user", + "content": "请回复'随机50个字。'来确认我们的连接正常。" + } + ], + "max_tokens": 50, + "temperature": 0.1 + }') - if [ -z "$answer" ] || [ "$answer" = "AI请求失败" ]; then - echo -e "${RED}❌ AI连接失败，使用备用方案${NC}" + # 检查响应 + if [ $? -ne 0 ]; then + echo -e "${RED}❌ 网络连接失败${NC}" return 1 fi - echo -e "${GREEN}✅ AI回复接收成功${NC}" - echo "$answer" -} - -# 执行命令并显示结果 -execute_command() { - local command="$1" - local description="$2" + # 提取AI回复 + ai_reply=$(echo "$response" | grep -o '"content":"[^"]*"' | cut -d'"' -f4) - log "执行: $description" - echo -e "${YELLOW}▶️ 执行: $command${NC}" - echo -e "${BLUE}📝 描述: $description${NC}" - - # 显示执行开始时间 - local start_time=$(date +%s) - - # 执行命令 - echo -e "${CYAN}────────────────────────────────────────────${NC}" - eval "$command" 2>&1 | while IFS= read -r line; do - echo " 📋 $line" - done - echo -e "${CYAN}────────────────────────────────────────────${NC}" - - local exit_code=${PIPESTATUS[0]} - local end_time=$(date +%s) - local duration=$((end_time - start_time)) - - if [ $exit_code -eq 0 ]; then - echo -e "${GREEN}✅ 执行成功 (耗时: ${duration}s)${NC}" - else - echo -e "${RED}❌ 执行失败 (退出码: $exit_code, 耗时: ${duration}s)${NC}" - fi - - echo "" - return $exit_code -} - -# 分析当前问题 -analyze_problem() { - show_progress 1 5 "分析系统环境" - - echo -e "${PURPLE}🔍 开始系统分析...${NC}" - - # 收集系统信息 - execute_command "uname -a" "获取系统内核信息" - execute_command "cat /etc/os-release" "获取操作系统信息" - execute_command "python3 --version 2>/dev/null || echo 'Python3 未安装'" "检查Python环境" - execute_command "bash --version | head -1" "检查Bash版本" - - show_progress 2 5 "测试问题脚本" - - # 测试问题脚本 - local test_url="https://github.com/xzx3344521/dock/raw/refs/heads/main/ai" - echo -e "${CYAN}🌐 测试问题脚本URL: $test_url${NC}" - - execute_command "curl -s -I '$test_url' | head -1" "检查URL可访问性" - - local script_content=$(curl -sSL "$test_url" 2>/dev/null || echo "下载失败") - echo -e "${YELLOW}📄 脚本内容预览:${NC}" - echo "$script_content" | head -10 | while IFS= read -r line; do - echo " 📃 $line" - done - - local problem_context="系统分析完成，检测到远程脚本执行问题" - echo "$problem_context" -} - -# 主要修复流程 -main_repair() { - show_status - - echo -e "${GREEN}🚀 启动AI自动修复流程...${NC}" - echo -e "${YELLOW}⚠️ 注意: 整个流程将由AI自动控制${NC}" - echo "" - - # 分析问题 - local problem_context=$(analyze_problem) - local problem_description="用户执行远程脚本失败，脚本内容被显示而不是执行，出现命令未找到错误" - - show_progress 3 5 "咨询AI解决方案" - - # 询问AI解决方案 - echo -e "${PURPLE}🤖 正在咨询AI助手...${NC}" - local ai_solution=$(ask_ai "$problem_description" "$problem_context") - - if [ $? -ne 0 ] || [ -z "$ai_solution" ]; then - echo -e "${RED}❌ AI咨询失败，启用备用方案${NC}" - backup_solution - return - fi - - show_progress 4 5 "执行AI方案" - - echo -e "${GREEN}🤖 AI提供的解决方案:${NC}" - echo -e "${CYAN}$ai_solution${NC}" - echo "" - - # 执行AI命令 - execute_ai_commands "$ai_solution" - - show_progress 5 5 "完成修复" - - echo -e "${GREEN}🎉 AI修复流程完成！${NC}" -} - -# 执行AI返回的命令 -execute_ai_commands() { - local solution="$1" - local command_count=0 - - echo -e "${PURPLE}🔧 开始执行AI建议的命令...${NC}" - - while IFS= read -r line; do - # 跳过空行和注释 - if [[ -z "$line" || "$line" =~ ^# ]]; then - continue + if [ -z "$ai_reply" ]; then + error_msg=$(echo "$response" | grep -o '"message":"[^"]*"' | cut -d'"' -f4) + if [ -n "$error_msg" ]; then + echo -e "${RED}❌ API错误: $error_msg${NC}" + else + echo -e "${RED}❌ 无法解析AI响应${NC}" fi - - # 识别命令 - if [[ "$line" =~ ^(sudo\s+|apt\s+|curl\s+|wget\s+|python|bash\s+|chmod\s+|mkdir\s+|cd\s+|echo\s+|pip\s+) ]]; then - command_count=$((command_count + 1)) - execute_command "$line" "AI建议命令 #$command_count" - fi - done <<< "$solution" + return 1 + fi - echo -e "${GREEN}✅ 共执行了 $command_count 个AI命令${NC}" + echo -e "${GREEN}✅ AI回复: $ai_reply${NC}" + return 0 } -# 备用解决方案 -backup_solution() { - echo -e "${YELLOW}🛠️ 启用备用修复方案...${NC}" +# 检查API密钥 +check_api_key() { + echo -e "${YELLOW}🔑 检查API密钥配置...${NC}" - show_progress 1 3 "下载和分析脚本" - execute_command "curl -sSL -o /tmp/problem_script 'https://github.com/xzx3344521/dock/raw/refs/heads/main/ai'" "下载问题脚本" - execute_command "file /tmp/problem_script" "检测脚本类型" - execute_command "head -3 /tmp/problem_script" "查看脚本头部" + if [ "$API_KEY" = "sk-your-api-key-here" ] || [ -z "$API_KEY" ]; then + echo -e "${RED}❌ 请先设置你的DeepSeek API密钥！${NC}" + echo "" + echo -e "${CYAN}修改方法:${NC}" + echo "1. 打开脚本: nano ai_test.sh" + echo "2. 找到 API_KEY= 这一行" + echo "3. 替换为你的真实API密钥" + echo "4. 保存并重新运行脚本" + echo "" + return 1 + fi - show_progress 2 3 "准备执行环境" - execute_command "sudo apt update" "更新软件包列表" - execute_command "sudo apt install -y python3 python3-pip curl wget file" "安装必要工具" + if [[ "$API_KEY" != sk-* ]]; then + echo -e "${RED}❌ API密钥格式不正确${NC}" + return 1 + fi - show_progress 3 3 "执行修复" - # 根据脚本类型执行 - if head -1 /tmp/problem_script 2>/dev/null | grep -q "python"; then - execute_command "python3 /tmp/problem_script" "执行Python脚本" + echo -e "${GREEN}✅ API密钥格式正确${NC}" + return 0 +} + +# 系统信息显示 +show_system_info() { + echo -e "${CYAN}🖥️ 系统信息:${NC}" + echo "主机名: $(hostname)" + echo "系统: $(uname -srm)" + echo "时间: $(date)" + echo "工作目录: $(pwd)" + echo "" +} + +# 主测试流程 +main_test() { + echo -e "${BLUE}开始AI连接测试...${NC}" + echo "" + + # 显示系统信息 + show_system_info + + # 检查API密钥 + if ! check_api_key; then + exit 1 + fi + + echo -e "${YELLOW}⏳ 正在进行连接测试...${NC}" + echo "" + + # 测试AI连接 + if test_ai_connection; then + echo "" + echo -e "${GREEN}🎉 恭喜！AI连接测试成功！${NC}" + echo -e "${GREEN}🤖 你现在已经正式接入AI助手了！${NC}" + echo "" + echo -e "${CYAN}接下来你可以:${NC}" + echo "✅ 运行完整的AI修复脚本" + echo "✅ 使用AI助手解决各种问题" + echo "✅ 享受AI驱动的自动化体验" else - execute_command "bash /tmp/problem_script" "执行Bash脚本" + echo "" + echo -e "${RED}😞 AI连接测试失败${NC}" + echo -e "${YELLOW}请检查:${NC}" + echo "🔹 API密钥是否正确" + echo "🔹 网络连接是否正常" + echo "🔹 账户余额是否充足" + echo "🔹 API密钥是否有访问权限" fi } -# 安装为系统命令 -install_systemwide() { - echo -e "${PURPLE}📥 安装到系统命令...${NC}" - - show_status - - execute_command "sudo cp '$0' '$INSTALL_DIR/ai-fix'" "复制脚本到系统目录" - execute_command "sudo chmod +x '$INSTALL_DIR/ai-fix'" "设置执行权限" - execute_command "sudo chown root:root '$INSTALL_DIR/ai-fix'" "设置所有权" - - echo -e "${GREEN}✅ 安装完成！现在你可以使用以下命令：${NC}" - echo -e "${CYAN} ai-fix repair - AI修复问题${NC}" - echo -e "${CYAN} ai-fix status - 显示状态${NC}" - echo -e "${CYAN} ai-fix log - 查看日志${NC}" -} - -# 交互式菜单 -interactive_menu() { - while true; do - show_status - - echo -e "${GREEN}请选择操作:${NC}" - echo -e "${CYAN}1. 🚀 AI自动修复问题${NC}" - echo -e "${CYAN}2. 📥 安装到系统命令${NC}" - echo -e "${CYAN}3. 📊 显示系统状态${NC}" - echo -e "${CYAN}4. 📋 查看执行日志${NC}" - echo -e "${CYAN}5. 🚪 退出${NC}" - echo "" - - read -p "请输入选择 (1-5): " choice - - case $choice in - 1) - main_repair - ;; - 2) - install_systemwide - ;; - 3) - show_status - ;; - 4) - execute_command "cat '$LOG_FILE' | tail -20" "显示最近日志" - ;; - 5) - echo -e "${GREEN}👋 再见！${NC}" - exit 0 - ;; - *) - echo -e "${RED}❌ 无效选择，请重新输入${NC}" - ;; - esac - - echo "" - read -p "按回车键继续..." - clear - done -} - -# 命令行参数处理 +# 交互式选项 case "${1:-}" in - "repair"|"fix") - main_repair - ;; - "install") - install_systemwide - ;; - "status") - show_status - ;; - "log") - execute_command "cat '$LOG_FILE'" "显示完整日志" - ;; - "menu"|"interactive") - interactive_menu + "help") + echo -e "${CYAN}使用方法:${NC}" + echo " ./ai_test.sh - 运行AI连接测试" + echo " ./ai_test.sh help - 显示帮助信息" ;; *) - echo -e "${GREEN}🤖 AI驱动交互式修复系统${NC}" - echo "" - echo -e "${CYAN}运行位置: $(pwd)${NC}" - echo -e "${CYAN}脚本路径: $0${NC}" - echo "" - echo -e "${YELLOW}使用方法:${NC}" - echo " $0 repair - 🚀 AI自动修复问题" - echo " $0 install - 📥 安装到系统命令" - echo " $0 status - 📊 显示系统状态" - echo " $0 log - 📋 查看执行日志" - echo " $0 menu - 🖥️ 交互式菜单" - echo "" - echo -e "${PURPLE}推荐使用: $0 menu 进入交互模式${NC}" + main_test ;; esac