@@ -12,7 +12,6 @@ import {
12
12
unrollAssistantFromContent ,
13
13
validateConfigYaml ,
14
14
} from "@continuedev/config-yaml" ;
15
- import { fetchwithRequestOptions } from "@continuedev/fetch" ;
16
15
17
16
import {
18
17
ContinueConfig ,
@@ -22,14 +21,12 @@ import {
22
21
IdeSettings ,
23
22
} from "../.." ;
24
23
import { slashFromCustomCommand } from "../../commands" ;
25
- import { AllRerankers } from "../../context/allRerankers" ;
26
24
import { MCPManagerSingleton } from "../../context/mcp" ;
27
25
import CodebaseContextProvider from "../../context/providers/CodebaseContextProvider" ;
28
26
import DocsContextProvider from "../../context/providers/DocsContextProvider" ;
29
27
import FileContextProvider from "../../context/providers/FileContextProvider" ;
30
28
import { contextProviderClassFromName } from "../../context/providers/index" ;
31
29
import { ControlPlaneClient } from "../../control-plane/client" ;
32
- import { allEmbeddingsProviders } from "../../indexing/allEmbeddingsProviders" ;
33
30
import FreeTrial from "../../llm/llms/FreeTrial" ;
34
31
import TransformersJsEmbeddingsProvider from "../../llm/llms/TransformersJsEmbeddingsProvider" ;
35
32
import { slashCommandFromPromptFileV1 } from "../../promptFiles/v1/slashCommandFromPromptFile" ;
@@ -247,58 +244,24 @@ async function configYamlToContinueConfig(
247
244
248
245
if ( model . roles ?. includes ( "embed" ) ) {
249
246
const { provider, ...options } = model ;
250
- const embeddingsProviderClass = allEmbeddingsProviders [ provider ] ;
251
- if ( embeddingsProviderClass ) {
252
- if (
253
- embeddingsProviderClass . name === "_TransformersJsEmbeddingsProvider"
254
- ) {
247
+ if ( provider === "transformers.js" ) {
248
+ if ( ideInfo . ideType === "vscode" ) {
255
249
continueConfig . modelsByRole . embed . push (
256
- new embeddingsProviderClass ( ) ,
250
+ new TransformersJsEmbeddingsProvider ( ) ,
257
251
) ;
258
252
} else {
259
- continueConfig . modelsByRole . embed . push (
260
- new embeddingsProviderClass (
261
- {
262
- ...options ,
263
- title : options . name ,
264
- } ,
265
- ( url : string | URL , init : any ) =>
266
- fetchwithRequestOptions ( url , init , {
267
- ...options . requestOptions ,
268
- } ) ,
269
- ) ,
270
- ) ;
253
+ localErrors . push ( {
254
+ fatal : false ,
255
+ message : `Transformers.js embeddings provider not supported in this IDE.` ,
256
+ } ) ;
271
257
}
272
258
} else {
273
- localErrors . push ( {
274
- fatal : false ,
275
- message : `Unsupported embeddings model provider found: ${ provider } ` ,
276
- } ) ;
259
+ continueConfig . modelsByRole . embed . push ( ...llms ) ;
277
260
}
278
261
}
279
262
280
263
if ( model . roles ?. includes ( "rerank" ) ) {
281
- const { provider, ...options } = model ;
282
- const rerankerClass = AllRerankers [ provider ] ;
283
- if ( rerankerClass ) {
284
- continueConfig . modelsByRole . rerank . push (
285
- new rerankerClass (
286
- {
287
- ...options ,
288
- title : options . name ,
289
- } ,
290
- ( url : string | URL , init : any ) =>
291
- fetchwithRequestOptions ( url , init , {
292
- ...options . requestOptions ,
293
- } ) ,
294
- ) ,
295
- ) ;
296
- } else {
297
- localErrors . push ( {
298
- fatal : false ,
299
- message : `Unsupported reranking model provider found: ${ provider } ` ,
300
- } ) ;
301
- }
264
+ continueConfig . modelsByRole . rerank . push ( ...llms ) ;
302
265
}
303
266
} catch ( e ) {
304
267
localErrors . push ( {
0 commit comments