81 lines
2.4 KiB
TypeScript
81 lines
2.4 KiB
TypeScript
import { ChatCompletionRole, ExtensionType } from '@janhq/core'
|
|
import { ConversationalExtension } from '@janhq/core'
|
|
import { useAtom, useAtomValue, useSetAtom } from 'jotai'
|
|
|
|
import { currentPromptAtom } from '@/containers/Providers/Jotai'
|
|
|
|
import { toaster } from '@/containers/Toast'
|
|
|
|
import { useActiveModel } from './useActiveModel'
|
|
|
|
import { extensionManager } from '@/extension/ExtensionManager'
|
|
|
|
import {
|
|
cleanConversationMessages,
|
|
deleteConversationMessage,
|
|
getCurrentChatMessagesAtom,
|
|
} from '@/helpers/atoms/ChatMessage.atom'
|
|
import {
|
|
threadsAtom,
|
|
getActiveThreadIdAtom,
|
|
setActiveThreadIdAtom,
|
|
} from '@/helpers/atoms/Conversation.atom'
|
|
|
|
export default function useDeleteThread() {
|
|
const { activeModel } = useActiveModel()
|
|
const [threads, setThreads] = useAtom(threadsAtom)
|
|
const setCurrentPrompt = useSetAtom(currentPromptAtom)
|
|
const activeThreadId = useAtomValue(getActiveThreadIdAtom)
|
|
const messages = useAtomValue(getCurrentChatMessagesAtom)
|
|
|
|
const setActiveConvoId = useSetAtom(setActiveThreadIdAtom)
|
|
const deleteMessages = useSetAtom(deleteConversationMessage)
|
|
const cleanMessages = useSetAtom(cleanConversationMessages)
|
|
|
|
const cleanThread = async () => {
|
|
if (activeThreadId) {
|
|
const thread = threads.filter((c) => c.id === activeThreadId)[0]
|
|
cleanMessages(activeThreadId)
|
|
if (thread)
|
|
await extensionManager
|
|
.get<ConversationalExtension>(ExtensionType.Conversational)
|
|
?.writeMessages(
|
|
activeThreadId,
|
|
messages.filter((msg) => msg.role === ChatCompletionRole.System)
|
|
)
|
|
}
|
|
}
|
|
|
|
const deleteThread = async () => {
|
|
if (!activeThreadId) {
|
|
alert('No active thread')
|
|
return
|
|
}
|
|
try {
|
|
await extensionManager
|
|
.get<ConversationalExtension>(ExtensionType.Conversational)
|
|
?.deleteThread(activeThreadId)
|
|
const availableThreads = threads.filter((c) => c.id !== activeThreadId)
|
|
setThreads(availableThreads)
|
|
deleteMessages(activeThreadId)
|
|
setCurrentPrompt('')
|
|
toaster({
|
|
title: 'Chat successfully deleted.',
|
|
description: `Chat with ${activeModel?.name} has been successfully deleted.`,
|
|
})
|
|
if (availableThreads.length > 0) {
|
|
setActiveConvoId(availableThreads[0].id)
|
|
} else {
|
|
setActiveConvoId(undefined)
|
|
}
|
|
} catch (err) {
|
|
console.error(err)
|
|
}
|
|
}
|
|
|
|
return {
|
|
cleanThread,
|
|
deleteThread,
|
|
}
|
|
}
|