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