@rowan-agent/agent 0.8.3 → 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: {

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) {
@@ -4495,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);
@@ -4600,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,
@@ -4626,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);
@@ -5533,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() };

package/package.json

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