@theokit/sdk 2.22.0 → 2.24.0

package/dist/a2a/subagent.d.cts

@@ -25,6 +25,12 @@ export interface MessageFilterArgs {
 export interface DelegationStartContext {
     input: string;
     name: string;
+    /**
+     * SE15 — 1-based count of times THIS subagent tool has been invoked (a
+     * per-`defineSubAgent`-instance counter). Incremented before this hook runs;
+     * a rejected delegation still counts. Enables reject-after-N patterns.
+     */
+    iteration: number;
 }
 /**
  * Decision returned from {@link SubAgentSpec.onDelegationStart}. Discriminated on
@@ -37,6 +43,8 @@ export type DelegationStartDecision = {
 } | {
     proceed?: true;
     modifiedInput?: string;
+    /** SE13 — cap the child's iteration count (forwarded as `SendOptions.maxIterations`). */
+    modifiedMaxSteps?: number;
 };
 /** Context passed to {@link SubAgentSpec.onDelegationComplete} after the child settles. */
 export interface DelegationCompleteContext {
@@ -46,7 +54,15 @@ export interface DelegationCompleteContext {
     result?: string;
     /** The error the child threw (present on failure); the error is still re-thrown. */
     error?: unknown;
+    /** SE15 — the same 1-based iteration this delegation's `onDelegationStart` saw. */
+    iteration: number;
 }
+/**
+ * The return of a delegation hook: a decision, a promise of one, or nothing —
+ * `void` lets a side-effect-only callback (`(ctx) => { log(ctx) }`) type-check,
+ * which is the common case (mirrors Mastra's `async ctx => { ... }` hooks).
+ */
+type DelegationHookResult<T> = T | void | Promise<T | void>;
 /** Decision returned from {@link SubAgentSpec.onDelegationComplete}. */
 export interface DelegationCompleteDecision {
     /** Appended to the child's result string. */
@@ -65,7 +81,7 @@ export interface SubAgentSpec {
      * result), or `{ modifiedInput }` to rewrite the delegated prompt. A throwing
      * hook surfaces (never silently swallowed).
      */
-    onDelegationStart?: (ctx: DelegationStartContext) => DelegationStartDecision | undefined | Promise<DelegationStartDecision | undefined>;
+    onDelegationStart?: (ctx: DelegationStartContext) => DelegationHookResult<DelegationStartDecision>;
     /**
      * SE11 — called after the delegation settles. On success `ctx.result` is set
      * and an optional `{ feedback }` is appended to it. On failure `ctx.error` is
@@ -73,7 +89,7 @@ export interface SubAgentSpec {
      * from this hook on the error path is suppressed so it cannot mask the
      * delegation's real failure (on the success path a throw does propagate).
      */
-    onDelegationComplete?: (ctx: DelegationCompleteContext) => DelegationCompleteDecision | undefined | Promise<DelegationCompleteDecision | undefined>;
+    onDelegationComplete?: (ctx: DelegationCompleteContext) => DelegationHookResult<DelegationCompleteDecision>;
     /**
      * SE12 — opt-in parent-context forwarding. When set, the supervisor transcript
      * (`ctx.messages`, a read-only text projection) is passed to this filter and the
@@ -84,6 +100,14 @@ export interface SubAgentSpec {
      * `onDelegationStart`); the delegation surfaces as a tool error.
      */
     messageFilter?: (args: MessageFilterArgs) => readonly ToolContextMessage[];
+    /**
+     * SE14 — opt-in subagent result-context control. When `true`, the child's
+     * completed tool-call results (name + result) are appended to the delegation
+     * payload returned to the supervisor, inside a `<subagent-tool-results>` block.
+     * When absent/`false` the delegation returns the child's final text only —
+     * text-only stays the default (Mastra's scoped posture). See ADR 0006.
+     */
+    includeToolResults?: boolean;
 }
 export declare class MaxDelegationDepthError extends Error {
     readonly currentDepth: number;
@@ -92,3 +116,4 @@ export declare class MaxDelegationDepthError extends Error {
     constructor(currentDepth: number, maxDepth: number);
 }
 export declare function defineSubAgent(spec: SubAgentSpec, _parentDepth?: number): CustomTool;
+export {};

package/dist/a2a/subagent.d.ts

@@ -25,6 +25,12 @@ export interface MessageFilterArgs {
 export interface DelegationStartContext {
     input: string;
     name: string;
+    /**
+     * SE15 — 1-based count of times THIS subagent tool has been invoked (a
+     * per-`defineSubAgent`-instance counter). Incremented before this hook runs;
+     * a rejected delegation still counts. Enables reject-after-N patterns.
+     */
+    iteration: number;
 }
 /**
  * Decision returned from {@link SubAgentSpec.onDelegationStart}. Discriminated on
@@ -37,6 +43,8 @@ export type DelegationStartDecision = {
 } | {
     proceed?: true;
     modifiedInput?: string;
+    /** SE13 — cap the child's iteration count (forwarded as `SendOptions.maxIterations`). */
+    modifiedMaxSteps?: number;
 };
 /** Context passed to {@link SubAgentSpec.onDelegationComplete} after the child settles. */
 export interface DelegationCompleteContext {
@@ -46,7 +54,15 @@ export interface DelegationCompleteContext {
     result?: string;
     /** The error the child threw (present on failure); the error is still re-thrown. */
     error?: unknown;
+    /** SE15 — the same 1-based iteration this delegation's `onDelegationStart` saw. */
+    iteration: number;
 }
+/**
+ * The return of a delegation hook: a decision, a promise of one, or nothing —
+ * `void` lets a side-effect-only callback (`(ctx) => { log(ctx) }`) type-check,
+ * which is the common case (mirrors Mastra's `async ctx => { ... }` hooks).
+ */
+type DelegationHookResult<T> = T | void | Promise<T | void>;
 /** Decision returned from {@link SubAgentSpec.onDelegationComplete}. */
 export interface DelegationCompleteDecision {
     /** Appended to the child's result string. */
@@ -65,7 +81,7 @@ export interface SubAgentSpec {
      * result), or `{ modifiedInput }` to rewrite the delegated prompt. A throwing
      * hook surfaces (never silently swallowed).
      */
-    onDelegationStart?: (ctx: DelegationStartContext) => DelegationStartDecision | undefined | Promise<DelegationStartDecision | undefined>;
+    onDelegationStart?: (ctx: DelegationStartContext) => DelegationHookResult<DelegationStartDecision>;
     /**
      * SE11 — called after the delegation settles. On success `ctx.result` is set
      * and an optional `{ feedback }` is appended to it. On failure `ctx.error` is
@@ -73,7 +89,7 @@ export interface SubAgentSpec {
      * from this hook on the error path is suppressed so it cannot mask the
      * delegation's real failure (on the success path a throw does propagate).
      */
-    onDelegationComplete?: (ctx: DelegationCompleteContext) => DelegationCompleteDecision | undefined | Promise<DelegationCompleteDecision | undefined>;
+    onDelegationComplete?: (ctx: DelegationCompleteContext) => DelegationHookResult<DelegationCompleteDecision>;
     /**
      * SE12 — opt-in parent-context forwarding. When set, the supervisor transcript
      * (`ctx.messages`, a read-only text projection) is passed to this filter and the
@@ -84,6 +100,14 @@ export interface SubAgentSpec {
      * `onDelegationStart`); the delegation surfaces as a tool error.
      */
     messageFilter?: (args: MessageFilterArgs) => readonly ToolContextMessage[];
+    /**
+     * SE14 — opt-in subagent result-context control. When `true`, the child's
+     * completed tool-call results (name + result) are appended to the delegation
+     * payload returned to the supervisor, inside a `<subagent-tool-results>` block.
+     * When absent/`false` the delegation returns the child's final text only —
+     * text-only stays the default (Mastra's scoped posture). See ADR 0006.
+     */
+    includeToolResults?: boolean;
 }
 export declare class MaxDelegationDepthError extends Error {
     readonly currentDepth: number;
@@ -92,3 +116,4 @@ export declare class MaxDelegationDepthError extends Error {
     constructor(currentDepth: number, maxDepth: number);
 }
 export declare function defineSubAgent(spec: SubAgentSpec, _parentDepth?: number): CustomTool;
+export {};

package/dist/{cron-Bpt_1khA.d.cts → cron-Bd2oRD7A.d.cts}

@@ -1,4 +1,4 @@
-import { C as CustomTool, M as ModelSelection, Z as SDKUserMessage, $ as SendOptions, b as Run, G as GenerateOptions, j as GenerateRunResult, F as RunToCompletionOptions, H as RunToCompletionResult, S as SDKMessage, a6 as StreamToCompletionResult, a as McpServerConfig } from './run-CrIulPF7.cjs';
+import { C as CustomTool, M as ModelSelection, Z as SDKUserMessage, $ as SendOptions, b as Run, G as GenerateOptions, j as GenerateRunResult, F as RunToCompletionOptions, H as RunToCompletionResult, S as SDKMessage, a6 as StreamToCompletionResult, a as McpServerConfig } from './run-Cr0C6cOM.cjs';
 import * as zod from 'zod';
 
 /**

package/dist/{cron-7CruUd_0.d.ts → cron-vjod0qVQ.d.ts}

@@ -1,4 +1,4 @@
-import { C as CustomTool, M as ModelSelection, Z as SDKUserMessage, $ as SendOptions, b as Run, G as GenerateOptions, j as GenerateRunResult, F as RunToCompletionOptions, H as RunToCompletionResult, S as SDKMessage, a6 as StreamToCompletionResult, a as McpServerConfig } from './run-CrIulPF7.js';
+import { C as CustomTool, M as ModelSelection, Z as SDKUserMessage, $ as SendOptions, b as Run, G as GenerateOptions, j as GenerateRunResult, F as RunToCompletionOptions, H as RunToCompletionResult, S as SDKMessage, a6 as StreamToCompletionResult, a as McpServerConfig } from './run-Cr0C6cOM.js';
 import * as zod from 'zod';
 
 /**

package/dist/cron.cjs

@@ -13307,6 +13307,7 @@ function levenshtein(a, b) {
 }
 
 // src/internal/runtime/local-agent/real-local-run.ts
+init_async_local_storage();
 function createRealLocalRun(options) {
   const { userText, id, startTime } = prepareRunContext(options.message);
   const supported = /* @__PURE__ */ new Set(["stream", "wait", "cancel", "conversation"]);
@@ -13418,6 +13419,7 @@ function buildLoopInputs(options, runId, userText) {
     ...options.onStep !== void 0 ? { onStep: options.onStep } : {},
     ...options.onDelta !== void 0 ? { onDelta: options.onDelta } : {},
     ...options.sendOptions.toolChoice !== void 0 ? { toolChoice: options.sendOptions.toolChoice } : {},
+    ...options.sendOptions.activeTools !== void 0 ? { activeTools: options.sendOptions.activeTools } : {},
     ...options.priorMessages !== void 0 ? { priorMessages: options.priorMessages } : {},
     ...options.memoryTools !== void 0 && options.memoryTools.length > 0 ? { memoryTools: options.memoryTools } : {},
     ...buildCustomToolsInput(
@@ -13548,7 +13550,7 @@ var RealLocalRun = class extends FixtureRunBase {
   }
   async executeAgentLoop(inputs) {
     try {
-      const output = await runAgentLoop(inputs);
+      const output = inputs.activeTools !== void 0 ? await withToolWhitelist(new Set(inputs.activeTools), () => runAgentLoop(inputs)) : await runAgentLoop(inputs);
       this.applyAgentLoopOutput(output);
       this.transitionTo(output.finalStatus);
     } catch (cause) {