From 5eed8a5ecaf25810f6a876f629597d4509c91d0f Mon Sep 17 00:00:00 2001 From: NamH Date: Wed, 27 Mar 2024 11:17:43 +0700 Subject: [PATCH] fix: rag is not working for nitro (#2511) Signed-off-by: James Co-authored-by: James --- extensions/assistant-extension/src/node/retrieval.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions/assistant-extension/src/node/retrieval.ts b/extensions/assistant-extension/src/node/retrieval.ts index 01ab413d3..e89357d5c 100644 --- a/extensions/assistant-extension/src/node/retrieval.ts +++ b/extensions/assistant-extension/src/node/retrieval.ts @@ -31,7 +31,6 @@ export class Retrieval { public updateEmbeddingEngine(engine: string): void { // Engine settings are not compatible with the current embedding model params // Switch case manually for now - const settings = readEmbeddingEngine(engine) if (engine === 'nitro') { this.embeddingModel = new OpenAIEmbeddings( { openAIApiKey: 'nitro-embedding' }, @@ -40,6 +39,7 @@ export class Retrieval { ) } else { // Fallback to OpenAI Settings + const settings = readEmbeddingEngine(engine) this.embeddingModel = new OpenAIEmbeddings({ openAIApiKey: settings.api_key, })