@@ -107,7 +107,7 @@ export class QuickAddApi {
107
107
ai : {
108
108
prompt : async (
109
109
prompt : string ,
110
- model : Model ,
110
+ model : Model | string ,
111
111
settings ?: Partial < {
112
112
variableName : string ;
113
113
shouldAssignVariables : boolean ;
@@ -131,17 +131,29 @@ export class QuickAddApi {
131
131
choiceExecutor
132
132
) . format ;
133
133
134
- const modelProvider = getModelProvider ( model . name ) ;
134
+ let _model : Model ;
135
+ if ( typeof model === "string" ) {
136
+ const foundModel = getModelByName ( model ) ;
137
+ if ( ! foundModel ) {
138
+ throw new Error ( `Model '${ model } ' not found.` ) ;
139
+ }
140
+
141
+ _model = foundModel ;
142
+ } else {
143
+ _model = model ;
144
+ }
145
+
146
+ const modelProvider = getModelProvider ( _model . name ) ;
135
147
136
148
if ( ! modelProvider ) {
137
149
throw new Error (
138
- `Model '${ model . name } ' not found in any provider`
150
+ `Model '${ _model . name } ' not found in any provider`
139
151
) ;
140
152
}
141
153
142
154
const assistantRes = await Prompt (
143
155
{
144
- model,
156
+ model : _model ,
145
157
prompt,
146
158
apiKey : modelProvider . apiKey ,
147
159
modelOptions : settings ?. modelOptions ?? { } ,
@@ -173,7 +185,7 @@ export class QuickAddApi {
173
185
chunkedPrompt : async (
174
186
text : string ,
175
187
promptTemplate : string ,
176
- model : string ,
188
+ model : Model | string ,
177
189
settings ?: Partial < {
178
190
variableName : string ;
179
191
shouldAssignVariables : boolean ;
@@ -201,13 +213,19 @@ export class QuickAddApi {
201
213
choiceExecutor
202
214
) . format ;
203
215
204
- const _model = getModelByName ( model ) ;
216
+ let _model : Model ;
217
+ if ( typeof model === "string" ) {
218
+ const foundModel = getModelByName ( model ) ;
219
+ if ( ! foundModel ) {
220
+ throw new Error ( `Model ${ model } not found.` ) ;
221
+ }
205
222
206
- if ( ! _model ) {
207
- throw new Error ( `Model ${ model } not found.` ) ;
223
+ _model = foundModel ;
224
+ } else {
225
+ _model = model ;
208
226
}
209
227
210
- const modelProvider = getModelProvider ( model ) ;
228
+ const modelProvider = getModelProvider ( _model . name ) ;
211
229
212
230
if ( ! modelProvider ) {
213
231
throw new Error (
0 commit comments