@mast-ai/core 0.1.1 → 0.2.0

package/dist/agentTool.js

@@ -22,14 +22,13 @@ export function createAgentTool(runner, agent, options) {
         definition: () => definition,
         async call(args, context) {
             const input = options.buildInput(args);
-            const builder = runner.runBuilder(agent);
+            const builder = runner.runBuilder(agent).forwardTo(context);
             if (context.signal)
                 builder.signal(context.signal);
             for await (const event of builder.runStream(input)) {
                 if (event.type === 'done') {
                     return event.output;
                 }
-                context.onEvent?.(event);
             }
             throw new AgentError(`Sub-agent '${agent.name}' stream ended without a 'done' event.`);
         },

package/dist/runner.d.ts

@@ -1,6 +1,6 @@
 import type { AgentConfig, AgentEvent, AgentResult, Message } from './types.js';
 import type { LlmAdapter } from './adapter/index.js';
-import { type ToolProvider } from './tool.js';
+import { type ToolContext, type ToolProvider } from './tool.js';
 import { Conversation } from './conversation.js';
 type StreamExecutor = (input: string, history: Message[], signal?: AbortSignal, onToolEvent?: (toolName: string, event: AgentEvent) => void) => AsyncIterable<AgentEvent>;
 /**
@@ -15,6 +15,7 @@ export declare class RunBuilder {
     private _history;
     private _signal?;
     private _onToolEvent?;
+    private _forwardContext?;
     constructor(agent: AgentConfig, execute: StreamExecutor);
     /** Prepend prior conversation turns. */
     history(messages: Message[]): this;
@@ -27,6 +28,18 @@ export declare class RunBuilder {
      * events — use `runBuilder().onToolEvent(...).runStream()` for that.
      */
     onToolEvent(handler: (toolName: string, event: AgentEvent) => void): this;
+    /**
+     * Forwards every non-`done` event yielded by this run to
+     * `parentContext.onEvent`. Intended for tools that internally run a
+     * sub-agent: chain this so the parent runner's UI can populate
+     * `subThinking` / `subText` / `nestedToolEvents` on the parent's tool
+     * entry without manual forwarding boilerplate.
+     *
+     * `done` events are filtered out to avoid leaking child conversation
+     * history to the parent's consumers. If `parentContext.onEvent` is
+     * undefined, this is a no-op.
+     */
+    forwardTo(parentContext: ToolContext): this;
     /** Executes the run and returns a stream of {@link AgentEvent} objects. */
     runStream(input: string): AsyncIterable<AgentEvent>;
     /** Executes the run and returns the final text output. */

package/dist/runner.js

@@ -15,6 +15,7 @@ export class RunBuilder {
     _history = [];
     _signal;
     _onToolEvent;
+    _forwardContext;
     constructor(agent, execute) {
         this.agent = agent;
         this.execute = execute;
@@ -39,9 +40,34 @@ export class RunBuilder {
         this._onToolEvent = handler;
         return this;
     }
+    /**
+     * Forwards every non-`done` event yielded by this run to
+     * `parentContext.onEvent`. Intended for tools that internally run a
+     * sub-agent: chain this so the parent runner's UI can populate
+     * `subThinking` / `subText` / `nestedToolEvents` on the parent's tool
+     * entry without manual forwarding boilerplate.
+     *
+     * `done` events are filtered out to avoid leaking child conversation
+     * history to the parent's consumers. If `parentContext.onEvent` is
+     * undefined, this is a no-op.
+     */
+    forwardTo(parentContext) {
+        this._forwardContext = parentContext;
+        return this;
+    }
     /** Executes the run and returns a stream of {@link AgentEvent} objects. */
     runStream(input) {
-        return this.execute(input, this._history, this._signal, this._onToolEvent);
+        const stream = this.execute(input, this._history, this._signal, this._onToolEvent);
+        const onEvent = this._forwardContext?.onEvent;
+        if (!onEvent)
+            return stream;
+        return (async function* () {
+            for await (const event of stream) {
+                if (event.type !== 'done')
+                    onEvent(event);
+                yield event;
+            }
+        })();
     }
     /** Executes the run and returns the final text output. */
     async run(input) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mast-ai/core",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "license": "Apache-2.0",
   "type": "module",
   "main": "./dist/index.js",