@xinghunm/ai-chat 1.4.2 → 1.4.3

package/dist/index.d.mts

@@ -738,6 +738,7 @@ interface ChatState {
     sessionMessageLoadStatusBySession: Record<string, ChatSessionMessageLoadStatus>;
     sessionMessageLoadErrorBySession: Record<string, string | null>;
     historyMessagePaginationBySession: Record<string, ChatHistoryMessagePaginationState>;
+    timelineAnchorStateBySession: Record<string, Record<string, ChatTimelineAnchorStateSnapshot>>;
 }
 interface ChatHistoryMessagePaginationState {
     previousCursor: string | null;
@@ -745,6 +746,11 @@ interface ChatHistoryMessagePaginationState {
     isLoadingPrevious: boolean;
     error: string | null;
 }
+interface ChatTimelineAnchorStateSnapshot {
+    previousBlockKeys: string[];
+    timelineBlockAnchors: Record<string, number>;
+    visibleTimelineBlockKeys: Record<string, true>;
+}
 interface ChatActions {
     createSession: (session: ChatSession) => void;
     startNewChat: () => void;
@@ -758,6 +764,7 @@ interface ChatActions {
     prependHistorySessionMessagesPage: (sessionId: string, page: ChatHistorySessionMessagesPage) => void;
     setHistorySessionPreviousMessagesLoadStatus: (sessionId: string, isLoading: boolean, error?: string | null) => void;
     setHistorySessionMessageLoadStatus: (sessionId: string, status: ChatSessionMessageLoadStatus, error?: string | null) => void;
+    setTimelineAnchorState: (sessionId: string, messageId: string, snapshot: ChatTimelineAnchorStateSnapshot) => void;
     startStreamingMessage: (sessionId: string, message: ChatMessage) => void;
     updateStreamingMessage: (sessionId: string, content: string) => void;
     patchStreamingMessage: (sessionId: string, patch: ChatStreamMessagePatch) => void;

package/dist/index.d.ts

@@ -738,6 +738,7 @@ interface ChatState {
     sessionMessageLoadStatusBySession: Record<string, ChatSessionMessageLoadStatus>;
     sessionMessageLoadErrorBySession: Record<string, string | null>;
     historyMessagePaginationBySession: Record<string, ChatHistoryMessagePaginationState>;
+    timelineAnchorStateBySession: Record<string, Record<string, ChatTimelineAnchorStateSnapshot>>;
 }
 interface ChatHistoryMessagePaginationState {
     previousCursor: string | null;
@@ -745,6 +746,11 @@ interface ChatHistoryMessagePaginationState {
     isLoadingPrevious: boolean;
     error: string | null;
 }
+interface ChatTimelineAnchorStateSnapshot {
+    previousBlockKeys: string[];
+    timelineBlockAnchors: Record<string, number>;
+    visibleTimelineBlockKeys: Record<string, true>;
+}
 interface ChatActions {
     createSession: (session: ChatSession) => void;
     startNewChat: () => void;
@@ -758,6 +764,7 @@ interface ChatActions {
     prependHistorySessionMessagesPage: (sessionId: string, page: ChatHistorySessionMessagesPage) => void;
     setHistorySessionPreviousMessagesLoadStatus: (sessionId: string, isLoading: boolean, error?: string | null) => void;
     setHistorySessionMessageLoadStatus: (sessionId: string, status: ChatSessionMessageLoadStatus, error?: string | null) => void;
+    setTimelineAnchorState: (sessionId: string, messageId: string, snapshot: ChatTimelineAnchorStateSnapshot) => void;
     startStreamingMessage: (sessionId: string, message: ChatMessage) => void;
     updateStreamingMessage: (sessionId: string, content: string) => void;
     patchStreamingMessage: (sessionId: string, patch: ChatStreamMessagePatch) => void;

package/dist/index.js

@@ -157,6 +157,18 @@ var resolveSessionTitleFromMessage = (message) => {
   }
   return DEFAULT_CHAT_SESSION_TITLE;
 };
+var areStringArraysEqual = (left, right) => left.length === right.length && left.every((value, index3) => value === right[index3]);
+var areNumberRecordsEqual = (left, right) => {
+  const leftKeys = Object.keys(left);
+  const rightKeys = Object.keys(right);
+  return leftKeys.length === rightKeys.length && leftKeys.every((key) => right[key] === left[key]);
+};
+var areTrueRecordsEqual = (left, right) => {
+  const leftKeys = Object.keys(left);
+  const rightKeys = Object.keys(right);
+  return leftKeys.length === rightKeys.length && leftKeys.every((key) => right[key] === true);
+};
+var isSameTimelineAnchorSnapshot = (current, next) => current !== void 0 && areStringArraysEqual(current.previousBlockKeys, next.previousBlockKeys) && areNumberRecordsEqual(current.timelineBlockAnchors, next.timelineBlockAnchors) && areTrueRecordsEqual(current.visibleTimelineBlockKeys, next.visibleTimelineBlockKeys);
 var mergeStreamingBlocks = (existingBlocks, incomingBlocks) => {
   const nextBlocks = [...existingBlocks ?? []];
   incomingBlocks.forEach((incomingBlock) => {
@@ -249,6 +261,7 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
   sessionMessageLoadStatusBySession: {},
   sessionMessageLoadErrorBySession: {},
   historyMessagePaginationBySession: {},
+  timelineAnchorStateBySession: {},
   // ---- Session management ------------------------------------------------
   createSession: (session) => {
     const state = get();
@@ -267,6 +280,9 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
     const nextHistoryMessagePaginationBySession = {
       ...state.historyMessagePaginationBySession
     };
+    const nextTimelineAnchorStateBySession = {
+      ...state.timelineAnchorStateBySession
+    };
     const sid = session.sessionId;
     if (nextMessagesBySession[sid] === void 0)
       nextMessagesBySession[sid] = [];
@@ -294,7 +310,8 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession
     });
   },
   startNewChat: () => {
@@ -364,10 +381,17 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
     const nextHistoryMessagePaginationBySession = {
       ...state.historyMessagePaginationBySession
     };
+    const nextTimelineAnchorStateBySession = {
+      ...state.timelineAnchorStateBySession
+    };
     if (previousSessionId in nextHistoryMessagePaginationBySession) {
       nextHistoryMessagePaginationBySession[nextSessionId] = nextHistoryMessagePaginationBySession[previousSessionId] ?? createHistoryMessagePaginationState();
       delete nextHistoryMessagePaginationBySession[previousSessionId];
     }
+    if (previousSessionId in nextTimelineAnchorStateBySession) {
+      nextTimelineAnchorStateBySession[nextSessionId] = nextTimelineAnchorStateBySession[previousSessionId] ?? {};
+      delete nextTimelineAnchorStateBySession[previousSessionId];
+    }
     const nextActiveSessionId = state.activeSessionId === previousSessionId ? nextSessionId : state.activeSessionId;
     set({
       sessions: nextSessions,
@@ -379,6 +403,7 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
       historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession,
       activeSessionId: nextActiveSessionId
     });
   },
@@ -419,6 +444,9 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
     const nextHistoryMessagePaginationBySession = {
       ...state.historyMessagePaginationBySession
     };
+    const nextTimelineAnchorStateBySession = {
+      ...state.timelineAnchorStateBySession
+    };
     nextSessions.forEach((session) => {
       const sid = session.sessionId;
       if (nextErrorBySession[sid] === void 0)
@@ -445,7 +473,8 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession
     });
   },
   // ---- Message operations ------------------------------------------------
@@ -560,6 +589,23 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
       }
     });
   },
+  setTimelineAnchorState: (sessionId, messageId, snapshot) => {
+    const state = get();
+    const currentSessionSnapshots = state.timelineAnchorStateBySession[sessionId] ?? {};
+    const currentSnapshot = currentSessionSnapshots[messageId];
+    if (isSameTimelineAnchorSnapshot(currentSnapshot, snapshot)) {
+      return;
+    }
+    set({
+      timelineAnchorStateBySession: {
+        ...state.timelineAnchorStateBySession,
+        [sessionId]: {
+          ...currentSessionSnapshots,
+          [messageId]: snapshot
+        }
+      }
+    });
+  },
   startStreamingMessage: (sessionId, message) => {
     const state = get();
     set({
@@ -1550,13 +1596,39 @@ var createTimelineAnchorState = ({
   timelineBlockAnchors: {},
   visibleTimelineBlockKeys: {}
 });
+var restoreTimelineAnchorState = ({
+  messageId,
+  currentBlockKeys,
+  persistedState
+}) => {
+  if (!persistedState) {
+    return createTimelineAnchorState({ messageId, currentBlockKeys });
+  }
+  const currentBlockKeySet = new Set(currentBlockKeys);
+  return {
+    messageId,
+    previousBlockKeys: persistedState.previousBlockKeys.filter(
+      (blockKey) => currentBlockKeySet.has(blockKey)
+    ),
+    timelineBlockAnchors: Object.fromEntries(
+      Object.entries(persistedState.timelineBlockAnchors).filter(
+        ([blockKey]) => currentBlockKeySet.has(blockKey)
+      )
+    ),
+    visibleTimelineBlockKeys: Object.fromEntries(
+      Object.entries(persistedState.visibleTimelineBlockKeys).filter(
+        ([blockKey]) => currentBlockKeySet.has(blockKey)
+      )
+    )
+  };
+};
 var createInitialTimelineAnchorState = ({
-  messageId
-}) => ({
   messageId,
-  previousBlockKeys: [],
-  timelineBlockAnchors: {},
-  visibleTimelineBlockKeys: {}
+  currentBlockKeys,
+  persistedState
+}) => ({
+  ...restoreTimelineAnchorState({ messageId, currentBlockKeys, persistedState }),
+  previousBlockKeys: persistedState?.previousBlockKeys ?? []
 });
 var timelineAnchorReducer = (state, action) => {
   switch (action.type) {
@@ -1564,7 +1636,7 @@ var timelineAnchorReducer = (state, action) => {
       if (state.messageId === action.messageId) {
         return state;
       }
-      return createTimelineAnchorState(action);
+      return restoreTimelineAnchorState(action);
     case "sync-anchors": {
       const previousBlockKeys = new Set(state.previousBlockKeys);
       const nextAnchors = action.currentBlockKeys.reduce(
@@ -1633,10 +1705,15 @@ var useTimelineBlockAnchors = ({
   message,
   messageRenderOrder
 }) => {
+  const { store } = useChatContext();
   const currentTimelineBlockKeys = (0, import_react6.useMemo)(
     () => blocks.map((block, index3) => getTimelineBlockKey(block, index3)).filter((blockKey) => Boolean(blockKey)),
     [blocks]
   );
+  const persistedState = (0, import_react6.useMemo)(
+    () => store.getState().timelineAnchorStateBySession[message.sessionId]?.[message.id],
+    [message.id, message.sessionId, store]
+  );
   const timelineTextStreamLength = (0, import_react6.useMemo)(
     () => getTimelineDisplayUnitCount(getTimelineTextStream(message.content, blocks)),
     [blocks, message.content]
@@ -1645,7 +1722,8 @@ var useTimelineBlockAnchors = ({
     timelineAnchorReducer,
     {
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     },
     createInitialTimelineAnchorState
   );
@@ -1680,9 +1758,10 @@ var useTimelineBlockAnchors = ({
     dispatch({
       type: "reset-message",
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     });
-  }, [currentTimelineBlockKeys, message.id]);
+  }, [currentTimelineBlockKeys, message.id, persistedState]);
   (0, import_react6.useEffect)(() => {
     if (messageRenderOrder !== "timeline" || !isAssistantStreaming) {
       return;
@@ -1709,6 +1788,24 @@ var useTimelineBlockAnchors = ({
     effectiveTimelineBlockAnchors,
     messageRenderOrder
   ]);
+  (0, import_react6.useEffect)(() => {
+    if (messageRenderOrder !== "timeline") {
+      return;
+    }
+    store.getState().setTimelineAnchorState(message.sessionId, message.id, {
+      previousBlockKeys: state.previousBlockKeys,
+      timelineBlockAnchors: state.timelineBlockAnchors,
+      visibleTimelineBlockKeys: state.visibleTimelineBlockKeys
+    });
+  }, [
+    message.id,
+    message.sessionId,
+    messageRenderOrder,
+    state.previousBlockKeys,
+    state.timelineBlockAnchors,
+    state.visibleTimelineBlockKeys,
+    store
+  ]);
   return {
     timelineBlockAnchors: messageRenderOrder === "timeline" ? effectiveTimelineBlockAnchors : {},
     visibleTimelineBlockKeys: messageRenderOrder === "timeline" ? state.visibleTimelineBlockKeys : {}
@@ -3018,10 +3115,10 @@ var areExecutionProposalsEqual = (previousProposal, nextProposal) => previousPro
   (warning, index3) => warning === nextProposal.warnings?.[index3]
 );
 var areResultSummariesEqual = (previousSummary, nextSummary) => previousSummary.summaryId === nextSummary.summaryId && previousSummary.status === nextSummary.status && previousSummary.headline === nextSummary.headline && previousSummary.details.length === nextSummary.details.length && previousSummary.details.every((detail, index3) => detail === nextSummary.details[index3]);
-var areStringArraysEqual = (previousValues, nextValues) => previousValues.length === nextValues.length && previousValues.every((value, index3) => value === nextValues[index3]);
+var areStringArraysEqual2 = (previousValues, nextValues) => previousValues.length === nextValues.length && previousValues.every((value, index3) => value === nextValues[index3]);
 var areQuestionAnswersEqual = (previousAnswer, nextAnswer) => {
   if (Array.isArray(previousAnswer) || Array.isArray(nextAnswer)) {
-    return Array.isArray(previousAnswer) && Array.isArray(nextAnswer) && areStringArraysEqual(previousAnswer, nextAnswer);
+    return Array.isArray(previousAnswer) && Array.isArray(nextAnswer) && areStringArraysEqual2(previousAnswer, nextAnswer);
   }
   return previousAnswer === nextAnswer;
 };
@@ -3108,7 +3205,7 @@ var areMessageBlocksEqual = (previousBlocks, nextBlocks) => {
     }
   });
 };
-var isSameMessage = (previousMessage, nextMessage, previousMode, nextMode, previousConfirmationSubmit, nextConfirmationSubmit, previousQuestionnaireSubmit, nextQuestionnaireSubmit, previousRenderMessageBlock, nextRenderMessageBlock) => previousMessage.id === nextMessage.id && previousMessage.sessionId === nextMessage.sessionId && previousMessage.role === nextMessage.role && previousMessage.content === nextMessage.content && areStringArraysEqual(previousMessage.skills ?? [], nextMessage.skills ?? []) && areMessageBlocksEqual(previousMessage.blocks, nextMessage.blocks) && previousMessage.localOnly === nextMessage.localOnly && areChatAttachmentsEqual(previousMessage.attachments, nextMessage.attachments) && previousMessage.status === nextMessage.status && previousMessage.createdAt === nextMessage.createdAt && previousMode === nextMode && previousConfirmationSubmit === nextConfirmationSubmit && previousQuestionnaireSubmit === nextQuestionnaireSubmit && previousRenderMessageBlock === nextRenderMessageBlock;
+var isSameMessage = (previousMessage, nextMessage, previousMode, nextMode, previousConfirmationSubmit, nextConfirmationSubmit, previousQuestionnaireSubmit, nextQuestionnaireSubmit, previousRenderMessageBlock, nextRenderMessageBlock) => previousMessage.id === nextMessage.id && previousMessage.sessionId === nextMessage.sessionId && previousMessage.role === nextMessage.role && previousMessage.content === nextMessage.content && areStringArraysEqual2(previousMessage.skills ?? [], nextMessage.skills ?? []) && areMessageBlocksEqual(previousMessage.blocks, nextMessage.blocks) && previousMessage.localOnly === nextMessage.localOnly && areChatAttachmentsEqual(previousMessage.attachments, nextMessage.attachments) && previousMessage.status === nextMessage.status && previousMessage.createdAt === nextMessage.createdAt && previousMode === nextMode && previousConfirmationSubmit === nextConfirmationSubmit && previousQuestionnaireSubmit === nextQuestionnaireSubmit && previousRenderMessageBlock === nextRenderMessageBlock;
 var ChatMessageItemView = ({
   message,
   mode = "agent",

package/dist/index.mjs

@@ -110,6 +110,18 @@ var resolveSessionTitleFromMessage = (message) => {
   }
   return DEFAULT_CHAT_SESSION_TITLE;
 };
+var areStringArraysEqual = (left, right) => left.length === right.length && left.every((value, index3) => value === right[index3]);
+var areNumberRecordsEqual = (left, right) => {
+  const leftKeys = Object.keys(left);
+  const rightKeys = Object.keys(right);
+  return leftKeys.length === rightKeys.length && leftKeys.every((key) => right[key] === left[key]);
+};
+var areTrueRecordsEqual = (left, right) => {
+  const leftKeys = Object.keys(left);
+  const rightKeys = Object.keys(right);
+  return leftKeys.length === rightKeys.length && leftKeys.every((key) => right[key] === true);
+};
+var isSameTimelineAnchorSnapshot = (current, next) => current !== void 0 && areStringArraysEqual(current.previousBlockKeys, next.previousBlockKeys) && areNumberRecordsEqual(current.timelineBlockAnchors, next.timelineBlockAnchors) && areTrueRecordsEqual(current.visibleTimelineBlockKeys, next.visibleTimelineBlockKeys);
 var mergeStreamingBlocks = (existingBlocks, incomingBlocks) => {
   const nextBlocks = [...existingBlocks ?? []];
   incomingBlocks.forEach((incomingBlock) => {
@@ -202,6 +214,7 @@ var createChatStore = (initialState) => createStore((set, get) => ({
   sessionMessageLoadStatusBySession: {},
   sessionMessageLoadErrorBySession: {},
   historyMessagePaginationBySession: {},
+  timelineAnchorStateBySession: {},
   // ---- Session management ------------------------------------------------
   createSession: (session) => {
     const state = get();
@@ -220,6 +233,9 @@ var createChatStore = (initialState) => createStore((set, get) => ({
     const nextHistoryMessagePaginationBySession = {
       ...state.historyMessagePaginationBySession
     };
+    const nextTimelineAnchorStateBySession = {
+      ...state.timelineAnchorStateBySession
+    };
     const sid = session.sessionId;
     if (nextMessagesBySession[sid] === void 0)
       nextMessagesBySession[sid] = [];
@@ -247,7 +263,8 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession
     });
   },
   startNewChat: () => {
@@ -317,10 +334,17 @@ var createChatStore = (initialState) => createStore((set, get) => ({
     const nextHistoryMessagePaginationBySession = {
       ...state.historyMessagePaginationBySession
     };
+    const nextTimelineAnchorStateBySession = {
+      ...state.timelineAnchorStateBySession
+    };
     if (previousSessionId in nextHistoryMessagePaginationBySession) {
       nextHistoryMessagePaginationBySession[nextSessionId] = nextHistoryMessagePaginationBySession[previousSessionId] ?? createHistoryMessagePaginationState();
       delete nextHistoryMessagePaginationBySession[previousSessionId];
     }
+    if (previousSessionId in nextTimelineAnchorStateBySession) {
+      nextTimelineAnchorStateBySession[nextSessionId] = nextTimelineAnchorStateBySession[previousSessionId] ?? {};
+      delete nextTimelineAnchorStateBySession[previousSessionId];
+    }
     const nextActiveSessionId = state.activeSessionId === previousSessionId ? nextSessionId : state.activeSessionId;
     set({
       sessions: nextSessions,
@@ -332,6 +356,7 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
       historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession,
       activeSessionId: nextActiveSessionId
     });
   },
@@ -372,6 +397,9 @@ var createChatStore = (initialState) => createStore((set, get) => ({
     const nextHistoryMessagePaginationBySession = {
       ...state.historyMessagePaginationBySession
     };
+    const nextTimelineAnchorStateBySession = {
+      ...state.timelineAnchorStateBySession
+    };
     nextSessions.forEach((session) => {
       const sid = session.sessionId;
       if (nextErrorBySession[sid] === void 0)
@@ -398,7 +426,8 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession
     });
   },
   // ---- Message operations ------------------------------------------------
@@ -513,6 +542,23 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       }
     });
   },
+  setTimelineAnchorState: (sessionId, messageId, snapshot) => {
+    const state = get();
+    const currentSessionSnapshots = state.timelineAnchorStateBySession[sessionId] ?? {};
+    const currentSnapshot = currentSessionSnapshots[messageId];
+    if (isSameTimelineAnchorSnapshot(currentSnapshot, snapshot)) {
+      return;
+    }
+    set({
+      timelineAnchorStateBySession: {
+        ...state.timelineAnchorStateBySession,
+        [sessionId]: {
+          ...currentSessionSnapshots,
+          [messageId]: snapshot
+        }
+      }
+    });
+  },
   startStreamingMessage: (sessionId, message) => {
     const state = get();
     set({
@@ -1503,13 +1549,39 @@ var createTimelineAnchorState = ({
   timelineBlockAnchors: {},
   visibleTimelineBlockKeys: {}
 });
+var restoreTimelineAnchorState = ({
+  messageId,
+  currentBlockKeys,
+  persistedState
+}) => {
+  if (!persistedState) {
+    return createTimelineAnchorState({ messageId, currentBlockKeys });
+  }
+  const currentBlockKeySet = new Set(currentBlockKeys);
+  return {
+    messageId,
+    previousBlockKeys: persistedState.previousBlockKeys.filter(
+      (blockKey) => currentBlockKeySet.has(blockKey)
+    ),
+    timelineBlockAnchors: Object.fromEntries(
+      Object.entries(persistedState.timelineBlockAnchors).filter(
+        ([blockKey]) => currentBlockKeySet.has(blockKey)
+      )
+    ),
+    visibleTimelineBlockKeys: Object.fromEntries(
+      Object.entries(persistedState.visibleTimelineBlockKeys).filter(
+        ([blockKey]) => currentBlockKeySet.has(blockKey)
+      )
+    )
+  };
+};
 var createInitialTimelineAnchorState = ({
-  messageId
-}) => ({
   messageId,
-  previousBlockKeys: [],
-  timelineBlockAnchors: {},
-  visibleTimelineBlockKeys: {}
+  currentBlockKeys,
+  persistedState
+}) => ({
+  ...restoreTimelineAnchorState({ messageId, currentBlockKeys, persistedState }),
+  previousBlockKeys: persistedState?.previousBlockKeys ?? []
 });
 var timelineAnchorReducer = (state, action) => {
   switch (action.type) {
@@ -1517,7 +1589,7 @@ var timelineAnchorReducer = (state, action) => {
       if (state.messageId === action.messageId) {
         return state;
       }
-      return createTimelineAnchorState(action);
+      return restoreTimelineAnchorState(action);
     case "sync-anchors": {
       const previousBlockKeys = new Set(state.previousBlockKeys);
       const nextAnchors = action.currentBlockKeys.reduce(
@@ -1586,10 +1658,15 @@ var useTimelineBlockAnchors = ({
   message,
   messageRenderOrder
 }) => {
+  const { store } = useChatContext();
   const currentTimelineBlockKeys = useMemo3(
     () => blocks.map((block, index3) => getTimelineBlockKey(block, index3)).filter((blockKey) => Boolean(blockKey)),
     [blocks]
   );
+  const persistedState = useMemo3(
+    () => store.getState().timelineAnchorStateBySession[message.sessionId]?.[message.id],
+    [message.id, message.sessionId, store]
+  );
   const timelineTextStreamLength = useMemo3(
     () => getTimelineDisplayUnitCount(getTimelineTextStream(message.content, blocks)),
     [blocks, message.content]
@@ -1598,7 +1675,8 @@ var useTimelineBlockAnchors = ({
     timelineAnchorReducer,
     {
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     },
     createInitialTimelineAnchorState
   );
@@ -1633,9 +1711,10 @@ var useTimelineBlockAnchors = ({
     dispatch({
       type: "reset-message",
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     });
-  }, [currentTimelineBlockKeys, message.id]);
+  }, [currentTimelineBlockKeys, message.id, persistedState]);
   useEffect3(() => {
     if (messageRenderOrder !== "timeline" || !isAssistantStreaming) {
       return;
@@ -1662,6 +1741,24 @@ var useTimelineBlockAnchors = ({
     effectiveTimelineBlockAnchors,
     messageRenderOrder
   ]);
+  useEffect3(() => {
+    if (messageRenderOrder !== "timeline") {
+      return;
+    }
+    store.getState().setTimelineAnchorState(message.sessionId, message.id, {
+      previousBlockKeys: state.previousBlockKeys,
+      timelineBlockAnchors: state.timelineBlockAnchors,
+      visibleTimelineBlockKeys: state.visibleTimelineBlockKeys
+    });
+  }, [
+    message.id,
+    message.sessionId,
+    messageRenderOrder,
+    state.previousBlockKeys,
+    state.timelineBlockAnchors,
+    state.visibleTimelineBlockKeys,
+    store
+  ]);
   return {
     timelineBlockAnchors: messageRenderOrder === "timeline" ? effectiveTimelineBlockAnchors : {},
     visibleTimelineBlockKeys: messageRenderOrder === "timeline" ? state.visibleTimelineBlockKeys : {}
@@ -2975,10 +3072,10 @@ var areExecutionProposalsEqual = (previousProposal, nextProposal) => previousPro
   (warning, index3) => warning === nextProposal.warnings?.[index3]
 );
 var areResultSummariesEqual = (previousSummary, nextSummary) => previousSummary.summaryId === nextSummary.summaryId && previousSummary.status === nextSummary.status && previousSummary.headline === nextSummary.headline && previousSummary.details.length === nextSummary.details.length && previousSummary.details.every((detail, index3) => detail === nextSummary.details[index3]);
-var areStringArraysEqual = (previousValues, nextValues) => previousValues.length === nextValues.length && previousValues.every((value, index3) => value === nextValues[index3]);
+var areStringArraysEqual2 = (previousValues, nextValues) => previousValues.length === nextValues.length && previousValues.every((value, index3) => value === nextValues[index3]);
 var areQuestionAnswersEqual = (previousAnswer, nextAnswer) => {
   if (Array.isArray(previousAnswer) || Array.isArray(nextAnswer)) {
-    return Array.isArray(previousAnswer) && Array.isArray(nextAnswer) && areStringArraysEqual(previousAnswer, nextAnswer);
+    return Array.isArray(previousAnswer) && Array.isArray(nextAnswer) && areStringArraysEqual2(previousAnswer, nextAnswer);
   }
   return previousAnswer === nextAnswer;
 };
@@ -3065,7 +3162,7 @@ var areMessageBlocksEqual = (previousBlocks, nextBlocks) => {
     }
   });
 };
-var isSameMessage = (previousMessage, nextMessage, previousMode, nextMode, previousConfirmationSubmit, nextConfirmationSubmit, previousQuestionnaireSubmit, nextQuestionnaireSubmit, previousRenderMessageBlock, nextRenderMessageBlock) => previousMessage.id === nextMessage.id && previousMessage.sessionId === nextMessage.sessionId && previousMessage.role === nextMessage.role && previousMessage.content === nextMessage.content && areStringArraysEqual(previousMessage.skills ?? [], nextMessage.skills ?? []) && areMessageBlocksEqual(previousMessage.blocks, nextMessage.blocks) && previousMessage.localOnly === nextMessage.localOnly && areChatAttachmentsEqual(previousMessage.attachments, nextMessage.attachments) && previousMessage.status === nextMessage.status && previousMessage.createdAt === nextMessage.createdAt && previousMode === nextMode && previousConfirmationSubmit === nextConfirmationSubmit && previousQuestionnaireSubmit === nextQuestionnaireSubmit && previousRenderMessageBlock === nextRenderMessageBlock;
+var isSameMessage = (previousMessage, nextMessage, previousMode, nextMode, previousConfirmationSubmit, nextConfirmationSubmit, previousQuestionnaireSubmit, nextQuestionnaireSubmit, previousRenderMessageBlock, nextRenderMessageBlock) => previousMessage.id === nextMessage.id && previousMessage.sessionId === nextMessage.sessionId && previousMessage.role === nextMessage.role && previousMessage.content === nextMessage.content && areStringArraysEqual2(previousMessage.skills ?? [], nextMessage.skills ?? []) && areMessageBlocksEqual(previousMessage.blocks, nextMessage.blocks) && previousMessage.localOnly === nextMessage.localOnly && areChatAttachmentsEqual(previousMessage.attachments, nextMessage.attachments) && previousMessage.status === nextMessage.status && previousMessage.createdAt === nextMessage.createdAt && previousMode === nextMode && previousConfirmationSubmit === nextConfirmationSubmit && previousQuestionnaireSubmit === nextQuestionnaireSubmit && previousRenderMessageBlock === nextRenderMessageBlock;
 var ChatMessageItemView = ({
   message,
   mode = "agent",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xinghunm/ai-chat",
-  "version": "1.4.2",
+  "version": "1.4.3",
   "description": "AI chat React component library",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -19,7 +19,7 @@
   "peerDependencies": {
     "@emotion/react": ">=11",
     "@emotion/styled": ">=11",
-    "@xinghunm/compass-ui": "0.8.3",
+    "@xinghunm/compass-ui": "0.9.0",
     "axios": ">=1.0",
     "react": ">=18",
     "react-dom": ">=18",