@rowan-agent/agent 0.14.0 → 0.14.2

package/dist/index.d.ts

@@ -110,6 +110,8 @@ type RunInteraction = Readonly<{
     status: RunInteractionStatus;
     answer?: JsonValue;
     reply?: string;
+    /** The Run is only waiting for the next user message, not asking a question. */
+    turnBoundary?: true;
     toolCallId?: string;
     result?: Readonly<{
         answered?: string;
@@ -543,7 +545,7 @@ type PhaseExecution = {
     }): Promise<ModelInvokeOutput>;
     reportStatus(status: PhaseStatus): Promise<void>;
     executeTool(context: AgentContext, toolCall: ToolCall): Promise<ToolResult>;
-    executeTools(context: AgentContext, toolCalls: readonly ToolCall[]): Promise<readonly ToolResult[]>;
+    executeTools(context: AgentContext, toolCalls: readonly ToolCall[], contentBlocks?: readonly ContentBlock[]): Promise<readonly ToolResult[]>;
     interaction: RunInteractionDriver;
     /** Message lifecycle manager for streaming updates from programmatic phases */
     messages: PhaseMessageManager;
@@ -1440,6 +1442,8 @@ interface OwnedStore {
         expectedRevision: number;
         requestMessageId: MessageId;
         calls: readonly ToolCallReservation[];
+        /** The model response's blocks, so the request message keeps what preceded the Tool Calls. */
+        contentBlocks?: readonly ContentBlock[];
     }): Promise<ToolBatchCommit>;
     startToolCall(input: {
         runId: RunId;
@@ -2386,12 +2390,9 @@ declare class InMemoryStore implements DurableStore {
         execution: ExecutionToken;
         expectedRevision: number;
         requestMessageId: MessageId;
-        calls: readonly Readonly<{
-            providerToolCallId: string;
-            name: string;
-            args: JsonValue;
-            toolCallId?: ToolCallId;
-        }>[];
+        calls: readonly ToolCallReservation[];
+        /** The model response's blocks, so the request message keeps what preceded the Tool Calls. */
+        contentBlocks?: readonly ContentBlock[];
     }): ToolBatchCommit;
     startToolCall(lease: OwnerLease, input: {
         runId: RunId;
@@ -2605,6 +2606,7 @@ declare class SqliteStore implements DurableStore {
             args: JsonValue;
             toolCallId?: ToolCallId;
         }>[];
+        contentBlocks?: readonly ContentBlock[];
     }): Promise<ToolBatchCommit>;
     startToolCall(lease: OwnerLease, input: {
         runId: RunId;

package/dist/index.js

@@ -2537,6 +2537,7 @@ async function executePhaseWithModel(ctx) {
       messages: ctx.messageManager.visible()
     };
     let executableToolCalls;
+    let roundContentBlocks;
     const checkpoint = ctx.execution.interaction.checkpoint();
     if (checkpoint && typeof checkpoint === "object" && checkpoint.kind === "tool_call_suspension") {
       const toolSuspension = checkpoint;
@@ -2575,11 +2576,12 @@ async function executePhaseWithModel(ctx) {
       executableToolCalls = collected.toolCalls.filter(
         (toolCall) => executableToolNames.has(toolCall.name)
       );
+      roundContentBlocks = collected.contentBlocks;
       if (executableToolCalls.length === 0) {
         return output;
       }
     }
-    const results = await ctx.execution.executeTools(roundContext, executableToolCalls);
+    const results = await ctx.execution.executeTools(roundContext, executableToolCalls, roundContentBlocks);
     for (const result of results) {
       const messageId = ctx.messageManager.start("tool", createToolResultContent(result), {
         phase: ctx.phase.name
@@ -3055,6 +3057,7 @@ async function executeToolCall(input) {
     result = await input.config.runtime.tools({
       config: input.config,
       toolCall: input.toolCall,
+      contentBlocks: input.contentBlocks,
       driver: input.driver
     });
   } else {
@@ -3082,6 +3085,7 @@ async function executeToolCalls(input) {
     const results2 = await input.config.runtime.toolsBatch({
       config: input.config,
       toolCalls: input.toolCalls,
+      contentBlocks: input.contentBlocks,
       driver: input.driver
     });
     if (results2.length !== input.toolCalls.length) throw new Error("Runtime returned an invalid Tool batch result");
@@ -3187,7 +3191,7 @@ function createPhaseExecution(config, state, phase, messageManager, toolExecutio
         return result;
       });
     },
-    async executeTools(phaseContext, toolCalls) {
+    async executeTools(phaseContext, toolCalls, contentBlocks) {
       return runTurn(async () => {
         const tools = phaseContext.tools.filter((tool) => tool.name !== PhaseRouteTool);
         for (const toolCall of toolCalls) {
@@ -3204,6 +3208,7 @@ function createPhaseExecution(config, state, phase, messageManager, toolExecutio
           },
           tools,
           toolCalls,
+          contentBlocks,
           driver: interaction
         });
         for (const result of results) {
@@ -3449,6 +3454,7 @@ async function executeOnce(input) {
         kind: "user_input",
         phase: error.request.phase,
         prompt: error.request.prompt,
+        turnBoundary: true,
         status: "pending",
         createdAt: error.request.requestedAt
       };
@@ -6530,7 +6536,7 @@ var AgentRuntime = class _AgentRuntime {
         },
         onContext: assembly.setContext,
         runtime: {
-          tools: ({ toolCall, driver }) => {
+          tools: ({ toolCall, contentBlocks, driver }) => {
             const task = toolQueue.then(async () => {
               const execution = await this.executeToolBatch({
                 run,
@@ -6539,6 +6545,7 @@ var AgentRuntime = class _AgentRuntime {
                 expectedRevision: executionRevision,
                 toolConfig: executionTools,
                 toolCalls: [toolCall],
+                contentBlocks,
                 signal: controller.signal,
                 driver,
                 onRevision: (rev) => {
@@ -6551,7 +6558,7 @@ var AgentRuntime = class _AgentRuntime {
             toolQueue = task.then(() => void 0, () => void 0);
             return task;
           },
-          toolsBatch: ({ toolCalls, driver }) => {
+          toolsBatch: ({ toolCalls, contentBlocks, driver }) => {
             const task = toolQueue.then(async () => {
               const execution = await this.executeToolBatch({
                 run,
@@ -6560,6 +6567,7 @@ var AgentRuntime = class _AgentRuntime {
                 expectedRevision: executionRevision,
                 toolConfig: executionTools,
                 toolCalls,
+                contentBlocks,
                 signal: controller.signal,
                 driver,
                 onRevision: (rev) => {
@@ -6675,7 +6683,12 @@ var AgentRuntime = class _AgentRuntime {
           execution: claim.execution,
           expectedRevision: executionRevision,
           failure: { code: "execution_failed", message: error instanceof Error ? error.message : "Execution failed." }
-        }).catch(() => void 0);
+        }).catch(async (commitError) => {
+          await this.owned.cancelRun({
+            runId: run.id,
+            reason: commitError instanceof Error ? commitError.message : "Run failure could not be committed."
+          }).catch(() => void 0);
+        });
         return;
       }
       throw error;
@@ -6755,6 +6768,7 @@ var AgentRuntime = class _AgentRuntime {
           execution: input.execution,
           expectedRevision: revision,
           requestMessageId: createId("msg"),
+          contentBlocks: input.contentBlocks,
           calls: input.toolCalls.map((tc, i) => ({
             toolCallId: toolCallIds[i],
             providerToolCallId: tc.id,
@@ -6929,6 +6943,7 @@ var AgentRuntime = class _AgentRuntime {
           execution: input.execution,
           expectedRevision: revision,
           requestMessageId: createId("msg"),
+          contentBlocks: input.contentBlocks,
           calls: [{
             toolCallId,
             providerToolCallId,
@@ -7011,6 +7026,7 @@ var AgentRuntime = class _AgentRuntime {
                 execution: input.execution,
                 expectedRevision: revision,
                 requestMessageId: createId("msg"),
+                contentBlocks: input.contentBlocks,
                 calls: [{ toolCallId, providerToolCallId, name: toolCall.name, args: toJsonValue(toolCall.args) }]
               });
               setRevision(reserved2.run.revision);
@@ -7024,6 +7040,7 @@ var AgentRuntime = class _AgentRuntime {
                 execution: input.execution,
                 expectedRevision: revision,
                 requestMessageId: createId("msg"),
+                contentBlocks: input.contentBlocks,
                 calls: [{ toolCallId, providerToolCallId, name: toolCall.name, args: toJsonValue(toolCall.args) }]
               });
               setRevision(reserved2.run.revision);
@@ -7052,6 +7069,7 @@ var AgentRuntime = class _AgentRuntime {
             execution: input.execution,
             expectedRevision: revision,
             requestMessageId: createId("msg"),
+            contentBlocks: input.contentBlocks,
             calls: [{ toolCallId, providerToolCallId, name: toolCall.name, args: toJsonValue(toolCall.args) }]
           });
           setRevision(reserved2.run.revision);
@@ -7074,6 +7092,7 @@ var AgentRuntime = class _AgentRuntime {
           execution: input.execution,
           expectedRevision: revision,
           requestMessageId: createId("msg"),
+          contentBlocks: input.contentBlocks,
           calls: [{ toolCallId, providerToolCallId, name: toolCall.name, args: toJsonValue(toolCall.args) }]
         });
         setRevision(committed.run.revision);
@@ -7872,7 +7891,10 @@ var InMemoryStore = class _InMemoryStore {
         this.commitInteractionRecord(waitingRun, interaction, "replied", void 0, replyText);
       }
       delete waitingRun.openInteractions;
-      delete waitingRun.interactionAnswers;
+      waitingRun.interactionAnswers = {
+        ...waitingRun.interactionAnswers ?? {},
+        ...Object.fromEntries(openInteractions.map(({ id }) => [id, { status: "replied", reply: replyText }]))
+      };
       if (waitingRun.checkpoint?.data && typeof waitingRun.checkpoint.data === "object") {
         const stored = waitingRun.checkpoint.data.runInteractions ?? waitingRun.checkpoint.data.phaseInteractions;
         if (stored?.requests && Array.isArray(stored.requests)) {
@@ -8144,7 +8166,8 @@ var InMemoryStore = class _InMemoryStore {
       input.expectedRevision,
       input.execution.executionId,
       input.requestMessageId,
-      input.calls
+      input.calls,
+      input.contentBlocks ?? null
     ]);
     const replay = this.replayOperation(operationKey, operationPayload);
     if (replay) return clone(replay);
@@ -8173,13 +8196,7 @@ var InMemoryStore = class _InMemoryStore {
       agentId: run.agentId,
       runId: run.id,
       role: "assistant",
-      content: toolCalls.map((toolCall) => ({
-        type: "tool_use",
-        toolCallId: toolCall.id,
-        providerToolCallId: toolCall.providerToolCallId,
-        name: toolCall.name,
-        input: clone(toolCall.args)
-      })),
+      content: requestMessageContent(toolCalls, input.contentBlocks),
       sequenceWithinRun: this.nextMessageSequence(run.id),
       createdAt: timestamp
     };
@@ -9003,6 +9020,32 @@ function digestToolEffects(toolCalls) {
 function assertToolValue(value, argument) {
   assertUtf8ByteLimit(canonicalJson(value), TOOL_VALUE_JSON_BYTES, argument);
 }
+function requestMessageContent(toolCalls, contentBlocks) {
+  const toolUse = (toolCall) => ({
+    type: "tool_use",
+    toolCallId: toolCall.id,
+    providerToolCallId: toolCall.providerToolCallId,
+    name: toolCall.name,
+    input: clone(toolCall.args)
+  });
+  if (!contentBlocks) return toolCalls.map(toolUse);
+  const reserved = new Map(toolCalls.map((toolCall) => [toolCall.providerToolCallId, toolCall]));
+  const parts = [];
+  for (const block of contentBlocks) {
+    if (block.type === "text") {
+      parts.push({ type: "text", text: block.text });
+    } else if (block.type === "thinking") {
+      parts.push({ type: "thinking", thinking: block.thinking, ...block.signature ? { signature: block.signature } : {} });
+    } else {
+      const toolCall = reserved.get(block.id);
+      if (!toolCall) continue;
+      reserved.delete(block.id);
+      parts.push(toolUse(toolCall));
+    }
+  }
+  for (const toolCall of reserved.values()) parts.push(toolUse(toolCall));
+  return parts;
+}
 
 // src/runtime/sqlite-durable-store.ts
 import { Database } from "bun:sqlite";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rowan-agent/agent",
-  "version": "0.14.0",
+  "version": "0.14.2",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -23,7 +23,7 @@
     "build": "tsup && bun scripts/check-public-interface.ts"
   },
   "dependencies": {
-    "@rowan-agent/models": "0.7.0",
+    "@rowan-agent/models": "0.7.1",
     "jiti": "2.7.0",
     "typebox": "^1.0.0",
     "yaml": "^2.7.0"