ai-lab-bridge 0.1.3 → 0.1.5

package/index.js

@@ -136,6 +136,7 @@ const BRIDGE_VERSION = (() => {
 // 连接参数由各子命令解析填充（connect/daemon 需要，status/stop 不需要）
 let SERVER = "";
 let TOKEN = "";
+let sessionCwd = ""; // 当前任务的实际工作目录（emitSession 回传用）
 
 // ---------------------------------------------------------------------------
 // CLI 能力探测
@@ -234,7 +235,11 @@ class Session {
   emitSession(cliSessionId) {
     if (!cliSessionId || this.cliSessionId) return;
     this.cliSessionId = cliSessionId;
-    this.emitEvent("session", { cli_session_id: cliSessionId });
+    this.emitEvent("session", {
+      cli_session_id: cliSessionId,
+      // 设备实际执行的工作目录（CLI cwd 或 bridge 进程启动目录），平台前端展示用
+      working_dir: sessionCwd || null,
+    });
   }
 
   /**
@@ -262,11 +267,26 @@ class Session {
     }
   }
 
+  /** 归一化后缓存 + 回传 usage：各家 CLI 字段名不同（input_tokens/input/prompt_tokens...），统一为平台口径 */
   emitUsage(usage) {
-    if (usage && typeof usage === "object") {
-      this.usage = usage;
-      this.emitEvent("usage", { usage });
-    }
+    const num = (v) => (typeof v === "number" && isFinite(v) ? v : undefined);
+    const raw = usage && typeof usage === "object" ? usage : null;
+    if (!raw) return;
+    // opencode 系: {input, output, cache:{read}} / claude: {input_tokens, output_tokens, cache_read_input_tokens}
+    // codex: {input_tokens, output_tokens, total_tokens} / gemini: {inputTokenCount, outputTokenCount, ...}
+    const input = num(raw.input_tokens) ?? num(raw.input) ?? num(raw.prompt_tokens) ?? num(raw.inputTokenCount);
+    const output =
+      num(raw.output_tokens) ?? num(raw.output) ?? num(raw.completion_tokens) ?? num(raw.outputTokenCount) ?? num(raw.candidatesTokenCount);
+    const cacheRead = num(raw.cache_read_input_tokens) ?? num(raw.cache && raw.cache.read) ?? num(raw.cachedContentTokenCount);
+    if (input === undefined && output === undefined && cacheRead === undefined) return;
+    const normalized = {
+      input_tokens: input ?? null,
+      output_tokens: output ?? null,
+      total_tokens: input !== undefined || output !== undefined ? (input ?? 0) + (output ?? 0) : null,
+      cache_read_input_tokens: cacheRead ?? null,
+    };
+    this.usage = normalized;
+    this.emitEvent("usage", { usage: normalized });
   }
 
   emitToolUse(summary) {
@@ -467,6 +487,9 @@ function handleOpencodeLine(session, obj) {
       session.emitToolUse(part.tool || part.name || (state && state.title));
     }
   }
+  // usage 提取：opencode 系在 step_finish 的 part.tokens / 顶层 usage 携带本轮 token 统计
+  const usageRaw = (part && (part.tokens || part.usage)) || obj.tokens || obj.usage;
+  if (usageRaw && typeof usageRaw === "object") session.emitUsage(usageRaw);
   if (obj.type === "error" || (obj.error && typeof obj.error === "object")) {
     const err = obj.error && typeof obj.error === "object" ? obj.error : null;
     // 提取可读错误消息（opencode 系: error.data.message / error.message；附带 name/ref 便于定位）
@@ -546,8 +569,10 @@ function handleCursorLine(session, obj) {
 
 /**
  * Gemini CLI 家族事件解析（gemini / qwen / iflow 的 `-p --output-format stream-json`，JSONL）。
- * 事件类型：init(session id) / message / tool_use / tool_result / result(stats)。
- * 各 fork 字段命名可能有差异，做宽松归一化；终态以进程退出码兜底。
+ * 事件格式两代并存，宽松归一化：
+ *   - Gemini 原生：message(text/content) / tool_use / result(response, stats)
+ *   - Claude 风格（qwen 0.24+ 同源仿制）：assistant(message.content[]) / tool_call / result(result, usage)
+ * 终态以进程退出码兜底。
  */
 function handleGeminiLine(session, obj) {
   if (!obj || typeof obj !== "object") return;
@@ -557,28 +582,50 @@ function handleGeminiLine(session, obj) {
   if (obj.type === "message" || obj.type === "assistant") {
     const role = obj.role || (obj.message && obj.message.role);
     if (role === "user") return;
+    // 文本提取顺序：obj.text → obj.content → obj.message.content[]（Claude 风格嵌套）
     let text = obj.text;
-    if (!text && obj.content) {
-      if (typeof obj.content === "string") text = obj.content;
-      else if (Array.isArray(obj.content)) {
-        text = obj.content
-          .filter((b) => (b && b.type === "text") || typeof b === "string")
-          .map((b) => (typeof b === "string" ? b : b.text))
-          .join("");
+    let blocks = obj.content;
+    if (!blocks && obj.message && Array.isArray(obj.message.content)) {
+      blocks = obj.message.content;
+    }
+    if (!text && blocks) {
+      if (typeof blocks === "string") text = blocks;
+      else if (Array.isArray(blocks)) {
+        const parts = [];
+        for (const b of blocks) {
+          if (!b) continue;
+          if (b.type === "text" && b.text) {
+            session.endThink();
+            parts.push(b.text);
+          } else if (b.type === "thinking" && b.thinking) {
+            session.emitThink(b.thinking);
+          } else if (typeof b === "string") {
+            parts.push(b);
+          }
+        }
+        text = parts.join("");
       }
     }
     if (text) session.emitText(text);
   } else if (obj.type === "tool_use" || obj.type === "tool_call") {
     session.emitToolUse(obj.name || obj.tool_name || obj.tool || "tool");
   } else if (obj.type === "result") {
-    session.emitUsage(obj.stats);
-    if (obj.error) {
-      const msg = typeof obj.error === "string" ? obj.error : (obj.error.message || JSON.stringify(obj.error));
+    // usage：gemini 原生在 stats，qwen/claude 风格在 usage
+    session.emitUsage(obj.stats || obj.usage);
+    const failed = obj.error || obj.is_error || (obj.subtype && obj.subtype !== "success");
+    if (failed) {
+      const msg = typeof obj.error === "string"
+        ? obj.error
+        : (obj.error && obj.error.message)
+          || (typeof obj.result === "string" ? obj.result : undefined)
+          || `执行异常: ${obj.subtype || "unknown"}`;
       session.finish("failed", msg);
       return;
     }
-    if (typeof obj.response === "string" && obj.response && !session.textParts.length) {
-      session.emitText(obj.response);
+    // 终态文本兜底：gemini 原生在 response，qwen/claude 风格在 result
+    const finalText = (typeof obj.response === "string" && obj.response) || (typeof obj.result === "string" && obj.result);
+    if (finalText && !session.textParts.length) {
+      session.emitText(finalText);
     }
     session.finish("completed");
   }
@@ -720,6 +767,8 @@ function startSession(message) {
 
   const timeoutSeconds = Number(options.timeout_seconds) > 0 ? Number(options.timeout_seconds) : 0;
   const cwd = typeof options.cwd === "string" && options.cwd ? options.cwd : process.cwd();
+  // 供 emitSession 回传实际工作目录（Session 构造在 cwd 解析之前，用模块级变量传递）
+  sessionCwd = cwd;
   let argv;
 
   let promptViaStdin = true;
@@ -942,7 +991,8 @@ async function connect() {
       for (const cap of capabilities) {
         console.log(`[bridge]   ${cap.cli}: ${cap.available ? `可用 (${cap.version})` : "未安装"}`);
       }
-      send({ type: "register", capabilities });
+      // working_dir：bridge 进程启动目录，作为未配置 cwd 时任务的默认工作目录（平台前端展示）
+      send({ type: "register", capabilities, working_dir: process.cwd() });
       startHeartbeat();
     } else if (message.type === "registered") {
       console.log("[bridge] 能力注册完成，等待任务下发");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-lab-bridge",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "AI 开发平台设备桥接进程：反连平台 WebSocket，驱动本地 Claude Code / Codex CLI 执行任务并回传事件流",
   "license": "MIT",
   "files": [