Commit 4dde116 1 parent 911ed64 commit 4dde116 Copy full SHA for 4dde116
File tree 5 files changed +31
-10
lines changed
5 files changed +31
-10
lines changed Original file line number Diff line number Diff line change 313
313
]
314
314
}
315
315
}
316
- }
316
+ }
Original file line number Diff line number Diff line change @@ -56,7 +56,7 @@ export class AIProviderFactory {
56
56
case AIProvider . DOUBAO :
57
57
provider = new DoubaoProvider ( ) ;
58
58
break ;
59
- case AIProvider . Gemini :
59
+ case AIProvider . GEMINI :
60
60
provider = new GeminiAIProvider ( ) ;
61
61
break ;
62
62
default :
Original file line number Diff line number Diff line change @@ -122,8 +122,30 @@ export abstract class BaseOpenAIProvider implements AIProvider {
122
122
}
123
123
}
124
124
125
- async getModels ( ) : Promise < AIModel [ ] > {
126
- return Promise . resolve ( this . config . models ) ;
125
+ async getModels ( ) : Promise < AIModel [ ] | any [ ] > {
126
+ try {
127
+ const response = await this . openai . models . list ( ) ;
128
+ return response . data . map ( ( model : any ) => {
129
+ console . log ( "model" , model ) ;
130
+ return {
131
+ id : model . id ,
132
+ name : model . id ,
133
+ maxTokens : {
134
+ input : model . context_window || 4096 ,
135
+ output : Math . floor ( ( model . context_window || 4096 ) / 2 ) ,
136
+ } ,
137
+ provider : this . provider ,
138
+ } ;
139
+ } ) ;
140
+ } catch ( error ) {
141
+ console . warn ( "Failed to fetch models:" , error ) ;
142
+ return this . config . models ;
143
+ }
144
+ }
145
+
146
+ async refreshModels ( ) : Promise < string [ ] > {
147
+ const models = await this . getModels ( ) ;
148
+ return models . map ( ( m ) => m . id ) ;
127
149
}
128
150
129
151
getName ( ) : string {
@@ -135,5 +157,4 @@ export abstract class BaseOpenAIProvider implements AIProvider {
135
157
}
136
158
137
159
abstract isAvailable ( ) : Promise < boolean > ;
138
- abstract refreshModels ( ) : Promise < string [ ] > ;
139
160
}
Original file line number Diff line number Diff line change @@ -54,7 +54,7 @@ export class GeminiAIProvider extends BaseOpenAIProvider {
54
54
apiKey : configManager . getConfig ( "PROVIDERS_GEMINI_APIKEY" ) ,
55
55
baseURL : "https://generativelanguage.googleapis.com/v1beta/openai/" ,
56
56
providerId : "gemini" ,
57
- providerName : "Gemini AI " ,
57
+ providerName : "Gemini" ,
58
58
models : geminiModels ,
59
59
defaultModel : "gemini-1" ,
60
60
} ) ;
Original file line number Diff line number Diff line change @@ -11,10 +11,10 @@ export abstract class BaseCommand {
11
11
}
12
12
13
13
protected async validateConfig ( ) : Promise < boolean > {
14
- if ( ! ( await ConfigurationManager . getInstance ( ) . validateConfiguration ( ) ) ) {
15
- await NotificationHandler . error ( "command.execution.failed" ) ;
16
- return false ;
17
- }
14
+ // if (!(await ConfigurationManager.getInstance().validateConfiguration())) {
15
+ // await NotificationHandler.error("command.execution.failed");
16
+ // return false;
17
+ // }
18
18
return true ;
19
19
}
20
20
You can’t perform that action at this time.
0 commit comments