✨ feat: support auto rename topic

lobehub · Jan 4, 2024 · 4c5a345 · 4c5a345
1 parent acae827
commit 4c5a345
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 7 deletions.
diff --git a/src/app/chat/features/TopicListContent/Topic/TopicContent.tsx b/src/app/chat/features/TopicListContent/Topic/TopicContent.tsx
@@ -1,7 +1,7 @@
 import { ActionIcon, EditableText, Icon } from '@lobehub/ui';
 import { App, Dropdown, type MenuProps, Typography } from 'antd';
 import { createStyles } from 'antd-style';
-import { MoreVertical, PencilLine, Star, Trash } from 'lucide-react';
+import { MoreVertical, PencilLine, Star, Trash, Wand2 } from 'lucide-react';
 import { memo, useMemo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
@@ -33,12 +33,14 @@ interface TopicContentProps {
 const TopicContent = memo<TopicContentProps>(({ id, title, fav, showMore }) => {
   const { t } = useTranslation('common');
 
-  const [editing, favoriteTopic, updateTopicTitle, removeTopic] = useChatStore((s) => [
-    s.topicRenamingId === id,
-    s.favoriteTopic,
-    s.updateTopicTitle,
-    s.removeTopic,
-  ]);
+  const [editing, favoriteTopic, updateTopicTitle, removeTopic, autoRenameTopicTitle] =
+    useChatStore((s) => [
+      s.topicRenamingId === id,
+      s.favoriteTopic,
+      s.updateTopicTitle,
+      s.removeTopic,
+      s.autoRenameTopicTitle,
+    ]);
   const { styles, theme } = useStyles();
 
   const toggleEditing = (visible?: boolean) => {
@@ -57,6 +59,14 @@ const TopicContent = memo<TopicContentProps>(({ id, title, fav, showMore }) => {
           toggleEditing(true);
         },
       },
+      {
+        icon: <Icon icon={Wand2} />,
+        key: 'autoRename',
+        label: t('topic.actions.autoRename', { ns: 'chat' }),
+        onClick: () => {
+          autoRenameTopicTitle(id);
+        },
+      },
       // {
       //   icon: <Icon icon={Share2} />,
       //   key: 'share',

diff --git a/src/locales/default/chat.ts b/src/locales/default/chat.ts
@@ -56,6 +56,9 @@ export default {
     used: '使用',
   },
   topic: {
+    actions: {
+      autoRename: '智能重命名',
+    },
     confirmRemoveAll: '即将删除全部话题，删除后将不可恢复，请谨慎操作。',
     confirmRemoveTopic: '即将删除该话题，删除后将不可恢复，请谨慎操作。',
     confirmRemoveUnstarred: '即将删除未收藏话题，删除后将不可恢复，请谨慎操作。',

diff --git a/src/store/chat/slices/topic/action.ts b/src/store/chat/slices/topic/action.ts
@@ -30,6 +30,7 @@ export interface ChatTopicAction {
   removeTopic: (id: string) => Promise<void>;
   removeUnstarredTopic: () => void;
   saveToTopic: () => Promise<string | undefined>;
+  autoRenameTopicTitle: (id: string) => Promise<void>;
   summaryTopicTitle: (topicId: string, messages: ChatMessage[]) => Promise<void>;
   switchTopic: (id?: string) => Promise<void>;
   updateTopicTitleInSummary: (id: string, title: string) => void;
@@ -115,6 +116,13 @@ export const chatTopic: StateCreator<
     await topicService.updateTitle(id, title);
     await get().refreshTopic();
   },
+
+  autoRenameTopicTitle: async (id) => {
+    const { activeId: sessionId, summaryTopicTitle } = get();
+    const messages = await messageService.getMessages(sessionId, id);
+    console.log(messages);
+    await summaryTopicTitle(id, messages);
+  },
   // query
   useFetchTopics: (sessionId) =>
     useSWR<ChatTopic[]>(sessionId, async (sessionId) => topicService.getTopics({ sessionId }), {