🚸 style: add delete and regenerate for function message (lobehub#992)

* 🚸 style: add delete and regenerate for function message * 🐛 fix: fix plugin state update error * ✅ test: fix test
CellLz · Jan 10, 2024 · 7f8c717 · 7f8c717
1 parent c0f1ca7
commit 7f8c717
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 7 deletions.
diff --git a/src/database/models/message.ts b/src/database/models/message.ts
@@ -102,6 +102,12 @@ class _MessageModel extends BaseModel {
     return super._update(id, data);
   }
 
+  async updatePluginState(id: string, key: string, value: any) {
+    const item = await this.findById(id);
+
+    return this.update(id, { pluginState: { ...item.pluginState, [key]: value } });
+  }
+
   /**
    * Batch updates multiple fields of the specified messages.
    *

diff --git a/src/features/Conversation/Actions/Function.tsx b/src/features/Conversation/Actions/Function.tsx
@@ -5,10 +5,10 @@ import { useChatListActionsBar } from '../hooks/useChatListActionsBar';
 import { RenderAction } from '../types';
 
 export const FunctionActionsBar: RenderAction = memo(({ onActionClick }) => {
-  const { regenerate, divider, del } = useChatListActionsBar();
+  const { regenerate, divider, delAndRegenerate, del } = useChatListActionsBar();
   return (
     <ActionIconGroup
-      dropdownMenu={[regenerate, divider, del]}
+      dropdownMenu={[regenerate, divider, delAndRegenerate, del]}
       items={[regenerate]}
       onActionClick={onActionClick}
       type="ghost"

diff --git a/src/services/__tests__/message.test.ts b/src/services/__tests__/message.test.ts
@@ -26,6 +26,7 @@ vi.mock('@/database/models/message', () => {
       clearTable: vi.fn(),
       batchUpdate: vi.fn(),
       queryAll: vi.fn(),
+      updatePluginState: vi.fn(),
     },
   };
 });
@@ -255,7 +256,7 @@ describe('MessageService', () => {
       const key = 'stateKey';
       const value = 'stateValue';
       const newPluginState = { [key]: value };
-      (MessageModel.update as Mock).mockResolvedValue({
+      (MessageModel.updatePluginState as Mock).mockResolvedValue({
         ...mockMessage,
         pluginState: newPluginState,
       });
@@ -264,9 +265,7 @@ describe('MessageService', () => {
       const result = await messageService.updateMessagePluginState(mockMessageId, key, value);
 
       // Assert
-      expect(MessageModel.update).toHaveBeenCalledWith(mockMessageId, {
-        pluginState: newPluginState,
-      });
+      expect(MessageModel.updatePluginState).toHaveBeenCalledWith(mockMessageId, key, value);
       expect(result).toEqual({ ...mockMessage, pluginState: newPluginState });
     });
   });

diff --git a/src/services/message.ts b/src/services/message.ts
@@ -53,8 +53,9 @@ export class MessageService {
   async updateMessagePlugin(id: string, plugin: ChatPluginPayload) {
     return MessageModel.update(id, { plugin });
   }
+
   async updateMessagePluginState(id: string, key: string, value: any) {
-    return MessageModel.update(id, { pluginState: { [key]: value } });
+    return MessageModel.updatePluginState(id, key, value);
   }
 
   async getAllMessages() {