diff --git a/src/LLMProviders/embeddingManager.ts b/src/LLMProviders/embeddingManager.ts index 34198056..beae4e81 100644 --- a/src/LLMProviders/embeddingManager.ts +++ b/src/LLMProviders/embeddingManager.ts @@ -25,7 +25,7 @@ const EMBEDDING_PROVIDER_CONSTRUCTORS = { [EmbeddingModelProviders.GOOGLE]: GoogleGenerativeAIEmbeddings, [EmbeddingModelProviders.AZURE_OPENAI]: AzureOpenAIEmbeddings, [EmbeddingModelProviders.OLLAMA]: OllamaEmbeddings, - [EmbeddingModelProviders.LM_STUDIO]: OpenAIEmbeddings, + [EmbeddingModelProviders.LM_STUDIO]: CustomOpenAIEmbeddings, [EmbeddingModelProviders.OPENAI_FORMAT]: OpenAIEmbeddings, } as const;