test: correct tests

web/hooks/useImportModel.test.ts

@@ -31,11 +31,11 @@ describe('useImportModel', () => {
     ] as any
 
     await act(async () => {
-      await result.current.importModels(models, 'local' as any)
+      await result.current.importModels(models, 'copy')
     })
 
-    expect(mockImportModels).toHaveBeenCalledWith('1', '/path/to/model1', undefined)
+    expect(mockImportModels).toHaveBeenCalledWith('1', '/path/to/model1', undefined,'copy')
-    expect(mockImportModels).toHaveBeenCalledWith('2', '/path/to/model2', undefined)
+    expect(mockImportModels).toHaveBeenCalledWith('2', '/path/to/model2', undefined, 'copy')
   })
 
   it('should update model info successfully', async () => {

web/hooks/useModels.test.ts

@@ -35,6 +35,8 @@ describe('useModels', () => {
           }),
         }),
         get: () => undefined,
+        has: () => true,
+        // set: () => {}
       },
     })
 

web/hooks/useModels.ts

@@ -58,13 +58,13 @@ const useModels = () => {
       }
     }
 
-    const getExtensionModels = async () => {
+    const getExtensionModels = () => {
       const models = ModelManager.instance().models.values().toArray()
       setExtensionModels(models)
     }
-
     // Fetch all data
-    getExtensionModels().then(getDownloadedModels)
+    getExtensionModels()
+    getDownloadedModels()
   }, [setDownloadedModels, setExtensionModels])
 
   const reloadData = useDebouncedCallback(() => getData(), 300)