diff --git a/src/renderer/src/pages/home/Messages/Messages.tsx b/src/renderer/src/pages/home/Messages/Messages.tsx index cae4237ffd..c7b97ae05a 100644 --- a/src/renderer/src/pages/home/Messages/Messages.tsx +++ b/src/renderer/src/pages/home/Messages/Messages.tsx @@ -1,6 +1,7 @@ import SvgSpinners180Ring from '@renderer/components/Icons/SvgSpinners180Ring' import Scrollbar from '@renderer/components/Scrollbar' import { LOAD_MORE_COUNT } from '@renderer/config/constant' +import db from '@renderer/databases' import { useAssistant } from '@renderer/hooks/useAssistant' import { useChatContext } from '@renderer/hooks/useChatContext' import { useMessageOperations, useTopicMessages } from '@renderer/hooks/useMessageOperations' @@ -11,6 +12,7 @@ import { autoRenameTopic, getTopic } from '@renderer/hooks/useTopic' import SelectionBox from '@renderer/pages/home/Messages/SelectionBox' import { getDefaultTopic } from '@renderer/services/AssistantService' import { EVENT_NAMES, EventEmitter } from '@renderer/services/EventService' +import FileManager from '@renderer/services/FileManager' import { getContextCount, getGroupedMessages, getUserMessage } from '@renderer/services/MessagesService' import { estimateHistoryTokens } from '@renderer/services/TokenService' import store, { useAppDispatch } from '@renderer/store' @@ -102,14 +104,22 @@ const Messages: React.FC = ({ assistant, topic, setActiveTopic, o async (data: Topic) => { const defaultTopic = getDefaultTopic(assistant.id) + async function handleTopicRelatedFiles(topicId: string) { + const topicFiles = await db.files.where({ topicId }).toArray() + if (topicFiles.length > 0) { + await Promise.all(topicFiles.map((file) => FileManager.deleteFile(file.id, false))) + } + } + if (data && data.id !== topic.id) { await clearTopicMessages(data.id) + await handleTopicRelatedFiles(data.id) updateTopic({ ...data, name: defaultTopic.name } as Topic) return } await clearTopicMessages() - + await handleTopicRelatedFiles(topic.id) setDisplayMessages([]) const _topic = getTopic(assistant, topic.id) diff --git a/src/renderer/src/services/FileManager.ts b/src/renderer/src/services/FileManager.ts index beda045a81..c76254a74c 100644 --- a/src/renderer/src/services/FileManager.ts +++ b/src/renderer/src/services/FileManager.ts @@ -8,8 +8,7 @@ import dayjs from 'dayjs' class FileManager { static async selectFiles(options?: Electron.OpenDialogOptions): Promise { - const files = await window.api.file.select(options) - return files + return await window.api.file.select(options) } static async addFile(file: FileType): Promise {