feat: AI全代理模式 + 版本自动检查（v1.5.4）

Author: zcj220

extension/package.json

@@ -2,7 +2,7 @@
   "name": "testpilot-ai",
   "displayName": "TestPilot AI",
   "description": "AI-powered automated testing robot — operates UI like a human, finds bugs, and auto-fixes them. Supports Web, Android, iOS, Mini Program & Desktop.",
-  "version": "1.5.3",
+  "version": "1.5.4",
   "publisher": "wenzhouxinzao",
   "engines": {
     "vscode": "^1.85.0"

extension/src/engineManager.ts

@@ -108,6 +108,9 @@ export class EngineManager {
    * 流程：已运行 → 直接返回；已缓存 → 启动；未缓存 → 先下载再启动。
    */
   async ensureRunning(): Promise<void> {
+    // 0. 版本检查（异步，不阻塞启动，但强制更新会提示）
+    this._checkVersion().catch(() => {});
+
     // 1. 已有引擎在跑，直接复用
     if (await this.isEngineRunning()) {
       this._outputChannel.appendLine("[引擎] 检测到引擎已运行，直接连接");
@@ -259,4 +262,63 @@ export class EngineManager {
     }
     this._setStatus("offline");
   }
+
+  /** 版本检查：低于最低版本强制提示，低于最新版本弱提示 */
+  private async _checkVersion(): Promise<void> {
+    const VERSION_URL = "https://xinzaoai.com/api/version/check";
+    const currentVersion = vscode.extensions.getExtension("wenzhouxinzao.testpilot-ai")
+      ?.packageJSON?.version as string | undefined;
+    if (!currentVersion) { return; }
+
+    try {
+      const data = await new Promise<{ latest: string; minimum: string; changelog?: string }>(
+        (resolve, reject) => {
+          const mod = VERSION_URL.startsWith("https") ? https : http;
+          mod.get(VERSION_URL, { timeout: 8000 }, (res) => {
+            if (res.statusCode !== 200) { reject(new Error(`HTTP ${res.statusCode}`)); return; }
+            let body = "";
+            res.on("data", (c: Buffer) => (body += c.toString()));
+            res.on("end", () => { try { resolve(JSON.parse(body)); } catch (e) { reject(e); } });
+          }).on("error", reject).on("timeout", () => reject(new Error("timeout")));
+        }
+      );
+
+      const cmp = (a: string, b: string) => {
+        const pa = a.split(".").map(Number);
+        const pb = b.split(".").map(Number);
+        for (let i = 0; i < 3; i++) {
+          if ((pa[i] || 0) < (pb[i] || 0)) { return -1; }
+          if ((pa[i] || 0) > (pb[i] || 0)) { return 1; }
+        }
+        return 0;
+      };
+
+      if (cmp(currentVersion, data.minimum) < 0) {
+        // 强制更新：弹出模态提示
+        const action = await vscode.window.showErrorMessage(
+          `TestPilot AI 当前版本 v${currentVersion} 已不再支持，请更新至 v${data.latest}。`,
+          { modal: true },
+          "前往下载"
+        );
+        if (action === "前往下载") {
+          vscode.env.openExternal(
+            vscode.Uri.parse("https://xinzaoai.com/downloads/testpilot-ai-" + data.latest + ".vsix")
+          );
+        }
+      } else if (cmp(currentVersion, data.latest) < 0) {
+        // 弱提示：状态栏提醒，不阻塞
+        const action = await vscode.window.showInformationMessage(
+          `TestPilot AI 有新版本 v${data.latest}（当前 v${currentVersion}）`,
+          "下载更新", "忽略"
+        );
+        if (action === "下载更新") {
+          vscode.env.openExternal(
+            vscode.Uri.parse("https://xinzaoai.com/downloads/testpilot-ai-" + data.latest + ".vsix")
+          );
+        }
+      }
+    } catch {
+      // 版本检查失败不影响正常使用，静默忽略
+    }
+  }
 }

src/app.py

@@ -18,7 +18,7 @@
 from src.core.config import PROJECT_ROOT
 from src.api.websocket import ws_manager
 from src.browser.automator import BrowserAutomator
-from src.core.ai_client import AIClient
+from src.core.ai_client import AIClient, ProxyAIClient
 from src.core.config import get_config
 from src.core.logger import setup_logger
 from src.memory.store import MemoryStore
@@ -52,15 +52,23 @@ def create_app() -> FastAPI:
     # 记忆系统（SQLite 本地存储，零外部依赖）
     memory_store = MemoryStore()
 
-    # AI 客户端（API Key 未配置时优雅降级，测试任务端点将返回503）
-    ai_client: AIClient | None = None
+    # AI 客户端：有 TP_AI_API_KEY 时直连豆包（开发模式），否则走代理（生产/用户端）
+    ai_client: AIClient | ProxyAIClient | None = None
     if config.ai.api_key:
         try:
             ai_client = AIClient(config.ai)
         except Exception as e:
-            logger.warning("AI 客户端初始化失败（测试任务不可用）: {}", e)
+            logger.warning("AI 客户端初始化失败，降级为代理模式: {}", e)
+            ai_client = ProxyAIClient(
+                reasoning_effort=config.ai.reasoning_effort,
+                max_tokens=config.ai.max_completion_tokens,
+            )
     else:
-        logger.warning("TP_AI_API_KEY 未配置，测试任务功能不可用")
+        logger.info("TP_AI_API_KEY 未配置，使用代理模式（xinzaoai.com）")
+        ai_client = ProxyAIClient(
+            reasoning_effort=config.ai.reasoning_effort,
+            max_tokens=config.ai.max_completion_tokens,
+        )
 
     @asynccontextmanager
     async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:

src/core/ai_client.py

@@ -11,9 +11,11 @@
 """
 
 import base64
+import json
 from pathlib import Path
 from typing import Optional
 
+import httpx
 from loguru import logger
 from openai import OpenAI
 
@@ -293,3 +295,109 @@ def _call_chat(
                     message="AI API 调用失败",
                     detail=error_msg,
                 )
+
+
+# 代理服务器地址和鉴权 secret（后端随时可切换模型，引擎无需更新）
+_PROXY_URL = "https://xinzaoai.com/api/ai/proxy"
+_ENGINE_SECRET = "testpilot-engine-secret-2026"
+
+
+class ProxyAIClient:
+    """通过 xinzaoai.com 服务器代理调用豆包 AI。
+
+    引擎 exe 默认使用此客户端（不需要 TP_AI_API_KEY）。
+    后端切换模型/Key 只需改服务器环境变量，引擎无需重新打包。
+    接口与 AIClient 完全相同，可互相替换。
+    """
+
+    def __init__(self, reasoning_effort: str = "medium", max_tokens: int = 65535) -> None:
+        self._reasoning_effort = reasoning_effort
+        self._max_tokens = max_tokens
+        self._http = httpx.Client(timeout=120.0)
+        logger.info("AI 代理客户端初始化完成 | 代理={}", _PROXY_URL)
+
+    def chat(
+        self,
+        prompt: str,
+        system_prompt: str = "",
+        reasoning_effort: Optional[str] = None,
+        timeout: Optional[float] = None,
+    ) -> str:
+        messages = []
+        if system_prompt:
+            messages.append({"role": "system", "content": system_prompt})
+        messages.append({"role": "user", "content": prompt})
+        return self._call(messages, reasoning_effort, timeout)
+
+    def analyze_screenshot(
+        self,
+        image_path: str,
+        prompt: str = "请描述这个页面的内容，并指出可能存在的UI问题或Bug。",
+        system_prompt: str = "",
+        reasoning_effort: Optional[str] = None,
+        timeout: Optional[float] = None,
+        max_tokens: Optional[int] = None,
+    ) -> str:
+        path = Path(image_path)
+        if not path.exists():
+            raise FileNotFoundError(f"截图文件不存在: {image_path}")
+        image_data = path.read_bytes()
+        b64 = base64.b64encode(image_data).decode("utf-8")
+        suffix = path.suffix.lower()
+        mime = {"png": "image/png", "jpg": "image/jpeg", "jpeg": "image/jpeg",
+                "webp": "image/webp", "gif": "image/gif"}.get(suffix.lstrip("."), "image/png")
+        messages = []
+        if system_prompt:
+            messages.append({"role": "system", "content": system_prompt})
+        messages.append({"role": "user", "content": [
+            {"type": "image_url", "image_url": {"url": f"data:{mime};base64,{b64}"}},
+            {"type": "text", "text": prompt},
+        ]})
+        logger.info("发送截图分析请求（代理）| 文件={} | 大小={}KB", path.name, len(image_data) // 1024)
+        return self._call(messages, reasoning_effort, timeout, max_tokens)
+
+    def analyze_screenshot_url(
+        self,
+        image_url: str,
+        prompt: str = "请描述这个页面的内容，并指出可能存在的UI问题或Bug。",
+        system_prompt: str = "",
+        reasoning_effort: Optional[str] = None,
+    ) -> str:
+        messages = []
+        if system_prompt:
+            messages.append({"role": "system", "content": system_prompt})
+        messages.append({"role": "user", "content": [
+            {"type": "image_url", "image_url": {"url": image_url}},
+            {"type": "text", "text": prompt},
+        ]})
+        return self._call(messages, reasoning_effort)
+
+    def _call(
+        self,
+        messages: list[dict],
+        reasoning_effort: Optional[str] = None,
+        timeout: Optional[float] = None,
+        max_tokens: Optional[int] = None,
+    ) -> str:
+        payload = {
+            "messages": messages,
+            "reasoning_effort": reasoning_effort or self._reasoning_effort,
+            "max_tokens": max_tokens or self._max_tokens,
+        }
+        try:
+            resp = self._http.post(
+                _PROXY_URL,
+                json=payload,
+                headers={"X-Engine-Secret": _ENGINE_SECRET},
+                timeout=timeout or 120.0,
+            )
+            resp.raise_for_status()
+            data = resp.json()
+            content = data["choices"][0]["message"]["content"]
+            if not content:
+                raise AIResponseError(message="代理 AI 返回了空响应", detail=str(data))
+            return content
+        except AIError:
+            raise
+        except Exception as e:
+            raise AIError(message="代理 AI 请求失败", detail=str(e))