doer-agent 0.9.1 → 0.9.3

package/dist/agent-connectivity.test.js

@@ -0,0 +1,42 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+import { buildNatsConnectionOptions } from "./agent-jetstream.js";
+import { heartbeatAgentSession } from "./agent-runtime-io.js";
+test("NATS connection detects stale networks quickly and keeps reconnecting", () => {
+    const options = buildNatsConnectionOptions({
+        servers: ["tls://nats.example.com:4222"],
+        token: "secret",
+    });
+    assert.equal(options.pingInterval, 2_000);
+    assert.equal(options.maxPingOut, 2);
+    assert.equal(options.maxReconnectAttempts, -1);
+    assert.equal(options.reconnectTimeWait, 1_000);
+    assert.equal(options.token, "secret");
+});
+test("heartbeat stops waiting when a NATS flush stalls", async () => {
+    const startedAt = Date.now();
+    await assert.rejects(heartbeatAgentSession({
+        nc: { flush: () => new Promise(() => undefined) },
+        serverBaseUrl: "https://doer.example.com",
+        userId: "user-1",
+        agentToken: "token-1",
+        timeoutMs: 20,
+        postJson: async () => ({ ok: true }),
+    }), /nats flush timed out after 20ms/);
+    assert.ok(Date.now() - startedAt < 500);
+});
+test("heartbeat forwards its timeout to the HTTP probe", async () => {
+    let observedTimeout;
+    await heartbeatAgentSession({
+        nc: { flush: async () => undefined },
+        serverBaseUrl: "https://doer.example.com",
+        userId: "user-1",
+        agentToken: "token-1",
+        timeoutMs: 3_000,
+        postJson: async (_url, _body, options) => {
+            observedTimeout = options?.timeoutMs;
+            return { ok: true };
+        },
+    });
+    assert.equal(observedTimeout, 3_000);
+});

package/dist/agent-jetstream.js

@@ -1,4 +1,20 @@
 import { AckPolicy, connect, DeliverPolicy, JSONCodec, RetentionPolicy, StorageType, } from "nats";
+const NATS_PING_INTERVAL_MS = 2_000;
+const NATS_MAX_PING_OUT = 2;
+const NATS_RECONNECT_TIME_WAIT_MS = 1_000;
+const NATS_RECONNECT_JITTER_MS = 250;
+export function buildNatsConnectionOptions(args) {
+    return {
+        servers: args.servers,
+        ...(args.token ? { token: args.token } : {}),
+        pingInterval: NATS_PING_INTERVAL_MS,
+        maxPingOut: NATS_MAX_PING_OUT,
+        maxReconnectAttempts: -1,
+        reconnectTimeWait: NATS_RECONNECT_TIME_WAIT_MS,
+        reconnectJitter: NATS_RECONNECT_JITTER_MS,
+        reconnectJitterTLS: NATS_RECONNECT_JITTER_MS,
+    };
+}
 export function normalizeNatsServers(value) {
     if (!Array.isArray(value)) {
         return [];
@@ -55,7 +71,7 @@ async function initJetStreamContext(args) {
     const stream = `DOER_AGENT_EVENTS_${sanitized}`;
     const subject = `doer.agent.events.${sanitized}`;
     const durable = `doer-agent-uploader-${sanitized}`;
-    const nc = await connect(args.token ? { servers: args.servers, token: args.token } : { servers: args.servers });
+    const nc = await connect(buildNatsConnectionOptions({ servers: args.servers, token: args.token }));
     const jsm = await nc.jetstreamManager();
     await ensureJetStreamInfra({ jsm, stream, subject, durable });
     void (async () => {

package/dist/agent-runtime-io.js

@@ -1,9 +1,19 @@
-export async function postJson(url, body) {
-    const res = await fetch(url, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify(body),
-    });
+export async function postJson(url, body, options = {}) {
+    let res;
+    try {
+        res = await fetch(url, {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify(body),
+            ...(options.timeoutMs ? { signal: AbortSignal.timeout(options.timeoutMs) } : {}),
+        });
+    }
+    catch (error) {
+        if (options.timeoutMs && error instanceof Error && error.name === "TimeoutError") {
+            throw new Error(`request timed out after ${options.timeoutMs}ms`);
+        }
+        throw error;
+    }
     const text = await res.text();
     let data = {};
     if (text) {
@@ -104,9 +114,22 @@ export function createEventPersistenceHelpers(args) {
     };
 }
 export async function heartbeatAgentSession(args) {
-    await args.nc.flush();
+    let timeout;
+    try {
+        await Promise.race([
+            args.nc.flush(),
+            new Promise((_, reject) => {
+                timeout = setTimeout(() => reject(new Error(`nats flush timed out after ${args.timeoutMs}ms`)), args.timeoutMs);
+            }),
+        ]);
+    }
+    finally {
+        if (timeout) {
+            clearTimeout(timeout);
+        }
+    }
     await args.postJson(`${args.serverBaseUrl}/api/agent/heartbeat`, {
         userId: args.userId,
         agentToken: args.agentToken,
-    });
+    }, { timeoutMs: args.timeoutMs });
 }

package/dist/agent.js

@@ -28,7 +28,8 @@ const AGENT_PROJECT_DIR = path.join(AGENT_MODULE_DIR, "..");
 const AGENT_PACKAGE_JSON_PATH = path.join(AGENT_PROJECT_DIR, "package.json");
 const BUNDLED_SKILLS_ROOT = path.join(AGENT_PROJECT_DIR, "runtime", "skills");
 const HEARTBEAT_INTERVAL_MS = 5_000;
-const HEARTBEAT_FAILURE_THRESHOLD = 3;
+const HEARTBEAT_FAILURE_THRESHOLD = 2;
+const HEARTBEAT_REQUEST_TIMEOUT_MS = 3_000;
 const codexAppEventCodec = StringCodec();
 let activeTaskLogContext = null;
 let workspaceRootOverride = null;
@@ -187,6 +188,7 @@ const eventPersistenceHelpers = createEventPersistenceHelpers({
 const heartbeatSession = async (args) => {
     await heartbeatAgentSession({
         ...args,
+        timeoutMs: HEARTBEAT_REQUEST_TIMEOUT_MS,
         postJson,
     });
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "doer-agent",
-  "version": "0.9.1",
+  "version": "0.9.3",
   "description": "Reverse-polling agent runtime for doer",
   "type": "module",
   "main": "dist/agent.js",
@@ -26,8 +26,8 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.29.0",
-    "@openai/codex": "^0.144.5",
-    "@openai/codex-sdk": "^0.144.5",
+    "@openai/codex": "^0.145.0",
+    "@openai/codex-sdk": "^0.145.0",
     "diff": "^9.0.0",
     "nats": "^2.29.3",
     "tar": "^7.5.15"