diff --git a/src/agents/tools/gateway-tool.ts b/src/agents/tools/gateway-tool.ts index b47293f0106..ca125deb6a6 100644 --- a/src/agents/tools/gateway-tool.ts +++ b/src/agents/tools/gateway-tool.ts @@ -157,7 +157,13 @@ export function createGatewayTool(opts?: { } : undefined; const extracted = extractDeliveryInfo(sessionKey); - const deliveryContext = liveContext ?? extracted.deliveryContext; + const deliveryContext = + liveContext != null + ? { + ...liveContext, + accountId: liveContext.accountId ?? extracted.deliveryContext?.accountId, + } + : extracted.deliveryContext; // Guard threadId with the same session check as deliveryContext. When // targeting another session, opts.agentThreadId belongs to the current // session's thread and must not be written into the sentinel — it would diff --git a/src/gateway/server-methods/config.ts b/src/gateway/server-methods/config.ts index 7c9e4cf95a0..e3300d02045 100644 --- a/src/gateway/server-methods/config.ts +++ b/src/gateway/server-methods/config.ts @@ -205,7 +205,13 @@ function resolveConfigRestartRequest(params: unknown): { const { deliveryContext: extractedDeliveryContext, threadId: extractedThreadId } = extractDeliveryInfo(sessionKey); const paramsDeliveryContext = parseDeliveryContextFromParams(params); - const deliveryContext = paramsDeliveryContext ?? extractedDeliveryContext; + const deliveryContext = + paramsDeliveryContext != null + ? { + ...paramsDeliveryContext, + accountId: paramsDeliveryContext.accountId ?? extractedDeliveryContext?.accountId, + } + : extractedDeliveryContext; const threadId = paramsDeliveryContext?.threadId ?? extractedThreadId; return {