@wrenautomation/aerie 0.5.1 → 0.5.2

package/dist/bin.js

@@ -36427,6 +36427,22 @@ function date4(params) {
   return _coercedDate(ZodDate, params);
 }
 
+// src/deadline.ts
+async function withDeadline(ms2, outer, call) {
+  const signal = AbortSignal.any([AbortSignal.timeout(ms2), ...outer ? [outer] : []]);
+  let stop;
+  const stopped = new Promise((_, reject) => {
+    stop = () => reject(new Error(outer?.aborted ? "aborted" : `no answer in ${Math.round(ms2 / 1e3)} s`));
+    if (signal.aborted) stop();
+    else signal.addEventListener("abort", stop, { once: true });
+  });
+  try {
+    return await Promise.race([call(signal), stopped]);
+  } finally {
+    if (stop) signal.removeEventListener("abort", stop);
+  }
+}
+
 // src/run/spawn.ts
 init_db2();
 import { spawn } from "node:child_process";
@@ -36714,11 +36730,16 @@ function apiRunner(o) {
       const m = modelFromSpec(spec, { ...process.env, ...o.env, ...ctx.env });
       await ctx.emit({ kind: "init", text: m.id, data: { spec, model: m.id } });
       try {
-        const { value, usage } = await m.ask(
-          `${promptOf(t, ctx.answers)}
+        const { value, usage } = await withDeadline(
+          o.timeoutMs ?? 12e4,
+          ctx.signal,
+          (signal) => m.ask(
+            `${promptOf(t, ctx.answers)}
 
 You have no tools. Answer as JSON: outcome (done, needs-input or failed), summary, and question when a person must decide.`,
-          ApiAnswer
+            ApiAnswer,
+            { signal }
+          )
         );
         await ctx.emit({ kind: "say", text: value.summary, data: { value, usage } });
         return {
@@ -36755,7 +36776,10 @@ function headlessRunners(get2) {
     codexRunner(opt("codex")),
     grokRunner(opt("grok")),
     autobrowseRunner(opt("autobrowse")),
-    apiRunner({ spec: get2("AERIE_API_MODEL") })
+    apiRunner({
+      spec: get2("AERIE_API_MODEL"),
+      timeoutMs: Number(get2("AERIE_API_TIMEOUT_MS")) || void 0
+    })
   ];
   return new Map(rs.map((r) => [r.name, r]));
 }