@ariaflowagents/core 0.6.3 → 0.7.1

package/README.md

@@ -20,9 +20,9 @@ This package is built on top of the Vercel AI SDK v6.
 This package exports:
 - **Agents:** `Agent`, `LLMAgent`, `FlowAgent`, `TriageAgent`, `CompositeAgent`
 - **Runtime:** `Runtime`, `createRuntime`
-- **Flows:** `FlowManager`, `FlowGraph`, `FlowNode`, `createFlowTransition`
-- **Session:** `SessionManager`, `SessionStore`, `MemoryStore`
-- **Tools:** `createTool`, `createToolWithFiller`, `createHandoffTool`
+- **Flows:** `FlowManager`, `FlowGraph`, `FlowNode`, `createFlowTransition`, `validateFlowConfig`
+- **Session:** `SessionManager`, `SessionStore`, `MemoryStore`, `RedisStore`
+- **Tools:** `createTool`, `createToolWithFiller`, `createHandoffTool`, `createHttpTool`
 - **System Injections:** `InjectionQueue`, `commonInjections`
 - **Prompts:** `PromptTemplateBuilder`, `PromptBuilder`
 - **Hooks:** `HookRunner`, `loggingHooks`, `createMetricsHooks`
@@ -68,6 +68,55 @@ const run = async () => {
 run();
 ```
 
+## Stream Callback Defaults (Message-Oriented)
+
+`streamCallback` is non-blocking and pluggable (file/http/db/custom sinks).  
+Current defaults are optimized for persistence pipelines:
+
+- no implicit sink (if `sinks` is omitted, callback is a no-op)
+- message mode by default (`input`, `done`, `error`, `tripwire`, tool events, and transition events)
+- token deltas off by default
+- final assistant text included as `fullText` on terminal events
+
+```ts
+import { Runtime, createFileStreamSink } from '@ariaflowagents/core';
+
+const runtime = new Runtime({
+  agents: [supportAgent],
+  defaultAgentId: 'support',
+  streamCallback: {
+    sinks: [createFileStreamSink({ directory: './transcripts' })],
+    // defaults shown explicitly:
+    eventMode: 'message',
+    emitToolEvents: true,
+    emitTransitionEvents: true,
+    emitTextDeltas: false,
+    emitFinalText: true,
+  },
+});
+```
+
+## Runtime Durability Defaults
+
+Runtime now applies lightweight durability hardening by default:
+
+- session checkpoints are persisted on `tool-result`, `tool-error`, and `flow-transition`
+- handoff state changes are checkpointed immediately after routing updates
+- replay-friendly per-turn events are stored in `session.workingMemory.runtimeEventLog`
+  (`user`, `assistant_final`, `tool_call`, `tool_result`, `tool_error`, `transition`)
+
+Tool execution context also includes `idempotencyKey` in `experimental_context`, which can be forwarded to external systems.
+
+For high-volume token streaming, opt in:
+
+```ts
+streamCallback: {
+  sinks: [createFileStreamSink({ directory: './transcripts' })],
+  eventMode: 'all',
+  emitTextDeltas: true,
+}
+```
+
 ## Routing & Handoffs (Important Defaults)
 
 AriaFlow supports invisible multi-agent routing via a `handoff` tool.
@@ -198,15 +247,19 @@ The `Runtime` class orchestrates multiple agents with seamless handoffs:
 - **TriageAgent**: Routes requests to the appropriate specialist
 - **Agent Handoffs**: Transfer conversation context between agents
 - **Session Persistence**: Maintains conversation state
+- **Flow Snapshot Memory**: Persists flow progress in `session.workingMemory.flowStateByAgent`
 
-### AgentFlowManager (Single Flow)
+### FlowManager (Single Flow)
 
-The `AgentFlowManager` class manages structured, node-based conversations:
+The `FlowManager` class manages structured, node-based conversations:
 
 - **Flow Nodes**: Each node has a specific purpose and tools
 - **State Transitions**: Tools drive transitions via `createFlowTransition()`
+- **Declarative Edge Tools**: `FlowManager` auto-injects tools for `transitions[].on` edges
 - **Flow Hooks**: Observe lifecycle events (onFlowStart, onTransition, etc.)
 - **Context Strategies**: Control memory management (append, reset, summarize)
+- **Prompt Composition**: Global role prompt + node prompt (per-node opt-out via `addGlobalPrompt: false`)
+- **Transition Contracts**: `contract.toolOnly` and `contract.requiresUserTurn` are enforced at runtime
 
 ### Date Parsing Utilities
 

package/dist/agents/Agent.d.ts

@@ -11,5 +11,6 @@ export declare abstract class Agent<TConfig extends AgentConfig = AgentConfig> {
     cleanup(_context: AgentContext): Promise<void>;
     get canHandoffTo(): string[];
     protected buildSystemPrompt(context: AgentContext): string;
+    protected getSystemPromptString(): string;
 }
 //# sourceMappingURL=Agent.d.ts.map

package/dist/agents/Agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Agent.d.ts","sourceRoot":"","sources":["../../src/agents/Agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpF,8BAAsB,KAAK,CAAC,OAAO,SAAS,WAAW,GAAG,WAAW;IACnE,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC;gBAEd,MAAM,EAAE,OAAO;IAO3B,QAAQ,CAAC,OAAO,CACd,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IAE5B,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC;IAInE,UAAU,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjD,OAAO,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpD,IAAI,YAAY,IAAI,MAAM,EAAE,CAE3B;IAED,SAAS,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM;CAU3D"}
+{"version":3,"file":"Agent.d.ts","sourceRoot":"","sources":["../../src/agents/Agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGpF,8BAAsB,KAAK,CAAC,OAAO,SAAS,WAAW,GAAG,WAAW;IACnE,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC;gBAEd,MAAM,EAAE,OAAO;IAO3B,QAAQ,CAAC,OAAO,CACd,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IAE5B,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC;IAInE,UAAU,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjD,OAAO,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpD,IAAI,YAAY,IAAI,MAAM,EAAE,CAE3B;IAED,SAAS,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM;IAW1D,SAAS,CAAC,qBAAqB,IAAI,MAAM;CAQ1C"}

package/dist/agents/Agent.js

@@ -1,3 +1,4 @@
+import { PromptBuilder } from '../prompts/PromptBuilder.js';
 export class Agent {
     id;
     name;
@@ -22,12 +23,20 @@ export class Agent {
         return this.config.canHandoffTo ?? [];
     }
     buildSystemPrompt(context) {
-        let prompt = this.config.systemPrompt;
+        let prompt = this.getSystemPromptString();
         const memory = context.workingMemory.toJSON();
         if (Object.keys(memory).length > 0) {
             prompt += `\n\n## Known Information\n${JSON.stringify(memory, null, 2)}`;
         }
         return prompt;
     }
+    getSystemPromptString() {
+        const sp = this.config.systemPrompt;
+        if (typeof sp === 'string') {
+            return sp;
+        }
+        // It's a PromptTemplate
+        return new PromptBuilder(sp).build();
+    }
 }
 //# sourceMappingURL=Agent.js.map

package/dist/agents/Agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"Agent.js","sourceRoot":"","sources":["../../src/agents/Agent.ts"],"names":[],"mappings":"AAEA,MAAM,OAAgB,KAAK;IAChB,EAAE,CAAS;IACX,IAAI,CAAS;IACb,WAAW,CAAU;IACpB,MAAM,CAAU;IAE1B,YAAY,MAAe;QACzB,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAOD,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,QAAsB;QACpD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAsB;QACrC,yBAAyB;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAsB;QAClC,yBAAyB;IAC3B,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;IACxC,CAAC;IAES,iBAAiB,CAAC,OAAqB;QAC/C,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAEtC,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;QAC9C,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,6BAA6B,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC3E,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}
+{"version":3,"file":"Agent.js","sourceRoot":"","sources":["../../src/agents/Agent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAE5D,MAAM,OAAgB,KAAK;IAChB,EAAE,CAAS;IACX,IAAI,CAAS;IACb,WAAW,CAAU;IACpB,MAAM,CAAU;IAE1B,YAAY,MAAe;QACzB,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAOD,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,QAAsB;QACpD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAsB;QACrC,yBAAyB;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAsB;QAClC,yBAAyB;IAC3B,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;IACxC,CAAC;IAES,iBAAiB,CAAC,OAAqB;QAC/C,IAAI,MAAM,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAE1C,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;QAC9C,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,6BAA6B,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC3E,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAES,qBAAqB;QAC7B,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QACpC,IAAI,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC3B,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,wBAAwB;QACxB,OAAO,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;IACvC,CAAC;CACF"}

package/dist/agents/CompositeAgent.d.ts

@@ -1,24 +1,20 @@
-import type { AgentContext, AgentStreamPart } from '../types/index.js';
-import type { Agent } from './Agent.js';
-export interface CompositeAgentConfig {
-    id: string;
-    name: string;
-    description?: string;
+import type { AgentConfig, AgentContext, AgentStreamPart } from '../types/index.js';
+import { Agent } from './Agent.js';
+export interface CompositeAgentConfig extends AgentConfig {
+    type: 'composite';
     agents: Agent[];
     entryAgent: string;
     maxHandoffs?: number;
 }
-export declare class CompositeAgent {
-    readonly id: string;
-    readonly name: string;
-    readonly description?: string;
+export declare class CompositeAgent extends Agent<CompositeAgentConfig> {
     private agents;
     private entryAgentId;
     private maxHandoffs;
     constructor(config: CompositeAgentConfig);
+    initialize(context: AgentContext): Promise<void>;
+    cleanup(context: AgentContext): Promise<void>;
     process(input: string, context: AgentContext): AsyncGenerator<AgentStreamPart>;
     getAgent(id: string): Agent | undefined;
-    get agentIds(): string[];
-    get entryAgent(): string;
+    get subAgentIds(): string[];
 }
 //# sourceMappingURL=CompositeAgent.d.ts.map

package/dist/agents/CompositeAgent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"CompositeAgent.d.ts","sourceRoot":"","sources":["../../src/agents/CompositeAgent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAExC,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,KAAK,EAAE,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,cAAc;IACzB,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,WAAW,CAAS;gBAEhB,MAAM,EAAE,oBAAoB;IAYjC,OAAO,CACZ,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IAwClC,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,KAAK,GAAG,SAAS;IAIvC,IAAI,QAAQ,IAAI,MAAM,EAAE,CAEvB;IAED,IAAI,UAAU,IAAI,MAAM,CAEvB;CACF"}
+{"version":3,"file":"CompositeAgent.d.ts","sourceRoot":"","sources":["../../src/agents/CompositeAgent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpF,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEnC,MAAM,WAAW,oBAAqB,SAAQ,WAAW;IACvD,IAAI,EAAE,WAAW,CAAC;IAClB,MAAM,EAAE,KAAK,EAAE,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,cAAe,SAAQ,KAAK,CAAC,oBAAoB,CAAC;IAC7D,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,WAAW,CAAS;gBAEhB,MAAM,EAAE,oBAAoB;IAUlC,UAAU,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAOhD,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAO5C,OAAO,CACZ,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IAmDlC,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,KAAK,GAAG,SAAS;IAIvC,IAAI,WAAW,IAAI,MAAM,EAAE,CAE1B;CACF"}

package/dist/agents/CompositeAgent.js

@@ -1,32 +1,41 @@
-export class CompositeAgent {
-    id;
-    name;
-    description;
+import { Agent } from './Agent.js';
+export class CompositeAgent extends Agent {
     agents = new Map();
     entryAgentId;
     maxHandoffs;
     constructor(config) {
-        this.id = config.id;
-        this.name = config.name;
-        this.description = config.description;
+        super(config);
         this.entryAgentId = config.entryAgent;
         this.maxHandoffs = config.maxHandoffs ?? 10;
         for (const agent of config.agents) {
             this.agents.set(agent.id, agent);
         }
     }
+    async initialize(context) {
+        await super.initialize(context);
+        for (const agent of this.agents.values()) {
+            await agent.initialize(context);
+        }
+    }
+    async cleanup(context) {
+        for (const agent of this.agents.values()) {
+            await agent.cleanup(context);
+        }
+        await super.cleanup(context);
+    }
     async *process(input, context) {
-        let currentAgentId = context.currentAgent || this.entryAgentId;
+        let currentAgentId = context.session.activeAgentId || this.entryAgentId;
         let currentInput = input;
         let handoffCount = 0;
         while (handoffCount < this.maxHandoffs) {
             const agent = this.agents.get(currentAgentId);
             if (!agent) {
-                yield { type: 'error', error: `Agent "${currentAgentId}" not found` };
+                yield { type: 'error', error: `Agent "${currentAgentId}" not found in composite agent "${this.id}"` };
                 return;
             }
             let handoffTriggered = false;
             let nextAgentId;
+            // Delegate processing to the sub-agent
             for await (const part of agent.process(currentInput, context)) {
                 yield part;
                 if (part.type === 'handoff') {
@@ -34,26 +43,31 @@ export class CompositeAgent {
                     nextAgentId = part.targetAgent;
                     handoffCount += 1;
                 }
+                if (part.type === 'turn-end' && part.metadata?.response) {
+                    // In a composite agent, we might want to collect all messages 
+                    // but for now we just yield it up so the orchestrator can persist.
+                }
             }
             if (!handoffTriggered || !nextAgentId) {
                 break;
             }
+            // If the target agent is not part of this composite, stop internal loop and let runtime handle it
+            if (!this.agents.has(nextAgentId)) {
+                break;
+            }
             currentAgentId = nextAgentId;
-            context.currentAgent = nextAgentId;
-            currentInput = '';
+            context.session.activeAgentId = nextAgentId;
+            currentInput = ''; // Clear input for the next agent in the sequence
         }
         if (handoffCount >= this.maxHandoffs) {
-            yield { type: 'error', error: 'Maximum handoffs exceeded' };
+            yield { type: 'error', error: `Maximum internal handoffs (${this.maxHandoffs}) exceeded in composite agent "${this.id}"` };
         }
     }
     getAgent(id) {
         return this.agents.get(id);
     }
-    get agentIds() {
+    get subAgentIds() {
         return Array.from(this.agents.keys());
     }
-    get entryAgent() {
-        return this.entryAgentId;
-    }
 }
 //# sourceMappingURL=CompositeAgent.js.map

package/dist/agents/CompositeAgent.js.map

@@ -1 +1 @@
-{"version":3,"file":"CompositeAgent.js","sourceRoot":"","sources":["../../src/agents/CompositeAgent.ts"],"names":[],"mappings":"AAYA,MAAM,OAAO,cAAc;IAChB,EAAE,CAAS;IACX,IAAI,CAAS;IACb,WAAW,CAAU;IACtB,MAAM,GAAG,IAAI,GAAG,EAAiB,CAAC;IAClC,YAAY,CAAS;IACrB,WAAW,CAAS;IAE5B,YAAY,MAA4B;QACtC,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,UAAU,CAAC;QACtC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;QAE5C,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,KAAa,EACb,OAAqB;QAErB,IAAI,cAAc,GAAG,OAAO,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC;QAC/D,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,IAAI,YAAY,GAAG,CAAC,CAAC;QAErB,OAAO,YAAY,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAE9C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,cAAc,aAAa,EAAE,CAAC;gBACtE,OAAO;YACT,CAAC;YAED,IAAI,gBAAgB,GAAG,KAAK,CAAC;YAC7B,IAAI,WAA+B,CAAC;YAEpC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC9D,MAAM,IAAI,CAAC;gBAEX,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBAC5B,gBAAgB,GAAG,IAAI,CAAC;oBACxB,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;oBAC/B,YAAY,IAAI,CAAC,CAAC;gBACpB,CAAC;YACH,CAAC;YAED,IAAI,CAAC,gBAAgB,IAAI,CAAC,WAAW,EAAE,CAAC;gBACtC,MAAM;YACR,CAAC;YAED,cAAc,GAAG,WAAW,CAAC;YAC7B,OAAO,CAAC,YAAY,GAAG,WAAW,CAAC;YACnC,YAAY,GAAG,EAAE,CAAC;QACpB,CAAC;QAED,IAAI,YAAY,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,2BAA2B,EAAE,CAAC;QAC9D,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;CACF"}
+{"version":3,"file":"CompositeAgent.js","sourceRoot":"","sources":["../../src/agents/CompositeAgent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AASnC,MAAM,OAAO,cAAe,SAAQ,KAA2B;IACrD,MAAM,GAAG,IAAI,GAAG,EAAiB,CAAC;IAClC,YAAY,CAAS;IACrB,WAAW,CAAS;IAE5B,YAAY,MAA4B;QACtC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,UAAU,CAAC;QACtC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;QAE5C,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAqB;QACpC,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAChC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAqB;QACjC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QACD,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,KAAa,EACb,OAAqB;QAErB,IAAI,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,IAAI,IAAI,CAAC,YAAY,CAAC;QACxE,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,IAAI,YAAY,GAAG,CAAC,CAAC;QAErB,OAAO,YAAY,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAE9C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,cAAc,mCAAmC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;gBACtG,OAAO;YACT,CAAC;YAED,IAAI,gBAAgB,GAAG,KAAK,CAAC;YAC7B,IAAI,WAA+B,CAAC;YAEpC,uCAAuC;YACvC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC9D,MAAM,IAAI,CAAC;gBAEX,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBAC5B,gBAAgB,GAAG,IAAI,CAAC;oBACxB,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;oBAC/B,YAAY,IAAI,CAAC,CAAC;gBACpB,CAAC;gBAED,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,CAAC;oBACxD,+DAA+D;oBAC/D,mEAAmE;gBACrE,CAAC;YACH,CAAC;YAED,IAAI,CAAC,gBAAgB,IAAI,CAAC,WAAW,EAAE,CAAC;gBACtC,MAAM;YACR,CAAC;YAED,kGAAkG;YAClG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;gBAClC,MAAM;YACR,CAAC;YAED,cAAc,GAAG,WAAW,CAAC;YAC7B,OAAO,CAAC,OAAO,CAAC,aAAa,GAAG,WAAW,CAAC;YAC5C,YAAY,GAAG,EAAE,CAAC,CAAC,iDAAiD;QACtE,CAAC;QAED,IAAI,YAAY,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,8BAA8B,IAAI,CAAC,WAAW,kCAAkC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QAC7H,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,WAAW;QACb,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IACxC,CAAC;CACF"}

package/dist/agents/FlowAgent.js

@@ -14,9 +14,9 @@ export class FlowAgent extends Agent {
                 flow: this.config.flow,
                 initialNode: this.config.initialNode,
                 model: this.config.model,
-                defaultRolePrompt: this.config.systemPrompt,
+                defaultRolePrompt: this.getSystemPromptString(),
                 contextMessages: context.messages,
-                sessionMessages: context.session.messages,
+                abortSignal: context.abortSignal,
             });
             for await (const part of this.flowManager.initialize()) {
                 yield part;

package/dist/agents/FlowAgent.js.map

@@ -1 +1 @@
-{"version":3,"file":"FlowAgent.js","sourceRoot":"","sources":["../../src/agents/FlowAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAGtD,MAAM,OAAO,SAAU,SAAQ,KAAsB;IAC3C,WAAW,CAAe;IAElC,YAAY,MAAuB;QACjC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,cAAc,MAAM,CAAC,EAAE,oBAAoB,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,KAAa,EACb,OAAqB;QAErB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAM;gBACzB,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBAC3C,eAAe,EAAE,OAAO,CAAC,QAAQ;gBACjC,eAAe,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ;aAC1C,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC;gBACvD,MAAM,IAAI,CAAC;YACb,CAAC;QACH,CAAC;QAED,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzD,MAAM,IAAI,CAAC;YAEX,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC7B,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;YAC/B,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,WAAW,EAAE,aAAa,IAAI,EAAE,CAAC;IAC/C,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAY;QAC7B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;CACF"}
+{"version":3,"file":"FlowAgent.js","sourceRoot":"","sources":["../../src/agents/FlowAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAGtD,MAAM,OAAO,SAAU,SAAQ,KAAsB;IAC3C,WAAW,CAAe;IAElC,YAAY,MAAuB;QACjC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,cAAc,MAAM,CAAC,EAAE,oBAAoB,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,KAAa,EACb,OAAqB;QAErB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAM;gBACzB,iBAAiB,EAAE,IAAI,CAAC,qBAAqB,EAAE;gBAC/C,eAAe,EAAE,OAAO,CAAC,QAAQ;gBACjC,WAAW,EAAE,OAAO,CAAC,WAAW;aACjC,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC;gBACvD,MAAM,IAAI,CAAC;YACb,CAAC;QACH,CAAC;QAED,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzD,MAAM,IAAI,CAAC;YAEX,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC7B,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;YAC/B,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,WAAW,EAAE,aAAa,IAAI,EAAE,CAAC;IAC/C,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAY;QAC7B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;CACF"}

package/dist/agents/LLMAgent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"LLMAgent.d.ts","sourceRoot":"","sources":["../../src/agents/LLMAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkC,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAClE,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAGvF,qBAAa,QAAS,SAAQ,KAAK,CAAC,cAAc,CAAC;IACjD,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,KAAK,CAAU;gBAEX,MAAM,EAAE,cAAc;IAS3B,OAAO,CACZ,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IAyClC,SAAS,CAAC,kBAAkB,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO;CAG9D"}
+{"version":3,"file":"LLMAgent.d.ts","sourceRoot":"","sources":["../../src/agents/LLMAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkC,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAClE,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAIvF,qBAAa,QAAS,SAAQ,KAAK,CAAC,cAAc,CAAC;IACjD,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,KAAK,CAAU;gBAEX,MAAM,EAAE,cAAc;IAS3B,OAAO,CACZ,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IAgBlC,SAAS,CAAC,kBAAkB,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO;CAG9D"}

package/dist/agents/LLMAgent.js

@@ -1,6 +1,6 @@
 import { streamText } from 'ai';
 import { Agent } from './Agent.js';
-import { isHandoffResult } from '../tools/handoff.js';
+import { processAIStream } from '../utils/aiStream.js';
 export class LLMAgent extends Agent {
     model;
     tools;
@@ -19,31 +19,10 @@ export class LLMAgent extends Agent {
             system: systemPrompt,
             messages: context.messages,
             tools: this.getToolsForContext(context),
+            abortSignal: context.abortSignal,
         });
-        for await (const chunk of result.fullStream) {
-            if (chunk.type === 'text-delta') {
-                yield { type: 'text-delta', text: chunk.text };
-            }
-            if (chunk.type === 'tool-call') {
-                const args = 'args' in chunk ? chunk.args : chunk.input;
-                yield { type: 'tool-call', toolName: chunk.toolName, args };
-            }
-            if (chunk.type === 'tool-result') {
-                const toolResult = 'result' in chunk ? chunk.result : chunk.output;
-                yield { type: 'tool-result', toolName: chunk.toolName, result: toolResult };
-                if (isHandoffResult(toolResult)) {
-                    const targetAgent = toolResult.targetAgent ?? toolResult.targetAgentId;
-                    yield {
-                        type: 'handoff',
-                        targetAgent,
-                        reason: toolResult.reason,
-                    };
-                }
-            }
-        }
-        const response = await result.response;
-        context.session.messages.push(...response.messages);
-        yield { type: 'turn-end' };
+        yield* processAIStream(result.fullStream);
+        yield { type: 'turn-end', metadata: { response: await result.response } };
     }
     getToolsForContext(_context) {
         return this.tools;

package/dist/agents/LLMAgent.js.map

@@ -1 +1 @@
-{"version":3,"file":"LLMAgent.js","sourceRoot":"","sources":["../../src/agents/LLMAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAoC,MAAM,IAAI,CAAC;AAClE,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,MAAM,OAAO,QAAS,SAAQ,KAAqB;IACzC,KAAK,CAAgB;IACrB,KAAK,CAAU;IAEvB,YAAY,MAAsB;QAChC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,aAAa,MAAM,CAAC,EAAE,oBAAoB,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,MAAc,EACd,OAAqB;QAErB,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAErD,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,KAAK,EAAE,IAAI,CAAC,KAAY;YACxB,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;SACxC,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;YACjD,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAE,KAAa,CAAC,KAAK,CAAC;gBACjE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC9D,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gBACjC,MAAM,UAAU,GAAG,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAE,KAAa,CAAC,MAAM,CAAC;gBAC5E,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;gBAE5E,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;oBAChC,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,aAAa,CAAC;oBACvE,MAAM;wBACJ,IAAI,EAAE,SAAS;wBACf,WAAW;wBACX,MAAM,EAAE,UAAU,CAAC,MAAM;qBAC1B,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;QACvC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEpD,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IAC7B,CAAC;IAES,kBAAkB,CAAC,QAAsB;QACjD,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;CACF"}
+{"version":3,"file":"LLMAgent.js","sourceRoot":"","sources":["../../src/agents/LLMAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAoC,MAAM,IAAI,CAAC;AAClE,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAGnC,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,OAAO,QAAS,SAAQ,KAAqB;IACzC,KAAK,CAAgB;IACrB,KAAK,CAAU;IAEvB,YAAY,MAAsB;QAChC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,aAAa,MAAM,CAAC,EAAE,oBAAoB,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,MAAc,EACd,OAAqB;QAErB,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAErD,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE,OAAO,CAAC,QAAe;YACjC,KAAK,EAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAS;YAChD,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC,CAAC;QAEH,KAAK,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAE1C,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;IAC5E,CAAC;IAES,kBAAkB,CAAC,QAAsB;QACjD,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;CACF"}

package/dist/agents/TriageAgent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TriageAgent.d.ts","sourceRoot":"","sources":["../../src/agents/TriageAgent.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,EAAE,YAAY,EAAc,eAAe,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAGtG,qBAAa,WAAY,SAAQ,KAAK,CAAC,iBAAiB,CAAC;IACvD,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,YAAY,CAAC,CAAS;gBAElB,MAAM,EAAE,iBAAiB;IAS9B,OAAO,CACZ,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IA6ClC,OAAO,CAAC,kBAAkB;IAsB1B,OAAO,CAAC,iBAAiB;CAmB1B"}
+{"version":3,"file":"TriageAgent.d.ts","sourceRoot":"","sources":["../../src/agents/TriageAgent.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,EAAE,YAAY,EAAc,eAAe,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAKtG,qBAAa,WAAY,SAAQ,KAAK,CAAC,iBAAiB,CAAC;IACvD,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,YAAY,CAAC,CAAS;gBAElB,MAAM,EAAE,iBAAiB;IAS9B,OAAO,CACZ,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,YAAY,GACpB,cAAc,CAAC,eAAe,CAAC;IAoBlC,OAAO,CAAC,kBAAkB;IAsB1B,OAAO,CAAC,iBAAiB;CAmB1B"}

package/dist/agents/TriageAgent.js

@@ -1,7 +1,7 @@
 import { streamText, tool } from 'ai';
 import { z } from 'zod';
 import { Agent } from './Agent.js';
-import { isHandoffResult } from '../tools/handoff.js';
+import { processAIStream } from '../utils/aiStream.js';
 export class TriageAgent extends Agent {
     routes;
     defaultAgent;
@@ -14,50 +14,29 @@ export class TriageAgent extends Agent {
         this.defaultAgent = config.defaultAgent;
     }
     async *process(_input, context) {
-        const systemPrompt = this.buildRoutingPrompt();
+        const systemPrompt = this.buildRoutingPrompt(context);
         const handoffTool = this.createHandoffTool();
         const result = streamText({
             model: this.config.model,
             system: systemPrompt,
             messages: context.messages,
+            abortSignal: context.abortSignal,
             tools: {
                 handoff_to_agent: handoffTool,
                 ...this.config.tools,
             },
         });
-        for await (const chunk of result.fullStream) {
-            if (chunk.type === 'text-delta') {
-                yield { type: 'text-delta', text: chunk.text };
-            }
-            if (chunk.type === 'tool-call') {
-                const args = 'args' in chunk ? chunk.args : chunk.input;
-                yield { type: 'tool-call', toolName: chunk.toolName, args };
-            }
-            if (chunk.type === 'tool-result') {
-                const toolResult = 'result' in chunk ? chunk.result : chunk.output;
-                yield { type: 'tool-result', toolName: chunk.toolName, result: toolResult };
-                if (isHandoffResult(toolResult)) {
-                    const targetAgent = toolResult.targetAgent ?? toolResult.targetAgentId;
-                    yield {
-                        type: 'handoff',
-                        targetAgent,
-                        reason: toolResult.reason,
-                    };
-                }
-            }
-        }
-        const response = await result.response;
-        context.session.messages.push(...response.messages);
-        yield { type: 'turn-end' };
+        yield* processAIStream(result.fullStream);
+        yield { type: 'turn-end', metadata: { response: await result.response } };
     }
-    buildRoutingPrompt() {
+    buildRoutingPrompt(context) {
         const routeDescriptions = this.routes
             .map(route => `- **${route.agentId}**: ${route.description}`)
             .join('\n');
         const defaultNote = this.defaultAgent
             ? `\n- Use "${this.defaultAgent}" when no specialist applies.`
             : '';
-        return `${this.config.systemPrompt}
+        return `${this.buildSystemPrompt(context)}
 
 ## Available Specialists
 ${routeDescriptions}
@@ -65,7 +44,7 @@ ${routeDescriptions}
 ## Instructions
 - Follow the system prompt above.
 - Use the handoff_to_agent tool when a specialist is needed. 
-- Provide a very brief, natural transition (e.g., "I'll check that for you...") instead of stating "I am handing you off".
+- You MAY provide a very brief, natural transition (e.g., "I'll check that for you...") or remain silent. Do not state "I am handing you off".
 - Keep the transition content separate from the specialist's likely greeting.
 - Always provide a reason for the handoff via the tool${defaultNote}`;
     }

package/dist/agents/TriageAgent.js.map

@@ -1 +1 @@
-{"version":3,"file":"TriageAgent.js","sourceRoot":"","sources":["../../src/agents/TriageAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AACtC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,MAAM,OAAO,WAAY,SAAQ,KAAwB;IAC/C,MAAM,CAAe;IACrB,YAAY,CAAU;IAE9B,YAAY,MAAyB;QACnC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,gBAAgB,MAAM,CAAC,EAAE,oBAAoB,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,MAAc,EACd,OAAqB;QAErB,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC/C,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE7C,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAY;YAC/B,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE;gBACL,gBAAgB,EAAE,WAAW;gBAC7B,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK;aACrB;SACF,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;YACjD,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAE,KAAa,CAAC,KAAK,CAAC;gBACjE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC9D,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gBACjC,MAAM,UAAU,GAAG,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAE,KAAa,CAAC,MAAM,CAAC;gBAC5E,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;gBAE5E,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;oBAChC,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,aAAa,CAAC;oBACvE,MAAM;wBACJ,IAAI,EAAE,SAAS;wBACf,WAAW;wBACX,MAAM,EAAE,UAAU,CAAC,MAAM;qBAC1B,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;QACvC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEpD,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IAC7B,CAAC;IAEO,kBAAkB;QACxB,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM;aAClC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,KAAK,CAAC,OAAO,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC;aAC5D,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY;YACnC,CAAC,CAAC,YAAY,IAAI,CAAC,YAAY,+BAA+B;YAC9D,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY;;;EAGpC,iBAAiB;;;;;;;wDAOqC,WAAW,EAAE,CAAC;IACpE,CAAC;IAEO,iBAAiB;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEzD,OAAO,IAAI,CAAC;YACV,WAAW,EAAE,wEAAwE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAC1G,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,QAAiC,CAAC;gBACxD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;aAC9D,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,EAAE;gBAC3C,OAAO;oBACL,SAAS,EAAE,IAAI;oBACf,aAAa;oBACb,WAAW,EAAE,aAAa;oBAC1B,MAAM;iBACP,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}
+{"version":3,"file":"TriageAgent.js","sourceRoot":"","sources":["../../src/agents/TriageAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AACtC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAInC,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,OAAO,WAAY,SAAQ,KAAwB;IAC/C,MAAM,CAAe;IACrB,YAAY,CAAU;IAE9B,YAAY,MAAyB;QACnC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,gBAAgB,MAAM,CAAC,EAAE,oBAAoB,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CACZ,MAAc,EACd,OAAqB;QAErB,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QACtD,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE7C,MAAM,MAAM,GAAG,UAAU,CAAC;YACxB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAM;YACzB,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE,OAAO,CAAC,QAAe;YACjC,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,KAAK,EAAE;gBACL,gBAAgB,EAAE,WAAW;gBAC7B,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK;aACd;SACT,CAAC,CAAC;QAEH,KAAK,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAE1C,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;IAC5E,CAAC;IAEO,kBAAkB,CAAC,OAAqB;QAC9C,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM;aAClC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,KAAK,CAAC,OAAO,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC;aAC5D,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY;YACnC,CAAC,CAAC,YAAY,IAAI,CAAC,YAAY,+BAA+B;YAC9D,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC;;;EAG3C,iBAAiB;;;;;;;wDAOqC,WAAW,EAAE,CAAC;IACpE,CAAC;IAEO,iBAAiB;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEzD,OAAO,IAAI,CAAC;YACV,WAAW,EAAE,wEAAwE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAC1G,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,QAAiC,CAAC;gBACxD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;aAC9D,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,EAAE;gBAC3C,OAAO;oBACL,SAAS,EAAE,IAAI;oBACf,aAAa;oBACb,WAAW,EAAE,aAAa;oBAC1B,MAAM;iBACP,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/callbacks/httpCallback.d.ts

@@ -6,6 +6,7 @@ export interface HttpCallbackConfig {
     allowList?: string[];
     denyList?: string[];
     includeFullText?: boolean;
+    timeoutMs?: number;
 }
 export interface HttpCallbackPayload {
     sessionId: string;

package/dist/callbacks/httpCallback.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"httpCallback.d.ts","sourceRoot":"","sources":["../../src/callbacks/httpCallback.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEvE,MAAM,WAAW,kBAAkB;IACjC,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,iBAAiB,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAoCD,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,KAAK,OAAO,CAAC,IAAI,CAAC,CAqC9H;AAED,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAE5D"}
+{"version":3,"file":"httpCallback.d.ts","sourceRoot":"","sources":["../../src/callbacks/httpCallback.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEvE,MAAM,WAAW,kBAAkB;IACjC,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,iBAAiB,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAkDD,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,KAAK,OAAO,CAAC,IAAI,CAAC,CAgD9H;AAED,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAE5D"}

package/dist/callbacks/httpCallback.js

@@ -14,31 +14,44 @@ function shouldSendEvent(eventType, config) {
     }
     return true;
 }
-async function sendHttpRequest(url, method, headers, payload) {
+async function sendHttpRequest(url, method, headers, payload, timeoutMs) {
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort('HTTP callback timeout'), timeoutMs);
     try {
-        await fetch(url, {
+        const response = await fetch(url, {
             method,
             headers: {
                 'Content-Type': 'application/json',
                 ...headers,
             },
             body: JSON.stringify(payload),
+            signal: controller.signal,
         });
+        if (!response.ok) {
+            console.error(`[HttpCallback] Webhook returned ${response.status}`);
+        }
     }
     catch (error) {
         console.error(`[HttpCallback] Failed to send webhook: ${error.message}`);
     }
+    finally {
+        clearTimeout(timeout);
+    }
 }
 export function createHttpCallback(config) {
     const resolvedHeaders = config.headers
         ? Object.fromEntries(Object.entries(config.headers).map(([key, value]) => [key, resolveEnvVars(value)]))
         : undefined;
     return async (context, part) => {
+        const isTerminal = part.type === 'turn-end' || part.type === 'error' || part.type === 'done';
         if (part.type === 'text-delta') {
             const currentText = TEXT_ACCUMULATOR.get(context.session.id) || '';
             TEXT_ACCUMULATOR.set(context.session.id, currentText + part.text);
         }
         if (!shouldSendEvent(part.type, config)) {
+            if (isTerminal) {
+                TEXT_ACCUMULATOR.delete(context.session.id);
+            }
             return;
         }
         const payload = {
@@ -53,10 +66,11 @@ export function createHttpCallback(config) {
                 payload.fullText = fullText;
             }
         }
-        if (part.type === 'turn-end' || part.type === 'error' || part.type === 'done') {
-            TEXT_ACCUMULATOR.delete(context.session.id);
-        }
-        sendHttpRequest(resolveEnvVars(config.url), config.method || 'POST', resolvedHeaders, payload);
+        void sendHttpRequest(resolveEnvVars(config.url), config.method || 'POST', resolvedHeaders, payload, config.timeoutMs ?? 5000).finally(() => {
+            if (isTerminal) {
+                TEXT_ACCUMULATOR.delete(context.session.id);
+            }
+        });
     };
 }
 export function clearTextAccumulator(sessionId) {

package/dist/callbacks/httpCallback.js.map

@@ -1 +1 @@
-{"version":3,"file":"httpCallback.js","sourceRoot":"","sources":["../../src/callbacks/httpCallback.ts"],"names":[],"mappings":"AAmBA,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAkB,CAAC;AAEnD,SAAS,cAAc,CAAC,KAAa;IACnC,OAAO,KAAK,CAAC,OAAO,CAAC,yCAAyC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QACxF,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpC,OAAO,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,SAAiB,EAAE,MAA0B;IACpE,IAAI,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpD,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,GAAW,EAAE,MAAsB,EAAE,OAA2C,EAAE,OAA4B;IAC3I,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,GAAG,EAAE;YACf,MAAM;YACN,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,OAAO;aACX;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;SAC9B,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,0CAA2C,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO;QACpC,CAAC,CAAC,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CACnF;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO,KAAK,EAAE,OAAmB,EAAE,IAAuB,EAAE,EAAE;QAC5D,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC/B,MAAM,WAAW,GAAG,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACnE,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAwB;YACnC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;YAC7B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,IAAI;SACL,CAAC;QAEF,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1D,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC9E,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,eAAe,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,IAAI,MAAM,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;IACjG,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,SAAiB;IACpD,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AACrC,CAAC"}
+{"version":3,"file":"httpCallback.js","sourceRoot":"","sources":["../../src/callbacks/httpCallback.ts"],"names":[],"mappings":"AAoBA,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAkB,CAAC;AAEnD,SAAS,cAAc,CAAC,KAAa;IACnC,OAAO,KAAK,CAAC,OAAO,CAAC,yCAAyC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QACxF,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpC,OAAO,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,SAAiB,EAAE,MAA0B;IACpE,IAAI,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpD,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,GAAW,EACX,MAAsB,EACtB,OAA2C,EAC3C,OAA4B,EAC5B,SAAiB;IAEjB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,uBAAuB,CAAC,EAAE,SAAS,CAAC,CAAC;IACvF,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM;YACN,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,OAAO;aACX;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC7B,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,OAAO,CAAC,KAAK,CAAC,mCAAmC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,0CAA2C,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;IACtF,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO;QACpC,CAAC,CAAC,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CACnF;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO,KAAK,EAAE,OAAmB,EAAE,IAAuB,EAAE,EAAE;QAC5D,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;QAE7F,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC/B,MAAM,WAAW,GAAG,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACnE,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC;YACxC,IAAI,UAAU,EAAE,CAAC;gBACf,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC9C,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAwB;YACnC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;YAC7B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,IAAI;SACL,CAAC;QAEF,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1D,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,KAAK,eAAe,CAClB,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,EAC1B,MAAM,CAAC,MAAM,IAAI,MAAM,EACvB,eAAe,EACf,OAAO,EACP,MAAM,CAAC,SAAS,IAAI,IAAI,CACzB,CAAC,OAAO,CAAC,GAAG,EAAE;YACb,IAAI,UAAU,EAAE,CAAC;gBACf,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,SAAiB;IACpD,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AACrC,CAAC"}

package/dist/callbacks/streamCallback.d.ts

@@ -0,0 +1,26 @@
+import type { HarnessStreamPart, RunContext, StreamCallbackConfig, StreamCallbackPayload, StreamCallbackSink } from '../types/index.js';
+export interface StreamCallbackAdapter {
+    onStreamPart: (context: RunContext, part: HarnessStreamPart) => Promise<void>;
+    flush: (timeoutMs?: number) => Promise<void>;
+    close: (timeoutMs?: number) => Promise<void>;
+}
+export declare function createConsoleStreamSink(options?: {
+    logger?: Pick<Console, 'log'>;
+    prefix?: string;
+}): StreamCallbackSink;
+export declare function createFileStreamSink(config: {
+    path?: string;
+    directory?: string;
+    filename?: string | ((payload: StreamCallbackPayload) => string);
+    mkdirp?: boolean;
+    format?: 'jsonl' | 'text';
+}): StreamCallbackSink;
+export declare function createFunctionStreamSink(handler: (payload: StreamCallbackPayload) => Promise<void> | void, name?: string): StreamCallbackSink;
+export declare function createHttpStreamSink(config: {
+    url: string;
+    method?: 'POST' | 'PUT';
+    headers?: Record<string, string>;
+    timeoutMs?: number;
+}): StreamCallbackSink;
+export declare function createStreamCallbackAdapter(config?: StreamCallbackConfig): StreamCallbackAdapter;
+//# sourceMappingURL=streamCallback.d.ts.map

package/dist/callbacks/streamCallback.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamCallback.d.ts","sourceRoot":"","sources":["../../src/callbacks/streamCallback.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AA8HxI,MAAM,WAAW,qBAAqB;IACpC,YAAY,EAAE,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9E,KAAK,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,KAAK,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9C;AAED,wBAAgB,uBAAuB,CAAC,OAAO,GAAE;IAC/C,MAAM,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;CACZ,GAAG,kBAAkB,CAU1B;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,GAAG,CAAC,CAAC,OAAO,EAAE,qBAAqB,KAAK,MAAM,CAAC,CAAC;IACjE,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,OAAO,GAAG,MAAM,CAAC;CAC3B,GAAG,kBAAkB,CA8CrB;AAED,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,CAAC,OAAO,EAAE,qBAAqB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,EACjE,IAAI,GAAE,MAAmB,GACxB,kBAAkB,CAOpB;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE;IAC3C,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,GAAG,kBAAkB,CAoCrB;AAED,wBAAgB,2BAA2B,CAAC,MAAM,GAAE,oBAAyB,GAAG,qBAAqB,CAsFpG"}