@langchain/langgraph-sdk 1.6.4 → 1.7.0

package/dist/ui/branching.d.ts

@@ -1,4 +1,5 @@
 import { ThreadState } from "../schema.js";
+import { Message } from "../types.messages.js";
 
 //#region src/ui/branching.d.ts
 interface Node<StateType = any> {
@@ -14,6 +15,30 @@ interface Sequence<StateType = any> {
   type: "sequence";
   items: Array<Node<StateType> | Fork<StateType>>;
 }
+type BranchByCheckpoint = Record<string, {
+  branch: string | undefined;
+  branchOptions: string[] | undefined;
+}>;
+declare function getBranchContext<StateType extends Record<string, unknown>>(branch: string, history: ThreadState<StateType>[] | undefined): {
+  branchTree: Sequence<any>;
+  flatHistory: ThreadState<any>[];
+  branchByCheckpoint: BranchByCheckpoint;
+  threadHead: ThreadState<any> | undefined;
+};
+declare function getMessagesMetadataMap<StateType extends Record<string, unknown>>(options: {
+  initialValues: StateType | null | undefined;
+  history: ThreadState<StateType>[] | null | undefined;
+  getMessages: (values: StateType) => Message[];
+  branchContext: {
+    threadHead: ThreadState<StateType> | undefined;
+    branchByCheckpoint: BranchByCheckpoint;
+  };
+}): {
+  messageId: string;
+  firstSeenState: ThreadState<StateType> | undefined;
+  branch: string | undefined;
+  branchOptions: string[] | undefined;
+}[];
 //#endregion
-export { Sequence };
+export { Sequence, getBranchContext, getMessagesMetadataMap };
 //# sourceMappingURL=branching.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"branching.d.ts","names":[],"sources":["../../src/ui/branching.ts"],"mappings":";;;UAGU,IAAA;EACR,IAAA;EACA,KAAA,EAAO,WAAA,CAAY,SAAA;EACnB,IAAA;AAAA;AAAA,UAIQ,IAAA;EACR,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,QAAA,CAAS,SAAA;AAAA;AAAA,UAIP,QAAA;EACf,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,IAAA,CAAK,SAAA,IAAa,IAAA,CAAK,SAAA;AAAA"}
+{"version":3,"file":"branching.d.ts","names":[],"sources":["../../src/ui/branching.ts"],"mappings":";;;;UAKU,IAAA;EACR,IAAA;EACA,KAAA,EAAO,WAAA,CAAY,SAAA;EACnB,IAAA;AAAA;AAAA,UAIQ,IAAA;EACR,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,QAAA,CAAS,SAAA;AAAA;AAAA,UAIP,QAAA;EACf,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,IAAA,CAAK,SAAA,IAAa,IAAA,CAAK,SAAA;AAAA;AAAA,KAqIjC,kBAAA,GAAqB,MAAA;EAEtB,MAAA;EAA4B,aAAA;AAAA;AAAA,iBA2DhB,gBAAA,mBAAmC,MAAA,kBAAA,CACjD,MAAA,UACA,OAAA,EAAS,WAAA,CAAY,SAAA;;;;;;iBAiBP,sBAAA,mBACI,MAAA,kBAAA,CAClB,OAAA;EACA,aAAA,EAAe,SAAA;EACf,OAAA,EAAS,WAAA,CAAY,SAAA;EACrB,WAAA,GAAc,MAAA,EAAQ,SAAA,KAAc,OAAA;EAEpC,aAAA;IACE,UAAA,EAAY,WAAA,CAAY,SAAA;IACxB,kBAAA,EAAoB,kBAAA;EAAA;AAAA"}

package/dist/ui/branching.js

@@ -1,3 +1,5 @@
+import { findLast } from "./utils.js";
+
 //#region src/ui/branching.ts
 function getBranchSequence(history) {
 	const nodeIds = /* @__PURE__ */ new Set();
@@ -146,7 +148,29 @@ function getBranchContext(branch, history) {
 		threadHead: flatHistory.at(-1)
 	};
 }
+function getMessagesMetadataMap(options) {
+	const currentValues = options.branchContext.threadHead?.values ?? options.initialValues ?? {};
+	const alreadyShown = /* @__PURE__ */ new Set();
+	return options.getMessages(currentValues).map((message, idx) => {
+		const messageId = message.id ?? idx;
+		const firstSeenState = findLast(options.history ?? [], (state) => options.getMessages(state.values).map((m, idx) => m.id ?? idx).includes(messageId));
+		const checkpointId = firstSeenState?.checkpoint?.checkpoint_id;
+		let branch = checkpointId != null ? options.branchContext.branchByCheckpoint[checkpointId] : void 0;
+		if (!branch?.branch?.length) branch = void 0;
+		const optionsShown = branch?.branchOptions?.flat(2).join(",");
+		if (optionsShown) {
+			if (alreadyShown.has(optionsShown)) branch = void 0;
+			alreadyShown.add(optionsShown);
+		}
+		return {
+			messageId: messageId.toString(),
+			firstSeenState,
+			branch: branch?.branch,
+			branchOptions: branch?.branchOptions
+		};
+	});
+}
 
 //#endregion
-export { getBranchContext };
+export { getBranchContext, getMessagesMetadataMap };
 //# sourceMappingURL=branching.js.map

package/dist/ui/branching.js.map

@@ -1 +1 @@
-{"version":3,"file":"branching.js","names":[],"sources":["../../src/ui/branching.ts"],"sourcesContent":["import type { ThreadState } from \"../schema.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Node<StateType = any> {\n  type: \"node\";\n  value: ThreadState<StateType>;\n  path: string[];\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Fork<StateType = any> {\n  type: \"fork\";\n  items: Array<Sequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport interface Sequence<StateType = any> {\n  type: \"sequence\";\n  items: Array<Node<StateType> | Fork<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidFork<StateType = any> {\n  type: \"fork\";\n  items: Array<ValidSequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidSequence<StateType = any> {\n  type: \"sequence\";\n  items: [Node<StateType>, ...(Node<StateType> | ValidFork<StateType>)[]];\n}\n\nexport function getBranchSequence<StateType extends Record<string, unknown>>(\n  history: ThreadState<StateType>[]\n) {\n  const nodeIds = new Set<string>();\n  const childrenMap: Record<string, ThreadState<StateType>[]> = {};\n\n  // Short circuit if there's only a singular one state\n  if (history.length <= 1) {\n    return {\n      rootSequence: {\n        type: \"sequence\",\n        items: history.map((value) => ({ type: \"node\", value, path: [] })),\n      } satisfies Sequence<StateType>,\n      paths: [],\n    };\n  }\n\n  // First pass - collect nodes for each checkpoint\n  history.forEach((state) => {\n    const checkpointId = state.parent_checkpoint?.checkpoint_id ?? \"$\";\n    childrenMap[checkpointId] ??= [];\n    childrenMap[checkpointId].push(state);\n\n    if (state.checkpoint?.checkpoint_id != null) {\n      nodeIds.add(state.checkpoint.checkpoint_id);\n    }\n  });\n\n  // If dealing with partial history, take the branch\n  // with the latest checkpoint and mark it as the root.\n  const maxId = (...ids: (string | null)[]) =>\n    ids\n      .filter((i): i is string => i != null)\n      .sort((a, b) => a.localeCompare(b))\n      .at(-1)!;\n\n  const lastOrphanedNode =\n    childrenMap.$ == null\n      ? Object.keys(childrenMap)\n          .filter((parentId) => !nodeIds.has(parentId))\n          .map((parentId) => {\n            const queue: string[] = [parentId];\n            const seen = new Set<string>();\n\n            let lastId = parentId;\n\n            while (queue.length > 0) {\n              const current = queue.shift()!;\n\n              if (seen.has(current)) continue;\n              seen.add(current);\n\n              const children = (childrenMap[current] ?? []).flatMap(\n                (i) => i.checkpoint?.checkpoint_id ?? []\n              );\n\n              lastId = maxId(lastId, ...children);\n              queue.push(...children);\n            }\n\n            return { parentId, lastId };\n          })\n          .sort((a, b) => a.lastId.localeCompare(b.lastId))\n          .at(-1)?.parentId\n      : undefined;\n\n  if (lastOrphanedNode != null) childrenMap.$ = childrenMap[lastOrphanedNode];\n\n  // Second pass - create a tree of sequences\n  type Task = { id: string; sequence: Sequence; path: string[] };\n  const rootSequence: Sequence = { type: \"sequence\", items: [] };\n  const queue: Task[] = [{ id: \"$\", sequence: rootSequence, path: [] }];\n\n  const paths: string[][] = [];\n\n  const visited = new Set<string>();\n  while (queue.length > 0) {\n    const task = queue.shift()!;\n    if (visited.has(task.id)) continue;\n    visited.add(task.id);\n\n    const children = childrenMap[task.id];\n    if (children == null || children.length === 0) continue;\n\n    // If we've encountered a fork (2+ children), push the fork\n    // to the sequence and add a new sequence for each child\n    let fork: Fork | undefined;\n    if (children.length > 1) {\n      fork = { type: \"fork\", items: [] };\n      task.sequence.items.push(fork);\n    }\n\n    for (const value of children) {\n      const id = value.checkpoint?.checkpoint_id;\n      if (id == null) continue;\n\n      let { sequence } = task;\n      let { path } = task;\n      if (fork != null) {\n        sequence = { type: \"sequence\", items: [] };\n        fork.items.unshift(sequence);\n\n        path = path.slice();\n        path.push(id);\n        paths.push(path);\n      }\n\n      sequence.items.push({ type: \"node\", value, path });\n      queue.push({ id, sequence, path });\n    }\n  }\n\n  return { rootSequence, paths };\n}\n\nconst PATH_SEP = \">\";\nconst ROOT_ID = \"$\";\n\n// Get flat view\nexport function getBranchView<StateType extends Record<string, unknown>>(\n  sequence: Sequence<StateType>,\n  paths: string[][],\n  branch: string\n) {\n  const path = branch.split(PATH_SEP);\n  const pathMap: Record<string, string[][]> = {};\n\n  for (const path of paths) {\n    const parent = path.at(-2) ?? ROOT_ID;\n    pathMap[parent] ??= [];\n    pathMap[parent].unshift(path);\n  }\n\n  const history: ThreadState<StateType>[] = [];\n  const branchByCheckpoint: Record<\n    string,\n    { branch: string | undefined; branchOptions: string[] | undefined }\n  > = {};\n\n  const forkStack = path.slice();\n  const queue: (Node<StateType> | Fork<StateType>)[] = [...sequence.items];\n\n  while (queue.length > 0) {\n    const item = queue.shift()!;\n\n    if (item.type === \"node\") {\n      history.push(item.value);\n      const checkpointId = item.value.checkpoint?.checkpoint_id;\n      if (checkpointId == null) continue;\n\n      branchByCheckpoint[checkpointId] = {\n        branch: item.path.join(PATH_SEP),\n        branchOptions: (item.path.length > 0\n          ? pathMap[item.path.at(-2) ?? ROOT_ID] ?? []\n          : []\n        ).map((p) => p.join(PATH_SEP)),\n      };\n    }\n    if (item.type === \"fork\") {\n      const forkId = forkStack.shift();\n      const index =\n        forkId != null\n          ? item.items.findIndex((value) => {\n              const firstItem = value.items.at(0);\n              if (!firstItem || firstItem.type !== \"node\") return false;\n              return firstItem.value.checkpoint?.checkpoint_id === forkId;\n            })\n          : -1;\n\n      const nextItems = item.items.at(index)?.items ?? [];\n      queue.push(...nextItems);\n    }\n  }\n\n  return { history, branchByCheckpoint };\n}\n\nexport function getBranchContext<StateType extends Record<string, unknown>>(\n  branch: string,\n  history: ThreadState<StateType>[] | undefined\n) {\n  const { rootSequence: branchTree, paths } = getBranchSequence(history ?? []);\n  const { history: flatHistory, branchByCheckpoint } = getBranchView(\n    branchTree,\n    paths,\n    branch\n  );\n\n  return {\n    branchTree,\n    flatHistory,\n    branchByCheckpoint,\n    threadHead: flatHistory.at(-1),\n  };\n}\n"],"mappings":";AAiCA,SAAgB,kBACd,SACA;CACA,MAAM,0BAAU,IAAI,KAAa;CACjC,MAAM,cAAwD,EAAE;AAGhE,KAAI,QAAQ,UAAU,EACpB,QAAO;EACL,cAAc;GACZ,MAAM;GACN,OAAO,QAAQ,KAAK,WAAW;IAAE,MAAM;IAAQ;IAAO,MAAM,EAAE;IAAE,EAAE;GACnE;EACD,OAAO,EAAE;EACV;AAIH,SAAQ,SAAS,UAAU;EACzB,MAAM,eAAe,MAAM,mBAAmB,iBAAiB;AAC/D,cAAY,kBAAkB,EAAE;AAChC,cAAY,cAAc,KAAK,MAAM;AAErC,MAAI,MAAM,YAAY,iBAAiB,KACrC,SAAQ,IAAI,MAAM,WAAW,cAAc;GAE7C;CAIF,MAAM,SAAS,GAAG,QAChB,IACG,QAAQ,MAAmB,KAAK,KAAK,CACrC,MAAM,GAAG,MAAM,EAAE,cAAc,EAAE,CAAC,CAClC,GAAG,GAAG;CAEX,MAAM,mBACJ,YAAY,KAAK,OACb,OAAO,KAAK,YAAY,CACrB,QAAQ,aAAa,CAAC,QAAQ,IAAI,SAAS,CAAC,CAC5C,KAAK,aAAa;EACjB,MAAM,QAAkB,CAAC,SAAS;EAClC,MAAM,uBAAO,IAAI,KAAa;EAE9B,IAAI,SAAS;AAEb,SAAO,MAAM,SAAS,GAAG;GACvB,MAAM,UAAU,MAAM,OAAO;AAE7B,OAAI,KAAK,IAAI,QAAQ,CAAE;AACvB,QAAK,IAAI,QAAQ;GAEjB,MAAM,YAAY,YAAY,YAAY,EAAE,EAAE,SAC3C,MAAM,EAAE,YAAY,iBAAiB,EAAE,CACzC;AAED,YAAS,MAAM,QAAQ,GAAG,SAAS;AACnC,SAAM,KAAK,GAAG,SAAS;;AAGzB,SAAO;GAAE;GAAU;GAAQ;GAC3B,CACD,MAAM,GAAG,MAAM,EAAE,OAAO,cAAc,EAAE,OAAO,CAAC,CAChD,GAAG,GAAG,EAAE,WACX;AAEN,KAAI,oBAAoB,KAAM,aAAY,IAAI,YAAY;CAI1D,MAAM,eAAyB;EAAE,MAAM;EAAY,OAAO,EAAE;EAAE;CAC9D,MAAM,QAAgB,CAAC;EAAE,IAAI;EAAK,UAAU;EAAc,MAAM,EAAE;EAAE,CAAC;CAErE,MAAM,QAAoB,EAAE;CAE5B,MAAM,0BAAU,IAAI,KAAa;AACjC,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAC1B,MAAI,QAAQ,IAAI,KAAK,GAAG,CAAE;AAC1B,UAAQ,IAAI,KAAK,GAAG;EAEpB,MAAM,WAAW,YAAY,KAAK;AAClC,MAAI,YAAY,QAAQ,SAAS,WAAW,EAAG;EAI/C,IAAI;AACJ,MAAI,SAAS,SAAS,GAAG;AACvB,UAAO;IAAE,MAAM;IAAQ,OAAO,EAAE;IAAE;AAClC,QAAK,SAAS,MAAM,KAAK,KAAK;;AAGhC,OAAK,MAAM,SAAS,UAAU;GAC5B,MAAM,KAAK,MAAM,YAAY;AAC7B,OAAI,MAAM,KAAM;GAEhB,IAAI,EAAE,aAAa;GACnB,IAAI,EAAE,SAAS;AACf,OAAI,QAAQ,MAAM;AAChB,eAAW;KAAE,MAAM;KAAY,OAAO,EAAE;KAAE;AAC1C,SAAK,MAAM,QAAQ,SAAS;AAE5B,WAAO,KAAK,OAAO;AACnB,SAAK,KAAK,GAAG;AACb,UAAM,KAAK,KAAK;;AAGlB,YAAS,MAAM,KAAK;IAAE,MAAM;IAAQ;IAAO;IAAM,CAAC;AAClD,SAAM,KAAK;IAAE;IAAI;IAAU;IAAM,CAAC;;;AAItC,QAAO;EAAE;EAAc;EAAO;;AAGhC,MAAM,WAAW;AACjB,MAAM,UAAU;AAGhB,SAAgB,cACd,UACA,OACA,QACA;CACA,MAAM,OAAO,OAAO,MAAM,SAAS;CACnC,MAAM,UAAsC,EAAE;AAE9C,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,SAAS,KAAK,GAAG,GAAG,IAAI;AAC9B,UAAQ,YAAY,EAAE;AACtB,UAAQ,QAAQ,QAAQ,KAAK;;CAG/B,MAAM,UAAoC,EAAE;CAC5C,MAAM,qBAGF,EAAE;CAEN,MAAM,YAAY,KAAK,OAAO;CAC9B,MAAM,QAA+C,CAAC,GAAG,SAAS,MAAM;AAExE,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAE1B,MAAI,KAAK,SAAS,QAAQ;AACxB,WAAQ,KAAK,KAAK,MAAM;GACxB,MAAM,eAAe,KAAK,MAAM,YAAY;AAC5C,OAAI,gBAAgB,KAAM;AAE1B,sBAAmB,gBAAgB;IACjC,QAAQ,KAAK,KAAK,KAAK,SAAS;IAChC,gBAAgB,KAAK,KAAK,SAAS,IAC/B,QAAQ,KAAK,KAAK,GAAG,GAAG,IAAI,YAAY,EAAE,GAC1C,EAAE,EACJ,KAAK,MAAM,EAAE,KAAK,SAAS,CAAC;IAC/B;;AAEH,MAAI,KAAK,SAAS,QAAQ;GACxB,MAAM,SAAS,UAAU,OAAO;GAChC,MAAM,QACJ,UAAU,OACN,KAAK,MAAM,WAAW,UAAU;IAC9B,MAAM,YAAY,MAAM,MAAM,GAAG,EAAE;AACnC,QAAI,CAAC,aAAa,UAAU,SAAS,OAAQ,QAAO;AACpD,WAAO,UAAU,MAAM,YAAY,kBAAkB;KACrD,GACF;GAEN,MAAM,YAAY,KAAK,MAAM,GAAG,MAAM,EAAE,SAAS,EAAE;AACnD,SAAM,KAAK,GAAG,UAAU;;;AAI5B,QAAO;EAAE;EAAS;EAAoB;;AAGxC,SAAgB,iBACd,QACA,SACA;CACA,MAAM,EAAE,cAAc,YAAY,UAAU,kBAAkB,WAAW,EAAE,CAAC;CAC5E,MAAM,EAAE,SAAS,aAAa,uBAAuB,cACnD,YACA,OACA,OACD;AAED,QAAO;EACL;EACA;EACA;EACA,YAAY,YAAY,GAAG,GAAG;EAC/B"}
+{"version":3,"file":"branching.js","names":[],"sources":["../../src/ui/branching.ts"],"sourcesContent":["import type { ThreadState } from \"../schema.js\";\nimport { Message } from \"../types.messages.js\";\nimport { findLast } from \"./utils.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Node<StateType = any> {\n  type: \"node\";\n  value: ThreadState<StateType>;\n  path: string[];\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Fork<StateType = any> {\n  type: \"fork\";\n  items: Array<Sequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport interface Sequence<StateType = any> {\n  type: \"sequence\";\n  items: Array<Node<StateType> | Fork<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidFork<StateType = any> {\n  type: \"fork\";\n  items: Array<ValidSequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidSequence<StateType = any> {\n  type: \"sequence\";\n  items: [Node<StateType>, ...(Node<StateType> | ValidFork<StateType>)[]];\n}\n\nexport function getBranchSequence<StateType extends Record<string, unknown>>(\n  history: ThreadState<StateType>[]\n) {\n  const nodeIds = new Set<string>();\n  const childrenMap: Record<string, ThreadState<StateType>[]> = {};\n\n  // Short circuit if there's only a singular one state\n  if (history.length <= 1) {\n    return {\n      rootSequence: {\n        type: \"sequence\",\n        items: history.map((value) => ({ type: \"node\", value, path: [] })),\n      } satisfies Sequence<StateType>,\n      paths: [],\n    };\n  }\n\n  // First pass - collect nodes for each checkpoint\n  history.forEach((state) => {\n    const checkpointId = state.parent_checkpoint?.checkpoint_id ?? \"$\";\n    childrenMap[checkpointId] ??= [];\n    childrenMap[checkpointId].push(state);\n\n    if (state.checkpoint?.checkpoint_id != null) {\n      nodeIds.add(state.checkpoint.checkpoint_id);\n    }\n  });\n\n  // If dealing with partial history, take the branch\n  // with the latest checkpoint and mark it as the root.\n  const maxId = (...ids: (string | null)[]) =>\n    ids\n      .filter((i): i is string => i != null)\n      .sort((a, b) => a.localeCompare(b))\n      .at(-1)!;\n\n  const lastOrphanedNode =\n    childrenMap.$ == null\n      ? Object.keys(childrenMap)\n          .filter((parentId) => !nodeIds.has(parentId))\n          .map((parentId) => {\n            const queue: string[] = [parentId];\n            const seen = new Set<string>();\n\n            let lastId = parentId;\n\n            while (queue.length > 0) {\n              const current = queue.shift()!;\n\n              if (seen.has(current)) continue;\n              seen.add(current);\n\n              const children = (childrenMap[current] ?? []).flatMap(\n                (i) => i.checkpoint?.checkpoint_id ?? []\n              );\n\n              lastId = maxId(lastId, ...children);\n              queue.push(...children);\n            }\n\n            return { parentId, lastId };\n          })\n          .sort((a, b) => a.lastId.localeCompare(b.lastId))\n          .at(-1)?.parentId\n      : undefined;\n\n  if (lastOrphanedNode != null) childrenMap.$ = childrenMap[lastOrphanedNode];\n\n  // Second pass - create a tree of sequences\n  type Task = { id: string; sequence: Sequence; path: string[] };\n  const rootSequence: Sequence = { type: \"sequence\", items: [] };\n  const queue: Task[] = [{ id: \"$\", sequence: rootSequence, path: [] }];\n\n  const paths: string[][] = [];\n\n  const visited = new Set<string>();\n  while (queue.length > 0) {\n    const task = queue.shift()!;\n    if (visited.has(task.id)) continue;\n    visited.add(task.id);\n\n    const children = childrenMap[task.id];\n    if (children == null || children.length === 0) continue;\n\n    // If we've encountered a fork (2+ children), push the fork\n    // to the sequence and add a new sequence for each child\n    let fork: Fork | undefined;\n    if (children.length > 1) {\n      fork = { type: \"fork\", items: [] };\n      task.sequence.items.push(fork);\n    }\n\n    for (const value of children) {\n      const id = value.checkpoint?.checkpoint_id;\n      if (id == null) continue;\n\n      let { sequence } = task;\n      let { path } = task;\n      if (fork != null) {\n        sequence = { type: \"sequence\", items: [] };\n        fork.items.unshift(sequence);\n\n        path = path.slice();\n        path.push(id);\n        paths.push(path);\n      }\n\n      sequence.items.push({ type: \"node\", value, path });\n      queue.push({ id, sequence, path });\n    }\n  }\n\n  return { rootSequence, paths };\n}\n\nconst PATH_SEP = \">\";\nconst ROOT_ID = \"$\";\n\ntype BranchByCheckpoint = Record<\n  string,\n  { branch: string | undefined; branchOptions: string[] | undefined }\n>;\n\n// Get flat view\nexport function getBranchView<StateType extends Record<string, unknown>>(\n  sequence: Sequence<StateType>,\n  paths: string[][],\n  branch: string\n) {\n  const path = branch.split(PATH_SEP);\n  const pathMap: Record<string, string[][]> = {};\n\n  for (const path of paths) {\n    const parent = path.at(-2) ?? ROOT_ID;\n    pathMap[parent] ??= [];\n    pathMap[parent].unshift(path);\n  }\n\n  const history: ThreadState<StateType>[] = [];\n  const branchByCheckpoint: BranchByCheckpoint = {};\n\n  const forkStack = path.slice();\n  const queue: (Node<StateType> | Fork<StateType>)[] = [...sequence.items];\n\n  while (queue.length > 0) {\n    const item = queue.shift()!;\n\n    if (item.type === \"node\") {\n      history.push(item.value);\n      const checkpointId = item.value.checkpoint?.checkpoint_id;\n      if (checkpointId == null) continue;\n\n      branchByCheckpoint[checkpointId] = {\n        branch: item.path.join(PATH_SEP),\n        branchOptions: (item.path.length > 0\n          ? pathMap[item.path.at(-2) ?? ROOT_ID] ?? []\n          : []\n        ).map((p) => p.join(PATH_SEP)),\n      };\n    }\n    if (item.type === \"fork\") {\n      const forkId = forkStack.shift();\n      const index =\n        forkId != null\n          ? item.items.findIndex((value) => {\n              const firstItem = value.items.at(0);\n              if (!firstItem || firstItem.type !== \"node\") return false;\n              return firstItem.value.checkpoint?.checkpoint_id === forkId;\n            })\n          : -1;\n\n      const nextItems = item.items.at(index)?.items ?? [];\n      queue.push(...nextItems);\n    }\n  }\n\n  return { history, branchByCheckpoint };\n}\n\nexport function getBranchContext<StateType extends Record<string, unknown>>(\n  branch: string,\n  history: ThreadState<StateType>[] | undefined\n) {\n  const { rootSequence: branchTree, paths } = getBranchSequence(history ?? []);\n  const { history: flatHistory, branchByCheckpoint } = getBranchView(\n    branchTree,\n    paths,\n    branch\n  );\n\n  return {\n    branchTree,\n    flatHistory,\n    branchByCheckpoint,\n    threadHead: flatHistory.at(-1),\n  };\n}\n\nexport function getMessagesMetadataMap<\n  StateType extends Record<string, unknown>\n>(options: {\n  initialValues: StateType | null | undefined;\n  history: ThreadState<StateType>[] | null | undefined;\n  getMessages: (values: StateType) => Message[];\n\n  branchContext: {\n    threadHead: ThreadState<StateType> | undefined;\n    branchByCheckpoint: BranchByCheckpoint;\n  };\n}) {\n  const currentValues =\n    options.branchContext.threadHead?.values ??\n    options.initialValues ??\n    ({} as StateType);\n\n  const alreadyShown = new Set<string>();\n  return options.getMessages(currentValues).map((message, idx) => {\n    const messageId = message.id ?? idx;\n\n    // Find the first checkpoint where the message was seen\n    const firstSeenState = findLast(options.history ?? [], (state) =>\n      options\n        .getMessages(state.values)\n        .map((m, idx) => m.id ?? idx)\n        .includes(messageId)\n    );\n\n    const checkpointId = firstSeenState?.checkpoint?.checkpoint_id;\n    let branch =\n      checkpointId != null\n        ? options.branchContext.branchByCheckpoint[checkpointId]\n        : undefined;\n    if (!branch?.branch?.length) branch = undefined;\n\n    // serialize branches\n    const optionsShown = branch?.branchOptions?.flat(2).join(\",\");\n    if (optionsShown) {\n      if (alreadyShown.has(optionsShown)) branch = undefined;\n      alreadyShown.add(optionsShown);\n    }\n\n    return {\n      messageId: messageId.toString(),\n      firstSeenState,\n\n      branch: branch?.branch,\n      branchOptions: branch?.branchOptions,\n    };\n  });\n}\n"],"mappings":";;;AAmCA,SAAgB,kBACd,SACA;CACA,MAAM,0BAAU,IAAI,KAAa;CACjC,MAAM,cAAwD,EAAE;AAGhE,KAAI,QAAQ,UAAU,EACpB,QAAO;EACL,cAAc;GACZ,MAAM;GACN,OAAO,QAAQ,KAAK,WAAW;IAAE,MAAM;IAAQ;IAAO,MAAM,EAAE;IAAE,EAAE;GACnE;EACD,OAAO,EAAE;EACV;AAIH,SAAQ,SAAS,UAAU;EACzB,MAAM,eAAe,MAAM,mBAAmB,iBAAiB;AAC/D,cAAY,kBAAkB,EAAE;AAChC,cAAY,cAAc,KAAK,MAAM;AAErC,MAAI,MAAM,YAAY,iBAAiB,KACrC,SAAQ,IAAI,MAAM,WAAW,cAAc;GAE7C;CAIF,MAAM,SAAS,GAAG,QAChB,IACG,QAAQ,MAAmB,KAAK,KAAK,CACrC,MAAM,GAAG,MAAM,EAAE,cAAc,EAAE,CAAC,CAClC,GAAG,GAAG;CAEX,MAAM,mBACJ,YAAY,KAAK,OACb,OAAO,KAAK,YAAY,CACrB,QAAQ,aAAa,CAAC,QAAQ,IAAI,SAAS,CAAC,CAC5C,KAAK,aAAa;EACjB,MAAM,QAAkB,CAAC,SAAS;EAClC,MAAM,uBAAO,IAAI,KAAa;EAE9B,IAAI,SAAS;AAEb,SAAO,MAAM,SAAS,GAAG;GACvB,MAAM,UAAU,MAAM,OAAO;AAE7B,OAAI,KAAK,IAAI,QAAQ,CAAE;AACvB,QAAK,IAAI,QAAQ;GAEjB,MAAM,YAAY,YAAY,YAAY,EAAE,EAAE,SAC3C,MAAM,EAAE,YAAY,iBAAiB,EAAE,CACzC;AAED,YAAS,MAAM,QAAQ,GAAG,SAAS;AACnC,SAAM,KAAK,GAAG,SAAS;;AAGzB,SAAO;GAAE;GAAU;GAAQ;GAC3B,CACD,MAAM,GAAG,MAAM,EAAE,OAAO,cAAc,EAAE,OAAO,CAAC,CAChD,GAAG,GAAG,EAAE,WACX;AAEN,KAAI,oBAAoB,KAAM,aAAY,IAAI,YAAY;CAI1D,MAAM,eAAyB;EAAE,MAAM;EAAY,OAAO,EAAE;EAAE;CAC9D,MAAM,QAAgB,CAAC;EAAE,IAAI;EAAK,UAAU;EAAc,MAAM,EAAE;EAAE,CAAC;CAErE,MAAM,QAAoB,EAAE;CAE5B,MAAM,0BAAU,IAAI,KAAa;AACjC,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAC1B,MAAI,QAAQ,IAAI,KAAK,GAAG,CAAE;AAC1B,UAAQ,IAAI,KAAK,GAAG;EAEpB,MAAM,WAAW,YAAY,KAAK;AAClC,MAAI,YAAY,QAAQ,SAAS,WAAW,EAAG;EAI/C,IAAI;AACJ,MAAI,SAAS,SAAS,GAAG;AACvB,UAAO;IAAE,MAAM;IAAQ,OAAO,EAAE;IAAE;AAClC,QAAK,SAAS,MAAM,KAAK,KAAK;;AAGhC,OAAK,MAAM,SAAS,UAAU;GAC5B,MAAM,KAAK,MAAM,YAAY;AAC7B,OAAI,MAAM,KAAM;GAEhB,IAAI,EAAE,aAAa;GACnB,IAAI,EAAE,SAAS;AACf,OAAI,QAAQ,MAAM;AAChB,eAAW;KAAE,MAAM;KAAY,OAAO,EAAE;KAAE;AAC1C,SAAK,MAAM,QAAQ,SAAS;AAE5B,WAAO,KAAK,OAAO;AACnB,SAAK,KAAK,GAAG;AACb,UAAM,KAAK,KAAK;;AAGlB,YAAS,MAAM,KAAK;IAAE,MAAM;IAAQ;IAAO;IAAM,CAAC;AAClD,SAAM,KAAK;IAAE;IAAI;IAAU;IAAM,CAAC;;;AAItC,QAAO;EAAE;EAAc;EAAO;;AAGhC,MAAM,WAAW;AACjB,MAAM,UAAU;AAQhB,SAAgB,cACd,UACA,OACA,QACA;CACA,MAAM,OAAO,OAAO,MAAM,SAAS;CACnC,MAAM,UAAsC,EAAE;AAE9C,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,SAAS,KAAK,GAAG,GAAG,IAAI;AAC9B,UAAQ,YAAY,EAAE;AACtB,UAAQ,QAAQ,QAAQ,KAAK;;CAG/B,MAAM,UAAoC,EAAE;CAC5C,MAAM,qBAAyC,EAAE;CAEjD,MAAM,YAAY,KAAK,OAAO;CAC9B,MAAM,QAA+C,CAAC,GAAG,SAAS,MAAM;AAExE,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAE1B,MAAI,KAAK,SAAS,QAAQ;AACxB,WAAQ,KAAK,KAAK,MAAM;GACxB,MAAM,eAAe,KAAK,MAAM,YAAY;AAC5C,OAAI,gBAAgB,KAAM;AAE1B,sBAAmB,gBAAgB;IACjC,QAAQ,KAAK,KAAK,KAAK,SAAS;IAChC,gBAAgB,KAAK,KAAK,SAAS,IAC/B,QAAQ,KAAK,KAAK,GAAG,GAAG,IAAI,YAAY,EAAE,GAC1C,EAAE,EACJ,KAAK,MAAM,EAAE,KAAK,SAAS,CAAC;IAC/B;;AAEH,MAAI,KAAK,SAAS,QAAQ;GACxB,MAAM,SAAS,UAAU,OAAO;GAChC,MAAM,QACJ,UAAU,OACN,KAAK,MAAM,WAAW,UAAU;IAC9B,MAAM,YAAY,MAAM,MAAM,GAAG,EAAE;AACnC,QAAI,CAAC,aAAa,UAAU,SAAS,OAAQ,QAAO;AACpD,WAAO,UAAU,MAAM,YAAY,kBAAkB;KACrD,GACF;GAEN,MAAM,YAAY,KAAK,MAAM,GAAG,MAAM,EAAE,SAAS,EAAE;AACnD,SAAM,KAAK,GAAG,UAAU;;;AAI5B,QAAO;EAAE;EAAS;EAAoB;;AAGxC,SAAgB,iBACd,QACA,SACA;CACA,MAAM,EAAE,cAAc,YAAY,UAAU,kBAAkB,WAAW,EAAE,CAAC;CAC5E,MAAM,EAAE,SAAS,aAAa,uBAAuB,cACnD,YACA,OACA,OACD;AAED,QAAO;EACL;EACA;EACA;EACA,YAAY,YAAY,GAAG,GAAG;EAC/B;;AAGH,SAAgB,uBAEd,SASC;CACD,MAAM,gBACJ,QAAQ,cAAc,YAAY,UAClC,QAAQ,iBACP,EAAE;CAEL,MAAM,+BAAe,IAAI,KAAa;AACtC,QAAO,QAAQ,YAAY,cAAc,CAAC,KAAK,SAAS,QAAQ;EAC9D,MAAM,YAAY,QAAQ,MAAM;EAGhC,MAAM,iBAAiB,SAAS,QAAQ,WAAW,EAAE,GAAG,UACtD,QACG,YAAY,MAAM,OAAO,CACzB,KAAK,GAAG,QAAQ,EAAE,MAAM,IAAI,CAC5B,SAAS,UAAU,CACvB;EAED,MAAM,eAAe,gBAAgB,YAAY;EACjD,IAAI,SACF,gBAAgB,OACZ,QAAQ,cAAc,mBAAmB,gBACzC;AACN,MAAI,CAAC,QAAQ,QAAQ,OAAQ,UAAS;EAGtC,MAAM,eAAe,QAAQ,eAAe,KAAK,EAAE,CAAC,KAAK,IAAI;AAC7D,MAAI,cAAc;AAChB,OAAI,aAAa,IAAI,aAAa,CAAE,UAAS;AAC7C,gBAAa,IAAI,aAAa;;AAGhC,SAAO;GACL,WAAW,UAAU,UAAU;GAC/B;GAEA,QAAQ,QAAQ;GAChB,eAAe,QAAQ;GACxB;GACD"}

package/dist/ui/index.cjs

@@ -0,0 +1,30 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_errors = require('./errors.cjs');
+const require_messages = require('./messages.cjs');
+const require_subagents = require('./subagents.cjs');
+const require_manager = require('./manager.cjs');
+const require_utils = require('./utils.cjs');
+const require_branching = require('./branching.cjs');
+const require_interrupts = require('./interrupts.cjs');
+const require_transport = require('./transport.cjs');
+const require_queue = require('./queue.cjs');
+
+exports.FetchStreamTransport = require_transport.FetchStreamTransport;
+exports.MessageTupleManager = require_messages.MessageTupleManager;
+exports.PendingRunsTracker = require_queue.PendingRunsTracker;
+exports.StreamError = require_errors.StreamError;
+exports.StreamManager = require_manager.StreamManager;
+exports.SubagentManager = require_subagents.SubagentManager;
+exports.calculateDepthFromNamespace = require_subagents.calculateDepthFromNamespace;
+exports.ensureMessageInstances = require_messages.ensureMessageInstances;
+exports.extractInterrupts = require_interrupts.extractInterrupts;
+exports.extractParentIdFromNamespace = require_subagents.extractParentIdFromNamespace;
+exports.extractToolCallIdFromNamespace = require_subagents.extractToolCallIdFromNamespace;
+exports.filterStream = require_utils.filterStream;
+exports.findLast = require_utils.findLast;
+exports.getBranchContext = require_branching.getBranchContext;
+exports.getMessagesMetadataMap = require_branching.getMessagesMetadataMap;
+exports.isSubagentNamespace = require_subagents.isSubagentNamespace;
+exports.toMessageClass = require_messages.toMessageClass;
+exports.toMessageDict = require_messages.toMessageDict;
+exports.unique = require_utils.unique;

package/dist/ui/index.d.cts

@@ -0,0 +1,15 @@
+import { AcceptBaseMessages, AgentMiddlewareLike, AgentTypeConfigLike, AnyStreamCustomOptions, AnyStreamOptions, BaseSubagentState, CompiledSubAgentLike, CustomSubmitOptions, DeepAgentTypeConfigLike, DefaultSubagentStates, ExtractAgentConfig, ExtractDeepAgentConfig, ExtractSubAgentMiddleware, ExtractToolCallsFromState, GetConfigurableType, GetCustomEventType, GetInterruptType, GetToolCallsType, GetUpdateType, InferAgentState, InferAgentToolCalls, InferDeepAgentSubagents, InferMiddlewareStatesFromArray, InferSubagentByName, InferSubagentNames, InferSubagentState, IsAgentLike, IsDeepAgentLike, MessageMetadata, RunCallbackMeta, StreamBase, SubAgentLike, SubagentApi, SubagentStateMap, SubagentStatus, SubagentStream, SubagentStreamInterface, SubagentToolCall, SubmitOptions, UseStreamCustomOptions, UseStreamOptions, UseStreamThread, UseStreamTransport, UseStreamTransportPayload } from "./types.cjs";
+import { Sequence, getBranchContext, getMessagesMetadataMap } from "./branching.cjs";
+import { PendingRunsTracker, QueueEntry, QueueInterface } from "./queue.cjs";
+import { BaseStream } from "./stream/base.cjs";
+import { UseAgentStream, UseAgentStreamOptions } from "./stream/agent.cjs";
+import { UseDeepAgentStream, UseDeepAgentStreamOptions } from "./stream/deep-agent.cjs";
+import { InferBag, InferNodeNames, InferNodeReturnTypes, InferStateType, InferSubagentStates, InferToolCalls, ResolveStreamInterface, ResolveStreamOptions } from "./stream/index.cjs";
+import { StreamError } from "./errors.cjs";
+import { MessageTupleManager, ensureMessageInstances, toMessageClass, toMessageDict } from "./messages.cjs";
+import { EventStreamEvent, StreamManager } from "./manager.cjs";
+import { extractInterrupts } from "./interrupts.cjs";
+import { FetchStreamTransport } from "./transport.cjs";
+import { filterStream, findLast, unique } from "./utils.cjs";
+import { SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace } from "./subagents.cjs";
+export { AcceptBaseMessages, AgentMiddlewareLike, AgentTypeConfigLike, AnyStreamCustomOptions, AnyStreamOptions, BaseStream, BaseSubagentState, CompiledSubAgentLike, CustomSubmitOptions, DeepAgentTypeConfigLike, DefaultSubagentStates, type EventStreamEvent, ExtractAgentConfig, ExtractDeepAgentConfig, ExtractSubAgentMiddleware, ExtractToolCallsFromState, FetchStreamTransport, GetConfigurableType, GetCustomEventType, GetInterruptType, GetToolCallsType, GetUpdateType, InferAgentState, InferAgentToolCalls, InferBag, InferDeepAgentSubagents, InferMiddlewareStatesFromArray, InferNodeNames, InferNodeReturnTypes, InferStateType, InferSubagentByName, InferSubagentNames, InferSubagentState, InferSubagentStates, InferToolCalls, IsAgentLike, IsDeepAgentLike, MessageMetadata, MessageTupleManager, PendingRunsTracker, type QueueEntry, type QueueInterface, ResolveStreamInterface, ResolveStreamOptions, RunCallbackMeta, type Sequence, StreamBase, StreamError, StreamManager, SubAgentLike, SubagentApi, SubagentManager, SubagentStateMap, SubagentStatus, SubagentStream, SubagentStreamInterface, SubagentToolCall, SubmitOptions, UseAgentStream, UseAgentStreamOptions, UseDeepAgentStream, UseDeepAgentStreamOptions, UseStreamCustomOptions, UseStreamOptions, UseStreamThread, UseStreamTransport, UseStreamTransportPayload, calculateDepthFromNamespace, ensureMessageInstances, extractInterrupts, extractParentIdFromNamespace, extractToolCallIdFromNamespace, filterStream, findLast, getBranchContext, getMessagesMetadataMap, isSubagentNamespace, toMessageClass, toMessageDict, unique };

package/dist/ui/index.d.ts

@@ -0,0 +1,15 @@
+import { AcceptBaseMessages, AgentMiddlewareLike, AgentTypeConfigLike, AnyStreamCustomOptions, AnyStreamOptions, BaseSubagentState, CompiledSubAgentLike, CustomSubmitOptions, DeepAgentTypeConfigLike, DefaultSubagentStates, ExtractAgentConfig, ExtractDeepAgentConfig, ExtractSubAgentMiddleware, ExtractToolCallsFromState, GetConfigurableType, GetCustomEventType, GetInterruptType, GetToolCallsType, GetUpdateType, InferAgentState, InferAgentToolCalls, InferDeepAgentSubagents, InferMiddlewareStatesFromArray, InferSubagentByName, InferSubagentNames, InferSubagentState, IsAgentLike, IsDeepAgentLike, MessageMetadata, RunCallbackMeta, StreamBase, SubAgentLike, SubagentApi, SubagentStateMap, SubagentStatus, SubagentStream, SubagentStreamInterface, SubagentToolCall, SubmitOptions, UseStreamCustomOptions, UseStreamOptions, UseStreamThread, UseStreamTransport, UseStreamTransportPayload } from "./types.js";
+import { Sequence, getBranchContext, getMessagesMetadataMap } from "./branching.js";
+import { PendingRunsTracker, QueueEntry, QueueInterface } from "./queue.js";
+import { BaseStream } from "./stream/base.js";
+import { UseAgentStream, UseAgentStreamOptions } from "./stream/agent.js";
+import { UseDeepAgentStream, UseDeepAgentStreamOptions } from "./stream/deep-agent.js";
+import { InferBag, InferNodeNames, InferNodeReturnTypes, InferStateType, InferSubagentStates, InferToolCalls, ResolveStreamInterface, ResolveStreamOptions } from "./stream/index.js";
+import { StreamError } from "./errors.js";
+import { MessageTupleManager, ensureMessageInstances, toMessageClass, toMessageDict } from "./messages.js";
+import { EventStreamEvent, StreamManager } from "./manager.js";
+import { extractInterrupts } from "./interrupts.js";
+import { FetchStreamTransport } from "./transport.js";
+import { filterStream, findLast, unique } from "./utils.js";
+import { SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace } from "./subagents.js";
+export { AcceptBaseMessages, AgentMiddlewareLike, AgentTypeConfigLike, AnyStreamCustomOptions, AnyStreamOptions, BaseStream, BaseSubagentState, CompiledSubAgentLike, CustomSubmitOptions, DeepAgentTypeConfigLike, DefaultSubagentStates, type EventStreamEvent, ExtractAgentConfig, ExtractDeepAgentConfig, ExtractSubAgentMiddleware, ExtractToolCallsFromState, FetchStreamTransport, GetConfigurableType, GetCustomEventType, GetInterruptType, GetToolCallsType, GetUpdateType, InferAgentState, InferAgentToolCalls, InferBag, InferDeepAgentSubagents, InferMiddlewareStatesFromArray, InferNodeNames, InferNodeReturnTypes, InferStateType, InferSubagentByName, InferSubagentNames, InferSubagentState, InferSubagentStates, InferToolCalls, IsAgentLike, IsDeepAgentLike, MessageMetadata, MessageTupleManager, PendingRunsTracker, type QueueEntry, type QueueInterface, ResolveStreamInterface, ResolveStreamOptions, RunCallbackMeta, type Sequence, StreamBase, StreamError, StreamManager, SubAgentLike, SubagentApi, SubagentManager, SubagentStateMap, SubagentStatus, SubagentStream, SubagentStreamInterface, SubagentToolCall, SubmitOptions, UseAgentStream, UseAgentStreamOptions, UseDeepAgentStream, UseDeepAgentStreamOptions, UseStreamCustomOptions, UseStreamOptions, UseStreamThread, UseStreamTransport, UseStreamTransportPayload, calculateDepthFromNamespace, ensureMessageInstances, extractInterrupts, extractParentIdFromNamespace, extractToolCallIdFromNamespace, filterStream, findLast, getBranchContext, getMessagesMetadataMap, isSubagentNamespace, toMessageClass, toMessageDict, unique };

package/dist/ui/index.js

@@ -0,0 +1,11 @@
+import { StreamError } from "./errors.js";
+import { MessageTupleManager, ensureMessageInstances, toMessageClass, toMessageDict } from "./messages.js";
+import { SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace } from "./subagents.js";
+import { StreamManager } from "./manager.js";
+import { filterStream, findLast, unique } from "./utils.js";
+import { getBranchContext, getMessagesMetadataMap } from "./branching.js";
+import { extractInterrupts } from "./interrupts.js";
+import { FetchStreamTransport } from "./transport.js";
+import { PendingRunsTracker } from "./queue.js";
+
+export { FetchStreamTransport, MessageTupleManager, PendingRunsTracker, StreamError, StreamManager, SubagentManager, calculateDepthFromNamespace, ensureMessageInstances, extractInterrupts, extractParentIdFromNamespace, extractToolCallIdFromNamespace, filterStream, findLast, getBranchContext, getMessagesMetadataMap, isSubagentNamespace, toMessageClass, toMessageDict, unique };

package/dist/ui/interrupts.cjs

@@ -0,0 +1,21 @@
+
+//#region src/ui/interrupts.ts
+function extractInterrupts(values, options) {
+	if (typeof values === "object" && values != null && "__interrupt__" in values && Array.isArray(values.__interrupt__)) {
+		const valueInterrupts = values.__interrupt__;
+		if (valueInterrupts.length === 0) return { when: "breakpoint" };
+		if (valueInterrupts.length === 1) return valueInterrupts[0];
+		return valueInterrupts;
+	}
+	if (options?.isLoading) return void 0;
+	const interrupts = options?.threadState?.tasks?.at(-1)?.interrupts;
+	if (interrupts == null || interrupts.length === 0) {
+		if (!(options?.threadState?.next ?? []).length || options?.error != null) return void 0;
+		return { when: "breakpoint" };
+	}
+	return interrupts.at(-1);
+}
+
+//#endregion
+exports.extractInterrupts = extractInterrupts;
+//# sourceMappingURL=interrupts.cjs.map

package/dist/ui/interrupts.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"interrupts.cjs","names":[],"sources":["../../src/ui/interrupts.ts"],"sourcesContent":["import { Interrupt, ThreadState } from \"../schema.js\";\n\nexport function extractInterrupts<InterruptType = unknown>(\n  values: unknown,\n  options?: {\n    isLoading: boolean;\n    threadState: ThreadState | undefined;\n    error: unknown;\n  }\n): Interrupt<InterruptType> | undefined {\n  if (\n    typeof values === \"object\" &&\n    values != null &&\n    \"__interrupt__\" in values &&\n    Array.isArray(values.__interrupt__)\n  ) {\n    const valueInterrupts = values.__interrupt__;\n    if (valueInterrupts.length === 0) return { when: \"breakpoint\" };\n    if (valueInterrupts.length === 1) return valueInterrupts[0];\n\n    // TODO: fix the typing of interrupts if multiple interrupts are returned\n    return valueInterrupts as unknown as Interrupt<InterruptType> | undefined;\n  }\n\n  // If we're deferring to old interrupt detection logic, don't show the interrupt if the stream is loading\n  if (options?.isLoading) return undefined;\n\n  const interrupts = options?.threadState?.tasks?.at(-1)?.interrupts;\n  if (interrupts == null || interrupts.length === 0) {\n    // check if there's a next task present\n    const next = options?.threadState?.next ?? [];\n    if (!next.length || options?.error != null) return undefined;\n    return { when: \"breakpoint\" };\n  }\n\n  // Return only the current interrupt\n  return interrupts.at(-1) as Interrupt<InterruptType> | undefined;\n}\n"],"mappings":";;AAEA,SAAgB,kBACd,QACA,SAKsC;AACtC,KACE,OAAO,WAAW,YAClB,UAAU,QACV,mBAAmB,UACnB,MAAM,QAAQ,OAAO,cAAc,EACnC;EACA,MAAM,kBAAkB,OAAO;AAC/B,MAAI,gBAAgB,WAAW,EAAG,QAAO,EAAE,MAAM,cAAc;AAC/D,MAAI,gBAAgB,WAAW,EAAG,QAAO,gBAAgB;AAGzD,SAAO;;AAIT,KAAI,SAAS,UAAW,QAAO;CAE/B,MAAM,aAAa,SAAS,aAAa,OAAO,GAAG,GAAG,EAAE;AACxD,KAAI,cAAc,QAAQ,WAAW,WAAW,GAAG;AAGjD,MAAI,EADS,SAAS,aAAa,QAAQ,EAAE,EACnC,UAAU,SAAS,SAAS,KAAM,QAAO;AACnD,SAAO,EAAE,MAAM,cAAc;;AAI/B,QAAO,WAAW,GAAG,GAAG"}

package/dist/ui/interrupts.d.cts

@@ -0,0 +1,11 @@
+import { Interrupt, ThreadState } from "../schema.cjs";
+
+//#region src/ui/interrupts.d.ts
+declare function extractInterrupts<InterruptType = unknown>(values: unknown, options?: {
+  isLoading: boolean;
+  threadState: ThreadState | undefined;
+  error: unknown;
+}): Interrupt<InterruptType> | undefined;
+//#endregion
+export { extractInterrupts };
+//# sourceMappingURL=interrupts.d.cts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"interrupts.d.cts","names":[],"sources":["../../src/ui/interrupts.ts"],"mappings":";;;iBAEgB,iBAAA,yBAAA,CACd,MAAA,WACA,OAAA;EACE,SAAA;EACA,WAAA,EAAa,WAAA;EACb,KAAA;AAAA,IAED,SAAA,CAAU,aAAA"}

package/dist/ui/interrupts.d.ts

@@ -0,0 +1,11 @@
+import { Interrupt, ThreadState } from "../schema.js";
+
+//#region src/ui/interrupts.d.ts
+declare function extractInterrupts<InterruptType = unknown>(values: unknown, options?: {
+  isLoading: boolean;
+  threadState: ThreadState | undefined;
+  error: unknown;
+}): Interrupt<InterruptType> | undefined;
+//#endregion
+export { extractInterrupts };
+//# sourceMappingURL=interrupts.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"interrupts.d.ts","names":[],"sources":["../../src/ui/interrupts.ts"],"mappings":";;;iBAEgB,iBAAA,yBAAA,CACd,MAAA,WACA,OAAA;EACE,SAAA;EACA,WAAA,EAAa,WAAA;EACb,KAAA;AAAA,IAED,SAAA,CAAU,aAAA"}

package/dist/ui/interrupts.js

@@ -0,0 +1,20 @@
+//#region src/ui/interrupts.ts
+function extractInterrupts(values, options) {
+	if (typeof values === "object" && values != null && "__interrupt__" in values && Array.isArray(values.__interrupt__)) {
+		const valueInterrupts = values.__interrupt__;
+		if (valueInterrupts.length === 0) return { when: "breakpoint" };
+		if (valueInterrupts.length === 1) return valueInterrupts[0];
+		return valueInterrupts;
+	}
+	if (options?.isLoading) return void 0;
+	const interrupts = options?.threadState?.tasks?.at(-1)?.interrupts;
+	if (interrupts == null || interrupts.length === 0) {
+		if (!(options?.threadState?.next ?? []).length || options?.error != null) return void 0;
+		return { when: "breakpoint" };
+	}
+	return interrupts.at(-1);
+}
+
+//#endregion
+export { extractInterrupts };
+//# sourceMappingURL=interrupts.js.map

package/dist/ui/interrupts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"interrupts.js","names":[],"sources":["../../src/ui/interrupts.ts"],"sourcesContent":["import { Interrupt, ThreadState } from \"../schema.js\";\n\nexport function extractInterrupts<InterruptType = unknown>(\n  values: unknown,\n  options?: {\n    isLoading: boolean;\n    threadState: ThreadState | undefined;\n    error: unknown;\n  }\n): Interrupt<InterruptType> | undefined {\n  if (\n    typeof values === \"object\" &&\n    values != null &&\n    \"__interrupt__\" in values &&\n    Array.isArray(values.__interrupt__)\n  ) {\n    const valueInterrupts = values.__interrupt__;\n    if (valueInterrupts.length === 0) return { when: \"breakpoint\" };\n    if (valueInterrupts.length === 1) return valueInterrupts[0];\n\n    // TODO: fix the typing of interrupts if multiple interrupts are returned\n    return valueInterrupts as unknown as Interrupt<InterruptType> | undefined;\n  }\n\n  // If we're deferring to old interrupt detection logic, don't show the interrupt if the stream is loading\n  if (options?.isLoading) return undefined;\n\n  const interrupts = options?.threadState?.tasks?.at(-1)?.interrupts;\n  if (interrupts == null || interrupts.length === 0) {\n    // check if there's a next task present\n    const next = options?.threadState?.next ?? [];\n    if (!next.length || options?.error != null) return undefined;\n    return { when: \"breakpoint\" };\n  }\n\n  // Return only the current interrupt\n  return interrupts.at(-1) as Interrupt<InterruptType> | undefined;\n}\n"],"mappings":";AAEA,SAAgB,kBACd,QACA,SAKsC;AACtC,KACE,OAAO,WAAW,YAClB,UAAU,QACV,mBAAmB,UACnB,MAAM,QAAQ,OAAO,cAAc,EACnC;EACA,MAAM,kBAAkB,OAAO;AAC/B,MAAI,gBAAgB,WAAW,EAAG,QAAO,EAAE,MAAM,cAAc;AAC/D,MAAI,gBAAgB,WAAW,EAAG,QAAO,gBAAgB;AAGzD,SAAO;;AAIT,KAAI,SAAS,UAAW,QAAO;CAE/B,MAAM,aAAa,SAAS,aAAa,OAAO,GAAG,GAAG,EAAE;AACxD,KAAI,cAAc,QAAQ,WAAW,WAAW,GAAG;AAGjD,MAAI,EADS,SAAS,aAAa,QAAQ,EAAE,EACnC,UAAU,SAAS,SAAS,KAAM,QAAO;AACnD,SAAO,EAAE,MAAM,cAAc;;AAI/B,QAAO,WAAW,GAAG,GAAG"}

package/dist/ui/manager.cjs

@@ -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 ?? require_messages.toMessageDict;
 		this.subagentManager = new require_subagents.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 && require_subagents.isSubagentNamespace(namespace)) {
 					const namespaceId = require_subagents.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 = require_messages.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();

package/dist/ui/manager.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"manager.cjs","names":["SubagentManager","StreamError","isSubagentNamespace","extractToolCallIdFromNamespace","toMessageDict"],"sources":["../../src/ui/manager.ts"],"sourcesContent":["import type {\n  CheckpointsStreamEvent,\n  CustomStreamEvent,\n  DebugStreamEvent,\n  ErrorStreamEvent,\n  EventsStreamEvent,\n  FeedbackStreamEvent,\n  MessagesTupleStreamEvent,\n  MetadataStreamEvent,\n  TasksStreamEvent,\n  UpdatesStreamEvent,\n  ValuesStreamEvent,\n} from \"../types.stream.js\";\nimport { MessageTupleManager, toMessageDict } from \"./messages.js\";\nimport { StreamError } from \"./errors.js\";\nimport type { Message } from \"../types.messages.js\";\nimport type { BagTemplate } from \"../types.template.js\";\nimport {\n  SubagentManager,\n  extractToolCallIdFromNamespace,\n  isSubagentNamespace,\n} from \"./subagents.js\";\nimport type { SubagentStreamInterface } from \"./types.js\";\n\n/**\n * Special ID used by LangGraph's messagesStateReducer to signal\n * that all messages should be removed from the state.\n */\nexport const REMOVE_ALL_MESSAGES = \"__remove_all__\";\n\ntype GetUpdateType<\n  Bag extends BagTemplate,\n  StateType extends Record<string, unknown>\n> = Bag extends { UpdateType: unknown }\n  ? Bag[\"UpdateType\"]\n  : Partial<StateType>;\n\ntype GetCustomEventType<Bag extends BagTemplate> = Bag extends {\n  CustomEventType: unknown;\n}\n  ? Bag[\"CustomEventType\"]\n  : unknown;\n\ntype EventStreamMap<StateType, UpdateType, CustomType> = {\n  values: ValuesStreamEvent<StateType>;\n  updates: UpdatesStreamEvent<UpdateType>;\n  custom: CustomStreamEvent<CustomType>;\n  debug: DebugStreamEvent;\n  messages: MessagesTupleStreamEvent;\n  events: EventsStreamEvent;\n  metadata: MetadataStreamEvent;\n  checkpoints: CheckpointsStreamEvent<StateType>;\n  tasks: TasksStreamEvent<StateType, UpdateType>;\n  error: ErrorStreamEvent;\n  feedback: FeedbackStreamEvent;\n};\n\nexport type EventStreamEvent<StateType, UpdateType, CustomType> =\n  EventStreamMap<StateType, UpdateType, CustomType>[keyof EventStreamMap<\n    StateType,\n    UpdateType,\n    CustomType\n  >];\n\ninterface StreamManagerEventCallbacks<\n  StateType extends Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate\n> {\n  onUpdateEvent?: (\n    data: UpdatesStreamEvent<GetUpdateType<Bag, StateType>>[\"data\"],\n    options: {\n      namespace: string[] | undefined;\n      mutate: (\n        update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n      ) => void;\n    }\n  ) => void;\n  onCustomEvent?: (\n    data: GetCustomEventType<Bag>,\n    options: {\n      namespace: string[] | undefined;\n      mutate: (\n        update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n      ) => void;\n    }\n  ) => void;\n  onMetadataEvent?: (data: MetadataStreamEvent[\"data\"]) => void;\n  onLangChainEvent?: (data: EventsStreamEvent[\"data\"]) => void;\n  onDebugEvent?: (\n    data: DebugStreamEvent[\"data\"],\n    options: { namespace: string[] | undefined }\n  ) => void;\n  onCheckpointEvent?: (\n    data: CheckpointsStreamEvent<StateType>[\"data\"],\n    options: { namespace: string[] | undefined }\n  ) => void;\n  onTaskEvent?: (\n    data: TasksStreamEvent<StateType, GetUpdateType<Bag, StateType>>[\"data\"],\n    options: { namespace: string[] | undefined }\n  ) => void;\n}\n\n/**\n * Options for StreamManager constructor.\n */\nexport interface StreamManagerOptions {\n  /**\n   * Throttle the stream updates.\n   * If a number is provided, updates are throttled to the given milliseconds.\n   * If `true`, updates are batched in a single macrotask.\n   * If `false`, updates are not throttled.\n   */\n  throttle: number | boolean;\n\n  /**\n   * Tool names that indicate subagent invocation.\n   *\n   * When an AI message contains tool calls with these names, they are\n   * automatically tracked as subagent executions. This enables the\n   * `subagents`, `activeSubagents`, `getSubagent()`, and `getSubagentsByType()`\n   * properties on the stream.\n   *\n   * @default [\"task\"]\n   *\n   * @example\n   * ```typescript\n   * // Track both \"task\" and \"delegate\" as subagent tools\n   * subagentToolNames: [\"task\", \"delegate\", \"spawn_agent\"]\n   * ```\n   */\n  subagentToolNames?: string[];\n\n  /**\n   * Filter out messages from subagent streams in the main messages array.\n   *\n   * When enabled, messages from subagraph executions (those with a `tools:` namespace)\n   * are excluded from `stream.messages`. Instead, these messages are tracked\n   * per-subagent and accessible via `stream.subagents.get(id).messages`.\n   *\n   * This is useful for deep agent architectures where you want to display\n   * the main conversation separately from subagent activity.\n   *\n   * @default false\n   *\n   * @example\n   * ```typescript\n   * const stream = useStream({\n   *   assistantId: \"my-agent\",\n   *   filterSubagentMessages: true,\n   * });\n   *\n   * // Main thread messages only (no subagent messages)\n   * stream.messages\n   *\n   * // Access subagent messages individually\n   * stream.subagents.get(\"call_xyz\").messages\n   * ```\n   */\n  filterSubagentMessages?: boolean;\n}\n\nexport class StreamManager<\n  StateType extends Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate\n> {\n  private abortRef = new AbortController();\n\n  private messages: MessageTupleManager;\n\n  private subagentManager: SubagentManager;\n\n  private listeners = new Set<() => void>();\n\n  private throttle: number | boolean;\n\n  private filterSubagentMessages: boolean;\n\n  private queue: Promise<unknown> = Promise.resolve();\n\n  private queueSize: number = 0;\n\n  private state: {\n    isLoading: boolean;\n    values: [values: StateType, kind: \"stream\" | \"stop\"] | null;\n    error: unknown;\n    /** Version counter to force React re-renders on subagent changes */\n    version: number;\n  };\n\n  constructor(messages: MessageTupleManager, options: StreamManagerOptions) {\n    this.messages = messages;\n    this.state = {\n      isLoading: false,\n      values: null,\n      error: undefined,\n      version: 0,\n    };\n    this.throttle = options.throttle;\n    this.filterSubagentMessages = options.filterSubagentMessages ?? false;\n    this.subagentManager = new SubagentManager({\n      subagentToolNames: options.subagentToolNames,\n      onSubagentChange: () => this.bumpVersion(),\n    });\n  }\n\n  /**\n   * Increment version counter to trigger React re-renders.\n   * Called when subagent state changes.\n   */\n  private bumpVersion = () => {\n    this.state = { ...this.state, version: this.state.version + 1 };\n    this.notifyListeners();\n  };\n\n  /**\n   * Get all subagents as a Map.\n   */\n  getSubagents(): Map<string, SubagentStreamInterface> {\n    return this.subagentManager.getSubagents();\n  }\n\n  /**\n   * Get all currently running subagents.\n   */\n  getActiveSubagents(): SubagentStreamInterface[] {\n    return this.subagentManager.getActiveSubagents();\n  }\n\n  /**\n   * Get a specific subagent by tool call ID.\n   */\n  getSubagent(toolCallId: string): SubagentStreamInterface | undefined {\n    return this.subagentManager.getSubagent(toolCallId);\n  }\n\n  /**\n   * Get all subagents of a specific type.\n   */\n  getSubagentsByType(type: string): SubagentStreamInterface[] {\n    return this.subagentManager.getSubagentsByType(type);\n  }\n\n  /**\n   * Get all subagents triggered by a specific AI message.\n   */\n  getSubagentsByMessage(messageId: string): SubagentStreamInterface[] {\n    return this.subagentManager.getSubagentsByMessage(messageId);\n  }\n\n  /**\n   * Reconstruct subagent state from historical messages.\n   *\n   * This method should be called when loading thread history to restore\n   * subagent visualization after:\n   * - Page refresh (when stream has already completed)\n   * - Loading thread history\n   * - Navigating between threads\n   *\n   * @param messages - Array of messages from thread history\n   * @param options - Optional configuration\n   * @param options.skipIfPopulated - If true, skip reconstruction if subagents already exist\n   */\n  reconstructSubagents(\n    messages: Message[],\n    options?: { skipIfPopulated?: boolean }\n  ): void {\n    this.subagentManager.reconstructFromMessages(messages, options);\n  }\n\n  /**\n   * Check if any subagents are currently tracked.\n   */\n  hasSubagents(): boolean {\n    return this.subagentManager.hasSubagents();\n  }\n\n  private setState = (newState: Partial<typeof this.state>) => {\n    this.state = { ...this.state, ...newState };\n    this.notifyListeners();\n  };\n\n  private notifyListeners = () => {\n    this.listeners.forEach((listener) => listener());\n  };\n\n  subscribe = (listener: () => void): (() => void) => {\n    if (this.throttle === false) {\n      this.listeners.add(listener);\n      return () => this.listeners.delete(listener);\n    }\n\n    const timeoutMs = this.throttle === true ? 0 : this.throttle;\n    let timeoutId: NodeJS.Timeout | number | undefined;\n\n    const throttledListener = () => {\n      clearTimeout(timeoutId);\n      timeoutId = setTimeout(() => {\n        clearTimeout(timeoutId);\n        listener();\n      }, timeoutMs);\n    };\n\n    this.listeners.add(throttledListener);\n    return () => {\n      clearTimeout(timeoutId);\n      this.listeners.delete(throttledListener);\n    };\n  };\n\n  getSnapshot = () => this.state;\n\n  get isLoading() {\n    return this.state.isLoading;\n  }\n\n  get values() {\n    return this.state.values?.[0] ?? null;\n  }\n\n  get error() {\n    return this.state.error;\n  }\n\n  setStreamValues = (\n    values:\n      | (StateType | null)\n      | ((prev: StateType | null, kind: \"stream\" | \"stop\") => StateType | null),\n    kind: \"stream\" | \"stop\" = \"stream\"\n  ) => {\n    if (typeof values === \"function\") {\n      const [prevValues, prevKind] = this.state.values ?? [null, \"stream\"];\n      const nextValues = values(prevValues, prevKind);\n      this.setState({ values: nextValues != null ? [nextValues, kind] : null });\n    } else {\n      const nextValues = values != null ? [values, kind] : null;\n      this.setState({ values: nextValues as [StateType, \"stream\" | \"stop\"] });\n    }\n  };\n\n  private getMutateFn = (kind: \"stream\" | \"stop\", historyValues: StateType) => {\n    return (\n      update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n    ) => {\n      const stateValues = (this.state.values ?? [null, \"stream\"])[0];\n      const prev = {\n        ...historyValues,\n        ...stateValues,\n      };\n      const next = typeof update === \"function\" ? update(prev) : update;\n      this.setStreamValues({ ...prev, ...next }, kind);\n    };\n  };\n\n  private matchEventType = <\n    T extends keyof EventStreamMap<\n      StateType,\n      GetUpdateType<Bag, StateType>,\n      GetCustomEventType<Bag>\n    >\n  >(\n    expected: T,\n    actual: EventStreamEvent<\n      StateType,\n      GetUpdateType<Bag, StateType>,\n      GetCustomEventType<Bag>\n    >[\"event\"],\n    _data: EventStreamEvent<\n      StateType,\n      GetUpdateType<Bag, StateType>,\n      GetCustomEventType<Bag>\n    >[\"data\"]\n  ): _data is EventStreamMap<\n    StateType,\n    GetUpdateType<Bag, StateType>,\n    GetCustomEventType<Bag>\n  >[T][\"data\"] => {\n    return expected === actual || actual.startsWith(`${expected}|`);\n  };\n\n  protected enqueue = async (\n    action: (\n      signal: AbortSignal\n    ) => Promise<\n      AsyncGenerator<\n        EventStreamEvent<\n          StateType,\n          GetUpdateType<Bag, StateType>,\n          GetCustomEventType<Bag>\n        >\n      >\n    >,\n    options: {\n      getMessages: (values: StateType) => Message[];\n\n      setMessages: (current: StateType, messages: Message[]) => StateType;\n\n      initialValues: StateType;\n\n      callbacks: StreamManagerEventCallbacks<StateType, Bag>;\n\n      onSuccess: () =>\n        | StateType\n        | null\n        | undefined\n        | void\n        | Promise<StateType | null | undefined | void>;\n\n      onError: (error: unknown) => void | Promise<void>;\n\n      onFinish?: () => void;\n    }\n  ) => {\n    try {\n      this.queueSize = Math.max(0, this.queueSize - 1);\n      this.setState({ isLoading: true, error: undefined });\n      this.abortRef = new AbortController();\n\n      const run = await action(this.abortRef.signal);\n\n      let streamError: StreamError | undefined;\n      for await (const { event, data } of run) {\n        if (event === \"error\") {\n          streamError = new StreamError(data);\n          break;\n        }\n\n        const namespace = event.includes(\"|\")\n          ? event.split(\"|\").slice(1)\n          : undefined;\n\n        const mutate = this.getMutateFn(\"stream\", options.initialValues);\n\n        if (event === \"metadata\") options.callbacks.onMetadataEvent?.(data);\n        if (event === \"events\") options.callbacks.onLangChainEvent?.(data);\n\n        if (this.matchEventType(\"updates\", event, data)) {\n          options.callbacks.onUpdateEvent?.(data, { namespace, mutate });\n\n          // Track subagent streaming updates from subgraph namespaces\n          // Mark the subagent as running when we receive updates\n          // The actual message content is handled via addMessageToSubagent\n          if (namespace && isSubagentNamespace(namespace)) {\n            const namespaceId = extractToolCallIdFromNamespace(namespace);\n            if (namespaceId && this.filterSubagentMessages) {\n              this.subagentManager.markRunningFromNamespace(\n                namespaceId,\n                namespace\n              );\n            }\n          }\n\n          // Also register subagents from main agent updates (tool_calls in messages)\n          // AND process tool results to complete subagents\n          // This is needed because tool_calls often appear complete in updates\n          // before they appear in the messages stream\n          if (!namespace || !isSubagentNamespace(namespace)) {\n            const updateData = data as Record<string, unknown>;\n            for (const nodeData of Object.values(updateData)) {\n              if (\n                nodeData &&\n                typeof nodeData === \"object\" &&\n                \"messages\" in nodeData\n              ) {\n                const { messages } = nodeData as { messages: unknown[] };\n                if (Array.isArray(messages)) {\n                  for (const msg of messages) {\n                    if (!msg || typeof msg !== \"object\") continue;\n                    const msgObj = msg as Record<string, unknown>;\n\n                    // Register subagents from AI messages with tool_calls\n                    if (\n                      msgObj.type === \"ai\" &&\n                      \"tool_calls\" in msgObj &&\n                      Array.isArray(msgObj.tool_calls)\n                    ) {\n                      this.subagentManager.registerFromToolCalls(\n                        msgObj.tool_calls as Array<{\n                          id?: string;\n                          name: string;\n                          args: Record<string, unknown> | string;\n                        }>,\n                        msgObj.id as string | undefined\n                      );\n                    }\n\n                    // Complete subagents from tool messages (task results)\n                    if (\n                      msgObj.type === \"tool\" &&\n                      \"tool_call_id\" in msgObj &&\n                      typeof msgObj.tool_call_id === \"string\"\n                    ) {\n                      const content =\n                        typeof msgObj.content === \"string\"\n                          ? msgObj.content\n                          : JSON.stringify(msgObj.content);\n                      const status =\n                        \"status\" in msgObj && msgObj.status === \"error\"\n                          ? \"error\"\n                          : \"success\";\n                      this.subagentManager.processToolMessage(\n                        msgObj.tool_call_id,\n                        content,\n                        status\n                      );\n                    }\n                  }\n                }\n              }\n            }\n          }\n        }\n\n        if (this.matchEventType(\"custom\", event, data)) {\n          options.callbacks.onCustomEvent?.(data, { namespace, mutate });\n        }\n\n        if (this.matchEventType(\"checkpoints\", event, data)) {\n          options.callbacks.onCheckpointEvent?.(data, { namespace });\n        }\n\n        if (this.matchEventType(\"tasks\", event, data)) {\n          options.callbacks.onTaskEvent?.(data, { namespace });\n        }\n\n        if (this.matchEventType(\"debug\", event, data)) {\n          options.callbacks.onDebugEvent?.(data, { namespace });\n        }\n\n        // Handle values events - use startsWith to match both \"values\" and \"values|tools:xxx\"\n        if (event === \"values\" || event.startsWith(\"values|\")) {\n          // Check if this is a subgraph values event (for namespace mapping and values)\n          if (namespace && isSubagentNamespace(namespace)) {\n            const namespaceId = extractToolCallIdFromNamespace(namespace);\n            if (namespaceId && this.filterSubagentMessages) {\n              const valuesData = data as Record<string, unknown>;\n\n              // Try to establish namespace mapping from the initial human message\n              const messages = valuesData.messages as unknown[];\n              if (Array.isArray(messages) && messages.length > 0) {\n                const firstMsg = messages[0] as Record<string, unknown>;\n                if (\n                  firstMsg?.type === \"human\" &&\n                  typeof firstMsg?.content === \"string\"\n                ) {\n                  this.subagentManager.matchSubgraphToSubagent(\n                    namespaceId,\n                    firstMsg.content\n                  );\n                }\n              }\n\n              // Update the subagent's values with the full state\n              this.subagentManager.updateSubagentValues(\n                namespaceId,\n                valuesData\n              );\n            }\n          } else if (\n            data &&\n            typeof data === \"object\" &&\n            \"__interrupt__\" in data\n          ) {\n            const interruptData = data as Partial<StateType>;\n            this.setStreamValues(\n              (prev) => ({ ...prev, ...interruptData } as StateType)\n            );\n          } else {\n            this.setStreamValues(data as StateType);\n          }\n        }\n\n        if (this.matchEventType(\"messages\", event, data)) {\n          const [serialized, metadata] = data;\n\n          // Check if this message is from a subagent namespace\n          const rawCheckpointNs =\n            (metadata?.langgraph_checkpoint_ns as string | undefined) ||\n            (metadata?.checkpoint_ns as string | undefined);\n          const checkpointNs: string | undefined =\n            typeof rawCheckpointNs === \"string\" ? rawCheckpointNs : undefined;\n          const isFromSubagent = isSubagentNamespace(checkpointNs);\n          const toolCallId = isFromSubagent\n            ? extractToolCallIdFromNamespace(checkpointNs?.split(\"|\"))\n            : undefined;\n\n          // If filtering is enabled and this is a subagent message,\n          // add it to the subagent's messages instead of the main stream\n          if (this.filterSubagentMessages && isFromSubagent && toolCallId) {\n            // Add to subagent's message list\n            this.subagentManager.addMessageToSubagent(\n              toolCallId,\n              serialized,\n              metadata\n            );\n            continue;\n          }\n\n          const messageId = this.messages.add(serialized, metadata);\n          if (!messageId) {\n            console.warn(\n              \"Failed to add message to manager, no message ID found\"\n            );\n            continue;\n          }\n\n          this.setStreamValues((streamValues) => {\n            const values = {\n              ...options.initialValues,\n              ...streamValues,\n            };\n\n            // Assumption: we're concatenating the message\n            let messages = options.getMessages(values).slice();\n            const { chunk, index } =\n              this.messages.get(messageId, messages.length) ?? {};\n\n            if (!chunk || index == null) return values;\n            if (chunk.getType() === \"remove\") {\n              // Check for special REMOVE_ALL_MESSAGES sentinel\n              if (chunk.id === REMOVE_ALL_MESSAGES) {\n                // Clear all messages when __remove_all__ is received\n                messages = [];\n              } else {\n                messages.splice(index, 1);\n              }\n            } else {\n              const msgDict = toMessageDict(chunk);\n              messages[index] = msgDict;\n\n              // Track subagents from AI messages with tool calls (main agent only)\n              if (\n                !isFromSubagent &&\n                msgDict.type === \"ai\" &&\n                \"tool_calls\" in msgDict &&\n                Array.isArray(msgDict.tool_calls)\n              ) {\n                this.subagentManager.registerFromToolCalls(\n                  msgDict.tool_calls,\n                  msgDict.id as string | undefined\n                );\n              }\n\n              // Complete subagents when tool messages arrive (main agent only)\n              if (\n                !isFromSubagent &&\n                msgDict.type === \"tool\" &&\n                \"tool_call_id\" in msgDict\n              ) {\n                const tcId = msgDict.tool_call_id as string;\n                const content =\n                  typeof msgDict.content === \"string\"\n                    ? msgDict.content\n                    : JSON.stringify(msgDict.content);\n                const status =\n                  \"status\" in msgDict && msgDict.status === \"error\"\n                    ? \"error\"\n                    : \"success\";\n                this.subagentManager.processToolMessage(tcId, content, status);\n              }\n            }\n\n            return options.setMessages(values, messages);\n          });\n        }\n      }\n\n      if (streamError != null) throw streamError;\n\n      // Skip onSuccess when the stream was aborted (e.g., by multitask interrupt).\n      // This avoids unnecessary HTTP calls (like history fetching) that would\n      // delay the next queued stream from starting.\n      if (!this.abortRef.signal.aborted) {\n        const values = await options.onSuccess?.();\n        if (typeof values !== \"undefined\" && this.queueSize === 0) {\n          this.setStreamValues(values);\n        }\n      }\n    } catch (error) {\n      if (\n        !(\n          error instanceof Error && // eslint-disable-line no-instanceof/no-instanceof\n          (error.name === \"AbortError\" || error.name === \"TimeoutError\")\n        )\n      ) {\n        console.error(error);\n        this.setState({ error });\n        await options.onError?.(error);\n      }\n    } finally {\n      this.setState({ isLoading: false });\n      this.abortRef = new AbortController();\n      options.onFinish?.();\n    }\n  };\n\n  start = async (\n    action: (\n      signal: AbortSignal\n    ) => Promise<\n      AsyncGenerator<\n        EventStreamEvent<\n          StateType,\n          GetUpdateType<Bag, StateType>,\n          GetCustomEventType<Bag>\n        >\n      >\n    >,\n    options: {\n      getMessages: (values: StateType) => Message[];\n\n      setMessages: (current: StateType, messages: Message[]) => StateType;\n\n      initialValues: StateType;\n\n      callbacks: StreamManagerEventCallbacks<StateType, Bag>;\n\n      onSuccess: () =>\n        | StateType\n        | null\n        | undefined\n        | void\n        | Promise<StateType | null | undefined | void>;\n\n      onError: (error: unknown) => void | Promise<void>;\n\n      onFinish?: () => void;\n    },\n    startOptions?: {\n      /**\n       * If true, abort any currently running stream before starting this one.\n       * Used for multitask_strategy: \"interrupt\" and \"rollback\" to unblock\n       * the queue so the new run request can proceed immediately.\n       */\n      abortPrevious?: boolean;\n    }\n  ): Promise<void> => {\n    if (startOptions?.abortPrevious) {\n      this.abortRef.abort();\n    }\n    this.queueSize += 1;\n    this.queue = this.queue.then(() => this.enqueue(action, options));\n  };\n\n  stop = async (\n    historyValues: StateType,\n    options: {\n      onStop?: (options: {\n        mutate: (\n          update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n        ) => void;\n      }) => void;\n    }\n  ): Promise<void> => {\n    this.abortRef.abort();\n    this.abortRef = new AbortController();\n\n    options.onStop?.({ mutate: this.getMutateFn(\"stop\", historyValues) });\n  };\n\n  clear = () => {\n    // Cancel any running streams\n    this.abortRef.abort();\n    this.abortRef = new AbortController();\n\n    // Set the stream state to null\n    this.setState({ error: undefined, values: null, isLoading: false });\n\n    // Clear any pending messages\n    this.messages.clear();\n\n    // Clear subagent state\n    this.subagentManager.clear();\n  };\n}\n"],"mappings":";;;;;;;;;AA4BA,MAAa,sBAAsB;AAqInC,IAAa,gBAAb,MAGE;CACA,AAAQ,WAAW,IAAI,iBAAiB;CAExC,AAAQ;CAER,AAAQ;CAER,AAAQ,4BAAY,IAAI,KAAiB;CAEzC,AAAQ;CAER,AAAQ;CAER,AAAQ,QAA0B,QAAQ,SAAS;CAEnD,AAAQ,YAAoB;CAE5B,AAAQ;CAQR,YAAY,UAA+B,SAA+B;AACxE,OAAK,WAAW;AAChB,OAAK,QAAQ;GACX,WAAW;GACX,QAAQ;GACR,OAAO;GACP,SAAS;GACV;AACD,OAAK,WAAW,QAAQ;AACxB,OAAK,yBAAyB,QAAQ,0BAA0B;AAChE,OAAK,kBAAkB,IAAIA,kCAAgB;GACzC,mBAAmB,QAAQ;GAC3B,wBAAwB,KAAK,aAAa;GAC3C,CAAC;;;;;;CAOJ,AAAQ,oBAAoB;AAC1B,OAAK,QAAQ;GAAE,GAAG,KAAK;GAAO,SAAS,KAAK,MAAM,UAAU;GAAG;AAC/D,OAAK,iBAAiB;;;;;CAMxB,eAAqD;AACnD,SAAO,KAAK,gBAAgB,cAAc;;;;;CAM5C,qBAAgD;AAC9C,SAAO,KAAK,gBAAgB,oBAAoB;;;;;CAMlD,YAAY,YAAyD;AACnE,SAAO,KAAK,gBAAgB,YAAY,WAAW;;;;;CAMrD,mBAAmB,MAAyC;AAC1D,SAAO,KAAK,gBAAgB,mBAAmB,KAAK;;;;;CAMtD,sBAAsB,WAA8C;AAClE,SAAO,KAAK,gBAAgB,sBAAsB,UAAU;;;;;;;;;;;;;;;CAgB9D,qBACE,UACA,SACM;AACN,OAAK,gBAAgB,wBAAwB,UAAU,QAAQ;;;;;CAMjE,eAAwB;AACtB,SAAO,KAAK,gBAAgB,cAAc;;CAG5C,AAAQ,YAAY,aAAyC;AAC3D,OAAK,QAAQ;GAAE,GAAG,KAAK;GAAO,GAAG;GAAU;AAC3C,OAAK,iBAAiB;;CAGxB,AAAQ,wBAAwB;AAC9B,OAAK,UAAU,SAAS,aAAa,UAAU,CAAC;;CAGlD,aAAa,aAAuC;AAClD,MAAI,KAAK,aAAa,OAAO;AAC3B,QAAK,UAAU,IAAI,SAAS;AAC5B,gBAAa,KAAK,UAAU,OAAO,SAAS;;EAG9C,MAAM,YAAY,KAAK,aAAa,OAAO,IAAI,KAAK;EACpD,IAAI;EAEJ,MAAM,0BAA0B;AAC9B,gBAAa,UAAU;AACvB,eAAY,iBAAiB;AAC3B,iBAAa,UAAU;AACvB,cAAU;MACT,UAAU;;AAGf,OAAK,UAAU,IAAI,kBAAkB;AACrC,eAAa;AACX,gBAAa,UAAU;AACvB,QAAK,UAAU,OAAO,kBAAkB;;;CAI5C,oBAAoB,KAAK;CAEzB,IAAI,YAAY;AACd,SAAO,KAAK,MAAM;;CAGpB,IAAI,SAAS;AACX,SAAO,KAAK,MAAM,SAAS,MAAM;;CAGnC,IAAI,QAAQ;AACV,SAAO,KAAK,MAAM;;CAGpB,mBACE,QAGA,OAA0B,aACvB;AACH,MAAI,OAAO,WAAW,YAAY;GAChC,MAAM,CAAC,YAAY,YAAY,KAAK,MAAM,UAAU,CAAC,MAAM,SAAS;GACpE,MAAM,aAAa,OAAO,YAAY,SAAS;AAC/C,QAAK,SAAS,EAAE,QAAQ,cAAc,OAAO,CAAC,YAAY,KAAK,GAAG,MAAM,CAAC;SACpE;GACL,MAAM,aAAa,UAAU,OAAO,CAAC,QAAQ,KAAK,GAAG;AACrD,QAAK,SAAS,EAAE,QAAQ,YAA8C,CAAC;;;CAI3E,AAAQ,eAAe,MAAyB,kBAA6B;AAC3E,UACE,WACG;GACH,MAAM,eAAe,KAAK,MAAM,UAAU,CAAC,MAAM,SAAS,EAAE;GAC5D,MAAM,OAAO;IACX,GAAG;IACH,GAAG;IACJ;GACD,MAAM,OAAO,OAAO,WAAW,aAAa,OAAO,KAAK,GAAG;AAC3D,QAAK,gBAAgB;IAAE,GAAG;IAAM,GAAG;IAAM,EAAE,KAAK;;;CAIpD,AAAQ,kBAON,UACA,QAKA,UASc;AACd,SAAO,aAAa,UAAU,OAAO,WAAW,GAAG,SAAS,GAAG;;CAGjE,AAAU,UAAU,OAClB,QAWA,YAoBG;AACH,MAAI;AACF,QAAK,YAAY,KAAK,IAAI,GAAG,KAAK,YAAY,EAAE;AAChD,QAAK,SAAS;IAAE,WAAW;IAAM,OAAO;IAAW,CAAC;AACpD,QAAK,WAAW,IAAI,iBAAiB;GAErC,MAAM,MAAM,MAAM,OAAO,KAAK,SAAS,OAAO;GAE9C,IAAI;AACJ,cAAW,MAAM,EAAE,OAAO,UAAU,KAAK;AACvC,QAAI,UAAU,SAAS;AACrB,mBAAc,IAAIC,2BAAY,KAAK;AACnC;;IAGF,MAAM,YAAY,MAAM,SAAS,IAAI,GACjC,MAAM,MAAM,IAAI,CAAC,MAAM,EAAE,GACzB;IAEJ,MAAM,SAAS,KAAK,YAAY,UAAU,QAAQ,cAAc;AAEhE,QAAI,UAAU,WAAY,SAAQ,UAAU,kBAAkB,KAAK;AACnE,QAAI,UAAU,SAAU,SAAQ,UAAU,mBAAmB,KAAK;AAElE,QAAI,KAAK,eAAe,WAAW,OAAO,KAAK,EAAE;AAC/C,aAAQ,UAAU,gBAAgB,MAAM;MAAE;MAAW;MAAQ,CAAC;AAK9D,SAAI,aAAaC,sCAAoB,UAAU,EAAE;MAC/C,MAAM,cAAcC,iDAA+B,UAAU;AAC7D,UAAI,eAAe,KAAK,uBACtB,MAAK,gBAAgB,yBACnB,aACA,UACD;;AAQL,SAAI,CAAC,aAAa,CAACD,sCAAoB,UAAU,EAAE;MACjD,MAAM,aAAa;AACnB,WAAK,MAAM,YAAY,OAAO,OAAO,WAAW,CAC9C,KACE,YACA,OAAO,aAAa,YACpB,cAAc,UACd;OACA,MAAM,EAAE,aAAa;AACrB,WAAI,MAAM,QAAQ,SAAS,CACzB,MAAK,MAAM,OAAO,UAAU;AAC1B,YAAI,CAAC,OAAO,OAAO,QAAQ,SAAU;QACrC,MAAM,SAAS;AAGf,YACE,OAAO,SAAS,QAChB,gBAAgB,UAChB,MAAM,QAAQ,OAAO,WAAW,CAEhC,MAAK,gBAAgB,sBACnB,OAAO,YAKP,OAAO,GACR;AAIH,YACE,OAAO,SAAS,UAChB,kBAAkB,UAClB,OAAO,OAAO,iBAAiB,UAC/B;SACA,MAAM,UACJ,OAAO,OAAO,YAAY,WACtB,OAAO,UACP,KAAK,UAAU,OAAO,QAAQ;SACpC,MAAM,SACJ,YAAY,UAAU,OAAO,WAAW,UACpC,UACA;AACN,cAAK,gBAAgB,mBACnB,OAAO,cACP,SACA,OACD;;;;;;AASf,QAAI,KAAK,eAAe,UAAU,OAAO,KAAK,CAC5C,SAAQ,UAAU,gBAAgB,MAAM;KAAE;KAAW;KAAQ,CAAC;AAGhE,QAAI,KAAK,eAAe,eAAe,OAAO,KAAK,CACjD,SAAQ,UAAU,oBAAoB,MAAM,EAAE,WAAW,CAAC;AAG5D,QAAI,KAAK,eAAe,SAAS,OAAO,KAAK,CAC3C,SAAQ,UAAU,cAAc,MAAM,EAAE,WAAW,CAAC;AAGtD,QAAI,KAAK,eAAe,SAAS,OAAO,KAAK,CAC3C,SAAQ,UAAU,eAAe,MAAM,EAAE,WAAW,CAAC;AAIvD,QAAI,UAAU,YAAY,MAAM,WAAW,UAAU,CAEnD,KAAI,aAAaA,sCAAoB,UAAU,EAAE;KAC/C,MAAM,cAAcC,iDAA+B,UAAU;AAC7D,SAAI,eAAe,KAAK,wBAAwB;MAC9C,MAAM,aAAa;MAGnB,MAAM,WAAW,WAAW;AAC5B,UAAI,MAAM,QAAQ,SAAS,IAAI,SAAS,SAAS,GAAG;OAClD,MAAM,WAAW,SAAS;AAC1B,WACE,UAAU,SAAS,WACnB,OAAO,UAAU,YAAY,SAE7B,MAAK,gBAAgB,wBACnB,aACA,SAAS,QACV;;AAKL,WAAK,gBAAgB,qBACnB,aACA,WACD;;eAGH,QACA,OAAO,SAAS,YAChB,mBAAmB,MACnB;KACA,MAAM,gBAAgB;AACtB,UAAK,iBACF,UAAU;MAAE,GAAG;MAAM,GAAG;MAAe,EACzC;UAED,MAAK,gBAAgB,KAAkB;AAI3C,QAAI,KAAK,eAAe,YAAY,OAAO,KAAK,EAAE;KAChD,MAAM,CAAC,YAAY,YAAY;KAG/B,MAAM,kBACH,UAAU,2BACV,UAAU;KACb,MAAM,eACJ,OAAO,oBAAoB,WAAW,kBAAkB;KAC1D,MAAM,iBAAiBD,sCAAoB,aAAa;KACxD,MAAM,aAAa,iBACfC,iDAA+B,cAAc,MAAM,IAAI,CAAC,GACxD;AAIJ,SAAI,KAAK,0BAA0B,kBAAkB,YAAY;AAE/D,WAAK,gBAAgB,qBACnB,YACA,YACA,SACD;AACD;;KAGF,MAAM,YAAY,KAAK,SAAS,IAAI,YAAY,SAAS;AACzD,SAAI,CAAC,WAAW;AACd,cAAQ,KACN,wDACD;AACD;;AAGF,UAAK,iBAAiB,iBAAiB;MACrC,MAAM,SAAS;OACb,GAAG,QAAQ;OACX,GAAG;OACJ;MAGD,IAAI,WAAW,QAAQ,YAAY,OAAO,CAAC,OAAO;MAClD,MAAM,EAAE,OAAO,UACb,KAAK,SAAS,IAAI,WAAW,SAAS,OAAO,IAAI,EAAE;AAErD,UAAI,CAAC,SAAS,SAAS,KAAM,QAAO;AACpC,UAAI,MAAM,SAAS,KAAK,SAEtB,KAAI,MAAM,OAAO,oBAEf,YAAW,EAAE;UAEb,UAAS,OAAO,OAAO,EAAE;WAEtB;OACL,MAAM,UAAUC,+BAAc,MAAM;AACpC,gBAAS,SAAS;AAGlB,WACE,CAAC,kBACD,QAAQ,SAAS,QACjB,gBAAgB,WAChB,MAAM,QAAQ,QAAQ,WAAW,CAEjC,MAAK,gBAAgB,sBACnB,QAAQ,YACR,QAAQ,GACT;AAIH,WACE,CAAC,kBACD,QAAQ,SAAS,UACjB,kBAAkB,SAClB;QACA,MAAM,OAAO,QAAQ;QACrB,MAAM,UACJ,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,QAAQ;QACrC,MAAM,SACJ,YAAY,WAAW,QAAQ,WAAW,UACtC,UACA;AACN,aAAK,gBAAgB,mBAAmB,MAAM,SAAS,OAAO;;;AAIlE,aAAO,QAAQ,YAAY,QAAQ,SAAS;OAC5C;;;AAIN,OAAI,eAAe,KAAM,OAAM;AAK/B,OAAI,CAAC,KAAK,SAAS,OAAO,SAAS;IACjC,MAAM,SAAS,MAAM,QAAQ,aAAa;AAC1C,QAAI,OAAO,WAAW,eAAe,KAAK,cAAc,EACtD,MAAK,gBAAgB,OAAO;;WAGzB,OAAO;AACd,OACE,EACE,iBAAiB,UAChB,MAAM,SAAS,gBAAgB,MAAM,SAAS,kBAEjD;AACA,YAAQ,MAAM,MAAM;AACpB,SAAK,SAAS,EAAE,OAAO,CAAC;AACxB,UAAM,QAAQ,UAAU,MAAM;;YAExB;AACR,QAAK,SAAS,EAAE,WAAW,OAAO,CAAC;AACnC,QAAK,WAAW,IAAI,iBAAiB;AACrC,WAAQ,YAAY;;;CAIxB,QAAQ,OACN,QAWA,SAoBA,iBAQkB;AAClB,MAAI,cAAc,cAChB,MAAK,SAAS,OAAO;AAEvB,OAAK,aAAa;AAClB,OAAK,QAAQ,KAAK,MAAM,WAAW,KAAK,QAAQ,QAAQ,QAAQ,CAAC;;CAGnE,OAAO,OACL,eACA,YAOkB;AAClB,OAAK,SAAS,OAAO;AACrB,OAAK,WAAW,IAAI,iBAAiB;AAErC,UAAQ,SAAS,EAAE,QAAQ,KAAK,YAAY,QAAQ,cAAc,EAAE,CAAC;;CAGvE,cAAc;AAEZ,OAAK,SAAS,OAAO;AACrB,OAAK,WAAW,IAAI,iBAAiB;AAGrC,OAAK,SAAS;GAAE,OAAO;GAAW,QAAQ;GAAM,WAAW;GAAO,CAAC;AAGnE,OAAK,SAAS,OAAO;AAGrB,OAAK,gBAAgB,OAAO"}
+{"version":3,"file":"manager.cjs","names":["toMessageDict","SubagentManager","StreamError","isSubagentNamespace","extractToolCallIdFromNamespace"],"sources":["../../src/ui/manager.ts"],"sourcesContent":["import type { BaseMessage } from \"@langchain/core/messages\";\n\nimport type {\n  CheckpointsStreamEvent,\n  CustomStreamEvent,\n  DebugStreamEvent,\n  ErrorStreamEvent,\n  EventsStreamEvent,\n  FeedbackStreamEvent,\n  MessagesTupleStreamEvent,\n  MetadataStreamEvent,\n  TasksStreamEvent,\n  ToolsStreamEvent,\n  UpdatesStreamEvent,\n  ValuesStreamEvent,\n} from \"../types.stream.js\";\nimport { MessageTupleManager, toMessageDict } from \"./messages.js\";\nimport { StreamError } from \"./errors.js\";\nimport type { Message } from \"../types.messages.js\";\nimport type { BagTemplate } from \"../types.template.js\";\nimport {\n  SubagentManager,\n  extractToolCallIdFromNamespace,\n  isSubagentNamespace,\n} from \"./subagents.js\";\nimport type { SubagentStreamInterface } from \"./types.js\";\n\n/**\n * Special ID used by LangGraph's messagesStateReducer to signal\n * that all messages should be removed from the state.\n */\nexport const REMOVE_ALL_MESSAGES = \"__remove_all__\";\n\ntype GetUpdateType<\n  Bag extends BagTemplate,\n  StateType extends Record<string, unknown>\n> = Bag extends { UpdateType: unknown }\n  ? Bag[\"UpdateType\"]\n  : Partial<StateType>;\n\ntype GetCustomEventType<Bag extends BagTemplate> = Bag extends {\n  CustomEventType: unknown;\n}\n  ? Bag[\"CustomEventType\"]\n  : unknown;\n\ntype EventStreamMap<StateType, UpdateType, CustomType> = {\n  values: ValuesStreamEvent<StateType>;\n  updates: UpdatesStreamEvent<UpdateType>;\n  custom: CustomStreamEvent<CustomType>;\n  debug: DebugStreamEvent;\n  messages: MessagesTupleStreamEvent;\n  events: EventsStreamEvent;\n  metadata: MetadataStreamEvent;\n  checkpoints: CheckpointsStreamEvent<StateType>;\n  tasks: TasksStreamEvent<StateType, UpdateType>;\n  error: ErrorStreamEvent;\n  feedback: FeedbackStreamEvent;\n  tools: ToolsStreamEvent;\n};\n\nexport type EventStreamEvent<StateType, UpdateType, CustomType> =\n  EventStreamMap<StateType, UpdateType, CustomType>[keyof EventStreamMap<\n    StateType,\n    UpdateType,\n    CustomType\n  >];\n\ninterface StreamManagerEventCallbacks<\n  StateType extends Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate\n> {\n  onUpdateEvent?: (\n    data: UpdatesStreamEvent<GetUpdateType<Bag, StateType>>[\"data\"],\n    options: {\n      namespace: string[] | undefined;\n      mutate: (\n        update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n      ) => void;\n    }\n  ) => void;\n  onCustomEvent?: (\n    data: GetCustomEventType<Bag>,\n    options: {\n      namespace: string[] | undefined;\n      mutate: (\n        update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n      ) => void;\n    }\n  ) => void;\n  onMetadataEvent?: (data: MetadataStreamEvent[\"data\"]) => void;\n  onLangChainEvent?: (data: EventsStreamEvent[\"data\"]) => void;\n  onDebugEvent?: (\n    data: DebugStreamEvent[\"data\"],\n    options: { namespace: string[] | undefined }\n  ) => void;\n  onCheckpointEvent?: (\n    data: CheckpointsStreamEvent<StateType>[\"data\"],\n    options: { namespace: string[] | undefined }\n  ) => void;\n  onTaskEvent?: (\n    data: TasksStreamEvent<StateType, GetUpdateType<Bag, StateType>>[\"data\"],\n    options: { namespace: string[] | undefined }\n  ) => void;\n  onToolEvent?: (\n    data: ToolsStreamEvent[\"data\"],\n    options: {\n      namespace: string[] | undefined;\n      mutate: (\n        update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n      ) => void;\n    }\n  ) => void;\n}\n\n/**\n * Options for StreamManager constructor.\n */\nexport interface StreamManagerOptions {\n  /**\n   * Throttle the stream updates.\n   * If a number is provided, updates are throttled to the given milliseconds.\n   * If `true`, updates are batched in a single macrotask.\n   * If `false`, updates are not throttled.\n   */\n  throttle: number | boolean;\n\n  /**\n   * Tool names that indicate subagent invocation.\n   *\n   * When an AI message contains tool calls with these names, they are\n   * automatically tracked as subagent executions. This enables the\n   * `subagents`, `activeSubagents`, `getSubagent()`, and `getSubagentsByType()`\n   * properties on the stream.\n   *\n   * @default [\"task\"]\n   *\n   * @example\n   * ```typescript\n   * // Track both \"task\" and \"delegate\" as subagent tools\n   * subagentToolNames: [\"task\", \"delegate\", \"spawn_agent\"]\n   * ```\n   */\n  subagentToolNames?: string[];\n\n  /**\n   * Filter out messages from subagent streams in the main messages array.\n   *\n   * When enabled, messages from subagraph executions (those with a `tools:` namespace)\n   * are excluded from `stream.messages`. Instead, these messages are tracked\n   * per-subagent and accessible via `stream.subagents.get(id).messages`.\n   *\n   * This is useful for deep agent architectures where you want to display\n   * the main conversation separately from subagent activity.\n   *\n   * @default false\n   *\n   * @example\n   * ```typescript\n   * const stream = useStream({\n   *   assistantId: \"my-agent\",\n   *   filterSubagentMessages: true,\n   * });\n   *\n   * // Main thread messages only (no subagent messages)\n   * stream.messages\n   *\n   * // Access subagent messages individually\n   * stream.subagents.get(\"call_xyz\").messages\n   * ```\n   */\n  filterSubagentMessages?: boolean;\n\n  /**\n   * Converts a @langchain/core BaseMessage to the desired output format.\n   *\n   * Defaults to `toMessageDict` which produces plain Message objects.\n   * Framework SDKs pass `toMessageClass` (identity) to keep class instances.\n   */\n  toMessage?: (chunk: BaseMessage) => Message | BaseMessage;\n}\n\nexport class StreamManager<\n  StateType extends Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate\n> {\n  private abortRef = new AbortController();\n\n  private messages: MessageTupleManager;\n\n  private subagentManager: SubagentManager;\n\n  private listeners = new Set<() => void>();\n\n  private throttle: number | boolean;\n\n  private filterSubagentMessages: boolean;\n\n  private toMessage: (chunk: BaseMessage) => Message | BaseMessage;\n\n  private queue: Promise<unknown> = Promise.resolve();\n\n  private queueSize: number = 0;\n\n  private state: {\n    isLoading: boolean;\n    values: [values: StateType, kind: \"stream\" | \"stop\"] | null;\n    error: unknown;\n    /** Version counter to force React re-renders on subagent changes */\n    version: number;\n  };\n\n  constructor(messages: MessageTupleManager, options: StreamManagerOptions) {\n    this.messages = messages;\n    this.state = {\n      isLoading: false,\n      values: null,\n      error: undefined,\n      version: 0,\n    };\n    this.throttle = options.throttle;\n    this.filterSubagentMessages = options.filterSubagentMessages ?? false;\n    this.toMessage = options.toMessage ?? toMessageDict;\n    this.subagentManager = new SubagentManager({\n      subagentToolNames: options.subagentToolNames,\n      onSubagentChange: () => this.bumpVersion(),\n      toMessage: this.toMessage,\n    });\n  }\n\n  /**\n   * Increment version counter to trigger React re-renders.\n   * Called when subagent state changes.\n   */\n  private bumpVersion = () => {\n    this.state = { ...this.state, version: this.state.version + 1 };\n    this.notifyListeners();\n  };\n\n  /**\n   * Get all subagents as a Map.\n   */\n  getSubagents(): Map<string, SubagentStreamInterface> {\n    return this.subagentManager.getSubagents();\n  }\n\n  /**\n   * Get all currently running subagents.\n   */\n  getActiveSubagents(): SubagentStreamInterface[] {\n    return this.subagentManager.getActiveSubagents();\n  }\n\n  /**\n   * Get a specific subagent by tool call ID.\n   */\n  getSubagent(toolCallId: string): SubagentStreamInterface | undefined {\n    return this.subagentManager.getSubagent(toolCallId);\n  }\n\n  /**\n   * Get all subagents of a specific type.\n   */\n  getSubagentsByType(type: string): SubagentStreamInterface[] {\n    return this.subagentManager.getSubagentsByType(type);\n  }\n\n  /**\n   * Get all subagents triggered by a specific AI message.\n   */\n  getSubagentsByMessage(messageId: string): SubagentStreamInterface[] {\n    return this.subagentManager.getSubagentsByMessage(messageId);\n  }\n\n  /**\n   * Reconstruct subagent state from historical messages.\n   *\n   * This method should be called when loading thread history to restore\n   * subagent visualization after:\n   * - Page refresh (when stream has already completed)\n   * - Loading thread history\n   * - Navigating between threads\n   *\n   * @param messages - Array of messages from thread history\n   * @param options - Optional configuration\n   * @param options.skipIfPopulated - If true, skip reconstruction if subagents already exist\n   */\n  reconstructSubagents(\n    messages: Message[],\n    options?: { skipIfPopulated?: boolean }\n  ): void {\n    this.subagentManager.reconstructFromMessages(messages, options);\n  }\n\n  /**\n   * Check if any subagents are currently tracked.\n   */\n  hasSubagents(): boolean {\n    return this.subagentManager.hasSubagents();\n  }\n\n  private setState = (newState: Partial<typeof this.state>) => {\n    this.state = { ...this.state, ...newState };\n    this.notifyListeners();\n  };\n\n  private notifyListeners = () => {\n    this.listeners.forEach((listener) => listener());\n  };\n\n  subscribe = (listener: () => void): (() => void) => {\n    if (this.throttle === false) {\n      this.listeners.add(listener);\n      return () => this.listeners.delete(listener);\n    }\n\n    const timeoutMs = this.throttle === true ? 0 : this.throttle;\n    let timeoutId: NodeJS.Timeout | number | undefined;\n\n    const throttledListener = () => {\n      clearTimeout(timeoutId);\n      timeoutId = setTimeout(() => {\n        clearTimeout(timeoutId);\n        listener();\n      }, timeoutMs);\n    };\n\n    this.listeners.add(throttledListener);\n    return () => {\n      clearTimeout(timeoutId);\n      this.listeners.delete(throttledListener);\n    };\n  };\n\n  getSnapshot = () => this.state;\n\n  get isLoading() {\n    return this.state.isLoading;\n  }\n\n  get values() {\n    return this.state.values?.[0] ?? null;\n  }\n\n  get error() {\n    return this.state.error;\n  }\n\n  setStreamValues = (\n    values:\n      | (StateType | null)\n      | ((prev: StateType | null, kind: \"stream\" | \"stop\") => StateType | null),\n    kind: \"stream\" | \"stop\" = \"stream\"\n  ) => {\n    if (typeof values === \"function\") {\n      const [prevValues, prevKind] = this.state.values ?? [null, \"stream\"];\n      const nextValues = values(prevValues, prevKind);\n      this.setState({ values: nextValues != null ? [nextValues, kind] : null });\n    } else {\n      const nextValues = values != null ? [values, kind] : null;\n      this.setState({ values: nextValues as [StateType, \"stream\" | \"stop\"] });\n    }\n  };\n\n  private getMutateFn = (kind: \"stream\" | \"stop\", historyValues: StateType) => {\n    return (\n      update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n    ) => {\n      const stateValues = (this.state.values ?? [null, \"stream\"])[0];\n      const prev = {\n        ...historyValues,\n        ...stateValues,\n      };\n      const next = typeof update === \"function\" ? update(prev) : update;\n      this.setStreamValues({ ...prev, ...next }, kind);\n    };\n  };\n\n  private matchEventType = <\n    T extends keyof EventStreamMap<\n      StateType,\n      GetUpdateType<Bag, StateType>,\n      GetCustomEventType<Bag>\n    >\n  >(\n    expected: T,\n    actual: EventStreamEvent<\n      StateType,\n      GetUpdateType<Bag, StateType>,\n      GetCustomEventType<Bag>\n    >[\"event\"],\n    _data: EventStreamEvent<\n      StateType,\n      GetUpdateType<Bag, StateType>,\n      GetCustomEventType<Bag>\n    >[\"data\"]\n  ): _data is EventStreamMap<\n    StateType,\n    GetUpdateType<Bag, StateType>,\n    GetCustomEventType<Bag>\n  >[T][\"data\"] => {\n    return expected === actual || actual.startsWith(`${expected}|`);\n  };\n\n  protected enqueue = async (\n    action: (\n      signal: AbortSignal\n    ) => Promise<\n      AsyncGenerator<\n        EventStreamEvent<\n          StateType,\n          GetUpdateType<Bag, StateType>,\n          GetCustomEventType<Bag>\n        >\n      >\n    >,\n    options: {\n      getMessages: (values: StateType) => Message[];\n\n      setMessages: (current: StateType, messages: Message[]) => StateType;\n\n      initialValues: StateType;\n\n      callbacks: StreamManagerEventCallbacks<StateType, Bag>;\n\n      onSuccess: () =>\n        | StateType\n        | null\n        | undefined\n        | void\n        | Promise<StateType | null | undefined | void>;\n\n      onError: (error: unknown) => void | Promise<void>;\n\n      onFinish?: () => void;\n    }\n  ) => {\n    try {\n      this.queueSize = Math.max(0, this.queueSize - 1);\n      this.setState({ isLoading: true, error: undefined });\n      this.abortRef = new AbortController();\n\n      const run = await action(this.abortRef.signal);\n\n      let streamError: StreamError | undefined;\n      for await (const { event, data } of run) {\n        if (event === \"error\") {\n          streamError = new StreamError(data);\n          break;\n        }\n\n        const namespace = event.includes(\"|\")\n          ? event.split(\"|\").slice(1)\n          : undefined;\n\n        const mutate = this.getMutateFn(\"stream\", options.initialValues);\n\n        if (event === \"metadata\") options.callbacks.onMetadataEvent?.(data);\n        if (event === \"events\") options.callbacks.onLangChainEvent?.(data);\n\n        if (this.matchEventType(\"updates\", event, data)) {\n          options.callbacks.onUpdateEvent?.(data, { namespace, mutate });\n\n          // Track subagent streaming updates from subgraph namespaces\n          // Mark the subagent as running when we receive updates\n          // The actual message content is handled via addMessageToSubagent\n          if (namespace && isSubagentNamespace(namespace)) {\n            const namespaceId = extractToolCallIdFromNamespace(namespace);\n            if (namespaceId && this.filterSubagentMessages) {\n              this.subagentManager.markRunningFromNamespace(\n                namespaceId,\n                namespace\n              );\n            }\n          }\n\n          // Also register subagents from main agent updates (tool_calls in messages)\n          // AND process tool results to complete subagents\n          // This is needed because tool_calls often appear complete in updates\n          // before they appear in the messages stream\n          if (!namespace || !isSubagentNamespace(namespace)) {\n            const updateData = data as Record<string, unknown>;\n            for (const nodeData of Object.values(updateData)) {\n              if (\n                nodeData &&\n                typeof nodeData === \"object\" &&\n                \"messages\" in nodeData\n              ) {\n                const { messages } = nodeData as { messages: unknown[] };\n                if (Array.isArray(messages)) {\n                  for (const msg of messages) {\n                    if (!msg || typeof msg !== \"object\") continue;\n                    const msgObj = msg as Record<string, unknown>;\n\n                    // Register subagents from AI messages with tool_calls\n                    if (\n                      msgObj.type === \"ai\" &&\n                      \"tool_calls\" in msgObj &&\n                      Array.isArray(msgObj.tool_calls)\n                    ) {\n                      this.subagentManager.registerFromToolCalls(\n                        msgObj.tool_calls as Array<{\n                          id?: string;\n                          name: string;\n                          args: Record<string, unknown> | string;\n                        }>,\n                        msgObj.id as string | undefined\n                      );\n                    }\n\n                    // Complete subagents from tool messages (task results)\n                    if (\n                      msgObj.type === \"tool\" &&\n                      \"tool_call_id\" in msgObj &&\n                      typeof msgObj.tool_call_id === \"string\"\n                    ) {\n                      const content =\n                        typeof msgObj.content === \"string\"\n                          ? msgObj.content\n                          : JSON.stringify(msgObj.content);\n                      const status =\n                        \"status\" in msgObj && msgObj.status === \"error\"\n                          ? \"error\"\n                          : \"success\";\n                      this.subagentManager.processToolMessage(\n                        msgObj.tool_call_id,\n                        content,\n                        status\n                      );\n                    }\n                  }\n                }\n              }\n            }\n          }\n        }\n\n        if (this.matchEventType(\"custom\", event, data)) {\n          options.callbacks.onCustomEvent?.(data, { namespace, mutate });\n        }\n\n        if (this.matchEventType(\"checkpoints\", event, data)) {\n          options.callbacks.onCheckpointEvent?.(data, { namespace });\n        }\n\n        if (this.matchEventType(\"tasks\", event, data)) {\n          options.callbacks.onTaskEvent?.(data, { namespace });\n        }\n\n        if (this.matchEventType(\"debug\", event, data)) {\n          options.callbacks.onDebugEvent?.(data, { namespace });\n        }\n\n        if (this.matchEventType(\"tools\", event, data)) {\n          options.callbacks.onToolEvent?.(data, { namespace, mutate });\n        }\n\n        // Handle values events - use startsWith to match both \"values\" and \"values|tools:xxx\"\n        if (event === \"values\" || event.startsWith(\"values|\")) {\n          // Check if this is a subgraph values event (for namespace mapping and values)\n          if (namespace && isSubagentNamespace(namespace)) {\n            const namespaceId = extractToolCallIdFromNamespace(namespace);\n            if (namespaceId && this.filterSubagentMessages) {\n              const valuesData = data as Record<string, unknown>;\n\n              // Try to establish namespace mapping from the initial human message\n              const messages = valuesData.messages as unknown[];\n              if (Array.isArray(messages) && messages.length > 0) {\n                const firstMsg = messages[0] as Record<string, unknown>;\n                if (\n                  firstMsg?.type === \"human\" &&\n                  typeof firstMsg?.content === \"string\"\n                ) {\n                  this.subagentManager.matchSubgraphToSubagent(\n                    namespaceId,\n                    firstMsg.content\n                  );\n                }\n              }\n\n              // Update the subagent's values with the full state\n              this.subagentManager.updateSubagentValues(\n                namespaceId,\n                valuesData\n              );\n            }\n          } else if (\n            data &&\n            typeof data === \"object\" &&\n            \"__interrupt__\" in data\n          ) {\n            const interruptData = data as Partial<StateType>;\n            this.setStreamValues(\n              (prev) => ({ ...prev, ...interruptData } as StateType)\n            );\n          } else {\n            this.setStreamValues(data as StateType);\n          }\n        }\n\n        if (this.matchEventType(\"messages\", event, data)) {\n          const [serialized, metadata] = data;\n\n          // Check if this message is from a subagent namespace\n          const rawCheckpointNs =\n            (metadata?.langgraph_checkpoint_ns as string | undefined) ||\n            (metadata?.checkpoint_ns as string | undefined);\n          const checkpointNs: string | undefined =\n            typeof rawCheckpointNs === \"string\" ? rawCheckpointNs : undefined;\n          const isFromSubagent = isSubagentNamespace(checkpointNs);\n          const toolCallId = isFromSubagent\n            ? extractToolCallIdFromNamespace(checkpointNs?.split(\"|\"))\n            : undefined;\n\n          // If filtering is enabled and this is a subagent message,\n          // add it to the subagent's messages instead of the main stream\n          if (this.filterSubagentMessages && isFromSubagent && toolCallId) {\n            // Add to subagent's message list\n            this.subagentManager.addMessageToSubagent(\n              toolCallId,\n              serialized,\n              metadata\n            );\n            continue;\n          }\n\n          const messageId = this.messages.add(serialized, metadata);\n          if (!messageId) {\n            console.warn(\n              \"Failed to add message to manager, no message ID found\"\n            );\n            continue;\n          }\n\n          this.setStreamValues((streamValues) => {\n            const values = {\n              ...options.initialValues,\n              ...streamValues,\n            };\n\n            // Assumption: we're concatenating the message\n            let messages = options.getMessages(values).slice();\n            const { chunk, index } =\n              this.messages.get(messageId, messages.length) ?? {};\n\n            if (!chunk || index == null) return values;\n            if (chunk.getType() === \"remove\") {\n              // Check for special REMOVE_ALL_MESSAGES sentinel\n              if (chunk.id === REMOVE_ALL_MESSAGES) {\n                // Clear all messages when __remove_all__ is received\n                messages = [];\n              } else {\n                messages.splice(index, 1);\n              }\n            } else {\n              const msgDict = this.toMessage(chunk) as Message;\n              messages[index] = msgDict;\n\n              // Track subagents from AI messages with tool calls (main agent only)\n              if (\n                !isFromSubagent &&\n                msgDict.type === \"ai\" &&\n                \"tool_calls\" in msgDict &&\n                Array.isArray(msgDict.tool_calls)\n              ) {\n                this.subagentManager.registerFromToolCalls(\n                  msgDict.tool_calls,\n                  msgDict.id as string | undefined\n                );\n              }\n\n              // Complete subagents when tool messages arrive (main agent only)\n              if (\n                !isFromSubagent &&\n                msgDict.type === \"tool\" &&\n                \"tool_call_id\" in msgDict\n              ) {\n                const tcId = msgDict.tool_call_id as string;\n                const content =\n                  typeof msgDict.content === \"string\"\n                    ? msgDict.content\n                    : JSON.stringify(msgDict.content);\n                const status =\n                  \"status\" in msgDict && msgDict.status === \"error\"\n                    ? \"error\"\n                    : \"success\";\n                this.subagentManager.processToolMessage(tcId, content, status);\n              }\n            }\n\n            return options.setMessages(values, messages);\n          });\n        }\n      }\n\n      if (streamError != null) throw streamError;\n\n      // Skip onSuccess when the stream was aborted (e.g., by multitask interrupt).\n      // This avoids unnecessary HTTP calls (like history fetching) that would\n      // delay the next queued stream from starting.\n      if (!this.abortRef.signal.aborted) {\n        const values = await options.onSuccess?.();\n        if (typeof values !== \"undefined\" && this.queueSize === 0) {\n          this.setStreamValues(values);\n        }\n      }\n    } catch (error) {\n      if (\n        !(\n          error instanceof Error && // eslint-disable-line no-instanceof/no-instanceof\n          (error.name === \"AbortError\" || error.name === \"TimeoutError\")\n        )\n      ) {\n        console.error(error);\n        this.setState({ error });\n        await options.onError?.(error);\n      }\n    } finally {\n      this.setState({ isLoading: false });\n      this.abortRef = new AbortController();\n      options.onFinish?.();\n    }\n  };\n\n  start = async (\n    action: (\n      signal: AbortSignal\n    ) => Promise<\n      AsyncGenerator<\n        EventStreamEvent<\n          StateType,\n          GetUpdateType<Bag, StateType>,\n          GetCustomEventType<Bag>\n        >\n      >\n    >,\n    options: {\n      getMessages: (values: StateType) => Message[];\n\n      setMessages: (current: StateType, messages: Message[]) => StateType;\n\n      initialValues: StateType;\n\n      callbacks: StreamManagerEventCallbacks<StateType, Bag>;\n\n      onSuccess: () =>\n        | StateType\n        | null\n        | undefined\n        | void\n        | Promise<StateType | null | undefined | void>;\n\n      onError: (error: unknown) => void | Promise<void>;\n\n      onFinish?: () => void;\n    },\n    startOptions?: {\n      /**\n       * If true, abort any currently running stream before starting this one.\n       * Used for multitask_strategy: \"interrupt\" and \"rollback\" to unblock\n       * the queue so the new run request can proceed immediately.\n       */\n      abortPrevious?: boolean;\n    }\n  ): Promise<void> => {\n    if (startOptions?.abortPrevious) {\n      this.abortRef.abort();\n    }\n    this.queueSize += 1;\n    const queued = this.queue.then(() => this.enqueue(action, options));\n    this.queue = queued;\n    await queued;\n  };\n\n  stop = async (\n    historyValues: StateType,\n    options: {\n      onStop?: (options: {\n        mutate: (\n          update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)\n        ) => void;\n      }) => void;\n    }\n  ): Promise<void> => {\n    this.abortRef.abort();\n    this.abortRef = new AbortController();\n\n    options.onStop?.({ mutate: this.getMutateFn(\"stop\", historyValues) });\n  };\n\n  clear = () => {\n    // Cancel any running streams\n    this.abortRef.abort();\n    this.abortRef = new AbortController();\n\n    // Set the stream state to null\n    this.setState({ error: undefined, values: null, isLoading: false });\n\n    // Clear any pending messages\n    this.messages.clear();\n\n    // Clear subagent state\n    this.subagentManager.clear();\n  };\n}\n"],"mappings":";;;;;;;;;AA+BA,MAAa,sBAAsB;AAuJnC,IAAa,gBAAb,MAGE;CACA,AAAQ,WAAW,IAAI,iBAAiB;CAExC,AAAQ;CAER,AAAQ;CAER,AAAQ,4BAAY,IAAI,KAAiB;CAEzC,AAAQ;CAER,AAAQ;CAER,AAAQ;CAER,AAAQ,QAA0B,QAAQ,SAAS;CAEnD,AAAQ,YAAoB;CAE5B,AAAQ;CAQR,YAAY,UAA+B,SAA+B;AACxE,OAAK,WAAW;AAChB,OAAK,QAAQ;GACX,WAAW;GACX,QAAQ;GACR,OAAO;GACP,SAAS;GACV;AACD,OAAK,WAAW,QAAQ;AACxB,OAAK,yBAAyB,QAAQ,0BAA0B;AAChE,OAAK,YAAY,QAAQ,aAAaA;AACtC,OAAK,kBAAkB,IAAIC,kCAAgB;GACzC,mBAAmB,QAAQ;GAC3B,wBAAwB,KAAK,aAAa;GAC1C,WAAW,KAAK;GACjB,CAAC;;;;;;CAOJ,AAAQ,oBAAoB;AAC1B,OAAK,QAAQ;GAAE,GAAG,KAAK;GAAO,SAAS,KAAK,MAAM,UAAU;GAAG;AAC/D,OAAK,iBAAiB;;;;;CAMxB,eAAqD;AACnD,SAAO,KAAK,gBAAgB,cAAc;;;;;CAM5C,qBAAgD;AAC9C,SAAO,KAAK,gBAAgB,oBAAoB;;;;;CAMlD,YAAY,YAAyD;AACnE,SAAO,KAAK,gBAAgB,YAAY,WAAW;;;;;CAMrD,mBAAmB,MAAyC;AAC1D,SAAO,KAAK,gBAAgB,mBAAmB,KAAK;;;;;CAMtD,sBAAsB,WAA8C;AAClE,SAAO,KAAK,gBAAgB,sBAAsB,UAAU;;;;;;;;;;;;;;;CAgB9D,qBACE,UACA,SACM;AACN,OAAK,gBAAgB,wBAAwB,UAAU,QAAQ;;;;;CAMjE,eAAwB;AACtB,SAAO,KAAK,gBAAgB,cAAc;;CAG5C,AAAQ,YAAY,aAAyC;AAC3D,OAAK,QAAQ;GAAE,GAAG,KAAK;GAAO,GAAG;GAAU;AAC3C,OAAK,iBAAiB;;CAGxB,AAAQ,wBAAwB;AAC9B,OAAK,UAAU,SAAS,aAAa,UAAU,CAAC;;CAGlD,aAAa,aAAuC;AAClD,MAAI,KAAK,aAAa,OAAO;AAC3B,QAAK,UAAU,IAAI,SAAS;AAC5B,gBAAa,KAAK,UAAU,OAAO,SAAS;;EAG9C,MAAM,YAAY,KAAK,aAAa,OAAO,IAAI,KAAK;EACpD,IAAI;EAEJ,MAAM,0BAA0B;AAC9B,gBAAa,UAAU;AACvB,eAAY,iBAAiB;AAC3B,iBAAa,UAAU;AACvB,cAAU;MACT,UAAU;;AAGf,OAAK,UAAU,IAAI,kBAAkB;AACrC,eAAa;AACX,gBAAa,UAAU;AACvB,QAAK,UAAU,OAAO,kBAAkB;;;CAI5C,oBAAoB,KAAK;CAEzB,IAAI,YAAY;AACd,SAAO,KAAK,MAAM;;CAGpB,IAAI,SAAS;AACX,SAAO,KAAK,MAAM,SAAS,MAAM;;CAGnC,IAAI,QAAQ;AACV,SAAO,KAAK,MAAM;;CAGpB,mBACE,QAGA,OAA0B,aACvB;AACH,MAAI,OAAO,WAAW,YAAY;GAChC,MAAM,CAAC,YAAY,YAAY,KAAK,MAAM,UAAU,CAAC,MAAM,SAAS;GACpE,MAAM,aAAa,OAAO,YAAY,SAAS;AAC/C,QAAK,SAAS,EAAE,QAAQ,cAAc,OAAO,CAAC,YAAY,KAAK,GAAG,MAAM,CAAC;SACpE;GACL,MAAM,aAAa,UAAU,OAAO,CAAC,QAAQ,KAAK,GAAG;AACrD,QAAK,SAAS,EAAE,QAAQ,YAA8C,CAAC;;;CAI3E,AAAQ,eAAe,MAAyB,kBAA6B;AAC3E,UACE,WACG;GACH,MAAM,eAAe,KAAK,MAAM,UAAU,CAAC,MAAM,SAAS,EAAE;GAC5D,MAAM,OAAO;IACX,GAAG;IACH,GAAG;IACJ;GACD,MAAM,OAAO,OAAO,WAAW,aAAa,OAAO,KAAK,GAAG;AAC3D,QAAK,gBAAgB;IAAE,GAAG;IAAM,GAAG;IAAM,EAAE,KAAK;;;CAIpD,AAAQ,kBAON,UACA,QAKA,UASc;AACd,SAAO,aAAa,UAAU,OAAO,WAAW,GAAG,SAAS,GAAG;;CAGjE,AAAU,UAAU,OAClB,QAWA,YAoBG;AACH,MAAI;AACF,QAAK,YAAY,KAAK,IAAI,GAAG,KAAK,YAAY,EAAE;AAChD,QAAK,SAAS;IAAE,WAAW;IAAM,OAAO;IAAW,CAAC;AACpD,QAAK,WAAW,IAAI,iBAAiB;GAErC,MAAM,MAAM,MAAM,OAAO,KAAK,SAAS,OAAO;GAE9C,IAAI;AACJ,cAAW,MAAM,EAAE,OAAO,UAAU,KAAK;AACvC,QAAI,UAAU,SAAS;AACrB,mBAAc,IAAIC,2BAAY,KAAK;AACnC;;IAGF,MAAM,YAAY,MAAM,SAAS,IAAI,GACjC,MAAM,MAAM,IAAI,CAAC,MAAM,EAAE,GACzB;IAEJ,MAAM,SAAS,KAAK,YAAY,UAAU,QAAQ,cAAc;AAEhE,QAAI,UAAU,WAAY,SAAQ,UAAU,kBAAkB,KAAK;AACnE,QAAI,UAAU,SAAU,SAAQ,UAAU,mBAAmB,KAAK;AAElE,QAAI,KAAK,eAAe,WAAW,OAAO,KAAK,EAAE;AAC/C,aAAQ,UAAU,gBAAgB,MAAM;MAAE;MAAW;MAAQ,CAAC;AAK9D,SAAI,aAAaC,sCAAoB,UAAU,EAAE;MAC/C,MAAM,cAAcC,iDAA+B,UAAU;AAC7D,UAAI,eAAe,KAAK,uBACtB,MAAK,gBAAgB,yBACnB,aACA,UACD;;AAQL,SAAI,CAAC,aAAa,CAACD,sCAAoB,UAAU,EAAE;MACjD,MAAM,aAAa;AACnB,WAAK,MAAM,YAAY,OAAO,OAAO,WAAW,CAC9C,KACE,YACA,OAAO,aAAa,YACpB,cAAc,UACd;OACA,MAAM,EAAE,aAAa;AACrB,WAAI,MAAM,QAAQ,SAAS,CACzB,MAAK,MAAM,OAAO,UAAU;AAC1B,YAAI,CAAC,OAAO,OAAO,QAAQ,SAAU;QACrC,MAAM,SAAS;AAGf,YACE,OAAO,SAAS,QAChB,gBAAgB,UAChB,MAAM,QAAQ,OAAO,WAAW,CAEhC,MAAK,gBAAgB,sBACnB,OAAO,YAKP,OAAO,GACR;AAIH,YACE,OAAO,SAAS,UAChB,kBAAkB,UAClB,OAAO,OAAO,iBAAiB,UAC/B;SACA,MAAM,UACJ,OAAO,OAAO,YAAY,WACtB,OAAO,UACP,KAAK,UAAU,OAAO,QAAQ;SACpC,MAAM,SACJ,YAAY,UAAU,OAAO,WAAW,UACpC,UACA;AACN,cAAK,gBAAgB,mBACnB,OAAO,cACP,SACA,OACD;;;;;;AASf,QAAI,KAAK,eAAe,UAAU,OAAO,KAAK,CAC5C,SAAQ,UAAU,gBAAgB,MAAM;KAAE;KAAW;KAAQ,CAAC;AAGhE,QAAI,KAAK,eAAe,eAAe,OAAO,KAAK,CACjD,SAAQ,UAAU,oBAAoB,MAAM,EAAE,WAAW,CAAC;AAG5D,QAAI,KAAK,eAAe,SAAS,OAAO,KAAK,CAC3C,SAAQ,UAAU,cAAc,MAAM,EAAE,WAAW,CAAC;AAGtD,QAAI,KAAK,eAAe,SAAS,OAAO,KAAK,CAC3C,SAAQ,UAAU,eAAe,MAAM,EAAE,WAAW,CAAC;AAGvD,QAAI,KAAK,eAAe,SAAS,OAAO,KAAK,CAC3C,SAAQ,UAAU,cAAc,MAAM;KAAE;KAAW;KAAQ,CAAC;AAI9D,QAAI,UAAU,YAAY,MAAM,WAAW,UAAU,CAEnD,KAAI,aAAaA,sCAAoB,UAAU,EAAE;KAC/C,MAAM,cAAcC,iDAA+B,UAAU;AAC7D,SAAI,eAAe,KAAK,wBAAwB;MAC9C,MAAM,aAAa;MAGnB,MAAM,WAAW,WAAW;AAC5B,UAAI,MAAM,QAAQ,SAAS,IAAI,SAAS,SAAS,GAAG;OAClD,MAAM,WAAW,SAAS;AAC1B,WACE,UAAU,SAAS,WACnB,OAAO,UAAU,YAAY,SAE7B,MAAK,gBAAgB,wBACnB,aACA,SAAS,QACV;;AAKL,WAAK,gBAAgB,qBACnB,aACA,WACD;;eAGH,QACA,OAAO,SAAS,YAChB,mBAAmB,MACnB;KACA,MAAM,gBAAgB;AACtB,UAAK,iBACF,UAAU;MAAE,GAAG;MAAM,GAAG;MAAe,EACzC;UAED,MAAK,gBAAgB,KAAkB;AAI3C,QAAI,KAAK,eAAe,YAAY,OAAO,KAAK,EAAE;KAChD,MAAM,CAAC,YAAY,YAAY;KAG/B,MAAM,kBACH,UAAU,2BACV,UAAU;KACb,MAAM,eACJ,OAAO,oBAAoB,WAAW,kBAAkB;KAC1D,MAAM,iBAAiBD,sCAAoB,aAAa;KACxD,MAAM,aAAa,iBACfC,iDAA+B,cAAc,MAAM,IAAI,CAAC,GACxD;AAIJ,SAAI,KAAK,0BAA0B,kBAAkB,YAAY;AAE/D,WAAK,gBAAgB,qBACnB,YACA,YACA,SACD;AACD;;KAGF,MAAM,YAAY,KAAK,SAAS,IAAI,YAAY,SAAS;AACzD,SAAI,CAAC,WAAW;AACd,cAAQ,KACN,wDACD;AACD;;AAGF,UAAK,iBAAiB,iBAAiB;MACrC,MAAM,SAAS;OACb,GAAG,QAAQ;OACX,GAAG;OACJ;MAGD,IAAI,WAAW,QAAQ,YAAY,OAAO,CAAC,OAAO;MAClD,MAAM,EAAE,OAAO,UACb,KAAK,SAAS,IAAI,WAAW,SAAS,OAAO,IAAI,EAAE;AAErD,UAAI,CAAC,SAAS,SAAS,KAAM,QAAO;AACpC,UAAI,MAAM,SAAS,KAAK,SAEtB,KAAI,MAAM,OAAO,oBAEf,YAAW,EAAE;UAEb,UAAS,OAAO,OAAO,EAAE;WAEtB;OACL,MAAM,UAAU,KAAK,UAAU,MAAM;AACrC,gBAAS,SAAS;AAGlB,WACE,CAAC,kBACD,QAAQ,SAAS,QACjB,gBAAgB,WAChB,MAAM,QAAQ,QAAQ,WAAW,CAEjC,MAAK,gBAAgB,sBACnB,QAAQ,YACR,QAAQ,GACT;AAIH,WACE,CAAC,kBACD,QAAQ,SAAS,UACjB,kBAAkB,SAClB;QACA,MAAM,OAAO,QAAQ;QACrB,MAAM,UACJ,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,QAAQ;QACrC,MAAM,SACJ,YAAY,WAAW,QAAQ,WAAW,UACtC,UACA;AACN,aAAK,gBAAgB,mBAAmB,MAAM,SAAS,OAAO;;;AAIlE,aAAO,QAAQ,YAAY,QAAQ,SAAS;OAC5C;;;AAIN,OAAI,eAAe,KAAM,OAAM;AAK/B,OAAI,CAAC,KAAK,SAAS,OAAO,SAAS;IACjC,MAAM,SAAS,MAAM,QAAQ,aAAa;AAC1C,QAAI,OAAO,WAAW,eAAe,KAAK,cAAc,EACtD,MAAK,gBAAgB,OAAO;;WAGzB,OAAO;AACd,OACE,EACE,iBAAiB,UAChB,MAAM,SAAS,gBAAgB,MAAM,SAAS,kBAEjD;AACA,YAAQ,MAAM,MAAM;AACpB,SAAK,SAAS,EAAE,OAAO,CAAC;AACxB,UAAM,QAAQ,UAAU,MAAM;;YAExB;AACR,QAAK,SAAS,EAAE,WAAW,OAAO,CAAC;AACnC,QAAK,WAAW,IAAI,iBAAiB;AACrC,WAAQ,YAAY;;;CAIxB,QAAQ,OACN,QAWA,SAoBA,iBAQkB;AAClB,MAAI,cAAc,cAChB,MAAK,SAAS,OAAO;AAEvB,OAAK,aAAa;EAClB,MAAM,SAAS,KAAK,MAAM,WAAW,KAAK,QAAQ,QAAQ,QAAQ,CAAC;AACnE,OAAK,QAAQ;AACb,QAAM;;CAGR,OAAO,OACL,eACA,YAOkB;AAClB,OAAK,SAAS,OAAO;AACrB,OAAK,WAAW,IAAI,iBAAiB;AAErC,UAAQ,SAAS,EAAE,QAAQ,KAAK,YAAY,QAAQ,cAAc,EAAE,CAAC;;CAGvE,cAAc;AAEZ,OAAK,SAAS,OAAO;AACrB,OAAK,WAAW,IAAI,iBAAiB;AAGrC,OAAK,SAAS;GAAE,OAAO;GAAW,QAAQ;GAAM,WAAW;GAAO,CAAC;AAGnE,OAAK,SAAS,OAAO;AAGrB,OAAK,gBAAgB,OAAO"}