Function repairAssistantMessage

demo/chat/session.ts:20–39

(message: ChatMessage)

const STORAGE_KEY = "htmlstream-chat-session-v1";

function repairAssistantMessage(message: ChatMessage): ChatMessage {
  if (message.role !== "assistant") {
    return message;
  }

  const content = typeof message.content === "string" ? message.content : "";
  const reasoning =
    typeof message.reasoning === "string" ? message.reasoning : "";

  if (!content.includes('"kind":"')) {
    return { ...message, content, reasoning: reasoning || undefined };
  }

  const repaired = splitStoredStreamText(content);
  return {
    ...message,
    reasoning: reasoning || repaired.reasoning || undefined,
    content: repaired.content,
  };
}

export function repairSession(session: ChatSession): ChatSession {
  return {