@assistant-ui/core 0.3.14 → 0.3.16

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

@@ -0,0 +1,221 @@
+import { isJSONValueEqual } from "../../utils/json/is-json-equal.js";
+import { generateErrorMessageId } from "../../utils/id.js";
+import { fromThreadMessageLike } from "./thread-message-like.js";
+import { FALLBACK_ID_PREFIX, bindExternalStoreMessage, getExternalStoreMessages, symbolInnerMessage } from "./external-store-message.js";
+import { getAutoStatus, isAutoStatus, isInterruptedToolCall, isPendingToolCall } from "./auto-status.js";
+//#region src/runtime/utils/external-message-conversion.ts
+const stringifyForError = (value) => {
+	let text;
+	try {
+		text = value instanceof Error ? String(value) : JSON.stringify(value);
+	} catch {}
+	text ??= String(value);
+	return text.length > 200 ? `${text.slice(0, 200)}…` : text;
+};
+const toCallbackOutputs = (output, input) => {
+	const outputs = Array.isArray(output) ? output : [output];
+	for (const o of outputs) if (!(typeof o === "object" && o !== null && (o.role === "tool" || (o.role === "assistant" || o.role === "user" || o.role === "system") && (typeof o.content === "string" || Array.isArray(o.content))))) throw new Error(`External message converter: the converter callback returned an invalid message (${stringifyForError(o)}) for input ${stringifyForError(input)}. Return an empty array to skip a message.`);
+	return outputs.filter((o) => {
+		if (o.role !== "tool") return true;
+		if (typeof o.toolCallId === "string" && o.toolCallId.length > 0) return true;
+		if (!warnedDroppedToolResults.has(o)) {
+			warnedDroppedToolResults.add(o);
+			console.warn(`External message converter: dropping a tool result without a toolCallId (${stringifyForError(o)}) for input ${stringifyForError(input)}.`);
+		}
+		return false;
+	});
+};
+const warnedDroppedToolResults = /* @__PURE__ */ new WeakSet();
+const convertExternalMessageCallback = (input, callback, metadata) => ({
+	input,
+	outputs: toCallbackOutputs(callback(input, metadata), input)
+});
+const mergeInnerMessages = (existing, incoming) => ({ [symbolInnerMessage]: [...existing[symbolInnerMessage] ?? [], ...incoming[symbolInnerMessage] ?? []] });
+const joinExternalMessages = (messages) => {
+	const assistantMessage = {
+		role: "assistant",
+		content: []
+	};
+	for (const output of messages) if (output.role === "tool") {
+		const toolCallIdx = assistantMessage.content.findIndex((c) => c.type === "tool-call" && c.toolCallId === output.toolCallId);
+		if (toolCallIdx !== -1) {
+			const 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}`);
+			}
+			assistantMessage.content[toolCallIdx] = {
+				...toolCall,
+				[symbolInnerMessage]: [...toolCall[symbolInnerMessage] ?? [], output],
+				result: output.result,
+				artifact: output.artifact,
+				isError: output.isError,
+				messages: output.messages
+			};
+		}
+	} else {
+		const role = output.role;
+		const content = (typeof output.content === "string" ? [{
+			type: "text",
+			text: output.content
+		}] : output.content).map((c) => ({
+			...c,
+			[symbolInnerMessage]: [output]
+		}));
+		switch (role) {
+			case "system":
+			case "user": return {
+				...output,
+				content
+			};
+			case "assistant":
+				if (assistantMessage.content.length === 0) {
+					assistantMessage.id = output.id;
+					assistantMessage.createdAt ??= output.createdAt;
+					assistantMessage.status ??= output.status;
+					if (output.attachments) assistantMessage.attachments = [...assistantMessage.attachments ?? [], ...output.attachments];
+				}
+				if (output.metadata) {
+					assistantMessage.metadata ??= {};
+					if (output.metadata.unstable_state !== void 0) assistantMessage.metadata.unstable_state = output.metadata.unstable_state;
+					if (output.metadata.unstable_annotations) assistantMessage.metadata.unstable_annotations = [...assistantMessage.metadata.unstable_annotations ?? [], ...output.metadata.unstable_annotations];
+					if (output.metadata.unstable_data) assistantMessage.metadata.unstable_data = [...assistantMessage.metadata.unstable_data ?? [], ...output.metadata.unstable_data];
+					if (output.metadata.steps) assistantMessage.metadata.steps = [...assistantMessage.metadata.steps ?? [], ...output.metadata.steps];
+					if (output.metadata.custom) assistantMessage.metadata.custom = {
+						...assistantMessage.metadata.custom ?? {},
+						...output.metadata.custom
+					};
+					if (output.metadata.timing) assistantMessage.metadata.timing = output.metadata.timing;
+					if (output.metadata.submittedFeedback) assistantMessage.metadata.submittedFeedback = output.metadata.submittedFeedback;
+					if (output.metadata.isOptimistic) assistantMessage.metadata.isOptimistic = true;
+				}
+				for (const part of content) {
+					if (part.type === "tool-call") {
+						const existingIdx = assistantMessage.content.findIndex((c) => c.type === "tool-call" && c.toolCallId === part.toolCallId);
+						if (existingIdx !== -1) {
+							const existing = assistantMessage.content[existingIdx];
+							assistantMessage.content[existingIdx] = {
+								...existing,
+								...part,
+								...mergeInnerMessages(existing, part)
+							};
+							continue;
+						}
+					}
+					if (part.type === "reasoning" && "parentId" in part && part.parentId) {
+						const existingIdx = assistantMessage.content.findIndex((c) => c.type === "reasoning" && "parentId" in c && c.parentId === part.parentId);
+						if (existingIdx !== -1) {
+							const existing = assistantMessage.content[existingIdx];
+							assistantMessage.content[existingIdx] = {
+								...existing,
+								text: `${existing.text}\n\n${part.text}`,
+								...mergeInnerMessages(existing, part)
+							};
+							continue;
+						}
+					}
+					assistantMessage.content.push(part);
+				}
+				break;
+			default: throw new Error(`Unknown message role: ${role}`);
+		}
+	}
+	return assistantMessage;
+};
+const chunkExternalMessages = (callbackResults, joinStrategy) => {
+	const results = [];
+	let isAssistant = false;
+	let pendingNone = false;
+	let inputs = [];
+	let outputs = [];
+	const flush = () => {
+		if (outputs.length) results.push({
+			inputs,
+			outputs
+		});
+		inputs = [];
+		outputs = [];
+		isAssistant = false;
+		pendingNone = false;
+	};
+	for (const callbackResult of callbackResults) for (const output of callbackResult.outputs) {
+		if (pendingNone && output.role !== "tool" || !isAssistant || output.role === "user" || output.role === "system") flush();
+		isAssistant = output.role === "assistant" || output.role === "tool";
+		if (inputs.at(-1) !== callbackResult.input) inputs.push(callbackResult.input);
+		outputs.push(output);
+		if (output.role === "assistant" && (output.convertConfig?.joinStrategy === "none" || joinStrategy === "none")) pendingNone = true;
+	}
+	flush();
+	return results;
+};
+const shallowArrayEqual = (a, b) => {
+	if (a.length !== b.length) return false;
+	for (let i = 0; i < a.length; i++) if (a[i] !== b[i]) return false;
+	return true;
+};
+const convertExternalMessageChunk = (message, idx, chunkCount, isRunning, error, cache) => {
+	const isLast = idx === chunkCount - 1;
+	const joined = joinExternalMessages(message.outputs);
+	const hasInterruptedToolCalls = typeof joined.content === "object" && joined.content.some(isInterruptedToolCall);
+	const hasPendingToolCalls = typeof joined.content === "object" && joined.content.some(isPendingToolCall);
+	const autoStatus = getAutoStatus(isLast, isRunning, hasInterruptedToolCalls, hasPendingToolCalls, isLast ? error : void 0);
+	const fallbackId = `${FALLBACK_ID_PREFIX}${idx}`;
+	const cachedMessage = cache?.message;
+	if (cachedMessage && (cachedMessage.role !== "assistant" || !isAutoStatus(cachedMessage.status) || cachedMessage.status === autoStatus || cachedMessage.status.type === "incomplete" && cachedMessage.status.reason === "error" && cachedMessage.status.error !== void 0 && autoStatus.type === "incomplete" && autoStatus.reason === "error" && autoStatus.error !== void 0 && (cachedMessage.status.error === autoStatus.error || isJSONValueEqual(cachedMessage.status.error, autoStatus.error)))) {
+		const inputs = getExternalStoreMessages(cachedMessage);
+		if (shallowArrayEqual(inputs, message.inputs)) {
+			if (cache.generatedFallbackMessages.has(cachedMessage) && cachedMessage.id !== fallbackId) {
+				const updated = {
+					...cachedMessage,
+					id: fallbackId
+				};
+				cache.generatedFallbackMessages.add(updated);
+				bindExternalStoreMessage(updated, message.inputs);
+				return updated;
+			}
+			return cachedMessage;
+		}
+	}
+	const newMessage = fromThreadMessageLike(joined, fallbackId, autoStatus);
+	if (cache && joined.id == null) cache.generatedFallbackMessages.add(newMessage);
+	bindExternalStoreMessage(newMessage, message.inputs);
+	return newMessage;
+};
+function createErrorAssistantMessage(error) {
+	const msg = {
+		id: generateErrorMessageId(),
+		role: "assistant",
+		content: [],
+		status: {
+			type: "incomplete",
+			reason: "error",
+			error
+		},
+		createdAt: /* @__PURE__ */ new Date(),
+		metadata: {
+			unstable_state: null,
+			unstable_annotations: [],
+			unstable_data: [],
+			custom: {},
+			steps: []
+		}
+	};
+	bindExternalStoreMessage(msg, []);
+	return msg;
+}
+const completeExternalMessageConversion = (messages, error) => {
+	if (error) {
+		const lastMessage = messages.at(-1);
+		if (!lastMessage || lastMessage.role !== "assistant") messages.push(createErrorAssistantMessage(error));
+	}
+	return messages;
+};
+const convertExternalMessages = (messages, callback, isRunning, metadata) => {
+	const callbackResults = messages.map((message) => convertExternalMessageCallback(message, callback, metadata));
+	const chunks = chunkExternalMessages(callbackResults);
+	const result = chunks.map((message, idx) => convertExternalMessageChunk(message, idx, chunks.length, isRunning, metadata.error));
+	return completeExternalMessageConversion(result, metadata.error);
+};
+//#endregion
+export { chunkExternalMessages, completeExternalMessageConversion, convertExternalMessageCallback, convertExternalMessageChunk, convertExternalMessages, joinExternalMessages, shallowArrayEqual };
+
+//# sourceMappingURL=external-message-conversion.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"external-message-conversion.js","names":["ReadonlyJSONValue","ToolExecutionStatus","ThreadAssistantMessage","ThreadMessage","ToolCallMessagePart","MessageTiming","generateErrorMessageId","isJSONValueEqual","getAutoStatus","isAutoStatus","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","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","joinExternalMessages","assistantMessage","Omit","Exclude","toolCallIdx","findIndex","c","type","toolCall","const","map","id","createdAt","status","attachments","unstable_state","undefined","unstable_annotations","unstable_data","steps","custom","timing","submittedFeedback","isOptimistic","part","existingIdx","parentId","push","unsupportedRole","chunkExternalMessages","callbackResults","results","isAssistant","pendingNone","flush","callbackResult","at","shallowArrayEqual","a","b","i","ExternalMessageConversionCache","generatedFallbackMessages","convertExternalMessageChunk","idx","chunkCount","isRunning","cache","isLast","joined","hasInterruptedToolCalls","some","hasPendingToolCalls","autoStatus","fallbackId","cachedMessage","reason","updated","newMessage","createErrorAssistantMessage","msg","Date","completeExternalMessageConversion","lastMessage","convertExternalMessages","WeakKey","chunks"],"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 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  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 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\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  for (const output of messages) {\n    if (output.role === \"tool\") {\n      const toolCallIdx = assistantMessage.content.findIndex(\n        (c) => c.type === \"tool-call\" && c.toolCallId === output.toolCallId,\n      );\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.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\") {\n              const existingIdx = assistantMessage.content.findIndex(\n                (c) =>\n                  c.type === \"tool-call\" && c.toolCallId === part.toolCallId,\n              );\n              if (existingIdx !== -1) {\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 = assistantMessage.content.findIndex(\n                (c) =>\n                  c.type === \"reasoning\" &&\n                  \"parentId\" in c &&\n                  c.parentId === part.parentId,\n              );\n              if (existingIdx !== -1) {\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            assistantMessage.content.push(part);\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      if (\n        (pendingNone && output.role !== \"tool\") ||\n        !isAssistant ||\n        output.role === \"user\" ||\n        output.role === \"system\"\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        (output.convertConfig?.joinStrategy === \"none\" ||\n          joinStrategy === \"none\")\n      ) {\n        pendingNone = true;\n      }\n    }\n  }\n  flush();\n  return results;\n};\n\nexport const shallowArrayEqual = (a: unknown[], b: unknown[]) => {\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 ExternalMessageConversionCache = {\n  message: ThreadMessage | undefined;\n  generatedFallbackMessages: WeakSet<object>;\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?: ExternalMessageConversionCache,\n) => {\n  const isLast = idx === chunkCount - 1;\n  const joined = joinExternalMessages(message.outputs);\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 autoStatus = getAutoStatus(\n    isLast,\n    isRunning,\n    hasInterruptedToolCalls,\n    hasPendingToolCalls,\n    isLast ? error : undefined,\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) => {\n  const callbackResults = messages.map((message) =>\n    convertExternalMessageCallback(message, callback, metadata),\n  );\n  const chunks = chunkExternalMessages(callbackResults);\n  const result = chunks.map((message, idx) =>\n    convertExternalMessageChunk(\n      message,\n      idx,\n      chunks.length,\n      isRunning,\n      metadata.error,\n    ),\n  );\n  return completeExternalMessageConversion(result, metadata.error);\n};\n"],"mappings":";;;;;;AAsEA,MAAM6C,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,EAAEpC,SAAS,WACRoC,EAAEpC,SAAS,eAAeoC,EAAEpC,SAAS,UAAUoC,EAAEpC,SAAS,cACzD,OAAOoC,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,EAAEpC,SAAS,QAAQ,OAAO;EAC9B,IAAI,OAAOoC,EAAEnC,eAAe,YAAYmC,EAAEnC,WAAW6B,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,GACjEzD,qBAAqB,CACpB,GAAKwD,SAAiBxD,uBAAuB,CAAA,GAC7C,GAAKyD,SAAiBzD,uBAAuB,CAAA,CAAG,EAEpD;AAEA,MAAa0D,wBACX7C,aACsB;CACtB,MAAM8C,mBAGF;EACFpD,MAAM;EACNsC,SAAS,CAAA;CACX;CACA,KAAK,MAAML,UAAU3B,UACnB,IAAI2B,OAAOjC,SAAS,QAAQ;EAC1B,MAAMuD,cAAcH,iBAAiBd,QAAQkB,WAC1CC,MAAMA,EAAEC,SAAS,eAAeD,EAAExD,eAAegC,OAAOhC,UAC3D;EAGA,IAAIsD,gBAAgB,IAAI;GACtB,MAAMI,WAAWP,iBAAiBd,QAChCiB;GAEF,IAAItB,OAAO/B,YAAY,MACjByD;QAAAA,SAASzD,aAAa+B,OAAO/B,UAC/B,MAAM,IAAIwB,MACR,kBAAkBO,OAAOhC,WAAU,GAAIgC,OAAO/B,SAAQ,qCAAsCyD,SAASzD,UACvG;GAAA;GAEJkD,iBAAiBd,QAAQiB,eAAe;IACtC,GAAGI;KAEAlE,qBAAqB,CACpB,GAAKkE,SAAiBlE,uBAAuB,CAAA,GAC7CwC,MAAM;IAGV9B,QAAQ8B,OAAO9B;IACfC,UAAU6B,OAAO7B;IACjBC,SAAS4B,OAAO5B;IAChBC,UAAU2B,OAAO3B;GACnB;EACF;CACF,OAAO;EACL,MAAMN,OAAOiC,OAAOjC;EACpB,MAAMsC,WACJ,OAAOL,OAAOK,YAAY,WACtB,CAAC;GAAEoB,MAAM;GAAiBjC,MAAMQ,OAAOK;EAAQ,CAAC,IAChDL,OAAOK,QAAAA,CACXuB,KAAKJ,OAAO;GACZ,GAAGA;IACGhE,qBAAqB,CAACwC,MAAM;EACpC,EAAE;EACF,QAAQjC,MAAR;GACE,KAAK;GACL,KAAK,QACH,OAAO;IACL,GAAGiC;IACHK;GACF;GACF,KAAK;IACH,IAAIc,iBAAiBd,QAAQR,WAAW,GAAG;KACzCsB,iBAAiBU,KAAK7B,OAAO6B;KAC7BV,iBAAiBW,cAAc9B,OAAO8B;KACtCX,iBAAiBY,WAAW/B,OAAO+B;KAEnC,IAAI/B,OAAOgC,aACTb,iBAAiBa,cAAc,CAC7B,GAAIb,iBAAiBa,eAAe,CAAA,GACpC,GAAGhC,OAAOgC,WAAW;IAG3B;IAEA,IAAIhC,OAAOlB,UAAU;KACnBqC,iBAAiBrC,aAAa,CAAC;KAC/B,IAAIkB,OAAOlB,SAASmD,mBAAmBC,KAAAA,GACrCf,iBAAiBrC,SAASmD,iBACxBjC,OAAOlB,SAASmD;KAEpB,IAAIjC,OAAOlB,SAASqD,sBAClBhB,iBAAiBrC,SAASqD,uBAAuB,CAC/C,GAAIhB,iBAAiBrC,SAASqD,wBAAwB,CAAA,GACtD,GAAGnC,OAAOlB,SAASqD,oBAAoB;KAG3C,IAAInC,OAAOlB,SAASsD,eAClBjB,iBAAiBrC,SAASsD,gBAAgB,CACxC,GAAIjB,iBAAiBrC,SAASsD,iBAAiB,CAAA,GAC/C,GAAGpC,OAAOlB,SAASsD,aAAa;KAGpC,IAAIpC,OAAOlB,SAASuD,OAClBlB,iBAAiBrC,SAASuD,QAAQ,CAChC,GAAIlB,iBAAiBrC,SAASuD,SAAS,CAAA,GACvC,GAAGrC,OAAOlB,SAASuD,KAAK;KAG5B,IAAIrC,OAAOlB,SAASwD,QAClBnB,iBAAiBrC,SAASwD,SAAS;MACjC,GAAInB,iBAAiBrC,SAASwD,UAAU,CAAC;MACzC,GAAGtC,OAAOlB,SAASwD;KACrB;KAGF,IAAItC,OAAOlB,SAASyD,QAClBpB,iBAAiBrC,SAASyD,SAASvC,OAAOlB,SAASyD;KAGrD,IAAIvC,OAAOlB,SAAS0D,mBAClBrB,iBAAiBrC,SAAS0D,oBACxBxC,OAAOlB,SAAS0D;KAGpB,IAAIxC,OAAOlB,SAAS2D,cAClBtB,iBAAiBrC,SAAS2D,eAAe;IAG7C;IAGA,KAAK,MAAMC,QAAQrC,SAAS;KAC1B,IAAIqC,KAAKjB,SAAS,aAAa;MAC7B,MAAMkB,cAAcxB,iBAAiBd,QAAQkB,WAC1CC,MACCA,EAAEC,SAAS,eAAeD,EAAExD,eAAe0E,KAAK1E,UACpD;MACA,IAAI2E,gBAAgB,IAAI;OACtB,MAAM3B,WAAWG,iBAAiBd,QAChCsC;OAEFxB,iBAAiBd,QAAQsC,eAAe;QACtC,GAAG3B;QACH,GAAG0B;QACH,GAAG3B,mBAAmBC,UAAU0B,IAAI;OACtC;OACA;MACF;KACF;KAEA,IACEA,KAAKjB,SAAS,eACd,cAAciB,QACdA,KAAKE,UACL;MACA,MAAMD,cAAcxB,iBAAiBd,QAAQkB,WAC1CC,MACCA,EAAEC,SAAS,eACX,cAAcD,KACdA,EAAEoB,aAAaF,KAAKE,QACxB;MACA,IAAID,gBAAgB,IAAI;OACtB,MAAM3B,WAAWG,iBAAiBd,QAChCsC;OAEFxB,iBAAiBd,QAAQsC,eAAe;QACtC,GAAG3B;QACHxB,MAAM,GAAGwB,SAASxB,KAAI,MAAOkD,KAAKlD;QAClC,GAAGuB,mBAAmBC,UAAU0B,IAAI;OACtC;OACA;MACF;KACF;KACAvB,iBAAiBd,QAAQwC,KAAKH,IAAI;IACpC;IACA;GACF,SAEE,MAAM,IAAIjD,MAAM,yBAAyBqD,MAAiB;EAE9D;CACF;CAEF,OAAO3B;AACT;AAEA,MAAa4B,yBACXC,iBACAlF,iBACG;CACH,MAAMmF,UAA8C,CAAA;CACpD,IAAIC,cAAc;CAClB,IAAIC,cAAc;CAClB,IAAIhE,SAAc,CAAA;CAClB,IAAIF,UAA6C,CAAA;CAEjD,MAAMmE,cAAc;EAClB,IAAInE,QAAQY,QACVoD,QAAQJ,KAAK;GACX1D;GACAF;EACF,CAAC;EAEHE,SAAS,CAAA;EACTF,UAAU,CAAA;EACViE,cAAc;EACdC,cAAc;CAChB;CAEA,KAAK,MAAME,kBAAkBL,iBAC3B,KAAK,MAAMhD,UAAUqD,eAAepE,SAAS;EAC3C,IACGkE,eAAenD,OAAOjC,SAAS,UAChC,CAACmF,eACDlD,OAAOjC,SAAS,UAChBiC,OAAOjC,SAAS,UAEhBqF,MAAM;EAERF,cAAclD,OAAOjC,SAAS,eAAeiC,OAAOjC,SAAS;EAE7D,IAAIoB,OAAOmE,GAAG,EAAE,MAAMD,eAAerE,OACnCG,OAAO0D,KAAKQ,eAAerE,KAAK;EAElCC,QAAQ4D,KAAK7C,MAAM;EAEnB,IACEA,OAAOjC,SAAS,gBACfiC,OAAOnC,eAAeC,iBAAiB,UACtCA,iBAAiB,SAEnBqF,cAAc;CAElB;CAEFC,MAAM;CACN,OAAOH;AACT;AAEA,MAAaM,qBAAqBC,GAAcC,MAAiB;CAC/D,IAAID,EAAE3D,WAAW4D,EAAE5D,QAAQ,OAAO;CAClC,KAAK,IAAI6D,IAAI,GAAGA,IAAIF,EAAE3D,QAAQ6D,KAC5B,IAAIF,EAAEE,OAAOD,EAAEC,IAAI,OAAO;CAE5B,OAAO;AACT;AAOA,MAAaG,+BACXjF,SACAkF,KACAC,YACAC,WACAvF,OACAwF,UACG;CACH,MAAMC,SAASJ,QAAQC,aAAa;CACpC,MAAMI,SAASjD,qBAAqBtC,QAAQK,OAAO;CACnD,MAAMmF,0BACJ,OAAOD,OAAO9D,YAAY,YAC1B8D,OAAO9D,QAAQgE,KAAKlH,qBAAqB;CAC3C,MAAMmH,sBACJ,OAAOH,OAAO9D,YAAY,YAC1B8D,OAAO9D,QAAQgE,KAAKjH,iBAAiB;CACvC,MAAMmH,aAAatH,cACjBiH,QACAF,WACAI,yBACAE,qBACAJ,SAASzF,QAAQyD,KAAAA,CACnB;CACA,MAAMsC,aAAa,GAAGlH,qBAAqBwG;CAE3C,MAAMW,gBAAgBR,OAAOrF;CAC7B,IACE6F,kBACCA,cAAc1G,SAAS,eACtB,CAACb,aAAauH,cAAc1C,MAAM,KAClC0C,cAAc1C,WAAWwC,cACxBE,cAAc1C,OAAON,SAAS,gBAC7BgD,cAAc1C,OAAO2C,WAAW,WAChCD,cAAc1C,OAAOtD,UAAUyD,KAAAA,KAC/BqC,WAAW9C,SAAS,gBACpB8C,WAAWG,WAAW,WACtBH,WAAW9F,UAAUyD,KAAAA,MACpBuC,cAAc1C,OAAOtD,UAAU8F,WAAW9F,SACzCzB,iBAAiByH,cAAc1C,OAAOtD,OAAO8F,WAAW9F,KAAK,KACnE;EACA,MAAMU,SAAS5B,yBAA4BkH,aAAa;EACxD,IAAIlB,kBAAkBpE,QAAQP,QAAQO,MAAM,GAAG;GAI7C,IACE8E,MAAML,0BAA0BpD,IAAIiE,aAAa,KACjDA,cAAc5C,OAAO2C,YACrB;IACA,MAAMG,UAAU;KAAE,GAAGF;KAAe5C,IAAI2C;IAAW;IACnDP,MAAML,0BAA0BnD,IAAIkE,OAAO;IAC3CtH,yBAAyBsH,SAAS/F,QAAQO,MAAM;IAChD,OAAOwF;GACT;GACA,OAAOF;EACT;CACF;CAEA,MAAMG,aAAanH,sBAAsB0G,QAAQK,YAAYD,UAAU;CACvE,IAAIN,SAASE,OAAOtC,MAAM,MACxBoC,MAAML,0BAA0BnD,IAAImE,UAAU;CAEhDvH,yBAAyBuH,YAAYhG,QAAQO,MAAM;CACnD,OAAOyF;AACT;AAEA,SAASC,4BACPpG,OACwB;CACxB,MAAMqG,MAA8B;EAClCjD,IAAI9E,uBAAuB;EAC3BgB,MAAM;EACNsC,SAAS,CAAA;EACT0B,QAAQ;GAAEN,MAAM;GAAciD,QAAQ;GAASjG;EAAM;EACrDqD,2BAAW,IAAIiD,KAAK;EACpBjG,UAAU;GACRmD,gBAAgB;GAChBE,sBAAsB,CAAA;GACtBC,eAAe,CAAA;GACfE,QAAQ,CAAC;GACTD,OAAO,CAAA;EACT;CACF;CACAhF,yBAAyByH,KAAK,CAAA,CAAE;CAChC,OAAOA;AACT;AAEA,MAAaE,qCACX3G,UACAI,UACG;CACH,IAAIA,OAAO;EACT,MAAMwG,cAAc5G,SAASiF,GAAG,EAAE;EAClC,IAAI,CAAC2B,eAAeA,YAAYlH,SAAS,aACvCM,SAASwE,KAAKgC,4BAA4BpG,KAAK,CAAC;CAEpD;CACA,OAAOJ;AACT;AAEA,MAAa6G,2BACX7G,UACAyC,UACAkD,WACAlF,aACG;CACH,MAAMkE,kBAAkB3E,SAASuD,KAAKhD,YACpCiC,+BAA+BjC,SAASkC,UAAUhC,QAAQ,CAC5D;CACA,MAAMsG,SAASrC,sBAAsBC,eAAe;CACpD,MAAM9E,SAASkH,OAAOxD,KAAKhD,SAASkF,QAClCD,4BACEjF,SACAkF,KACAsB,OAAOvF,QACPmE,WACAlF,SAASL,KACX,CACF;CACA,OAAOuG,kCAAkC9G,QAAQY,SAASL,KAAK;AACjE"}

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

@@ -12,6 +12,7 @@ declare const bindExternalStoreMessage: <T>(target: object, message: T | T[]) =>
 declare const getExternalStoreMessages: <T>(input: {
   messages: readonly ThreadMessage[];
 } | ThreadMessage | ThreadMessage["content"][number]) => T[];
+declare const FALLBACK_ID_PREFIX = "__external_store_fallback_";
 //#endregion
-export { bindExternalStoreMessage, getExternalStoreMessages, symbolInnerMessage };
+export { FALLBACK_ID_PREFIX, bindExternalStoreMessage, getExternalStoreMessages, symbolInnerMessage };
 //# 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":";;cAEa;;;;;;;;cAiBA,2BAA4B,GACvC,gBACA,SAAS,IAAI;cAMF,2BAA4B,GACvC;EACM,mBAAmB;IACrB,gBACA,qCAAgC"}
+{"version":3,"file":"external-store-message.d.ts","names":[],"sources":["../../../src/runtime/utils/external-store-message.ts"],"mappings":";;cAEa;;;;;;;;cAiBA,2BAA4B,GACvC,gBACA,SAAS,IAAI;cAMF,2BAA4B,GACvC;EACM,mBAAmB;IACrB,gBACA,qCAAgC;cAczB"}

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

@@ -21,7 +21,8 @@ const getExternalStoreMessages = (input) => {
 	container[symbolInnerMessages] = [value];
 	return container[symbolInnerMessages];
 };
+const FALLBACK_ID_PREFIX = "__external_store_fallback_";
 //#endregion
-export { bindExternalStoreMessage, getExternalStoreMessages, symbolInnerMessage };
+export { FALLBACK_ID_PREFIX, bindExternalStoreMessage, getExternalStoreMessages, symbolInnerMessage };
 
 //# sourceMappingURL=external-store-message.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"external-store-message.js","names":["ThreadMessage","symbolInnerMessage","Symbol","symbolInnerMessages","WithInnerMessages","T","EMPTY_ARRAY","bindExternalStoreMessage","target","message","getExternalStoreMessages","input","messages","container","value","Array","isArray"],"sources":["../../../src/runtime/utils/external-store-message.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\n\nexport const symbolInnerMessage = Symbol(\"innerMessage\");\nconst symbolInnerMessages = Symbol(\"innerMessages\");\n\ntype WithInnerMessages<T> = {\n  [symbolInnerMessage]?: T | T[];\n  [symbolInnerMessages]?: T[];\n};\n\nconst EMPTY_ARRAY: never[] = [];\n\n/**\n * Attach the original external store message(s) to a ThreadMessage or message part.\n * This is a no-op if the target already has a bound message.\n * Use `getExternalStoreMessages` to retrieve the bound messages later.\n *\n * @deprecated This API is experimental and may change without notice.\n */\nexport const bindExternalStoreMessage = <T>(\n  target: object,\n  message: T | T[],\n): void => {\n  if (symbolInnerMessage in target) return;\n  (target as WithInnerMessages<T>)[symbolInnerMessage] = message;\n};\n\nexport const getExternalStoreMessages = <T>(\n  input:\n    | { messages: readonly ThreadMessage[] }\n    | ThreadMessage\n    | ThreadMessage[\"content\"][number],\n) => {\n  const container = (\n    \"messages\" in input ? input.messages : input\n  ) as WithInnerMessages<T>;\n  const value = container[symbolInnerMessages] || container[symbolInnerMessage];\n  if (!value) return EMPTY_ARRAY;\n  if (Array.isArray(value)) {\n    return value;\n  }\n  container[symbolInnerMessages] = [value];\n  return container[symbolInnerMessages];\n};\n"],"mappings":";AAEA,MAAaC,qBAAqBC,OAAO,cAAc;AACvD,MAAMC,sBAAsBD,OAAO,eAAe;AAOlD,MAAMI,cAAuB,CAAA;;;;;;;;AAS7B,MAAaC,4BACXC,QACAC,YACS;CACT,IAAIR,sBAAsBO,QAAQ;CAClC,OAAiCP,sBAAsBQ;AACzD;AAEA,MAAaC,4BACXC,UAIG;CACH,MAAME,YACJ,cAAcF,QAAQA,MAAMC,WAAWD;CAEzC,MAAMG,QAAQD,UAAUV,wBAAwBU,UAAUZ;CAC1D,IAAI,CAACa,OAAO,OAAOR;CACnB,IAAIS,MAAMC,QAAQF,KAAK,GACrB,OAAOA;CAETD,UAAUV,uBAAuB,CAACW,KAAK;CACvC,OAAOD,UAAUV;AACnB"}
+{"version":3,"file":"external-store-message.js","names":["ThreadMessage","symbolInnerMessage","Symbol","symbolInnerMessages","WithInnerMessages","T","EMPTY_ARRAY","bindExternalStoreMessage","target","message","getExternalStoreMessages","input","messages","container","value","Array","isArray","FALLBACK_ID_PREFIX"],"sources":["../../../src/runtime/utils/external-store-message.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\n\nexport const symbolInnerMessage = Symbol(\"innerMessage\");\nconst symbolInnerMessages = Symbol(\"innerMessages\");\n\ntype WithInnerMessages<T> = {\n  [symbolInnerMessage]?: T | T[];\n  [symbolInnerMessages]?: T[];\n};\n\nconst EMPTY_ARRAY: never[] = [];\n\n/**\n * Attach the original external store message(s) to a ThreadMessage or message part.\n * This is a no-op if the target already has a bound message.\n * Use `getExternalStoreMessages` to retrieve the bound messages later.\n *\n * @deprecated This API is experimental and may change without notice.\n */\nexport const bindExternalStoreMessage = <T>(\n  target: object,\n  message: T | T[],\n): void => {\n  if (symbolInnerMessage in target) return;\n  (target as WithInnerMessages<T>)[symbolInnerMessage] = message;\n};\n\nexport const getExternalStoreMessages = <T>(\n  input:\n    | { messages: readonly ThreadMessage[] }\n    | ThreadMessage\n    | ThreadMessage[\"content\"][number],\n) => {\n  const container = (\n    \"messages\" in input ? input.messages : input\n  ) as WithInnerMessages<T>;\n  const value = container[symbolInnerMessages] || container[symbolInnerMessage];\n  if (!value) return EMPTY_ARRAY;\n  if (Array.isArray(value)) {\n    return value;\n  }\n  container[symbolInnerMessages] = [value];\n  return container[symbolInnerMessages];\n};\n\nexport const FALLBACK_ID_PREFIX = \"__external_store_fallback_\";\n"],"mappings":";AAEA,MAAaC,qBAAqBC,OAAO,cAAc;AACvD,MAAMC,sBAAsBD,OAAO,eAAe;AAOlD,MAAMI,cAAuB,CAAA;;;;;;;;AAS7B,MAAaC,4BACXC,QACAC,YACS;CACT,IAAIR,sBAAsBO,QAAQ;CAClC,OAAiCP,sBAAsBQ;AACzD;AAEA,MAAaC,4BACXC,UAIG;CACH,MAAME,YACJ,cAAcF,QAAQA,MAAMC,WAAWD;CAEzC,MAAMG,QAAQD,UAAUV,wBAAwBU,UAAUZ;CAC1D,IAAI,CAACa,OAAO,OAAOR;CACnB,IAAIS,MAAMC,QAAQF,KAAK,GACrB,OAAOA;CAETD,UAAUV,uBAAuB,CAACW,KAAK;CACvC,OAAOD,UAAUV;AACnB;AAEA,MAAac,qBAAqB"}

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

@@ -0,0 +1,30 @@
+import { ThreadMessage } from "../../types/message.js";
+import { ExportedMessageRepository, MessageRepository } from "./message-repository.js";
+//#region src/runtime/utils/message-repository-session.d.ts
+type MessageRepositorySessionOptions = {
+  decorateExport?: (exported: ExportedMessageRepository, repository: MessageRepository) => ExportedMessageRepository;
+};
+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;
+};
+type MessageRepositorySession = ReturnType<typeof createMessageRepositorySession>;
+//#endregion
+export { MessageRepositorySession, MessageRepositorySessionOptions, createMessageRepositorySession };
+//# sourceMappingURL=message-repository-session.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"message-repository-session.d.ts","names":[],"sources":["../../../src/runtime/utils/message-repository-session.ts"],"mappings":";;;KAMY;EACV,kBACE,UAAU,2BACV,YAAY,sBACT;;cAGM,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;;KA8EA,2BAA2B,kBAC9B"}

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

@@ -0,0 +1,130 @@
+import { MessageRepository } from "./message-repository.js";
+//#region src/runtime/utils/message-repository-session.ts
+const createMessageRepositorySession = (options = {}) => {
+	const repository = new MessageRepository();
+	let exportedRepository;
+	const getMessages = () => {
+		return repository.getMessages();
+	};
+	const tryGetMessage = (messageId) => {
+		try {
+			return repository.getMessage(messageId);
+		} catch {
+			return;
+		}
+	};
+	const tryGetMessages = (messageId) => {
+		try {
+			return repository.getMessages(messageId);
+		} catch {
+			return;
+		}
+	};
+	const hasMessage = (messageId) => {
+		return tryGetMessage(messageId) !== void 0;
+	};
+	const addOrUpdateMessage = (parentId, message) => {
+		repository.addOrUpdateMessage(parentId, message);
+		exportedRepository = void 0;
+	};
+	const deleteMessage = (messageId, replacementId) => {
+		repository.deleteMessage(messageId, replacementId);
+		exportedRepository = void 0;
+	};
+	const tryDeleteMessage = (messageId) => {
+		if (!hasMessage(messageId)) return false;
+		deleteMessage(messageId);
+		return true;
+	};
+	const switchToBranch = (messageId) => {
+		repository.switchToBranch(messageId);
+		exportedRepository = void 0;
+	};
+	const resetHead = (messageId) => {
+		repository.resetHead(messageId);
+		exportedRepository = void 0;
+	};
+	const clear = () => {
+		repository.clear();
+		exportedRepository = void 0;
+	};
+	const updateMessage = (messageId, updater) => {
+		const item = tryGetMessage(messageId);
+		if (!item) return false;
+		const message = updater(item.message);
+		if (message === item.message) return false;
+		addOrUpdateMessage(item.parentId, message);
+		return true;
+	};
+	const applyExternalMessageRepository = (loaded) => {
+		const headId = loaded.headId ?? loaded.messages.at(-1)?.message.id ?? null;
+		const ids = /* @__PURE__ */ new Set();
+		let degenerate = false;
+		for (const { message } of loaded.messages) {
+			if (ids.has(message.id)) {
+				degenerate = true;
+				break;
+			}
+			ids.add(message.id);
+		}
+		if (headId !== null && !ids.has(headId)) degenerate = true;
+		if (!degenerate) {
+			clear();
+			let pending = [...loaded.messages];
+			const importedIds = /* @__PURE__ */ new Set();
+			while (pending.length > 0) {
+				const unresolved = [];
+				let progressed = false;
+				for (const item of pending) {
+					if (item.parentId !== null && !importedIds.has(item.parentId)) {
+						unresolved.push(item);
+						continue;
+					}
+					addOrUpdateMessage(item.parentId, item.message);
+					importedIds.add(item.message.id);
+					progressed = true;
+				}
+				if (!progressed) {
+					degenerate = true;
+					break;
+				}
+				pending = unresolved;
+			}
+		}
+		if (degenerate) {
+			clear();
+			let previousId = null;
+			for (const { message } of loaded.messages) {
+				const existing = tryGetMessage(message.id);
+				addOrUpdateMessage(existing ? existing.parentId : previousId, message);
+				previousId = message.id;
+			}
+			resetHead(previousId);
+		} else resetHead(headId);
+	};
+	return {
+		getMessages,
+		get headId() {
+			return repository.headId;
+		},
+		export: () => {
+			exportedRepository ??= options.decorateExport ? options.decorateExport(repository.export(), repository) : repository.export();
+			return exportedRepository;
+		},
+		tryGetMessage,
+		tryGetMessages,
+		hasMessage,
+		addOrUpdateMessage,
+		deleteMessage,
+		tryDeleteMessage,
+		switchToBranch,
+		resetHead,
+		clear,
+		updateMessage,
+		applyExternalMessageRepository
+	};
+};
+//#endregion
+export { createMessageRepositorySession };
+
+//# sourceMappingURL=message-repository-session.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"message-repository-session.js","names":["ThreadMessage","MessageRepository","ExportedMessageRepository","MessageRepositorySessionOptions","decorateExport","exported","repository","createMessageRepositorySession","options","exportedRepository","getMessages","tryGetMessage","messageId","getMessage","undefined","tryGetMessages","hasMessage","addOrUpdateMessage","parentId","message","deleteMessage","replacementId","tryDeleteMessage","switchToBranch","resetHead","clear","updateMessage","updater","item","applyExternalMessageRepository","loaded","headId","messages","at","id","ids","Set","degenerate","has","add","pending","importedIds","length","unresolved","progressed","push","previousId","existing","export","MessageRepositorySession","ReturnType"],"sources":["../../../src/runtime/utils/message-repository-session.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\nimport {\n  MessageRepository,\n  type ExportedMessageRepository,\n} from \"./message-repository\";\n\nexport type MessageRepositorySessionOptions = {\n  decorateExport?: (\n    exported: ExportedMessageRepository,\n    repository: MessageRepository,\n  ) => ExportedMessageRepository;\n};\n\nexport const createMessageRepositorySession = (\n  options: MessageRepositorySessionOptions = {},\n) => {\n  const repository = new MessageRepository();\n  let exportedRepository: ExportedMessageRepository | undefined;\n\n  const getMessages = (): readonly ThreadMessage[] => {\n    return repository.getMessages();\n  };\n\n  const tryGetMessage = (messageId: string) => {\n    try {\n      return repository.getMessage(messageId);\n    } catch {\n      return undefined;\n    }\n  };\n\n  const tryGetMessages = (\n    messageId: string,\n  ): readonly ThreadMessage[] | undefined => {\n    try {\n      return repository.getMessages(messageId);\n    } catch {\n      return undefined;\n    }\n  };\n\n  const hasMessage = (messageId: string): boolean => {\n    return tryGetMessage(messageId) !== undefined;\n  };\n\n  const addOrUpdateMessage = (\n    parentId: string | null,\n    message: ThreadMessage,\n  ): void => {\n    repository.addOrUpdateMessage(parentId, message);\n    exportedRepository = undefined;\n  };\n\n  const deleteMessage = (\n    messageId: string,\n    replacementId?: string | null,\n  ): void => {\n    repository.deleteMessage(messageId, replacementId);\n    exportedRepository = undefined;\n  };\n\n  const tryDeleteMessage = (messageId: string): boolean => {\n    if (!hasMessage(messageId)) return false;\n    deleteMessage(messageId);\n    return true;\n  };\n\n  const switchToBranch = (messageId: string): void => {\n    repository.switchToBranch(messageId);\n    exportedRepository = undefined;\n  };\n\n  const resetHead = (messageId: string | null): void => {\n    repository.resetHead(messageId);\n    exportedRepository = undefined;\n  };\n\n  const clear = (): void => {\n    repository.clear();\n    exportedRepository = undefined;\n  };\n\n  const updateMessage = (\n    messageId: string,\n    updater: (message: ThreadMessage) => ThreadMessage,\n  ): boolean => {\n    const item = tryGetMessage(messageId);\n    if (!item) return false;\n    const message = updater(item.message);\n    if (message === item.message) return false;\n    addOrUpdateMessage(item.parentId, message);\n    return true;\n  };\n\n  const applyExternalMessageRepository = (\n    loaded: ExportedMessageRepository,\n  ): void => {\n    const headId = loaded.headId ?? loaded.messages.at(-1)?.message.id ?? null;\n    const ids = new Set<string>();\n    let degenerate = false;\n    for (const { message } of loaded.messages) {\n      if (ids.has(message.id)) {\n        degenerate = true;\n        break;\n      }\n      ids.add(message.id);\n    }\n    if (headId !== null && !ids.has(headId)) degenerate = true;\n\n    if (!degenerate) {\n      clear();\n      let pending = [...loaded.messages];\n      const importedIds = new Set<string>();\n\n      while (pending.length > 0) {\n        const unresolved: typeof pending = [];\n        let progressed = false;\n        for (const item of pending) {\n          if (item.parentId !== null && !importedIds.has(item.parentId)) {\n            unresolved.push(item);\n            continue;\n          }\n          addOrUpdateMessage(item.parentId, item.message);\n          importedIds.add(item.message.id);\n          progressed = true;\n        }\n        if (!progressed) {\n          degenerate = true;\n          break;\n        }\n        pending = unresolved;\n      }\n    }\n\n    if (degenerate) {\n      clear();\n      let previousId: string | null = null;\n      for (const { message } of loaded.messages) {\n        const existing = tryGetMessage(message.id);\n        addOrUpdateMessage(existing ? existing.parentId : previousId, message);\n        previousId = message.id;\n      }\n      resetHead(previousId);\n    } else {\n      resetHead(headId);\n    }\n  };\n\n  return {\n    getMessages,\n    get headId() {\n      return repository.headId;\n    },\n    export: (): ExportedMessageRepository => {\n      exportedRepository ??= options.decorateExport\n        ? options.decorateExport(repository.export(), repository)\n        : repository.export();\n      return exportedRepository;\n    },\n    tryGetMessage,\n    tryGetMessages,\n    hasMessage,\n    addOrUpdateMessage,\n    deleteMessage,\n    tryDeleteMessage,\n    switchToBranch,\n    resetHead,\n    clear,\n    updateMessage,\n    applyExternalMessageRepository,\n  };\n};\n\nexport type MessageRepositorySession = ReturnType<\n  typeof createMessageRepositorySession\n>;\n"],"mappings":";;AAaA,MAAaO,kCACXC,UAA2C,CAAC,MACzC;CACH,MAAMF,aAAa,IAAIL,kBAAkB;CACzC,IAAIQ;CAEJ,MAAMC,oBAA8C;EAClD,OAAOJ,WAAWI,YAAY;CAChC;CAEA,MAAMC,iBAAiBC,cAAsB;EAC3C,IAAI;GACF,OAAON,WAAWO,WAAWD,SAAS;EACxC,QAAQ;GACN;EACF;CACF;CAEA,MAAMG,kBACJH,cACyC;EACzC,IAAI;GACF,OAAON,WAAWI,YAAYE,SAAS;EACzC,QAAQ;GACN;EACF;CACF;CAEA,MAAMI,cAAcJ,cAA+B;EACjD,OAAOD,cAAcC,SAAS,MAAME,KAAAA;CACtC;CAEA,MAAMG,sBACJC,UACAC,YACS;EACTb,WAAWW,mBAAmBC,UAAUC,OAAO;EAC/CV,qBAAqBK,KAAAA;CACvB;CAEA,MAAMM,iBACJR,WACAS,kBACS;EACTf,WAAWc,cAAcR,WAAWS,aAAa;EACjDZ,qBAAqBK,KAAAA;CACvB;CAEA,MAAMQ,oBAAoBV,cAA+B;EACvD,IAAI,CAACI,WAAWJ,SAAS,GAAG,OAAO;EACnCQ,cAAcR,SAAS;EACvB,OAAO;CACT;CAEA,MAAMW,kBAAkBX,cAA4B;EAClDN,WAAWiB,eAAeX,SAAS;EACnCH,qBAAqBK,KAAAA;CACvB;CAEA,MAAMU,aAAaZ,cAAmC;EACpDN,WAAWkB,UAAUZ,SAAS;EAC9BH,qBAAqBK,KAAAA;CACvB;CAEA,MAAMW,cAAoB;EACxBnB,WAAWmB,MAAM;EACjBhB,qBAAqBK,KAAAA;CACvB;CAEA,MAAMY,iBACJd,WACAe,YACY;EACZ,MAAMC,OAAOjB,cAAcC,SAAS;EACpC,IAAI,CAACgB,MAAM,OAAO;EAClB,MAAMT,UAAUQ,QAAQC,KAAKT,OAAO;EACpC,IAAIA,YAAYS,KAAKT,SAAS,OAAO;EACrCF,mBAAmBW,KAAKV,UAAUC,OAAO;EACzC,OAAO;CACT;CAEA,MAAMU,kCACJC,WACS;EACT,MAAMC,SAASD,OAAOC,UAAUD,OAAOE,SAASC,GAAG,EAAE,CAAC,EAAEd,QAAQe,MAAM;EACtE,MAAMC,sBAAM,IAAIC,IAAY;EAC5B,IAAIC,aAAa;EACjB,KAAK,MAAM,EAAElB,aAAaW,OAAOE,UAAU;GACzC,IAAIG,IAAIG,IAAInB,QAAQe,EAAE,GAAG;IACvBG,aAAa;IACb;GACF;GACAF,IAAII,IAAIpB,QAAQe,EAAE;EACpB;EACA,IAAIH,WAAW,QAAQ,CAACI,IAAIG,IAAIP,MAAM,GAAGM,aAAa;EAEtD,IAAI,CAACA,YAAY;GACfZ,MAAM;GACN,IAAIe,UAAU,CAAC,GAAGV,OAAOE,QAAQ;GACjC,MAAMS,8BAAc,IAAIL,IAAY;GAEpC,OAAOI,QAAQE,SAAS,GAAG;IACzB,MAAMC,aAA6B,CAAA;IACnC,IAAIC,aAAa;IACjB,KAAK,MAAMhB,QAAQY,SAAS;KAC1B,IAAIZ,KAAKV,aAAa,QAAQ,CAACuB,YAAYH,IAAIV,KAAKV,QAAQ,GAAG;MAC7DyB,WAAWE,KAAKjB,IAAI;MACpB;KACF;KACAX,mBAAmBW,KAAKV,UAAUU,KAAKT,OAAO;KAC9CsB,YAAYF,IAAIX,KAAKT,QAAQe,EAAE;KAC/BU,aAAa;IACf;IACA,IAAI,CAACA,YAAY;KACfP,aAAa;KACb;IACF;IACAG,UAAUG;GACZ;EACF;EAEA,IAAIN,YAAY;GACdZ,MAAM;GACN,IAAIqB,aAA4B;GAChC,KAAK,MAAM,EAAE3B,aAAaW,OAAOE,UAAU;IACzC,MAAMe,WAAWpC,cAAcQ,QAAQe,EAAE;IACzCjB,mBAAmB8B,WAAWA,SAAS7B,WAAW4B,YAAY3B,OAAO;IACrE2B,aAAa3B,QAAQe;GACvB;GACAV,UAAUsB,UAAU;EACtB,OACEtB,UAAUO,MAAM;CAEpB;CAEA,OAAO;EACLrB;EACA,IAAIqB,SAAS;GACX,OAAOzB,WAAWyB;EACpB;EACAiB,cAAyC;GACvCvC,uBAAuBD,QAAQJ,iBAC3BI,QAAQJ,eAAeE,WAAW0C,OAAO,GAAG1C,UAAU,IACtDA,WAAW0C,OAAO;GACtB,OAAOvC;EACT;EACAE;EACAI;EACAC;EACAC;EACAG;EACAE;EACAC;EACAC;EACAC;EACAC;EACAG;CACF;AACF"}

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":";;;KAOY;EACV,SAAS;EACT;EACA,YAAY;;KAGF;EACV;EACA,UAAU;IACR,SAAS;IACT;IACA,YAAY;;;cAIH;EAEC,YAAA,mBAAS,wBAClB;EAkBM,sBAAA;IACL,SAAS;IACT;OACC;IACS;QACX;;cA6DQ;UACH;UACA;UACA;UAKA;UAUA;UA0DA;MAQJ;MAIA;EAUJ,YAAY,2BAAe;EAuB3B,mBAAmB,yBAAyB,SAAS;EAiCrD,WAAW;;;;;EAcX,cAAc,mBAAmB;EAsCjC,YAAY;;;;;;;;;UAmBJ;EA+BR,eAAe;EAkBf,UAAU;EAgDV;EAUA,UAAU;EAwBV,SAAS,QAAQ,YAAY"}
+{"version":3,"file":"message-repository.d.ts","names":[],"sources":["../../../src/runtime/utils/message-repository.ts"],"mappings":";;;KAOY;EACV,SAAS;EACT;EACA,YAAY;;KAGF;EACV;EACA,UAAU;IACR,SAAS;IACT;IACA,YAAY;;;cAIH;EAEC,YAAA,mBAAS,wBAClB;EAkBM,sBAAA;IACL,SAAS;IACT;OACC;IACS;QACX;;cAgEQ;UACH;UACA;UACA;UAKA;UAgBA;UA2DA;MAQJ;MAIA;EAUJ,YAAY,2BAAe;EAuB3B,mBAAmB,yBAAyB,SAAS;EAiCrD,WAAW;;;;;EAcX,cAAc,mBAAmB;EAsCjC,YAAY;;;;;;;;;UAmBJ;EA+BR,eAAe;EAkBf,UAAU;EAiDV;EAUA,UAAU;EAwBV,SAAS,QAAQ,YAAY"}

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

@@ -1,33 +1,32 @@
 import { generateId } from "../../utils/id.js";
 import { fromThreadMessageLike } from "./thread-message-like.js";
-import { getAutoStatus } from "./auto-status.js";
+import { getContentAutoStatus } from "./auto-status.js";
 //#region src/runtime/utils/message-repository.ts
 const ExportedMessageRepository = {
 	fromArray: (messages) => {
-		const conv = messages.map((m) => fromThreadMessageLike(m, generateId(), getAutoStatus(false, false, false, false, void 0)));
+		const conv = messages.map((m) => fromThreadMessageLike(m, generateId(), getContentAutoStatus(m.content, false, false)));
 		return { messages: conv.map((m, idx) => ({
 			parentId: idx > 0 ? conv[idx - 1].id : null,
 			message: m
 		})) };
 	},
 	fromBranchableArray: (items, options) => {
-		const fallbackStatus = getAutoStatus(false, false, false, false, void 0);
 		return {
 			...options?.headId !== void 0 ? { headId: options.headId } : void 0,
 			messages: items.map(({ message, parentId }) => {
 				if (!message.id) throw new Error("ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.");
 				return {
 					parentId,
-					message: fromThreadMessageLike(message, message.id, fallbackStatus)
+					message: fromThreadMessageLike(message, message.id, getContentAutoStatus(message.content, false, false))
 				};
 			})
 		};
 	}
 };
 const findHead = (message) => {
-	if (message.next) return findHead(message.next);
-	if ("current" in message) return message;
-	return null;
+	let current = message;
+	while (current.next) current = current.next;
+	return "current" in current ? current : null;
 };
 var CachedValue = class {
 	_value = null;
@@ -51,18 +50,28 @@ var MessageRepository = class {
 		next: null
 	};
 	updateLevels(message, newLevel) {
-		message.level = newLevel;
-		for (const childId of message.children) {
-			const childMessage = this.messages.get(childId);
-			if (childMessage) this.updateLevels(childMessage, newLevel + 1);
+		const pending = [{
+			message,
+			level: newLevel
+		}];
+		while (pending.length > 0) {
+			const current = pending.pop();
+			current.message.level = current.level;
+			for (const childId of current.message.children) {
+				const childMessage = this.messages.get(childId);
+				if (childMessage) pending.push({
+					message: childMessage,
+					level: current.level + 1
+				});
+			}
 		}
 	}
 	performOp(newParent, child, operation) {
 		const parentOrRoot = child.prev ?? this.root;
 		const newParentOrRoot = newParent ?? this.root;
 		if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
-		if (operation !== "cut") {
-			for (let current = newParent; current; current = current.prev) if (current.current.id === child.current.id) throw new Error("MessageRepository(performOp/link): 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.");
+		if (operation === "relink") {
+			for (let current = newParent; current; current = current.prev) if (current.current.id === child.current.id) throw new Error("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.");
 		}
 		if (operation !== "link") {
 			parentOrRoot.children = parentOrRoot.children.filter((m) => m !== child.current.id);
@@ -192,16 +201,15 @@ var MessageRepository = class {
 		if (!message) throw new Error("MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui.");
 		const previousHead = this.head;
 		if (message.children.length > 0) {
-			const deleteDescendants = (msg) => {
-				for (const childId of msg.children) {
-					const childMessage = this.messages.get(childId);
-					if (childMessage) {
-						deleteDescendants(childMessage);
-						this.messages.delete(childId);
-					}
+			const pending = [...message.children];
+			while (pending.length > 0) {
+				const childId = pending.pop();
+				const childMessage = this.messages.get(childId);
+				if (childMessage) {
+					for (const descendantId of childMessage.children) pending.push(descendantId);
+					this.messages.delete(childId);
 				}
-			};
-			deleteDescendants(message);
+			}
 			message.children = [];
 			message.next = null;
 		}