@xinghunm/ai-chat 1.4.2 → 1.4.4

package/dist/index.d.mts

@@ -738,6 +738,8 @@ interface ChatState {
     sessionMessageLoadStatusBySession: Record<string, ChatSessionMessageLoadStatus>;
     sessionMessageLoadErrorBySession: Record<string, string | null>;
     historyMessagePaginationBySession: Record<string, ChatHistoryMessagePaginationState>;
+    timelineAnchorStateBySession: Record<string, Record<string, ChatTimelineAnchorStateSnapshot>>;
+    newlyCreatedSessionIds: string[];
 }
 interface ChatHistoryMessagePaginationState {
     previousCursor: string | null;
@@ -745,6 +747,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 +765,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,8 @@ interface ChatState {
     sessionMessageLoadStatusBySession: Record<string, ChatSessionMessageLoadStatus>;
     sessionMessageLoadErrorBySession: Record<string, string | null>;
     historyMessagePaginationBySession: Record<string, ChatHistoryMessagePaginationState>;
+    timelineAnchorStateBySession: Record<string, Record<string, ChatTimelineAnchorStateSnapshot>>;
+    newlyCreatedSessionIds: string[];
 }
 interface ChatHistoryMessagePaginationState {
     previousCursor: string | null;
@@ -745,6 +747,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 +765,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,8 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
   sessionMessageLoadStatusBySession: {},
   sessionMessageLoadErrorBySession: {},
   historyMessagePaginationBySession: {},
+  timelineAnchorStateBySession: {},
+  newlyCreatedSessionIds: [],
   // ---- Session management ------------------------------------------------
   createSession: (session) => {
     const state = get();
@@ -267,6 +281,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 +311,8 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession
     });
   },
   startNewChat: () => {
@@ -364,11 +382,19 @@ 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;
+    const nextNewlyCreatedSessionIds = state.newlyCreatedSessionIds.includes(nextSessionId) ? state.newlyCreatedSessionIds : [...state.newlyCreatedSessionIds.filter((id) => id !== previousSessionId), nextSessionId];
     set({
       sessions: nextSessions,
       messagesBySession: nextMessagesBySession,
@@ -379,7 +405,9 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
       historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
-      activeSessionId: nextActiveSessionId
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession,
+      activeSessionId: nextActiveSessionId,
+      newlyCreatedSessionIds: nextNewlyCreatedSessionIds
     });
   },
   setPreferredMode: (mode) => {
@@ -396,15 +424,28 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
     const hasActiveInIncoming = sessions.some(
       (session) => session.sessionId === state.activeSessionId
     );
-    const keepActiveSession = activePersistedSession && !hasActiveInIncoming ? [activePersistedSession] : [];
+    let prependActiveSession = [];
+    let appendActiveSession = [];
+    if (activePersistedSession && !hasActiveInIncoming) {
+      const isNewlyCreated = state.newlyCreatedSessionIds.includes(
+        activePersistedSession.sessionId
+      );
+      if (isNewlyCreated) {
+        prependActiveSession = [activePersistedSession];
+      } else {
+        appendActiveSession = [activePersistedSession];
+      }
+    }
     const excludeIds = /* @__PURE__ */ new Set([
       ...draftSessions.map((s) => s.sessionId),
-      ...keepActiveSession.map((s) => s.sessionId)
+      ...prependActiveSession.map((s) => s.sessionId),
+      ...appendActiveSession.map((s) => s.sessionId)
     ]);
     const nextSessions = [
       ...draftSessions,
+      ...prependActiveSession,
       ...sessions.filter((session) => !excludeIds.has(session.sessionId)),
-      ...keepActiveSession
+      ...appendActiveSession
     ];
     const nextMessagesBySession = { ...state.messagesBySession };
     const nextErrorBySession = { ...state.errorBySession };
@@ -419,6 +460,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)
@@ -437,6 +481,10 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
         nextHistoryMessagePaginationBySession[sid] = createHistoryMessagePaginationState();
       }
     });
+    const incomingSessionIds = new Set(sessions.map((s) => s.sessionId));
+    const nextNewlyCreatedSessionIds = state.newlyCreatedSessionIds.filter(
+      (id) => !incomingSessionIds.has(id)
+    );
     set({
       sessions: nextSessions,
       messagesBySession: nextMessagesBySession,
@@ -445,7 +493,9 @@ var createChatStore = (initialState) => (0, import_vanilla.createStore)((set, ge
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession,
+      newlyCreatedSessionIds: nextNewlyCreatedSessionIds
     });
   },
   // ---- Message operations ------------------------------------------------
@@ -560,6 +610,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 +1617,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 +1657,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 +1726,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 +1743,8 @@ var useTimelineBlockAnchors = ({
     timelineAnchorReducer,
     {
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     },
     createInitialTimelineAnchorState
   );
@@ -1680,9 +1779,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 +1809,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 +3136,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 +3226,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,8 @@ var createChatStore = (initialState) => createStore((set, get) => ({
   sessionMessageLoadStatusBySession: {},
   sessionMessageLoadErrorBySession: {},
   historyMessagePaginationBySession: {},
+  timelineAnchorStateBySession: {},
+  newlyCreatedSessionIds: [],
   // ---- Session management ------------------------------------------------
   createSession: (session) => {
     const state = get();
@@ -220,6 +234,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 +264,8 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession
     });
   },
   startNewChat: () => {
@@ -317,11 +335,19 @@ 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;
+    const nextNewlyCreatedSessionIds = state.newlyCreatedSessionIds.includes(nextSessionId) ? state.newlyCreatedSessionIds : [...state.newlyCreatedSessionIds.filter((id) => id !== previousSessionId), nextSessionId];
     set({
       sessions: nextSessions,
       messagesBySession: nextMessagesBySession,
@@ -332,7 +358,9 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
       historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
-      activeSessionId: nextActiveSessionId
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession,
+      activeSessionId: nextActiveSessionId,
+      newlyCreatedSessionIds: nextNewlyCreatedSessionIds
     });
   },
   setPreferredMode: (mode) => {
@@ -349,15 +377,28 @@ var createChatStore = (initialState) => createStore((set, get) => ({
     const hasActiveInIncoming = sessions.some(
       (session) => session.sessionId === state.activeSessionId
     );
-    const keepActiveSession = activePersistedSession && !hasActiveInIncoming ? [activePersistedSession] : [];
+    let prependActiveSession = [];
+    let appendActiveSession = [];
+    if (activePersistedSession && !hasActiveInIncoming) {
+      const isNewlyCreated = state.newlyCreatedSessionIds.includes(
+        activePersistedSession.sessionId
+      );
+      if (isNewlyCreated) {
+        prependActiveSession = [activePersistedSession];
+      } else {
+        appendActiveSession = [activePersistedSession];
+      }
+    }
     const excludeIds = /* @__PURE__ */ new Set([
       ...draftSessions.map((s) => s.sessionId),
-      ...keepActiveSession.map((s) => s.sessionId)
+      ...prependActiveSession.map((s) => s.sessionId),
+      ...appendActiveSession.map((s) => s.sessionId)
     ]);
     const nextSessions = [
       ...draftSessions,
+      ...prependActiveSession,
       ...sessions.filter((session) => !excludeIds.has(session.sessionId)),
-      ...keepActiveSession
+      ...appendActiveSession
     ];
     const nextMessagesBySession = { ...state.messagesBySession };
     const nextErrorBySession = { ...state.errorBySession };
@@ -372,6 +413,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)
@@ -390,6 +434,10 @@ var createChatStore = (initialState) => createStore((set, get) => ({
         nextHistoryMessagePaginationBySession[sid] = createHistoryMessagePaginationState();
       }
     });
+    const incomingSessionIds = new Set(sessions.map((s) => s.sessionId));
+    const nextNewlyCreatedSessionIds = state.newlyCreatedSessionIds.filter(
+      (id) => !incomingSessionIds.has(id)
+    );
     set({
       sessions: nextSessions,
       messagesBySession: nextMessagesBySession,
@@ -398,7 +446,9 @@ var createChatStore = (initialState) => createStore((set, get) => ({
       isStoppingBySession: nextIsStoppingBySession,
       sessionMessageLoadStatusBySession: nextSessionMessageLoadStatusBySession,
       sessionMessageLoadErrorBySession: nextSessionMessageLoadErrorBySession,
-      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession
+      historyMessagePaginationBySession: nextHistoryMessagePaginationBySession,
+      timelineAnchorStateBySession: nextTimelineAnchorStateBySession,
+      newlyCreatedSessionIds: nextNewlyCreatedSessionIds
     });
   },
   // ---- Message operations ------------------------------------------------
@@ -513,6 +563,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 +1570,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 +1610,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 +1679,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 +1696,8 @@ var useTimelineBlockAnchors = ({
     timelineAnchorReducer,
     {
       messageId: message.id,
-      currentBlockKeys: currentTimelineBlockKeys
+      currentBlockKeys: currentTimelineBlockKeys,
+      persistedState
     },
     createInitialTimelineAnchorState
   );
@@ -1633,9 +1732,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 +1762,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 +3093,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 +3183,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.4",
   "description": "AI chat React component library",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",