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

janhq · 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
  }
  }