@apo-ai/sdk 0.9.3 → 0.9.4

package/dist/agent-task/public.js

@@ -560,8 +560,19 @@ function parseJudgeJson(raw) {
     reasoning: "Judge response could not be parsed as JSON \u2014 the verdict is unknown, so this check is treated as a failure. The model's raw response is available in the judge metadata."
   };
 }
-var JUDGE_TIMEOUT_MS = 3e5;
 var JUDGE_IDLE_TIMEOUT_MS = 9e4;
+function envMs(name, fallback) {
+  const raw = process.env[name]?.trim();
+  if (!raw) return fallback;
+  const ms = Number(raw);
+  return Number.isFinite(ms) && ms > 0 ? ms : fallback;
+}
+function judgeFirstDataTimeoutMs() {
+  return envMs("APO_JUDGE_TIMEOUT_MS", 3e5);
+}
+function judgeMaxDurationMs() {
+  return envMs("APO_JUDGE_MAX_DURATION_MS", 12e5);
+}
 var JUDGE_RETRY_DELAY_MS = 1e3;
 var JUDGE_RETRY_AFTER_CAP_MS = 2e4;
 var JUDGE_MIN_RETRY_BUDGET_MS = 6e4;
@@ -736,16 +747,29 @@ ${instructionText}`,
     response_format: { type: "json_object" },
     ...stream ? { stream: true, stream_options: { include_usage: true } } : {}
   });
-  const attempt = async (stream, deadline) => {
+  const attempt = async (stream, maxDeadline) => {
     const controller = new AbortController();
     const expire = (reason, name) => () => controller.abort(new DOMException(reason, name));
-    const total = setTimeout(
-      expire(`no complete response within ${JUDGE_TIMEOUT_MS / 1e3}s`, "TimeoutError"),
-      Math.max(deadline - Date.now(), 0)
+    const firstDataMs = judgeFirstDataTimeoutMs();
+    const runaway = setTimeout(
+      expire(
+        `still streaming after ${Math.round(judgeMaxDurationMs() / 1e3)}s (APO_JUDGE_MAX_DURATION_MS)`,
+        "TimeoutError"
+      ),
+      Math.max(maxDeadline - Date.now(), 0)
+    );
+    let waiting = setTimeout(
+      expire(
+        `no data within ${Math.round(firstDataMs / 1e3)}s (APO_JUDGE_TIMEOUT_MS)`,
+        "FirstDataTimeoutError"
+      ),
+      firstDataMs
     );
     const stalled = expire(`no data for ${JUDGE_IDLE_TIMEOUT_MS / 1e3}s`, "IdleTimeoutError");
     let idle;
     const onData = () => {
+      clearTimeout(waiting);
+      waiting = void 0;
       clearTimeout(idle);
       idle = setTimeout(stalled, JUDGE_IDLE_TIMEOUT_MS);
     };
@@ -782,7 +806,8 @@ ${instructionText}`,
         return transportFailure("Judge response failed", controller.signal.reason ?? error);
       }
     } finally {
-      clearTimeout(total);
+      clearTimeout(runaway);
+      clearTimeout(waiting);
       clearTimeout(idle);
       controller.abort();
     }
@@ -802,7 +827,7 @@ ${instructionText}`,
   };
   return runWithSharedPrefix(cacheKey, async () => {
     const startedAt = Date.now();
-    const deadline = startedAt + JUDGE_TIMEOUT_MS;
+    const deadline = startedAt + judgeMaxDurationMs();
     let stream = true;
     let result = await attempt(stream, deadline);
     const truncated = result.kind === "completion" && isTruncated(result.completion);