Skip to content

Commit 5886841

Browse files
committed
embed/rerank fixes
1 parent 17c9427 commit 5886841

File tree

2 files changed

+12
-46
lines changed

2 files changed

+12
-46
lines changed

core/config/load.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -423,6 +423,7 @@ async function intermediateToFinalConfig(
423423
if (!embedConfig) {
424424
return null;
425425
}
426+
// config.ts-injected ILLM
426427
if ("providerName" in embedConfig) {
427428
return embedConfig;
428429
}
@@ -464,6 +465,7 @@ async function intermediateToFinalConfig(
464465
if (!rerankingConfig) {
465466
return null;
466467
}
468+
// config.ts-injected ILLM
467469
if ("providerName" in rerankingConfig) {
468470
return rerankingConfig;
469471
}
@@ -488,6 +490,7 @@ async function intermediateToFinalConfig(
488490
};
489491
return new rerankerClass(llmOptions, (url: string | URL, init: any) =>
490492
fetchwithRequestOptions(url, init, {
493+
...config.requestOptions,
491494
...params?.requestOptions,
492495
}),
493496
);

core/config/yaml/loadYaml.ts

Lines changed: 9 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ import {
1212
unrollAssistantFromContent,
1313
validateConfigYaml,
1414
} from "@continuedev/config-yaml";
15-
import { fetchwithRequestOptions } from "@continuedev/fetch";
1615

1716
import {
1817
ContinueConfig,
@@ -22,14 +21,12 @@ import {
2221
IdeSettings,
2322
} from "../..";
2423
import { slashFromCustomCommand } from "../../commands";
25-
import { AllRerankers } from "../../context/allRerankers";
2624
import { MCPManagerSingleton } from "../../context/mcp";
2725
import CodebaseContextProvider from "../../context/providers/CodebaseContextProvider";
2826
import DocsContextProvider from "../../context/providers/DocsContextProvider";
2927
import FileContextProvider from "../../context/providers/FileContextProvider";
3028
import { contextProviderClassFromName } from "../../context/providers/index";
3129
import { ControlPlaneClient } from "../../control-plane/client";
32-
import { allEmbeddingsProviders } from "../../indexing/allEmbeddingsProviders";
3330
import FreeTrial from "../../llm/llms/FreeTrial";
3431
import TransformersJsEmbeddingsProvider from "../../llm/llms/TransformersJsEmbeddingsProvider";
3532
import { slashCommandFromPromptFileV1 } from "../../promptFiles/v1/slashCommandFromPromptFile";
@@ -247,58 +244,24 @@ async function configYamlToContinueConfig(
247244

248245
if (model.roles?.includes("embed")) {
249246
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") {
255249
continueConfig.modelsByRole.embed.push(
256-
new embeddingsProviderClass(),
250+
new TransformersJsEmbeddingsProvider(),
257251
);
258252
} 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+
});
271257
}
272258
} else {
273-
localErrors.push({
274-
fatal: false,
275-
message: `Unsupported embeddings model provider found: ${provider}`,
276-
});
259+
continueConfig.modelsByRole.embed.push(...llms);
277260
}
278261
}
279262

280263
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);
302265
}
303266
} catch (e) {
304267
localErrors.push({

0 commit comments

Comments
 (0)