@assistant-ui/core 0.3.20 → 0.3.21

package/dist/runtime/utils/external-message-conversion.js

@@ -1,5 +1,5 @@
-import { isJSONValueEqual } from "../../utils/json/is-json-equal.js";
 import { generateErrorMessageId } from "../../utils/id.js";
+import { isJSONValueEqual } from "../../utils/json/is-json-equal.js";
 import { fromThreadMessageLike } from "./thread-message-like.js";
 import { FALLBACK_ID_PREFIX, bindExternalStoreMessage, getExternalStoreMessages, symbolInnerMessage } from "./external-store-message.js";
 import { getAutoStatus, isAutoStatus, isBackgroundToolCall, isInterruptedToolCall, isPendingToolCall } from "./auto-status.js";
@@ -43,7 +43,7 @@ const joinExternalMessages = (messages) => {
 	for (const output of messages) if (output.role === "tool") {
 		const toolCallIdx = !isNaNToolCallId(output.toolCallId) ? toolCallIndices.get(output.toolCallId) ?? -1 : -1;
 		if (toolCallIdx !== -1) {
-			const toolCall = assistantMessage.content[toolCallIdx];
+			const { isPreliminary: _isPreliminary, ...toolCall } = assistantMessage.content[toolCallIdx];
 			if (output.toolName != null) {
 				if (toolCall.toolName !== output.toolName) throw new Error(`Tool call name ${output.toolCallId} ${output.toolName} does not match existing tool call ${toolCall.toolName}`);
 			}
@@ -220,7 +220,7 @@ function createErrorAssistantMessage(error) {
 	return msg;
 }
 const completeExternalMessageConversion = (messages, error) => {
-	if (error) {
+	if (error !== void 0 && error !== null) {
 		const lastMessage = messages.at(-1);
 		if (!lastMessage || lastMessage.role !== "assistant") messages.push(createErrorAssistantMessage(error));
 	}

package/dist/runtime/utils/external-message-conversion.js.map

@@ -1 +1 @@
-{"version":3,"file":"external-message-conversion.js","names":["ReadonlyJSONValue","ToolExecutionStatus","ThreadMessageConverter","ThreadAssistantMessage","ThreadMessage","ToolCallMessagePart","MessageTiming","generateErrorMessageId","isJSONValueEqual","getAutoStatus","isAutoStatus","isBackgroundToolCall","isInterruptedToolCall","isPendingToolCall","bindExternalStoreMessage","FALLBACK_ID_PREFIX","getExternalStoreMessages","symbolInnerMessage","fromThreadMessageLike","ThreadMessageLike","JoinStrategy","ExternalMessageConverterMessage","convertConfig","joinStrategy","role","toolCallId","toolName","result","artifact","isError","messages","ExternalMessageConverterMetadata","toolStatuses","Record","error","cancelledMessageIds","ReadonlySet","messageTiming","ExternalMessageConverterCallback","message","T","metadata","ExternalMessageConverterCallbackResult","input","outputs","ExternalMessageConverterChunk","inputs","Mutable","P","stringifyForError","value","text","Error","String","JSON","stringify","length","slice","toCallbackOutputs","output","Array","isArray","o","valid","content","filter","warnedDroppedToolResults","has","add","console","warn","WeakSet","convertExternalMessageCallback","callback","mergeInnerMessages","existing","incoming","isNaNToolCallId","Number","isNaN","joinExternalMessages","assistantMessage","Omit","Exclude","toolCallIndices","Map","reasoningIndices","toolCallIdx","get","toolCall","type","const","map","c","id","createdAt","status","attachments","unstable_state","undefined","unstable_annotations","unstable_data","steps","custom","timing","modality","submittedFeedback","isOptimistic","part","existingIdx","parentId","partIdx","push","set","unsupportedRole","chunkExternalMessages","callbackResults","results","isAssistant","pendingNone","flush","callbackResult","isVoice","at","shallowArrayEqual","a","b","i","ExternalMessageChunkConversionCache","generatedFallbackMessages","ExternalMessageConversionCallbackCacheEntry","InternalExternalMessageConversionCache","WeakKey","callbackCache","WeakMap","chunkCache","converterCache","createExternalMessageConversionCache","convertExternalMessageChunk","idx","chunkCount","isRunning","cache","isLast","joined","isCancelled","some","hasInterruptedToolCalls","hasPendingToolCalls","hasBackgroundToolCalls","autoStatus","fallbackId","cachedMessage","reason","updated","newMessage","createErrorAssistantMessage","msg","Date","completeExternalMessageConversion","lastMessage","convertExternalMessages","chunks","key","cached","convertMessages"],"sources":["../../../src/runtime/utils/external-message-conversion.ts"],"sourcesContent":["import type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport type { ToolExecutionStatus } from \"../../runtimes/tool-invocations/ToolInvocationTracker\";\nimport { ThreadMessageConverter } from \"../../runtimes/external-store/thread-message-converter\";\nimport type {\n  ThreadAssistantMessage,\n  ThreadMessage,\n  ToolCallMessagePart,\n} from \"../../types/message\";\nimport type { MessageTiming } from \"../../types/message\";\nimport { generateErrorMessageId } from \"../../utils/id\";\nimport { isJSONValueEqual } from \"../../utils/json/is-json-equal\";\nimport {\n  getAutoStatus,\n  isAutoStatus,\n  isBackgroundToolCall,\n  isInterruptedToolCall,\n  isPendingToolCall,\n} from \"./auto-status\";\nimport {\n  bindExternalStoreMessage,\n  FALLBACK_ID_PREFIX,\n  getExternalStoreMessages,\n  symbolInnerMessage,\n} from \"./external-store-message\";\nimport {\n  fromThreadMessageLike,\n  type ThreadMessageLike,\n} from \"./thread-message-like\";\n\nexport type JoinStrategy = \"concat-content\" | \"none\";\n\nexport type ExternalMessageConverterMessage =\n  | (ThreadMessageLike & {\n      readonly convertConfig?: {\n        readonly joinStrategy?: JoinStrategy;\n      };\n    })\n  | {\n      role: \"tool\";\n      toolCallId: string;\n      toolName?: string | undefined;\n      result: any;\n      artifact?: any;\n      isError?: boolean;\n      messages?: readonly ThreadMessage[];\n    };\n\nexport type ExternalMessageConverterMetadata = {\n  readonly toolStatuses?: Record<string, ToolExecutionStatus>;\n  readonly error?: ReadonlyJSONValue;\n  readonly cancelledMessageIds?: ReadonlySet<string>;\n  readonly messageTiming?: Record<string, MessageTiming>;\n};\n\nexport type ExternalMessageConverterCallback<T> = (\n  message: T,\n  metadata: ExternalMessageConverterMetadata,\n) => ExternalMessageConverterMessage | ExternalMessageConverterMessage[];\n\nexport type ExternalMessageConverterCallbackResult<T> = {\n  input: T;\n  outputs: ExternalMessageConverterMessage[];\n};\n\nexport type ExternalMessageConverterChunk<T> = {\n  inputs: T[];\n  outputs: ExternalMessageConverterMessage[];\n};\n\ntype Mutable<T> = {\n  -readonly [P in keyof T]: T[P];\n};\n\nconst stringifyForError = (value: unknown) => {\n  let text;\n  try {\n    text = value instanceof Error ? String(value) : JSON.stringify(value);\n  } catch {\n    /* circular */\n  }\n  text ??= String(value);\n  return text.length > 200 ? `${text.slice(0, 200)}…` : text;\n};\n\nconst toCallbackOutputs = (\n  output: ExternalMessageConverterMessage | ExternalMessageConverterMessage[],\n  input: unknown,\n): ExternalMessageConverterMessage[] => {\n  const outputs = Array.isArray(output) ? output : [output];\n  for (const o of outputs) {\n    const valid =\n      typeof o === \"object\" &&\n      o !== null &&\n      (o.role === \"tool\" ||\n        ((o.role === \"assistant\" || o.role === \"user\" || o.role === \"system\") &&\n          (typeof o.content === \"string\" || Array.isArray(o.content))));\n    if (!valid)\n      throw new Error(\n        `External message converter: the converter callback returned an invalid message (${stringifyForError(o)}) for input ${stringifyForError(input)}. Return an empty array to skip a message.`,\n      );\n  }\n  // Providers can emit tool results without a usable id (react-langchain maps\n  // a missing tool_call_id to \"\"); converters must not throw on those, so the\n  // result is dropped like any other orphaned tool output.\n  return outputs.filter((o) => {\n    if (o.role !== \"tool\") return true;\n    if (typeof o.toolCallId === \"string\" && o.toolCallId.length > 0)\n      return true;\n    if (!warnedDroppedToolResults.has(o)) {\n      warnedDroppedToolResults.add(o);\n      console.warn(\n        `External message converter: dropping a tool result without a toolCallId (${stringifyForError(o)}) for input ${stringifyForError(input)}.`,\n      );\n    }\n    return false;\n  });\n};\n\nconst warnedDroppedToolResults = new WeakSet<object>();\n\nexport const convertExternalMessageCallback = <T>(\n  input: T,\n  callback: ExternalMessageConverterCallback<T>,\n  metadata: ExternalMessageConverterMetadata,\n): ExternalMessageConverterCallbackResult<T> => ({\n  input,\n  outputs: toCallbackOutputs(callback(input, metadata), input),\n});\n\nconst mergeInnerMessages = (existing: object, incoming: object) => ({\n  [symbolInnerMessage]: [\n    ...((existing as any)[symbolInnerMessage] ?? []),\n    ...((incoming as any)[symbolInnerMessage] ?? []),\n  ],\n});\n\nconst isNaNToolCallId = (toolCallId: unknown) =>\n  typeof toolCallId === \"number\" && Number.isNaN(toolCallId);\n\nexport const joinExternalMessages = (\n  messages: readonly ExternalMessageConverterMessage[],\n): ThreadMessageLike => {\n  const assistantMessage: Mutable<Omit<ThreadMessageLike, \"metadata\">> & {\n    content: Exclude<ThreadMessageLike[\"content\"][0], string>[];\n    metadata?: Mutable<ThreadMessageLike[\"metadata\"]>;\n  } = {\n    role: \"assistant\",\n    content: [],\n  };\n  const toolCallIndices = new Map<unknown, number>();\n  const reasoningIndices = new Map<string, number>();\n  for (const output of messages) {\n    if (output.role === \"tool\") {\n      const toolCallIdx = !isNaNToolCallId(output.toolCallId)\n        ? (toolCallIndices.get(output.toolCallId) ?? -1)\n        : -1;\n      // Ignore orphaned tool results so one bad tool message does not\n      // prevent rendering the rest of the conversation.\n      if (toolCallIdx !== -1) {\n        const toolCall = assistantMessage.content[\n          toolCallIdx\n        ]! as ToolCallMessagePart;\n        if (output.toolName != null) {\n          if (toolCall.toolName !== output.toolName)\n            throw new Error(\n              `Tool call name ${output.toolCallId} ${output.toolName} does not match existing tool call ${toolCall.toolName}`,\n            );\n        }\n        assistantMessage.content[toolCallIdx] = {\n          ...toolCall,\n          ...{\n            [symbolInnerMessage]: [\n              ...((toolCall as any)[symbolInnerMessage] ?? []),\n              output,\n            ],\n          },\n          result: output.result,\n          artifact: output.artifact,\n          isError: output.isError,\n          messages: output.messages,\n        };\n      }\n    } else {\n      const role = output.role;\n      const content = (\n        typeof output.content === \"string\"\n          ? [{ type: \"text\" as const, text: output.content }]\n          : output.content\n      ).map((c) => ({\n        ...c,\n        ...{ [symbolInnerMessage]: [output] },\n      }));\n      switch (role) {\n        case \"system\":\n        case \"user\":\n          return {\n            ...output,\n            content,\n          };\n        case \"assistant\":\n          if (assistantMessage.content.length === 0) {\n            assistantMessage.id = output.id;\n            assistantMessage.createdAt ??= output.createdAt;\n            assistantMessage.status ??= output.status;\n\n            if (output.attachments) {\n              assistantMessage.attachments = [\n                ...(assistantMessage.attachments ?? []),\n                ...output.attachments,\n              ];\n            }\n          }\n\n          if (output.metadata) {\n            assistantMessage.metadata ??= {};\n            if (output.metadata.unstable_state !== undefined) {\n              assistantMessage.metadata.unstable_state =\n                output.metadata.unstable_state;\n            }\n            if (output.metadata.unstable_annotations) {\n              assistantMessage.metadata.unstable_annotations = [\n                ...(assistantMessage.metadata.unstable_annotations ?? []),\n                ...output.metadata.unstable_annotations,\n              ];\n            }\n            if (output.metadata.unstable_data) {\n              assistantMessage.metadata.unstable_data = [\n                ...(assistantMessage.metadata.unstable_data ?? []),\n                ...output.metadata.unstable_data,\n              ];\n            }\n            if (output.metadata.steps) {\n              assistantMessage.metadata.steps = [\n                ...(assistantMessage.metadata.steps ?? []),\n                ...output.metadata.steps,\n              ];\n            }\n            if (output.metadata.custom) {\n              assistantMessage.metadata.custom = {\n                ...(assistantMessage.metadata.custom ?? {}),\n                ...output.metadata.custom,\n              };\n            }\n\n            if (output.metadata.timing) {\n              assistantMessage.metadata.timing = output.metadata.timing;\n            }\n\n            if (output.metadata.modality) {\n              assistantMessage.metadata.modality = output.metadata.modality;\n            }\n\n            if (output.metadata.submittedFeedback) {\n              assistantMessage.metadata.submittedFeedback =\n                output.metadata.submittedFeedback;\n            }\n\n            if (output.metadata.isOptimistic) {\n              assistantMessage.metadata.isOptimistic = true;\n            }\n            // TODO keep this in sync with ThreadMessageLike[\"metadata\"] / fromThreadMessageLike\n          }\n\n          // Add content parts, merging reasoning parts with same parentId\n          for (const part of content) {\n            if (part.type === \"tool-call\" && part.toolCallId) {\n              const existingIdx = toolCallIndices.get(part.toolCallId);\n              if (existingIdx !== undefined) {\n                const existing = assistantMessage.content[\n                  existingIdx\n                ] as typeof part;\n                assistantMessage.content[existingIdx] = {\n                  ...existing,\n                  ...part,\n                  ...mergeInnerMessages(existing, part),\n                };\n                continue;\n              }\n            }\n\n            if (\n              part.type === \"reasoning\" &&\n              \"parentId\" in part &&\n              part.parentId\n            ) {\n              const existingIdx = reasoningIndices.get(part.parentId);\n              if (existingIdx !== undefined) {\n                const existing = assistantMessage.content[\n                  existingIdx\n                ] as typeof part;\n                assistantMessage.content[existingIdx] = {\n                  ...existing,\n                  text: `${existing.text}\\n\\n${part.text}`,\n                  ...mergeInnerMessages(existing, part),\n                };\n                continue;\n              }\n            }\n            const partIdx = assistantMessage.content.length;\n            assistantMessage.content.push(part);\n            if (\n              part.type === \"tool-call\" &&\n              !isNaNToolCallId(part.toolCallId) &&\n              !toolCallIndices.has(part.toolCallId)\n            ) {\n              toolCallIndices.set(part.toolCallId, partIdx);\n            } else if (\n              part.type === \"reasoning\" &&\n              \"parentId\" in part &&\n              part.parentId\n            ) {\n              reasoningIndices.set(part.parentId, partIdx);\n            }\n          }\n          break;\n        default: {\n          const unsupportedRole: never = role;\n          throw new Error(`Unknown message role: ${unsupportedRole}`);\n        }\n      }\n    }\n  }\n  return assistantMessage;\n};\n\nexport const chunkExternalMessages = <T>(\n  callbackResults: ExternalMessageConverterCallbackResult<T>[],\n  joinStrategy?: JoinStrategy,\n) => {\n  const results: ExternalMessageConverterChunk<T>[] = [];\n  let isAssistant = false;\n  let pendingNone = false; // true if the previous assistant message had joinStrategy \"none\"\n  let inputs: T[] = [];\n  let outputs: ExternalMessageConverterMessage[] = [];\n\n  const flush = () => {\n    if (outputs.length) {\n      results.push({\n        inputs,\n        outputs,\n      });\n    }\n    inputs = [];\n    outputs = [];\n    isAssistant = false;\n    pendingNone = false;\n  };\n\n  for (const callbackResult of callbackResults) {\n    for (const output of callbackResult.outputs) {\n      const isVoice =\n        output.role === \"assistant\" && output.metadata?.modality === \"voice\";\n      if (\n        (pendingNone && output.role !== \"tool\") ||\n        !isAssistant ||\n        output.role === \"user\" ||\n        output.role === \"system\" ||\n        isVoice\n      ) {\n        flush();\n      }\n      isAssistant = output.role === \"assistant\" || output.role === \"tool\";\n\n      if (inputs.at(-1) !== callbackResult.input) {\n        inputs.push(callbackResult.input);\n      }\n      outputs.push(output);\n\n      if (\n        output.role === \"assistant\" &&\n        (isVoice ||\n          output.convertConfig?.joinStrategy === \"none\" ||\n          joinStrategy === \"none\")\n      ) {\n        pendingNone = true;\n      }\n    }\n  }\n  flush();\n  return results;\n};\n\nexport const shallowArrayEqual = (\n  a: readonly unknown[],\n  b: readonly unknown[],\n) => {\n  if (a.length !== b.length) return false;\n  for (let i = 0; i < a.length; i++) {\n    if (a[i] !== b[i]) return false;\n  }\n  return true;\n};\n\ntype ExternalMessageChunkConversionCache = {\n  message: ThreadMessage | undefined;\n  generatedFallbackMessages: WeakSet<object>;\n};\n\ntype ExternalMessageConversionCallbackCacheEntry<T> =\n  ExternalMessageConverterCallbackResult<T> & {\n    metadata: ExternalMessageConverterMetadata;\n    callback: ExternalMessageConverterCallback<T>;\n  };\n\nexport type InternalExternalMessageConversionCache<\n  T extends WeakKey = WeakKey,\n> = {\n  callbackCache: WeakMap<T, ExternalMessageConversionCallbackCacheEntry<T>>;\n  chunkCache: WeakMap<\n    ExternalMessageConverterMessage,\n    ExternalMessageConverterChunk<T>\n  >;\n  converterCache: ThreadMessageConverter;\n  // Generatedness is tracked by identity, not by id shape: a caller-supplied id that happens to match the generated pattern must never be rewritten.\n  generatedFallbackMessages: WeakSet<object>;\n};\n\nexport const createExternalMessageConversionCache = <\n  T extends WeakKey = WeakKey,\n>(): InternalExternalMessageConversionCache<T> => ({\n  callbackCache: new WeakMap(),\n  chunkCache: new WeakMap(),\n  converterCache: new ThreadMessageConverter(),\n  generatedFallbackMessages: new WeakSet(),\n});\n\nexport const convertExternalMessageChunk = <T>(\n  message: ExternalMessageConverterChunk<T>,\n  idx: number,\n  chunkCount: number,\n  isRunning: boolean,\n  error: ReadonlyJSONValue | undefined,\n  cache?: ExternalMessageChunkConversionCache,\n  cancelledMessageIds?: ReadonlySet<string>,\n) => {\n  const isLast = idx === chunkCount - 1;\n  const joined = joinExternalMessages(message.outputs);\n  const isCancelled =\n    cancelledMessageIds !== undefined &&\n    message.outputs.some(\n      (output) =>\n        output.role !== \"tool\" &&\n        output.id != null &&\n        cancelledMessageIds.has(output.id),\n    );\n  const hasInterruptedToolCalls =\n    typeof joined.content === \"object\" &&\n    joined.content.some(isInterruptedToolCall);\n  const hasPendingToolCalls =\n    typeof joined.content === \"object\" &&\n    joined.content.some(isPendingToolCall);\n  const hasBackgroundToolCalls =\n    typeof joined.content === \"object\" &&\n    joined.content.some(isBackgroundToolCall);\n  const autoStatus = getAutoStatus(\n    isLast,\n    isRunning,\n    hasInterruptedToolCalls,\n    hasPendingToolCalls,\n    isLast ? error : undefined,\n    isCancelled,\n    hasBackgroundToolCalls,\n  );\n  const fallbackId = `${FALLBACK_ID_PREFIX}${idx}`;\n\n  const cachedMessage = cache?.message;\n  if (\n    cachedMessage &&\n    (cachedMessage.role !== \"assistant\" ||\n      !isAutoStatus(cachedMessage.status) ||\n      cachedMessage.status === autoStatus ||\n      (cachedMessage.status.type === \"incomplete\" &&\n        cachedMessage.status.reason === \"error\" &&\n        cachedMessage.status.error !== undefined &&\n        autoStatus.type === \"incomplete\" &&\n        autoStatus.reason === \"error\" &&\n        autoStatus.error !== undefined &&\n        (cachedMessage.status.error === autoStatus.error ||\n          isJSONValueEqual(cachedMessage.status.error, autoStatus.error))))\n  ) {\n    const inputs = getExternalStoreMessages<T>(cachedMessage);\n    if (shallowArrayEqual(inputs, message.inputs)) {\n      // A positional fallback id goes stale when messages are prepended\n      // or reordered; serving it unchanged makes two messages collide on\n      // one id and the dedup downstream drops one of them.\n      if (\n        cache.generatedFallbackMessages.has(cachedMessage) &&\n        cachedMessage.id !== fallbackId\n      ) {\n        const updated = { ...cachedMessage, id: fallbackId };\n        cache.generatedFallbackMessages.add(updated);\n        bindExternalStoreMessage(updated, message.inputs);\n        return updated;\n      }\n      return cachedMessage;\n    }\n  }\n\n  const newMessage = fromThreadMessageLike(joined, fallbackId, autoStatus);\n  if (cache && joined.id == null) {\n    cache.generatedFallbackMessages.add(newMessage);\n  }\n  bindExternalStoreMessage(newMessage, message.inputs);\n  return newMessage;\n};\n\nfunction createErrorAssistantMessage(\n  error: ReadonlyJSONValue,\n): ThreadAssistantMessage {\n  const msg: ThreadAssistantMessage = {\n    id: generateErrorMessageId(),\n    role: \"assistant\",\n    content: [],\n    status: { type: \"incomplete\", reason: \"error\", error },\n    createdAt: new Date(),\n    metadata: {\n      unstable_state: null,\n      unstable_annotations: [],\n      unstable_data: [],\n      custom: {},\n      steps: [],\n    },\n  };\n  bindExternalStoreMessage(msg, []);\n  return msg;\n}\n\nexport const completeExternalMessageConversion = (\n  messages: ThreadMessage[],\n  error: ReadonlyJSONValue | undefined,\n) => {\n  if (error) {\n    const lastMessage = messages.at(-1);\n    if (!lastMessage || lastMessage.role !== \"assistant\") {\n      messages.push(createErrorAssistantMessage(error));\n    }\n  }\n  return messages;\n};\n\nexport const convertExternalMessages = <T extends WeakKey>(\n  messages: T[],\n  callback: ExternalMessageConverterCallback<T>,\n  isRunning: boolean,\n  metadata: ExternalMessageConverterMetadata,\n  joinStrategy?: JoinStrategy,\n  cache?: InternalExternalMessageConversionCache<T>,\n) => {\n  const callbackResults = messages.map((message) => {\n    let result = cache?.callbackCache.get(message);\n    if (\n      !result ||\n      result.metadata !== metadata ||\n      result.callback !== callback\n    ) {\n      result = {\n        ...convertExternalMessageCallback(message, callback, metadata),\n        metadata,\n        callback,\n      };\n      cache?.callbackCache.set(message, result);\n    }\n    return result;\n  });\n  const chunks = chunkExternalMessages(callbackResults, joinStrategy).map(\n    (message) => {\n      const key = message.outputs[0];\n      if (!key || !cache) return message;\n\n      const cached = cache.chunkCache.get(key);\n      if (cached && shallowArrayEqual(cached.outputs, message.outputs)) {\n        return cached;\n      }\n      cache.chunkCache.set(key, message);\n      return message;\n    },\n  );\n  const result = cache\n    ? cache.converterCache.convertMessages(chunks, (cached, message, idx) =>\n        convertExternalMessageChunk(\n          message,\n          idx,\n          chunks.length,\n          isRunning,\n          metadata.error,\n          {\n            message: cached,\n            generatedFallbackMessages: cache.generatedFallbackMessages,\n          },\n          metadata.cancelledMessageIds,\n        ),\n      )\n    : chunks.map((message, idx) =>\n        convertExternalMessageChunk(\n          message,\n          idx,\n          chunks.length,\n          isRunning,\n          metadata.error,\n          undefined,\n          metadata.cancelledMessageIds,\n        ),\n      );\n  return completeExternalMessageConversion(result, metadata.error);\n};\n"],"mappings":";;;;;;;AAyEA,MAAMiD,qBAAqBC,UAAmB;CAC5C,IAAIC;CACJ,IAAI;EACFA,OAAOD,iBAAiBE,QAAQC,OAAOH,KAAK,IAAII,KAAKC,UAAUL,KAAK;CACtE,QAAQ,CACN;CAEFC,SAASE,OAAOH,KAAK;CACrB,OAAOC,KAAKK,SAAS,MAAM,GAAGL,KAAKM,MAAM,GAAG,GAAG,EAAC,KAAMN;AACxD;AAEA,MAAMO,qBACJC,QACAhB,UACsC;CACtC,MAAMC,UAAUgB,MAAMC,QAAQF,MAAM,IAAIA,SAAS,CAACA,MAAM;CACxD,KAAK,MAAMG,KAAKlB,SAOd,IAAI,EALF,OAAOkB,MAAM,YACbA,MAAM,SACLA,EAAEtC,SAAS,WACRsC,EAAEtC,SAAS,eAAesC,EAAEtC,SAAS,UAAUsC,EAAEtC,SAAS,cACzD,OAAOsC,EAAEE,YAAY,YAAYJ,MAAMC,QAAQC,EAAEE,OAAO,MAE7D,MAAM,IAAIZ,MACR,mFAAmFH,kBAAkBa,CAAC,EAAC,cAAeb,kBAAkBN,KAAK,EAAC,2CAChJ;CAKJ,OAAOC,QAAQqB,QAAQH,MAAM;EAC3B,IAAIA,EAAEtC,SAAS,QAAQ,OAAO;EAC9B,IAAI,OAAOsC,EAAErC,eAAe,YAAYqC,EAAErC,WAAW+B,SAAS,GAC5D,OAAO;EACT,IAAI,CAACU,yBAAyBC,IAAIL,CAAC,GAAG;GACpCI,yBAAyBE,IAAIN,CAAC;GAC9BO,QAAQC,KACN,4EAA4ErB,kBAAkBa,CAAC,EAAC,cAAeb,kBAAkBN,KAAK,EAAC,EACzI;EACF;EACA,OAAO;CACT,CAAC;AACH;AAEA,MAAMuB,2CAA2B,IAAIK,QAAgB;AAErD,MAAaC,kCACX7B,OACA8B,UACAhC,cAC+C;CAC/CE;CACAC,SAASc,kBAAkBe,SAAS9B,OAAOF,QAAQ,GAAGE,KAAK;AAC7D;AAEA,MAAM+B,sBAAsBC,UAAkBC,cAAsB,GACjE3D,qBAAqB,CACpB,GAAK0D,SAAiB1D,uBAAuB,CAAA,GAC7C,GAAK2D,SAAiB3D,uBAAuB,CAAA,CAAG,EAEpD;AAEA,MAAM4D,mBAAmBpD,eACvB,OAAOA,eAAe,YAAYqD,OAAOC,MAAMtD,UAAU;AAE3D,MAAauD,wBACXlD,aACsB;CACtB,MAAMmD,mBAGF;EACFzD,MAAM;EACNwC,SAAS,CAAA;CACX;CACA,MAAMoB,kCAAkB,IAAIC,IAAqB;CACjD,MAAMC,mCAAmB,IAAID,IAAoB;CACjD,KAAK,MAAM1B,UAAU7B,UACnB,IAAI6B,OAAOnC,SAAS,QAAQ;EAC1B,MAAM+D,cAAc,CAACV,gBAAgBlB,OAAOlC,UAAU,IACjD2D,gBAAgBI,IAAI7B,OAAOlC,UAAU,KAAK,KAC3C;EAGJ,IAAI8D,gBAAgB,IAAI;GACtB,MAAME,WAAWR,iBAAiBjB,QAChCuB;GAEF,IAAI5B,OAAOjC,YAAY,MACjB+D;QAAAA,SAAS/D,aAAaiC,OAAOjC,UAC/B,MAAM,IAAI0B,MACR,kBAAkBO,OAAOlC,WAAU,GAAIkC,OAAOjC,SAAQ,qCAAsC+D,SAAS/D,UACvG;GAAA;GAEJuD,iBAAiBjB,QAAQuB,eAAe;IACtC,GAAGE;KAEAxE,qBAAqB,CACpB,GAAKwE,SAAiBxE,uBAAuB,CAAA,GAC7C0C,MAAM;IAGVhC,QAAQgC,OAAOhC;IACfC,UAAU+B,OAAO/B;IACjBC,SAAS8B,OAAO9B;IAChBC,UAAU6B,OAAO7B;GACnB;EACF;CACF,OAAO;EACL,MAAMN,OAAOmC,OAAOnC;EACpB,MAAMwC,WACJ,OAAOL,OAAOK,YAAY,WACtB,CAAC;GAAE0B,MAAM;GAAiBvC,MAAMQ,OAAOK;EAAQ,CAAC,IAChDL,OAAOK,QAAAA,CACX4B,KAAKC,OAAO;GACZ,GAAGA;IACG5E,qBAAqB,CAAC0C,MAAM;EACpC,EAAE;EACF,QAAQnC,MAAR;GACE,KAAK;GACL,KAAK,QACH,OAAO;IACL,GAAGmC;IACHK;GACF;GACF,KAAK;IACH,IAAIiB,iBAAiBjB,QAAQR,WAAW,GAAG;KACzCyB,iBAAiBa,KAAKnC,OAAOmC;KAC7Bb,iBAAiBc,cAAcpC,OAAOoC;KACtCd,iBAAiBe,WAAWrC,OAAOqC;KAEnC,IAAIrC,OAAOsC,aACThB,iBAAiBgB,cAAc,CAC7B,GAAIhB,iBAAiBgB,eAAe,CAAA,GACpC,GAAGtC,OAAOsC,WAAW;IAG3B;IAEA,IAAItC,OAAOlB,UAAU;KACnBwC,iBAAiBxC,aAAa,CAAC;KAC/B,IAAIkB,OAAOlB,SAASyD,mBAAmBC,KAAAA,GACrClB,iBAAiBxC,SAASyD,iBACxBvC,OAAOlB,SAASyD;KAEpB,IAAIvC,OAAOlB,SAAS2D,sBAClBnB,iBAAiBxC,SAAS2D,uBAAuB,CAC/C,GAAInB,iBAAiBxC,SAAS2D,wBAAwB,CAAA,GACtD,GAAGzC,OAAOlB,SAAS2D,oBAAoB;KAG3C,IAAIzC,OAAOlB,SAAS4D,eAClBpB,iBAAiBxC,SAAS4D,gBAAgB,CACxC,GAAIpB,iBAAiBxC,SAAS4D,iBAAiB,CAAA,GAC/C,GAAG1C,OAAOlB,SAAS4D,aAAa;KAGpC,IAAI1C,OAAOlB,SAAS6D,OAClBrB,iBAAiBxC,SAAS6D,QAAQ,CAChC,GAAIrB,iBAAiBxC,SAAS6D,SAAS,CAAA,GACvC,GAAG3C,OAAOlB,SAAS6D,KAAK;KAG5B,IAAI3C,OAAOlB,SAAS8D,QAClBtB,iBAAiBxC,SAAS8D,SAAS;MACjC,GAAItB,iBAAiBxC,SAAS8D,UAAU,CAAC;MACzC,GAAG5C,OAAOlB,SAAS8D;KACrB;KAGF,IAAI5C,OAAOlB,SAAS+D,QAClBvB,iBAAiBxC,SAAS+D,SAAS7C,OAAOlB,SAAS+D;KAGrD,IAAI7C,OAAOlB,SAASgE,UAClBxB,iBAAiBxC,SAASgE,WAAW9C,OAAOlB,SAASgE;KAGvD,IAAI9C,OAAOlB,SAASiE,mBAClBzB,iBAAiBxC,SAASiE,oBACxB/C,OAAOlB,SAASiE;KAGpB,IAAI/C,OAAOlB,SAASkE,cAClB1B,iBAAiBxC,SAASkE,eAAe;IAG7C;IAGA,KAAK,MAAMC,QAAQ5C,SAAS;KAC1B,IAAI4C,KAAKlB,SAAS,eAAekB,KAAKnF,YAAY;MAChD,MAAMoF,cAAczB,gBAAgBI,IAAIoB,KAAKnF,UAAU;MACvD,IAAIoF,gBAAgBV,KAAAA,GAAW;OAC7B,MAAMxB,WAAWM,iBAAiBjB,QAChC6C;OAEF5B,iBAAiBjB,QAAQ6C,eAAe;QACtC,GAAGlC;QACH,GAAGiC;QACH,GAAGlC,mBAAmBC,UAAUiC,IAAI;OACtC;OACA;MACF;KACF;KAEA,IACEA,KAAKlB,SAAS,eACd,cAAckB,QACdA,KAAKE,UACL;MACA,MAAMD,cAAcvB,iBAAiBE,IAAIoB,KAAKE,QAAQ;MACtD,IAAID,gBAAgBV,KAAAA,GAAW;OAC7B,MAAMxB,WAAWM,iBAAiBjB,QAChC6C;OAEF5B,iBAAiBjB,QAAQ6C,eAAe;QACtC,GAAGlC;QACHxB,MAAM,GAAGwB,SAASxB,KAAI,MAAOyD,KAAKzD;QAClC,GAAGuB,mBAAmBC,UAAUiC,IAAI;OACtC;OACA;MACF;KACF;KACA,MAAMG,UAAU9B,iBAAiBjB,QAAQR;KACzCyB,iBAAiBjB,QAAQgD,KAAKJ,IAAI;KAClC,IACEA,KAAKlB,SAAS,eACd,CAACb,gBAAgB+B,KAAKnF,UAAU,KAChC,CAAC2D,gBAAgBjB,IAAIyC,KAAKnF,UAAU,GAEpC2D,gBAAgB6B,IAAIL,KAAKnF,YAAYsF,OAAO;UACvC,IACLH,KAAKlB,SAAS,eACd,cAAckB,QACdA,KAAKE,UAELxB,iBAAiB2B,IAAIL,KAAKE,UAAUC,OAAO;IAE/C;IACA;GACF,SAEE,MAAM,IAAI3D,MAAM,yBAAyB8D,MAAiB;EAE9D;CACF;CAEF,OAAOjC;AACT;AAEA,MAAakC,yBACXC,iBACA7F,iBACG;CACH,MAAM8F,UAA8C,CAAA;CACpD,IAAIC,cAAc;CAClB,IAAIC,cAAc;CAClB,IAAIzE,SAAc,CAAA;CAClB,IAAIF,UAA6C,CAAA;CAEjD,MAAM4E,cAAc;EAClB,IAAI5E,QAAQY,QACV6D,QAAQL,KAAK;GACXlE;GACAF;EACF,CAAC;EAEHE,SAAS,CAAA;EACTF,UAAU,CAAA;EACV0E,cAAc;EACdC,cAAc;CAChB;CAEA,KAAK,MAAME,kBAAkBL,iBAC3B,KAAK,MAAMzD,UAAU8D,eAAe7E,SAAS;EAC3C,MAAM8E,UACJ/D,OAAOnC,SAAS,eAAemC,OAAOlB,UAAUgE,aAAa;EAC/D,IACGc,eAAe5D,OAAOnC,SAAS,UAChC,CAAC8F,eACD3D,OAAOnC,SAAS,UAChBmC,OAAOnC,SAAS,YAChBkG,SAEAF,MAAM;EAERF,cAAc3D,OAAOnC,SAAS,eAAemC,OAAOnC,SAAS;EAE7D,IAAIsB,OAAO6E,GAAG,EAAE,MAAMF,eAAe9E,OACnCG,OAAOkE,KAAKS,eAAe9E,KAAK;EAElCC,QAAQoE,KAAKrD,MAAM;EAEnB,IACEA,OAAOnC,SAAS,gBACfkG,WACC/D,OAAOrC,eAAeC,iBAAiB,UACvCA,iBAAiB,SAEnBgG,cAAc;CAElB;CAEFC,MAAM;CACN,OAAOH;AACT;AAEA,MAAaO,qBACXC,GACAC,MACG;CACH,IAAID,EAAErE,WAAWsE,EAAEtE,QAAQ,OAAO;CAClC,KAAK,IAAIuE,IAAI,GAAGA,IAAIF,EAAErE,QAAQuE,KAC5B,IAAIF,EAAEE,OAAOD,EAAEC,IAAI,OAAO;CAE5B,OAAO;AACT;AA0BA,MAAaU,8CAEsC;CACjDJ,+BAAe,IAAIC,QAAQ;CAC3BC,4BAAY,IAAID,QAAQ;CACxBE,gBAAgB,IAAItI,uBAAuB;CAC3C+H,2CAA2B,IAAI1D,QAAQ;AACzC;AAEA,MAAamE,+BACXnG,SACAoG,KACAC,YACAC,WACA3G,OACA4G,OACA3G,wBACG;CACH,MAAM4G,SAASJ,QAAQC,aAAa;CACpC,MAAMI,SAAShE,qBAAqBzC,QAAQK,OAAO;CACnD,MAAMqG,cACJ9G,wBAAwBgE,KAAAA,KACxB5D,QAAQK,QAAQsG,MACbvF,WACCA,OAAOnC,SAAS,UAChBmC,OAAOmC,MAAM,QACb3D,oBAAoBgC,IAAIR,OAAOmC,EAAE,CACrC;CACF,MAAMqD,0BACJ,OAAOH,OAAOhF,YAAY,YAC1BgF,OAAOhF,QAAQkF,KAAKtI,qBAAqB;CAC3C,MAAMwI,sBACJ,OAAOJ,OAAOhF,YAAY,YAC1BgF,OAAOhF,QAAQkF,KAAKrI,iBAAiB;CACvC,MAAMwI,yBACJ,OAAOL,OAAOhF,YAAY,YAC1BgF,OAAOhF,QAAQkF,KAAKvI,oBAAoB;CAC1C,MAAM2I,aAAa7I,cACjBsI,QACAF,WACAM,yBACAC,qBACAL,SAAS7G,QAAQiE,KAAAA,GACjB8C,aACAI,sBACF;CACA,MAAME,aAAa,GAAGxI,qBAAqB4H;CAE3C,MAAMa,gBAAgBV,OAAOvG;CAC7B,IACEiH,kBACCA,cAAchI,SAAS,eACtB,CAACd,aAAa8I,cAAcxD,MAAM,KAClCwD,cAAcxD,WAAWsD,cACxBE,cAAcxD,OAAON,SAAS,gBAC7B8D,cAAcxD,OAAOyD,WAAW,WAChCD,cAAcxD,OAAO9D,UAAUiE,KAAAA,KAC/BmD,WAAW5D,SAAS,gBACpB4D,WAAWG,WAAW,WACtBH,WAAWpH,UAAUiE,KAAAA,MACpBqD,cAAcxD,OAAO9D,UAAUoH,WAAWpH,SACzC1B,iBAAiBgJ,cAAcxD,OAAO9D,OAAOoH,WAAWpH,KAAK,KACnE;EACA,MAAMY,SAAS9B,yBAA4BwI,aAAa;EACxD,IAAI5B,kBAAkB9E,QAAQP,QAAQO,MAAM,GAAG;GAI7C,IACEgG,MAAMb,0BAA0B9D,IAAIqF,aAAa,KACjDA,cAAc1D,OAAOyD,YACrB;IACA,MAAMG,UAAU;KAAE,GAAGF;KAAe1D,IAAIyD;IAAW;IACnDT,MAAMb,0BAA0B7D,IAAIsF,OAAO;IAC3C5I,yBAAyB4I,SAASnH,QAAQO,MAAM;IAChD,OAAO4G;GACT;GACA,OAAOF;EACT;CACF;CAEA,MAAMG,aAAazI,sBAAsB8H,QAAQO,YAAYD,UAAU;CACvE,IAAIR,SAASE,OAAOlD,MAAM,MACxBgD,MAAMb,0BAA0B7D,IAAIuF,UAAU;CAEhD7I,yBAAyB6I,YAAYpH,QAAQO,MAAM;CACnD,OAAO6G;AACT;AAEA,SAASC,4BACP1H,OACwB;CACxB,MAAM2H,MAA8B;EAClC/D,IAAIvF,uBAAuB;EAC3BiB,MAAM;EACNwC,SAAS,CAAA;EACTgC,QAAQ;GAAEN,MAAM;GAAc+D,QAAQ;GAASvH;EAAM;EACrD6D,2BAAW,IAAI+D,KAAK;EACpBrH,UAAU;GACRyD,gBAAgB;GAChBE,sBAAsB,CAAA;GACtBC,eAAe,CAAA;GACfE,QAAQ,CAAC;GACTD,OAAO,CAAA;EACT;CACF;CACAxF,yBAAyB+I,KAAK,CAAA,CAAE;CAChC,OAAOA;AACT;AAEA,MAAaE,qCACXjI,UACAI,UACG;CACH,IAAIA,OAAO;EACT,MAAM8H,cAAclI,SAAS6F,GAAG,EAAE;EAClC,IAAI,CAACqC,eAAeA,YAAYxI,SAAS,aACvCM,SAASkF,KAAK4C,4BAA4B1H,KAAK,CAAC;CAEpD;CACA,OAAOJ;AACT;AAEA,MAAamI,2BACXnI,UACA2C,UACAoE,WACApG,UACAlB,cACAuH,UACG;CACH,MAAM1B,kBAAkBtF,SAAS8D,KAAKrD,YAAY;EAChD,IAAIZ,SAASmH,OAAOT,cAAc7C,IAAIjD,OAAO;EAC7C,IACE,CAACZ,UACDA,OAAOc,aAAaA,YACpBd,OAAO8C,aAAaA,UACpB;GACA9C,SAAS;IACP,GAAG6C,+BAA+BjC,SAASkC,UAAUhC,QAAQ;IAC7DA;IACAgC;GACF;GACAqE,OAAOT,cAAcpB,IAAI1E,SAASZ,MAAM;EAC1C;EACA,OAAOA;CACT,CAAC;CACD,MAAMuI,SAAS/C,sBAAsBC,iBAAiB7F,YAAY,CAAC,CAACqE,KACjErD,YAAY;EACX,MAAM4H,MAAM5H,QAAQK,QAAQ;EAC5B,IAAI,CAACuH,OAAO,CAACrB,OAAO,OAAOvG;EAE3B,MAAM6H,SAAStB,MAAMP,WAAW/C,IAAI2E,GAAG;EACvC,IAAIC,UAAUxC,kBAAkBwC,OAAOxH,SAASL,QAAQK,OAAO,GAC7D,OAAOwH;EAETtB,MAAMP,WAAWtB,IAAIkD,KAAK5H,OAAO;EACjC,OAAOA;CACT,CACF;CACA,MAAMZ,SAASmH,QACXA,MAAMN,eAAe6B,gBAAgBH,SAASE,QAAQ7H,SAASoG,QAC7DD,4BACEnG,SACAoG,KACAuB,OAAO1G,QACPqF,WACApG,SAASP,OACT;EACEK,SAAS6H;EACTnC,2BAA2Ba,MAAMb;CACnC,GACAxF,SAASN,mBACX,CACF,IACA+H,OAAOtE,KAAKrD,SAASoG,QACnBD,4BACEnG,SACAoG,KACAuB,OAAO1G,QACPqF,WACApG,SAASP,OACTiE,KAAAA,GACA1D,SAASN,mBACX,CACF;CACJ,OAAO4H,kCAAkCpI,QAAQc,SAASP,KAAK;AACjE"}
+{"version":3,"file":"external-message-conversion.js","names":["ReadonlyJSONValue","ToolExecutionStatus","ThreadMessageConverter","ThreadAssistantMessage","ThreadMessage","ToolCallMessagePart","MessageTiming","generateErrorMessageId","isJSONValueEqual","getAutoStatus","isAutoStatus","isBackgroundToolCall","isInterruptedToolCall","isPendingToolCall","bindExternalStoreMessage","FALLBACK_ID_PREFIX","getExternalStoreMessages","symbolInnerMessage","fromThreadMessageLike","ThreadMessageLike","JoinStrategy","ExternalMessageConverterMessage","convertConfig","joinStrategy","role","toolCallId","toolName","result","artifact","isError","messages","ExternalMessageConverterMetadata","toolStatuses","Record","error","cancelledMessageIds","ReadonlySet","messageTiming","ExternalMessageConverterCallback","message","T","metadata","ExternalMessageConverterCallbackResult","input","outputs","ExternalMessageConverterChunk","inputs","Mutable","P","stringifyForError","value","text","Error","String","JSON","stringify","length","slice","toCallbackOutputs","output","Array","isArray","o","valid","content","filter","warnedDroppedToolResults","has","add","console","warn","WeakSet","convertExternalMessageCallback","callback","mergeInnerMessages","existing","incoming","isNaNToolCallId","Number","isNaN","joinExternalMessages","assistantMessage","Omit","Exclude","toolCallIndices","Map","reasoningIndices","toolCallIdx","get","isPreliminary","_isPreliminary","toolCall","type","const","map","c","id","createdAt","status","attachments","unstable_state","undefined","unstable_annotations","unstable_data","steps","custom","timing","modality","submittedFeedback","isOptimistic","part","existingIdx","parentId","partIdx","push","set","unsupportedRole","chunkExternalMessages","callbackResults","results","isAssistant","pendingNone","flush","callbackResult","isVoice","at","shallowArrayEqual","a","b","i","ExternalMessageChunkConversionCache","generatedFallbackMessages","ExternalMessageConversionCallbackCacheEntry","InternalExternalMessageConversionCache","WeakKey","callbackCache","WeakMap","chunkCache","converterCache","createExternalMessageConversionCache","convertExternalMessageChunk","idx","chunkCount","isRunning","cache","isLast","joined","isCancelled","some","hasInterruptedToolCalls","hasPendingToolCalls","hasBackgroundToolCalls","autoStatus","fallbackId","cachedMessage","reason","updated","newMessage","createErrorAssistantMessage","msg","Date","completeExternalMessageConversion","lastMessage","convertExternalMessages","chunks","key","cached","convertMessages"],"sources":["../../../src/runtime/utils/external-message-conversion.ts"],"sourcesContent":["import type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport type { ToolExecutionStatus } from \"../../runtimes/tool-invocations/ToolInvocationTracker\";\nimport { ThreadMessageConverter } from \"../../runtimes/external-store/thread-message-converter\";\nimport type {\n  ThreadAssistantMessage,\n  ThreadMessage,\n  ToolCallMessagePart,\n} from \"../../types/message\";\nimport type { MessageTiming } from \"../../types/message\";\nimport { generateErrorMessageId } from \"../../utils/id\";\nimport { isJSONValueEqual } from \"../../utils/json/is-json-equal\";\nimport {\n  getAutoStatus,\n  isAutoStatus,\n  isBackgroundToolCall,\n  isInterruptedToolCall,\n  isPendingToolCall,\n} from \"./auto-status\";\nimport {\n  bindExternalStoreMessage,\n  FALLBACK_ID_PREFIX,\n  getExternalStoreMessages,\n  symbolInnerMessage,\n} from \"./external-store-message\";\nimport {\n  fromThreadMessageLike,\n  type ThreadMessageLike,\n} from \"./thread-message-like\";\n\nexport type JoinStrategy = \"concat-content\" | \"none\";\n\nexport type ExternalMessageConverterMessage =\n  | (ThreadMessageLike & {\n      readonly convertConfig?: {\n        readonly joinStrategy?: JoinStrategy;\n      };\n    })\n  | {\n      role: \"tool\";\n      toolCallId: string;\n      toolName?: string | undefined;\n      result: any;\n      artifact?: any;\n      isError?: boolean;\n      messages?: readonly ThreadMessage[];\n    };\n\nexport type ExternalMessageConverterMetadata = {\n  readonly toolStatuses?: Record<string, ToolExecutionStatus>;\n  readonly error?: ReadonlyJSONValue;\n  readonly cancelledMessageIds?: ReadonlySet<string>;\n  readonly messageTiming?: Record<string, MessageTiming>;\n};\n\nexport type ExternalMessageConverterCallback<T> = (\n  message: T,\n  metadata: ExternalMessageConverterMetadata,\n) => ExternalMessageConverterMessage | ExternalMessageConverterMessage[];\n\nexport type ExternalMessageConverterCallbackResult<T> = {\n  input: T;\n  outputs: ExternalMessageConverterMessage[];\n};\n\nexport type ExternalMessageConverterChunk<T> = {\n  inputs: T[];\n  outputs: ExternalMessageConverterMessage[];\n};\n\ntype Mutable<T> = {\n  -readonly [P in keyof T]: T[P];\n};\n\nconst stringifyForError = (value: unknown) => {\n  let text;\n  try {\n    text = value instanceof Error ? String(value) : JSON.stringify(value);\n  } catch {\n    /* circular */\n  }\n  text ??= String(value);\n  return text.length > 200 ? `${text.slice(0, 200)}…` : text;\n};\n\nconst toCallbackOutputs = (\n  output: ExternalMessageConverterMessage | ExternalMessageConverterMessage[],\n  input: unknown,\n): ExternalMessageConverterMessage[] => {\n  const outputs = Array.isArray(output) ? output : [output];\n  for (const o of outputs) {\n    const valid =\n      typeof o === \"object\" &&\n      o !== null &&\n      (o.role === \"tool\" ||\n        ((o.role === \"assistant\" || o.role === \"user\" || o.role === \"system\") &&\n          (typeof o.content === \"string\" || Array.isArray(o.content))));\n    if (!valid)\n      throw new Error(\n        `External message converter: the converter callback returned an invalid message (${stringifyForError(o)}) for input ${stringifyForError(input)}. Return an empty array to skip a message.`,\n      );\n  }\n  // Providers can emit tool results without a usable id (react-langchain maps\n  // a missing tool_call_id to \"\"); converters must not throw on those, so the\n  // result is dropped like any other orphaned tool output.\n  return outputs.filter((o) => {\n    if (o.role !== \"tool\") return true;\n    if (typeof o.toolCallId === \"string\" && o.toolCallId.length > 0)\n      return true;\n    if (!warnedDroppedToolResults.has(o)) {\n      warnedDroppedToolResults.add(o);\n      console.warn(\n        `External message converter: dropping a tool result without a toolCallId (${stringifyForError(o)}) for input ${stringifyForError(input)}.`,\n      );\n    }\n    return false;\n  });\n};\n\nconst warnedDroppedToolResults = new WeakSet<object>();\n\nexport const convertExternalMessageCallback = <T>(\n  input: T,\n  callback: ExternalMessageConverterCallback<T>,\n  metadata: ExternalMessageConverterMetadata,\n): ExternalMessageConverterCallbackResult<T> => ({\n  input,\n  outputs: toCallbackOutputs(callback(input, metadata), input),\n});\n\nconst mergeInnerMessages = (existing: object, incoming: object) => ({\n  [symbolInnerMessage]: [\n    ...((existing as any)[symbolInnerMessage] ?? []),\n    ...((incoming as any)[symbolInnerMessage] ?? []),\n  ],\n});\n\nconst isNaNToolCallId = (toolCallId: unknown) =>\n  typeof toolCallId === \"number\" && Number.isNaN(toolCallId);\n\nexport const joinExternalMessages = (\n  messages: readonly ExternalMessageConverterMessage[],\n): ThreadMessageLike => {\n  const assistantMessage: Mutable<Omit<ThreadMessageLike, \"metadata\">> & {\n    content: Exclude<ThreadMessageLike[\"content\"][0], string>[];\n    metadata?: Mutable<ThreadMessageLike[\"metadata\"]>;\n  } = {\n    role: \"assistant\",\n    content: [],\n  };\n  const toolCallIndices = new Map<unknown, number>();\n  const reasoningIndices = new Map<string, number>();\n  for (const output of messages) {\n    if (output.role === \"tool\") {\n      const toolCallIdx = !isNaNToolCallId(output.toolCallId)\n        ? (toolCallIndices.get(output.toolCallId) ?? -1)\n        : -1;\n      // Ignore orphaned tool results so one bad tool message does not\n      // prevent rendering the rest of the conversation.\n      if (toolCallIdx !== -1) {\n        const { isPreliminary: _isPreliminary, ...toolCall } = assistantMessage\n          .content[toolCallIdx]! as ToolCallMessagePart;\n        if (output.toolName != null) {\n          if (toolCall.toolName !== output.toolName)\n            throw new Error(\n              `Tool call name ${output.toolCallId} ${output.toolName} does not match existing tool call ${toolCall.toolName}`,\n            );\n        }\n        assistantMessage.content[toolCallIdx] = {\n          ...toolCall,\n          ...{\n            [symbolInnerMessage]: [\n              ...((toolCall as any)[symbolInnerMessage] ?? []),\n              output,\n            ],\n          },\n          result: output.result,\n          artifact: output.artifact,\n          isError: output.isError,\n          messages: output.messages,\n        };\n      }\n    } else {\n      const role = output.role;\n      const content = (\n        typeof output.content === \"string\"\n          ? [{ type: \"text\" as const, text: output.content }]\n          : output.content\n      ).map((c) => ({\n        ...c,\n        ...{ [symbolInnerMessage]: [output] },\n      }));\n      switch (role) {\n        case \"system\":\n        case \"user\":\n          return {\n            ...output,\n            content,\n          };\n        case \"assistant\":\n          if (assistantMessage.content.length === 0) {\n            assistantMessage.id = output.id;\n            assistantMessage.createdAt ??= output.createdAt;\n            assistantMessage.status ??= output.status;\n\n            if (output.attachments) {\n              assistantMessage.attachments = [\n                ...(assistantMessage.attachments ?? []),\n                ...output.attachments,\n              ];\n            }\n          }\n\n          if (output.metadata) {\n            assistantMessage.metadata ??= {};\n            if (output.metadata.unstable_state !== undefined) {\n              assistantMessage.metadata.unstable_state =\n                output.metadata.unstable_state;\n            }\n            if (output.metadata.unstable_annotations) {\n              assistantMessage.metadata.unstable_annotations = [\n                ...(assistantMessage.metadata.unstable_annotations ?? []),\n                ...output.metadata.unstable_annotations,\n              ];\n            }\n            if (output.metadata.unstable_data) {\n              assistantMessage.metadata.unstable_data = [\n                ...(assistantMessage.metadata.unstable_data ?? []),\n                ...output.metadata.unstable_data,\n              ];\n            }\n            if (output.metadata.steps) {\n              assistantMessage.metadata.steps = [\n                ...(assistantMessage.metadata.steps ?? []),\n                ...output.metadata.steps,\n              ];\n            }\n            if (output.metadata.custom) {\n              assistantMessage.metadata.custom = {\n                ...(assistantMessage.metadata.custom ?? {}),\n                ...output.metadata.custom,\n              };\n            }\n\n            if (output.metadata.timing) {\n              assistantMessage.metadata.timing = output.metadata.timing;\n            }\n\n            if (output.metadata.modality) {\n              assistantMessage.metadata.modality = output.metadata.modality;\n            }\n\n            if (output.metadata.submittedFeedback) {\n              assistantMessage.metadata.submittedFeedback =\n                output.metadata.submittedFeedback;\n            }\n\n            if (output.metadata.isOptimistic) {\n              assistantMessage.metadata.isOptimistic = true;\n            }\n            // TODO keep this in sync with ThreadMessageLike[\"metadata\"] / fromThreadMessageLike\n          }\n\n          // Add content parts, merging reasoning parts with same parentId\n          for (const part of content) {\n            if (part.type === \"tool-call\" && part.toolCallId) {\n              const existingIdx = toolCallIndices.get(part.toolCallId);\n              if (existingIdx !== undefined) {\n                const existing = assistantMessage.content[\n                  existingIdx\n                ] as typeof part;\n                assistantMessage.content[existingIdx] = {\n                  ...existing,\n                  ...part,\n                  ...mergeInnerMessages(existing, part),\n                };\n                continue;\n              }\n            }\n\n            if (\n              part.type === \"reasoning\" &&\n              \"parentId\" in part &&\n              part.parentId\n            ) {\n              const existingIdx = reasoningIndices.get(part.parentId);\n              if (existingIdx !== undefined) {\n                const existing = assistantMessage.content[\n                  existingIdx\n                ] as typeof part;\n                assistantMessage.content[existingIdx] = {\n                  ...existing,\n                  text: `${existing.text}\\n\\n${part.text}`,\n                  ...mergeInnerMessages(existing, part),\n                };\n                continue;\n              }\n            }\n            const partIdx = assistantMessage.content.length;\n            assistantMessage.content.push(part);\n            if (\n              part.type === \"tool-call\" &&\n              !isNaNToolCallId(part.toolCallId) &&\n              !toolCallIndices.has(part.toolCallId)\n            ) {\n              toolCallIndices.set(part.toolCallId, partIdx);\n            } else if (\n              part.type === \"reasoning\" &&\n              \"parentId\" in part &&\n              part.parentId\n            ) {\n              reasoningIndices.set(part.parentId, partIdx);\n            }\n          }\n          break;\n        default: {\n          const unsupportedRole: never = role;\n          throw new Error(`Unknown message role: ${unsupportedRole}`);\n        }\n      }\n    }\n  }\n  return assistantMessage;\n};\n\nexport const chunkExternalMessages = <T>(\n  callbackResults: ExternalMessageConverterCallbackResult<T>[],\n  joinStrategy?: JoinStrategy,\n) => {\n  const results: ExternalMessageConverterChunk<T>[] = [];\n  let isAssistant = false;\n  let pendingNone = false; // true if the previous assistant message had joinStrategy \"none\"\n  let inputs: T[] = [];\n  let outputs: ExternalMessageConverterMessage[] = [];\n\n  const flush = () => {\n    if (outputs.length) {\n      results.push({\n        inputs,\n        outputs,\n      });\n    }\n    inputs = [];\n    outputs = [];\n    isAssistant = false;\n    pendingNone = false;\n  };\n\n  for (const callbackResult of callbackResults) {\n    for (const output of callbackResult.outputs) {\n      const isVoice =\n        output.role === \"assistant\" && output.metadata?.modality === \"voice\";\n      if (\n        (pendingNone && output.role !== \"tool\") ||\n        !isAssistant ||\n        output.role === \"user\" ||\n        output.role === \"system\" ||\n        isVoice\n      ) {\n        flush();\n      }\n      isAssistant = output.role === \"assistant\" || output.role === \"tool\";\n\n      if (inputs.at(-1) !== callbackResult.input) {\n        inputs.push(callbackResult.input);\n      }\n      outputs.push(output);\n\n      if (\n        output.role === \"assistant\" &&\n        (isVoice ||\n          output.convertConfig?.joinStrategy === \"none\" ||\n          joinStrategy === \"none\")\n      ) {\n        pendingNone = true;\n      }\n    }\n  }\n  flush();\n  return results;\n};\n\nexport const shallowArrayEqual = (\n  a: readonly unknown[],\n  b: readonly unknown[],\n) => {\n  if (a.length !== b.length) return false;\n  for (let i = 0; i < a.length; i++) {\n    if (a[i] !== b[i]) return false;\n  }\n  return true;\n};\n\ntype ExternalMessageChunkConversionCache = {\n  message: ThreadMessage | undefined;\n  generatedFallbackMessages: WeakSet<object>;\n};\n\ntype ExternalMessageConversionCallbackCacheEntry<T> =\n  ExternalMessageConverterCallbackResult<T> & {\n    metadata: ExternalMessageConverterMetadata;\n    callback: ExternalMessageConverterCallback<T>;\n  };\n\nexport type InternalExternalMessageConversionCache<\n  T extends WeakKey = WeakKey,\n> = {\n  callbackCache: WeakMap<T, ExternalMessageConversionCallbackCacheEntry<T>>;\n  chunkCache: WeakMap<\n    ExternalMessageConverterMessage,\n    ExternalMessageConverterChunk<T>\n  >;\n  converterCache: ThreadMessageConverter;\n  // Generatedness is tracked by identity, not by id shape: a caller-supplied id that happens to match the generated pattern must never be rewritten.\n  generatedFallbackMessages: WeakSet<object>;\n};\n\nexport const createExternalMessageConversionCache = <\n  T extends WeakKey = WeakKey,\n>(): InternalExternalMessageConversionCache<T> => ({\n  callbackCache: new WeakMap(),\n  chunkCache: new WeakMap(),\n  converterCache: new ThreadMessageConverter(),\n  generatedFallbackMessages: new WeakSet(),\n});\n\nexport const convertExternalMessageChunk = <T>(\n  message: ExternalMessageConverterChunk<T>,\n  idx: number,\n  chunkCount: number,\n  isRunning: boolean,\n  error: ReadonlyJSONValue | undefined,\n  cache?: ExternalMessageChunkConversionCache,\n  cancelledMessageIds?: ReadonlySet<string>,\n) => {\n  const isLast = idx === chunkCount - 1;\n  const joined = joinExternalMessages(message.outputs);\n  const isCancelled =\n    cancelledMessageIds !== undefined &&\n    message.outputs.some(\n      (output) =>\n        output.role !== \"tool\" &&\n        output.id != null &&\n        cancelledMessageIds.has(output.id),\n    );\n  const hasInterruptedToolCalls =\n    typeof joined.content === \"object\" &&\n    joined.content.some(isInterruptedToolCall);\n  const hasPendingToolCalls =\n    typeof joined.content === \"object\" &&\n    joined.content.some(isPendingToolCall);\n  const hasBackgroundToolCalls =\n    typeof joined.content === \"object\" &&\n    joined.content.some(isBackgroundToolCall);\n  const autoStatus = getAutoStatus(\n    isLast,\n    isRunning,\n    hasInterruptedToolCalls,\n    hasPendingToolCalls,\n    isLast ? error : undefined,\n    isCancelled,\n    hasBackgroundToolCalls,\n  );\n  const fallbackId = `${FALLBACK_ID_PREFIX}${idx}`;\n\n  const cachedMessage = cache?.message;\n  if (\n    cachedMessage &&\n    (cachedMessage.role !== \"assistant\" ||\n      !isAutoStatus(cachedMessage.status) ||\n      cachedMessage.status === autoStatus ||\n      (cachedMessage.status.type === \"incomplete\" &&\n        cachedMessage.status.reason === \"error\" &&\n        cachedMessage.status.error !== undefined &&\n        autoStatus.type === \"incomplete\" &&\n        autoStatus.reason === \"error\" &&\n        autoStatus.error !== undefined &&\n        (cachedMessage.status.error === autoStatus.error ||\n          isJSONValueEqual(cachedMessage.status.error, autoStatus.error))))\n  ) {\n    const inputs = getExternalStoreMessages<T>(cachedMessage);\n    if (shallowArrayEqual(inputs, message.inputs)) {\n      // A positional fallback id goes stale when messages are prepended\n      // or reordered; serving it unchanged makes two messages collide on\n      // one id and the dedup downstream drops one of them.\n      if (\n        cache.generatedFallbackMessages.has(cachedMessage) &&\n        cachedMessage.id !== fallbackId\n      ) {\n        const updated = { ...cachedMessage, id: fallbackId };\n        cache.generatedFallbackMessages.add(updated);\n        bindExternalStoreMessage(updated, message.inputs);\n        return updated;\n      }\n      return cachedMessage;\n    }\n  }\n\n  const newMessage = fromThreadMessageLike(joined, fallbackId, autoStatus);\n  if (cache && joined.id == null) {\n    cache.generatedFallbackMessages.add(newMessage);\n  }\n  bindExternalStoreMessage(newMessage, message.inputs);\n  return newMessage;\n};\n\nfunction createErrorAssistantMessage(\n  error: ReadonlyJSONValue,\n): ThreadAssistantMessage {\n  const msg: ThreadAssistantMessage = {\n    id: generateErrorMessageId(),\n    role: \"assistant\",\n    content: [],\n    status: { type: \"incomplete\", reason: \"error\", error },\n    createdAt: new Date(),\n    metadata: {\n      unstable_state: null,\n      unstable_annotations: [],\n      unstable_data: [],\n      custom: {},\n      steps: [],\n    },\n  };\n  bindExternalStoreMessage(msg, []);\n  return msg;\n}\n\nexport const completeExternalMessageConversion = (\n  messages: ThreadMessage[],\n  error: ReadonlyJSONValue | undefined,\n) => {\n  if (error !== undefined && error !== null) {\n    const lastMessage = messages.at(-1);\n    if (!lastMessage || lastMessage.role !== \"assistant\") {\n      messages.push(createErrorAssistantMessage(error));\n    }\n  }\n  return messages;\n};\n\nexport const convertExternalMessages = <T extends WeakKey>(\n  messages: T[],\n  callback: ExternalMessageConverterCallback<T>,\n  isRunning: boolean,\n  metadata: ExternalMessageConverterMetadata,\n  joinStrategy?: JoinStrategy,\n  cache?: InternalExternalMessageConversionCache<T>,\n) => {\n  const callbackResults = messages.map((message) => {\n    let result = cache?.callbackCache.get(message);\n    if (\n      !result ||\n      result.metadata !== metadata ||\n      result.callback !== callback\n    ) {\n      result = {\n        ...convertExternalMessageCallback(message, callback, metadata),\n        metadata,\n        callback,\n      };\n      cache?.callbackCache.set(message, result);\n    }\n    return result;\n  });\n  const chunks = chunkExternalMessages(callbackResults, joinStrategy).map(\n    (message) => {\n      const key = message.outputs[0];\n      if (!key || !cache) return message;\n\n      const cached = cache.chunkCache.get(key);\n      if (cached && shallowArrayEqual(cached.outputs, message.outputs)) {\n        return cached;\n      }\n      cache.chunkCache.set(key, message);\n      return message;\n    },\n  );\n  const result = cache\n    ? cache.converterCache.convertMessages(chunks, (cached, message, idx) =>\n        convertExternalMessageChunk(\n          message,\n          idx,\n          chunks.length,\n          isRunning,\n          metadata.error,\n          {\n            message: cached,\n            generatedFallbackMessages: cache.generatedFallbackMessages,\n          },\n          metadata.cancelledMessageIds,\n        ),\n      )\n    : chunks.map((message, idx) =>\n        convertExternalMessageChunk(\n          message,\n          idx,\n          chunks.length,\n          isRunning,\n          metadata.error,\n          undefined,\n          metadata.cancelledMessageIds,\n        ),\n      );\n  return completeExternalMessageConversion(result, metadata.error);\n};\n"],"mappings":";;;;;;;AAyEA,MAAMiD,qBAAqBC,UAAmB;CAC5C,IAAIC;CACJ,IAAI;EACFA,OAAOD,iBAAiBE,QAAQC,OAAOH,KAAK,IAAII,KAAKC,UAAUL,KAAK;CACtE,QAAQ,CACN;CAEFC,SAASE,OAAOH,KAAK;CACrB,OAAOC,KAAKK,SAAS,MAAM,GAAGL,KAAKM,MAAM,GAAG,GAAG,EAAC,KAAMN;AACxD;AAEA,MAAMO,qBACJC,QACAhB,UACsC;CACtC,MAAMC,UAAUgB,MAAMC,QAAQF,MAAM,IAAIA,SAAS,CAACA,MAAM;CACxD,KAAK,MAAMG,KAAKlB,SAOd,IAAI,EALF,OAAOkB,MAAM,YACbA,MAAM,SACLA,EAAEtC,SAAS,WACRsC,EAAEtC,SAAS,eAAesC,EAAEtC,SAAS,UAAUsC,EAAEtC,SAAS,cACzD,OAAOsC,EAAEE,YAAY,YAAYJ,MAAMC,QAAQC,EAAEE,OAAO,MAE7D,MAAM,IAAIZ,MACR,mFAAmFH,kBAAkBa,CAAC,EAAC,cAAeb,kBAAkBN,KAAK,EAAC,2CAChJ;CAKJ,OAAOC,QAAQqB,QAAQH,MAAM;EAC3B,IAAIA,EAAEtC,SAAS,QAAQ,OAAO;EAC9B,IAAI,OAAOsC,EAAErC,eAAe,YAAYqC,EAAErC,WAAW+B,SAAS,GAC5D,OAAO;EACT,IAAI,CAACU,yBAAyBC,IAAIL,CAAC,GAAG;GACpCI,yBAAyBE,IAAIN,CAAC;GAC9BO,QAAQC,KACN,4EAA4ErB,kBAAkBa,CAAC,EAAC,cAAeb,kBAAkBN,KAAK,EAAC,EACzI;EACF;EACA,OAAO;CACT,CAAC;AACH;AAEA,MAAMuB,2CAA2B,IAAIK,QAAgB;AAErD,MAAaC,kCACX7B,OACA8B,UACAhC,cAC+C;CAC/CE;CACAC,SAASc,kBAAkBe,SAAS9B,OAAOF,QAAQ,GAAGE,KAAK;AAC7D;AAEA,MAAM+B,sBAAsBC,UAAkBC,cAAsB,GACjE3D,qBAAqB,CACpB,GAAK0D,SAAiB1D,uBAAuB,CAAA,GAC7C,GAAK2D,SAAiB3D,uBAAuB,CAAA,CAAG,EAEpD;AAEA,MAAM4D,mBAAmBpD,eACvB,OAAOA,eAAe,YAAYqD,OAAOC,MAAMtD,UAAU;AAE3D,MAAauD,wBACXlD,aACsB;CACtB,MAAMmD,mBAGF;EACFzD,MAAM;EACNwC,SAAS,CAAA;CACX;CACA,MAAMoB,kCAAkB,IAAIC,IAAqB;CACjD,MAAMC,mCAAmB,IAAID,IAAoB;CACjD,KAAK,MAAM1B,UAAU7B,UACnB,IAAI6B,OAAOnC,SAAS,QAAQ;EAC1B,MAAM+D,cAAc,CAACV,gBAAgBlB,OAAOlC,UAAU,IACjD2D,gBAAgBI,IAAI7B,OAAOlC,UAAU,KAAK,KAC3C;EAGJ,IAAI8D,gBAAgB,IAAI;GACtB,MAAM,EAAEE,eAAeC,gBAAgB,GAAGC,aAAaV,iBACpDjB,QAAQuB;GACX,IAAI5B,OAAOjC,YAAY,MACjBiE;QAAAA,SAASjE,aAAaiC,OAAOjC,UAC/B,MAAM,IAAI0B,MACR,kBAAkBO,OAAOlC,WAAU,GAAIkC,OAAOjC,SAAQ,qCAAsCiE,SAASjE,UACvG;GAAA;GAEJuD,iBAAiBjB,QAAQuB,eAAe;IACtC,GAAGI;KAEA1E,qBAAqB,CACpB,GAAK0E,SAAiB1E,uBAAuB,CAAA,GAC7C0C,MAAM;IAGVhC,QAAQgC,OAAOhC;IACfC,UAAU+B,OAAO/B;IACjBC,SAAS8B,OAAO9B;IAChBC,UAAU6B,OAAO7B;GACnB;EACF;CACF,OAAO;EACL,MAAMN,OAAOmC,OAAOnC;EACpB,MAAMwC,WACJ,OAAOL,OAAOK,YAAY,WACtB,CAAC;GAAE4B,MAAM;GAAiBzC,MAAMQ,OAAOK;EAAQ,CAAC,IAChDL,OAAOK,QAAAA,CACX8B,KAAKC,OAAO;GACZ,GAAGA;IACG9E,qBAAqB,CAAC0C,MAAM;EACpC,EAAE;EACF,QAAQnC,MAAR;GACE,KAAK;GACL,KAAK,QACH,OAAO;IACL,GAAGmC;IACHK;GACF;GACF,KAAK;IACH,IAAIiB,iBAAiBjB,QAAQR,WAAW,GAAG;KACzCyB,iBAAiBe,KAAKrC,OAAOqC;KAC7Bf,iBAAiBgB,cAActC,OAAOsC;KACtChB,iBAAiBiB,WAAWvC,OAAOuC;KAEnC,IAAIvC,OAAOwC,aACTlB,iBAAiBkB,cAAc,CAC7B,GAAIlB,iBAAiBkB,eAAe,CAAA,GACpC,GAAGxC,OAAOwC,WAAW;IAG3B;IAEA,IAAIxC,OAAOlB,UAAU;KACnBwC,iBAAiBxC,aAAa,CAAC;KAC/B,IAAIkB,OAAOlB,SAAS2D,mBAAmBC,KAAAA,GACrCpB,iBAAiBxC,SAAS2D,iBACxBzC,OAAOlB,SAAS2D;KAEpB,IAAIzC,OAAOlB,SAAS6D,sBAClBrB,iBAAiBxC,SAAS6D,uBAAuB,CAC/C,GAAIrB,iBAAiBxC,SAAS6D,wBAAwB,CAAA,GACtD,GAAG3C,OAAOlB,SAAS6D,oBAAoB;KAG3C,IAAI3C,OAAOlB,SAAS8D,eAClBtB,iBAAiBxC,SAAS8D,gBAAgB,CACxC,GAAItB,iBAAiBxC,SAAS8D,iBAAiB,CAAA,GAC/C,GAAG5C,OAAOlB,SAAS8D,aAAa;KAGpC,IAAI5C,OAAOlB,SAAS+D,OAClBvB,iBAAiBxC,SAAS+D,QAAQ,CAChC,GAAIvB,iBAAiBxC,SAAS+D,SAAS,CAAA,GACvC,GAAG7C,OAAOlB,SAAS+D,KAAK;KAG5B,IAAI7C,OAAOlB,SAASgE,QAClBxB,iBAAiBxC,SAASgE,SAAS;MACjC,GAAIxB,iBAAiBxC,SAASgE,UAAU,CAAC;MACzC,GAAG9C,OAAOlB,SAASgE;KACrB;KAGF,IAAI9C,OAAOlB,SAASiE,QAClBzB,iBAAiBxC,SAASiE,SAAS/C,OAAOlB,SAASiE;KAGrD,IAAI/C,OAAOlB,SAASkE,UAClB1B,iBAAiBxC,SAASkE,WAAWhD,OAAOlB,SAASkE;KAGvD,IAAIhD,OAAOlB,SAASmE,mBAClB3B,iBAAiBxC,SAASmE,oBACxBjD,OAAOlB,SAASmE;KAGpB,IAAIjD,OAAOlB,SAASoE,cAClB5B,iBAAiBxC,SAASoE,eAAe;IAG7C;IAGA,KAAK,MAAMC,QAAQ9C,SAAS;KAC1B,IAAI8C,KAAKlB,SAAS,eAAekB,KAAKrF,YAAY;MAChD,MAAMsF,cAAc3B,gBAAgBI,IAAIsB,KAAKrF,UAAU;MACvD,IAAIsF,gBAAgBV,KAAAA,GAAW;OAC7B,MAAM1B,WAAWM,iBAAiBjB,QAChC+C;OAEF9B,iBAAiBjB,QAAQ+C,eAAe;QACtC,GAAGpC;QACH,GAAGmC;QACH,GAAGpC,mBAAmBC,UAAUmC,IAAI;OACtC;OACA;MACF;KACF;KAEA,IACEA,KAAKlB,SAAS,eACd,cAAckB,QACdA,KAAKE,UACL;MACA,MAAMD,cAAczB,iBAAiBE,IAAIsB,KAAKE,QAAQ;MACtD,IAAID,gBAAgBV,KAAAA,GAAW;OAC7B,MAAM1B,WAAWM,iBAAiBjB,QAChC+C;OAEF9B,iBAAiBjB,QAAQ+C,eAAe;QACtC,GAAGpC;QACHxB,MAAM,GAAGwB,SAASxB,KAAI,MAAO2D,KAAK3D;QAClC,GAAGuB,mBAAmBC,UAAUmC,IAAI;OACtC;OACA;MACF;KACF;KACA,MAAMG,UAAUhC,iBAAiBjB,QAAQR;KACzCyB,iBAAiBjB,QAAQkD,KAAKJ,IAAI;KAClC,IACEA,KAAKlB,SAAS,eACd,CAACf,gBAAgBiC,KAAKrF,UAAU,KAChC,CAAC2D,gBAAgBjB,IAAI2C,KAAKrF,UAAU,GAEpC2D,gBAAgB+B,IAAIL,KAAKrF,YAAYwF,OAAO;UACvC,IACLH,KAAKlB,SAAS,eACd,cAAckB,QACdA,KAAKE,UAEL1B,iBAAiB6B,IAAIL,KAAKE,UAAUC,OAAO;IAE/C;IACA;GACF,SAEE,MAAM,IAAI7D,MAAM,yBAAyBgE,MAAiB;EAE9D;CACF;CAEF,OAAOnC;AACT;AAEA,MAAaoC,yBACXC,iBACA/F,iBACG;CACH,MAAMgG,UAA8C,CAAA;CACpD,IAAIC,cAAc;CAClB,IAAIC,cAAc;CAClB,IAAI3E,SAAc,CAAA;CAClB,IAAIF,UAA6C,CAAA;CAEjD,MAAM8E,cAAc;EAClB,IAAI9E,QAAQY,QACV+D,QAAQL,KAAK;GACXpE;GACAF;EACF,CAAC;EAEHE,SAAS,CAAA;EACTF,UAAU,CAAA;EACV4E,cAAc;EACdC,cAAc;CAChB;CAEA,KAAK,MAAME,kBAAkBL,iBAC3B,KAAK,MAAM3D,UAAUgE,eAAe/E,SAAS;EAC3C,MAAMgF,UACJjE,OAAOnC,SAAS,eAAemC,OAAOlB,UAAUkE,aAAa;EAC/D,IACGc,eAAe9D,OAAOnC,SAAS,UAChC,CAACgG,eACD7D,OAAOnC,SAAS,UAChBmC,OAAOnC,SAAS,YAChBoG,SAEAF,MAAM;EAERF,cAAc7D,OAAOnC,SAAS,eAAemC,OAAOnC,SAAS;EAE7D,IAAIsB,OAAO+E,GAAG,EAAE,MAAMF,eAAehF,OACnCG,OAAOoE,KAAKS,eAAehF,KAAK;EAElCC,QAAQsE,KAAKvD,MAAM;EAEnB,IACEA,OAAOnC,SAAS,gBACfoG,WACCjE,OAAOrC,eAAeC,iBAAiB,UACvCA,iBAAiB,SAEnBkG,cAAc;CAElB;CAEFC,MAAM;CACN,OAAOH;AACT;AAEA,MAAaO,qBACXC,GACAC,MACG;CACH,IAAID,EAAEvE,WAAWwE,EAAExE,QAAQ,OAAO;CAClC,KAAK,IAAIyE,IAAI,GAAGA,IAAIF,EAAEvE,QAAQyE,KAC5B,IAAIF,EAAEE,OAAOD,EAAEC,IAAI,OAAO;CAE5B,OAAO;AACT;AA0BA,MAAaU,8CAEsC;CACjDJ,+BAAe,IAAIC,QAAQ;CAC3BC,4BAAY,IAAID,QAAQ;CACxBE,gBAAgB,IAAIxI,uBAAuB;CAC3CiI,2CAA2B,IAAI5D,QAAQ;AACzC;AAEA,MAAaqE,+BACXrG,SACAsG,KACAC,YACAC,WACA7G,OACA8G,OACA7G,wBACG;CACH,MAAM8G,SAASJ,QAAQC,aAAa;CACpC,MAAMI,SAASlE,qBAAqBzC,QAAQK,OAAO;CACnD,MAAMuG,cACJhH,wBAAwBkE,KAAAA,KACxB9D,QAAQK,QAAQwG,MACbzF,WACCA,OAAOnC,SAAS,UAChBmC,OAAOqC,MAAM,QACb7D,oBAAoBgC,IAAIR,OAAOqC,EAAE,CACrC;CACF,MAAMqD,0BACJ,OAAOH,OAAOlF,YAAY,YAC1BkF,OAAOlF,QAAQoF,KAAKxI,qBAAqB;CAC3C,MAAM0I,sBACJ,OAAOJ,OAAOlF,YAAY,YAC1BkF,OAAOlF,QAAQoF,KAAKvI,iBAAiB;CACvC,MAAM0I,yBACJ,OAAOL,OAAOlF,YAAY,YAC1BkF,OAAOlF,QAAQoF,KAAKzI,oBAAoB;CAC1C,MAAM6I,aAAa/I,cACjBwI,QACAF,WACAM,yBACAC,qBACAL,SAAS/G,QAAQmE,KAAAA,GACjB8C,aACAI,sBACF;CACA,MAAME,aAAa,GAAG1I,qBAAqB8H;CAE3C,MAAMa,gBAAgBV,OAAOzG;CAC7B,IACEmH,kBACCA,cAAclI,SAAS,eACtB,CAACd,aAAagJ,cAAcxD,MAAM,KAClCwD,cAAcxD,WAAWsD,cACxBE,cAAcxD,OAAON,SAAS,gBAC7B8D,cAAcxD,OAAOyD,WAAW,WAChCD,cAAcxD,OAAOhE,UAAUmE,KAAAA,KAC/BmD,WAAW5D,SAAS,gBACpB4D,WAAWG,WAAW,WACtBH,WAAWtH,UAAUmE,KAAAA,MACpBqD,cAAcxD,OAAOhE,UAAUsH,WAAWtH,SACzC1B,iBAAiBkJ,cAAcxD,OAAOhE,OAAOsH,WAAWtH,KAAK,KACnE;EACA,MAAMY,SAAS9B,yBAA4B0I,aAAa;EACxD,IAAI5B,kBAAkBhF,QAAQP,QAAQO,MAAM,GAAG;GAI7C,IACEkG,MAAMb,0BAA0BhE,IAAIuF,aAAa,KACjDA,cAAc1D,OAAOyD,YACrB;IACA,MAAMG,UAAU;KAAE,GAAGF;KAAe1D,IAAIyD;IAAW;IACnDT,MAAMb,0BAA0B/D,IAAIwF,OAAO;IAC3C9I,yBAAyB8I,SAASrH,QAAQO,MAAM;IAChD,OAAO8G;GACT;GACA,OAAOF;EACT;CACF;CAEA,MAAMG,aAAa3I,sBAAsBgI,QAAQO,YAAYD,UAAU;CACvE,IAAIR,SAASE,OAAOlD,MAAM,MACxBgD,MAAMb,0BAA0B/D,IAAIyF,UAAU;CAEhD/I,yBAAyB+I,YAAYtH,QAAQO,MAAM;CACnD,OAAO+G;AACT;AAEA,SAASC,4BACP5H,OACwB;CACxB,MAAM6H,MAA8B;EAClC/D,IAAIzF,uBAAuB;EAC3BiB,MAAM;EACNwC,SAAS,CAAA;EACTkC,QAAQ;GAAEN,MAAM;GAAc+D,QAAQ;GAASzH;EAAM;EACrD+D,2BAAW,IAAI+D,KAAK;EACpBvH,UAAU;GACR2D,gBAAgB;GAChBE,sBAAsB,CAAA;GACtBC,eAAe,CAAA;GACfE,QAAQ,CAAC;GACTD,OAAO,CAAA;EACT;CACF;CACA1F,yBAAyBiJ,KAAK,CAAA,CAAE;CAChC,OAAOA;AACT;AAEA,MAAaE,qCACXnI,UACAI,UACG;CACH,IAAIA,UAAUmE,KAAAA,KAAanE,UAAU,MAAM;EACzC,MAAMgI,cAAcpI,SAAS+F,GAAG,EAAE;EAClC,IAAI,CAACqC,eAAeA,YAAY1I,SAAS,aACvCM,SAASoF,KAAK4C,4BAA4B5H,KAAK,CAAC;CAEpD;CACA,OAAOJ;AACT;AAEA,MAAaqI,2BACXrI,UACA2C,UACAsE,WACAtG,UACAlB,cACAyH,UACG;CACH,MAAM1B,kBAAkBxF,SAASgE,KAAKvD,YAAY;EAChD,IAAIZ,SAASqH,OAAOT,cAAc/C,IAAIjD,OAAO;EAC7C,IACE,CAACZ,UACDA,OAAOc,aAAaA,YACpBd,OAAO8C,aAAaA,UACpB;GACA9C,SAAS;IACP,GAAG6C,+BAA+BjC,SAASkC,UAAUhC,QAAQ;IAC7DA;IACAgC;GACF;GACAuE,OAAOT,cAAcpB,IAAI5E,SAASZ,MAAM;EAC1C;EACA,OAAOA;CACT,CAAC;CACD,MAAMyI,SAAS/C,sBAAsBC,iBAAiB/F,YAAY,CAAC,CAACuE,KACjEvD,YAAY;EACX,MAAM8H,MAAM9H,QAAQK,QAAQ;EAC5B,IAAI,CAACyH,OAAO,CAACrB,OAAO,OAAOzG;EAE3B,MAAM+H,SAAStB,MAAMP,WAAWjD,IAAI6E,GAAG;EACvC,IAAIC,UAAUxC,kBAAkBwC,OAAO1H,SAASL,QAAQK,OAAO,GAC7D,OAAO0H;EAETtB,MAAMP,WAAWtB,IAAIkD,KAAK9H,OAAO;EACjC,OAAOA;CACT,CACF;CACA,MAAMZ,SAASqH,QACXA,MAAMN,eAAe6B,gBAAgBH,SAASE,QAAQ/H,SAASsG,QAC7DD,4BACErG,SACAsG,KACAuB,OAAO5G,QACPuF,WACAtG,SAASP,OACT;EACEK,SAAS+H;EACTnC,2BAA2Ba,MAAMb;CACnC,GACA1F,SAASN,mBACX,CACF,IACAiI,OAAOtE,KAAKvD,SAASsG,QACnBD,4BACErG,SACAsG,KACAuB,OAAO5G,QACPuF,WACAtG,SAASP,OACTmE,KAAAA,GACA5D,SAASN,mBACX,CACF;CACJ,OAAO8H,kCAAkCtI,QAAQc,SAASP,KAAK;AACjE"}

package/dist/runtime/utils/external-store-message.d.ts

@@ -1,5 +1,4 @@
-import { ThreadMessage } from "../../types/message.js";
-//#region src/runtime/utils/external-store-message.d.ts
+import type { ThreadMessage } from "../../types/message.js";
 export declare const symbolInnerMessage: unique symbol;
 /**
  * Attach the original external store message(s) to a ThreadMessage or message part.
@@ -10,8 +9,7 @@ export declare const symbolInnerMessage: unique symbol;
  */
 export declare const bindExternalStoreMessage: <T>(target: object, message: T | T[]) => void;
 export declare const getExternalStoreMessages: <T>(input: {
-  messages: readonly ThreadMessage[];
+    messages: readonly ThreadMessage[];
 } | ThreadMessage | ThreadMessage["content"][number]) => T[];
 export declare const FALLBACK_ID_PREFIX = "__external_store_fallback_";
-//#endregion
 //# sourceMappingURL=external-store-message.d.ts.map

package/dist/runtime/utils/external-store-message.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"external-store-message.d.ts","names":[],"sources":["../../../src/runtime/utils/external-store-message.ts"],"mappings":";;qBAEa;;;;;;;;qBAiBA,2BAA4B,GACvC,gBACA,SAAS,IAAI;qBAMF,2BAA4B,GACvC;EACM,mBAAmB;IACrB,gBACA,qCAAgC;qBAczB"}
+{"version":3,"file":"external-store-message.d.ts","sourceRoot":"","sources":["../../../src/runtime/utils/external-store-message.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,+BAA4B;AAEzD,eAAO,MAAM,kBAAkB,eAAyB,CAAC;AAUzD;;;;;;GAMG;AACH,eAAO,MAAM,wBAAwB,GAAI,CAAC,EACxC,QAAQ,MAAM,EACd,SAAS,CAAC,GAAG,CAAC,EAAE,KACf,IAGF,CAAC;AAEF,eAAO,MAAM,wBAAwB,GAAI,CAAC,EACxC,OACI;IAAE,QAAQ,EAAE,SAAS,aAAa,EAAE,CAAA;CAAE,GACtC,aAAa,GACb,aAAa,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,QAYrC,CAAC;AAEF,eAAO,MAAM,kBAAkB,+BAA+B,CAAC"}

package/dist/runtime/utils/message-repository-session.d.ts

@@ -1,29 +1,27 @@
-import { ThreadMessage } from "../../types/message.js";
-import { ExportedMessageRepository, MessageRepository } from "./message-repository.js";
-//#region src/runtime/utils/message-repository-session.d.ts
+import type { ThreadMessage } from "../../types/message.js";
+import { MessageRepository, type ExportedMessageRepository } from "./message-repository.js";
 export type MessageRepositorySessionOptions = {
-  decorateExport?: (exported: ExportedMessageRepository, repository: MessageRepository) => ExportedMessageRepository;
+    decorateExport?: (exported: ExportedMessageRepository, repository: MessageRepository) => ExportedMessageRepository;
 };
 export declare const createMessageRepositorySession: (options?: MessageRepositorySessionOptions) => {
-  getMessages: () => readonly ThreadMessage[];
-  readonly headId: string | null;
-  export: () => ExportedMessageRepository;
-  tryGetMessage: (messageId: string) => {
-    parentId: string | null;
-    message: ThreadMessage;
-    index: number;
-  } | undefined;
-  tryGetMessages: (messageId: string) => readonly ThreadMessage[] | undefined;
-  hasMessage: (messageId: string) => boolean;
-  addOrUpdateMessage: (parentId: string | null, message: ThreadMessage) => void;
-  deleteMessage: (messageId: string, replacementId?: string | null) => void;
-  tryDeleteMessage: (messageId: string) => boolean;
-  switchToBranch: (messageId: string) => void;
-  resetHead: (messageId: string | null) => void;
-  clear: () => void;
-  updateMessage: (messageId: string, updater: (message: ThreadMessage) => ThreadMessage) => boolean;
-  applyExternalMessageRepository: (loaded: ExportedMessageRepository) => void;
+    getMessages: () => readonly ThreadMessage[];
+    readonly headId: string | null;
+    export: () => ExportedMessageRepository;
+    tryGetMessage: (messageId: string) => {
+        parentId: string | null;
+        message: ThreadMessage;
+        index: number;
+    } | undefined;
+    tryGetMessages: (messageId: string) => readonly ThreadMessage[] | undefined;
+    hasMessage: (messageId: string) => boolean;
+    addOrUpdateMessage: (parentId: string | null, message: ThreadMessage) => void;
+    deleteMessage: (messageId: string, replacementId?: string | null) => void;
+    tryDeleteMessage: (messageId: string) => boolean;
+    switchToBranch: (messageId: string) => void;
+    resetHead: (messageId: string | null) => void;
+    clear: () => void;
+    updateMessage: (messageId: string, updater: (message: ThreadMessage) => ThreadMessage) => boolean;
+    applyExternalMessageRepository: (loaded: ExportedMessageRepository) => void;
 };
 export type MessageRepositorySession = ReturnType<typeof createMessageRepositorySession>;
-//#endregion
 //# sourceMappingURL=message-repository-session.d.ts.map

package/dist/runtime/utils/message-repository-session.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"message-repository-session.d.ts","names":[],"sources":["../../../src/runtime/utils/message-repository-session.ts"],"mappings":";;;YAMY;EACV,kBACE,UAAU,2BACV,YAAY,sBACT;;qBAGM,iCACX,UAAS;EAKe,4BAAS;;EAsInB,cAAA;EAlIoB,gBAAA;;;;;EASrB,iBAAA,+BACD;EAQmB,aAAA;EAKnB,qBAAA,yBAAa,SACd;EAOE,gBAAA,mBAAM;EAOkB,mBAAA;EAMF,iBAAA;EAKL,YAAA;EAKZ;EAML,gBAAA,mBAAM,UACP,SAAS,kBAAkB;EAW7B,iCAAA,QAAA;;YA8EA,2BAA2B,kBAC9B"}
+{"version":3,"file":"message-repository-session.d.ts","sourceRoot":"","sources":["../../../src/runtime/utils/message-repository-session.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,+BAA4B;AACzD,OAAO,EACL,iBAAiB,EACjB,KAAK,yBAAyB,EAC/B,gCAA6B;AAE9B,MAAM,MAAM,+BAA+B,GAAG;IAC5C,cAAc,CAAC,EAAE,CACf,QAAQ,EAAE,yBAAyB,EACnC,UAAU,EAAE,iBAAiB,KAC1B,yBAAyB,CAAC;CAChC,CAAC;AAEF,eAAO,MAAM,8BAA8B,GACzC,UAAS,+BAAoC;uBAKrB,SAAS,aAAa,EAAE;;kBAsIlC,yBAAyB;+BAlIL,MAAM;;;;;gCAS3B,MAAM,KAChB,SAAS,aAAa,EAAE,GAAG,SAAS;4BAQR,MAAM,KAAG,OAAO;mCAKnC,MAAM,GAAG,IAAI,WACd,aAAa,KACrB,IAAI;+BAMM,MAAM,kBACD,MAAM,GAAG,IAAI,KAC5B,IAAI;kCAK8B,MAAM,KAAG,OAAO;gCAMlB,MAAM,KAAG,IAAI;2BAKlB,MAAM,GAAG,IAAI,KAAG,IAAI;iBAKhC,IAAI;+BAMT,MAAM,WACR,CAAC,OAAO,EAAE,aAAa,KAAK,aAAa,KACjD,OAAO;6CAUA,yBAAyB,KAChC,IAAI;CA2ER,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAC/C,OAAO,8BAA8B,CACtC,CAAC"}

package/dist/runtime/utils/message-repository.d.ts

@@ -1,61 +1,60 @@
-import { RunConfig, ThreadMessage } from "../../types/message.js";
-import { ThreadMessageLike } from "./thread-message-like.js";
-//#region src/runtime/utils/message-repository.d.ts
+import type { ThreadMessage } from "../../types/message.js";
+import type { RunConfig } from "../../types/message.js";
+import type { ThreadMessageLike } from "./thread-message-like.js";
 export type ExportedMessageRepositoryItem = {
-  message: ThreadMessage;
-  parentId: string | null;
-  runConfig?: RunConfig;
-};
-export type ExportedMessageRepository = {
-  headId?: string | null;
-  messages: Array<{
     message: ThreadMessage;
     parentId: string | null;
     runConfig?: RunConfig;
-  }>;
 };
-export declare const ExportedMessageRepository: {
-  fromArray: (messages: readonly ThreadMessageLike[]) => ExportedMessageRepository;
-  fromBranchableArray: (items: readonly {
-    message: ThreadMessageLike;
-    parentId: string | null;
-  }[], options?: {
+export type ExportedMessageRepository = {
     headId?: string | null;
-  }) => ExportedMessageRepository;
+    messages: Array<{
+        message: ThreadMessage;
+        parentId: string | null;
+        runConfig?: RunConfig;
+    }>;
+};
+export declare const ExportedMessageRepository: {
+    fromArray: (messages: readonly ThreadMessageLike[]) => ExportedMessageRepository;
+    fromBranchableArray: (items: readonly {
+        message: ThreadMessageLike;
+        parentId: string | null;
+    }[], options?: {
+        headId?: string | null;
+    }) => ExportedMessageRepository;
 };
 export declare class MessageRepository {
-  private messages;
-  private head;
-  private root;
-  private updateLevels;
-  private selectPathTo;
-  private performOp;
-  private _messages;
-  get headId(): string | null;
-  get canonicalHeadId(): string | null;
-  getMessages(headId?: string): readonly ThreadMessage[];
-  addOrUpdateMessage(parentId: string | null, message: ThreadMessage): void;
-  getMessage(messageId: string): {
-    parentId: string | null;
-    message: ThreadMessage;
-    index: number;
-  };
-  deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
-  getBranches(messageId: string): string[];
-  /**
-   * Evicts optimistic messages (`metadata.isOptimistic`) the head just moved
-   * away from. Since eviction runs on every head move, the only optimistic
-   * messages in the repository live on the branch the head previously pointed
-   * at — so we walk just that branch rather than the whole repository. Keeps a
-   * client→server id swap from leaving a phantom sibling, and drops off-branch
-   * placeholders.
-   */
-  private evictOffBranchOptimisticMessages;
-  switchToBranch(messageId: string): void;
-  resetHead(messageId: string | null): void;
-  clear(): void;
-  export(): ExportedMessageRepository;
-  import({ headId, messages }: ExportedMessageRepository): void;
+    private messages;
+    private head;
+    private root;
+    private updateLevels;
+    private selectPathTo;
+    private performOp;
+    private _messages;
+    get headId(): string | null;
+    get canonicalHeadId(): string | null;
+    getMessages(headId?: string): readonly ThreadMessage[];
+    addOrUpdateMessage(parentId: string | null, message: ThreadMessage): void;
+    getMessage(messageId: string): {
+        parentId: string | null;
+        message: ThreadMessage;
+        index: number;
+    };
+    deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
+    getBranches(messageId: string): string[];
+    /**
+     * Evicts optimistic messages (`metadata.isOptimistic`) on the branch the head
+     * just moved away from. Only that branch is walked, so an optimistic message
+     * added off the head branch (such as the server-id copy that replaces a
+     * client-id placeholder before the head moves to it) is kept until a
+     * `switchToBranch` or `resetHead` moves the head off the branch it is on.
+     * Keeps a client→server id swap from leaving a phantom sibling.
+     */
+    private evictOffBranchOptimisticMessages;
+    switchToBranch(messageId: string): void;
+    resetHead(messageId: string | null): void;
+    clear(): void;
+    export(): ExportedMessageRepository;
+    import({ headId, messages }: ExportedMessageRepository): void;
 }
-//#endregion
 //# sourceMappingURL=message-repository.d.ts.map

package/dist/runtime/utils/message-repository.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"message-repository.d.ts","names":[],"sources":["../../../src/runtime/utils/message-repository.ts"],"mappings":";;;YAOY;EACV,SAAS;EACT;EACA,YAAY;;YAGF;EACV;EACA,UAAU;IACR,SAAS;IACT;IACA,YAAY;;;qBAIH;EAEC,YAAA,mBAAS,wBAClB;EAkBM,sBAAA;IACL,SAAS;IACT;OACC;IACS;QACX;;qBAgEQ;UACH;UACA;UACA;UAKA;UAgBA;UAUA;UAgEA;MAQJ;MAIA;EAUJ,YAAY,2BAAe;EAuB3B,mBAAmB,yBAAyB,SAAS;EAiCrD,WAAW;;;;;EAcX,cAAc,mBAAmB;EAsCjC,YAAY;;;;;;;;;UAmBJ;EA+BR,eAAe;EAiBf,UAAU;EAuCV;EAUA,UAAU;EAgCV,SAAS,QAAQ,YAAY"}
+{"version":3,"file":"message-repository.d.ts","sourceRoot":"","sources":["../../../src/runtime/utils/message-repository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,+BAA4B;AACzD,OAAO,KAAK,EAAE,SAAS,EAAE,+BAA4B;AAErD,OAAO,KAAK,EAAE,iBAAiB,EAAE,iCAA8B;AAI/D,MAAM,MAAM,6BAA6B,GAAG;IAC1C,OAAO,EAAE,aAAa,CAAC;IACvB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,SAAS,CAAC,EAAE,SAAS,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,QAAQ,EAAE,KAAK,CAAC;QACd,OAAO,EAAE,aAAa,CAAC;QACvB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QACxB,SAAS,CAAC,EAAE,SAAS,CAAC;KACvB,CAAC,CAAC;CACJ,CAAC;AAEF,eAAO,MAAM,yBAAyB;0BAExB,SAAS,iBAAiB,EAAE,KACrC,yBAAyB;iCAkBnB,SAAS;QACd,OAAO,EAAE,iBAAiB,CAAC;QAC3B,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;KACzB,EAAE,YACO;QAAE,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,KACnC,yBAAyB;CAsB7B,CAAC;AA0CF,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAAwC;IACxD,OAAO,CAAC,IAAI,CAAkC;IAC9C,OAAO,CAAC,IAAI,CAGV;IAEF,OAAO,CAAC,YAAY;IAgBpB,OAAO,CAAC,YAAY;IAUpB,OAAO,CAAC,SAAS;IAgEjB,OAAO,CAAC,SAAS,CAMd;IAEH,IAAI,MAAM,kBAET;IAED,IAAI,eAAe,kBAQlB;IAED,WAAW,CAAC,MAAM,CAAC,EAAE,MAAM;IAuB3B,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,EAAE,OAAO,EAAE,aAAa;IAiClE,UAAU,CAAC,SAAS,EAAE,MAAM;;;;;IAc5B,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS;IA6C1E,WAAW,CAAC,SAAS,EAAE,MAAM;IAW7B;;;;;;;OAOG;IACH,OAAO,CAAC,gCAAgC;IA+BxC,cAAc,CAAC,SAAS,EAAE,MAAM;IAiBhC,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAuClC,KAAK,IAAI,IAAI;IAUb,MAAM,IAAI,yBAAyB;IAgCnC,MAAM,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,yBAAyB;CAOvD"}

package/dist/runtime/utils/message-repository.js

@@ -123,8 +123,8 @@ var MessageRepository = class {
 		const prev = parentId ? this.messages.get(parentId) : null;
 		if (prev === void 0) throw new Error("MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui.");
 		if (existingItem) {
-			existingItem.current = message;
 			this.performOp(prev, existingItem, "relink");
+			existingItem.current = message;
 			this._messages.dirty();
 			return;
 		}
@@ -154,6 +154,7 @@ var MessageRepository = class {
 		if (!message) throw new Error("MessageRepository(deleteMessage): Message not found. This is likely an internal bug in assistant-ui.");
 		const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
 		if (replacement === void 0) throw new Error("MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui.");
+		for (let current = replacement; current; current = current.prev) if (current === message) throw new Error("MessageRepository(deleteMessage): Replacement is the deleted message or one of its descendants. This is likely an internal bug in assistant-ui.");
 		for (const child of message.children) {
 			const childMessage = this.messages.get(child);
 			if (!childMessage) throw new Error("MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui.");
@@ -171,12 +172,12 @@ var MessageRepository = class {
 		return children;
 	}
 	/**
-	* Evicts optimistic messages (`metadata.isOptimistic`) the head just moved
-	* away from. Since eviction runs on every head move, the only optimistic
-	* messages in the repository live on the branch the head previously pointed
-	* at — so we walk just that branch rather than the whole repository. Keeps a
-	* client→server id swap from leaving a phantom sibling, and drops off-branch
-	* placeholders.
+	* Evicts optimistic messages (`metadata.isOptimistic`) on the branch the head
+	* just moved away from. Only that branch is walked, so an optimistic message
+	* added off the head branch (such as the server-id copy that replaces a
+	* client-id placeholder before the head moves to it) is kept until a
+	* `switchToBranch` or `resetHead` moves the head off the branch it is on.
+	* Keeps a client→server id swap from leaving a phantom sibling.
 	*/
 	evictOffBranchOptimisticMessages(previousHead, currentHead) {
 		if (!previousHead) return;

package/dist/runtime/utils/message-repository.js.map

@@ -1 +1 @@
-{"version":3,"file":"message-repository.js","names":["ThreadMessage","RunConfig","generateId","ThreadMessageLike","getRepositoryContentAutoStatus","fromThreadMessageLike","ExportedMessageRepositoryItem","message","parentId","runConfig","ExportedMessageRepository","headId","messages","Array","fromArray","conv","map","m","content","idx","id","fromBranchableArray","items","options","undefined","Error","RepositoryParent","children","next","RepositoryMessage","prev","current","level","findHead","CachedValue","_value","T","func","constructor","value","dirty","MessageRepository","Map","head","root","updateLevels","newLevel","pending","length","pop","childId","childMessage","get","push","selectPathTo","performOp","newParent","child","operation","parentOrRoot","newParentOrRoot","filter","fallbackId","at","fallback","_messages","canonicalHeadId","metadata","isOptimistic","getMessages","headMessage","addOrUpdateMessage","existingItem","newItem","set","getMessage","messageId","index","deleteMessage","replacementId","replacement","delete","getBranches","evictOffBranchOptimisticMessages","previousHead","currentHead","onHeadBranch","Set","add","stale","has","switchToBranch","resetHead","clear","descendantId","export","exportItems","reverse","i","import"],"sources":["../../../src/runtime/utils/message-repository.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\nimport type { RunConfig } from \"../../types/message\";\nimport { generateId } from \"../../utils/id\";\nimport type { ThreadMessageLike } from \"./thread-message-like\";\nimport { getRepositoryContentAutoStatus } from \"./auto-status\";\nimport { fromThreadMessageLike } from \"./thread-message-like\";\n\nexport type ExportedMessageRepositoryItem = {\n  message: ThreadMessage;\n  parentId: string | null;\n  runConfig?: RunConfig;\n};\n\nexport type ExportedMessageRepository = {\n  headId?: string | null;\n  messages: Array<{\n    message: ThreadMessage;\n    parentId: string | null;\n    runConfig?: RunConfig;\n  }>;\n};\n\nexport const ExportedMessageRepository = {\n  fromArray: (\n    messages: readonly ThreadMessageLike[],\n  ): ExportedMessageRepository => {\n    const conv = messages.map((m) =>\n      fromThreadMessageLike(\n        m,\n        generateId(),\n        getRepositoryContentAutoStatus(m.content),\n      ),\n    );\n\n    return {\n      messages: conv.map((m, idx) => ({\n        parentId: idx > 0 ? conv[idx - 1]!.id : null,\n        message: m,\n      })),\n    };\n  },\n\n  fromBranchableArray: (\n    items: readonly {\n      message: ThreadMessageLike;\n      parentId: string | null;\n    }[],\n    options?: { headId?: string | null },\n  ): ExportedMessageRepository => {\n    return {\n      ...(options?.headId !== undefined\n        ? { headId: options.headId }\n        : undefined),\n      messages: items.map(({ message, parentId }) => {\n        if (!message.id) {\n          throw new Error(\n            \"ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.\",\n          );\n        }\n        return {\n          parentId,\n          message: fromThreadMessageLike(\n            message,\n            message.id,\n            getRepositoryContentAutoStatus(message.content),\n          ),\n        };\n      }),\n    };\n  },\n};\n\ntype RepositoryParent = {\n  children: string[];\n  next: RepositoryMessage | null;\n};\n\ntype RepositoryMessage = RepositoryParent & {\n  prev: RepositoryMessage | null;\n  current: ThreadMessage;\n  level: number;\n};\n\nconst findHead = (\n  message: RepositoryMessage | RepositoryParent,\n): RepositoryMessage | null => {\n  let current = message;\n  while (current.next) current = current.next;\n  return \"current\" in current ? current : null;\n};\n\nclass CachedValue<T> {\n  private _value: T | null = null;\n\n  private func: () => T;\n\n  constructor(func: () => T) {\n    this.func = func;\n  }\n\n  get value() {\n    if (this._value === null) {\n      this._value = this.func();\n    }\n    return this._value;\n  }\n\n  dirty() {\n    this._value = null;\n  }\n}\n\nexport class MessageRepository {\n  private messages = new Map<string, RepositoryMessage>();\n  private head: RepositoryMessage | null = null;\n  private root: RepositoryParent = {\n    children: [],\n    next: null,\n  };\n\n  private updateLevels(message: RepositoryMessage, newLevel: number) {\n    const pending = [{ message, level: newLevel }];\n\n    while (pending.length > 0) {\n      const current = pending.pop()!;\n      current.message.level = current.level;\n\n      for (const childId of current.message.children) {\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          pending.push({ message: childMessage, level: current.level + 1 });\n        }\n      }\n    }\n  }\n\n  private selectPathTo(message: RepositoryMessage) {\n    for (\n      let current: RepositoryMessage | null = message;\n      current;\n      current = current.prev\n    ) {\n      (current.prev ?? this.root).next = current;\n    }\n  }\n\n  private performOp(\n    newParent: RepositoryMessage | null,\n    child: RepositoryMessage,\n    operation: \"cut\" | \"link\" | \"relink\",\n  ) {\n    const parentOrRoot = child.prev ?? this.root;\n    const newParentOrRoot = newParent ?? this.root;\n\n    if (operation === \"relink\" && parentOrRoot === newParentOrRoot) return;\n\n    // `link` receives a fresh ID from `addOrUpdateMessage`; only `relink` can introduce a cycle.\n    if (operation === \"relink\") {\n      for (\n        let current: RepositoryMessage | null = newParent;\n        current;\n        current = current.prev\n      ) {\n        if (current.current.id === child.current.id) {\n          throw new Error(\n            \"MessageRepository(performOp/relink): A message with the same id already exists in the parent tree. This error occurs if the same message id is found multiple times. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n      }\n    }\n\n    if (operation !== \"link\") {\n      parentOrRoot.children = parentOrRoot.children.filter(\n        (m) => m !== child.current.id,\n      );\n\n      if (parentOrRoot.next === child) {\n        const fallbackId = parentOrRoot.children.at(-1);\n        const fallback = fallbackId ? this.messages.get(fallbackId) : null;\n        if (fallback === undefined) {\n          throw new Error(\n            \"MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n        parentOrRoot.next = fallback;\n      }\n    }\n\n    if (operation !== \"cut\") {\n      newParentOrRoot.children = [\n        ...newParentOrRoot.children,\n        child.current.id,\n      ];\n\n      child.prev = newParent;\n\n      if (findHead(child) === this.head) {\n        this.selectPathTo(child);\n      } else if (newParentOrRoot.next === null) {\n        newParentOrRoot.next = child;\n        if (this.head === newParentOrRoot) {\n          this.head = findHead(child);\n        }\n      }\n\n      const newLevel = newParent ? newParent.level + 1 : 0;\n      this.updateLevels(child, newLevel);\n    }\n  }\n\n  private _messages = new CachedValue<readonly ThreadMessage[]>(() => {\n    const messages = new Array<ThreadMessage>((this.head?.level ?? -1) + 1);\n    for (let current = this.head; current; current = current.prev) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  });\n\n  get headId() {\n    return this.head?.current.id ?? null;\n  }\n\n  get canonicalHeadId() {\n    // Optimistic messages are ephemeral, so persisted callers need the nearest\n    // non-optimistic ancestor rather than the raw head.\n    let head = this.head;\n    while (head?.current.metadata?.isOptimistic) {\n      head = head.prev;\n    }\n    return head?.current.id ?? null;\n  }\n\n  getMessages(headId?: string) {\n    if (headId === undefined || headId === this.head?.current.id) {\n      return this._messages.value;\n    }\n\n    const headMessage = this.messages.get(headId);\n    if (!headMessage) {\n      throw new Error(\n        \"MessageRepository(getMessages): Head message not found. This is likely an internal bug in assistant-ui.\",\n      );\n    }\n\n    const messages = new Array<ThreadMessage>(headMessage.level + 1);\n    for (\n      let current: RepositoryMessage | null = headMessage;\n      current;\n      current = current.prev\n    ) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  }\n\n  addOrUpdateMessage(parentId: string | null, message: ThreadMessage) {\n    const existingItem = this.messages.get(message.id);\n    const prev = parentId ? this.messages.get(parentId) : null;\n    if (prev === undefined)\n      throw new Error(\n        \"MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    if (existingItem) {\n      existingItem.current = message;\n      this.performOp(prev, existingItem, \"relink\");\n      this._messages.dirty();\n      return;\n    }\n\n    const newItem: RepositoryMessage = {\n      prev,\n      current: message,\n      next: null,\n      children: [],\n      level: prev ? prev.level + 1 : 0,\n    };\n\n    this.messages.set(message.id, newItem);\n    this.performOp(prev, newItem, \"link\");\n\n    if (this.head === prev) {\n      this.head = newItem;\n    }\n\n    this._messages.dirty();\n  }\n\n  getMessage(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    return {\n      parentId: message.prev?.current.id ?? null,\n      message: message.current,\n      index: message.level,\n    };\n  }\n\n  deleteMessage(messageId: string, replacementId?: string | null | undefined) {\n    const message = this.messages.get(messageId);\n\n    if (!message)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const replacement =\n      replacementId === undefined\n        ? message.prev\n        : replacementId === null\n          ? null\n          : this.messages.get(replacementId);\n    if (replacement === undefined)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    for (const child of message.children) {\n      const childMessage = this.messages.get(child);\n      if (!childMessage)\n        throw new Error(\n          \"MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui.\",\n        );\n      this.performOp(replacement, childMessage, \"relink\");\n    }\n\n    this.performOp(null, message, \"cut\");\n    this.messages.delete(messageId);\n\n    if (this.head === message) {\n      this.head = findHead(replacement ?? this.root);\n    }\n\n    this._messages.dirty();\n  }\n\n  getBranches(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const { children } = message.prev ?? this.root;\n    return children;\n  }\n\n  /**\n   * Evicts optimistic messages (`metadata.isOptimistic`) the head just moved\n   * away from. Since eviction runs on every head move, the only optimistic\n   * messages in the repository live on the branch the head previously pointed\n   * at — so we walk just that branch rather than the whole repository. Keeps a\n   * client→server id swap from leaving a phantom sibling, and drops off-branch\n   * placeholders.\n   */\n  private evictOffBranchOptimisticMessages(\n    previousHead: RepositoryMessage | null,\n    currentHead: RepositoryMessage | null,\n  ) {\n    if (!previousHead) return;\n\n    const onHeadBranch = new Set<string>();\n    for (let current = currentHead; current; current = current.prev) {\n      onHeadBranch.add(current.current.id);\n    }\n\n    const stale: string[] = [];\n    for (\n      let current: RepositoryMessage | null = previousHead;\n      current;\n      current = current.prev\n    ) {\n      // Stop at the first node shared with the current head branch: every\n      // ancestor above it is shared too, so nothing further can be off-branch.\n      if (onHeadBranch.has(current.current.id)) break;\n      if (current.current.metadata?.isOptimistic) {\n        stale.push(current.current.id);\n      }\n    }\n\n    for (const id of stale) {\n      // A prior deletion may have already removed this node.\n      if (this.messages.has(id)) this.deleteMessage(id);\n    }\n  }\n\n  switchToBranch(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n    this.selectPathTo(message);\n\n    this.head = findHead(message);\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  resetHead(messageId: string | null) {\n    if (messageId === null) {\n      this.clear();\n      return;\n    }\n\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n\n    if (message.children.length > 0) {\n      const pending = [...message.children];\n      while (pending.length > 0) {\n        const childId = pending.pop()!;\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          for (const descendantId of childMessage.children) {\n            pending.push(descendantId);\n          }\n          this.messages.delete(childId);\n        }\n      }\n\n      message.children = [];\n      message.next = null;\n    }\n\n    this.head = message;\n    this.selectPathTo(message);\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  clear(): void {\n    this.messages.clear();\n    this.head = null;\n    this.root = {\n      children: [],\n      next: null,\n    };\n    this._messages.dirty();\n  }\n\n  export(): ExportedMessageRepository {\n    const exportItems: ExportedMessageRepository[\"messages\"] = [];\n\n    // Optimistic messages are ephemeral and never persisted. A persisted child\n    // of an optimistic node is re-parented onto its nearest persisted ancestor\n    // so the exported tree never references a skipped id.\n    // Import and external-state conversion require parents before children, so\n    // the tree is walked in pre-order rather than iterated in insertion order.\n    const pending = [...this.root.children].reverse();\n    while (pending.length > 0) {\n      const message = this.messages.get(pending.pop()!);\n      if (!message) continue;\n      for (let i = message.children.length - 1; i >= 0; i--) {\n        pending.push(message.children[i]!);\n      }\n      if (message.current.metadata?.isOptimistic) continue;\n      let prev = message.prev;\n      while (prev && prev.current.metadata?.isOptimistic) {\n        prev = prev.prev;\n      }\n      exportItems.push({\n        message: message.current,\n        parentId: prev?.current.id ?? null,\n      });\n    }\n\n    return {\n      headId: this.canonicalHeadId,\n      messages: exportItems,\n    };\n  }\n\n  import({ headId, messages }: ExportedMessageRepository) {\n    for (const { message, parentId } of messages) {\n      this.addOrUpdateMessage(parentId, message);\n    }\n\n    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);\n  }\n}\n"],"mappings":";;;;AAsBA,MAAaU,4BAA4B;CACvCI,YACEF,aAC8B;EAC9B,MAAMG,OAAOH,SAASI,KAAKC,MACzBZ,sBACEY,GACAf,WAAW,GACXE,+BAA+Ba,EAAEC,OAAO,CAC1C,CACF;EAEA,OAAO,EACLN,UAAUG,KAAKC,KAAKC,GAAGE,SAAS;GAC9BX,UAAUW,MAAM,IAAIJ,KAAKI,MAAM,EAAE,CAAEC,KAAK;GACxCb,SAASU;EACX,EAAE,EACJ;CACF;CAEAI,sBACEC,OAIAC,YAC8B;EAC9B,OAAO;GACL,GAAIA,SAASZ,WAAWa,KAAAA,IACpB,EAAEb,QAAQY,QAAQZ,OAAO,IACzBa,KAAAA;GACJZ,UAAUU,MAAMN,KAAK,EAAET,SAASC,eAAe;IAC7C,IAAI,CAACD,QAAQa,IACX,MAAM,IAAIK,MACR,0FACF;IAEF,OAAO;KACLjB;KACAD,SAASF,sBACPE,SACAA,QAAQa,IACRhB,+BAA+BG,QAAQW,OAAO,CAChD;IACF;GACF,CAAC;EACH;CACF;AACF;AAaA,MAAMe,YACJ1B,YAC6B;CAC7B,IAAIwB,UAAUxB;CACd,OAAOwB,QAAQH,MAAMG,UAAUA,QAAQH;CACvC,OAAO,aAAaG,UAAUA,UAAU;AAC1C;AAEA,IAAMG,cAAN,MAAqB;CACnB,SAA2B;CAE3B;CAEAI,YAAYD,MAAe;EACzB,KAAKA,OAAOA;CACd;CAEA,IAAIE,QAAQ;EACV,IAAI,KAAKJ,WAAW,MAClB,KAAKA,SAAS,KAAKE,KAAK;EAE1B,OAAO,KAAKF;CACd;CAEAK,QAAQ;EACN,KAAKL,SAAS;CAChB;AACF;AAEA,IAAaM,oBAAb,MAA+B;CAC7B,2BAAmB,IAAIC,IAA+B;CACtD,OAAyC;CACzC,OAAiC;EAC/Bf,UAAU,CAAA;EACVC,MAAM;CACR;CAEA,aAAqBrB,SAA4BuC,UAAkB;EACjE,MAAMC,UAAU,CAAC;GAAExC;GAASyB,OAAOc;EAAS,CAAC;EAE7C,OAAOC,QAAQC,SAAS,GAAG;GACzB,MAAMjB,UAAUgB,QAAQE,IAAI;GAC5BlB,QAAQxB,QAAQyB,QAAQD,QAAQC;GAEhC,KAAK,MAAMkB,WAAWnB,QAAQxB,QAAQoB,UAAU;IAC9C,MAAMwB,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cACFJ,QAAQM,KAAK;KAAE9C,SAAS4C;KAAcnB,OAAOD,QAAQC,QAAQ;IAAE,CAAC;GAEpE;EACF;CACF;CAEA,aAAqBzB,SAA4B;EAC/C,KACE,IAAIwB,UAAoCxB,SACxCwB,SACAA,UAAUA,QAAQD,MAElB,CAACC,QAAQD,QAAQ,KAAKc,KAAAA,CAAMhB,OAAOG;CAEvC;CAEA,UACEyB,WACAC,OACAC,WACA;EACA,MAAMC,eAAeF,MAAM3B,QAAQ,KAAKc;EACxC,MAAMgB,kBAAkBJ,aAAa,KAAKZ;EAE1C,IAAIc,cAAc,YAAYC,iBAAiBC,iBAAiB;EAGhE,IAAIF,cAAc,UAEd;QAAA,IAAI3B,UAAoCyB,WACxCzB,SACAA,UAAUA,QAAQD,MAElB,IAAIC,QAAQA,QAAQX,OAAOqC,MAAM1B,QAAQX,IACvC,MAAM,IAAIK,MACR,sNACF;EAAA;EAKN,IAAIiC,cAAc,QAAQ;GACxBC,aAAahC,WAAWgC,aAAahC,SAASkC,QAC3C5C,MAAMA,MAAMwC,MAAM1B,QAAQX,EAC7B;GAEA,IAAIuC,aAAa/B,SAAS6B,OAAO;IAC/B,MAAMK,aAAaH,aAAahC,SAASoC,GAAG,EAAE;IAC9C,MAAMC,WAAWF,aAAa,KAAKlD,SAASwC,IAAIU,UAAU,IAAI;IAC9D,IAAIE,aAAaxC,KAAAA,GACf,MAAM,IAAIC,MACR,uHACF;IAEFkC,aAAa/B,OAAOoC;GACtB;EACF;EAEA,IAAIN,cAAc,OAAO;GACvBE,gBAAgBjC,WAAW,CACzB,GAAGiC,gBAAgBjC,UACnB8B,MAAM1B,QAAQX,EAAE;GAGlBqC,MAAM3B,OAAO0B;GAEb,IAAIvB,SAASwB,KAAK,MAAM,KAAKd,MAC3B,KAAKW,aAAaG,KAAK;QAClB,IAAIG,gBAAgBhC,SAAS,MAAM;IACxCgC,gBAAgBhC,OAAO6B;IACvB,IAAI,KAAKd,SAASiB,iBAChB,KAAKjB,OAAOV,SAASwB,KAAK;GAE9B;GAEA,MAAMX,WAAWU,YAAYA,UAAUxB,QAAQ,IAAI;GACnD,KAAKa,aAAaY,OAAOX,QAAQ;EACnC;CACF;CAEA,YAAoB,IAAIZ,kBAA4C;EAClE,MAAMtB,WAAW,IAAIC,OAAsB,KAAK8B,MAAMX,SAAS,MAAM,CAAC;EACtE,KAAK,IAAID,UAAU,KAAKY,MAAMZ,SAASA,UAAUA,QAAQD,MACvDlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT,CAAC;CAED,IAAID,SAAS;EACX,OAAO,KAAKgC,MAAMZ,QAAQX,MAAM;CAClC;CAEA,IAAI8C,kBAAkB;EAGpB,IAAIvB,OAAO,KAAKA;EAChB,OAAOA,MAAMZ,QAAQoC,UAAUC,cAC7BzB,OAAOA,KAAKb;EAEd,OAAOa,MAAMZ,QAAQX,MAAM;CAC7B;CAEAiD,YAAY1D,QAAiB;EAC3B,IAAIA,WAAWa,KAAAA,KAAab,WAAW,KAAKgC,MAAMZ,QAAQX,IACxD,OAAO,KAAK6C,UAAU1B;EAGxB,MAAM+B,cAAc,KAAK1D,SAASwC,IAAIzC,MAAM;EAC5C,IAAI,CAAC2D,aACH,MAAM,IAAI7C,MACR,yGACF;EAGF,MAAMb,WAAW,IAAIC,MAAqByD,YAAYtC,QAAQ,CAAC;EAC/D,KACE,IAAID,UAAoCuC,aACxCvC,SACAA,UAAUA,QAAQD,MAElBlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT;CAEA2D,mBAAmB/D,UAAyBD,SAAwB;EAClE,MAAMiE,eAAe,KAAK5D,SAASwC,IAAI7C,QAAQa,EAAE;EACjD,MAAMU,OAAOtB,WAAW,KAAKI,SAASwC,IAAI5C,QAAQ,IAAI;EACtD,IAAIsB,SAASN,KAAAA,GACX,MAAM,IAAIC,MACR,kHACF;EAEF,IAAI+C,cAAc;GAChBA,aAAazC,UAAUxB;GACvB,KAAKgD,UAAUzB,MAAM0C,cAAc,QAAQ;GAC3C,KAAKP,UAAUzB,MAAM;GACrB;EACF;EAEA,MAAMiC,UAA6B;GACjC3C;GACAC,SAASxB;GACTqB,MAAM;GACND,UAAU,CAAA;GACVK,OAAOF,OAAOA,KAAKE,QAAQ,IAAI;EACjC;EAEA,KAAKpB,SAAS8D,IAAInE,QAAQa,IAAIqD,OAAO;EACrC,KAAKlB,UAAUzB,MAAM2C,SAAS,MAAM;EAEpC,IAAI,KAAK9B,SAASb,MAChB,KAAKa,OAAO8B;EAGd,KAAKR,UAAUzB,MAAM;CACvB;CAEAmC,WAAWC,WAAmB;EAC5B,MAAMrE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,OAAO;GACLjB,UAAUD,QAAQuB,MAAMC,QAAQX,MAAM;GACtCb,SAASA,QAAQwB;GACjB8C,OAAOtE,QAAQyB;EACjB;CACF;CAEA8C,cAAcF,WAAmBG,eAA2C;EAC1E,MAAMxE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAE3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAMuD,cACJD,kBAAkBvD,KAAAA,IACdjB,QAAQuB,OACRiD,kBAAkB,OAChB,OACA,KAAKnE,SAASwC,IAAI2B,aAAa;EACvC,IAAIC,gBAAgBxD,KAAAA,GAClB,MAAM,IAAIC,MACR,0GACF;EAEF,KAAK,MAAMgC,SAASlD,QAAQoB,UAAU;GACpC,MAAMwB,eAAe,KAAKvC,SAASwC,IAAIK,KAAK;GAC5C,IAAI,CAACN,cACH,MAAM,IAAI1B,MACR,4GACF;GACF,KAAK8B,UAAUyB,aAAa7B,cAAc,QAAQ;EACpD;EAEA,KAAKI,UAAU,MAAMhD,SAAS,KAAK;EACnC,KAAKK,SAASqE,OAAOL,SAAS;EAE9B,IAAI,KAAKjC,SAASpC,SAChB,KAAKoC,OAAOV,SAAS+C,eAAe,KAAKpC,IAAI;EAG/C,KAAKqB,UAAUzB,MAAM;CACvB;CAEA0C,YAAYN,WAAmB;EAC7B,MAAMrE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,oGACF;EAEF,MAAM,EAAEE,aAAapB,QAAQuB,QAAQ,KAAKc;EAC1C,OAAOjB;CACT;;;;;;;;;CAUA,iCACEyD,cACAC,aACA;EACA,IAAI,CAACD,cAAc;EAEnB,MAAME,+BAAe,IAAIC,IAAY;EACrC,KAAK,IAAIxD,UAAUsD,aAAatD,SAASA,UAAUA,QAAQD,MACzDwD,aAAaE,IAAIzD,QAAQA,QAAQX,EAAE;EAGrC,MAAMqE,QAAkB,CAAA;EACxB,KACE,IAAI1D,UAAoCqD,cACxCrD,SACAA,UAAUA,QAAQD,MAClB;GAGA,IAAIwD,aAAaI,IAAI3D,QAAQA,QAAQX,EAAE,GAAG;GAC1C,IAAIW,QAAQA,QAAQoC,UAAUC,cAC5BqB,MAAMpC,KAAKtB,QAAQA,QAAQX,EAAE;EAEjC;EAEA,KAAK,MAAMA,MAAMqE,OAEf,IAAI,KAAK7E,SAAS8E,IAAItE,EAAE,GAAG,KAAK0D,cAAc1D,EAAE;CAEpD;CAEAuE,eAAef,WAAmB;EAChC,MAAMrE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAM2D,eAAe,KAAKzC;EAC1B,KAAKW,aAAa/C,OAAO;EAEzB,KAAKoC,OAAOV,SAAS1B,OAAO;EAE5B,KAAK4E,iCAAiCC,cAAc,KAAKzC,IAAI;EAE7D,KAAKsB,UAAUzB,MAAM;CACvB;CAEAoD,UAAUhB,WAA0B;EAClC,IAAIA,cAAc,MAAM;GACtB,KAAKiB,MAAM;GACX;EACF;EAEA,MAAMtF,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,iGACF;EAEF,MAAM2D,eAAe,KAAKzC;EAE1B,IAAIpC,QAAQoB,SAASqB,SAAS,GAAG;GAC/B,MAAMD,UAAU,CAAC,GAAGxC,QAAQoB,QAAQ;GACpC,OAAOoB,QAAQC,SAAS,GAAG;IACzB,MAAME,UAAUH,QAAQE,IAAI;IAC5B,MAAME,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cAAc;KAChB,KAAK,MAAM2C,gBAAgB3C,aAAaxB,UACtCoB,QAAQM,KAAKyC,YAAY;KAE3B,KAAKlF,SAASqE,OAAO/B,OAAO;IAC9B;GACF;GAEA3C,QAAQoB,WAAW,CAAA;GACnBpB,QAAQqB,OAAO;EACjB;EAEA,KAAKe,OAAOpC;EACZ,KAAK+C,aAAa/C,OAAO;EAEzB,KAAK4E,iCAAiCC,cAAc,KAAKzC,IAAI;EAE7D,KAAKsB,UAAUzB,MAAM;CACvB;CAEAqD,QAAc;EACZ,KAAKjF,SAASiF,MAAM;EACpB,KAAKlD,OAAO;EACZ,KAAKC,OAAO;GACVjB,UAAU,CAAA;GACVC,MAAM;EACR;EACA,KAAKqC,UAAUzB,MAAM;CACvB;CAEAuD,SAAoC;EAClC,MAAMC,cAAqD,CAAA;EAO3D,MAAMjD,UAAU,CAAC,GAAG,KAAKH,KAAKjB,QAAQ,CAAC,CAACsE,QAAQ;EAChD,OAAOlD,QAAQC,SAAS,GAAG;GACzB,MAAMzC,UAAU,KAAKK,SAASwC,IAAIL,QAAQE,IAAI,CAAE;GAChD,IAAI,CAAC1C,SAAS;GACd,KAAK,IAAI2F,IAAI3F,QAAQoB,SAASqB,SAAS,GAAGkD,KAAK,GAAGA,KAChDnD,QAAQM,KAAK9C,QAAQoB,SAASuE,EAAG;GAEnC,IAAI3F,QAAQwB,QAAQoC,UAAUC,cAAc;GAC5C,IAAItC,OAAOvB,QAAQuB;GACnB,OAAOA,QAAQA,KAAKC,QAAQoC,UAAUC,cACpCtC,OAAOA,KAAKA;GAEdkE,YAAY3C,KAAK;IACf9C,SAASA,QAAQwB;IACjBvB,UAAUsB,MAAMC,QAAQX,MAAM;GAChC,CAAC;EACH;EAEA,OAAO;GACLT,QAAQ,KAAKuD;GACbtD,UAAUoF;EACZ;CACF;CAEAG,OAAO,EAAExF,QAAQC,YAAuC;EACtD,KAAK,MAAM,EAAEL,SAASC,cAAcI,UAClC,KAAK2D,mBAAmB/D,UAAUD,OAAO;EAG3C,KAAKqF,UAAUjF,UAAUC,SAASmD,GAAG,EAAE,CAAC,EAAExD,QAAQa,MAAM,IAAI;CAC9D;AACF"}
+{"version":3,"file":"message-repository.js","names":["ThreadMessage","RunConfig","generateId","ThreadMessageLike","getRepositoryContentAutoStatus","fromThreadMessageLike","ExportedMessageRepositoryItem","message","parentId","runConfig","ExportedMessageRepository","headId","messages","Array","fromArray","conv","map","m","content","idx","id","fromBranchableArray","items","options","undefined","Error","RepositoryParent","children","next","RepositoryMessage","prev","current","level","findHead","CachedValue","_value","T","func","constructor","value","dirty","MessageRepository","Map","head","root","updateLevels","newLevel","pending","length","pop","childId","childMessage","get","push","selectPathTo","performOp","newParent","child","operation","parentOrRoot","newParentOrRoot","filter","fallbackId","at","fallback","_messages","canonicalHeadId","metadata","isOptimistic","getMessages","headMessage","addOrUpdateMessage","existingItem","newItem","set","getMessage","messageId","index","deleteMessage","replacementId","replacement","delete","getBranches","evictOffBranchOptimisticMessages","previousHead","currentHead","onHeadBranch","Set","add","stale","has","switchToBranch","resetHead","clear","descendantId","export","exportItems","reverse","i","import"],"sources":["../../../src/runtime/utils/message-repository.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\nimport type { RunConfig } from \"../../types/message\";\nimport { generateId } from \"../../utils/id\";\nimport type { ThreadMessageLike } from \"./thread-message-like\";\nimport { getRepositoryContentAutoStatus } from \"./auto-status\";\nimport { fromThreadMessageLike } from \"./thread-message-like\";\n\nexport type ExportedMessageRepositoryItem = {\n  message: ThreadMessage;\n  parentId: string | null;\n  runConfig?: RunConfig;\n};\n\nexport type ExportedMessageRepository = {\n  headId?: string | null;\n  messages: Array<{\n    message: ThreadMessage;\n    parentId: string | null;\n    runConfig?: RunConfig;\n  }>;\n};\n\nexport const ExportedMessageRepository = {\n  fromArray: (\n    messages: readonly ThreadMessageLike[],\n  ): ExportedMessageRepository => {\n    const conv = messages.map((m) =>\n      fromThreadMessageLike(\n        m,\n        generateId(),\n        getRepositoryContentAutoStatus(m.content),\n      ),\n    );\n\n    return {\n      messages: conv.map((m, idx) => ({\n        parentId: idx > 0 ? conv[idx - 1]!.id : null,\n        message: m,\n      })),\n    };\n  },\n\n  fromBranchableArray: (\n    items: readonly {\n      message: ThreadMessageLike;\n      parentId: string | null;\n    }[],\n    options?: { headId?: string | null },\n  ): ExportedMessageRepository => {\n    return {\n      ...(options?.headId !== undefined\n        ? { headId: options.headId }\n        : undefined),\n      messages: items.map(({ message, parentId }) => {\n        if (!message.id) {\n          throw new Error(\n            \"ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.\",\n          );\n        }\n        return {\n          parentId,\n          message: fromThreadMessageLike(\n            message,\n            message.id,\n            getRepositoryContentAutoStatus(message.content),\n          ),\n        };\n      }),\n    };\n  },\n};\n\ntype RepositoryParent = {\n  children: string[];\n  next: RepositoryMessage | null;\n};\n\ntype RepositoryMessage = RepositoryParent & {\n  prev: RepositoryMessage | null;\n  current: ThreadMessage;\n  level: number;\n};\n\nconst findHead = (\n  message: RepositoryMessage | RepositoryParent,\n): RepositoryMessage | null => {\n  let current = message;\n  while (current.next) current = current.next;\n  return \"current\" in current ? current : null;\n};\n\nclass CachedValue<T> {\n  private _value: T | null = null;\n\n  private func: () => T;\n\n  constructor(func: () => T) {\n    this.func = func;\n  }\n\n  get value() {\n    if (this._value === null) {\n      this._value = this.func();\n    }\n    return this._value;\n  }\n\n  dirty() {\n    this._value = null;\n  }\n}\n\nexport class MessageRepository {\n  private messages = new Map<string, RepositoryMessage>();\n  private head: RepositoryMessage | null = null;\n  private root: RepositoryParent = {\n    children: [],\n    next: null,\n  };\n\n  private updateLevels(message: RepositoryMessage, newLevel: number) {\n    const pending = [{ message, level: newLevel }];\n\n    while (pending.length > 0) {\n      const current = pending.pop()!;\n      current.message.level = current.level;\n\n      for (const childId of current.message.children) {\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          pending.push({ message: childMessage, level: current.level + 1 });\n        }\n      }\n    }\n  }\n\n  private selectPathTo(message: RepositoryMessage) {\n    for (\n      let current: RepositoryMessage | null = message;\n      current;\n      current = current.prev\n    ) {\n      (current.prev ?? this.root).next = current;\n    }\n  }\n\n  private performOp(\n    newParent: RepositoryMessage | null,\n    child: RepositoryMessage,\n    operation: \"cut\" | \"link\" | \"relink\",\n  ) {\n    const parentOrRoot = child.prev ?? this.root;\n    const newParentOrRoot = newParent ?? this.root;\n\n    if (operation === \"relink\" && parentOrRoot === newParentOrRoot) return;\n\n    // `link` receives a fresh ID from `addOrUpdateMessage`; only `relink` can introduce a cycle.\n    if (operation === \"relink\") {\n      for (\n        let current: RepositoryMessage | null = newParent;\n        current;\n        current = current.prev\n      ) {\n        if (current.current.id === child.current.id) {\n          throw new Error(\n            \"MessageRepository(performOp/relink): A message with the same id already exists in the parent tree. This error occurs if the same message id is found multiple times. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n      }\n    }\n\n    if (operation !== \"link\") {\n      parentOrRoot.children = parentOrRoot.children.filter(\n        (m) => m !== child.current.id,\n      );\n\n      if (parentOrRoot.next === child) {\n        const fallbackId = parentOrRoot.children.at(-1);\n        const fallback = fallbackId ? this.messages.get(fallbackId) : null;\n        if (fallback === undefined) {\n          throw new Error(\n            \"MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n        parentOrRoot.next = fallback;\n      }\n    }\n\n    if (operation !== \"cut\") {\n      newParentOrRoot.children = [\n        ...newParentOrRoot.children,\n        child.current.id,\n      ];\n\n      child.prev = newParent;\n\n      if (findHead(child) === this.head) {\n        this.selectPathTo(child);\n      } else if (newParentOrRoot.next === null) {\n        newParentOrRoot.next = child;\n        if (this.head === newParentOrRoot) {\n          this.head = findHead(child);\n        }\n      }\n\n      const newLevel = newParent ? newParent.level + 1 : 0;\n      this.updateLevels(child, newLevel);\n    }\n  }\n\n  private _messages = new CachedValue<readonly ThreadMessage[]>(() => {\n    const messages = new Array<ThreadMessage>((this.head?.level ?? -1) + 1);\n    for (let current = this.head; current; current = current.prev) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  });\n\n  get headId() {\n    return this.head?.current.id ?? null;\n  }\n\n  get canonicalHeadId() {\n    // Optimistic messages are ephemeral, so persisted callers need the nearest\n    // non-optimistic ancestor rather than the raw head.\n    let head = this.head;\n    while (head?.current.metadata?.isOptimistic) {\n      head = head.prev;\n    }\n    return head?.current.id ?? null;\n  }\n\n  getMessages(headId?: string) {\n    if (headId === undefined || headId === this.head?.current.id) {\n      return this._messages.value;\n    }\n\n    const headMessage = this.messages.get(headId);\n    if (!headMessage) {\n      throw new Error(\n        \"MessageRepository(getMessages): Head message not found. This is likely an internal bug in assistant-ui.\",\n      );\n    }\n\n    const messages = new Array<ThreadMessage>(headMessage.level + 1);\n    for (\n      let current: RepositoryMessage | null = headMessage;\n      current;\n      current = current.prev\n    ) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  }\n\n  addOrUpdateMessage(parentId: string | null, message: ThreadMessage) {\n    const existingItem = this.messages.get(message.id);\n    const prev = parentId ? this.messages.get(parentId) : null;\n    if (prev === undefined)\n      throw new Error(\n        \"MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    if (existingItem) {\n      this.performOp(prev, existingItem, \"relink\");\n      existingItem.current = message;\n      this._messages.dirty();\n      return;\n    }\n\n    const newItem: RepositoryMessage = {\n      prev,\n      current: message,\n      next: null,\n      children: [],\n      level: prev ? prev.level + 1 : 0,\n    };\n\n    this.messages.set(message.id, newItem);\n    this.performOp(prev, newItem, \"link\");\n\n    if (this.head === prev) {\n      this.head = newItem;\n    }\n\n    this._messages.dirty();\n  }\n\n  getMessage(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    return {\n      parentId: message.prev?.current.id ?? null,\n      message: message.current,\n      index: message.level,\n    };\n  }\n\n  deleteMessage(messageId: string, replacementId?: string | null | undefined) {\n    const message = this.messages.get(messageId);\n\n    if (!message)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const replacement =\n      replacementId === undefined\n        ? message.prev\n        : replacementId === null\n          ? null\n          : this.messages.get(replacementId);\n    if (replacement === undefined)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    for (let current = replacement; current; current = current.prev) {\n      if (current === message)\n        throw new Error(\n          \"MessageRepository(deleteMessage): Replacement is the deleted message or one of its descendants. This is likely an internal bug in assistant-ui.\",\n        );\n    }\n\n    for (const child of message.children) {\n      const childMessage = this.messages.get(child);\n      if (!childMessage)\n        throw new Error(\n          \"MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui.\",\n        );\n      this.performOp(replacement, childMessage, \"relink\");\n    }\n\n    this.performOp(null, message, \"cut\");\n    this.messages.delete(messageId);\n\n    if (this.head === message) {\n      this.head = findHead(replacement ?? this.root);\n    }\n\n    this._messages.dirty();\n  }\n\n  getBranches(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const { children } = message.prev ?? this.root;\n    return children;\n  }\n\n  /**\n   * Evicts optimistic messages (`metadata.isOptimistic`) on the branch the head\n   * just moved away from. Only that branch is walked, so an optimistic message\n   * added off the head branch (such as the server-id copy that replaces a\n   * client-id placeholder before the head moves to it) is kept until a\n   * `switchToBranch` or `resetHead` moves the head off the branch it is on.\n   * Keeps a client→server id swap from leaving a phantom sibling.\n   */\n  private evictOffBranchOptimisticMessages(\n    previousHead: RepositoryMessage | null,\n    currentHead: RepositoryMessage | null,\n  ) {\n    if (!previousHead) return;\n\n    const onHeadBranch = new Set<string>();\n    for (let current = currentHead; current; current = current.prev) {\n      onHeadBranch.add(current.current.id);\n    }\n\n    const stale: string[] = [];\n    for (\n      let current: RepositoryMessage | null = previousHead;\n      current;\n      current = current.prev\n    ) {\n      // Stop at the first node shared with the current head branch: every\n      // ancestor above it is shared too, so nothing further can be off-branch.\n      if (onHeadBranch.has(current.current.id)) break;\n      if (current.current.metadata?.isOptimistic) {\n        stale.push(current.current.id);\n      }\n    }\n\n    for (const id of stale) {\n      // A prior deletion may have already removed this node.\n      if (this.messages.has(id)) this.deleteMessage(id);\n    }\n  }\n\n  switchToBranch(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n    this.selectPathTo(message);\n\n    this.head = findHead(message);\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  resetHead(messageId: string | null) {\n    if (messageId === null) {\n      this.clear();\n      return;\n    }\n\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n\n    if (message.children.length > 0) {\n      const pending = [...message.children];\n      while (pending.length > 0) {\n        const childId = pending.pop()!;\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          for (const descendantId of childMessage.children) {\n            pending.push(descendantId);\n          }\n          this.messages.delete(childId);\n        }\n      }\n\n      message.children = [];\n      message.next = null;\n    }\n\n    this.head = message;\n    this.selectPathTo(message);\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  clear(): void {\n    this.messages.clear();\n    this.head = null;\n    this.root = {\n      children: [],\n      next: null,\n    };\n    this._messages.dirty();\n  }\n\n  export(): ExportedMessageRepository {\n    const exportItems: ExportedMessageRepository[\"messages\"] = [];\n\n    // Optimistic messages are ephemeral and never persisted. A persisted child\n    // of an optimistic node is re-parented onto its nearest persisted ancestor\n    // so the exported tree never references a skipped id.\n    // Import and external-state conversion require parents before children, so\n    // the tree is walked in pre-order rather than iterated in insertion order.\n    const pending = [...this.root.children].reverse();\n    while (pending.length > 0) {\n      const message = this.messages.get(pending.pop()!);\n      if (!message) continue;\n      for (let i = message.children.length - 1; i >= 0; i--) {\n        pending.push(message.children[i]!);\n      }\n      if (message.current.metadata?.isOptimistic) continue;\n      let prev = message.prev;\n      while (prev && prev.current.metadata?.isOptimistic) {\n        prev = prev.prev;\n      }\n      exportItems.push({\n        message: message.current,\n        parentId: prev?.current.id ?? null,\n      });\n    }\n\n    return {\n      headId: this.canonicalHeadId,\n      messages: exportItems,\n    };\n  }\n\n  import({ headId, messages }: ExportedMessageRepository) {\n    for (const { message, parentId } of messages) {\n      this.addOrUpdateMessage(parentId, message);\n    }\n\n    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);\n  }\n}\n"],"mappings":";;;;AAsBA,MAAaU,4BAA4B;CACvCI,YACEF,aAC8B;EAC9B,MAAMG,OAAOH,SAASI,KAAKC,MACzBZ,sBACEY,GACAf,WAAW,GACXE,+BAA+Ba,EAAEC,OAAO,CAC1C,CACF;EAEA,OAAO,EACLN,UAAUG,KAAKC,KAAKC,GAAGE,SAAS;GAC9BX,UAAUW,MAAM,IAAIJ,KAAKI,MAAM,EAAE,CAAEC,KAAK;GACxCb,SAASU;EACX,EAAE,EACJ;CACF;CAEAI,sBACEC,OAIAC,YAC8B;EAC9B,OAAO;GACL,GAAIA,SAASZ,WAAWa,KAAAA,IACpB,EAAEb,QAAQY,QAAQZ,OAAO,IACzBa,KAAAA;GACJZ,UAAUU,MAAMN,KAAK,EAAET,SAASC,eAAe;IAC7C,IAAI,CAACD,QAAQa,IACX,MAAM,IAAIK,MACR,0FACF;IAEF,OAAO;KACLjB;KACAD,SAASF,sBACPE,SACAA,QAAQa,IACRhB,+BAA+BG,QAAQW,OAAO,CAChD;IACF;GACF,CAAC;EACH;CACF;AACF;AAaA,MAAMe,YACJ1B,YAC6B;CAC7B,IAAIwB,UAAUxB;CACd,OAAOwB,QAAQH,MAAMG,UAAUA,QAAQH;CACvC,OAAO,aAAaG,UAAUA,UAAU;AAC1C;AAEA,IAAMG,cAAN,MAAqB;CACnB,SAA2B;CAE3B;CAEAI,YAAYD,MAAe;EACzB,KAAKA,OAAOA;CACd;CAEA,IAAIE,QAAQ;EACV,IAAI,KAAKJ,WAAW,MAClB,KAAKA,SAAS,KAAKE,KAAK;EAE1B,OAAO,KAAKF;CACd;CAEAK,QAAQ;EACN,KAAKL,SAAS;CAChB;AACF;AAEA,IAAaM,oBAAb,MAA+B;CAC7B,2BAAmB,IAAIC,IAA+B;CACtD,OAAyC;CACzC,OAAiC;EAC/Bf,UAAU,CAAA;EACVC,MAAM;CACR;CAEA,aAAqBrB,SAA4BuC,UAAkB;EACjE,MAAMC,UAAU,CAAC;GAAExC;GAASyB,OAAOc;EAAS,CAAC;EAE7C,OAAOC,QAAQC,SAAS,GAAG;GACzB,MAAMjB,UAAUgB,QAAQE,IAAI;GAC5BlB,QAAQxB,QAAQyB,QAAQD,QAAQC;GAEhC,KAAK,MAAMkB,WAAWnB,QAAQxB,QAAQoB,UAAU;IAC9C,MAAMwB,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cACFJ,QAAQM,KAAK;KAAE9C,SAAS4C;KAAcnB,OAAOD,QAAQC,QAAQ;IAAE,CAAC;GAEpE;EACF;CACF;CAEA,aAAqBzB,SAA4B;EAC/C,KACE,IAAIwB,UAAoCxB,SACxCwB,SACAA,UAAUA,QAAQD,MAElB,CAACC,QAAQD,QAAQ,KAAKc,KAAAA,CAAMhB,OAAOG;CAEvC;CAEA,UACEyB,WACAC,OACAC,WACA;EACA,MAAMC,eAAeF,MAAM3B,QAAQ,KAAKc;EACxC,MAAMgB,kBAAkBJ,aAAa,KAAKZ;EAE1C,IAAIc,cAAc,YAAYC,iBAAiBC,iBAAiB;EAGhE,IAAIF,cAAc,UAEd;QAAA,IAAI3B,UAAoCyB,WACxCzB,SACAA,UAAUA,QAAQD,MAElB,IAAIC,QAAQA,QAAQX,OAAOqC,MAAM1B,QAAQX,IACvC,MAAM,IAAIK,MACR,sNACF;EAAA;EAKN,IAAIiC,cAAc,QAAQ;GACxBC,aAAahC,WAAWgC,aAAahC,SAASkC,QAC3C5C,MAAMA,MAAMwC,MAAM1B,QAAQX,EAC7B;GAEA,IAAIuC,aAAa/B,SAAS6B,OAAO;IAC/B,MAAMK,aAAaH,aAAahC,SAASoC,GAAG,EAAE;IAC9C,MAAMC,WAAWF,aAAa,KAAKlD,SAASwC,IAAIU,UAAU,IAAI;IAC9D,IAAIE,aAAaxC,KAAAA,GACf,MAAM,IAAIC,MACR,uHACF;IAEFkC,aAAa/B,OAAOoC;GACtB;EACF;EAEA,IAAIN,cAAc,OAAO;GACvBE,gBAAgBjC,WAAW,CACzB,GAAGiC,gBAAgBjC,UACnB8B,MAAM1B,QAAQX,EAAE;GAGlBqC,MAAM3B,OAAO0B;GAEb,IAAIvB,SAASwB,KAAK,MAAM,KAAKd,MAC3B,KAAKW,aAAaG,KAAK;QAClB,IAAIG,gBAAgBhC,SAAS,MAAM;IACxCgC,gBAAgBhC,OAAO6B;IACvB,IAAI,KAAKd,SAASiB,iBAChB,KAAKjB,OAAOV,SAASwB,KAAK;GAE9B;GAEA,MAAMX,WAAWU,YAAYA,UAAUxB,QAAQ,IAAI;GACnD,KAAKa,aAAaY,OAAOX,QAAQ;EACnC;CACF;CAEA,YAAoB,IAAIZ,kBAA4C;EAClE,MAAMtB,WAAW,IAAIC,OAAsB,KAAK8B,MAAMX,SAAS,MAAM,CAAC;EACtE,KAAK,IAAID,UAAU,KAAKY,MAAMZ,SAASA,UAAUA,QAAQD,MACvDlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT,CAAC;CAED,IAAID,SAAS;EACX,OAAO,KAAKgC,MAAMZ,QAAQX,MAAM;CAClC;CAEA,IAAI8C,kBAAkB;EAGpB,IAAIvB,OAAO,KAAKA;EAChB,OAAOA,MAAMZ,QAAQoC,UAAUC,cAC7BzB,OAAOA,KAAKb;EAEd,OAAOa,MAAMZ,QAAQX,MAAM;CAC7B;CAEAiD,YAAY1D,QAAiB;EAC3B,IAAIA,WAAWa,KAAAA,KAAab,WAAW,KAAKgC,MAAMZ,QAAQX,IACxD,OAAO,KAAK6C,UAAU1B;EAGxB,MAAM+B,cAAc,KAAK1D,SAASwC,IAAIzC,MAAM;EAC5C,IAAI,CAAC2D,aACH,MAAM,IAAI7C,MACR,yGACF;EAGF,MAAMb,WAAW,IAAIC,MAAqByD,YAAYtC,QAAQ,CAAC;EAC/D,KACE,IAAID,UAAoCuC,aACxCvC,SACAA,UAAUA,QAAQD,MAElBlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT;CAEA2D,mBAAmB/D,UAAyBD,SAAwB;EAClE,MAAMiE,eAAe,KAAK5D,SAASwC,IAAI7C,QAAQa,EAAE;EACjD,MAAMU,OAAOtB,WAAW,KAAKI,SAASwC,IAAI5C,QAAQ,IAAI;EACtD,IAAIsB,SAASN,KAAAA,GACX,MAAM,IAAIC,MACR,kHACF;EAEF,IAAI+C,cAAc;GAChB,KAAKjB,UAAUzB,MAAM0C,cAAc,QAAQ;GAC3CA,aAAazC,UAAUxB;GACvB,KAAK0D,UAAUzB,MAAM;GACrB;EACF;EAEA,MAAMiC,UAA6B;GACjC3C;GACAC,SAASxB;GACTqB,MAAM;GACND,UAAU,CAAA;GACVK,OAAOF,OAAOA,KAAKE,QAAQ,IAAI;EACjC;EAEA,KAAKpB,SAAS8D,IAAInE,QAAQa,IAAIqD,OAAO;EACrC,KAAKlB,UAAUzB,MAAM2C,SAAS,MAAM;EAEpC,IAAI,KAAK9B,SAASb,MAChB,KAAKa,OAAO8B;EAGd,KAAKR,UAAUzB,MAAM;CACvB;CAEAmC,WAAWC,WAAmB;EAC5B,MAAMrE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,OAAO;GACLjB,UAAUD,QAAQuB,MAAMC,QAAQX,MAAM;GACtCb,SAASA,QAAQwB;GACjB8C,OAAOtE,QAAQyB;EACjB;CACF;CAEA8C,cAAcF,WAAmBG,eAA2C;EAC1E,MAAMxE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAE3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAMuD,cACJD,kBAAkBvD,KAAAA,IACdjB,QAAQuB,OACRiD,kBAAkB,OAChB,OACA,KAAKnE,SAASwC,IAAI2B,aAAa;EACvC,IAAIC,gBAAgBxD,KAAAA,GAClB,MAAM,IAAIC,MACR,0GACF;EAEF,KAAK,IAAIM,UAAUiD,aAAajD,SAASA,UAAUA,QAAQD,MACzD,IAAIC,YAAYxB,SACd,MAAM,IAAIkB,MACR,iJACF;EAGJ,KAAK,MAAMgC,SAASlD,QAAQoB,UAAU;GACpC,MAAMwB,eAAe,KAAKvC,SAASwC,IAAIK,KAAK;GAC5C,IAAI,CAACN,cACH,MAAM,IAAI1B,MACR,4GACF;GACF,KAAK8B,UAAUyB,aAAa7B,cAAc,QAAQ;EACpD;EAEA,KAAKI,UAAU,MAAMhD,SAAS,KAAK;EACnC,KAAKK,SAASqE,OAAOL,SAAS;EAE9B,IAAI,KAAKjC,SAASpC,SAChB,KAAKoC,OAAOV,SAAS+C,eAAe,KAAKpC,IAAI;EAG/C,KAAKqB,UAAUzB,MAAM;CACvB;CAEA0C,YAAYN,WAAmB;EAC7B,MAAMrE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,oGACF;EAEF,MAAM,EAAEE,aAAapB,QAAQuB,QAAQ,KAAKc;EAC1C,OAAOjB;CACT;;;;;;;;;CAUA,iCACEyD,cACAC,aACA;EACA,IAAI,CAACD,cAAc;EAEnB,MAAME,+BAAe,IAAIC,IAAY;EACrC,KAAK,IAAIxD,UAAUsD,aAAatD,SAASA,UAAUA,QAAQD,MACzDwD,aAAaE,IAAIzD,QAAQA,QAAQX,EAAE;EAGrC,MAAMqE,QAAkB,CAAA;EACxB,KACE,IAAI1D,UAAoCqD,cACxCrD,SACAA,UAAUA,QAAQD,MAClB;GAGA,IAAIwD,aAAaI,IAAI3D,QAAQA,QAAQX,EAAE,GAAG;GAC1C,IAAIW,QAAQA,QAAQoC,UAAUC,cAC5BqB,MAAMpC,KAAKtB,QAAQA,QAAQX,EAAE;EAEjC;EAEA,KAAK,MAAMA,MAAMqE,OAEf,IAAI,KAAK7E,SAAS8E,IAAItE,EAAE,GAAG,KAAK0D,cAAc1D,EAAE;CAEpD;CAEAuE,eAAef,WAAmB;EAChC,MAAMrE,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAM2D,eAAe,KAAKzC;EAC1B,KAAKW,aAAa/C,OAAO;EAEzB,KAAKoC,OAAOV,SAAS1B,OAAO;EAE5B,KAAK4E,iCAAiCC,cAAc,KAAKzC,IAAI;EAE7D,KAAKsB,UAAUzB,MAAM;CACvB;CAEAoD,UAAUhB,WAA0B;EAClC,IAAIA,cAAc,MAAM;GACtB,KAAKiB,MAAM;GACX;EACF;EAEA,MAAMtF,UAAU,KAAKK,SAASwC,IAAIwB,SAAS;EAC3C,IAAI,CAACrE,SACH,MAAM,IAAIkB,MACR,iGACF;EAEF,MAAM2D,eAAe,KAAKzC;EAE1B,IAAIpC,QAAQoB,SAASqB,SAAS,GAAG;GAC/B,MAAMD,UAAU,CAAC,GAAGxC,QAAQoB,QAAQ;GACpC,OAAOoB,QAAQC,SAAS,GAAG;IACzB,MAAME,UAAUH,QAAQE,IAAI;IAC5B,MAAME,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cAAc;KAChB,KAAK,MAAM2C,gBAAgB3C,aAAaxB,UACtCoB,QAAQM,KAAKyC,YAAY;KAE3B,KAAKlF,SAASqE,OAAO/B,OAAO;IAC9B;GACF;GAEA3C,QAAQoB,WAAW,CAAA;GACnBpB,QAAQqB,OAAO;EACjB;EAEA,KAAKe,OAAOpC;EACZ,KAAK+C,aAAa/C,OAAO;EAEzB,KAAK4E,iCAAiCC,cAAc,KAAKzC,IAAI;EAE7D,KAAKsB,UAAUzB,MAAM;CACvB;CAEAqD,QAAc;EACZ,KAAKjF,SAASiF,MAAM;EACpB,KAAKlD,OAAO;EACZ,KAAKC,OAAO;GACVjB,UAAU,CAAA;GACVC,MAAM;EACR;EACA,KAAKqC,UAAUzB,MAAM;CACvB;CAEAuD,SAAoC;EAClC,MAAMC,cAAqD,CAAA;EAO3D,MAAMjD,UAAU,CAAC,GAAG,KAAKH,KAAKjB,QAAQ,CAAC,CAACsE,QAAQ;EAChD,OAAOlD,QAAQC,SAAS,GAAG;GACzB,MAAMzC,UAAU,KAAKK,SAASwC,IAAIL,QAAQE,IAAI,CAAE;GAChD,IAAI,CAAC1C,SAAS;GACd,KAAK,IAAI2F,IAAI3F,QAAQoB,SAASqB,SAAS,GAAGkD,KAAK,GAAGA,KAChDnD,QAAQM,KAAK9C,QAAQoB,SAASuE,EAAG;GAEnC,IAAI3F,QAAQwB,QAAQoC,UAAUC,cAAc;GAC5C,IAAItC,OAAOvB,QAAQuB;GACnB,OAAOA,QAAQA,KAAKC,QAAQoC,UAAUC,cACpCtC,OAAOA,KAAKA;GAEdkE,YAAY3C,KAAK;IACf9C,SAASA,QAAQwB;IACjBvB,UAAUsB,MAAMC,QAAQX,MAAM;GAChC,CAAC;EACH;EAEA,OAAO;GACLT,QAAQ,KAAKuD;GACbtD,UAAUoF;EACZ;CACF;CAEAG,OAAO,EAAExF,QAAQC,YAAuC;EACtD,KAAK,MAAM,EAAEL,SAASC,cAAcI,UAClC,KAAK2D,mBAAmB/D,UAAUD,OAAO;EAG3C,KAAKqF,UAAUjF,UAAUC,SAASmD,GAAG,EAAE,CAAC,EAAExD,QAAQa,MAAM,IAAI;CAC9D;AACF"}

package/dist/runtime/utils/pending-tool-calls.d.ts

@@ -1,19 +1,18 @@
-//#region src/runtime/utils/pending-tool-calls.d.ts
 type PendingToolCallMessage<TToolCall> = {
-  toolCalls: readonly TToolCall[];
+    toolCalls: readonly TToolCall[];
 } | {
-  toolCallId: string;
+    toolCallId: string;
 } | undefined;
 export declare const scanPendingToolCalls: <TMessage, TToolCall>(messages: readonly TMessage[], getMessage: (message: TMessage) => PendingToolCallMessage<TToolCall>, getToolCallId: (toolCall: TToolCall) => string) => TToolCall[];
 export declare const createToolCallCancellationStub: (toolCall: {
-  readonly id: string;
-  readonly name: string;
+    readonly id: string;
+    readonly name: string;
 }) => {
-  type: "tool";
-  name: string;
-  tool_call_id: string;
-  content: string;
-  status: "error";
+    type: "tool";
+    name: string;
+    tool_call_id: string;
+    content: string;
+    status: "error";
 };
-//#endregion
+export {};
 //# sourceMappingURL=pending-tool-calls.d.ts.map

package/dist/runtime/utils/pending-tool-calls.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"pending-tool-calls.d.ts","names":[],"sources":["../../../src/runtime/utils/pending-tool-calls.ts"],"mappings":";KAAK,uBAAuB;EACtB,oBAAoB;;EACpB;;qBAGO,uBAAwB,UAAU,WAC7C,mBAAmB,YACnB,aAAa,SAAS,aAAa,uBAAuB,YAC1D,gBAAgB,UAAU,yBACzB;qBAgBU,iCAAkC;WACpC;WACA"}
+{"version":3,"file":"pending-tool-calls.d.ts","sourceRoot":"","sources":["../../../src/runtime/utils/pending-tool-calls.ts"],"names":[],"mappings":"AAAA,KAAK,sBAAsB,CAAC,SAAS,IACjC;IAAE,SAAS,EAAE,SAAS,SAAS,EAAE,CAAA;CAAE,GACnC;IAAE,UAAU,EAAE,MAAM,CAAA;CAAE,GACtB,SAAS,CAAC;AAEd,eAAO,MAAM,oBAAoB,GAAI,QAAQ,EAAE,SAAS,EACtD,UAAU,SAAS,QAAQ,EAAE,EAC7B,YAAY,CAAC,OAAO,EAAE,QAAQ,KAAK,sBAAsB,CAAC,SAAS,CAAC,EACpE,eAAe,CAAC,QAAQ,EAAE,SAAS,KAAK,MAAM,KAC7C,SAAS,EAcX,CAAC;AAEF,eAAO,MAAM,8BAA8B,GAAI,UAAU;IACvD,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;CACvB;;;;;;CAMC,CAAC"}