fix: model dropdown should show recommended models to download (janhq…

…#3742)
destenson · Sep 30, 2024 · 7c63914 · 7c63914
1 parent ba1ddac
commit 7c63914
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/web/helpers/atoms/Model.atom.test.ts b/web/helpers/atoms/Model.atom.test.ts
@@ -38,7 +38,7 @@ describe('Model.atom.ts', () => {
 
   describe('showEngineListModelAtom', () => {
     it('should initialize as an empty array', () => {
-      expect(ModelAtoms.showEngineListModelAtom.init).toEqual([])
+      expect(ModelAtoms.showEngineListModelAtom.init).toEqual(['nitro'])
     })
   })
 

diff --git a/web/helpers/atoms/Model.atom.ts b/web/helpers/atoms/Model.atom.ts
@@ -1,4 +1,4 @@
-import { ImportingModel, Model, ModelFile } from '@janhq/core'
+import { ImportingModel, InferenceEngine, Model, ModelFile } from '@janhq/core'
 import { atom } from 'jotai'
 
 export const stateModel = atom({ state: 'start', loading: false, model: '' })
@@ -133,4 +133,4 @@ export const updateImportingModelAtom = atom(
 
 export const selectedModelAtom = atom<ModelFile | undefined>(undefined)
 
-export const showEngineListModelAtom = atom<string[]>([])
+export const showEngineListModelAtom = atom<string[]>([InferenceEngine.nitro])