diff --git a/web/helpers/atoms/Model.atom.test.ts b/web/helpers/atoms/Model.atom.test.ts index 36f2ce71c..4ab02cad9 100644 --- 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 index 28a6384eb..c817ee74b 100644 --- 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(undefined) -export const showEngineListModelAtom = atom([]) +export const showEngineListModelAtom = atom([InferenceEngine.nitro])