@liy/agent-runner 0.2.0 → 0.2.1

package/dist/index.js

@@ -15682,6 +15682,7 @@ function buildPiRpcArgs(harness) {
     model,
     "--thinking",
     thinking,
+    "--no-session",
     "--no-extensions",
     "--no-prompt-templates",
     "--no-themes",
@@ -15897,6 +15898,7 @@ async function runTaskAgentRunner(spec, options = {}) {
         spec,
         harness,
         channel,
+        ...spec.timeouts?.taskMs !== void 0 ? { timeoutMs: spec.timeouts.taskMs } : {},
         ...options.env ? { env: options.env } : {}
       });
       sendTaskStatus(channel, {
@@ -16034,7 +16036,28 @@ async function runHarnessAndReadCompletion(input) {
       ...input.env ? { env: input.env } : {}
     })
   });
-  const result = await activeRun.result;
+  const result = await waitForHarnessResult({
+    run: activeRun,
+    ...input.timeoutMs !== void 0 ? { timeoutMs: input.timeoutMs } : {}
+  });
+  if (result === "timeout") {
+    activeRun.cancel();
+    await waitForHarnessCleanup(activeRun.result);
+    const recovered = await readTaskCompletion(paths).catch(() => void 0);
+    if (recovered) {
+      await validateTaskArtifactFiles({
+        paths,
+        completion: recovered
+      });
+      return recovered;
+    }
+    const completion2 = createTimeoutCompletion(input.timeoutMs);
+    await writeTaskCompletion({
+      paths,
+      completion: completion2
+    });
+    return completion2;
+  }
   if (cancelRequested) {
     const completion2 = {
       status: "cancelled",
@@ -16064,6 +16087,38 @@ async function runHarnessAndReadCompletion(input) {
   }
   return completion;
 }
+async function waitForHarnessResult(input) {
+  const timeoutMs = normalizeTimeoutMs(input.timeoutMs);
+  if (timeoutMs === void 0) {
+    return input.run.result;
+  }
+  let timer;
+  try {
+    return await Promise.race([
+      input.run.result,
+      new Promise((resolve) => {
+        timer = setTimeout(() => resolve("timeout"), timeoutMs);
+      })
+    ]);
+  } finally {
+    if (timer) {
+      clearTimeout(timer);
+    }
+  }
+}
+async function waitForHarnessCleanup(result) {
+  await Promise.race([
+    result.catch(() => void 0),
+    sleep(1e4)
+  ]);
+}
+function createTimeoutCompletion(timeoutMs) {
+  return {
+    status: "failed",
+    summary: timeoutMs === void 0 ? "execute_task harness timed out" : `execute_task harness timed out after ${timeoutMs}ms`,
+    artifactIds: []
+  };
+}
 function createMissingCompletionFailure(error48) {
   const detail = error48 instanceof Error ? error48.message : String(error48);
   return {
@@ -16085,6 +16140,9 @@ function validateRunnerSpec(spec) {
   if (!spec.rpcUrl?.trim()) {
     throw new Error("task spec rpcUrl is required");
   }
+  if (spec.timeouts?.taskMs !== void 0 && normalizeTimeoutMs(spec.timeouts.taskMs) === void 0) {
+    throw new Error("task spec timeouts.taskMs must be a positive finite number");
+  }
   if (spec.agentRunner?.harness !== void 0 && typeof spec.agentRunner.harness !== "string") {
     throw new Error("task spec agentRunner.harness must be a string");
   }
@@ -16097,6 +16155,14 @@ function validateRunnerSpec(spec) {
   }
   assertNoForbiddenSpecKeys(spec);
 }
+function normalizeTimeoutMs(value) {
+  return typeof value === "number" && Number.isFinite(value) && value > 0 ? value : void 0;
+}
+function sleep(ms) {
+  return new Promise((resolve) => {
+    setTimeout(resolve, ms);
+  });
+}
 async function openControlChannel(url2) {
   const WebSocketCtor = globalThis.WebSocket;
   if (!WebSocketCtor) {