From ffbfaf1fd14e4e8d93488d08fbfd5101a05d6769 Mon Sep 17 00:00:00 2001
From: hiro <vuonghoainam.work@gmail.com>
Date: Tue, 5 Dec 2023 00:26:19 +0700
Subject: [PATCH] feat: Add Event OnModelStop emission to web

---
 web/hooks/useActiveModel.ts | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/web/hooks/useActiveModel.ts b/web/hooks/useActiveModel.ts
index 3a1343489..699b16279 100644
--- a/web/hooks/useActiveModel.ts
+++ b/web/hooks/useActiveModel.ts
@@ -1,8 +1,6 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import {
   EventName,
-  ExtensionType,
-  InferenceExtension,
   events,
 } from '@janhq/core'
 import { Model, ModelSettingParams } from '@janhq/core'
@@ -60,8 +58,9 @@ export function useActiveModel() {
   }
 
   const stopModel = async (modelId: string) => {
+    const model = downloadedModels.find((e) => e.id === modelId)
     setStateModel({ state: 'stop', loading: true, model: modelId })
-    events.emit(EventName.OnModelStop, modelId)
+    events.emit(EventName.OnModelStop, model)
   }
 
   return { activeModel, startModel, stopModel, stateModel }