chrome-ai-bridge 2.5.1 → 2.5.2

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,31 @@ 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 = 30000; // ストップボタン消失後、30秒間無活動でタイムアウト
     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 は絶対安全上限）
+    while (Date.now() - startWait < maxWaitMs && Date.now() - lastActivityAt < IDLE_TIMEOUT_MS) {
         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 +1099,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 +1312,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 +1333,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 +1379,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 +1387,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;
@@ -1409,9 +1437,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 モデル対応:
     // 回答が「思考」として折りたたまれている場合は展開してからテキストを取得

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.2",
   "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