Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Add OnModelStopped Event

Browse Source
This commit is contained in:
hiro 2023-12-05 00:25:50 +07:00
parent a8e33c2a97
commit a51b206cb8
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
web/containers/Providers/EventHandler.tsx
View File

@ -7,6 +7,7 @@ import {
ThreadMessage, ThreadMessage,
ExtensionType, ExtensionType,
MessageStatus, MessageStatus,
Model,
} from '@janhq/core' } from '@janhq/core'
import { ConversationalExtension } from '@janhq/core' import { ConversationalExtension } from '@janhq/core'
import { useAtomValue, useSetAtom } from 'jotai' import { useAtomValue, useSetAtom } from 'jotai'
@ -64,13 +65,13 @@ export default function EventHandler({ children }: { children: ReactNode }) {
})) }))
} }
async function handleModelStop(res: any) { async function handleModelStopped(model: Model) {
setTimeout(async () => { setTimeout(async () => {
setActiveModel(undefined) setActiveModel(undefined)
setStateModel({ state: 'start', loading: false, model: '' }) setStateModel({ state: 'start', loading: false, model: '' })
toaster({ toaster({
title: 'Success!', title: 'Success!',
description: `Model ${res.modelId} has been stopped.`, description: `Model ${model.id} has been stopped.`,
}) })
}, 500) }, 500)
} }
@ -123,7 +124,7 @@ export default function EventHandler({ children }: { children: ReactNode }) {
events.on(EventName.OnMessageUpdate, handleMessageResponseUpdate) events.on(EventName.OnMessageUpdate, handleMessageResponseUpdate)
events.on(EventName.OnModelReady, handleModelReady) events.on(EventName.OnModelReady, handleModelReady)
events.on(EventName.OnModelFail, handleModelFail) events.on(EventName.OnModelFail, handleModelFail)
events.on(EventName.OnModelStop, handleModelStop) events.on(EventName.OnModelStopped, handleModelStopped)
} }
// eslint-disable-next-line react-hooks/exhaustive-deps // eslint-disable-next-line react-hooks/exhaustive-deps
}, []) }, [])