aws-runtime-bridge 1.9.2 → 1.9.5

package/dist/services/session-output.js

@@ -4,6 +4,7 @@
  * 管理终端会话的输出缓冲和刷新机制
  */
 import axios from "axios";
+import { requestRuntimeAccessTokenRefreshForServer } from "../services/auto-register.js";
 import { resolveSchedulerBaseUrlFrom } from "../config.js";
 import { logger } from "../utils/logger.js";
 import { getRuntimeAccessToken } from "./runtime-binding.js";
@@ -22,6 +23,42 @@ function resolveRuntimeCallbackBaseUrl(options) {
 }
 /** 输出刷新防抖时间：合并启动期高频小块输出，降低 HTTP/WebSocket 压力。 */
 const OUTPUT_FLUSH_DEBOUNCE_MS = 100;
+function getAxiosStatusCode(error) {
+    if (typeof error !== "object" || error === null)
+        return undefined;
+    const response = error.response;
+    if (typeof response !== "object" || response === null)
+        return undefined;
+    const status = response.status;
+    return typeof status === "number" ? status : undefined;
+}
+/** 防止并发回调同时触发多次 token 刷新。 */
+let tokenRefreshPromise = null;
+async function tryRefreshRuntimeToken(serverBaseUrl) {
+    if (tokenRefreshPromise)
+        return tokenRefreshPromise;
+    tokenRefreshPromise = (async () => {
+        try {
+            const result = await requestRuntimeAccessTokenRefreshForServer(serverBaseUrl);
+            if (result.success && result.runtimeAccessToken) {
+                logger.info("[session-output] Runtime token refreshed after 401");
+                return true;
+            }
+            logger.warn("[session-output] Runtime token refresh failed", { error: result.error });
+            return false;
+        }
+        catch (error) {
+            logger.warn("[session-output] Runtime token refresh threw", { error });
+            return false;
+        }
+    })();
+    try {
+        return await tokenRefreshPromise;
+    }
+    finally {
+        tokenRefreshPromise = null;
+    }
+}
 /** 单个会话待发送输出上限，避免调度器不可达时内存无限增长。 */
 const MAX_PENDING_OUTPUT_CHARS = 64 * 1024;
 /** 单次回调最大发送量，避免一次 POST 携带过大的终端噪声。 */
@@ -66,6 +103,23 @@ export async function sendStatus(agentId, sessionId, status, actionInfo, usage,
         return await axios.post(`${callbackBaseUrl}/api/runtime/callback/status`, payload, { headers });
     }
     catch (error) {
+        if (getAxiosStatusCode(error) === 401) {
+            const refreshed = await tryRefreshRuntimeToken(callbackBaseUrl);
+            if (refreshed) {
+                const retryHeaders = runtimeAuthHeaders(authOptions, callbackBaseUrl);
+                if (retryHeaders) {
+                    try {
+                        return await axios.post(`${callbackBaseUrl}/api/runtime/callback/status`, payload, { headers: retryHeaders });
+                    }
+                    catch (retryError) {
+                        logger.warn("Runtime status callback retry failed after token refresh", {
+                            agentId, sessionId, status, error: retryError,
+                        });
+                        return undefined;
+                    }
+                }
+            }
+        }
         logger.warn("Runtime status callback failed", {
             agentId,
             sessionId,
@@ -85,8 +139,8 @@ export async function sendStatus(agentId, sessionId, status, actionInfo, usage,
  * @param seq - 序列号
  */
 export async function sendOutput(agentId, output, sessionId, seq, metadata, authOptions) {
+    const callbackBaseUrl = resolveRuntimeCallbackBaseUrl(authOptions);
     try {
-        const callbackBaseUrl = resolveRuntimeCallbackBaseUrl(authOptions);
         const headers = runtimeAuthHeaders(authOptions, callbackBaseUrl);
         if (!headers) {
             logger.warn("Skip runtime output callback because runtime token is missing", {
@@ -99,6 +153,21 @@ export async function sendOutput(agentId, output, sessionId, seq, metadata, auth
         await axios.post(`${callbackBaseUrl}/api/runtime/callback/output`, { agentId, output, sessionId, seq, ...metadata }, { headers });
     }
     catch (error) {
+        if (getAxiosStatusCode(error) === 401) {
+            const refreshed = await tryRefreshRuntimeToken(callbackBaseUrl);
+            if (refreshed) {
+                const retryHeaders = runtimeAuthHeaders(authOptions, callbackBaseUrl);
+                if (retryHeaders) {
+                    try {
+                        await axios.post(`${callbackBaseUrl}/api/runtime/callback/output`, { agentId, output, sessionId, seq, ...metadata }, { headers: retryHeaders });
+                        return;
+                    }
+                    catch {
+                        // retry also failed
+                    }
+                }
+            }
+        }
         logger.warn("Runtime output callback failed", {
             agentId,
             sessionId,
@@ -114,15 +183,31 @@ export async function sendOutput(agentId, output, sessionId, seq, metadata, auth
  * @param questions - 结构化问题列表
  */
 export async function sendQuestionRequest(agentId, sessionId, questions, authOptions) {
+    const callbackBaseUrl = resolveRuntimeCallbackBaseUrl(authOptions);
+    const payload = { agentId, sessionId, questions };
     try {
-        const callbackBaseUrl = resolveRuntimeCallbackBaseUrl(authOptions);
         const headers = runtimeAuthHeaders(authOptions, callbackBaseUrl);
         if (!headers) {
             return;
         }
-        await axios.post(`${callbackBaseUrl}/api/runtime/callback/question`, { agentId, sessionId, questions }, { headers });
+        await axios.post(`${callbackBaseUrl}/api/runtime/callback/question`, payload, { headers });
     }
-    catch {
+    catch (error) {
+        if (getAxiosStatusCode(error) === 401) {
+            const refreshed = await tryRefreshRuntimeToken(callbackBaseUrl);
+            if (refreshed) {
+                const retryHeaders = runtimeAuthHeaders(authOptions, callbackBaseUrl);
+                if (retryHeaders) {
+                    try {
+                        await axios.post(`${callbackBaseUrl}/api/runtime/callback/question`, payload, { headers: retryHeaders });
+                        return;
+                    }
+                    catch {
+                        // retry also failed
+                    }
+                }
+            }
+        }
         // ignore transient callback failures
     }
 }
@@ -130,15 +215,30 @@ export async function sendQuestionRequest(agentId, sessionId, questions, authOpt
  * 发送工作区文件变更事件到调度器，由调度器通过 WebSocket 通知前端刷新预览。
  */
 export async function sendFileChanged(payload) {
+    const callbackBaseUrl = resolveRuntimeCallbackBaseUrl();
     try {
-        const callbackBaseUrl = resolveRuntimeCallbackBaseUrl();
         const headers = runtimeAuthHeaders(undefined, callbackBaseUrl);
         if (!headers) {
             return;
         }
         await axios.post(`${callbackBaseUrl}/api/runtime/callback/file-changed`, payload, { headers });
     }
-    catch {
+    catch (error) {
+        if (getAxiosStatusCode(error) === 401) {
+            const refreshed = await tryRefreshRuntimeToken(callbackBaseUrl);
+            if (refreshed) {
+                const retryHeaders = runtimeAuthHeaders(undefined, callbackBaseUrl);
+                if (retryHeaders) {
+                    try {
+                        await axios.post(`${callbackBaseUrl}/api/runtime/callback/file-changed`, payload, { headers: retryHeaders });
+                        return;
+                    }
+                    catch {
+                        // retry also failed
+                    }
+                }
+            }
+        }
         // ignore transient callback failures
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aws-runtime-bridge",
-  "version": "1.9.2",
+  "version": "1.9.5",
   "description": "AgentsWorkStudio runtime bridge service for machine-level agent runtime integration",
   "type": "module",
   "main": "dist/index.js",