@assistant-ui/react-langchain 0.0.34 → 0.0.36

package/dist/langChainThreadState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langChainThreadState.js","names":[],"sources":["../src/langChainThreadState.ts"],"sourcesContent":["import type { AppendMessage } from \"@assistant-ui/core\";\nimport type { LangChainBaseMessage } from \"./types\";\nimport { getMessageType } from \"./convertMessages\";\n\nexport type StagedMessage = LangChainBaseMessage & { id: string };\n\nexport type StagedEntry = {\n  message: StagedMessage;\n  runConfig: AppendMessage[\"runConfig\"];\n  reconcileOnEcho: boolean;\n  baseMessageCount: number;\n  transcriptStatus?: \"unsent\" | \"sent\";\n};\n\nexport type LangChainThreadState = {\n  forkGeneration: number;\n  forkPending: boolean;\n  stagedEntries: ReadonlyMap<string, StagedEntry>;\n  stagedBaseMessages: LangChainBaseMessage[] | null;\n  visibleStagedMessages: LangChainBaseMessage[] | null;\n};\n\nexport type LangChainThreadAction =\n  | { type: \"supersedeFork\" }\n  | { type: \"finishFork\"; generation: number }\n  | { type: \"cancelFork\" }\n  | {\n      type: \"stage\";\n      entry: StagedEntry;\n      visibleMessages: readonly LangChainBaseMessage[];\n    }\n  | {\n      type: \"stageEdit\";\n      entry: StagedEntry;\n      baseMessages: LangChainBaseMessage[];\n    }\n  | {\n      type: \"reconcile\";\n      messages: LangChainBaseMessage[];\n      visibleMessages: readonly LangChainBaseMessage[];\n    }\n  | {\n      type: \"markTranscript\";\n      messages: readonly LangChainBaseMessage[];\n      status: \"unsent\" | \"sent\";\n    }\n  | {\n      type: \"remove\";\n      id: string;\n      visibleMessages: readonly LangChainBaseMessage[];\n    }\n  | {\n      type: \"promote\";\n      messages: readonly LangChainBaseMessage[];\n      visibleMessages: readonly LangChainBaseMessage[];\n    };\n\nexport const createLangChainThreadState = (): LangChainThreadState => ({\n  forkGeneration: 0,\n  forkPending: false,\n  stagedEntries: new Map(),\n  stagedBaseMessages: null,\n  visibleStagedMessages: null,\n});\n\nconst humanContentText = (content: LangChainBaseMessage[\"content\"]) => {\n  if (typeof content === \"string\") return content;\n  if (!Array.isArray(content)) return \"\";\n  return content\n    .filter(\n      (part): part is { type: \"text\"; text: string } =>\n        typeof part === \"object\" &&\n        part !== null &&\n        part.type === \"text\" &&\n        typeof part.text === \"string\",\n    )\n    .map((part) => part.text)\n    .join(\"\");\n};\n\nconst hasSameMessageContent = (\n  a: LangChainBaseMessage,\n  b: LangChainBaseMessage,\n) => humanContentText(a.content) === humanContentText(b.content);\n\nexport const reduceLangChainThreadState = (\n  state: LangChainThreadState,\n  action: LangChainThreadAction,\n): LangChainThreadState => {\n  switch (action.type) {\n    case \"supersedeFork\":\n      return {\n        ...state,\n        forkGeneration: state.forkGeneration + 1,\n        forkPending: true,\n      };\n    case \"finishFork\":\n      return state.forkGeneration === action.generation\n        ? { ...state, forkPending: false }\n        : state;\n    case \"cancelFork\":\n      return {\n        ...state,\n        forkGeneration: state.forkGeneration + 1,\n        forkPending: false,\n      };\n    case \"stage\": {\n      const stagedEntries = new Map(state.stagedEntries);\n      stagedEntries.set(action.entry.message.id, action.entry);\n      return {\n        ...state,\n        stagedEntries,\n        visibleStagedMessages: [\n          ...action.visibleMessages,\n          action.entry.message,\n        ],\n      };\n    }\n    case \"stageEdit\": {\n      const stagedEntries = new Map(state.stagedEntries);\n      stagedEntries.set(action.entry.message.id, action.entry);\n      return {\n        ...state,\n        stagedEntries,\n        stagedBaseMessages: action.baseMessages,\n        visibleStagedMessages: [...action.baseMessages, action.entry.message],\n      };\n    }\n    case \"reconcile\": {\n      if (state.stagedEntries.size === 0) return state;\n      // Staged edits must keep their truncated base while stream updates arrive before promotion.\n      const baseMessages = state.stagedBaseMessages ?? action.messages;\n      const stagedEntries = new Map(state.stagedEntries);\n      const remainingStagedMessages: LangChainBaseMessage[] = [];\n      const matchedBaseMessageIndexes = new Set<number>();\n      const visibleStagedIds = new Set(\n        action.visibleMessages.flatMap((message) =>\n          message.id ? [message.id] : [],\n        ),\n      );\n      for (const [id, staged] of state.stagedEntries) {\n        if (!visibleStagedIds.has(id)) continue;\n        const echoed = baseMessages.some((message, index) => {\n          if (matchedBaseMessageIndexes.has(index)) return false;\n          if (message.id === id) {\n            matchedBaseMessageIndexes.add(index);\n            return true;\n          }\n          if (\n            !staged.reconcileOnEcho ||\n            index < staged.baseMessageCount ||\n            getMessageType(message) !== \"human\" ||\n            !hasSameMessageContent(message, staged.message)\n          ) {\n            return false;\n          }\n          matchedBaseMessageIndexes.add(index);\n          return true;\n        });\n        if (echoed) stagedEntries.delete(id);\n        else remainingStagedMessages.push(staged.message);\n      }\n      if (remainingStagedMessages.length === 0) {\n        if (\n          stagedEntries.size === state.stagedEntries.size &&\n          state.stagedBaseMessages === null &&\n          state.visibleStagedMessages === null\n        ) {\n          return state;\n        }\n        return {\n          ...state,\n          stagedEntries,\n          stagedBaseMessages: null,\n          visibleStagedMessages: null,\n        };\n      }\n      return {\n        ...state,\n        stagedEntries,\n        visibleStagedMessages: [...baseMessages, ...remainingStagedMessages],\n      };\n    }\n    case \"markTranscript\": {\n      let stagedEntries: Map<string, StagedEntry> | null = null;\n      for (const message of action.messages) {\n        const entry = message.id\n          ? (stagedEntries ?? state.stagedEntries).get(message.id)\n          : undefined;\n        if (entry?.transcriptStatus) {\n          stagedEntries ??= new Map(state.stagedEntries);\n          stagedEntries.set(message.id!, {\n            ...entry,\n            transcriptStatus: action.status,\n          });\n        }\n      }\n      return stagedEntries ? { ...state, stagedEntries } : state;\n    }\n    case \"remove\": {\n      if (!state.stagedEntries.has(action.id)) return state;\n      const stagedEntries = new Map(state.stagedEntries);\n      stagedEntries.delete(action.id);\n      return {\n        ...state,\n        stagedEntries,\n        stagedBaseMessages:\n          stagedEntries.size === 0 ? null : state.stagedBaseMessages,\n        visibleStagedMessages:\n          stagedEntries.size === 0\n            ? null\n            : action.visibleMessages.filter(\n                (message) => message.id !== action.id,\n              ),\n      };\n    }\n    case \"promote\": {\n      const stagedEntries = new Map(state.stagedEntries);\n      const promotedIds = new Set<string>();\n      for (const message of action.messages) {\n        if (!message.id || stagedEntries.get(message.id)?.transcriptStatus)\n          continue;\n        promotedIds.add(message.id);\n        stagedEntries.delete(message.id);\n      }\n      return {\n        ...state,\n        stagedEntries,\n        stagedBaseMessages: null,\n        visibleStagedMessages:\n          stagedEntries.size > 0\n            ? action.visibleMessages.filter(\n                (message) => !message.id || !promotedIds.has(message.id),\n              )\n            : null,\n      };\n    }\n  }\n};\n"],"mappings":";;AAyDA,MAAa,oCAA0D;CACrE,gBAAgB;CAChB,aAAa;CACb,+BAAe,IAAI,IAAI;CACvB,oBAAoB;CACpB,uBAAuB;AACzB;AAEA,MAAM,oBAAoB,YAA6C;CACrE,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;CACpC,OAAO,QACJ,QACE,SACC,OAAO,SAAS,YAChB,SAAS,QACT,KAAK,SAAS,UACd,OAAO,KAAK,SAAS,QACzB,CAAC,CACA,KAAK,SAAS,KAAK,IAAI,CAAC,CACxB,KAAK,EAAE;AACZ;AAEA,MAAM,yBACJ,GACA,MACG,iBAAiB,EAAE,OAAO,MAAM,iBAAiB,EAAE,OAAO;AAE/D,MAAa,8BACX,OACA,WACyB;CACzB,QAAQ,OAAO,MAAf;EACE,KAAK,iBACH,OAAO;GACL,GAAG;GACH,gBAAgB,MAAM,iBAAiB;GACvC,aAAa;EACf;EACF,KAAK,cACH,OAAO,MAAM,mBAAmB,OAAO,aACnC;GAAE,GAAG;GAAO,aAAa;EAAM,IAC/B;EACN,KAAK,cACH,OAAO;GACL,GAAG;GACH,gBAAgB,MAAM,iBAAiB;GACvC,aAAa;EACf;EACF,KAAK,SAAS;GACZ,MAAM,gBAAgB,IAAI,IAAI,MAAM,aAAa;GACjD,cAAc,IAAI,OAAO,MAAM,QAAQ,IAAI,OAAO,KAAK;GACvD,OAAO;IACL,GAAG;IACH;IACA,uBAAuB,CACrB,GAAG,OAAO,iBACV,OAAO,MAAM,OACf;GACF;EACF;EACA,KAAK,aAAa;GAChB,MAAM,gBAAgB,IAAI,IAAI,MAAM,aAAa;GACjD,cAAc,IAAI,OAAO,MAAM,QAAQ,IAAI,OAAO,KAAK;GACvD,OAAO;IACL,GAAG;IACH;IACA,oBAAoB,OAAO;IAC3B,uBAAuB,CAAC,GAAG,OAAO,cAAc,OAAO,MAAM,OAAO;GACtE;EACF;EACA,KAAK,aAAa;GAChB,IAAI,MAAM,cAAc,SAAS,GAAG,OAAO;GAE3C,MAAM,eAAe,MAAM,sBAAsB,OAAO;GACxD,MAAM,gBAAgB,IAAI,IAAI,MAAM,aAAa;GACjD,MAAM,0BAAkD,CAAC;GACzD,MAAM,4CAA4B,IAAI,IAAY;GAClD,MAAM,mBAAmB,IAAI,IAC3B,OAAO,gBAAgB,SAAS,YAC9B,QAAQ,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAC/B,CACF;GACA,KAAK,MAAM,CAAC,IAAI,WAAW,MAAM,eAAe;IAC9C,IAAI,CAAC,iBAAiB,IAAI,EAAE,GAAG;IAkB/B,IAjBe,aAAa,MAAM,SAAS,UAAU;KACnD,IAAI,0BAA0B,IAAI,KAAK,GAAG,OAAO;KACjD,IAAI,QAAQ,OAAO,IAAI;MACrB,0BAA0B,IAAI,KAAK;MACnC,OAAO;KACT;KACA,IACE,CAAC,OAAO,mBACR,QAAQ,OAAO,oBACf,eAAe,OAAO,MAAM,WAC5B,CAAC,sBAAsB,SAAS,OAAO,OAAO,GAE9C,OAAO;KAET,0BAA0B,IAAI,KAAK;KACnC,OAAO;IACT,CACS,GAAG,cAAc,OAAO,EAAE;SAC9B,wBAAwB,KAAK,OAAO,OAAO;GAClD;GACA,IAAI,wBAAwB,WAAW,GAAG;IACxC,IACE,cAAc,SAAS,MAAM,cAAc,QAC3C,MAAM,uBAAuB,QAC7B,MAAM,0BAA0B,MAEhC,OAAO;IAET,OAAO;KACL,GAAG;KACH;KACA,oBAAoB;KACpB,uBAAuB;IACzB;GACF;GACA,OAAO;IACL,GAAG;IACH;IACA,uBAAuB,CAAC,GAAG,cAAc,GAAG,uBAAuB;GACrE;EACF;EACA,KAAK,kBAAkB;GACrB,IAAI,gBAAiD;GACrD,KAAK,MAAM,WAAW,OAAO,UAAU;IACrC,MAAM,QAAQ,QAAQ,MACjB,iBAAiB,MAAM,cAAA,CAAe,IAAI,QAAQ,EAAE,IACrD,KAAA;IACJ,IAAI,OAAO,kBAAkB;KAC3B,kBAAkB,IAAI,IAAI,MAAM,aAAa;KAC7C,cAAc,IAAI,QAAQ,IAAK;MAC7B,GAAG;MACH,kBAAkB,OAAO;KAC3B,CAAC;IACH;GACF;GACA,OAAO,gBAAgB;IAAE,GAAG;IAAO;GAAc,IAAI;EACvD;EACA,KAAK,UAAU;GACb,IAAI,CAAC,MAAM,cAAc,IAAI,OAAO,EAAE,GAAG,OAAO;GAChD,MAAM,gBAAgB,IAAI,IAAI,MAAM,aAAa;GACjD,cAAc,OAAO,OAAO,EAAE;GAC9B,OAAO;IACL,GAAG;IACH;IACA,oBACE,cAAc,SAAS,IAAI,OAAO,MAAM;IAC1C,uBACE,cAAc,SAAS,IACnB,OACA,OAAO,gBAAgB,QACpB,YAAY,QAAQ,OAAO,OAAO,EACrC;GACR;EACF;EACA,KAAK,WAAW;GACd,MAAM,gBAAgB,IAAI,IAAI,MAAM,aAAa;GACjD,MAAM,8BAAc,IAAI,IAAY;GACpC,KAAK,MAAM,WAAW,OAAO,UAAU;IACrC,IAAI,CAAC,QAAQ,MAAM,cAAc,IAAI,QAAQ,EAAE,CAAC,EAAE,kBAChD;IACF,YAAY,IAAI,QAAQ,EAAE;IAC1B,cAAc,OAAO,QAAQ,EAAE;GACjC;GACA,OAAO;IACL,GAAG;IACH;IACA,oBAAoB;IACpB,uBACE,cAAc,OAAO,IACjB,OAAO,gBAAgB,QACpB,YAAY,CAAC,QAAQ,MAAM,CAAC,YAAY,IAAI,QAAQ,EAAE,CACzD,IACA;GACR;EACF;CACF;AACF"}

package/dist/sdkIdentity.js

@@ -1,7 +1,7 @@
 //#region src/sdkIdentity.ts
 const LANGCHAIN_SDK = {
 	name: "@assistant-ui/react-langchain",
-	version: "0.0.34"
+	version: "0.0.36"
 };
 //#endregion
 export { LANGCHAIN_SDK };

package/dist/types.d.ts

@@ -89,6 +89,11 @@ export type LangChainRuntimeExtraOptions = ExternalStoreSharedOptions & {
      */
     onThreadIdChange?: ((threadId: string | undefined) => void) | undefined;
     cloud?: AssistantCloud | undefined;
+    /**
+     * Stable identity for the account or workspace owning Cloud runtime state.
+     * Provide it from the first render and change it when that scope changes.
+     */
+    scopeId?: string | undefined;
     adapters?: {
         attachments?: AttachmentAdapter | undefined;
         speech?: SpeechSynthesisAdapter | undefined;
@@ -106,11 +111,15 @@ export type LangChainRuntimeExtraOptions = ExternalStoreSharedOptions & {
     /**
      * Routes the Cancel button's click to `useStream().stop()`. On by
      * default. Pass `false` to disable the Cancel button.
+     *
+     * @deprecated Experimental since 2025-01-03. Not scheduled for removal; the API may change in any release.
      */
     unstable_allowCancellation?: boolean | undefined;
     /**
      * Custom `RemoteThreadListAdapter`. When provided, replaces the
      * cloud-backed thread list adapter.
+     *
+     * @deprecated Experimental since 2026-04-16. Not scheduled for removal; the API may change in any release.
      */
     unstable_threadListAdapter?: RemoteThreadListAdapter | undefined;
     /** Custom thread-creation hook, forwarded to the cloud adapter. */

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,0BAA0B,EAC1B,eAAe,EACf,aAAa,EACb,oBAAoB,EACpB,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EACV,gBAAgB,EAChB,iBAAiB,EACjB,SAAS,EACT,yBAAyB,EACzB,yBAAyB,EAC1B,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EAAE,qBAAqB,EAAE,uBAAoB;AAEzD,MAAM,MAAM,iBAAiB,GAAG;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,SAAS,CACnB,KAAK,SAAS,MAAM,GAAG,MAAM,EAC7B,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAC9D;IACF,IAAI,EAAE,IAAI,CAAC;IACX,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,KAAK,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE;QACT,KAAK,CAAC,EAAE,OAAO,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACxB,CAAC;CACH,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;CACZ,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,EAAE,MAAM,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;IACxD,2BAA2B;IAC3B,UAAU,CAAC,EAAE,SAAS,iBAAiB,EAAE,GAAG,SAAS,CAAC;IACtD,6BAA6B;IAC7B,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,oEAAoE;IACpE,MAAM,CAAC,EAAE,aAAa,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;IACzD,6BAA6B;IAC7B,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,SAAS,EAAE;QAAE,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,SAAS,CAAC;IAC3C,UAAU,EAAE,SAAS;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,EAAE,CAAC;IACxD,SAAS,EAAE,SAAS,iBAAiB,EAAE,CAAC;IACxC,SAAS,EAAE,WAAW,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC;IAC1D,SAAS,EAAE,WAAW,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC;IAC1D,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,CACN,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,SAAS,EAClD,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,OAAO,EAAE,CACP,QAAQ,EAAE,OAAO,EACjB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,UAAU,EAAE,CACV,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,0BAA0B,GAAG;IACtE;;;;;;;OAOG;IACH,gBAAgB,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IACxE,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,GAAG,SAAS,CAAC;QAC5C,MAAM,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;QAC5C,SAAS,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAAC;QACzC,KAAK,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;QACzC,QAAQ,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC;KACxC,GACD,SAAS,CAAC;IACd;;;;;OAKG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;OAGG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;OAGG;IACH,0BAA0B,CAAC,EAAE,uBAAuB,GAAG,SAAS,CAAC;IACjE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,MAAM,OAAO,CAAC;QAAE,UAAU,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC,CAAC,GAAG,SAAS,CAAC;IACzE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;IAC3D;;;;;OAKG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACjC,CAAC;AAMF,MAAM,MAAM,uBAAuB,GAAG,gBAAgB,SAAS,MAAM,CAAC,GAClE,CAAC,SAAS,gBAAgB,GACxB,CAAC,GAAG,4BAA4B,GAChC,KAAK,GACP,KAAK,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,0BAA0B,EAC1B,eAAe,EACf,aAAa,EACb,oBAAoB,EACpB,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EACV,gBAAgB,EAChB,iBAAiB,EACjB,SAAS,EACT,yBAAyB,EACzB,yBAAyB,EAC1B,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EAAE,qBAAqB,EAAE,uBAAoB;AAEzD,MAAM,MAAM,iBAAiB,GAAG;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,SAAS,CACnB,KAAK,SAAS,MAAM,GAAG,MAAM,EAC7B,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAC9D;IACF,IAAI,EAAE,IAAI,CAAC;IACX,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,KAAK,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE;QACT,KAAK,CAAC,EAAE,OAAO,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACxB,CAAC;CACH,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;CACZ,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,EAAE,MAAM,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;IACxD,2BAA2B;IAC3B,UAAU,CAAC,EAAE,SAAS,iBAAiB,EAAE,GAAG,SAAS,CAAC;IACtD,6BAA6B;IAC7B,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,oEAAoE;IACpE,MAAM,CAAC,EAAE,aAAa,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;IACzD,6BAA6B;IAC7B,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,SAAS,EAAE;QAAE,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,SAAS,CAAC;IAC3C,UAAU,EAAE,SAAS;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,EAAE,CAAC;IACxD,SAAS,EAAE,SAAS,iBAAiB,EAAE,CAAC;IACxC,SAAS,EAAE,WAAW,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC;IAC1D,SAAS,EAAE,WAAW,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC;IAC1D,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,CACN,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,SAAS,EAClD,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,OAAO,EAAE,CACP,QAAQ,EAAE,OAAO,EACjB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,UAAU,EAAE,CACV,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,0BAA0B,GAAG;IACtE;;;;;;;OAOG;IACH,gBAAgB,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IACxE,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,GAAG,SAAS,CAAC;QAC5C,MAAM,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;QAC5C,SAAS,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAAC;QACzC,KAAK,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;QACzC,QAAQ,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC;KACxC,GACD,SAAS,CAAC;IACd;;;;;OAKG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;;;OAKG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;;;OAKG;IACH,0BAA0B,CAAC,EAAE,uBAAuB,GAAG,SAAS,CAAC;IACjE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,MAAM,OAAO,CAAC;QAAE,UAAU,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC,CAAC,GAAG,SAAS,CAAC;IACzE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;IAC3D;;;;;OAKG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACjC,CAAC;AAMF,MAAM,MAAM,uBAAuB,GAAG,gBAAgB,SAAS,MAAM,CAAC,GAClE,CAAC,SAAS,gBAAgB,GACxB,CAAC,GAAG,4BAA4B,GAChC,KAAK,GACP,KAAK,CAAC"}

package/dist/useStreamRuntime.d.ts

@@ -1,16 +1,17 @@
 /// <reference types="@assistant-ui/core/store" preserve="true" />
 import type { AppendMessage } from "@assistant-ui/core";
-import type { UseStreamRuntimeOptions } from "./types.js";
+import type { LangChainBaseMessage, LangChainToolCall, UseStreamRuntimeOptions } from "./types.js";
 export { groupUIMessagesByParent } from "./converter.js";
 export declare const runConfigToSubmitOptions: (runConfig: AppendMessage["runConfig"]) => {
     config: {
         configurable: Record<string, unknown>;
     };
 } | undefined;
+export declare const getPendingToolCalls: (messages: readonly LangChainBaseMessage[]) => LangChainToolCall[];
 /**
  * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
  * Accepts the same options as `useStream` from `@langchain/react`, plus
- * `cloud` and `adapters`.
+ * `cloud`, `scopeId`, and `adapters`.
  *
  * @example
  * ```tsx

package/dist/useStreamRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.d.ts","sourceRoot":"","sources":["../src/useStreamRuntime.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EAAE,aAAa,EAAuB,MAAM,oBAAoB,CAAC;AAqB7E,OAAO,KAAK,EAIV,uBAAuB,EACxB,mBAAgB;AAEjB,OAAO,EAAE,uBAAuB,EAAE,uBAAoB;AAwBtD,eAAO,MAAM,wBAAwB,GACnC,WAAW,aAAa,CAAC,WAAW,CAAC;;;;aAIxB,CAAC;AAgsBhB;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,gBAAgB,GAAI,YAAY,uBAAuB,kDA8BnE,CAAC"}
+{"version":3,"file":"useStreamRuntime.d.ts","sourceRoot":"","sources":["../src/useStreamRuntime.ts"],"names":[],"mappings":";AAYA,OAAO,KAAK,EAAE,aAAa,EAAuB,MAAM,oBAAoB,CAAC;AAqB7E,OAAO,KAAK,EACV,oBAAoB,EACpB,iBAAiB,EAEjB,uBAAuB,EACxB,mBAAgB;AAEjB,OAAO,EAAE,uBAAuB,EAAE,uBAAoB;AA4BtD,eAAO,MAAM,wBAAwB,GACnC,WAAW,aAAa,CAAC,WAAW,CAAC;;;;aAIxB,CAAC;AAMhB,eAAO,MAAM,mBAAmB,GAC9B,UAAU,SAAS,oBAAoB,EAAE,KACxC,iBAAiB,EAkBjB,CAAC;AAmoBJ;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,gBAAgB,GAAI,YAAY,uBAAuB,kDAgCnE,CAAC"}

package/dist/useStreamRuntime.js

@@ -1,24 +1,25 @@
 "use client";
-import { attachSubagentTranscripts, createAttachMemo } from "./attachSubagentTranscripts.js";
 import { getMessageContent, getMessageModality, groupUIMessagesByParent } from "./converter.js";
-import { convertLangChainBaseMessage, getMessageType } from "./convertMessages.js";
+import { convertLangChainBaseMessage, createLangChainMetadataKey, getMessageType } from "./convertMessages.js";
+import { LangChainThreadController } from "./LangChainThreadController.js";
+import { attachSubagentTranscripts, createAttachMemo } from "./attachSubagentTranscripts.js";
 import { langChainExtras } from "./runtimeExtras.js";
 import { useLangChainStreamingTiming } from "./streamingTiming.js";
 import { UI_CUSTOM_CHANNELS, createUIFoldMemo, createUISnapshotMemo, foldUIUpdates, mergeUIMessages, reconcileUISnapshot } from "./uiMessages.js";
 import { useSubagentTranscripts } from "./useSubagentTranscripts.js";
 import { resolveForkCheckpoint } from "./resolveForkCheckpoint.js";
 import { LANGCHAIN_SDK } from "./sdkIdentity.js";
-import { generateId, getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 import { createCloudThreadListAdapterCreateFallback, createToolCallCancellationStub, getThreadMessageText, scanPendingToolCalls } from "@assistant-ui/core/internal";
+import { generateId, getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { useCloudThreadListAdapter, useExternalMessageConverter, useExternalStoreRuntime, useRemoteThreadListRuntime } from "@assistant-ui/core/react";
-import { useCallback, useEffect, useInsertionEffect, useMemo, useRef, useState } from "react";
+import { useCallback, useEffect, useInsertionEffect, useMemo, useRef, useState, useSyncExternalStore } from "react";
 import { STREAM_CONTROLLER, useChannel, useStream } from "@langchain/react";
 //#region src/useStreamRuntime.ts
 const runConfigToSubmitOptions = (runConfig) => runConfig?.custom ? { config: { configurable: runConfig.custom } } : void 0;
 const getPendingToolCalls = (messages) => scanPendingToolCalls(messages, (message) => {
 	const type = getMessageType(message);
-	if (type === "ai") return { toolCalls: message.tool_calls ?? [] };
+	if (type === "ai") return { toolCalls: (message.tool_calls ?? []).filter((toolCall) => typeof toolCall === "object" && toolCall !== null) };
 	if (type === "tool" && message.tool_call_id) return { toolCallId: message.tool_call_id };
 }, (toolCall) => toolCall.id);
 const toStagedHumanMessage = (msg, id = generateId()) => ({
@@ -32,12 +33,6 @@ const toStagedMessageInput = (message) => ({
 	content: message.content,
 	...message.additional_kwargs && { additional_kwargs: message.additional_kwargs }
 });
-const humanContentText = (content) => {
-	if (typeof content === "string") return content;
-	if (!Array.isArray(content)) return "";
-	return content.filter((part) => typeof part === "object" && part !== null && part.type === "text" && typeof part.text === "string").map((part) => part.text).join("");
-};
-const hasSameMessageContent = (a, b) => humanContentText(a.content) === humanContentText(b.content);
 const truncateLangChainBaseMessages = (threadMessages, parentId) => {
 	if (parentId === null) return [];
 	const parentIndex = threadMessages.findIndex((m) => m.id === parentId);
@@ -53,7 +48,10 @@ const useStreamThreadRuntime = (options) => {
 	const uiStateKey = options.uiStateKey ?? "ui";
 	const externalId = useAuiState((s) => s.threadListItem.externalId);
 	const stream = useStream(Object.assign({}, options, { threadId: externalId }));
-	const [stagedMessages, setStagedMessages] = useState(null);
+	const [threadController] = useState(() => new LangChainThreadController());
+	const getVisibleStagedMessages = () => threadController.getState().visibleStagedMessages;
+	const visibleStagedMessages = useSyncExternalStore(threadController.subscribe, getVisibleStagedMessages, getVisibleStagedMessages);
+	const forkPending = useSyncExternalStore(threadController.subscribe, () => threadController.getState().forkPending, () => threadController.getState().forkPending);
 	const [toolStatuses, setToolStatuses] = useState({});
 	const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
 	const effectiveIsRunning = stream.isLoading || hasExecutingTools;
@@ -64,18 +62,20 @@ const useStreamThreadRuntime = (options) => {
 	const liveUiMessages = useMemo(() => foldUIUpdates(customEvents, uiFoldMemo), [customEvents, uiFoldMemo]);
 	const mergedUiMessages = useMemo(() => mergeUIMessages(liveUiMessages, uiStateValue), [liveUiMessages, uiStateValue]);
 	const uiMessagesByParent = useMemo(() => groupUIMessagesByParent(mergedUiMessages), [mergedUiMessages]);
-	const visibleMessages = stagedMessages ?? stream.messages;
+	const visibleMessages = visibleStagedMessages ?? stream.messages;
 	const messageTiming = useLangChainStreamingTiming(visibleMessages, effectiveIsRunning);
 	const subagentTranscripts = useSubagentTranscripts(stream, uiMessagesByParent);
-	const convertWithUI = useMemo(() => (message, metadata) => convertLangChainBaseMessage(message, {
-		...metadata,
+	const [getConverterMetadataKey] = useState(createLangChainMetadataKey);
+	const converterMetadata = useMemo(() => ({
 		uiMessagesByParent,
 		messageTiming
 	}), [uiMessagesByParent, messageTiming]);
 	const threadMessages = useExternalMessageConverter({
-		callback: convertWithUI,
+		callback: convertLangChainBaseMessage,
 		messages: visibleMessages,
-		isRunning: effectiveIsRunning
+		isRunning: effectiveIsRunning,
+		metadata: converterMetadata,
+		getMetadataKey: getConverterMetadataKey
 	});
 	const [memo] = useState(createAttachMemo);
 	const messagesWithTranscripts = useMemo(() => attachSubagentTranscripts(threadMessages, subagentTranscripts, memo), [
@@ -126,48 +126,35 @@ const useStreamThreadRuntime = (options) => {
 	useInsertionEffect(() => {
 		visibleMessagesRef.current = visibleMessages;
 	}, [visibleMessages]);
+	const dispatchStaging = useCallback((action, fallback = visibleMessagesRef.current) => {
+		threadController.dispatch(action);
+		visibleMessagesRef.current = threadController.getState().visibleStagedMessages ?? fallback;
+	}, [threadController]);
 	const threadMessagesRef = useRef(messagesWithTranscripts);
 	useInsertionEffect(() => {
 		threadMessagesRef.current = messagesWithTranscripts;
 	}, [messagesWithTranscripts]);
-	const stagedMessagesRef = useRef(/* @__PURE__ */ new Map());
-	const stagedBaseMessagesRef = useRef(null);
 	useEffect(() => {
-		if (stagedMessagesRef.current.size === 0) return;
-		const baseMessages = stagedBaseMessagesRef.current ?? stream.messages;
-		const remainingStagedMessages = [];
-		const matchedBaseMessageIndexes = /* @__PURE__ */ new Set();
-		const visibleStagedIds = new Set(visibleMessagesRef.current.flatMap((m) => m.id ? [m.id] : []));
-		for (const [id, staged] of stagedMessagesRef.current) {
-			if (!visibleStagedIds.has(id)) continue;
-			if (baseMessages.some((message, index) => {
-				if (matchedBaseMessageIndexes.has(index)) return false;
-				if (message.id === id) {
-					matchedBaseMessageIndexes.add(index);
-					return true;
-				}
-				if (!staged.reconcileOnEcho || index < staged.baseMessageCount || getMessageType(message) !== "human" || !hasSameMessageContent(message, staged.message)) return false;
-				matchedBaseMessageIndexes.add(index);
-				return true;
-			})) stagedMessagesRef.current.delete(id);
-			else remainingStagedMessages.push(staged.message);
-		}
-		if (remainingStagedMessages.length === 0) {
-			stagedBaseMessagesRef.current = null;
-			visibleMessagesRef.current = baseMessages;
-			setStagedMessages(null);
-			return;
-		}
-		const nextMessages = [...baseMessages, ...remainingStagedMessages];
-		visibleMessagesRef.current = nextMessages;
-		setStagedMessages(nextMessages);
-	}, [stream.messages]);
+		const state = threadController.getState();
+		if (state.stagedEntries.size === 0) return;
+		const messages = stream.messages;
+		dispatchStaging({
+			type: "reconcile",
+			messages,
+			visibleMessages: visibleMessagesRef.current
+		}, state.stagedBaseMessages ?? messages);
+	}, [
+		stream.messages,
+		threadController,
+		dispatchStaging
+	]);
 	const getStagedRun = (parentId) => {
-		const parent = parentId ? stagedMessagesRef.current.get(parentId) : void 0;
+		const stagedEntries = threadController.getState().stagedEntries;
+		const parent = parentId ? stagedEntries.get(parentId) : void 0;
 		if (!parent || parent.transcriptStatus === "sent") return null;
 		const staged = [];
 		for (const message of visibleMessagesRef.current) {
-			const entry = message.id ? stagedMessagesRef.current.get(message.id) : void 0;
+			const entry = message.id ? stagedEntries.get(message.id) : void 0;
 			if (entry && entry.transcriptStatus !== "sent") staged.push(entry.message);
 			if (message.id === parentId) break;
 		}
@@ -183,23 +170,28 @@ const useStreamThreadRuntime = (options) => {
 			content: getThreadMessageText(message),
 			...message.metadata.modality && { additional_kwargs: { modality: message.metadata.modality } }
 		};
-		stagedMessagesRef.current.set(transcript.id, {
-			message: transcript,
-			runConfig: void 0,
-			reconcileOnEcho: false,
-			baseMessageCount: streamRef.current.messages.length,
-			transcriptStatus: "unsent"
+		dispatchStaging({
+			type: "stage",
+			entry: {
+				message: transcript,
+				runConfig: void 0,
+				reconcileOnEcho: false,
+				baseMessageCount: streamRef.current.messages.length,
+				transcriptStatus: "unsent"
+			},
+			visibleMessages: visibleMessagesRef.current
 		});
-		const nextMessages = [...visibleMessagesRef.current, transcript];
-		visibleMessagesRef.current = nextMessages;
-		setStagedMessages(nextMessages);
 	};
-	const getUnsentTranscripts = () => visibleMessagesRef.current.filter((message) => message.id !== void 0 && stagedMessagesRef.current.get(message.id)?.transcriptStatus === "unsent");
+	const getUnsentTranscripts = () => {
+		const stagedEntries = threadController.getState().stagedEntries;
+		return visibleMessagesRef.current.filter((message) => message.id !== void 0 && stagedEntries.get(message.id)?.transcriptStatus === "unsent");
+	};
 	const setTranscriptStatus = (messages, status) => {
-		for (const message of messages) {
-			const staged = message.id ? stagedMessagesRef.current.get(message.id) : void 0;
-			if (staged?.transcriptStatus) staged.transcriptStatus = status;
-		}
+		dispatchStaging({
+			type: "markTranscript",
+			messages,
+			status
+		});
 	};
 	const submitCarryingTranscripts = async (transcripts, submit) => {
 		setTranscriptStatus(transcripts, "sent");
@@ -211,10 +203,10 @@ const useStreamThreadRuntime = (options) => {
 		}
 	};
 	const dropTranscripts = (messages) => {
-		for (const message of messages) if (message.id && stagedMessagesRef.current.get(message.id)?.transcriptStatus) removeStagedMessage(message.id);
+		for (const message of messages) if (message.id && threadController.getState().stagedEntries.get(message.id)?.transcriptStatus) removeStagedMessage(message.id);
 	};
 	const isTranscriptMessage = (message) => {
-		if ((message.id ? stagedMessagesRef.current.get(message.id) : void 0)?.transcriptStatus === "unsent") return true;
+		if ((message.id ? threadController.getState().stagedEntries.get(message.id) : void 0)?.transcriptStatus === "unsent") return true;
 		const type = getMessageType(message);
 		return (type === "human" || type === "ai") && getMessageModality(message.additional_kwargs) !== void 0;
 	};
@@ -236,25 +228,31 @@ const useStreamThreadRuntime = (options) => {
 	};
 	const stageUserMessage = (msg, reconcileOnEcho = false) => {
 		const stagedMessage = toStagedHumanMessage(msg);
-		stagedMessagesRef.current.set(stagedMessage.id, {
-			message: stagedMessage,
-			runConfig: msg.runConfig,
-			reconcileOnEcho,
-			baseMessageCount: streamRef.current.messages.length
+		dispatchStaging({
+			type: "stage",
+			entry: {
+				message: stagedMessage,
+				runConfig: msg.runConfig,
+				reconcileOnEcho,
+				baseMessageCount: streamRef.current.messages.length
+			},
+			visibleMessages: visibleMessagesRef.current
 		});
-		const nextMessages = [...visibleMessagesRef.current, stagedMessage];
-		visibleMessagesRef.current = nextMessages;
-		setStagedMessages(nextMessages);
 		return stagedMessage;
 	};
 	const removeStagedMessage = (id) => {
-		if (!stagedMessagesRef.current.delete(id)) return;
-		const nextMessages = visibleMessagesRef.current.filter((message) => message.id !== id);
-		visibleMessagesRef.current = nextMessages;
-		if (stagedMessagesRef.current.size === 0) {
-			stagedBaseMessagesRef.current = null;
-			setStagedMessages(null);
-		} else setStagedMessages(nextMessages);
+		if (!threadController.getState().stagedEntries.has(id)) return;
+		const visible = visibleMessagesRef.current;
+		const nextMessages = visible.filter((message) => message.id !== id);
+		dispatchStaging({
+			type: "remove",
+			id,
+			visibleMessages: visible
+		}, nextMessages);
+	};
+	const supersedeFork = () => {
+		threadController.dispatch({ type: "supersedeFork" });
+		return threadController.getState().forkGeneration;
 	};
 	const extras = useMemo(() => langChainExtras.provide({
 		interrupt: stream.interrupt,
@@ -279,7 +277,7 @@ const useStreamThreadRuntime = (options) => {
 	]);
 	return useExternalStoreRuntime({
 		...pickExternalStoreSharedOptions(options),
-		isRunning: stream.isLoading,
+		isRunning: stream.isLoading || forkPending,
 		isLoading: stream.isThreadLoading,
 		messages: messagesWithTranscripts,
 		adapters,
@@ -328,73 +326,89 @@ const useStreamThreadRuntime = (options) => {
 			}] }, runConfig === void 0 ? void 0 : { config: runConfig });
 		},
 		onReload: async (parentId, config) => {
-			const stagedRun = getStagedRun(parentId);
-			if (stagedRun) {
-				if (config.sourceId && stagedMessagesRef.current.get(config.sourceId)?.transcriptStatus) removeStagedMessage(config.sourceId);
-				const promotedIds = /* @__PURE__ */ new Set();
-				for (const message of stagedRun.messages) {
-					if (!message.id || stagedMessagesRef.current.get(message.id)?.transcriptStatus) continue;
-					promotedIds.add(message.id);
-					stagedMessagesRef.current.delete(message.id);
+			const forkGeneration = supersedeFork();
+			try {
+				await streamRef.current.stop();
+				if (threadController.getState().forkGeneration !== forkGeneration) return;
+				const stagedRun = getStagedRun(parentId);
+				if (stagedRun) {
+					if (config.sourceId && threadController.getState().stagedEntries.get(config.sourceId)?.transcriptStatus) removeStagedMessage(config.sourceId);
+					dispatchStaging({
+						type: "promote",
+						messages: stagedRun.messages,
+						visibleMessages: visibleMessagesRef.current
+					});
+					const runConfig = config.runConfig ?? stagedRun.runConfig;
+					setActiveRunConfig(runConfig);
+					await submitCarryingTranscripts(stagedRun.messages, () => stream.submit({ [messagesKey]: stagedRun.messages.map(toStagedMessageInput) }, runConfigToSubmitOptions(runConfig)));
+					return;
 				}
-				stagedBaseMessagesRef.current = null;
-				if (stagedMessagesRef.current.size > 0) {
-					const nextMessages = visibleMessagesRef.current.filter((message) => !message.id || !promotedIds.has(message.id));
-					visibleMessagesRef.current = nextMessages;
-					setStagedMessages(nextMessages);
-				} else setStagedMessages(null);
-				const runConfig = config.runConfig ?? stagedRun.runConfig;
-				setActiveRunConfig(runConfig);
-				await submitCarryingTranscripts(stagedRun.messages, () => stream.submit({ [messagesKey]: stagedRun.messages.map(toStagedMessageInput) }, runConfigToSubmitOptions(runConfig)));
-				return;
+				const threadId = externalId;
+				if (!threadId || parentId == null) return;
+				const s = streamRef.current;
+				const fork = planForkTranscripts(parentId);
+				const checkpointId = await resolveForkCheckpoint(s.client, threadId, s.messages, fork.forkParentId, config.sourceId, s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(), messagesKey);
+				if (threadController.getState().forkGeneration !== forkGeneration) return;
+				if (!checkpointId) return;
+				dropTranscripts(fork.truncated);
+				setActiveRunConfig(config.runConfig);
+				await submitCarryingTranscripts(fork.transcripts, () => s.submit(fork.transcripts.length > 0 ? { [messagesKey]: fork.transcripts.map(toStagedMessageInput) } : null, {
+					forkFrom: checkpointId,
+					...runConfigToSubmitOptions(config.runConfig)
+				}));
+			} finally {
+				threadController.dispatch({
+					type: "finishFork",
+					generation: forkGeneration
+				});
 			}
-			const threadId = externalId;
-			if (!threadId || parentId == null) return;
-			const s = streamRef.current;
-			const fork = planForkTranscripts(parentId);
-			const checkpointId = await resolveForkCheckpoint(s.client, threadId, s.messages, fork.forkParentId, config.sourceId, s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(), messagesKey);
-			if (!checkpointId) return;
-			dropTranscripts(fork.truncated);
-			setActiveRunConfig(config.runConfig);
-			await submitCarryingTranscripts(fork.transcripts, () => s.submit(fork.transcripts.length > 0 ? { [messagesKey]: fork.transcripts.map(toStagedMessageInput) } : null, {
-				forkFrom: checkpointId,
-				...runConfigToSubmitOptions(config.runConfig)
-			}));
 		},
 		onEdit: async (message) => {
-			if (!(message.startRun ?? message.role === "user")) {
-				const truncated = truncateLangChainBaseMessages(threadMessagesRef.current, message.parentId);
-				const stagedMessage = toStagedHumanMessage(message);
-				stagedMessagesRef.current.set(stagedMessage.id, {
-					message: stagedMessage,
-					runConfig: message.runConfig,
-					reconcileOnEcho: false,
-					baseMessageCount: 0
+			const forkGeneration = supersedeFork();
+			try {
+				await streamRef.current.stop();
+				if (threadController.getState().forkGeneration !== forkGeneration) return;
+				if (!(message.startRun ?? message.role === "user")) {
+					const truncated = truncateLangChainBaseMessages(threadMessagesRef.current, message.parentId);
+					const stagedMessage = toStagedHumanMessage(message);
+					dispatchStaging({
+						type: "stageEdit",
+						entry: {
+							message: stagedMessage,
+							runConfig: message.runConfig,
+							reconcileOnEcho: false,
+							baseMessageCount: 0
+						},
+						baseMessages: truncated
+					});
+					return;
+				}
+				const threadId = externalId;
+				if (!threadId) return;
+				const s = streamRef.current;
+				const fork = planForkTranscripts(message.parentId);
+				const checkpointId = await resolveForkCheckpoint(s.client, threadId, s.messages, fork.forkParentId, message.sourceId, s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(), messagesKey);
+				if (threadController.getState().forkGeneration !== forkGeneration) return;
+				if (!checkpointId) return;
+				dropTranscripts(fork.truncated);
+				const content = getMessageContent(message);
+				setActiveRunConfig(message.runConfig);
+				await submitCarryingTranscripts(fork.transcripts, () => s.submit({ [messagesKey]: [...fork.transcripts.map(toStagedMessageInput), {
+					type: "human",
+					content
+				}] }, {
+					forkFrom: checkpointId,
+					...runConfigToSubmitOptions(message.runConfig)
+				}));
+			} finally {
+				threadController.dispatch({
+					type: "finishFork",
+					generation: forkGeneration
 				});
-				stagedBaseMessagesRef.current = truncated;
-				const nextMessages = [...truncated, stagedMessage];
-				visibleMessagesRef.current = nextMessages;
-				setStagedMessages(nextMessages);
-				return;
 			}
-			const threadId = externalId;
-			if (!threadId) return;
-			const s = streamRef.current;
-			const fork = planForkTranscripts(message.parentId);
-			const checkpointId = await resolveForkCheckpoint(s.client, threadId, s.messages, fork.forkParentId, message.sourceId, s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(), messagesKey);
-			if (!checkpointId) return;
-			dropTranscripts(fork.truncated);
-			const content = getMessageContent(message);
-			setActiveRunConfig(message.runConfig);
-			await submitCarryingTranscripts(fork.transcripts, () => s.submit({ [messagesKey]: [...fork.transcripts.map(toStagedMessageInput), {
-				type: "human",
-				content
-			}] }, {
-				forkFrom: checkpointId,
-				...runConfigToSubmitOptions(message.runConfig)
-			}));
 		},
 		onCancel: unstable_allowCancellation !== false ? async () => {
+			threadController.dispatch({ type: "cancelFork" });
 			activeRunConfigRef.current = void 0;
 			await stream.stop();
 		} : void 0
@@ -403,7 +417,7 @@ const useStreamThreadRuntime = (options) => {
 /**
 * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
 * Accepts the same options as `useStream` from `@langchain/react`, plus
-* `cloud` and `adapters`.
+* `cloud`, `scopeId`, and `adapters`.
 *
 * @example
 * ```tsx
@@ -425,11 +439,12 @@ const useStreamThreadRuntime = (options) => {
 * ```
 */
 const useStreamRuntime = (rawOptions) => {
-	const { cloud, unstable_threadListAdapter, create, delete: deleteFn, onThreadIdChange, ...options } = rawOptions;
+	const { cloud, scopeId, unstable_threadListAdapter, create, delete: deleteFn, onThreadIdChange, ...options } = rawOptions;
 	const aui = useAui();
 	const cloudAdapter = useCloudThreadListAdapter({
 		sdk: LANGCHAIN_SDK,
 		cloud,
+		scopeId,
 		create: createCloudThreadListAdapterCreateFallback(create, aui.threadListItem),
 		delete: deleteFn
 	});
@@ -443,6 +458,6 @@ const useStreamRuntime = (rawOptions) => {
 	});
 };
 //#endregion
-export { groupUIMessagesByParent, runConfigToSubmitOptions, useStreamRuntime };
+export { getPendingToolCalls, groupUIMessagesByParent, runConfigToSubmitOptions, useStreamRuntime };
 
 //# sourceMappingURL=useStreamRuntime.js.map