112 lines
2.9 KiB
TypeScript
112 lines
2.9 KiB
TypeScript
import {
|
|
Assistant,
|
|
ConversationalExtension,
|
|
ExtensionType,
|
|
Thread,
|
|
ThreadAssistantInfo,
|
|
ThreadState,
|
|
} from '@janhq/core'
|
|
import { atom, useAtom, useAtomValue, useSetAtom } from 'jotai'
|
|
|
|
import { generateThreadId } from '@/utils/conversation'
|
|
|
|
import { extensionManager } from '@/extension'
|
|
import {
|
|
threadsAtom,
|
|
setActiveThreadIdAtom,
|
|
threadStatesAtom,
|
|
activeThreadAtom,
|
|
updateThreadAtom,
|
|
} from '@/helpers/atoms/Conversation.atom'
|
|
|
|
const createNewThreadAtom = atom(null, (get, set, newThread: Thread) => {
|
|
// create thread state for this new thread
|
|
const currentState = { ...get(threadStatesAtom) }
|
|
|
|
const threadState: ThreadState = {
|
|
hasMore: false,
|
|
waitingForResponse: false,
|
|
}
|
|
currentState[newThread.id] = threadState
|
|
set(threadStatesAtom, currentState)
|
|
|
|
// add the new thread on top of the thread list to the state
|
|
const threads = get(threadsAtom)
|
|
set(threadsAtom, [newThread, ...threads])
|
|
})
|
|
|
|
export const useCreateNewThread = () => {
|
|
const createNewThread = useSetAtom(createNewThreadAtom)
|
|
const setActiveThreadId = useSetAtom(setActiveThreadIdAtom)
|
|
const [threadStates, setThreadStates] = useAtom(threadStatesAtom)
|
|
const threads = useAtomValue(threadsAtom)
|
|
const updateThread = useSetAtom(updateThreadAtom)
|
|
|
|
const requestCreateNewThread = async (assistant: Assistant) => {
|
|
const unfinishedThreads = threads.filter((t) => t.isFinishInit === false)
|
|
if (unfinishedThreads.length > 0) {
|
|
return
|
|
}
|
|
|
|
const createdAt = Date.now()
|
|
const assistantInfo: ThreadAssistantInfo = {
|
|
assistant_id: assistant.id,
|
|
assistant_name: assistant.name,
|
|
model: {
|
|
id: '*',
|
|
settings: {
|
|
ctx_len: 0,
|
|
ngl: 0,
|
|
embedding: false,
|
|
n_parallel: 0,
|
|
},
|
|
parameters: {
|
|
temperature: 0,
|
|
token_limit: 0,
|
|
top_k: 0,
|
|
top_p: 0,
|
|
stream: false,
|
|
},
|
|
engine: undefined
|
|
},
|
|
instructions: assistant.instructions,
|
|
}
|
|
const threadId = generateThreadId(assistant.id)
|
|
const thread: Thread = {
|
|
id: threadId,
|
|
object: 'thread',
|
|
title: 'New Thread',
|
|
assistants: [assistantInfo],
|
|
created: createdAt,
|
|
updated: createdAt,
|
|
isFinishInit: false,
|
|
}
|
|
|
|
// TODO: move isFinishInit here
|
|
const threadState: ThreadState = {
|
|
hasMore: false,
|
|
waitingForResponse: false,
|
|
lastMessage: undefined,
|
|
}
|
|
setThreadStates({ ...threadStates, [threadId]: threadState })
|
|
// add the new thread on top of the thread list to the state
|
|
createNewThread(thread)
|
|
setActiveThreadId(thread.id)
|
|
}
|
|
|
|
function updateThreadMetadata(thread: Thread) {
|
|
const updatedThread: Thread = {
|
|
...thread,
|
|
}
|
|
updateThread(updatedThread)
|
|
extensionManager
|
|
.get<ConversationalExtension>(ExtensionType.Conversational)
|
|
?.saveThread(updatedThread)
|
|
}
|
|
|
|
return {
|
|
requestCreateNewThread,
|
|
updateThreadMetadata,
|
|
}
|
|
}
|