Refactor openai embedder (#1444)

This commit is contained in:
Vatsal Rathod
2024-06-26 13:58:12 -04:00
committed by GitHub
parent f80be2d2ea
commit edaeb78ccf
3 changed files with 32 additions and 16 deletions

View File

@@ -50,7 +50,7 @@ class LlmFactory:
class EmbedderFactory:
provider_to_class = {
"azure_openai": "embedchain.embedder.openai.OpenAIEmbedder",
"azure_openai": "embedchain.embedder.azure_openai.AzureOpenAIEmbedder",
"gpt4all": "embedchain.embedder.gpt4all.GPT4AllEmbedder",
"huggingface": "embedchain.embedder.huggingface.HuggingFaceEmbedder",
"openai": "embedchain.embedder.openai.OpenAIEmbedder",