✨ feat: support delete and regenerate message

lobehub · Jan 4, 2024 · a05be1c · a05be1c
1 parent d8f55d5
commit a05be1c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/features/Conversation/Actions/customAction.ts b/src/features/Conversation/Actions/customAction.ts
@@ -1,5 +1,6 @@
 import { ActionIconGroupItems } from '@lobehub/ui/es/ActionIconGroup';
 import { LanguagesIcon, Play } from 'lucide-react';
+import { useMemo } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import { localeOptions } from '@/locales/resources';
@@ -23,8 +24,5 @@ export const useCustomActions = () => {
     label: t('tts.action'),
   } as ActionIconGroupItems;
 
-  return {
-    translate,
-    tts,
-  };
+  return useMemo(() => ({ translate, tts }), []);
 };
diff --git a/src/features/Conversation/Actions/index.ts b/src/features/Conversation/Actions/index.ts
@@ -49,6 +49,12 @@ export const useActionsClick = (): OnActionsClick => {
         break;
       }
 
+      case 'delAndRegenerate': {
+        resendMessage(id);
+        deleteMessage(id);
+        break;
+      }
+
       case 'tts': {
         ttsMessage(id);
         break;