From a51b206cb8e769f2839f9279c071fcd68e7fd3f0 Mon Sep 17 00:00:00 2001
From: hiro <vuonghoainam.work@gmail.com>
Date: Tue, 5 Dec 2023 00:25:50 +0700
Subject: [PATCH] fix: Add OnModelStopped Event

---
 web/containers/Providers/EventHandler.tsx | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/web/containers/Providers/EventHandler.tsx b/web/containers/Providers/EventHandler.tsx
index d24caf34d..533e8cd4a 100644
--- a/web/containers/Providers/EventHandler.tsx
+++ b/web/containers/Providers/EventHandler.tsx
@@ -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
   }, [])