89 lines
2.5 KiB
TypeScript
89 lines
2.5 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
import { ReactNode, useEffect, useRef } from 'react'
|
|
|
|
import {
|
|
events,
|
|
EventName,
|
|
ThreadMessage,
|
|
ExtensionType,
|
|
MessageStatus,
|
|
} from '@janhq/core'
|
|
import { ConversationalExtension } from '@janhq/core'
|
|
import { useAtomValue, useSetAtom } from 'jotai'
|
|
|
|
import { extensionManager } from '@/extension'
|
|
import {
|
|
addNewMessageAtom,
|
|
updateMessageAtom,
|
|
} from '@/helpers/atoms/ChatMessage.atom'
|
|
import {
|
|
updateThreadWaitingForResponseAtom,
|
|
threadsAtom,
|
|
} from '@/helpers/atoms/Conversation.atom'
|
|
|
|
export default function EventHandler({ children }: { children: ReactNode }) {
|
|
const addNewMessage = useSetAtom(addNewMessageAtom)
|
|
const updateMessage = useSetAtom(updateMessageAtom)
|
|
|
|
const updateThreadWaiting = useSetAtom(updateThreadWaitingForResponseAtom)
|
|
const threads = useAtomValue(threadsAtom)
|
|
const threadsRef = useRef(threads)
|
|
|
|
useEffect(() => {
|
|
threadsRef.current = threads
|
|
}, [threads])
|
|
|
|
async function handleNewMessageResponse(message: ThreadMessage) {
|
|
addNewMessage(message)
|
|
}
|
|
|
|
async function handleMessageResponseUpdate(message: ThreadMessage) {
|
|
updateMessage(
|
|
message.id,
|
|
message.thread_id,
|
|
message.content,
|
|
message.status
|
|
)
|
|
if (message.status === MessageStatus.Ready) {
|
|
// Mark the thread as not waiting for response
|
|
updateThreadWaiting(message.thread_id, false)
|
|
|
|
const thread = threadsRef.current?.find((e) => e.id == message.thread_id)
|
|
if (thread) {
|
|
const messageContent = message.content[0]?.text.value ?? ''
|
|
const metadata = {
|
|
...thread.metadata,
|
|
lastMessage: messageContent,
|
|
}
|
|
extensionManager
|
|
.get<ConversationalExtension>(ExtensionType.Conversational)
|
|
?.saveThread({
|
|
...thread,
|
|
metadata,
|
|
})
|
|
|
|
extensionManager
|
|
.get<ConversationalExtension>(ExtensionType.Conversational)
|
|
?.addNewMessage(message)
|
|
}
|
|
}
|
|
}
|
|
|
|
useEffect(() => {
|
|
if (window.core.events) {
|
|
events.on(EventName.OnMessageResponse, handleNewMessageResponse)
|
|
events.on(EventName.OnMessageUpdate, handleMessageResponseUpdate)
|
|
}
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, [])
|
|
|
|
useEffect(() => {
|
|
return () => {
|
|
events.off(EventName.OnMessageResponse, handleNewMessageResponse)
|
|
events.off(EventName.OnMessageUpdate, handleMessageResponseUpdate)
|
|
}
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, [])
|
|
return <>{children}</>
|
|
}
|