@rowan-agent/agent 0.9.9 → 0.9.11

package/dist/index.d.ts

@@ -359,6 +359,16 @@ type MessageDelta = Readonly<{
     offset: number;
     text: string;
 }>;
+type ThinkingDelta = Readonly<{
+    kind: "thinking_delta";
+    durability: "transient";
+    runId: RunId;
+    executionId: ExecutionId;
+    messageId: MessageId;
+    blockIndex: number;
+    offset: number;
+    text: string;
+}>;
 type ToolProgress = Readonly<{
     kind: "tool_progress";
     durability: "transient";
@@ -375,7 +385,7 @@ type PhaseStatusEvent = Readonly<{
     phaseId: string;
     status: PhaseStatus;
 }>;
-type RunEvent = DurableRunEvent | MessageDelta | ToolProgress | PhaseStatusEvent;
+type RunEvent = DurableRunEvent | MessageDelta | ThinkingDelta | ToolProgress | PhaseStatusEvent;
 
 interface BeforePhaseEvent {
     type: "before_phase";
@@ -2657,4 +2667,4 @@ declare function parseFrontmatter<T = Record<string, unknown>>(raw: string): Fro
 
 declare function createCoreTools(input?: CoreToolContext): Tool[];
 
-export { type AfterToolCall, type AgentConfig, type AgentConfigRequest, type AgentConfiguration, type AgentDefinition, type AgentId, type AgentListCursor, type AgentRecord, type AgentResources, type AgentRun, AgentRuntime, type AgentRuntimeOptions, type AgentSummary, type AnyRuntimeError, type AssistantContent, type AssistantMessage, type BeforeToolCall, COMPACT_PHASE_ID, type ConfigProvider, type ConfigPutResult, type ConfigResolution, type ConfigToken, type ConfigurationSnapshot, type ContextCandidate, type ContextCompactionRecord, type ContextStatus, type CoreToolContext, DEFAULT_PHASE_ID, type DefinitionLayer, type DurableConsumer, type DurableRunEvent, type DurableStore, type DurableToolResult, type EventCursor, type EventId, type ExecutionCheckpoint, type ExecutionId, type ExecutionToken, type ExtensionAPI, type ExtensionActivationError, type ExtensionActivationResult, type ExtensionContribution, type ExtensionDisposer, type ExtensionFactory, type ExtensionFactoryResult, ExtensionLifetimeError, type ExtensionLifetimeErrorCode, type ExtensionLoadInput, type FrontmatterResult, type HistorySeed, type HookEvent, type HookEventType, type HookHandler, InMemoryConfigProvider, InMemoryStore, type InputRequest, type InputRequestId, type InputRequiredCommit, type InvocationCatalogEntry, type InvocationSource, type JsonObject, type JsonPrimitive, type JsonValue, type LoadExtensionsResult, type LoadInput, type LoadResult, type LoadedExtension, type Message, type MessageBase, type MessageCommitted, type MessageContent, type MessageDelta, type MessageId, type MessageRevised, type MessageRevisionResult, type Metadata, type OpaqueId, type Outcome, type OwnerLease, type OwnerToken, type Page, type Phase, type PhaseContext, type PhaseContribution, type PhaseExecution, type PhaseExecutionIdentity, type PhaseInteraction, PhaseInteractionBoundary, PhaseInteractionCancelledError, type PhaseInteractionDriver, type PhaseInteractionKind, type PhaseInteractionState, type PhaseInteractionStatus, type PhaseInvocation, type PhaseOutput, type PhaseRegistry, type PhaseRegistrySelection, type PhaseState, type PhaseStatus, type PhaseStatusState, type ResolvedResourceView, type ResourceDiagnostic, type ResourceKind, type ResourceRef, ResourceRegistry, ResourceRegistryError, type ResourceRegistryErrorCode, type ResourceSourceId, type ResourceView, type RetentionResult, type RunBoundary, type RunClaim, type RunEvent, type RunFailure, type RunId, type RunListCursor, type RunRecord, type RunSnapshot, type RunState, type RunStateChanged, type RunSummary, RuntimeBootstrapRegistry, RuntimeError, type RuntimeErrorCode, type RuntimeErrorDetails, RuntimeExtensionLifetime, STOP_PHASE_ID, type Skill, SqliteStore, type TextContent, type ThinkingContent, type Tool, type ToolCallId, type ToolCallSnapshot, type ToolCallState, type ToolContribution, type ToolDefinition, type ToolExecutionResult$1 as ToolExecutionResult, type ToolInvocationContext, type ToolMessage, type ToolMessageContent, type ToolProgress, type ToolResultContent, type ToolStateChanged, type ToolUseContent, type UserContent, type UserInput, type UserMessage, brandConfigToken, createCompactPhase, createCorePhases, createCoreTools, createDefaultPhase, createStopPhase, isRuntimeError, loadExtensionsFromPath as loadExtensions, loadPhase, loadPhases, loadSkill, loadSkills, materializeConfigurationSnapshot, parseAgentDefinition, parseFrontmatter, resolveConfigurationSnapshot };
+export { type AfterToolCall, type AgentConfig, type AgentConfigRequest, type AgentConfiguration, type AgentDefinition, type AgentId, type AgentListCursor, type AgentRecord, type AgentResources, type AgentRun, AgentRuntime, type AgentRuntimeOptions, type AgentSummary, type AnyRuntimeError, type AssistantContent, type AssistantMessage, type BeforeToolCall, COMPACT_PHASE_ID, type ConfigProvider, type ConfigPutResult, type ConfigResolution, type ConfigToken, type ConfigurationSnapshot, type ContextCandidate, type ContextCompactionRecord, type ContextStatus, type CoreToolContext, DEFAULT_PHASE_ID, type DefinitionLayer, type DurableConsumer, type DurableRunEvent, type DurableStore, type DurableToolResult, type EventCursor, type EventId, type ExecutionCheckpoint, type ExecutionId, type ExecutionToken, type ExtensionAPI, type ExtensionActivationError, type ExtensionActivationResult, type ExtensionContribution, type ExtensionDisposer, type ExtensionFactory, type ExtensionFactoryResult, ExtensionLifetimeError, type ExtensionLifetimeErrorCode, type ExtensionLoadInput, type FrontmatterResult, type HistorySeed, type HookEvent, type HookEventType, type HookHandler, InMemoryConfigProvider, InMemoryStore, type InputRequest, type InputRequestId, type InputRequiredCommit, type InvocationCatalogEntry, type InvocationSource, type JsonObject, type JsonPrimitive, type JsonValue, type LoadExtensionsResult, type LoadInput, type LoadResult, type LoadedExtension, type Message, type MessageBase, type MessageCommitted, type MessageContent, type MessageDelta, type MessageId, type MessageRevised, type MessageRevisionResult, type Metadata, type OpaqueId, type Outcome, type OwnerLease, type OwnerToken, type Page, type Phase, type PhaseContext, type PhaseContribution, type PhaseExecution, type PhaseExecutionIdentity, type PhaseInteraction, PhaseInteractionBoundary, PhaseInteractionCancelledError, type PhaseInteractionDriver, type PhaseInteractionKind, type PhaseInteractionState, type PhaseInteractionStatus, type PhaseInvocation, type PhaseOutput, type PhaseRegistry, type PhaseRegistrySelection, type PhaseState, type PhaseStatus, type PhaseStatusState, type ResolvedResourceView, type ResourceDiagnostic, type ResourceKind, type ResourceRef, ResourceRegistry, ResourceRegistryError, type ResourceRegistryErrorCode, type ResourceSourceId, type ResourceView, type RetentionResult, type RunBoundary, type RunClaim, type RunEvent, type RunFailure, type RunId, type RunListCursor, type RunRecord, type RunSnapshot, type RunState, type RunStateChanged, type RunSummary, RuntimeBootstrapRegistry, RuntimeError, type RuntimeErrorCode, type RuntimeErrorDetails, RuntimeExtensionLifetime, STOP_PHASE_ID, type Skill, SqliteStore, type TextContent, type ThinkingContent, type ThinkingDelta, type Tool, type ToolCallId, type ToolCallSnapshot, type ToolCallState, type ToolContribution, type ToolDefinition, type ToolExecutionResult$1 as ToolExecutionResult, type ToolInvocationContext, type ToolMessage, type ToolMessageContent, type ToolProgress, type ToolResultContent, type ToolStateChanged, type ToolUseContent, type UserContent, type UserInput, type UserMessage, brandConfigToken, createCompactPhase, createCorePhases, createCoreTools, createDefaultPhase, createStopPhase, isRuntimeError, loadExtensionsFromPath as loadExtensions, loadPhase, loadPhases, loadSkill, loadSkills, materializeConfigurationSnapshot, parseAgentDefinition, parseFrontmatter, resolveConfigurationSnapshot };

package/dist/index.js

@@ -858,7 +858,8 @@ async function invokeModel(input) {
     message: input.message,
     events,
     metadataPhase: input.phaseId,
-    output: input.output
+    output: input.output,
+    onThinkingDelta: input.onThinkingDelta
   });
   const response = result.doneResponse ?? {
     content: result.text,
@@ -888,6 +889,7 @@ async function collectStreamResult(input) {
   let doneResponse;
   let streamedOutputCharacters = 0;
   const toolCallsWithDeltas = /* @__PURE__ */ new Set();
+  const thinkingOffsets = /* @__PURE__ */ new Map();
   for await (const event of input.events) {
     if (event.type === "text_delta") {
       streamedOutputCharacters += event.text.length;
@@ -939,7 +941,29 @@ async function collectStreamResult(input) {
       lastPartial = event.partial;
     }
     if (event.type === "thinking_delta") {
+      if (persistAssistant && !activeMessageId) {
+        activeMessageId = input.message.reserve("assistant", {
+          phase: input.metadataPhase
+        });
+      }
       lastPartial = event.partial;
+      if (persistAssistant && activeMessageId) {
+        let blockIndex = 0;
+        for (let index = event.partial.contentBlocks.length - 1; index >= 0; index -= 1) {
+          if (event.partial.contentBlocks[index]?.type === "thinking") {
+            blockIndex = index;
+            break;
+          }
+        }
+        const offset = thinkingOffsets.get(blockIndex) ?? 0;
+        input.onThinkingDelta?.({
+          messageId: activeMessageId,
+          blockIndex,
+          offset,
+          text: event.thinking
+        });
+        thinkingOffsets.set(blockIndex, offset + event.thinking.length);
+      }
     }
     if (event.type === "done") {
       doneResponse = event.response;
@@ -2875,7 +2899,8 @@ function createPhaseExecution(config, state, phase, messageManager, toolExecutio
             message: messageManager,
             request,
             phaseId: phase.name,
-            output: options.output
+            output: options.output,
+            onThinkingDelta: config.onThinkingDelta
           }),
           {
             signal: config.signal,
@@ -3139,6 +3164,7 @@ async function executeOnce(input) {
     beforePrompt: input.beforePrompt,
     onMessage: input.onMessage,
     onMessageDelta: input.onMessageDelta,
+    onThinkingDelta: input.onThinkingDelta,
     onOutcome: input.onOutcome,
     onModelTranscript: input.onModelTranscript,
     onPhaseStatus: input.onPhaseStatus,
@@ -4051,6 +4077,11 @@ var TransientRunEventSubscription = class {
         ...previous,
         text: previous.text + event.text
       };
+    } else if (previous?.kind === "thinking_delta" && event.kind === "thinking_delta" && previous.executionId === event.executionId && previous.messageId === event.messageId && previous.blockIndex === event.blockIndex && previous.offset + previous.text.length === event.offset) {
+      this.queue[this.queue.length - 1] = {
+        ...previous,
+        text: previous.text + event.text
+      };
     } else {
       if (this.queue.length >= MAX_BUFFERED_EVENTS) this.queue.shift();
       this.queue.push(event);
@@ -4076,7 +4107,7 @@ var TransientRunEventSubscription = class {
   clearMessage(messageId) {
     for (let index = this.queue.length - 1; index >= 0; index -= 1) {
       const event = this.queue[index];
-      if (event?.kind === "message_delta" && event.messageId === messageId) this.queue.splice(index, 1);
+      if ((event?.kind === "message_delta" || event?.kind === "thinking_delta") && event.messageId === messageId) this.queue.splice(index, 1);
     }
     this.notify();
   }
@@ -4108,11 +4139,11 @@ var TransientRunEventSubscription = class {
   }
   trimText() {
     let buffered = this.queue.reduce(
-      (length, event) => length + (event.kind === "message_delta" ? event.text.length : 0),
+      (length, event) => length + (event.kind === "message_delta" || event.kind === "thinking_delta" ? event.text.length : 0),
       0
     );
     while (buffered > MAX_BUFFERED_TEXT) {
-      const index = this.queue.findIndex((event2) => event2.kind === "message_delta");
+      const index = this.queue.findIndex((event2) => event2.kind === "message_delta" || event2.kind === "thinking_delta");
       if (index < 0) return;
       const event = this.queue[index];
       const overflow = buffered - MAX_BUFFERED_TEXT;
@@ -6036,6 +6067,20 @@ ${instructions}`,
             text: event.text
           });
         },
+        onThinkingDelta: (event) => {
+          const active = this.executions.get(run.id);
+          if (this.closed || controller.signal.aborted || active?.executionId !== claim.execution.executionId) return;
+          this.transientEvents.publish({
+            kind: "thinking_delta",
+            durability: "transient",
+            runId: run.id,
+            executionId: claim.execution.executionId,
+            messageId: event.messageId,
+            blockIndex: event.blockIndex,
+            offset: event.offset,
+            text: event.text
+          });
+        },
         onContext: assembly.setContext,
         runtime: {
           tools: ({ toolCall }) => {
@@ -6106,7 +6151,12 @@ ${instructions}`,
         return;
       }
       if (result.type === "input_required") {
-        const prompt = promptMessage(run, result.request.prompt, result.messages.length);
+        const output = latestAssistant(
+          run,
+          result.messages.slice(modelMessages.length),
+          modelMessages.length
+        );
+        const prompt = output ?? promptMessage(run, result.request.prompt, result.messages.length);
         await this.owned.commitInputRequired({
           runId: run.id,
           execution: claim.execution,

package/package.json

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