From 1c63da09d87b7f61eb48a69a8b45f681cc00a2f2 Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Sun, 21 Jun 2026 23:34:23 +0200
Subject: [PATCH] fix(qa-lab): cap agent wait timeout payloads

---
 extensions/qa-lab/src/suite-runtime-agent-process.test.ts | 2 +-
 extensions/qa-lab/src/suite-runtime-agent-process.ts      | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/extensions/qa-lab/src/suite-runtime-agent-process.test.ts b/extensions/qa-lab/src/suite-runtime-agent-process.test.ts
index dae6c2ad484..a3527eea4d1 100644
--- a/extensions/qa-lab/src/suite-runtime-agent-process.test.ts
+++ b/extensions/qa-lab/src/suite-runtime-agent-process.test.ts
@@ -698,7 +698,7 @@ describe("qa suite runtime agent process helpers", () => {
 
     expect(gatewayCall).toHaveBeenCalledWith(
       "agent.wait",
-      { runId: "run-oversized", timeoutMs: 9e15 },
+      { runId: "run-oversized", timeoutMs: MAX_TIMER_TIMEOUT_MS },
       { timeoutMs: MAX_TIMER_TIMEOUT_MS },
     );
   });
diff --git a/extensions/qa-lab/src/suite-runtime-agent-process.ts b/extensions/qa-lab/src/suite-runtime-agent-process.ts
index 14a5ca4b293..96691351c3d 100644
--- a/extensions/qa-lab/src/suite-runtime-agent-process.ts
+++ b/extensions/qa-lab/src/suite-runtime-agent-process.ts
@@ -347,15 +347,16 @@ async function waitForAgentRun(
   runId: string,
   timeoutMs = 30_000,
 ) {
+  const waitTimeoutMs = resolveTimerTimeoutMs(timeoutMs, 30_000);
   try {
     return (await env.gateway.call(
       "agent.wait",
       {
         runId,
-        timeoutMs,
+        timeoutMs: waitTimeoutMs,
       },
       {
-        timeoutMs: resolveQaGatewayTimeoutWithGraceMs(timeoutMs),
+        timeoutMs: resolveQaGatewayTimeoutWithGraceMs(waitTimeoutMs),
       },
     )) as { status?: string; error?: string };
   } catch (error) {