@theokit/agents 0.28.0 → 0.30.0

package/dist/{bridge-entry-CsUiO2YU.d.ts → bridge-entry-DVhOQc_f.d.ts}

@@ -1,9 +1,9 @@
 import { ExecutionContext } from '@theokit/http';
-import { A as AgentOptions, T as ToolOptions, g as MainLoopMeta, a as ApprovalOptions, B as BudgetOptions, f as GatewayOptions, l as MemoryOptions, t as SkillsOptions, e as ContextWindowOptions, q as ProjectContextOptions, j as McpServersMap, b as CompactionDecoratorConfig, R as ReasoningEffort } from './skills-FcUdNDbn.js';
-import { SystemPromptResolver, SkillsSettings, ContextSettings, PluginsSettings, Plugin, ProviderRoutingSettings, AgentDefinition, BudgetTracker, ConversationStorageAdapter, CustomTool, ModelSelection } from '@theokit/sdk';
+import { A as AgentOptions, D as ToolOptions, m as MainLoopMeta, a as ApprovalOptions, B as BudgetOptions, l as HumanInTheLoopOptions, k as GatewayOptions, r as MemoryOptions, z as SkillsOptions, j as ContextWindowOptions, w as ProjectContextOptions, p as McpServersMap, g as CompactionDecoratorConfig, b as CheckpointOptions, R as ReasoningEffort } from './skills-Dx_KJ6Eg.js';
+import { SystemPromptResolver, SkillsSettings, ContextSettings, PluginsSettings, Plugin, ProviderRoutingSettings, AgentDefinition as AgentDefinition$1, BudgetTracker, ConversationStorageAdapter, CustomTool, ModelSelection } from '@theokit/sdk';
 import { UIMessageChunk } from 'ai';
-import { RetryOptions } from '@theokit/sdk/retry';
 import { z } from 'zod';
+import { RetryOptions } from '@theokit/sdk/retry';
 
 /**
  * AgentExecutionContext — extends http-decorators' ExecutionContext with agent-specific methods.
@@ -49,6 +49,7 @@ declare const AgentWarningCode: {
     readonly BUDGET_TOP_LEVEL_METADATA_ONLY: "THEO_AGENT_BUDGET_TOP_LEVEL_METADATA_ONLY";
     readonly CONTEXT_STRATEGY_METADATA_ONLY: "THEO_AGENT_CONTEXT_STRATEGY_METADATA_ONLY";
     readonly PROJECT_CONTEXT_KNOB_METADATA_ONLY: "THEO_AGENT_PROJECT_CONTEXT_KNOB_METADATA_ONLY";
+    readonly CHECKPOINT_STORAGE_METADATA_ONLY: "THEO_AGENT_CHECKPOINT_STORAGE_METADATA_ONLY";
 };
 interface AgentWalkResult {
     agentConfig: AgentOptions;
@@ -66,6 +67,8 @@ interface AgentWalkResult {
     projectContext?: ProjectContextOptions;
     mcpServers?: McpServersMap;
     compaction?: CompactionDecoratorConfig;
+    /** `@Checkpoint` config (M4) when the agent declares resumable execution; absent ⇒ no checkpoint. */
+    checkpoint?: CheckpointOptions;
 }
 interface ToolboxWalkResult {
     class: Function;
@@ -82,6 +85,8 @@ interface ToolWalkResult {
     budget?: BudgetOptions;
     trace: boolean;
     audit: boolean;
+    /** `@HumanInTheLoop` config when the tool method is gated (M4); absent ⇒ not gated. */
+    hitl?: HumanInTheLoopOptions;
 }
 /**
  * Walk all metadata on an agent class and its toolboxes.
@@ -156,6 +161,16 @@ interface CompiledAgentOptions {
     maxIterations?: number;
     timeoutMs?: number;
     stream: boolean;
+    /**
+     * HITL gate map (M4): runtime tool name → `@HumanInTheLoop` config. Absent/empty ⇒ no gated
+     * tools. The harness (`mountAgent`) turns this into the `pre_tool_call` pause wiring.
+     */
+    hitl?: Map<string, HumanInTheLoopOptions>;
+    /**
+     * `@Checkpoint` config (M4): when present the harness emits `checkpoint_saved` and selects the
+     * durable SDK conversation storage (`storage: 'filesystem'`) so a same-`sessionId` request resumes.
+     */
+    checkpoint?: CheckpointOptions;
 }
 /**
  * Compile @Agent metadata into SDK-compatible options.
@@ -469,7 +484,7 @@ interface RuntimeOverrides {
     /** Per-run provider routing. */
     providers?: ProviderRoutingSettings;
     /** Per-run sub-agent definitions (opts-only; `compiled.agents` stays deferred — ADR D3). */
-    agents?: Record<string, AgentDefinition>;
+    agents?: Record<string, AgentDefinition$1>;
     /** Per-run SDK budget tracker (inner tool-loop cap; distinct from the loop's USD `budget`). */
     budgetTracker?: BudgetTracker;
     /**
@@ -582,6 +597,140 @@ declare function translateToUIMessageStream(events: AsyncIterable<AgentStreamEve
     textId: string;
 }): AsyncGenerator<UIMessageChunk, void, unknown>;
 
+/**
+ * M2 (theokit-ai-first) — `defineAgent`, the zero-config imperative agent surface.
+ *
+ * ADR-B1: `defineAgent({...})` (default-exported from a top-level `agents/<name>.ts`) is
+ * the canonical zero-config surface; the `@Agent` class decorator stays the advanced/DI
+ * surface. Both compile to {@link CompiledAgentOptions} and run through the same SDK
+ * runtime (`createSdkAgentStream`) — one runtime, two syntaxes.
+ *
+ * This module is PURE metadata (sdk-runtime.md / G2): `defineAgent` describes an agent, it
+ * NEVER calls an LLM. It imports only `zod` (types) + the compiler shape — no `theokit`
+ * core, preserving the agents → (nothing) dependency direction (G1).
+ */
+
+/**
+ * Brand tag for a `defineAgent` value. `Symbol.for` (global registry, not `Symbol()`) so
+ * the brand survives duplicate module instances (dual-package / bundling) — the scanner's
+ * brand-check then works regardless of which copy created the definition.
+ */
+declare const AGENT_BRAND: unique symbol;
+/** Config accepted by {@link defineAgent}. */
+interface DefineAgentConfig<TInput extends z.ZodType = z.ZodType> {
+    /** Zod schema for the request body — lifted into the typed client (M2, {@link InferAgentInput}). */
+    input?: TInput;
+    /** Model id (e.g. `claude-sonnet-4-6`). Falls back to the SDK default when omitted. */
+    model?: string;
+    /** Static system prompt. */
+    system?: string;
+    /** Extended-thinking effort (mirrors `@Agent({ reasoningEffort })`). */
+    reasoningEffort?: ReasoningEffort;
+    /** Pre-built tools (SDK-ready `CompiledTool` shape). */
+    tools?: CompiledTool[];
+}
+/** A branded agent definition — the value {@link defineAgent} returns. */
+type AgentDefinition<TInput extends z.ZodType = z.ZodType> = DefineAgentConfig<TInput> & {
+    readonly [AGENT_BRAND]: true;
+};
+/** Infer the request type of an agent definition from its `input` Zod schema. */
+type InferAgentInput<T> = T extends AgentDefinition<infer S> ? (S extends z.ZodType ? z.infer<S> : never) : never;
+/**
+ * Define a zero-config agent. Identity/normalizer (like `defineRoute`) — returns the config
+ * branded so the scanner recognizes it. Compilation is deferred to {@link compileAgentDefinition}.
+ */
+declare function defineAgent<TInput extends z.ZodType = z.ZodType>(config: DefineAgentConfig<TInput>): AgentDefinition<TInput>;
+/** Brand-check: is `value` a {@link defineAgent} result? */
+declare function isAgentDefinition(value: unknown): value is AgentDefinition;
+/**
+ * Lower a definition to the SDK-ready {@link CompiledAgentOptions} — the same shape
+ * `compileAgent` (decorator path) produces, so both surfaces converge on one runtime.
+ */
+declare function compileAgentDefinition(def: AgentDefinition): CompiledAgentOptions;
+
+/**
+ * M4 (theokit-ai-first) — the HITL producer: a `pre_tool_call` plugin that pauses the SDK run
+ * for a human-in-the-loop tool approval.
+ *
+ * ADR 0038: this is the ADAPTER seam — it makes the shipped-but-dead `@HumanInTheLoop` decorator
+ * FUNCTIONAL by wiring the SDK's OWN async `pre_tool_call` veto hook (which the SDK loop `await`s,
+ * so returning a pending Promise genuinely PAUSES the run) to a human approval. It calls no LLM,
+ * dispatches no tool, and runs no second loop — the SDK owns all of that.
+ *
+ * Flow: gated tool about to run → emit an `ApprovalRequiredEvent` (which the translator maps to
+ * the ai-sdk `tool-approval-request` chunk) → `await awaitApproval(approvalId)` (resolved by the
+ * out-of-band approve route) → allow (`undefined`) or veto (`{ block, message }`, which the SDK
+ * surfaces as a tool result the model self-corrects on).
+ */
+
+/** Injected wiring — the harness (mount-agent) supplies these; the plugin stays pure. */
+interface HitlWiring {
+    /** Tool name → its `@HumanInTheLoop` config. A tool absent here is NOT gated. */
+    gated: Map<string, HumanInTheLoopOptions>;
+    /** Push the approval-required event into the agent stream (the translator emits the chunk). */
+    emit: (event: ApprovalRequiredEvent) => void;
+    /** Await the human decision for `approvalId`; resolves true=approve, false=deny (or timeout). */
+    awaitApproval: (approvalId: string, opts: HumanInTheLoopOptions) => Promise<boolean>;
+}
+
+/**
+ * M2 (theokit-ai-first) — the file-convention runtime bridge.
+ *
+ * Turns a loaded `agents/<name>.ts` module into the M0/M1 canonical `UIMessageStream`:
+ *
+ *   module (defineAgent value | @Agent class) ──compileAgentModule──▶ CompiledAgentOptions
+ *   CompiledAgentOptions ──createSdkAgentStream──▶ AgentStreamEvent* ──translate──▶ UIMessageChunk*
+ *
+ * Both agent surfaces converge here (ADR-B1): a `defineAgent` value lowers via
+ * `compileAgentDefinition`, an `@Agent`-decorated class lowers via `compileAgent`
+ * (which requires the full `@MainLoop` decoration — its existing errors surface for
+ * DI-heavy classes). Neither runs an LLM directly — `@theokit/sdk` stays the sole
+ * runtime (G2 / sdk-runtime.md); this module only wires its output onto the wire.
+ */
+
+/** Thrown when an `agents/` file default-exports neither a `defineAgent` value nor an `@Agent` class. */
+declare class AgentDefinitionError extends Error {
+    constructor(source: string);
+}
+/**
+ * Compile a loaded `agents/` module to SDK-ready options. Accepts a `defineAgent` value
+ * (zero-config surface) OR an `@Agent`-decorated class (advanced surface). `source` labels
+ * the fail-fast error (typically the file path).
+ *
+ * For a class agent, its `@Mixin(...)` toolboxes are gathered (the declared tool-association
+ * mechanism, same as `app.ts`/`theokit-plugin.ts`) and instantiated with a no-arg `new` — the
+ * zero-config file convention has no DI container. This is what makes a `@HumanInTheLoop`-gated
+ * tool on a mixin actually gate through the M2 endpoint (M4): its config reaches `compiled.hitl`.
+ */
+declare function compileAgentModule(mod: unknown, source?: string): CompiledAgentOptions;
+/** HITL wiring supplied by the harness (mount-agent): the gated-tool map + the approval resolver. */
+interface StreamHitlOptions {
+    gated: HitlWiring['gated'];
+    awaitApproval: HitlWiring['awaitApproval'];
+}
+interface StreamAgentOptions {
+    message: string;
+    sessionId: string;
+    /** Enable human-in-the-loop tool approval (M4). Absent ⇒ the M2 non-HITL path, byte-unchanged. */
+    hitl?: StreamHitlOptions;
+    /** Durable conversation storage for resume (M4); defaults to the SDK per-run in-memory store. */
+    conversationStorage?: RuntimeOverrides['conversationStorage'];
+    /**
+     * The request's abort signal (M4). On client disconnect, the HITL merge queue is closed so the
+     * detached SDK pump stops buffering (bounded memory) and the client stream terminates at once.
+     * The paused SDK run itself is released by the approval timeout, not instantly — a durable-store
+     * follow-up. The non-HITL path is pull-based and already tears down on the consumer's return.
+     */
+    signal?: AbortSignal;
+}
+/**
+ * Run a compiled agent and yield the M0/M1 `UIMessageStream` chunks. `apiKey` is resolved by the
+ * caller (`resolveProvider`). One `textId` per run (G8: `crypto.randomUUID`). When `hitl` is
+ * supplied (M4), a HITL `pre_tool_call` plugin pauses the run for gated tools — the pause is the
+ * SDK's own awaited hook, never a second loop (ADR 0038).
+ */
+declare function streamAgentUIMessages(compiled: CompiledAgentOptions, apiKey: string, input: StreamAgentOptions): AsyncGenerator<UIMessageChunk>;
+
 /**
  * LoopStrategy — the per-round terminal-decision contract that gives runtime to
  * `@MainLoop({ strategy })`.
@@ -801,7 +950,7 @@ interface DelegateOptions {
     /** Per-run provider routing (e.g. OpenRouter). */
     providers?: ProviderRoutingSettings;
     /** Per-run sub-agent definitions. */
-    agents?: Record<string, AgentDefinition>;
+    agents?: Record<string, AgentDefinition$1>;
     /** Per-run SDK budget tracker (inner tool-loop cap). */
     budgetTracker?: BudgetTracker;
     /** Per-run conversation store (cross-round history). */
@@ -912,4 +1061,4 @@ declare function agentsPlugin(opts: AgentsPluginOptions): {
     register(app: PluginApp): void;
 };
 
-export { compileTools as $, type AgentExecutionContext as A, BudgetExceededError as B, type CompiledAgentOptions as C, type DelegationResult as D, type ErrorEvent as E, type FileEditEvent as F, type RunStartedEvent as G, type SdkMessage as H, type IterationEvent as I, type Segment as J, type StateUpdateEvent as K, type LoopStrategy as L, type ThinkingEvent as M, type ToolCallEvent as N, type ToolResultEvent as O, type PartialToolCallEvent as P, type ToolWalkResult as Q, type ReflectionStrategy as R, type StreamEvent as S, type TextDeltaEvent as T, type ToolboxWalkResult as U, agentsPlugin as V, buildModelSelection as W, compileAgent as X, compileContextWindow as Y, compileProjectContext as Z, compileSkills as _, type CompiledTool as a, createAgentExecutionContext as a0, createSdkAgentStream as a1, createThinkTagExtractor as a2, delegate as a3, extractThinkTagStream as a4, generateAgentManifest as a5, generateAgentRoutes as a6, isAgentContext as a7, isApprovalRequired as a8, isDone as a9, isError as aa, isPartialToolCall as ab, isTextDelta as ac, isToolCall as ad, isToolResult as ae, ladderReflectionStrategy as af, loopStrategyConfigSchema as ag, noopReflectionStrategy as ah, projectContextMetadataOnlyKnobs as ai, reflectionStrategyConfigSchema as aj, resolveLoopStrategy as ak, streamAgentResponse as al, translateSdkEvent as am, translateToUIMessageStream as an, validateUniqueRoutes as ao, walkAgentMetadata as ap, type AgentManifest as b, type AgentManifestEntry as c, type AgentManifestTool as d, type AgentRoute as e, type AgentRouteContext as f, type AgentRunInfo as g, type AgentStreamEvent as h, type AgentWalkResult as i, AgentWarningCode as j, type AgentsPluginOptions as k, type ApprovalRequiredEvent as l, type ArtifactChunkEvent as m, type ArtifactStartEvent as n, type CheckpointSavedEvent as o, type CompiledContextWindow as p, DEFAULT_MAX_ITERATIONS as q, type DelegateOptions as r, DelegationError as s, type DoneEvent as t, type LoopFinishReason as u, type LoopOutcome as v, type LoopStrategyConfig as w, type ReflectionContext as x, type ReflectionResult as y, type ReflectionStrategyConfig as z };
+export { buildModelSelection as $, AGENT_BRAND as A, BudgetExceededError as B, type CompiledAgentOptions as C, type DelegationResult as D, type ErrorEvent as E, type FileEditEvent as F, type LoopOutcome as G, type LoopStrategyConfig as H, type InferAgentInput as I, type ReflectionContext as J, type ReflectionResult as K, type LoopStrategy as L, type ReflectionStrategyConfig as M, type RunStartedEvent as N, type SdkMessage as O, type PartialToolCallEvent as P, type Segment as Q, type ReflectionStrategy as R, type StreamEvent as S, type StateUpdateEvent as T, type TextDeltaEvent as U, type ThinkingEvent as V, type ToolCallEvent as W, type ToolResultEvent as X, type ToolWalkResult as Y, type ToolboxWalkResult as Z, agentsPlugin as _, type CompiledTool as a, compileAgent as a0, compileAgentDefinition as a1, compileAgentModule as a2, compileContextWindow as a3, compileProjectContext as a4, compileSkills as a5, compileTools as a6, createAgentExecutionContext as a7, createSdkAgentStream as a8, createThinkTagExtractor as a9, defineAgent as aa, delegate as ab, extractThinkTagStream as ac, generateAgentManifest as ad, generateAgentRoutes as ae, isAgentContext as af, isAgentDefinition as ag, isApprovalRequired as ah, isDone as ai, isError as aj, isPartialToolCall as ak, isTextDelta as al, isToolCall as am, isToolResult as an, ladderReflectionStrategy as ao, loopStrategyConfigSchema as ap, noopReflectionStrategy as aq, projectContextMetadataOnlyKnobs as ar, reflectionStrategyConfigSchema as as, resolveLoopStrategy as at, streamAgentResponse as au, streamAgentUIMessages as av, translateSdkEvent as aw, translateToUIMessageStream as ax, validateUniqueRoutes as ay, walkAgentMetadata as az, type AgentDefinition as b, AgentDefinitionError as c, type AgentExecutionContext as d, type AgentManifest as e, type AgentManifestEntry as f, type AgentManifestTool as g, type AgentRoute as h, type AgentRouteContext as i, type AgentRunInfo as j, type AgentStreamEvent as k, type AgentWalkResult as l, AgentWarningCode as m, type AgentsPluginOptions as n, type ApprovalRequiredEvent as o, type ArtifactChunkEvent as p, type ArtifactStartEvent as q, type CheckpointSavedEvent as r, type CompiledContextWindow as s, DEFAULT_MAX_ITERATIONS as t, type DefineAgentConfig as u, type DelegateOptions as v, DelegationError as w, type DoneEvent as x, type IterationEvent as y, type LoopFinishReason as z };

package/dist/bridge.d.ts

@@ -1,6 +1,6 @@
-export { A as AgentExecutionContext, b as AgentManifest, c as AgentManifestEntry, d as AgentManifestTool, e as AgentRoute, f as AgentRouteContext, g as AgentRunInfo, h as AgentStreamEvent, i as AgentWalkResult, j as AgentWarningCode, k as AgentsPluginOptions, l as ApprovalRequiredEvent, m as ArtifactChunkEvent, n as ArtifactStartEvent, B as BudgetExceededError, o as CheckpointSavedEvent, C as CompiledAgentOptions, p as CompiledContextWindow, a as CompiledTool, r as DelegateOptions, s as DelegationError, D as DelegationResult, t as DoneEvent, E as ErrorEvent, F as FileEditEvent, I as IterationEvent, P as PartialToolCallEvent, G as RunStartedEvent, H as SdkMessage, J as Segment, K as StateUpdateEvent, S as StreamEvent, T as TextDeltaEvent, M as ThinkingEvent, N as ToolCallEvent, O as ToolResultEvent, Q as ToolWalkResult, U as ToolboxWalkResult, V as agentsPlugin, W as buildModelSelection, X as compileAgent, Y as compileContextWindow, Z as compileProjectContext, _ as compileSkills, $ as compileTools, a0 as createAgentExecutionContext, a1 as createSdkAgentStream, a2 as createThinkTagExtractor, a3 as delegate, a4 as extractThinkTagStream, a5 as generateAgentManifest, a6 as generateAgentRoutes, a7 as isAgentContext, a8 as isApprovalRequired, a9 as isDone, aa as isError, ab as isPartialToolCall, ac as isTextDelta, ad as isToolCall, ae as isToolResult, ai as projectContextMetadataOnlyKnobs, al as streamAgentResponse, am as translateSdkEvent, an as translateToUIMessageStream, ao as validateUniqueRoutes, ap as walkAgentMetadata } from './bridge-entry-CsUiO2YU.js';
+export { A as AGENT_BRAND, b as AgentDefinition, c as AgentDefinitionError, d as AgentExecutionContext, e as AgentManifest, f as AgentManifestEntry, g as AgentManifestTool, h as AgentRoute, i as AgentRouteContext, j as AgentRunInfo, k as AgentStreamEvent, l as AgentWalkResult, m as AgentWarningCode, n as AgentsPluginOptions, o as ApprovalRequiredEvent, p as ArtifactChunkEvent, q as ArtifactStartEvent, B as BudgetExceededError, r as CheckpointSavedEvent, C as CompiledAgentOptions, s as CompiledContextWindow, a as CompiledTool, u as DefineAgentConfig, v as DelegateOptions, w as DelegationError, D as DelegationResult, x as DoneEvent, E as ErrorEvent, F as FileEditEvent, I as InferAgentInput, y as IterationEvent, P as PartialToolCallEvent, N as RunStartedEvent, O as SdkMessage, Q as Segment, T as StateUpdateEvent, S as StreamEvent, U as TextDeltaEvent, V as ThinkingEvent, W as ToolCallEvent, X as ToolResultEvent, Y as ToolWalkResult, Z as ToolboxWalkResult, _ as agentsPlugin, $ as buildModelSelection, a0 as compileAgent, a1 as compileAgentDefinition, a2 as compileAgentModule, a3 as compileContextWindow, a4 as compileProjectContext, a5 as compileSkills, a6 as compileTools, a7 as createAgentExecutionContext, a8 as createSdkAgentStream, a9 as createThinkTagExtractor, aa as defineAgent, ab as delegate, ac as extractThinkTagStream, ad as generateAgentManifest, ae as generateAgentRoutes, af as isAgentContext, ag as isAgentDefinition, ah as isApprovalRequired, ai as isDone, aj as isError, ak as isPartialToolCall, al as isTextDelta, am as isToolCall, an as isToolResult, ar as projectContextMetadataOnlyKnobs, au as streamAgentResponse, av as streamAgentUIMessages, aw as translateSdkEvent, ax as translateToUIMessageStream, ay as validateUniqueRoutes, az as walkAgentMetadata } from './bridge-entry-DVhOQc_f.js';
 import '@theokit/http';
-import './skills-FcUdNDbn.js';
+import './skills-Dx_KJ6Eg.js';
 import '@theokit/sdk';
 import 'zod';
 import 'ai';

package/dist/bridge.js

@@ -1,10 +1,14 @@
 import {
+  AGENT_BRAND,
+  AgentDefinitionError,
   AgentWarningCode,
   BudgetExceededError,
   DelegationError,
   agentsPlugin,
   buildModelSelection,
   compileAgent,
+  compileAgentDefinition,
+  compileAgentModule,
   compileContextWindow,
   compileProjectContext,
   compileSkills,
@@ -12,11 +16,13 @@ import {
   createAgentExecutionContext,
   createSdkAgentStream,
   createThinkTagExtractor,
+  defineAgent,
   delegate,
   extractThinkTagStream,
   generateAgentManifest,
   generateAgentRoutes,
   isAgentContext,
+  isAgentDefinition,
   isApprovalRequired,
   isDone,
   isError,
@@ -26,20 +32,25 @@ import {
   isToolResult,
   projectContextMetadataOnlyKnobs,
   streamAgentResponse,
+  streamAgentUIMessages,
   translateSdkEvent,
   translateToUIMessageStream,
   validateUniqueRoutes,
   walkAgentMetadata
-} from "./chunk-DA7HHFL5.js";
-import "./chunk-GVPUUKKE.js";
+} from "./chunk-AUOCWGPS.js";
+import "./chunk-FI6ZG2YP.js";
 import "./chunk-7QVYU63E.js";
 export {
+  AGENT_BRAND,
+  AgentDefinitionError,
   AgentWarningCode,
   BudgetExceededError,
   DelegationError,
   agentsPlugin,
   buildModelSelection,
   compileAgent,
+  compileAgentDefinition,
+  compileAgentModule,
   compileContextWindow,
   compileProjectContext,
   compileSkills,
@@ -47,11 +58,13 @@ export {
   createAgentExecutionContext,
   createSdkAgentStream,
   createThinkTagExtractor,
+  defineAgent,
   delegate,
   extractThinkTagStream,
   generateAgentManifest,
   generateAgentRoutes,
   isAgentContext,
+  isAgentDefinition,
   isApprovalRequired,
   isDone,
   isError,
@@ -61,6 +74,7 @@ export {
   isToolResult,
   projectContextMetadataOnlyKnobs,
   streamAgentResponse,
+  streamAgentUIMessages,
   translateSdkEvent,
   translateToUIMessageStream,
   validateUniqueRoutes,