@rowan-agent/agent 0.8.2 → 0.8.4

package/dist/index.d.ts

@@ -1391,7 +1391,7 @@ declare class InMemoryStore implements DurableStore {
         ownerId: string;
         leaseMs: number;
     }): Promise<OwnedStore>;
-    assertOwner(lease: OwnerLease): void;
+    assertOwner(lease: OwnerLease, requireLive?: boolean): void;
     renewOwner(lease: OwnerLease, leaseMs: number): OwnerLease;
     releaseOwner(lease: OwnerLease): void;
     reserveAgent(lease: OwnerLease, input: {
@@ -1667,6 +1667,7 @@ declare class SqliteStore implements DurableStore {
     private persistState;
     private requireMatchingOwner;
     private immediateTransaction;
+    private readTransaction;
     private assertOpen;
 }
 

package/dist/index.js

@@ -767,6 +767,15 @@ var EmptyResponseError = class extends Error {
     this.name = "EmptyResponseError";
   }
 };
+var ModelOutputLimitError = class extends Error {
+  constructor(limit) {
+    super(`Model output exceeded the ${limit}-character safety limit.`);
+    this.limit = limit;
+    this.name = "ModelOutputLimitError";
+  }
+  limit;
+  code = "model_output_limit";
+};
 var LoopGuard = {
   /** Returns abort result if signal is aborted */
   checkAbort(signal) {
@@ -778,6 +787,7 @@ var LoopGuard = {
 };
 
 // src/loop/stream-collector.ts
+var MAX_STREAMED_OUTPUT_CHARACTERS = 1024 * 1024;
 async function invokeModel(input) {
   const events = input.config.stream(input.request, { signal: input.config.signal });
   const result = await collectStreamResult({
@@ -811,7 +821,22 @@ async function collectStreamResult(input) {
   let lastPartial;
   let stopReason;
   let doneResponse;
+  let streamedOutputCharacters = 0;
+  const toolCallsWithDeltas = /* @__PURE__ */ new Set();
   for await (const event of input.events) {
+    if (event.type === "text_delta") {
+      streamedOutputCharacters += event.text.length;
+    } else if (event.type === "thinking_delta") {
+      streamedOutputCharacters += event.thinking.length;
+    } else if (event.type === "tool_call_delta") {
+      streamedOutputCharacters += event.arguments.length;
+      toolCallsWithDeltas.add(event.id);
+    } else if (event.type === "tool_call_end" && !toolCallsWithDeltas.has(event.id)) {
+      streamedOutputCharacters += event.arguments.length;
+    }
+    if (streamedOutputCharacters > MAX_STREAMED_OUTPUT_CHARACTERS) {
+      throw new ModelOutputLimitError(MAX_STREAMED_OUTPUT_CHARACTERS);
+    }
     const abortResult2 = LoopGuard.checkAbort(input.config.signal);
     if (abortResult2.stopReason !== "none") {
       if (activeMessageId) {
@@ -3816,6 +3841,7 @@ var TransientRunEventHub = class {
 // src/runtime/durable-runtime.ts
 var DEFAULT_CONCURRENCY = 10;
 var DEFAULT_POLL_MS = 25;
+var MAX_CONSUMER_IDLE_POLL_MS = 250;
 var OWNER_LEASE_MS = 3e4;
 var OWNER_RENEWAL_MS = 1e4;
 var AgentRuntime = class _AgentRuntime {
@@ -4218,6 +4244,7 @@ var AgentRuntime = class _AgentRuntime {
     let cursor = registration.cursor;
     const waterline = cursorSequence(registration.waterline);
     let caughtUp = false;
+    let pollMs = DEFAULT_POLL_MS;
     while (!subscription.controller.signal.aborted && !this.closed) {
       const events = await this.owned.listEvents(cursor ? { after: cursor } : {});
       for (const event of events) {
@@ -4243,7 +4270,8 @@ var AgentRuntime = class _AgentRuntime {
         caughtUp = true;
         subscription.caughtUp.resolve();
       }
-      await delay(DEFAULT_POLL_MS);
+      pollMs = events.length === 0 ? Math.min(MAX_CONSUMER_IDLE_POLL_MS, pollMs * 2) : DEFAULT_POLL_MS;
+      await delay(pollMs);
     }
     if (!caughtUp) subscription.caughtUp.resolve();
   }
@@ -4492,17 +4520,17 @@ var InMemoryStore = class _InMemoryStore {
     };
     return new MemoryOwnedStore(this, clone(this.owner));
   }
-  assertOwner(lease) {
+  assertOwner(lease, requireLive = true) {
     const actual = this.owner;
     if (!actual || actual.token !== lease.token || actual.epoch !== lease.epoch || actual.ownerId !== lease.ownerId) {
       throw ownershipLost(lease, actual, "epoch_advanced");
     }
-    if (Date.parse(actual.expiresAt) <= Date.now()) {
+    if (requireLive && Date.parse(actual.expiresAt) <= Date.now()) {
       throw ownershipLost(lease, actual, "expired");
     }
   }
   renewOwner(lease, leaseMs) {
-    this.assertOwner(lease);
+    this.assertOwner(lease, false);
     assertLeaseDuration(leaseMs);
     this.owner.expiresAt = expiry(leaseMs);
     return clone(this.owner);
@@ -4597,23 +4625,25 @@ var InMemoryStore = class _InMemoryStore {
     if ([...this.runs.values()].some((candidate) => candidate.agentId === run.agentId && candidate.id !== run.id && ["running", "input_required"].includes(candidate.state))) {
       throw new RuntimeError("run_state_conflict", { runId: run.id, expected: ["queued"], actual: run.state });
     }
-    const messageId = input.messageId ?? createId("msg");
     if (run.pinnedConfigToken && input.configToken && run.pinnedConfigToken !== input.configToken) {
       throw new RuntimeError("run_state_conflict", { runId: run.id, expected: ["queued"], actual: run.state });
     }
     if (!run.pinnedConfigToken && input.configToken) run.pinnedConfigToken = input.configToken;
-    const userInput = normalizeUserInput(run.input);
-    const message = {
-      id: messageId,
-      agentId: run.agentId,
-      runId: run.id,
-      role: "user",
-      content: userInputContent(userInput),
-      ...userInputMetadata(userInput) ? { metadata: clone(userInputMetadata(userInput)) } : {},
-      sequenceWithinRun: this.nextMessageSequence(run.id),
-      createdAt: createTimestamp()
-    };
-    this.messages.set(message.id, message);
+    let message;
+    if (!run.checkpoint) {
+      const userInput = normalizeUserInput(run.input);
+      message = {
+        id: input.messageId ?? createId("msg"),
+        agentId: run.agentId,
+        runId: run.id,
+        role: "user",
+        content: userInputContent(userInput),
+        ...userInputMetadata(userInput) ? { metadata: clone(userInputMetadata(userInput)) } : {},
+        sequenceWithinRun: this.nextMessageSequence(run.id),
+        createdAt: createTimestamp()
+      };
+      this.messages.set(message.id, message);
+    }
     const execution = {
       runId: run.id,
       ownerEpoch: lease.epoch,
@@ -4623,7 +4653,7 @@ var InMemoryStore = class _InMemoryStore {
     run.execution = execution;
     run.revision += 1;
     run.updatedAt = createTimestamp();
-    this.appendMessage(run, message);
+    if (message) this.appendMessage(run, message);
     this.appendTransition(run, "queued", "running");
     const result = { run: clone(run), execution: clone(execution), history: this.history(run.agentId, run.agentSequence) };
     this.writeOperationReceipt(operationKey, operationPayload, result);
@@ -5497,7 +5527,28 @@ var SqliteStore = class {
     return this.invoke(lease, (store, current) => store.listRuns(current, input), false);
   }
   async listEvents(lease, input = {}) {
-    return this.invoke(lease, (store, current) => store.listEvents(current, input), false);
+    this.assertOpen();
+    return this.readTransaction(() => {
+      this.requireMatchingOwner(lease, true);
+      const after = input.after ? parseEventCursor(lease, input.after) : 0;
+      if (input.after) {
+        const latest = this.database.query(
+          "SELECT payload_json FROM run_events ORDER BY sequence DESC LIMIT 1"
+        ).get();
+        const waterline = latest ? parseEventCursor(
+          lease,
+          JSON.parse(latest.payload_json).cursor
+        ) : 0;
+        if (after > waterline) {
+          throw new RuntimeError("invalid_cursor", { cursorType: "event", reason: "beyond_waterline" });
+        }
+        if (after === waterline) return [];
+      }
+      const rows = this.database.query(
+        "SELECT payload_json FROM run_events ORDER BY sequence LIMIT -1 OFFSET ?"
+      ).all(after);
+      return rows.map((row) => JSON.parse(row.payload_json));
+    });
   }
   async openConsumer(lease, consumerId) {
     return this.invoke(lease, (store, current) => store.openConsumer(current, consumerId), false);
@@ -5509,7 +5560,7 @@ var SqliteStore = class {
     this.assertOpen();
     assertOwnerInput(lease.ownerId, leaseMs);
     return this.immediateTransaction(() => {
-      const row = this.requireMatchingOwner(lease, true);
+      const row = this.requireMatchingOwner(lease, false);
       const expiresAt = Date.now() + leaseMs;
       this.writeOwner({ ...row, expires_at: expiresAt });
       return { ...lease, expiresAt: new Date(expiresAt).toISOString() };
@@ -5686,6 +5737,20 @@ var SqliteStore = class {
       throw error;
     }
   }
+  readTransaction(operation) {
+    this.database.run("BEGIN");
+    try {
+      const result = operation();
+      this.database.run("COMMIT");
+      return result;
+    } catch (error) {
+      try {
+        this.database.run("ROLLBACK");
+      } catch {
+      }
+      throw error;
+    }
+  }
   assertOpen() {
     if (this.closed) throw new RuntimeError("runtime_closed", null);
   }
@@ -5774,6 +5839,19 @@ function ownerLease(row) {
     expiresAt: new Date(row.expires_at).toISOString()
   };
 }
+function parseEventCursor(lease, cursor) {
+  const value = String(cursor);
+  const separator = value.lastIndexOf(":");
+  const incarnation = separator < 0 ? "" : value.slice(0, separator);
+  const sequence = separator < 0 ? "" : value.slice(separator + 1);
+  const token = String(lease.token);
+  const ownerSuffix = `:${lease.epoch}:${lease.ownerId}`;
+  const expectedIncarnation = token.endsWith(ownerSuffix) ? token.slice(0, -ownerSuffix.length) : "";
+  if (incarnation !== expectedIncarnation || !/^\d+$/.test(sequence)) {
+    throw new RuntimeError("invalid_cursor", { cursorType: "event", reason: "wrong_store" });
+  }
+  return Number(sequence);
+}
 function ownershipLost2(lease, row) {
   return new RuntimeError("runtime_ownership_lost", {
     reason: row.epoch > lease.epoch ? "epoch_advanced" : row.owner_id ? "released" : "released",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rowan-agent/agent",
-  "version": "0.8.2",
+  "version": "0.8.4",
   "license": "MIT",
   "repository": {
     "type": "git",