@xinghunm/ai-chat 1.4.1 → 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
     });
   },
@@ -387,13 +412,24 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
   },
   hydrateHistorySessions: (sessions) => {
     const state = get();
-    const localSessions = state.sessions.filter(
+    const draftSessions = state.sessions.filter(
       (session) => isDraftChatSessionId(session.sessionId)
     );
-    const localSessionIds = new Set(localSessions.map((session) => session.sessionId));
+    const activePersistedSession = state.sessions.find(
+      (session) => session.sessionId === state.activeSessionId && !isDraftChatSessionId(session.sessionId)
+    );
+    const hasActiveInIncoming = sessions.some(
+      (session) => session.sessionId === state.activeSessionId
+    );
+    const keepActiveSession = activePersistedSession && !hasActiveInIncoming ? [activePersistedSession] : [];
+    const excludeIds = /* @__PURE__ */ new Set([
+      ...draftSessions.map((s) => s.sessionId),
+      ...keepActiveSession.map((s) => s.sessionId)
+    ]);
     const nextSessions = [
-      ...localSessions,
-      ...sessions.filter((session) => !localSessionIds.has(session.sessionId))
+      ...draftSessions,
+      ...sessions.filter((session) => !excludeIds.has(session.sessionId)),
+      ...keepActiveSession
     ];
     const nextMessagesBySession = { ...state.messagesBySession };
     const nextErrorBySession = { ...state.errorBySession };
@@ -408,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)
@@ -434,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 ------------------------------------------------
@@ -549,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({
@@ -1539,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) {
@@ -1553,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(
@@ -1622,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]
@@ -1634,7 +1722,8 @@ var useTimelineBlockAnchors = ({
     timelineAnchorReducer,
     {
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     },
     createInitialTimelineAnchorState
   );
@@ -1669,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;
@@ -1698,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 : {}
@@ -3007,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;
 };
@@ -3097,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",
@@ -9071,19 +9179,7 @@ var ChatConversationList = () => {
     isLoadingMoreRef.current = false;
     hasSeenLoadingMoreRef.current = false;
   }, [historySessionList?.sessions.length, historySessionList?.hasMore]);
-  const sessions = (0, import_react20.useMemo)(() => {
-    if (!historySessionList) {
-      return localSessions;
-    }
-    const persistedLocalSessions = localSessions.filter(
-      (session) => !isDraftChatSessionId(session.sessionId)
-    );
-    const localSessionIds = new Set(persistedLocalSessions.map((session) => session.sessionId));
-    return [
-      ...persistedLocalSessions,
-      ...historySessionList.sessions.filter((session) => !localSessionIds.has(session.sessionId))
-    ];
-  }, [historySessionList, localSessions]);
+  const sessions = localSessions;
   const handleSessionListScroll = (event) => {
     if (!historySessionList?.hasMore || historySessionList.isLoading || !onLoadMoreSessions || isLoadingMoreRef.current) {
       return;

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
     });
   },
@@ -340,13 +365,24 @@ var createChatStore = (initialState) => createStore((set, get) => ({
   },
   hydrateHistorySessions: (sessions) => {
     const state = get();
-    const localSessions = state.sessions.filter(
+    const draftSessions = state.sessions.filter(
       (session) => isDraftChatSessionId(session.sessionId)
     );
-    const localSessionIds = new Set(localSessions.map((session) => session.sessionId));
+    const activePersistedSession = state.sessions.find(
+      (session) => session.sessionId === state.activeSessionId && !isDraftChatSessionId(session.sessionId)
+    );
+    const hasActiveInIncoming = sessions.some(
+      (session) => session.sessionId === state.activeSessionId
+    );
+    const keepActiveSession = activePersistedSession && !hasActiveInIncoming ? [activePersistedSession] : [];
+    const excludeIds = /* @__PURE__ */ new Set([
+      ...draftSessions.map((s) => s.sessionId),
+      ...keepActiveSession.map((s) => s.sessionId)
+    ]);
     const nextSessions = [
-      ...localSessions,
-      ...sessions.filter((session) => !localSessionIds.has(session.sessionId))
+      ...draftSessions,
+      ...sessions.filter((session) => !excludeIds.has(session.sessionId)),
+      ...keepActiveSession
     ];
     const nextMessagesBySession = { ...state.messagesBySession };
     const nextErrorBySession = { ...state.errorBySession };
@@ -361,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)
@@ -387,7 +426,8 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession
     });
   },
   // ---- Message operations ------------------------------------------------
@@ -502,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({
@@ -1492,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) {
@@ -1506,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(
@@ -1575,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]
@@ -1587,7 +1675,8 @@ var useTimelineBlockAnchors = ({
     timelineAnchorReducer,
     {
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     },
     createInitialTimelineAnchorState
   );
@@ -1622,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;
@@ -1651,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 : {}
@@ -2964,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;
 };
@@ -3054,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",
@@ -8906,7 +9014,7 @@ var SkillButton = styled14.button`
 `;
 
 // src/components/chat-conversation-list/index.tsx
-import { useEffect as useEffect10, useMemo as useMemo7, useRef as useRef12 } from "react";
+import { useEffect as useEffect10, useRef as useRef12 } from "react";
 import styled16 from "@emotion/styled";
 
 // src/components/chat-conversation-list/components/chat-session-item.tsx
@@ -9028,19 +9136,7 @@ var ChatConversationList = () => {
     isLoadingMoreRef.current = false;
     hasSeenLoadingMoreRef.current = false;
   }, [historySessionList?.sessions.length, historySessionList?.hasMore]);
-  const sessions = useMemo7(() => {
-    if (!historySessionList) {
-      return localSessions;
-    }
-    const persistedLocalSessions = localSessions.filter(
-      (session) => !isDraftChatSessionId(session.sessionId)
-    );
-    const localSessionIds = new Set(persistedLocalSessions.map((session) => session.sessionId));
-    return [
-      ...persistedLocalSessions,
-      ...historySessionList.sessions.filter((session) => !localSessionIds.has(session.sessionId))
-    ];
-  }, [historySessionList, localSessions]);
+  const sessions = localSessions;
   const handleSessionListScroll = (event) => {
     if (!historySessionList?.hasMore || historySessionList.isLoading || !onLoadMoreSessions || isLoadingMoreRef.current) {
       return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xinghunm/ai-chat",
-  "version": "1.4.1",
+  "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",