@lobehub/chat 1.121.1 → 1.122.0

package/src/store/chat/slices/message/action.ts

@@ -22,6 +22,7 @@ import {
   ModelReasoning,
 } from '@/types/message';
 import { ChatImageItem } from '@/types/message/image';
+import { UpdateMessageRAGParams } from '@/types/message/rag';
 import { GroundingSearch } from '@/types/search';
 import { TraceEventPayloads } from '@/types/trace';
 import { Action, setNamespace } from '@/utils/storeDebug';
@@ -39,6 +40,7 @@ const SWR_USE_FETCH_MESSAGES = 'SWR_USE_FETCH_MESSAGES';
 export interface ChatMessageAction {
   // create
   addAIMessage: () => Promise<void>;
+  addUserMessage: (params: { message: string; fileList?: string[] }) => Promise<void>;
   // delete
   /**
    * clear message on the active session
@@ -59,10 +61,11 @@ export interface ChatMessageAction {
   ) => SWRResponse<ChatMessage[]>;
   copyMessage: (id: string, content: string) => Promise<void>;
   refreshMessages: () => Promise<void>;
-
+  replaceMessages: (messages: ChatMessage[]) => void;
   // =========  ↓ Internal Method ↓  ========== //
   // ========================================== //
   // ========================================== //
+  internal_updateMessageRAG: (id: string, input: UpdateMessageRAGParams) => Promise<void>;
 
   /**
    * update message at the frontend
@@ -213,6 +216,21 @@ export const chatMessage: StateCreator<
 
     updateInputMessage('');
   },
+  addUserMessage: async ({ message, fileList }) => {
+    const { internal_createMessage, updateInputMessage, activeTopicId, activeId } = get();
+    if (!activeId) return;
+
+    await internal_createMessage({
+      content: message,
+      files: fileList,
+      role: 'user',
+      sessionId: activeId,
+      // if there is activeTopicId，then add topicId to message
+      topicId: activeTopicId,
+    });
+
+    updateInputMessage('');
+  },
   copyMessage: async (id, content) => {
     await copyToClipboard(content);
 
@@ -266,6 +284,25 @@ export const chatMessage: StateCreator<
   refreshMessages: async () => {
     await mutate([SWR_USE_FETCH_MESSAGES, get().activeId, get().activeTopicId]);
   },
+  replaceMessages: (messages) => {
+    set(
+      {
+        messagesMap: {
+          ...get().messagesMap,
+          [messageMapKey(get().activeId, get().activeTopicId)]: messages,
+        },
+      },
+      false,
+      'replaceMessages',
+    );
+  },
+
+  internal_updateMessageRAG: async (id, data) => {
+    const { refreshMessages } = get();
+
+    await messageService.updateMessageRAG(id, data);
+    await refreshMessages();
+  },
 
   // the internal process method of the AI message
   internal_dispatchMessage: (payload) => {

package/src/store/chat/slices/message/reducer.ts

@@ -11,6 +11,11 @@ import {
 import { merge } from '@/utils/merge';
 
 interface UpdateMessages {
+  type: 'updateMessages';
+  value: ChatMessage[];
+}
+
+interface UpdateMessage {
   id: string;
   type: 'updateMessage';
   value: Partial<ChatMessage>;
@@ -72,6 +77,7 @@ interface UpdateMessageExtra {
 
 export type MessageDispatch =
   | CreateMessage
+  | UpdateMessage
   | UpdateMessages
   | UpdatePluginState
   | UpdateMessageExtra
@@ -194,6 +200,11 @@ export const messagesReducer = (state: ChatMessage[], payload: MessageDispatch):
         draftState.push({ ...value, createdAt: Date.now(), id, meta: {}, updatedAt: Date.now() });
       });
     }
+
+    case 'updateMessages': {
+      return payload.value;
+    }
+
     case 'deleteMessage': {
       return produce(state, (draft) => {
         const { id } = payload;

package/src/store/chat/slices/topic/reducer.ts

@@ -14,12 +14,21 @@ interface UpdateChatTopicAction {
   value: Partial<ChatTopic>;
 }
 
+interface UpdateTopicsAction {
+  type: 'updateTopics';
+  value: ChatTopic[];
+}
+
 interface DeleteChatTopicAction {
   id: string;
   type: 'deleteTopic';
 }
 
-export type ChatTopicDispatch = AddChatTopicAction | UpdateChatTopicAction | DeleteChatTopicAction;
+export type ChatTopicDispatch =
+  | AddChatTopicAction
+  | UpdateChatTopicAction
+  | DeleteChatTopicAction
+  | UpdateTopicsAction;
 
 export const topicReducer = (state: ChatTopic[] = [], payload: ChatTopicDispatch): ChatTopic[] => {
   switch (payload.type) {
@@ -51,6 +60,10 @@ export const topicReducer = (state: ChatTopic[] = [], payload: ChatTopicDispatch
       });
     }
 
+    case 'updateTopics': {
+      return payload.value;
+    }
+
     case 'deleteTopic': {
       return produce(state, (draftState) => {
         const topicIndex = draftState.findIndex((topic) => topic.id === payload.id);