chrome-ai-bridge 2.5.1 → 2.5.3

package/build/src/fast-cdp/fast-chat.js

@@ -61,8 +61,8 @@ function envWithFallback(newName, oldName, defaultVal) {
 }
 const CONNECT_REUSE_TIMEOUT_MS = Number(envWithFallback('CAI_CONNECT_REUSE_TIMEOUT_MS', 'MCP_CONNECT_REUSE_TIMEOUT_MS', '12000'));
 const CONNECT_NEWTAB_TIMEOUT_MS = Number(envWithFallback('CAI_CONNECT_NEWTAB_TIMEOUT_MS', 'MCP_CONNECT_NEWTAB_TIMEOUT_MS', '20000'));
-const TOOL_BUDGET_MS = Number(envWithFallback('CAI_TOOL_BUDGET_MS', 'CAI_MCP_TOOL_BUDGET_MS', '50000'));
-const RESPONSE_WAIT_MAX_MS = Number(process.env.CAI_RESPONSE_WAIT_MAX_MS || '40000');
+const TOOL_BUDGET_MS = Number(envWithFallback('CAI_TOOL_BUDGET_MS', 'CAI_MCP_TOOL_BUDGET_MS', '300000'));
+const RESPONSE_WAIT_MAX_MS = Number(process.env.CAI_RESPONSE_WAIT_MAX_MS || '300000');
 const BUDGET_RESERVE_MS = Number(envWithFallback('CAI_BUDGET_RESERVE_MS', 'CAI_MCP_BUDGET_RESERVE_MS', '3000'));
 function getRemainingBudgetMs(startMs, overrideBudgetMs) {
     return (overrideBudgetMs ?? TOOL_BUDGET_MS) - (nowMs() - startMs) - BUDGET_RESERVE_MS;
@@ -1037,21 +1037,33 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
     // 60秒 caller deadline を超えないよう、残り予算内で待機する。
     const maxWaitMs = getResponseWaitBudgetMs(t0, RESPONSE_WAIT_MAX_MS, 'chatgpt-response', budgetMs);
     const pollIntervalMs = 1000;
+    const IDLE_TIMEOUT_MS = 60000; // ストップボタン消失後、60秒間無活動でタイムアウト
     const startWait = Date.now();
+    let lastActivityAt = Date.now(); // 最後にストップボタンorテキスト成長を検出した時刻
     let lastLoggedState = '';
     let sawStopButton = false; // 生成中状態を検出したかどうか
     let streamingText = ''; // ストリーミング中に取得したテキスト（完了後に折りたたまれる対策）
     let textStableCount = 0; // テキスト長が安定した回数（2-poll confirmation）
     let lastTextLength = -1; // 前回のテキスト長
-    while (Date.now() - startWait < maxWaitMs) {
+    let stopButtonGoneCount = 0; // ストップボタンが連続不在のポール数（Thinkingフェーズ切替誤判定防止）
+    let textGrowingCount = 0; // テキストが成長中のポール数（成長中はフォールバック抑止）
+    // ストップボタンが見えている間は maxWaitMs を無視し、IDLE_TIMEOUT のみで判定する。
+    // sawStopButton=false の初期フェーズでは maxWaitMs も併用。
+    while (Date.now() - lastActivityAt < IDLE_TIMEOUT_MS &&
+        (sawStopButton || Date.now() - startWait < maxWaitMs)) {
         const state = await client.evaluate(`
         (() => {
           ${DOM_UTILS_CODE}
 
           // 停止ボタン検出（フォールバックセレクター付き）
           const stopBtn = document.querySelector('button[data-testid="stop-button"]') ||
+                          document.querySelector('button[aria-label="ストリーミングの停止"]') ||
+                          document.querySelector('button[aria-label="Stop streaming"]') ||
                           document.querySelector('button[aria-label*="停止"]') ||
-                          document.querySelector('button[aria-label*="Stop"]');
+                          document.querySelector('button[aria-label*="Stop"]') ||
+                          [...document.querySelectorAll('button')].find(b =>
+                            b.querySelector('rect') && (b.textContent || '').trim() === ''
+                          );
           const buttons = __collectDeep(['button', '[role="button"]']).nodes;
           // 送信ボタン検出（フォールバックセレクター付き）
           // 注意: 応答完了後は音声ボタンに置き換わり、送信ボタンがDOMから消える
@@ -1089,7 +1101,7 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
           // 「今すぐ回答」「Skip thinking」ボタンがある場合はThinking進行中
           const hasSkipThinkingButton = bodyText.includes('今すぐ回答') ||
                                         bodyText.includes('Skip thinking');
-          const isStillGenerating = (hasGeneratingText && !hasThinkingComplete) || hasSkipThinkingButton;
+          const isStillGenerating = Boolean(stopBtn) || (hasGeneratingText && !hasThinkingComplete) || hasSkipThinkingButton;
 
           // 最後のアシスタントメッセージに実際のテキストがあるかチェック
           // 旧UIセレクター + 新UI(article)の両方を試す
@@ -1302,14 +1314,16 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
         })()
       `);
         // stopボタンを検出したらフラグを立てる（生成が始まった証拠）
+        // ストップボタンが見えている間はアクティブとみなし、タイムアウトを延長し続ける
         if (state.hasStopButton) {
             sawStopButton = true;
+            lastActivityAt = Date.now();
         }
         // 状態が変化した場合のみログ出力
         const currentState = JSON.stringify(state);
         if (currentState !== lastLoggedState) {
             const elapsed = Math.round((Date.now() - startWait) / 1000);
-            console.error(`[ChatGPT] State @${elapsed}s: stop=${state.hasStopButton}, send=${state.sendButtonFound}(disabled=${state.sendButtonDisabled}), assistant=${state.assistantMsgCount}, inputHasText=${state.inputBoxHasText}, sawStop=${sawStopButton}, generating=${state.isStillGenerating}, skipThink=${state.hasSkipThinkingButton}, hasText=${state.hasResponseText}`);
+            console.error(`[ChatGPT] State @${elapsed}s: stop=${state.hasStopButton}, send=${state.sendButtonFound}(disabled=${state.sendButtonDisabled}), assistant=${state.assistantMsgCount}, inputHasText=${state.inputBoxHasText}, sawStop=${sawStopButton}, generating=${state.isStillGenerating}, skipThink=${state.hasSkipThinkingButton}, hasText=${state.hasResponseText}, textGrow=${textGrowingCount}`);
             lastLoggedState = currentState;
         }
         // 応答完了条件（Thinkingモード対応版）:
@@ -1321,16 +1335,31 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
         const currentTextLen = state.debug_lastAssistantInnerTextLen;
         if (currentTextLen === lastTextLength && currentTextLen > 0) {
             textStableCount++;
+            textGrowingCount = 0;
+        }
+        else if (currentTextLen > lastTextLength) {
+            textStableCount = 0;
+            textGrowingCount++;
+            lastTextLength = currentTextLen;
+            lastActivityAt = Date.now();
         }
         else {
             textStableCount = 0;
+            textGrowingCount = 0;
             lastTextLength = currentTextLen;
         }
-        if (sawStopButton && !state.hasStopButton && !state.inputBoxHasText &&
+        // ストップボタン不在の連続カウント（Thinkingフェーズ切替時の一瞬消失を除外）
+        if (sawStopButton && !state.hasStopButton) {
+            stopButtonGoneCount++;
+        }
+        else {
+            stopButtonGoneCount = 0;
+        }
+        if (sawStopButton && stopButtonGoneCount >= 3 && !state.inputBoxHasText &&
             state.assistantMsgCount > initialAssistantCount) {
             // 2-poll confirmation: テキスト長が2回連続安定してから完了とする
             if (textStableCount >= 2) {
-                console.error(`[ChatGPT] Response complete - stop gone, text stable for ${textStableCount} polls (len=${currentTextLen})`);
+                console.error(`[ChatGPT] Response complete - stop gone for ${stopButtonGoneCount} polls, text stable for ${textStableCount} polls (len=${currentTextLen})`);
                 streamingText = await client.evaluate(`
             (() => {
               const msgs = document.querySelectorAll('[data-message-author-role="assistant"]');
@@ -1352,7 +1381,7 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
                 break;
             }
             // まだ安定していない — 次のポールまで待機
-            console.error(`[ChatGPT] Stop button gone but text not stable yet (len=${currentTextLen}, stableCount=${textStableCount})`);
+            console.error(`[ChatGPT] Stop button gone (${stopButtonGoneCount} polls) but text not stable yet (len=${currentTextLen}, stableCount=${textStableCount})`);
             await new Promise(r => setTimeout(r, pollIntervalMs));
             continue;
         }
@@ -1360,7 +1389,8 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
         // （stopボタンを見逃した場合の救済）
         const elapsed = Date.now() - startWait;
         if (elapsed > 5000 && !state.hasStopButton && !state.inputBoxHasText &&
-            state.assistantMsgCount > initialAssistantCount && !state.isStillGenerating) {
+            state.assistantMsgCount > initialAssistantCount && !state.isStillGenerating &&
+            textGrowingCount === 0) {
             if (textStableCount >= 2) {
                 console.error(`[ChatGPT] Response complete - fallback after 5s, text stable (len=${currentTextLen}, stableCount=${textStableCount})`);
                 break;
@@ -1379,8 +1409,10 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
         }
         await new Promise(r => setTimeout(r, pollIntervalMs));
     }
-    // タイムアウトチェック
-    if (Date.now() - startWait >= maxWaitMs) {
+    // タイムアウトチェック（IDLE_TIMEOUT で抜けた場合も含む）
+    const loopElapsed = Date.now() - startWait;
+    const loopIdle = Date.now() - lastActivityAt;
+    if (loopIdle >= IDLE_TIMEOUT_MS || (!sawStopButton && loopElapsed >= maxWaitMs)) {
         const finalState = await client.evaluate(`
         (() => {
           // フォールバックセレクター付きの検出
@@ -1409,9 +1441,14 @@ async function askChatGPTFastInternal(question, debug, budgetMs) {
           };
         })()
       `);
-        console.error(`[ChatGPT] Timeout - final state: ${JSON.stringify(finalState)}`);
+        const elapsedMs = Date.now() - startWait;
+        const idleMs = Date.now() - lastActivityAt;
+        const reason = idleMs >= IDLE_TIMEOUT_MS
+            ? `idle for ${Math.round(idleMs / 1000)}s (no stop button or text growth)`
+            : `absolute ceiling ${maxWaitMs}ms reached`;
+        console.error(`[ChatGPT] Timeout - ${reason}, elapsed=${Math.round(elapsedMs / 1000)}s, final state: ${JSON.stringify(finalState)}`);
         await resetConnection('chatgpt');
-        throw new Error(`Timed out waiting for ChatGPT response (${maxWaitMs}ms). Final state: ${JSON.stringify(finalState)}`);
+        throw new Error(`Timed out waiting for ChatGPT response (${Math.round(elapsedMs / 1000)}s, ${reason}). Final state: ${JSON.stringify(finalState)}`);
     }
     // ChatGPT 5.2 Thinking モデル対応:
     // 回答が「思考」として折りたたまれている場合は展開してからテキストを取得
@@ -2714,12 +2751,16 @@ async function askGeminiFastInternal(question, debug, budgetMs) {
     // ChatGPT側と同様のポーリングループで応答完了を検出
     const maxWaitMs = getResponseWaitBudgetMs(t0, RESPONSE_WAIT_MAX_MS, 'gemini-response', budgetMs);
     const pollIntervalMs = 1000;
+    const IDLE_TIMEOUT_MS = 60000; // ストップボタン消失後、60秒間無活動でタイムアウト
     const startWait = Date.now();
+    let lastActivityAt = Date.now(); // 最後にストップボタンorテキスト成長を検出した時刻
     let lastLoggedState = '';
     let sawStopButton = false; // 停止ボタンを見たかどうか（生成が始まった証拠）
     let lastTextLength = 0;
     let textStableCount = 0; // テキスト長が変わらなかった回数
-    while (Date.now() - startWait < maxWaitMs) {
+    // ストップボタンが見えている間は maxWaitMs を無視し、IDLE_TIMEOUT のみで判定する。
+    while (Date.now() - lastActivityAt < IDLE_TIMEOUT_MS &&
+        (sawStopButton || Date.now() - startWait < maxWaitMs)) {
         const state = await client.evaluate(`
       (() => {
         ${DOM_UTILS_CODE}
@@ -2821,14 +2862,20 @@ async function askGeminiFastInternal(question, debug, budgetMs) {
         };
       })()
     `);
-        // 停止ボタンを検出したらフラグを立てる（生成が始まった証拠）
+        // 停止ボタンを検出したらフラグを立て、アクティビティを更新
         if (state.hasStopButton) {
             sawStopButton = true;
+            lastActivityAt = Date.now();
         }
         // テキスト長安定化検出
         if (state.lastResponseTextLength === lastTextLength && state.lastResponseTextLength > 0) {
             textStableCount++;
         }
+        else if (state.lastResponseTextLength > lastTextLength) {
+            textStableCount = 0;
+            lastTextLength = state.lastResponseTextLength;
+            lastActivityAt = Date.now(); // テキスト成長もアクティビティとみなす
+        }
         else {
             textStableCount = 0;
             lastTextLength = state.lastResponseTextLength;
@@ -2878,8 +2925,13 @@ async function askGeminiFastInternal(question, debug, budgetMs) {
         }
         await new Promise(r => setTimeout(r, pollIntervalMs));
     }
-    // タイムアウトチェック
-    if (Date.now() - startWait >= maxWaitMs) {
+    // タイムアウトチェック（IDLE_TIMEOUT で抜けた場合も含む）
+    const geminiLoopElapsed = Date.now() - startWait;
+    const geminiLoopIdle = Date.now() - lastActivityAt;
+    if (geminiLoopIdle >= IDLE_TIMEOUT_MS || (!sawStopButton && geminiLoopElapsed >= maxWaitMs)) {
+        const reason = geminiLoopIdle >= IDLE_TIMEOUT_MS
+            ? `idle for ${Math.round(geminiLoopIdle / 1000)}s (no stop button or text growth)`
+            : `absolute ceiling ${maxWaitMs}ms reached (stop button never seen)`;
         const finalState = await client.evaluate(`
       (() => {
         const textIncludes = (needle) => document.body && document.body.innerText && document.body.innerText.includes(needle);
@@ -2908,9 +2960,9 @@ async function askGeminiFastInternal(question, debug, budgetMs) {
         };
       })()
     `);
-        console.error(`[Gemini] Timeout - final state: ${JSON.stringify(finalState)}`);
+        console.error(`[Gemini] Timeout - ${reason}, elapsed=${Math.round(geminiLoopElapsed / 1000)}s, final state: ${JSON.stringify(finalState)}`);
         await resetConnection('gemini');
-        throw new Error(`Timed out waiting for Gemini response (${maxWaitMs}ms). sawStopButton=${sawStopButton}, textStableCount=${textStableCount}. Final state: ${JSON.stringify(finalState)}`);
+        throw new Error(`Timed out waiting for Gemini response (${Math.round(geminiLoopElapsed / 1000)}s, ${reason}). sawStopButton=${sawStopButton}, textStableCount=${textStableCount}. Final state: ${JSON.stringify(finalState)}`);
     }
     // 重要: タブをフォアグラウンドに持ってくる（バックグラウンドタブ対策）
     // GeminiもChatGPTと同様、バックグラウンドタブではDOMの状態が正しく取得できない

package/build/src/main.js

@@ -202,7 +202,7 @@ const httpServer = http.createServer(async (req, res) => {
                 return;
             }
             const { target, question, debug: debugFlag, budgetMs: requestBudgetMs } = parsed;
-            const effectiveBudgetMs = requestBudgetMs ?? 120000;
+            const effectiveBudgetMs = requestBudgetMs ?? 300000;
             if (!target || !question) {
                 res.writeHead(400, { 'Content-Type': 'application/json' }).end(JSON.stringify({ success: false, error: 'Missing required fields: target, question' }));
                 return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chrome-ai-bridge",
-  "version": "2.5.1",
+  "version": "2.5.3",
   "description": "CLI tool for querying ChatGPT and Gemini via Chrome extension. No Puppeteer required.",
   "type": "module",
   "bin": {

package/scripts/cab

@@ -11,7 +11,9 @@ set -euo pipefail
 CAB_PORT="${CAI_IPC_PORT:-9321}"
 CAB_HOST="127.0.0.1"
 CAB_BASE="http://${CAB_HOST}:${CAB_PORT}"
-CAB_DIR="$(cd "$(dirname "$0")" && pwd)"
+# Resolve symlinks to get the real script path (macOS compatible)
+_CAB_SELF="$(realpath "$0" 2>/dev/null || python3 -c "import os,sys; print(os.path.realpath(sys.argv[1]))" "$0")"
+CAB_DIR="$(cd "$(dirname "$_CAB_SELF")" && pwd)"
 CAB_LOG_DIR="${HOME}/.cache/chrome-ai-bridge"
 CAB_LOG="${CAB_LOG_DIR}/cab-daemon.log"
 CAB_STARTUP_TIMEOUT=30