fix: properly fixed model list not updating

Vali-98 · Jul 19, 2024 · c530008 · c530008
1 parent fa3fe73
commit c530008
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/components/Endpoint/Local.tsx b/components/Endpoint/Local.tsx
@@ -73,6 +73,7 @@ const Local = () => {
                     Llama.deleteModel(currentModel ?? '')
                         .then(() => {
                             Logger.log('Model Deleted Successfully', true)
+                            setCurrentModel(undefined)
                             getModels()
                         })
                         .catch(() => Logger.log('Could Not Delete Model', true))
@@ -94,9 +95,7 @@ const Local = () => {
 
     const handleImport = async () => {
         setModelLoading(true)
-        await Llama.importModel().then(() => {
-            getModels()
-        })
+        await Llama.importModel()
         await getModels()
         setModelLoading(false)
     }

diff --git a/constants/LlamaLocal.ts b/constants/LlamaLocal.ts
@@ -269,7 +269,7 @@ export namespace Llama {
                 .catch((error) => {
                     Logger.log(`Import Failed: ${error.message}`, true)
                 })
-        }).catch
+        })
     }
 
     export const getKVSizeMB = async () => {