@springbrand/agent-runtime 0.1.2 → 0.1.3-alpha.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/agent-runtime",
-  "version": "0.1.2",
+  "version": "0.1.3-alpha.0",
   "type": "module",
   "files": [
     "src",

package/src/pi/tool/core-host.ts

@@ -14,6 +14,8 @@ import type { PiToolCandidate } from "./compiler";
 
 // 本文件沿用 `../../index.ts` 入口定义的 Workspace、Port 和 Tool Candidate 术语。
 
+const CODEMODE_SANDBOX_TIMEOUT_MS = 55_000;
+
 /**
  * 把宿主的 Worker Loader、出站网络和 Workspace 组装成 Pi 代码执行工具候选项。
  *
@@ -32,6 +34,8 @@ export function workspaceCodeExecutionPiToolCandidate(options: {
     executor: new DynamicWorkerExecutor({
       loader: options.loader,
       globalOutbound: options.outbound,
+      // 先于外层 60s 截止结束，给 Runtime RPC 结算和 Worker 释放留出时间。
+      timeout: CODEMODE_SANDBOX_TIMEOUT_MS,
     }),
     connectors: [
       new StateConnector(

package/src/pi/tool/core.ts

@@ -205,6 +205,7 @@ const executeParameters = Type.Object({
       "Plain JavaScript async function. TypeScript annotations are not supported.",
   }),
 });
+const CODEMODE_EXECUTE_TIMEOUT_MS = 60_000;
 
 /**
  * 把 Cloudflare Codemode Runtime handle 包装为高风险 Pi 代码执行工具候选项。
@@ -235,7 +236,33 @@ export function codeExecutionPiToolCandidate(
     // 必须通过 Runtime handle 而不是直接调用 executor，因为 Cloudflare Codemode 把重放、审批和执行日志放在持久化 Runtime 层。
     async execute(_toolCallId, input, signal) {
       signal?.throwIfAborted();
-      return result(await runtime.execute(input));
+      // ponytail: 外层截止只保证 Agent 继续；Codemode 支持 AbortSignal 或宿主 dispose 后再终止底层执行。
+      let timeout: ReturnType<typeof setTimeout> | undefined;
+      const deadline = new Promise<{
+        status: "error";
+        code: "timeout";
+        error: string;
+        retryable: false;
+        outcome: "unknown";
+      }>((resolve) => {
+        timeout = setTimeout(() => resolve({
+          status: "error",
+          code: "timeout",
+          error:
+            `Code Mode execute exceeded ${CODEMODE_EXECUTE_TIMEOUT_MS} ms; ` +
+            "use existing results and state the missing evidence.",
+          retryable: false,
+          outcome: "unknown",
+        }), CODEMODE_EXECUTE_TIMEOUT_MS);
+      });
+      try {
+        return result(await Promise.race([
+          runtime.execute(input),
+          deadline,
+        ]));
+      } finally {
+        if (timeout !== undefined) clearTimeout(timeout);
+      }
     },
   };
   return {