@langchain/langgraph-sdk 1.6.4 → 1.7.0

package/dist/ui/manager.d.cts

@@ -0,0 +1,224 @@
+import { Message } from "../types.messages.cjs";
+import { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, ErrorStreamEvent, EventsStreamEvent, FeedbackStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, TasksStreamEvent, ToolsStreamEvent, UpdatesStreamEvent, ValuesStreamEvent } from "../types.stream.cjs";
+import { BagTemplate } from "../types.template.cjs";
+import { SubagentStreamInterface } from "./types.cjs";
+import { MessageTupleManager } from "./messages.cjs";
+import { BaseMessage } from "@langchain/core/messages";
+
+//#region src/ui/manager.d.ts
+type GetUpdateType<Bag extends BagTemplate, StateType extends Record<string, unknown>> = Bag extends {
+  UpdateType: unknown;
+} ? Bag["UpdateType"] : Partial<StateType>;
+type GetCustomEventType<Bag extends BagTemplate> = Bag extends {
+  CustomEventType: unknown;
+} ? Bag["CustomEventType"] : unknown;
+type EventStreamMap<StateType, UpdateType, CustomType> = {
+  values: ValuesStreamEvent<StateType>;
+  updates: UpdatesStreamEvent<UpdateType>;
+  custom: CustomStreamEvent<CustomType>;
+  debug: DebugStreamEvent;
+  messages: MessagesTupleStreamEvent;
+  events: EventsStreamEvent;
+  metadata: MetadataStreamEvent;
+  checkpoints: CheckpointsStreamEvent<StateType>;
+  tasks: TasksStreamEvent<StateType, UpdateType>;
+  error: ErrorStreamEvent;
+  feedback: FeedbackStreamEvent;
+  tools: ToolsStreamEvent;
+};
+type EventStreamEvent<StateType, UpdateType, CustomType> = EventStreamMap<StateType, UpdateType, CustomType>[keyof EventStreamMap<StateType, UpdateType, CustomType>];
+interface StreamManagerEventCallbacks<StateType extends Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
+  onUpdateEvent?: (data: UpdatesStreamEvent<GetUpdateType<Bag, StateType>>["data"], options: {
+    namespace: string[] | undefined;
+    mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+  }) => void;
+  onCustomEvent?: (data: GetCustomEventType<Bag>, options: {
+    namespace: string[] | undefined;
+    mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+  }) => void;
+  onMetadataEvent?: (data: MetadataStreamEvent["data"]) => void;
+  onLangChainEvent?: (data: EventsStreamEvent["data"]) => void;
+  onDebugEvent?: (data: DebugStreamEvent["data"], options: {
+    namespace: string[] | undefined;
+  }) => void;
+  onCheckpointEvent?: (data: CheckpointsStreamEvent<StateType>["data"], options: {
+    namespace: string[] | undefined;
+  }) => void;
+  onTaskEvent?: (data: TasksStreamEvent<StateType, GetUpdateType<Bag, StateType>>["data"], options: {
+    namespace: string[] | undefined;
+  }) => void;
+  onToolEvent?: (data: ToolsStreamEvent["data"], options: {
+    namespace: string[] | undefined;
+    mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+  }) => void;
+}
+/**
+ * Options for StreamManager constructor.
+ */
+interface StreamManagerOptions {
+  /**
+   * Throttle the stream updates.
+   * If a number is provided, updates are throttled to the given milliseconds.
+   * If `true`, updates are batched in a single macrotask.
+   * If `false`, updates are not throttled.
+   */
+  throttle: number | boolean;
+  /**
+   * Tool names that indicate subagent invocation.
+   *
+   * When an AI message contains tool calls with these names, they are
+   * automatically tracked as subagent executions. This enables the
+   * `subagents`, `activeSubagents`, `getSubagent()`, and `getSubagentsByType()`
+   * properties on the stream.
+   *
+   * @default ["task"]
+   *
+   * @example
+   * ```typescript
+   * // Track both "task" and "delegate" as subagent tools
+   * subagentToolNames: ["task", "delegate", "spawn_agent"]
+   * ```
+   */
+  subagentToolNames?: string[];
+  /**
+   * Filter out messages from subagent streams in the main messages array.
+   *
+   * When enabled, messages from subagraph executions (those with a `tools:` namespace)
+   * are excluded from `stream.messages`. Instead, these messages are tracked
+   * per-subagent and accessible via `stream.subagents.get(id).messages`.
+   *
+   * This is useful for deep agent architectures where you want to display
+   * the main conversation separately from subagent activity.
+   *
+   * @default false
+   *
+   * @example
+   * ```typescript
+   * const stream = useStream({
+   *   assistantId: "my-agent",
+   *   filterSubagentMessages: true,
+   * });
+   *
+   * // Main thread messages only (no subagent messages)
+   * stream.messages
+   *
+   * // Access subagent messages individually
+   * stream.subagents.get("call_xyz").messages
+   * ```
+   */
+  filterSubagentMessages?: boolean;
+  /**
+   * Converts a @langchain/core BaseMessage to the desired output format.
+   *
+   * Defaults to `toMessageDict` which produces plain Message objects.
+   * Framework SDKs pass `toMessageClass` (identity) to keep class instances.
+   */
+  toMessage?: (chunk: BaseMessage) => Message | BaseMessage;
+}
+declare class StreamManager<StateType extends Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
+  private abortRef;
+  private messages;
+  private subagentManager;
+  private listeners;
+  private throttle;
+  private filterSubagentMessages;
+  private toMessage;
+  private queue;
+  private queueSize;
+  private state;
+  constructor(messages: MessageTupleManager, options: StreamManagerOptions);
+  /**
+   * Increment version counter to trigger React re-renders.
+   * Called when subagent state changes.
+   */
+  private bumpVersion;
+  /**
+   * Get all subagents as a Map.
+   */
+  getSubagents(): Map<string, SubagentStreamInterface>;
+  /**
+   * Get all currently running subagents.
+   */
+  getActiveSubagents(): SubagentStreamInterface[];
+  /**
+   * Get a specific subagent by tool call ID.
+   */
+  getSubagent(toolCallId: string): SubagentStreamInterface | undefined;
+  /**
+   * Get all subagents of a specific type.
+   */
+  getSubagentsByType(type: string): SubagentStreamInterface[];
+  /**
+   * Get all subagents triggered by a specific AI message.
+   */
+  getSubagentsByMessage(messageId: string): SubagentStreamInterface[];
+  /**
+   * Reconstruct subagent state from historical messages.
+   *
+   * This method should be called when loading thread history to restore
+   * subagent visualization after:
+   * - Page refresh (when stream has already completed)
+   * - Loading thread history
+   * - Navigating between threads
+   *
+   * @param messages - Array of messages from thread history
+   * @param options - Optional configuration
+   * @param options.skipIfPopulated - If true, skip reconstruction if subagents already exist
+   */
+  reconstructSubagents(messages: Message[], options?: {
+    skipIfPopulated?: boolean;
+  }): void;
+  /**
+   * Check if any subagents are currently tracked.
+   */
+  hasSubagents(): boolean;
+  private setState;
+  private notifyListeners;
+  subscribe: (listener: () => void) => () => void;
+  getSnapshot: () => {
+    isLoading: boolean;
+    values: [values: StateType, kind: "stop" | "stream"] | null;
+    error: unknown; /** Version counter to force React re-renders on subagent changes */
+    version: number;
+  };
+  get isLoading(): boolean;
+  get values(): StateType | null;
+  get error(): unknown;
+  setStreamValues: (values: StateType | ((prev: StateType | null, kind: "stop" | "stream") => StateType | null) | null, kind?: "stop" | "stream") => void;
+  private getMutateFn;
+  private matchEventType;
+  protected enqueue: (action: (signal: AbortSignal) => Promise<AsyncGenerator<EventStreamEvent<StateType, GetUpdateType<Bag, StateType>, GetCustomEventType<Bag>>, any, any>>, options: {
+    getMessages: (values: StateType) => Message[];
+    setMessages: (current: StateType, messages: Message[]) => StateType;
+    initialValues: StateType;
+    callbacks: StreamManagerEventCallbacks<StateType, Bag>;
+    onSuccess: () => void | StateType | Promise<void | StateType | null | undefined> | null | undefined;
+    onError: (error: unknown) => void | Promise<void>;
+    onFinish?: (() => void) | undefined;
+  }) => Promise<void>;
+  start: (action: (signal: AbortSignal) => Promise<AsyncGenerator<EventStreamEvent<StateType, GetUpdateType<Bag, StateType>, GetCustomEventType<Bag>>, any, any>>, options: {
+    getMessages: (values: StateType) => Message[];
+    setMessages: (current: StateType, messages: Message[]) => StateType;
+    initialValues: StateType;
+    callbacks: StreamManagerEventCallbacks<StateType, Bag>;
+    onSuccess: () => void | StateType | Promise<void | StateType | null | undefined> | null | undefined;
+    onError: (error: unknown) => void | Promise<void>;
+    onFinish?: (() => void) | undefined;
+  }, startOptions?: {
+    /**
+     * If true, abort any currently running stream before starting this one.
+     * Used for multitask_strategy: "interrupt" and "rollback" to unblock
+     * the queue so the new run request can proceed immediately.
+     */
+    abortPrevious?: boolean | undefined;
+  } | undefined) => Promise<void>;
+  stop: (historyValues: StateType, options: {
+    onStop?: ((options: {
+      mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+    }) => void) | undefined;
+  }) => Promise<void>;
+  clear: () => void;
+}
+//#endregion
+export { EventStreamEvent, StreamManager };
+//# sourceMappingURL=manager.d.cts.map

package/dist/ui/manager.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.d.cts","names":[],"sources":["../../src/ui/manager.ts"],"mappings":";;;;;;;;KAiCK,aAAA,aACS,WAAA,oBACM,MAAA,qBAChB,GAAA;EAAc,UAAA;AAAA,IACd,GAAA,iBACA,OAAA,CAAQ,SAAA;AAAA,KAEP,kBAAA,aAA+B,WAAA,IAAe,GAAA;EACjD,eAAA;AAAA,IAEE,GAAA;AAAA,KAGC,cAAA;EACH,MAAA,EAAQ,iBAAA,CAAkB,SAAA;EAC1B,OAAA,EAAS,kBAAA,CAAmB,UAAA;EAC5B,MAAA,EAAQ,iBAAA,CAAkB,UAAA;EAC1B,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,wBAAA;EACV,MAAA,EAAQ,iBAAA;EACR,QAAA,EAAU,mBAAA;EACV,WAAA,EAAa,sBAAA,CAAuB,SAAA;EACpC,KAAA,EAAO,gBAAA,CAAiB,SAAA,EAAW,UAAA;EACnC,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,mBAAA;EACV,KAAA,EAAO,gBAAA;AAAA;AAAA,KAGG,gBAAA,sCACV,cAAA,CAAe,SAAA,EAAW,UAAA,EAAY,UAAA,QAAkB,cAAA,CACtD,SAAA,EACA,UAAA,EACA,UAAA;AAAA,UAGM,2BAAA,mBACU,MAAA,+BACN,WAAA,GAAc,WAAA;EAE1B,aAAA,IACE,IAAA,EAAM,kBAAA,CAAmB,aAAA,CAAc,GAAA,EAAK,SAAA,YAC5C,OAAA;IACE,SAAA;IACA,MAAA,GACE,MAAA,EAAQ,OAAA,CAAQ,SAAA,MAAe,IAAA,EAAM,SAAA,KAAc,OAAA,CAAQ,SAAA;EAAA;EAIjE,aAAA,IACE,IAAA,EAAM,kBAAA,CAAmB,GAAA,GACzB,OAAA;IACE,SAAA;IACA,MAAA,GACE,MAAA,EAAQ,OAAA,CAAQ,SAAA,MAAe,IAAA,EAAM,SAAA,KAAc,OAAA,CAAQ,SAAA;EAAA;EAIjE,eAAA,IAAmB,IAAA,EAAM,mBAAA;EACzB,gBAAA,IAAoB,IAAA,EAAM,iBAAA;EAC1B,YAAA,IACE,IAAA,EAAM,gBAAA,UACN,OAAA;IAAW,SAAA;EAAA;EAEb,iBAAA,IACE,IAAA,EAAM,sBAAA,CAAuB,SAAA,WAC7B,OAAA;IAAW,SAAA;EAAA;EAEb,WAAA,IACE,IAAA,EAAM,gBAAA,CAAiB,SAAA,EAAW,aAAA,CAAc,GAAA,EAAK,SAAA,YACrD,OAAA;IAAW,SAAA;EAAA;EAEb,WAAA,IACE,IAAA,EAAM,gBAAA,UACN,OAAA;IACE,SAAA;IACA,MAAA,GACE,MAAA,EAAQ,OAAA,CAAQ,SAAA,MAAe,IAAA,EAAM,SAAA,KAAc,OAAA,CAAQ,SAAA;EAAA;AAAA;;;;UASlD,oBAAA;EA/DoB;;;;;;EAsEnC,QAAA;EA/EkB;;;;;;;;;;;;;;;;EAiGlB,iBAAA;EA3FQ;;;;;;;;;;;;;;;;;;AASV;;;;;;;;EA8GE,sBAAA;EA1GE;;;;;;EAkHF,SAAA,IAAa,KAAA,EAAO,WAAA,KAAgB,OAAA,GAAU,WAAA;AAAA;AAAA,cAGnC,aAAA,mBACO,MAAA,+BACN,WAAA,GAAc,WAAA;EAAA,QAElB,QAAA;EAAA,QAEA,QAAA;EAAA,QAEA,eAAA;EAAA,QAEA,SAAA;EAAA,QAEA,QAAA;EAAA,QAEA,sBAAA;EAAA,QAEA,SAAA;EAAA,QAEA,KAAA;EAAA,QAEA,SAAA;EAAA,QAEA,KAAA;EAQR,WAAA,CAAY,QAAA,EAAU,mBAAA,EAAqB,OAAA,EAAS,oBAAA;EA/IlC;;;;EAAA,QAqKV,WAAA;EAjKmB;;;EAyK3B,YAAA,CAAA,GAAgB,GAAA,SAAY,uBAAA;EArKe;;;EA4K3C,kBAAA,CAAA,GAAsB,uBAAA;EAvKd;;;EA8KR,WAAA,CAAY,UAAA,WAAqB,uBAAA;EA1KgC;;;EAiLjE,kBAAA,CAAmB,IAAA,WAAe,uBAAA;EA1K1B;;;EAiLR,qBAAA,CAAsB,SAAA,WAAoB,uBAAA;EAzKQ;;;;;;;;;;;;;EA0LlD,oBAAA,CACE,QAAA,EAAU,OAAA,IACV,OAAA;IAAY,eAAA;EAAA;EA3NY;;;EAmO1B,YAAA,CAAA;EAAA,QAIQ,QAAA;EAAA,QAKA,eAAA;EAIR,SAAA,GAAS,QAAA;EAwBT,WAAA;;;oBAjQ2C;;;MAmQvC,SAAA,CAAA;EAAA,IAIA,MAAA,CAAA,GAAM,SAAA;EAAA,IAIN,KAAA,CAAA;EAIJ,eAAA,GAAe,MAAA,EAAA,SAAA,KAAA,IAAA,EAAA,SAAA,SAAA,IAAA,wBAAA,SAAA,iBAAA,IAAA;EAAA,QAgBP,WAAA;EAAA,QAcA,cAAA;EAAA,UA0BE,OAAA,GAAO,MAAA,GAAA,MAAA,EAAA,WAAA,KAAA,OAAA,CAAA,cAAA,CAAA,gBAAA,CAAA,SAAA,EAAA,aAAA,CAAA,GAAA,EAAA,SAAA,GAAA,kBAAA,CAAA,GAAA,gBAAA,OAAA;;;;;;;;;EAgUjB,KAAA,GAAK,MAAA,GAAA,MAAA,EAAA,WAAA,KAAA,OAAA,CAAA,cAAA,CAAA,gBAAA,CAAA,SAAA,EAAA,aAAA,CAAA,GAAA,EAAA,SAAA,GAAA,kBAAA,CAAA,GAAA,gBAAA,OAAA;;;;;;;;;IAvnBG;;;;;;;EAyqBR,IAAA,GAAI,aAAA,EAAA,SAAA,EAAA,OAAA;;;;;EAgBJ,KAAA;AAAA"}

package/dist/ui/manager.d.ts

@@ -0,0 +1,224 @@
+import { Message } from "../types.messages.js";
+import { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, ErrorStreamEvent, EventsStreamEvent, FeedbackStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, TasksStreamEvent, ToolsStreamEvent, UpdatesStreamEvent, ValuesStreamEvent } from "../types.stream.js";
+import { BagTemplate } from "../types.template.js";
+import { SubagentStreamInterface } from "./types.js";
+import { MessageTupleManager } from "./messages.js";
+import { BaseMessage } from "@langchain/core/messages";
+
+//#region src/ui/manager.d.ts
+type GetUpdateType<Bag extends BagTemplate, StateType extends Record<string, unknown>> = Bag extends {
+  UpdateType: unknown;
+} ? Bag["UpdateType"] : Partial<StateType>;
+type GetCustomEventType<Bag extends BagTemplate> = Bag extends {
+  CustomEventType: unknown;
+} ? Bag["CustomEventType"] : unknown;
+type EventStreamMap<StateType, UpdateType, CustomType> = {
+  values: ValuesStreamEvent<StateType>;
+  updates: UpdatesStreamEvent<UpdateType>;
+  custom: CustomStreamEvent<CustomType>;
+  debug: DebugStreamEvent;
+  messages: MessagesTupleStreamEvent;
+  events: EventsStreamEvent;
+  metadata: MetadataStreamEvent;
+  checkpoints: CheckpointsStreamEvent<StateType>;
+  tasks: TasksStreamEvent<StateType, UpdateType>;
+  error: ErrorStreamEvent;
+  feedback: FeedbackStreamEvent;
+  tools: ToolsStreamEvent;
+};
+type EventStreamEvent<StateType, UpdateType, CustomType> = EventStreamMap<StateType, UpdateType, CustomType>[keyof EventStreamMap<StateType, UpdateType, CustomType>];
+interface StreamManagerEventCallbacks<StateType extends Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
+  onUpdateEvent?: (data: UpdatesStreamEvent<GetUpdateType<Bag, StateType>>["data"], options: {
+    namespace: string[] | undefined;
+    mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+  }) => void;
+  onCustomEvent?: (data: GetCustomEventType<Bag>, options: {
+    namespace: string[] | undefined;
+    mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+  }) => void;
+  onMetadataEvent?: (data: MetadataStreamEvent["data"]) => void;
+  onLangChainEvent?: (data: EventsStreamEvent["data"]) => void;
+  onDebugEvent?: (data: DebugStreamEvent["data"], options: {
+    namespace: string[] | undefined;
+  }) => void;
+  onCheckpointEvent?: (data: CheckpointsStreamEvent<StateType>["data"], options: {
+    namespace: string[] | undefined;
+  }) => void;
+  onTaskEvent?: (data: TasksStreamEvent<StateType, GetUpdateType<Bag, StateType>>["data"], options: {
+    namespace: string[] | undefined;
+  }) => void;
+  onToolEvent?: (data: ToolsStreamEvent["data"], options: {
+    namespace: string[] | undefined;
+    mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+  }) => void;
+}
+/**
+ * Options for StreamManager constructor.
+ */
+interface StreamManagerOptions {
+  /**
+   * Throttle the stream updates.
+   * If a number is provided, updates are throttled to the given milliseconds.
+   * If `true`, updates are batched in a single macrotask.
+   * If `false`, updates are not throttled.
+   */
+  throttle: number | boolean;
+  /**
+   * Tool names that indicate subagent invocation.
+   *
+   * When an AI message contains tool calls with these names, they are
+   * automatically tracked as subagent executions. This enables the
+   * `subagents`, `activeSubagents`, `getSubagent()`, and `getSubagentsByType()`
+   * properties on the stream.
+   *
+   * @default ["task"]
+   *
+   * @example
+   * ```typescript
+   * // Track both "task" and "delegate" as subagent tools
+   * subagentToolNames: ["task", "delegate", "spawn_agent"]
+   * ```
+   */
+  subagentToolNames?: string[];
+  /**
+   * Filter out messages from subagent streams in the main messages array.
+   *
+   * When enabled, messages from subagraph executions (those with a `tools:` namespace)
+   * are excluded from `stream.messages`. Instead, these messages are tracked
+   * per-subagent and accessible via `stream.subagents.get(id).messages`.
+   *
+   * This is useful for deep agent architectures where you want to display
+   * the main conversation separately from subagent activity.
+   *
+   * @default false
+   *
+   * @example
+   * ```typescript
+   * const stream = useStream({
+   *   assistantId: "my-agent",
+   *   filterSubagentMessages: true,
+   * });
+   *
+   * // Main thread messages only (no subagent messages)
+   * stream.messages
+   *
+   * // Access subagent messages individually
+   * stream.subagents.get("call_xyz").messages
+   * ```
+   */
+  filterSubagentMessages?: boolean;
+  /**
+   * Converts a @langchain/core BaseMessage to the desired output format.
+   *
+   * Defaults to `toMessageDict` which produces plain Message objects.
+   * Framework SDKs pass `toMessageClass` (identity) to keep class instances.
+   */
+  toMessage?: (chunk: BaseMessage) => Message | BaseMessage;
+}
+declare class StreamManager<StateType extends Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
+  private abortRef;
+  private messages;
+  private subagentManager;
+  private listeners;
+  private throttle;
+  private filterSubagentMessages;
+  private toMessage;
+  private queue;
+  private queueSize;
+  private state;
+  constructor(messages: MessageTupleManager, options: StreamManagerOptions);
+  /**
+   * Increment version counter to trigger React re-renders.
+   * Called when subagent state changes.
+   */
+  private bumpVersion;
+  /**
+   * Get all subagents as a Map.
+   */
+  getSubagents(): Map<string, SubagentStreamInterface>;
+  /**
+   * Get all currently running subagents.
+   */
+  getActiveSubagents(): SubagentStreamInterface[];
+  /**
+   * Get a specific subagent by tool call ID.
+   */
+  getSubagent(toolCallId: string): SubagentStreamInterface | undefined;
+  /**
+   * Get all subagents of a specific type.
+   */
+  getSubagentsByType(type: string): SubagentStreamInterface[];
+  /**
+   * Get all subagents triggered by a specific AI message.
+   */
+  getSubagentsByMessage(messageId: string): SubagentStreamInterface[];
+  /**
+   * Reconstruct subagent state from historical messages.
+   *
+   * This method should be called when loading thread history to restore
+   * subagent visualization after:
+   * - Page refresh (when stream has already completed)
+   * - Loading thread history
+   * - Navigating between threads
+   *
+   * @param messages - Array of messages from thread history
+   * @param options - Optional configuration
+   * @param options.skipIfPopulated - If true, skip reconstruction if subagents already exist
+   */
+  reconstructSubagents(messages: Message[], options?: {
+    skipIfPopulated?: boolean;
+  }): void;
+  /**
+   * Check if any subagents are currently tracked.
+   */
+  hasSubagents(): boolean;
+  private setState;
+  private notifyListeners;
+  subscribe: (listener: () => void) => () => void;
+  getSnapshot: () => {
+    isLoading: boolean;
+    values: [values: StateType, kind: "stop" | "stream"] | null;
+    error: unknown; /** Version counter to force React re-renders on subagent changes */
+    version: number;
+  };
+  get isLoading(): boolean;
+  get values(): StateType | null;
+  get error(): unknown;
+  setStreamValues: (values: StateType | ((prev: StateType | null, kind: "stop" | "stream") => StateType | null) | null, kind?: "stop" | "stream") => void;
+  private getMutateFn;
+  private matchEventType;
+  protected enqueue: (action: (signal: AbortSignal) => Promise<AsyncGenerator<EventStreamEvent<StateType, GetUpdateType<Bag, StateType>, GetCustomEventType<Bag>>, any, any>>, options: {
+    getMessages: (values: StateType) => Message[];
+    setMessages: (current: StateType, messages: Message[]) => StateType;
+    initialValues: StateType;
+    callbacks: StreamManagerEventCallbacks<StateType, Bag>;
+    onSuccess: () => void | StateType | Promise<void | StateType | null | undefined> | null | undefined;
+    onError: (error: unknown) => void | Promise<void>;
+    onFinish?: (() => void) | undefined;
+  }) => Promise<void>;
+  start: (action: (signal: AbortSignal) => Promise<AsyncGenerator<EventStreamEvent<StateType, GetUpdateType<Bag, StateType>, GetCustomEventType<Bag>>, any, any>>, options: {
+    getMessages: (values: StateType) => Message[];
+    setMessages: (current: StateType, messages: Message[]) => StateType;
+    initialValues: StateType;
+    callbacks: StreamManagerEventCallbacks<StateType, Bag>;
+    onSuccess: () => void | StateType | Promise<void | StateType | null | undefined> | null | undefined;
+    onError: (error: unknown) => void | Promise<void>;
+    onFinish?: (() => void) | undefined;
+  }, startOptions?: {
+    /**
+     * If true, abort any currently running stream before starting this one.
+     * Used for multitask_strategy: "interrupt" and "rollback" to unblock
+     * the queue so the new run request can proceed immediately.
+     */
+    abortPrevious?: boolean | undefined;
+  } | undefined) => Promise<void>;
+  stop: (historyValues: StateType, options: {
+    onStop?: ((options: {
+      mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+    }) => void) | undefined;
+  }) => Promise<void>;
+  clear: () => void;
+}
+//#endregion
+export { EventStreamEvent, StreamManager };
+//# sourceMappingURL=manager.d.ts.map

package/dist/ui/manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.d.ts","names":[],"sources":["../../src/ui/manager.ts"],"mappings":";;;;;;;;KAiCK,aAAA,aACS,WAAA,oBACM,MAAA,qBAChB,GAAA;EAAc,UAAA;AAAA,IACd,GAAA,iBACA,OAAA,CAAQ,SAAA;AAAA,KAEP,kBAAA,aAA+B,WAAA,IAAe,GAAA;EACjD,eAAA;AAAA,IAEE,GAAA;AAAA,KAGC,cAAA;EACH,MAAA,EAAQ,iBAAA,CAAkB,SAAA;EAC1B,OAAA,EAAS,kBAAA,CAAmB,UAAA;EAC5B,MAAA,EAAQ,iBAAA,CAAkB,UAAA;EAC1B,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,wBAAA;EACV,MAAA,EAAQ,iBAAA;EACR,QAAA,EAAU,mBAAA;EACV,WAAA,EAAa,sBAAA,CAAuB,SAAA;EACpC,KAAA,EAAO,gBAAA,CAAiB,SAAA,EAAW,UAAA;EACnC,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,mBAAA;EACV,KAAA,EAAO,gBAAA;AAAA;AAAA,KAGG,gBAAA,sCACV,cAAA,CAAe,SAAA,EAAW,UAAA,EAAY,UAAA,QAAkB,cAAA,CACtD,SAAA,EACA,UAAA,EACA,UAAA;AAAA,UAGM,2BAAA,mBACU,MAAA,+BACN,WAAA,GAAc,WAAA;EAE1B,aAAA,IACE,IAAA,EAAM,kBAAA,CAAmB,aAAA,CAAc,GAAA,EAAK,SAAA,YAC5C,OAAA;IACE,SAAA;IACA,MAAA,GACE,MAAA,EAAQ,OAAA,CAAQ,SAAA,MAAe,IAAA,EAAM,SAAA,KAAc,OAAA,CAAQ,SAAA;EAAA;EAIjE,aAAA,IACE,IAAA,EAAM,kBAAA,CAAmB,GAAA,GACzB,OAAA;IACE,SAAA;IACA,MAAA,GACE,MAAA,EAAQ,OAAA,CAAQ,SAAA,MAAe,IAAA,EAAM,SAAA,KAAc,OAAA,CAAQ,SAAA;EAAA;EAIjE,eAAA,IAAmB,IAAA,EAAM,mBAAA;EACzB,gBAAA,IAAoB,IAAA,EAAM,iBAAA;EAC1B,YAAA,IACE,IAAA,EAAM,gBAAA,UACN,OAAA;IAAW,SAAA;EAAA;EAEb,iBAAA,IACE,IAAA,EAAM,sBAAA,CAAuB,SAAA,WAC7B,OAAA;IAAW,SAAA;EAAA;EAEb,WAAA,IACE,IAAA,EAAM,gBAAA,CAAiB,SAAA,EAAW,aAAA,CAAc,GAAA,EAAK,SAAA,YACrD,OAAA;IAAW,SAAA;EAAA;EAEb,WAAA,IACE,IAAA,EAAM,gBAAA,UACN,OAAA;IACE,SAAA;IACA,MAAA,GACE,MAAA,EAAQ,OAAA,CAAQ,SAAA,MAAe,IAAA,EAAM,SAAA,KAAc,OAAA,CAAQ,SAAA;EAAA;AAAA;;;;UASlD,oBAAA;EA/DoB;;;;;;EAsEnC,QAAA;EA/EkB;;;;;;;;;;;;;;;;EAiGlB,iBAAA;EA3FQ;;;;;;;;;;;;;;;;;;AASV;;;;;;;;EA8GE,sBAAA;EA1GE;;;;;;EAkHF,SAAA,IAAa,KAAA,EAAO,WAAA,KAAgB,OAAA,GAAU,WAAA;AAAA;AAAA,cAGnC,aAAA,mBACO,MAAA,+BACN,WAAA,GAAc,WAAA;EAAA,QAElB,QAAA;EAAA,QAEA,QAAA;EAAA,QAEA,eAAA;EAAA,QAEA,SAAA;EAAA,QAEA,QAAA;EAAA,QAEA,sBAAA;EAAA,QAEA,SAAA;EAAA,QAEA,KAAA;EAAA,QAEA,SAAA;EAAA,QAEA,KAAA;EAQR,WAAA,CAAY,QAAA,EAAU,mBAAA,EAAqB,OAAA,EAAS,oBAAA;EA/IlC;;;;EAAA,QAqKV,WAAA;EAjKmB;;;EAyK3B,YAAA,CAAA,GAAgB,GAAA,SAAY,uBAAA;EArKe;;;EA4K3C,kBAAA,CAAA,GAAsB,uBAAA;EAvKd;;;EA8KR,WAAA,CAAY,UAAA,WAAqB,uBAAA;EA1KgC;;;EAiLjE,kBAAA,CAAmB,IAAA,WAAe,uBAAA;EA1K1B;;;EAiLR,qBAAA,CAAsB,SAAA,WAAoB,uBAAA;EAzKQ;;;;;;;;;;;;;EA0LlD,oBAAA,CACE,QAAA,EAAU,OAAA,IACV,OAAA;IAAY,eAAA;EAAA;EA3NY;;;EAmO1B,YAAA,CAAA;EAAA,QAIQ,QAAA;EAAA,QAKA,eAAA;EAIR,SAAA,GAAS,QAAA;EAwBT,WAAA;;;oBAjQ2C;;;MAmQvC,SAAA,CAAA;EAAA,IAIA,MAAA,CAAA,GAAM,SAAA;EAAA,IAIN,KAAA,CAAA;EAIJ,eAAA,GAAe,MAAA,EAAA,SAAA,KAAA,IAAA,EAAA,SAAA,SAAA,IAAA,wBAAA,SAAA,iBAAA,IAAA;EAAA,QAgBP,WAAA;EAAA,QAcA,cAAA;EAAA,UA0BE,OAAA,GAAO,MAAA,GAAA,MAAA,EAAA,WAAA,KAAA,OAAA,CAAA,cAAA,CAAA,gBAAA,CAAA,SAAA,EAAA,aAAA,CAAA,GAAA,EAAA,SAAA,GAAA,kBAAA,CAAA,GAAA,gBAAA,OAAA;;;;;;;;;EAgUjB,KAAA,GAAK,MAAA,GAAA,MAAA,EAAA,WAAA,KAAA,OAAA,CAAA,cAAA,CAAA,gBAAA,CAAA,SAAA,EAAA,aAAA,CAAA,GAAA,EAAA,SAAA,GAAA,kBAAA,CAAA,GAAA,gBAAA,OAAA;;;;;;;;;IAvnBG;;;;;;;EAyqBR,IAAA,GAAI,aAAA,EAAA,SAAA,EAAA,OAAA;;;;;EAgBJ,KAAA;AAAA"}

package/dist/ui/manager.js

@@ -15,6 +15,7 @@ var StreamManager = class {
 	listeners = /* @__PURE__ */ new Set();
 	throttle;
 	filterSubagentMessages;
+	toMessage;
 	queue = Promise.resolve();
 	queueSize = 0;
 	state;
@@ -28,9 +29,11 @@ var StreamManager = class {
 		};
 		this.throttle = options.throttle;
 		this.filterSubagentMessages = options.filterSubagentMessages ?? false;
+		this.toMessage = options.toMessage ?? toMessageDict;
 		this.subagentManager = new SubagentManager({
 			subagentToolNames: options.subagentToolNames,
-			onSubagentChange: () => this.bumpVersion()
+			onSubagentChange: () => this.bumpVersion(),
+			toMessage: this.toMessage
 		});
 	}
 	/**
@@ -215,6 +218,10 @@ var StreamManager = class {
 				if (this.matchEventType("checkpoints", event, data)) options.callbacks.onCheckpointEvent?.(data, { namespace });
 				if (this.matchEventType("tasks", event, data)) options.callbacks.onTaskEvent?.(data, { namespace });
 				if (this.matchEventType("debug", event, data)) options.callbacks.onDebugEvent?.(data, { namespace });
+				if (this.matchEventType("tools", event, data)) options.callbacks.onToolEvent?.(data, {
+					namespace,
+					mutate
+				});
 				if (event === "values" || event.startsWith("values|")) if (namespace && isSubagentNamespace(namespace)) {
 					const namespaceId = extractToolCallIdFromNamespace(namespace);
 					if (namespaceId && this.filterSubagentMessages) {
@@ -259,7 +266,7 @@ var StreamManager = class {
 						if (chunk.getType() === "remove") if (chunk.id === REMOVE_ALL_MESSAGES) messages = [];
 						else messages.splice(index, 1);
 						else {
-							const msgDict = toMessageDict(chunk);
+							const msgDict = this.toMessage(chunk);
 							messages[index] = msgDict;
 							if (!isFromSubagent && msgDict.type === "ai" && "tool_calls" in msgDict && Array.isArray(msgDict.tool_calls)) this.subagentManager.registerFromToolCalls(msgDict.tool_calls, msgDict.id);
 							if (!isFromSubagent && msgDict.type === "tool" && "tool_call_id" in msgDict) {
@@ -293,7 +300,9 @@ var StreamManager = class {
 	start = async (action, options, startOptions) => {
 		if (startOptions?.abortPrevious) this.abortRef.abort();
 		this.queueSize += 1;
-		this.queue = this.queue.then(() => this.enqueue(action, options));
+		const queued = this.queue.then(() => this.enqueue(action, options));
+		this.queue = queued;
+		await queued;
 	};
 	stop = async (historyValues, options) => {
 		this.abortRef.abort();