100 lines
3.1 KiB
TypeScript
100 lines
3.1 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
import { useCallback } from 'react'
|
|
|
|
import { defaultThreadTitle } from '@/constants/Threads'
|
|
|
|
import useAssistantQuery from './useAssistantQuery'
|
|
|
|
import useCortex from './useCortex'
|
|
import useMessageCreateMutation from './useMessageCreateMutation'
|
|
import useThreads from './useThreads'
|
|
|
|
const useMigratingData = () => {
|
|
const { createThread } = useThreads()
|
|
const { updateThread } = useCortex()
|
|
const createMessage = useMessageCreateMutation()
|
|
const { data: assistants } = useAssistantQuery()
|
|
|
|
const getJanThreadsAndMessages = useCallback(async (): Promise<{
|
|
messages: any[]
|
|
threads: any[]
|
|
}> => {
|
|
return window?.electronAPI?.getAllMessagesAndThreads()
|
|
}, [])
|
|
|
|
const getJanLocalModels = useCallback(async (): Promise<boolean> => {
|
|
// TODO: change the name of this function
|
|
return window?.electronAPI?.getAllLocalModels()
|
|
}, [])
|
|
|
|
const migrateModels = useCallback(async () => {
|
|
return window?.electronAPI?.syncModelFileToCortex()
|
|
}, [])
|
|
|
|
const migrateThreadsAndMessages = useCallback(async () => {
|
|
if (!assistants || assistants.length === 0) {
|
|
console.error('No assistant found')
|
|
return
|
|
}
|
|
const threadsAndMessages = await getJanThreadsAndMessages()
|
|
const janThreads = threadsAndMessages.threads
|
|
|
|
for (const thread of janThreads) {
|
|
const modelId: string | undefined = thread.assistants[0]?.model?.id
|
|
if (!modelId || modelId.trim().length === 0 || modelId === '*') {
|
|
console.error(`Ignore thread ${thread.id} because modelId is not found`)
|
|
continue
|
|
}
|
|
const threadTitle: string = thread.title ?? defaultThreadTitle
|
|
const instructions: string = thread.assistants[0]?.instructions ?? ''
|
|
// currently, we don't have api support for creating thread with messages
|
|
const cortexThread = await createThread(modelId, assistants[0])
|
|
|
|
console.log('createThread', cortexThread)
|
|
// update instruction
|
|
cortexThread.assistants[0].instructions = instructions
|
|
cortexThread.title = threadTitle
|
|
|
|
// update thread name
|
|
await updateThread(cortexThread)
|
|
console.log('updateThread', cortexThread)
|
|
|
|
// we finished with thread, now continue with messages
|
|
const janMessages = threadsAndMessages.messages.filter(
|
|
(m) => m.thread_id === thread.id
|
|
)
|
|
|
|
for (let j = 0; j < janMessages.length; ++j) {
|
|
const janMessage = janMessages[j]
|
|
// filter out the system message if any
|
|
if (janMessage.role === 'system') continue
|
|
const messageContent: string = janMessage.content[0]?.text.value ?? ''
|
|
|
|
// can speed up here with Promise.allSettled
|
|
await createMessage.mutateAsync({
|
|
threadId: cortexThread.id,
|
|
createMessageParams: {
|
|
content: messageContent,
|
|
role: janMessage.role,
|
|
},
|
|
})
|
|
}
|
|
}
|
|
}, [
|
|
assistants,
|
|
getJanThreadsAndMessages,
|
|
createThread,
|
|
updateThread,
|
|
createMessage,
|
|
])
|
|
|
|
return {
|
|
migrateModels,
|
|
migrateThreadsAndMessages,
|
|
getJanThreadsAndMessages,
|
|
getJanLocalModels,
|
|
}
|
|
}
|
|
|
|
export default useMigratingData
|