@lobehub/lobehub 2.0.0-next.35 → 2.0.0-next.36

package/src/store/chat/slices/message/selectors/displayMessage.ts

@@ -0,0 +1,301 @@
+import { UIChatMessage } from '@lobechat/types';
+
+import { DEFAULT_USER_AVATAR } from '@/const/meta';
+import { INBOX_SESSION_ID } from '@/const/session';
+import { useAgentStore } from '@/store/agent';
+import { agentChatConfigSelectors } from '@/store/agent/selectors';
+import { useSessionStore } from '@/store/session';
+import { sessionMetaSelectors } from '@/store/session/selectors';
+import { useUserStore } from '@/store/user';
+import { userProfileSelectors } from '@/store/user/selectors';
+
+import { chatHelpers } from '../../../helpers';
+import type { ChatStoreState } from '../../../initialState';
+import { messageMapKey } from '../../../utils/messageMapKey';
+
+/**
+ * Display Message Selectors
+ *
+ * These selectors access processed messages from messagesMap (parsed display data).
+ * Use these selectors when you need to:
+ * - Render messages in UI components
+ * - Display assistantGroup messages with children
+ * - Show messages with proper meta information
+ * - Present message history with filters
+ *
+ * DO NOT use these for data mutations - use dbMessage.ts selectors instead.
+ */
+
+// ============= Meta Information ========== //
+
+const getMeta = (message: UIChatMessage) => {
+  switch (message.role) {
+    case 'user': {
+      return {
+        avatar: userProfileSelectors.userAvatar(useUserStore.getState()) || DEFAULT_USER_AVATAR,
+      };
+    }
+
+    case 'system': {
+      return message.meta;
+    }
+
+    default: {
+      // For group chat, get meta from agent session
+      if (message.groupId && message.agentId) {
+        return sessionMetaSelectors.getAgentMetaByAgentId(message.agentId)(
+          useSessionStore.getState(),
+        );
+      }
+
+      // Otherwise, use the current session's agent meta for single agent chat
+      return sessionMetaSelectors.currentAgentMeta(useSessionStore.getState());
+    }
+  }
+};
+
+// ============= Basic Display Message Access ========== //
+
+/**
+ * Get the current chat key for accessing messagesMap
+ */
+export const currentDisplayChatKey = (s: ChatStoreState) =>
+  messageMapKey(s.activeId, s.activeTopicId);
+
+/**
+ * Get display messages by key (with meta information)
+ */
+const getDisplayMessagesByKey =
+  (key: string) =>
+  (s: ChatStoreState): UIChatMessage[] => {
+    const messages = s.messagesMap[key] || [];
+    return messages.map((i) => ({ ...i, meta: getMeta(i) }));
+  };
+
+/**
+ * Get current active session's display messages (includes assistantGroup messages)
+ */
+const activeDisplayMessages = (s: ChatStoreState): UIChatMessage[] => {
+  if (!s.activeId) return [];
+  return getDisplayMessagesByKey(currentDisplayChatKey(s))(s);
+};
+
+// ============= Display Message Queries ========== //
+
+/**
+ * Get display message by ID (searches in messagesMap including assistantGroup children)
+ */
+const getDisplayMessageById = (id: string) => (s: ChatStoreState) =>
+  chatHelpers.getMessageById(activeDisplayMessages(s), id);
+
+const lastDisplayMessageId = (s: ChatStoreState) => {
+  const messages = activeDisplayMessages(s);
+  if (messages.length === 0) return undefined;
+  return messages.at(-1)?.id;
+};
+
+// ============= Thread Handling ========== //
+
+const getChatsWithThread = (s: ChatStoreState, messages: UIChatMessage[]) => {
+  // 如果没有 activeThreadId，则返回所有的主消息
+  if (!s.activeThreadId) return messages.filter((m) => !m.threadId);
+
+  const thread = s.threadMaps[s.activeTopicId!]?.find((t) => t.id === s.activeThreadId);
+
+  if (!thread) return messages.filter((m) => !m.threadId);
+
+  const sourceIndex = messages.findIndex((m) => m.id === thread.sourceMessageId);
+  const sliced = messages.slice(0, sourceIndex + 1);
+
+  return [...sliced, ...messages.filter((m) => m.threadId === s.activeThreadId)];
+};
+
+// ============= Main Display Chats ========== //
+
+/**
+ * Main display chats for UI rendering (without tool messages, with thread handling)
+ */
+const mainDisplayChats = (s: ChatStoreState): UIChatMessage[] => {
+  const displayChats = activeDisplayMessages(s);
+  return getChatsWithThread(s, displayChats);
+};
+
+/**
+ * Main display chat IDs
+ */
+const mainDisplayChatIDs = (s: ChatStoreState) => mainDisplayChats(s).map((s) => s.id);
+
+/**
+ * Main AI chats (includes tool messages, with thread handling)
+ */
+const mainAIChats = (s: ChatStoreState): UIChatMessage[] => {
+  const messages = activeDisplayMessages(s);
+  return getChatsWithThread(s, messages);
+};
+
+/**
+ * Main AI chats with history configuration applied
+ */
+const mainAIChatsWithHistoryConfig = (s: ChatStoreState): UIChatMessage[] => {
+  const chats = mainAIChats(s);
+  const enableHistoryCount = agentChatConfigSelectors.enableHistoryCount(useAgentStore.getState());
+  const historyCount = agentChatConfigSelectors.historyCount(useAgentStore.getState());
+
+  return chatHelpers.getSlicedMessages(chats, {
+    enableHistoryCount,
+    historyCount,
+  });
+};
+
+/**
+ * Concatenated message string from AI chats with history config
+ */
+const mainAIChatsMessageString = (s: ChatStoreState): string => {
+  const chats = mainAIChatsWithHistoryConfig(s);
+  return chats.map((m) => m.content).join('');
+};
+
+/**
+ * Latest message reasoning content
+ */
+const mainAILatestMessageReasoningContent = (s: ChatStoreState) =>
+  mainAIChats(s).at(-1)?.reasoning?.content;
+
+// ============= Display Message Status ========== //
+
+/**
+ * Check if current chat messages are loaded
+ */
+const currentChatLoadingState = (s: ChatStoreState) => !s.messagesInit;
+
+/**
+ * Check if current chat is loaded in messagesMap
+ */
+const isCurrentDisplayChatLoaded = (s: ChatStoreState) => !!s.messagesMap[currentDisplayChatKey(s)];
+
+/**
+ * Show inbox welcome screen
+ */
+const showInboxWelcome = (s: ChatStoreState): boolean => {
+  const isInbox = s.activeId === INBOX_SESSION_ID;
+  if (!isInbox) return false;
+
+  const data = activeDisplayMessages(s);
+  return data.length === 0;
+};
+
+// ============= Thread Messages ========== //
+
+/**
+ * Gets messages between the current user and a specific agent (thread messages)
+ * This is like a DM (Direct Message) view between user and agent
+ */
+const getThreadMessages =
+  (agentId: string) =>
+  (s: ChatStoreState): UIChatMessage[] => {
+    if (!agentId) return [];
+
+    const allMessages = activeDisplayMessages(s);
+
+    // Filter messages to only include:
+    // 1. User messages sent TO the specific agent (role: 'user' && targetId matches agentId)
+    // 2. Assistant messages FROM the specific agent sent TO user (role: 'assistant' && agentId matches && targetId is 'user')
+    return allMessages.filter((message) => {
+      if (message.role === 'user' && message.targetId === agentId) {
+        return true; // Include user messages sent to the specific agent
+      }
+
+      if (
+        message.role === 'assistant' &&
+        message.agentId === agentId &&
+        message.targetId === 'user'
+      ) {
+        return true; // Include messages from the specific agent sent to user
+      }
+
+      return false; // Exclude all other messages
+    });
+  };
+
+/**
+ * Gets thread message IDs for a specific agent
+ */
+const getThreadMessageIDs =
+  (agentId: string) =>
+  (s: ChatStoreState): string[] => {
+    return getThreadMessages(agentId)(s).map((message) => message.id);
+  };
+
+// ============= Group Chat Selectors ========== //
+
+/**
+ * Gets the latest message block from a group message that doesn't contain tools
+ * Returns null if the last block contains tools or if message is not a group message
+ */
+const getGroupLatestMessageWithoutTools = (id: string) => (s: ChatStoreState) => {
+  const message = getDisplayMessageById(id)(s);
+
+  if (
+    !message ||
+    message.role !== 'assistantGroup' ||
+    !message.children ||
+    message.children.length === 0
+  )
+    return;
+
+  // Get the last child
+  const lastChild = message.children.at(-1);
+
+  if (!lastChild) return;
+
+  // Return the last child only if it doesn't have tools
+  if (!lastChild.tools || lastChild.tools.length === 0) {
+    return lastChild;
+  }
+
+  return;
+};
+
+// ============= Supervisor Selectors ========== //
+
+const isSupervisorLoading = (groupId: string) => (s: ChatStoreState) =>
+  s.supervisorDecisionLoading.includes(groupId);
+
+const getSupervisorTodos = (groupId?: string, topicId?: string | null) => (s: ChatStoreState) => {
+  if (!groupId) return [];
+  return s.supervisorTodos[messageMapKey(groupId, topicId)] || [];
+};
+
+// ============= Inbox Selectors ========== //
+
+/**
+ * Get inbox active topic display messages
+ */
+const inboxActiveTopicDisplayMessages = (state: ChatStoreState) => {
+  const activeTopicId = state.activeTopicId;
+  const key = messageMapKey(INBOX_SESSION_ID, activeTopicId);
+  return state.messagesMap[key] || [];
+};
+
+export const displayMessageSelectors = {
+  activeDisplayMessages,
+  currentChatLoadingState,
+  currentDisplayChatKey,
+  getDisplayMessageById,
+  getDisplayMessagesByKey,
+  getGroupLatestMessageWithoutTools,
+  getSupervisorTodos,
+  getThreadMessageIDs,
+  getThreadMessages,
+  inboxActiveTopicDisplayMessages,
+  isCurrentDisplayChatLoaded,
+  isSupervisorLoading,
+  lastDisplayMessageId,
+  mainAIChats,
+  mainAIChatsMessageString,
+  mainAIChatsWithHistoryConfig,
+  mainAILatestMessageReasoningContent,
+  mainDisplayChatIDs,
+  mainDisplayChats,
+  showInboxWelcome,
+};

package/src/store/chat/slices/message/selectors/messageState.ts

@@ -1,8 +1,10 @@
 import type { ChatStoreState } from '../../../initialState';
-import { getMessageByToolCallId, mainDisplayChatIDs } from './chat';
+import { mainDisplayChatIDs } from './chat';
+import { getDbMessageByToolCallId } from './dbMessage';
 
 const isMessageEditing = (id: string) => (s: ChatStoreState) => s.messageEditingIds.includes(id);
 const isMessageLoading = (id: string) => (s: ChatStoreState) => s.messageLoadingIds.includes(id);
+const isMessageRegenerating = (id: string) => (s: ChatStoreState) => s.regeneratingIds.includes(id);
 
 const isMessageGenerating = (id: string) => (s: ChatStoreState) => s.chatLoadingIds.includes(id);
 const isMessageInRAGFlow = (id: string) => (s: ChatStoreState) =>
@@ -31,7 +33,7 @@ const isInToolsCalling = (id: string, index: number) => (s: ChatStoreState) => {
 
 const isToolApiNameShining =
   (messageId: string, index: number, toolCallId: string) => (s: ChatStoreState) => {
-    const toolMessageId = getMessageByToolCallId(toolCallId)(s)?.id;
+    const toolMessageId = getDbMessageByToolCallId(toolCallId)(s)?.id;
     const isStreaming = isToolCallStreaming(messageId, index)(s);
     const isPluginInvoking = !toolMessageId ? true : isPluginApiInvoking(toolMessageId)(s);
 
@@ -73,6 +75,7 @@ export const messageStateSelectors = {
   isMessageInChatReasoning,
   isMessageInRAGFlow,
   isMessageLoading,
+  isMessageRegenerating,
   isPluginApiInvoking,
   isSendButtonDisabledByMessage,
   isToolApiNameShining,

package/src/store/chat/slices/plugin/action.test.ts

@@ -44,13 +44,13 @@ describe('ChatPluginAction', () => {
         content: 'Tool content to summarize',
       } as UIChatMessage;
 
-      const internal_coreProcessMessageMock = vi.fn();
+      const internal_execAgentRuntimeMock = vi.fn();
 
       act(() => {
         useChatStore.setState({
           activeId: 'session-id',
           messagesMap: { [messageMapKey('session-id')]: [toolMessage] },
-          internal_coreProcessMessage: internal_coreProcessMessageMock,
+          internal_execAgentRuntime: internal_execAgentRuntimeMock,
         });
       });
 
@@ -60,27 +60,30 @@ describe('ChatPluginAction', () => {
         await result.current.summaryPluginContent(messageId);
       });
 
-      expect(internal_coreProcessMessageMock).toHaveBeenCalledWith(
-        [
-          {
-            role: 'assistant',
-            content: '作为一名总结专家，请结合以上系统提示词，将以下内容进行总结：',
-          },
-          {
-            ...toolMessage,
-            meta: {
-              avatar: DEFAULT_INBOX_AVATAR,
-              backgroundColor: 'rgba(0,0,0,0)',
-              description: undefined,
-              title: undefined,
+      expect(internal_execAgentRuntimeMock).toHaveBeenCalledWith(
+        expect.objectContaining({
+          messages: [
+            {
+              role: 'assistant',
+              content: '作为一名总结专家，请结合以上系统提示词，将以下内容进行总结：',
             },
-            content: toolMessage.content,
-            role: 'assistant',
-            name: undefined,
-            tool_call_id: undefined,
-          },
-        ],
-        messageId,
+            {
+              ...toolMessage,
+              meta: {
+                avatar: DEFAULT_INBOX_AVATAR,
+                backgroundColor: 'rgba(0,0,0,0)',
+                description: undefined,
+                title: undefined,
+              },
+              content: toolMessage.content,
+              role: 'assistant',
+              name: undefined,
+              tool_call_id: undefined,
+            },
+          ],
+          parentMessageId: messageId,
+          parentMessageType: 'assistant',
+        }),
       );
     });
 
@@ -92,13 +95,13 @@ describe('ChatPluginAction', () => {
         content: 'User message',
       } as UIChatMessage;
 
-      const internal_coreProcessMessageMock = vi.fn();
+      const internal_execAgentRuntimeMock = vi.fn();
 
       act(() => {
         useChatStore.setState({
           activeId: 'session-id',
           messagesMap: { [messageMapKey('session-id')]: [nonToolMessage] },
-          internal_coreProcessMessage: internal_coreProcessMessageMock,
+          internal_execAgentRuntime: internal_execAgentRuntimeMock,
         });
       });
 
@@ -108,7 +111,7 @@ describe('ChatPluginAction', () => {
         await result.current.summaryPluginContent(messageId);
       });
 
-      expect(internal_coreProcessMessageMock).not.toHaveBeenCalled();
+      expect(internal_execAgentRuntimeMock).not.toHaveBeenCalled();
     });
   });
 
@@ -156,8 +159,9 @@ describe('ChatPluginAction', () => {
       // 设置初始状态
       const initialState = {
         messages: [],
-        internal_coreProcessMessage: vi.fn(),
+        internal_execAgentRuntime: vi.fn(),
         refreshMessages: vi.fn(),
+        optimisticUpdateMessageContent: vi.fn(),
       };
       useChatStore.setState(initialState);
 
@@ -170,21 +174,19 @@ describe('ChatPluginAction', () => {
         await result.current.fillPluginMessageContent(messageId, newContent, true);
       });
 
-      // 验证 messageService.internal_updateMessageContent 是否被正确调用
-      expect(messageService.updateMessage).toHaveBeenCalledWith(
+      // 验证 optimisticUpdateMessageContent 是否被正确调用
+      expect(result.current.optimisticUpdateMessageContent).toHaveBeenCalledWith(
         messageId,
-        { content: newContent },
-        { sessionId: 'inbox', topicId: null },
+        newContent,
       );
 
-      // 验证 refreshMessages 是否被调用
-      expect(result.current.refreshMessages).toHaveBeenCalled();
-
       // 验证 coreProcessMessage 是否被正确调用
-      expect(result.current.internal_coreProcessMessage).toHaveBeenCalledWith(
-        mockCurrentChats,
-        messageId,
-        {},
+      expect(result.current.internal_execAgentRuntime).toHaveBeenCalledWith(
+        expect.objectContaining({
+          messages: mockCurrentChats,
+          parentMessageId: messageId,
+          parentMessageType: 'user',
+        }),
       );
     });
     it('should update message content and not trigger ai message', async () => {
@@ -196,8 +198,9 @@ describe('ChatPluginAction', () => {
       const initialState = {
         messages: [],
         coreProcessMessage: vi.fn(),
-        internal_coreProcessMessage: vi.fn(),
+        internal_execAgentRuntime: vi.fn(),
         refreshMessages: vi.fn(),
+        optimisticUpdateMessageContent: vi.fn(),
       };
       useChatStore.setState(initialState);
 
@@ -210,18 +213,14 @@ describe('ChatPluginAction', () => {
         await result.current.fillPluginMessageContent(messageId, newContent);
       });
 
-      // 验证 messageService.internal_updateMessageContent 是否被正确调用
-      expect(messageService.updateMessage).toHaveBeenCalledWith(
+      // 验证 optimisticUpdateMessageContent 是否被正确调用
+      expect(result.current.optimisticUpdateMessageContent).toHaveBeenCalledWith(
         messageId,
-        { content: newContent },
-        { sessionId: 'inbox', topicId: null },
+        newContent,
       );
 
-      // 验证 refreshMessages 是否被调用
-      expect(result.current.refreshMessages).toHaveBeenCalled();
-
       // 验证 coreProcessMessage 没有被正确调用
-      expect(result.current.internal_coreProcessMessage).not.toHaveBeenCalled();
+      expect(result.current.internal_execAgentRuntime).not.toHaveBeenCalled();
     });
   });
 
@@ -236,6 +235,7 @@ describe('ChatPluginAction', () => {
       vi.spyOn(storeState, 'refreshMessages');
       vi.spyOn(storeState, 'triggerAIMessage').mockResolvedValue(undefined);
       vi.spyOn(storeState, 'internal_togglePluginApiCalling').mockReturnValue(undefined);
+      vi.spyOn(storeState, 'optimisticUpdateMessageContent').mockResolvedValue();
 
       const runSpy = vi.spyOn(chatService, 'runPluginApi').mockResolvedValue({
         text: pluginApiResponse,
@@ -254,12 +254,10 @@ describe('ChatPluginAction', () => {
         expect.any(String),
       );
       expect(runSpy).toHaveBeenCalledWith(pluginPayload, { signal: undefined, trace: {} });
-      expect(messageService.updateMessage).toHaveBeenCalledWith(
+      expect(storeState.optimisticUpdateMessageContent).toHaveBeenCalledWith(
         messageId,
-        { content: pluginApiResponse },
-        { sessionId: 'inbox', topicId: null },
+        pluginApiResponse,
       );
-      expect(storeState.refreshMessages).toHaveBeenCalled();
       expect(storeState.internal_togglePluginApiCalling).toHaveBeenCalledWith(
         false,
         'message-id',
@@ -352,7 +350,7 @@ describe('ChatPluginAction', () => {
       const invokeBuiltinToolMock = vi.fn();
       const invokeDefaultTypePluginMock = vi.fn().mockResolvedValue('Default tool response');
       const triggerAIMessageMock = vi.fn();
-      const internal_createMessageMock = vi
+      const optimisticCreateMessageMock = vi
         .fn()
         .mockResolvedValue({ id: 'tool-message-id', messages: [] });
       const getTraceIdByMessageIdMock = vi.fn().mockReturnValue('trace-id');
@@ -367,7 +365,7 @@ describe('ChatPluginAction', () => {
           invokeBuiltinTool: invokeBuiltinToolMock,
           invokeDefaultTypePlugin: invokeDefaultTypePluginMock,
           triggerAIMessage: triggerAIMessageMock,
-          internal_createMessage: internal_createMessageMock,
+          optimisticCreateMessage: optimisticCreateMessageMock,
           activeId: 'session-id',
           activeTopicId: 'topic-id',
         });
@@ -380,8 +378,8 @@ describe('ChatPluginAction', () => {
       });
 
       // Verify that tool messages were created for each tool call
-      expect(internal_createMessageMock).toHaveBeenCalledTimes(4);
-      expect(internal_createMessageMock).toHaveBeenNthCalledWith(1, {
+      expect(optimisticCreateMessageMock).toHaveBeenCalledTimes(4);
+      expect(optimisticCreateMessageMock).toHaveBeenNthCalledWith(1, {
         content: '',
         parentId: assistantId,
         plugin: message.tools![0],
@@ -392,7 +390,7 @@ describe('ChatPluginAction', () => {
         threadId: undefined,
         groupId: undefined,
       });
-      expect(internal_createMessageMock).toHaveBeenNthCalledWith(2, {
+      expect(optimisticCreateMessageMock).toHaveBeenNthCalledWith(2, {
         content: '',
         parentId: assistantId,
         plugin: message.tools![1],
@@ -403,7 +401,7 @@ describe('ChatPluginAction', () => {
         threadId: undefined,
         groupId: undefined,
       });
-      expect(internal_createMessageMock).toHaveBeenNthCalledWith(3, {
+      expect(optimisticCreateMessageMock).toHaveBeenNthCalledWith(3, {
         content: '',
         parentId: assistantId,
         plugin: message.tools![2],
@@ -414,7 +412,7 @@ describe('ChatPluginAction', () => {
         threadId: undefined,
         groupId: undefined,
       });
-      expect(internal_createMessageMock).toHaveBeenNthCalledWith(4, {
+      expect(optimisticCreateMessageMock).toHaveBeenNthCalledWith(4, {
         content: '',
         parentId: assistantId,
         plugin: message.tools![3],
@@ -481,7 +479,7 @@ describe('ChatPluginAction', () => {
       const invokeMarkdownTypePluginMock = vi.fn();
       const invokeBuiltinToolMock = vi.fn();
       const triggerAIMessageMock = vi.fn();
-      const internal_createMessageMock = vi
+      const optimisticCreateMessageMock = vi
         .fn()
         .mockResolvedValue({ id: 'tool-message-id', messages: [] });
 
@@ -491,7 +489,7 @@ describe('ChatPluginAction', () => {
           invokeMarkdownTypePlugin: invokeMarkdownTypePluginMock,
           invokeBuiltinTool: invokeBuiltinToolMock,
           triggerAIMessage: triggerAIMessageMock,
-          internal_createMessage: internal_createMessageMock,
+          optimisticCreateMessage: optimisticCreateMessageMock,
           activeId: 'session-id',
           messagesMap: {
             [messageMapKey('session-id', 'topic-id')]: [message],
@@ -507,7 +505,7 @@ describe('ChatPluginAction', () => {
       });
 
       // Verify that tool messages were created for each tool call
-      expect(internal_createMessageMock).toHaveBeenCalledTimes(3);
+      expect(optimisticCreateMessageMock).toHaveBeenCalledTimes(3);
 
       // Verify that the appropriate plugin types were invoked
       expect(invokeStandaloneTypePluginMock).toHaveBeenCalledWith(
@@ -801,7 +799,7 @@ describe('ChatPluginAction', () => {
           activeId: 'session-id',
           messagesMap: { [messageMapKey('session-id')]: [message] },
           internal_invokeDifferentTypePlugin: internal_invokeDifferentTypePluginMock,
-          internal_updateMessageError: vi.fn(),
+          optimisticUpdateMessagePluginError: vi.fn(),
         });
       });
 
@@ -840,7 +838,7 @@ describe('ChatPluginAction', () => {
           activeId: 'session-id',
           messagesMap: { [messageMapKey('session-id')]: [message] },
           internal_invokeDifferentTypePlugin: vi.fn(),
-          internal_updateMessagePluginError: internal_updateMessageErrorMock,
+          optimisticUpdateMessagePluginError: internal_updateMessageErrorMock,
         });
       });
 
@@ -925,7 +923,7 @@ describe('ChatPluginAction', () => {
       act(() => {
         useChatStore.setState({
           internal_togglePluginApiCalling: vi.fn(),
-          internal_updateMessageContent: vi.fn(),
+          optimisticUpdateMessageContent: vi.fn(),
           refreshMessages: vi.fn(),
         });
       });
@@ -937,7 +935,7 @@ describe('ChatPluginAction', () => {
       });
 
       expect(chatService.runPluginApi).toHaveBeenCalledWith(payload, expect.any(Object));
-      expect(result.current.internal_updateMessageContent).toHaveBeenCalledWith(
+      expect(result.current.optimisticUpdateMessageContent).toHaveBeenCalledWith(
         messageId,
         apiResponse,
       );