File tree 3 files changed +15
-8
lines changed
store/global/slices/settings/selectors
3 files changed +15
-8
lines changed Original file line number Diff line number Diff line change @@ -103,18 +103,18 @@ export const DEFAULT_LLM_CONFIG: GlobalLLMConfig = {
103
103
} ,
104
104
ollama : {
105
105
enabled : false ,
106
+ enabledModels : filterEnabledModels ( OllamaProvider ) ,
106
107
endpoint : '' ,
107
- models : filterEnabledModels ( OllamaProvider ) ,
108
108
} ,
109
109
openAI : {
110
110
OPENAI_API_KEY : '' ,
111
111
enabled : true ,
112
- models : filterEnabledModels ( OpenAIProvider ) ,
112
+ enabledModels : filterEnabledModels ( OpenAIProvider ) ,
113
113
} ,
114
114
openrouter : {
115
115
apiKey : '' ,
116
116
enabled : false ,
117
- models : filterEnabledModels ( OpenRouterProvider ) ,
117
+ enabledModels : filterEnabledModels ( OpenRouterProvider ) ,
118
118
} ,
119
119
perplexity : {
120
120
apiKey : '' ,
Original file line number Diff line number Diff line change @@ -16,15 +16,22 @@ describe('modelConfigSelectors', () => {
16
16
settings : {
17
17
languageModel : {
18
18
ollama : {
19
- models : [ 'llava' ] ,
19
+ enabledModels : [ 'llava' ] ,
20
20
} ,
21
21
} ,
22
22
} ,
23
23
} as GlobalSettingsState ) as unknown as GlobalStore ;
24
24
25
25
const ollamaList = modelConfigSelectors . modelSelectList ( s ) . find ( ( r ) => r . id === 'ollama' ) ;
26
26
27
- expect ( ollamaList ?. chatModels ) . toEqual ( [ ] ) ;
27
+ expect ( ollamaList ?. chatModels . find ( ( c ) => c . id === 'llava' ) ) . toEqual ( {
28
+ displayName : 'LLaVA 7B' ,
29
+ functionCall : false ,
30
+ hidden : false ,
31
+ id : 'llava' ,
32
+ tokens : 4000 ,
33
+ vision : true ,
34
+ } ) ;
28
35
} ) ;
29
36
} ) ;
30
37
} ) ;
Original file line number Diff line number Diff line change @@ -21,8 +21,8 @@ export interface OpenAIConfig {
21
21
*/
22
22
customModelName ?: string ;
23
23
enabled : boolean ;
24
+ enabledModels ?: string [ ] ;
24
25
endpoint ?: string ;
25
- models ?: string [ ] ;
26
26
useAzure ?: boolean ;
27
27
}
28
28
@@ -45,15 +45,15 @@ export interface AWSBedrockConfig {
45
45
export interface OllamaConfig {
46
46
customModelName ?: string ;
47
47
enabled ?: boolean ;
48
+ enabledModels : string [ ] ;
48
49
endpoint ?: string ;
49
- models : string [ ] ;
50
50
}
51
51
52
52
export interface OpenRouterConfig {
53
53
apiKey ?: string ;
54
54
customModelName ?: string ;
55
55
enabled ?: boolean ;
56
- models : string [ ] ;
56
+ enabledModels : string [ ] ;
57
57
}
58
58
59
59
export interface TogetherAIConfig {
You can’t perform that action at this time.
0 commit comments