fix: load model fails - generating button is not updated (janhq#2498)

Bura56 · Mar 27, 2024 · fe730fb · fe730fb
1 parent c5c4649
commit fe730fb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/web/hooks/useActiveModel.ts b/web/hooks/useActiveModel.ts
@@ -43,7 +43,10 @@ export function useActiveModel() {
 
     let model = downloadedModelsRef?.current.find((e) => e.id === modelId)
 
-    await stopModel().catch()
+    const error = await stopModel().catch((error: Error) => error)
+    if (error) {
+      return Promise.reject(error)
+    }
 
     setLoadModelError(undefined)
 

diff --git a/web/hooks/useSendChatMessage.ts b/web/hooks/useSendChatMessage.ts
@@ -112,6 +112,7 @@ export default function useSendChatMessage() {
     if (modelRef.current?.id !== modelId) {
       const error = await startModel(modelId).catch((error: Error) => error)
       if (error) {
+        updateThreadWaiting(activeThreadRef.current.id, false)
         return
       }
     }