From 7c63914e64885b4570c475e1b1c653a78faeda76 Mon Sep 17 00:00:00 2001 From: Louis Date: Mon, 30 Sep 2024 14:34:27 +0700 Subject: [PATCH] fix: model dropdown should show recommended models to download (#3742) --- web/helpers/atoms/Model.atom.test.ts | 2 +- web/helpers/atoms/Model.atom.ts | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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])