55 updateSettings ,
66} from '../providers'
77import { models as providerModels } from 'token.js'
8- import { predefinedProviders } from '@/mock/data '
8+ import { predefinedProviders } from '@/consts/providers '
99import { EngineManager } from '@janhq/core'
1010import { fetchModels } from '../models'
1111import { ExtensionManager } from '@/lib/extension'
@@ -21,7 +21,7 @@ vi.mock('token.js', () => ({
2121 } ,
2222} ) )
2323
24- vi . mock ( '@/mock/data ' , ( ) => ( {
24+ vi . mock ( '@/consts/providers ' , ( ) => ( {
2525 predefinedProviders : [
2626 {
2727 active : true ,
@@ -69,6 +69,7 @@ vi.mock('../models', () => ({
6969 { id : 'llama-2-7b' , name : 'Llama 2 7B' , description : 'Llama model' } ,
7070 ] )
7171 ) ,
72+ isToolSupported : vi . fn ( ( ) => Promise . resolve ( false ) ) ,
7273} ) )
7374
7475vi . mock ( '@/lib/extension' , ( ) => ( {
@@ -116,7 +117,7 @@ describe('providers service', () => {
116117 it ( 'should return builtin and runtime providers' , async ( ) => {
117118 const providers = await getProviders ( )
118119
119- expect ( providers ) . toHaveLength ( 9 ) // 8 runtime + 1 builtin
120+ expect ( providers ) . toHaveLength ( 2 ) // 1 runtime + 1 builtin (mocked)
120121 expect ( providers . some ( ( p ) => p . provider === 'llamacpp' ) ) . toBe ( true )
121122 expect ( providers . some ( ( p ) => p . provider === 'openai' ) ) . toBe ( true )
122123 } )
@@ -156,7 +157,7 @@ describe('providers service', () => {
156157 provider : 'openai' ,
157158 base_url : 'https://api.openai.com/v1' ,
158159 api_key : 'test-key' ,
159- } as ModelProvider
160+ }
160161
161162 const models = await fetchModelsFromProvider ( provider )
162163
@@ -185,7 +186,7 @@ describe('providers service', () => {
185186 provider : 'custom' ,
186187 base_url : 'https://api.custom.com' ,
187188 api_key : '' ,
188- } as ModelProvider
189+ }
189190
190191 const models = await fetchModelsFromProvider ( provider )
191192
@@ -204,7 +205,7 @@ describe('providers service', () => {
204205 const provider = {
205206 provider : 'custom' ,
206207 base_url : 'https://api.custom.com' ,
207- } as ModelProvider
208+ }
208209
209210 const models = await fetchModelsFromProvider ( provider )
210211
@@ -214,7 +215,7 @@ describe('providers service', () => {
214215 it ( 'should throw error when provider has no base_url' , async ( ) => {
215216 const provider = {
216217 provider : 'custom' ,
217- } as ModelProvider
218+ }
218219
219220 await expect ( fetchModelsFromProvider ( provider ) ) . rejects . toThrow (
220221 'Provider must have base_url configured'
@@ -232,10 +233,10 @@ describe('providers service', () => {
232233 const provider = {
233234 provider : 'custom' ,
234235 base_url : 'https://api.custom.com' ,
235- } as ModelProvider
236+ }
236237
237238 await expect ( fetchModelsFromProvider ( provider ) ) . rejects . toThrow (
238- 'Cannot connect to custom at https://api.custom.com'
239+ 'Cannot connect to custom at https://api.custom.com. Please check that the service is running and accessible. '
239240 )
240241 } )
241242
@@ -245,10 +246,10 @@ describe('providers service', () => {
245246 const provider = {
246247 provider : 'custom' ,
247248 base_url : 'https://api.custom.com' ,
248- } as ModelProvider
249+ }
249250
250251 await expect ( fetchModelsFromProvider ( provider ) ) . rejects . toThrow (
251- 'Cannot connect to custom at https://api.custom.com'
252+ 'Cannot connect to custom at https://api.custom.com. Please check that the service is running and accessible. '
252253 )
253254 } )
254255
@@ -264,7 +265,7 @@ describe('providers service', () => {
264265 const provider = {
265266 provider : 'custom' ,
266267 base_url : 'https://api.custom.com' ,
267- } as ModelProvider
268+ }
268269
269270 const models = await fetchModelsFromProvider ( provider )
270271
@@ -298,7 +299,7 @@ describe('providers service', () => {
298299 controller_type : 'input' ,
299300 controller_props : { value : 'test-key' } ,
300301 } ,
301- ] as ProviderSetting [ ]
302+ ]
302303
303304 await updateSettings ( 'openai' , settings )
304305
@@ -324,7 +325,7 @@ describe('providers service', () => {
324325 mockExtensionManager
325326 )
326327
327- const settings = [ ] as ProviderSetting [ ]
328+ const settings = [ ]
328329
329330 const result = await updateSettings ( 'nonexistent' , settings )
330331
@@ -350,7 +351,7 @@ describe('providers service', () => {
350351 controller_type : 'input' ,
351352 controller_props : { value : undefined } ,
352353 } ,
353- ] as ProviderSetting [ ]
354+ ]
354355
355356 await updateSettings ( 'openai' , settings )
356357
0 commit comments