78 lines
2.0 KiB
TypeScript
78 lines
2.0 KiB
TypeScript
import { useEffect } from 'react'
|
|
|
|
import {
|
|
ExtensionTypeEnum,
|
|
Thread,
|
|
ThreadState,
|
|
ConversationalExtension,
|
|
} from '@janhq/core'
|
|
|
|
import { useSetAtom } from 'jotai'
|
|
|
|
import useSetActiveThread from './useSetActiveThread'
|
|
|
|
import { extensionManager } from '@/extension/ExtensionManager'
|
|
import {
|
|
ModelParams,
|
|
threadModelParamsAtom,
|
|
threadStatesAtom,
|
|
threadsAtom,
|
|
} from '@/helpers/atoms/Thread.atom'
|
|
|
|
const useThreads = () => {
|
|
const setThreadStates = useSetAtom(threadStatesAtom)
|
|
const setThreads = useSetAtom(threadsAtom)
|
|
const setThreadModelRuntimeParams = useSetAtom(threadModelParamsAtom)
|
|
const { setActiveThread } = useSetActiveThread()
|
|
|
|
useEffect(() => {
|
|
const getThreads = async () => {
|
|
const localThreads = await getLocalThreads()
|
|
const localThreadStates: Record<string, ThreadState> = {}
|
|
const threadModelParams: Record<string, ModelParams> = {}
|
|
|
|
localThreads.forEach((thread) => {
|
|
if (thread.id != null) {
|
|
const lastMessage = (thread.metadata?.lastMessage as string) ?? ''
|
|
|
|
localThreadStates[thread.id] = {
|
|
hasMore: false,
|
|
waitingForResponse: false,
|
|
lastMessage,
|
|
}
|
|
|
|
const modelParams = thread.assistants?.[0]?.model?.parameters
|
|
const engineParams = thread.assistants?.[0]?.model?.settings
|
|
threadModelParams[thread.id] = {
|
|
...modelParams,
|
|
...engineParams,
|
|
}
|
|
}
|
|
})
|
|
|
|
// updating app states
|
|
setThreadStates(localThreadStates)
|
|
setThreads(localThreads)
|
|
setThreadModelRuntimeParams(threadModelParams)
|
|
|
|
if (localThreads.length > 0) {
|
|
setActiveThread(localThreads[0])
|
|
}
|
|
}
|
|
|
|
getThreads()
|
|
}, [
|
|
setActiveThread,
|
|
setThreadModelRuntimeParams,
|
|
setThreadStates,
|
|
setThreads,
|
|
])
|
|
}
|
|
|
|
const getLocalThreads = async (): Promise<Thread[]> =>
|
|
(await extensionManager
|
|
.get<ConversationalExtension>(ExtensionTypeEnum.Conversational)
|
|
?.getThreads()) ?? []
|
|
|
|
export default useThreads
|