@langchain/langgraph-sdk 1.7.1 → 1.7.3

package/dist/types.d.cts

@@ -169,12 +169,23 @@ interface RunsCreatePayload extends RunsInvokePayload {
    * If true, the stream can be resumed and replayed in its entirety even after disconnection.
    */
   streamResumable?: boolean;
+  /**
+   * Pass one or more feedbackKeys if you want to request short-lived signed URLs
+   * for submitting feedback to LangSmith with this key for this run.
+   */
+  feedbackKeys?: string[];
 }
 interface CronsCreatePayload extends RunsCreatePayload {
   /**
-   * Schedule for running the Cron Job. Schedules are interpreted in UTC.
+   * Schedule for running the Cron Job. Schedules are interpreted in UTC unless
+   * a timezone is specified.
    */
   schedule: string;
+  /**
+   * IANA timezone string for interpreting the schedule (e.g. "America/New_York").
+   * If not provided, the schedule is interpreted in UTC.
+   */
+  timezone?: string;
   /**
    * What to do with the thread after the run completes.
    * - "delete" (default): Automatically deletes the thread after the run completes.
@@ -186,10 +197,30 @@ interface CronsCreatePayload extends RunsCreatePayload {
    * Whether the cron is enabled.
    */
   enabled?: boolean;
+  /**
+   * The end date to stop running the cron job (ISO 8601 datetime string).
+   */
+  endTime?: string;
 }
 interface CronsUpdatePayload extends RunsInvokePayload {
   schedule?: string;
   endTime?: string;
+  /**
+   * IANA timezone string for interpreting the schedule (e.g. "America/New_York").
+   */
+  timezone?: string;
+  /**
+   * One of `"values"`, `"messages"`, `"messages-tuple"`, `"updates"`, `"events"`, `"debug"`, `"custom"`.
+   */
+  streamMode?: StreamMode | Array<StreamMode>;
+  /**
+   * Stream output from subgraphs. By default, streams only the top graph.
+   */
+  streamSubgraphs?: boolean;
+  /**
+   * Whether the stream is considered resumable.
+   */
+  streamResumable?: boolean;
   /**
    * What to do with the thread after the run completes.
    * - "delete" (default): Automatically deletes the thread after the run completes.

package/dist/types.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.cts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;KAIY,iBAAA;AAAA,KACA,kBAAA;AAAA,KACA,oBAAA;AAAA,KACA,cAAA;AAAA,KACA,UAAA;AAAA,KACA,WAAA;AAAA,UAYK,IAAA;EACf,IAAA;EACA,KAAA;AAAA;AAAA,UAGe,OAAA;EArBa;AAC9B;;EAwBE,MAAA,GAAS,MAAA;EAxBqB;;AAChC;EA4BE,MAAA;;;;AA3BF;;EAkCE,IAAA,GAAO,IAAA,GAAO,IAAA;AAAA;AAAA,UAGC,iBAAA;EApCL;;;EAwCV,KAAA,GAAQ,MAAA;EAxCa;AAYvB;;EAiCE,QAAA,GAAW,QAAA;EAhCX;;AAIF;EAiCE,MAAA,GAAS,MAAA;;;;;EAMT,OAAA;EAvBkB;;;EA4BlB,YAAA;EA5BA;;;EAiCA,UAAA,GAAa,IAAA,CAAK,UAAA;EAjCA;AAGpB;;;EAoCE,gBAAA;EA3BW;;;;;;;EAoCX,UAAA,GAAa,UAAA;EAgDE;;;EA3Cf,eAAA;EAsEkC;;;EAjElC,cAAA;EA9CW;;;;;;;;;;EA0DX,iBAAA,GAAoB,iBAAA;EAjBpB;;;EAsBA,MAAA,GAAS,eAAA;EAAT;;;;;;EAQA,YAAA,GAAe,oBAAA;EAmBf;;;EAdA,OAAA;EA6BA;;;;;;EArBA,YAAA,GAAe,cAAA;EA2BmB;AAGpC;;;EAxBE,YAAA;EAyBiC;;;EApBjC,WAAA;EAsByB;;;EAjBzB,OAAA,GAAU,OAAA;EAeuB;;;EAVjC,YAAA,IAAgB,MAAA;IAAU,MAAA;IAAgB,SAAA;EAAA;EA2B1C;;;;EArBA,gBAAA,GAAmB,eAAA;AAAA;AAAA,UAGJ,iBAAA,qBACK,UAAA,GAAa,UAAA,qDAEzB,iBAAA;EA4BK;;;EAxBb,UAAA,GAAa,WAAA;EAoB6C;;;EAf1D,eAAA,GAAkB,UAAA;EAmBL;;;;EAbb,eAAA;EAwBe;;AAGjB;;EArBE,YAAA;AAAA;AAAA,UAGe,iBAAA,SAA0B,iBAAA;EAsBzC;;;EAlBA,UAAA,GAAa,UAAA,GAAa,KAAA,CAAM,UAAA;EA+BzB;AAGT;;EA7BE,eAAA;EA6B2D;;;;EAvB3D,eAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EA8BnC;AAGT;;EA7BE,QAAA;EA6BuC;;;;;;EArBvC,cAAA;;;;EAKA,OAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EAC1C,QAAA;EACA,OAAA;;;;;;;EAOA,cAAA;EACA,OAAA;AAAA;AAAA,UAGe,eAAA,SAAwB,iBAAA;;;;EAIvC,UAAA;AAAA"}
+{"version":3,"file":"types.d.cts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;KAIY,iBAAA;AAAA,KACA,kBAAA;AAAA,KACA,oBAAA;AAAA,KACA,cAAA;AAAA,KACA,UAAA;AAAA,KACA,WAAA;AAAA,UAYK,IAAA;EACf,IAAA;EACA,KAAA;AAAA;AAAA,UAGe,OAAA;EArBa;AAC9B;;EAwBE,MAAA,GAAS,MAAA;EAxBqB;;AAChC;EA4BE,MAAA;;;;AA3BF;;EAkCE,IAAA,GAAO,IAAA,GAAO,IAAA;AAAA;AAAA,UAGC,iBAAA;EApCL;;;EAwCV,KAAA,GAAQ,MAAA;EAxCa;AAYvB;;EAiCE,QAAA,GAAW,QAAA;EAhCX;;AAIF;EAiCE,MAAA,GAAS,MAAA;;;;;EAMT,OAAA;EAvBkB;;;EA4BlB,YAAA;EA5BA;;;EAiCA,UAAA,GAAa,IAAA,CAAK,UAAA;EAjCA;AAGpB;;;EAoCE,gBAAA;EA3BW;;;;;;;EAoCX,UAAA,GAAa,UAAA;EAgDE;;;EA3Cf,eAAA;EAsEkC;;;EAjElC,cAAA;EA9CW;;;;;;;;;;EA0DX,iBAAA,GAAoB,iBAAA;EAjBpB;;;EAsBA,MAAA,GAAS,eAAA;EAAT;;;;;;EAQA,YAAA,GAAe,oBAAA;EAmBf;;;EAdA,OAAA;EA6BA;;;;;;EArBA,YAAA,GAAe,cAAA;EA2BmB;AAGpC;;;EAxBE,YAAA;EAyBiC;;;EApBjC,WAAA;EAsByB;;;EAjBzB,OAAA,GAAU,OAAA;EAeuB;;;EAVjC,YAAA,IAAgB,MAAA;IAAU,MAAA;IAAgB,SAAA;EAAA;EA2B1C;;;;EArBA,gBAAA,GAAmB,eAAA;AAAA;AAAA,UAGJ,iBAAA,qBACK,UAAA,GAAa,UAAA,qDAEzB,iBAAA;EA4BK;;;EAxBb,UAAA,GAAa,WAAA;EAoB6C;;;EAf1D,eAAA,GAAkB,UAAA;EAmBL;;;;EAbb,eAAA;EA8BA;;;AAGF;EA3BE,YAAA;AAAA;AAAA,UAGe,iBAAA,SAA0B,iBAAA;EAwBC;;;EApB1C,UAAA,GAAa,UAAA,GAAa,KAAA,CAAM,UAAA;EA4ChC;;;EAvCA,eAAA;EA+Ce;;;;EAzCf,eAAA;EAqD0B;;;;EA/C1B,YAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EAuC1C;;;;EAlCA,QAAA;EA4CA;;;;EAtCA,QAAA;EAoDO;AAGT;;;;;EA/CE,cAAA;;;;EAKA,OAAA;;;;EAKA,OAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EAC1C,QAAA;EACA,OAAA;;;;EAKA,QAAA;;;;EAKA,UAAA,GAAa,UAAA,GAAa,KAAA,CAAM,UAAA;;;;EAKhC,eAAA;;;;EAKA,eAAA;;;;;;;EAQA,cAAA;EACA,OAAA;AAAA;AAAA,UAGe,eAAA,SAAwB,iBAAA;;;;EAIvC,UAAA;AAAA"}

package/dist/types.d.ts

@@ -169,12 +169,23 @@ interface RunsCreatePayload extends RunsInvokePayload {
    * If true, the stream can be resumed and replayed in its entirety even after disconnection.
    */
   streamResumable?: boolean;
+  /**
+   * Pass one or more feedbackKeys if you want to request short-lived signed URLs
+   * for submitting feedback to LangSmith with this key for this run.
+   */
+  feedbackKeys?: string[];
 }
 interface CronsCreatePayload extends RunsCreatePayload {
   /**
-   * Schedule for running the Cron Job. Schedules are interpreted in UTC.
+   * Schedule for running the Cron Job. Schedules are interpreted in UTC unless
+   * a timezone is specified.
    */
   schedule: string;
+  /**
+   * IANA timezone string for interpreting the schedule (e.g. "America/New_York").
+   * If not provided, the schedule is interpreted in UTC.
+   */
+  timezone?: string;
   /**
    * What to do with the thread after the run completes.
    * - "delete" (default): Automatically deletes the thread after the run completes.
@@ -186,10 +197,30 @@ interface CronsCreatePayload extends RunsCreatePayload {
    * Whether the cron is enabled.
    */
   enabled?: boolean;
+  /**
+   * The end date to stop running the cron job (ISO 8601 datetime string).
+   */
+  endTime?: string;
 }
 interface CronsUpdatePayload extends RunsInvokePayload {
   schedule?: string;
   endTime?: string;
+  /**
+   * IANA timezone string for interpreting the schedule (e.g. "America/New_York").
+   */
+  timezone?: string;
+  /**
+   * One of `"values"`, `"messages"`, `"messages-tuple"`, `"updates"`, `"events"`, `"debug"`, `"custom"`.
+   */
+  streamMode?: StreamMode | Array<StreamMode>;
+  /**
+   * Stream output from subgraphs. By default, streams only the top graph.
+   */
+  streamSubgraphs?: boolean;
+  /**
+   * Whether the stream is considered resumable.
+   */
+  streamResumable?: boolean;
   /**
    * What to do with the thread after the run completes.
    * - "delete" (default): Automatically deletes the thread after the run completes.

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;KAIY,iBAAA;AAAA,KACA,kBAAA;AAAA,KACA,oBAAA;AAAA,KACA,cAAA;AAAA,KACA,UAAA;AAAA,KACA,WAAA;AAAA,UAYK,IAAA;EACf,IAAA;EACA,KAAA;AAAA;AAAA,UAGe,OAAA;EArBa;AAC9B;;EAwBE,MAAA,GAAS,MAAA;EAxBqB;;AAChC;EA4BE,MAAA;;;;AA3BF;;EAkCE,IAAA,GAAO,IAAA,GAAO,IAAA;AAAA;AAAA,UAGC,iBAAA;EApCL;;;EAwCV,KAAA,GAAQ,MAAA;EAxCa;AAYvB;;EAiCE,QAAA,GAAW,QAAA;EAhCX;;AAIF;EAiCE,MAAA,GAAS,MAAA;;;;;EAMT,OAAA;EAvBkB;;;EA4BlB,YAAA;EA5BA;;;EAiCA,UAAA,GAAa,IAAA,CAAK,UAAA;EAjCA;AAGpB;;;EAoCE,gBAAA;EA3BW;;;;;;;EAoCX,UAAA,GAAa,UAAA;EAgDE;;;EA3Cf,eAAA;EAsEkC;;;EAjElC,cAAA;EA9CW;;;;;;;;;;EA0DX,iBAAA,GAAoB,iBAAA;EAjBpB;;;EAsBA,MAAA,GAAS,eAAA;EAAT;;;;;;EAQA,YAAA,GAAe,oBAAA;EAmBf;;;EAdA,OAAA;EA6BA;;;;;;EArBA,YAAA,GAAe,cAAA;EA2BmB;AAGpC;;;EAxBE,YAAA;EAyBiC;;;EApBjC,WAAA;EAsByB;;;EAjBzB,OAAA,GAAU,OAAA;EAeuB;;;EAVjC,YAAA,IAAgB,MAAA;IAAU,MAAA;IAAgB,SAAA;EAAA;EA2B1C;;;;EArBA,gBAAA,GAAmB,eAAA;AAAA;AAAA,UAGJ,iBAAA,qBACK,UAAA,GAAa,UAAA,qDAEzB,iBAAA;EA4BK;;;EAxBb,UAAA,GAAa,WAAA;EAoB6C;;;EAf1D,eAAA,GAAkB,UAAA;EAmBL;;;;EAbb,eAAA;EAwBe;;AAGjB;;EArBE,YAAA;AAAA;AAAA,UAGe,iBAAA,SAA0B,iBAAA;EAsBzC;;;EAlBA,UAAA,GAAa,UAAA,GAAa,KAAA,CAAM,UAAA;EA+BzB;AAGT;;EA7BE,eAAA;EA6B2D;;;;EAvB3D,eAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EA8BnC;AAGT;;EA7BE,QAAA;EA6BuC;;;;;;EArBvC,cAAA;;;;EAKA,OAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EAC1C,QAAA;EACA,OAAA;;;;;;;EAOA,cAAA;EACA,OAAA;AAAA;AAAA,UAGe,eAAA,SAAwB,iBAAA;;;;EAIvC,UAAA;AAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;KAIY,iBAAA;AAAA,KACA,kBAAA;AAAA,KACA,oBAAA;AAAA,KACA,cAAA;AAAA,KACA,UAAA;AAAA,KACA,WAAA;AAAA,UAYK,IAAA;EACf,IAAA;EACA,KAAA;AAAA;AAAA,UAGe,OAAA;EArBa;AAC9B;;EAwBE,MAAA,GAAS,MAAA;EAxBqB;;AAChC;EA4BE,MAAA;;;;AA3BF;;EAkCE,IAAA,GAAO,IAAA,GAAO,IAAA;AAAA;AAAA,UAGC,iBAAA;EApCL;;;EAwCV,KAAA,GAAQ,MAAA;EAxCa;AAYvB;;EAiCE,QAAA,GAAW,QAAA;EAhCX;;AAIF;EAiCE,MAAA,GAAS,MAAA;;;;;EAMT,OAAA;EAvBkB;;;EA4BlB,YAAA;EA5BA;;;EAiCA,UAAA,GAAa,IAAA,CAAK,UAAA;EAjCA;AAGpB;;;EAoCE,gBAAA;EA3BW;;;;;;;EAoCX,UAAA,GAAa,UAAA;EAgDE;;;EA3Cf,eAAA;EAsEkC;;;EAjElC,cAAA;EA9CW;;;;;;;;;;EA0DX,iBAAA,GAAoB,iBAAA;EAjBpB;;;EAsBA,MAAA,GAAS,eAAA;EAAT;;;;;;EAQA,YAAA,GAAe,oBAAA;EAmBf;;;EAdA,OAAA;EA6BA;;;;;;EArBA,YAAA,GAAe,cAAA;EA2BmB;AAGpC;;;EAxBE,YAAA;EAyBiC;;;EApBjC,WAAA;EAsByB;;;EAjBzB,OAAA,GAAU,OAAA;EAeuB;;;EAVjC,YAAA,IAAgB,MAAA;IAAU,MAAA;IAAgB,SAAA;EAAA;EA2B1C;;;;EArBA,gBAAA,GAAmB,eAAA;AAAA;AAAA,UAGJ,iBAAA,qBACK,UAAA,GAAa,UAAA,qDAEzB,iBAAA;EA4BK;;;EAxBb,UAAA,GAAa,WAAA;EAoB6C;;;EAf1D,eAAA,GAAkB,UAAA;EAmBL;;;;EAbb,eAAA;EA8BA;;;AAGF;EA3BE,YAAA;AAAA;AAAA,UAGe,iBAAA,SAA0B,iBAAA;EAwBC;;;EApB1C,UAAA,GAAa,UAAA,GAAa,KAAA,CAAM,UAAA;EA4ChC;;;EAvCA,eAAA;EA+Ce;;;;EAzCf,eAAA;EAqD0B;;;;EA/C1B,YAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EAuC1C;;;;EAlCA,QAAA;EA4CA;;;;EAtCA,QAAA;EAoDO;AAGT;;;;;EA/CE,cAAA;;;;EAKA,OAAA;;;;EAKA,OAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,iBAAA;EAC1C,QAAA;EACA,OAAA;;;;EAKA,QAAA;;;;EAKA,UAAA,GAAa,UAAA,GAAa,KAAA,CAAM,UAAA;;;;EAKhC,eAAA;;;;EAKA,eAAA;;;;;;;EAQA,cAAA;EACA,OAAA;AAAA;AAAA,UAGe,eAAA,SAAwB,iBAAA;;;;EAIvC,UAAA;AAAA"}

package/dist/ui/branching.cjs

@@ -152,7 +152,7 @@ function getMessagesMetadataMap(options) {
 	const alreadyShown = /* @__PURE__ */ new Set();
 	return options.getMessages(currentValues).map((message, idx) => {
 		const messageId = message.id ?? idx;
-		const firstSeenState = require_utils.findLast(options.history ?? [], (state) => options.getMessages(state.values).map((m, idx) => m.id ?? idx).includes(messageId));
+		const firstSeenState = require_utils.findLast(options.history ?? [], (state) => state.values != null && 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;

package/dist/ui/branching.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"branching.cjs","names":["findLast"],"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,KAAA;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,iBAAiBA,cAAAA,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,KAAA;AACN,MAAI,CAAC,QAAQ,QAAQ,OAAQ,UAAS,KAAA;EAGtC,MAAM,eAAe,QAAQ,eAAe,KAAK,EAAE,CAAC,KAAK,IAAI;AAC7D,MAAI,cAAc;AAChB,OAAI,aAAa,IAAI,aAAa,CAAE,UAAS,KAAA;AAC7C,gBAAa,IAAI,aAAa;;AAGhC,SAAO;GACL,WAAW,UAAU,UAAU;GAC/B;GAEA,QAAQ,QAAQ;GAChB,eAAe,QAAQ;GACxB;GACD"}
+{"version":3,"file":"branching.cjs","names":["findLast"],"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(\n      options.history ?? [],\n      (state) =>\n        state.values != null &&\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,KAAA;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,iBAAiBA,cAAAA,SACrB,QAAQ,WAAW,EAAE,GACpB,UACC,MAAM,UAAU,QAChB,QACG,YAAY,MAAM,OAAO,CACzB,KAAK,GAAG,QAAQ,EAAE,MAAM,IAAI,CAC5B,SAAS,UAAU,CACzB;EAED,MAAM,eAAe,gBAAgB,YAAY;EACjD,IAAI,SACF,gBAAgB,OACZ,QAAQ,cAAc,mBAAmB,gBACzC,KAAA;AACN,MAAI,CAAC,QAAQ,QAAQ,OAAQ,UAAS,KAAA;EAGtC,MAAM,eAAe,QAAQ,eAAe,KAAK,EAAE,CAAC,KAAK,IAAI;AAC7D,MAAI,cAAc;AAChB,OAAI,aAAa,IAAI,aAAa,CAAE,UAAS,KAAA;AAC7C,gBAAa,IAAI,aAAa;;AAGhC,SAAO;GACL,WAAW,UAAU,UAAU;GAC/B;GAEA,QAAQ,QAAQ;GAChB,eAAe,QAAQ;GACxB;GACD"}

package/dist/ui/branching.js

@@ -152,7 +152,7 @@ function getMessagesMetadataMap(options) {
 	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 firstSeenState = findLast(options.history ?? [], (state) => state.values != null && 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;

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\";\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,KAAA;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,KAAA;AACN,MAAI,CAAC,QAAQ,QAAQ,OAAQ,UAAS,KAAA;EAGtC,MAAM,eAAe,QAAQ,eAAe,KAAK,EAAE,CAAC,KAAK,IAAI;AAC7D,MAAI,cAAc;AAChB,OAAI,aAAa,IAAI,aAAa,CAAE,UAAS,KAAA;AAC7C,gBAAa,IAAI,aAAa;;AAGhC,SAAO;GACL,WAAW,UAAU,UAAU;GAC/B;GAEA,QAAQ,QAAQ;GAChB,eAAe,QAAQ;GACxB;GACD"}
+{"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(\n      options.history ?? [],\n      (state) =>\n        state.values != null &&\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,KAAA;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,SACrB,QAAQ,WAAW,EAAE,GACpB,UACC,MAAM,UAAU,QAChB,QACG,YAAY,MAAM,OAAO,CACzB,KAAK,GAAG,QAAQ,EAAE,MAAM,IAAI,CAC5B,SAAS,UAAU,CACzB;EAED,MAAM,eAAe,gBAAgB,YAAY;EACjD,IAAI,SACF,gBAAgB,OACZ,QAAQ,cAAc,mBAAmB,gBACzC,KAAA;AACN,MAAI,CAAC,QAAQ,QAAQ,OAAQ,UAAS,KAAA;EAGtC,MAAM,eAAe,QAAQ,eAAe,KAAK,EAAE,CAAC,KAAK,IAAI;AAC7D,MAAI,cAAc;AAChB,OAAI,aAAa,IAAI,aAAa,CAAE,UAAS,KAAA;AAC7C,gBAAa,IAAI,aAAa;;AAGhC,SAAO;GACL,WAAW,UAAU,UAAU;GAC/B;GAEA,QAAQ,QAAQ;GAChB,eAAe,QAAQ;GACxB;GACD"}

package/dist/ui/index.d.cts

@@ -6,10 +6,10 @@ 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 { SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace } from "./subagents.cjs";
 import { HistoryWithBaseMessages, MessageTupleManager, ensureHistoryMessageInstances, 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, type HistoryWithBaseMessages, 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, ensureHistoryMessageInstances, ensureMessageInstances, extractInterrupts, extractParentIdFromNamespace, extractToolCallIdFromNamespace, filterStream, findLast, getBranchContext, getMessagesMetadataMap, isSubagentNamespace, toMessageClass, toMessageDict, unique };

package/dist/ui/index.d.ts

@@ -6,10 +6,10 @@ 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 { SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace } from "./subagents.js";
 import { HistoryWithBaseMessages, MessageTupleManager, ensureHistoryMessageInstances, 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, type HistoryWithBaseMessages, 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, ensureHistoryMessageInstances, ensureMessageInstances, extractInterrupts, extractParentIdFromNamespace, extractToolCallIdFromNamespace, filterStream, findLast, getBranchContext, getMessagesMetadataMap, isSubagentNamespace, toMessageClass, toMessageDict, unique };

package/dist/ui/messages.cjs

@@ -86,6 +86,7 @@ function ensureMessageInstances(messages) {
 */
 function ensureHistoryMessageInstances(history, messagesKey = "messages") {
 	return history.map((state) => {
+		if (state.values == null) return state;
 		const messages = state.values[messagesKey];
 		if (!Array.isArray(messages)) return state;
 		return {

package/dist/ui/messages.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"messages.cjs","names":["HumanMessageChunk","AIMessageChunk","SystemMessageChunk","ToolMessageChunk","RemoveMessage"],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  isBaseMessageChunk,\n  HumanMessageChunk,\n  SystemMessageChunk,\n  AIMessageChunk,\n  ToolMessageChunk,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends Record<string, unknown>\n  ? \"messages\" extends keyof S\n    ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n    : S\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (isBaseMessageChunk(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(message);\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToMessage(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;;AAmCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,OAAA,GAAA,yBAAA,oBAAuB,QAAQ,CAAE,QAAO;AACxC,UAAA,GAAA,yBAAA,gBAAsB,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAIA,yBAAAA,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAIC,yBAAAA,eAAe,QAAQ;AAGpC,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAIC,yBAAAA,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAIC,yBAAAA,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAIC,yBAAAA,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,SAAA,GAAA,yBAAA,4BAAkC,QAAQ;;AAG5C,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,8BAA8B,WAAW;EACzD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,UAAA,GAAA,yBAAA,oBACM,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAAI;QAE7D,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;EAC5B,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}
+{"version":3,"file":"messages.cjs","names":["HumanMessageChunk","AIMessageChunk","SystemMessageChunk","ToolMessageChunk","RemoveMessage"],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  isBaseMessageChunk,\n  HumanMessageChunk,\n  SystemMessageChunk,\n  AIMessageChunk,\n  ToolMessageChunk,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends Record<string, unknown>\n  ? \"messages\" extends keyof S\n    ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n    : S\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (isBaseMessageChunk(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(message);\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToMessage(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    if (state.values == null) return state;\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;;AAmCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,OAAA,GAAA,yBAAA,oBAAuB,QAAQ,CAAE,QAAO;AACxC,UAAA,GAAA,yBAAA,gBAAsB,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAIA,yBAAAA,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAIC,yBAAAA,eAAe,QAAQ;AAGpC,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAIC,yBAAAA,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAIC,yBAAAA,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAIC,yBAAAA,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,SAAA,GAAA,yBAAA,4BAAkC,QAAQ;;AAG5C,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,8BAA8B,WAAW;EACzD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,UAAA,GAAA,yBAAA,oBACM,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAAI;QAE7D,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;AAC5B,MAAI,MAAM,UAAU,KAAM,QAAO;EACjC,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}

package/dist/ui/messages.js

@@ -85,6 +85,7 @@ function ensureMessageInstances(messages) {
 */
 function ensureHistoryMessageInstances(history, messagesKey = "messages") {
 	return history.map((state) => {
+		if (state.values == null) return state;
 		const messages = state.values[messagesKey];
 		if (!Array.isArray(messages)) return state;
 		return {

package/dist/ui/messages.js.map

@@ -1 +1 @@
-{"version":3,"file":"messages.js","names":[],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  isBaseMessageChunk,\n  HumanMessageChunk,\n  SystemMessageChunk,\n  AIMessageChunk,\n  ToolMessageChunk,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends Record<string, unknown>\n  ? \"messages\" extends keyof S\n    ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n    : S\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (isBaseMessageChunk(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(message);\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToMessage(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;AAmCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,MAAI,mBAAmB,QAAQ,CAAE,QAAO;AACxC,SAAO,eAAe,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAI,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAI,eAAe,QAAQ;AAGpC,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAI,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAI,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAI,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,QAAO,2BAA2B,QAAQ;;AAG5C,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,8BAA8B,WAAW;EACzD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,SACb,mBAAmB,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAAI;QAE7D,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;EAC5B,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}
+{"version":3,"file":"messages.js","names":[],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  isBaseMessageChunk,\n  HumanMessageChunk,\n  SystemMessageChunk,\n  AIMessageChunk,\n  ToolMessageChunk,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends Record<string, unknown>\n  ? \"messages\" extends keyof S\n    ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n    : S\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (isBaseMessageChunk(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(message);\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToMessage(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    if (state.values == null) return state;\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;AAmCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,MAAI,mBAAmB,QAAQ,CAAE,QAAO;AACxC,SAAO,eAAe,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAI,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAI,eAAe,QAAQ;AAGpC,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAI,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAI,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAI,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,QAAO,2BAA2B,QAAQ;;AAG5C,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,8BAA8B,WAAW;EACzD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,SACb,mBAAmB,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAAI;QAE7D,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;AAC5B,MAAI,MAAM,UAAU,KAAM,QAAO;EACjC,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}

package/dist/ui/subagents.cjs

@@ -312,9 +312,11 @@ var SubagentManager = class {
 				const oldDesc = existing.toolCall.args.description || "";
 				const shouldUpdateType = this.isValidSubagentType(newType) && newType.length > oldType.length;
 				const shouldUpdateDesc = newDesc.length > oldDesc.length;
-				if (shouldUpdateType || shouldUpdateDesc) {
+				const shouldUpdateMessageId = aiMessageId != null && aiMessageId !== existing.aiMessageId;
+				if (shouldUpdateType || shouldUpdateDesc || shouldUpdateMessageId) {
 					this.subagents.set(toolCall.id, {
 						...existing,
+						...shouldUpdateMessageId ? { aiMessageId } : {},
 						toolCall: {
 							...existing.toolCall,
 							args: {