klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Ollama as a supported embedding provider (#1344)

Browse Source
This commit is contained in:
Colin O'Brien
2024-05-02 01:08:47 -04:00
committed by GitHub
parent 1a66f961f4
commit a795798156
9 changed files with 46 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
embedchain/factory.py
View File

@@ -58,6 +58,7 @@ class EmbedderFactory:
"mistralai": "embedchain.embedder.mistralai.MistralAIEmbedder",
"nvidia": "embedchain.embedder.nvidia.NvidiaEmbedder",
"cohere": "embedchain.embedder.cohere.CohereEmbedder",
"ollama": "embedchain.embedder.ollama.OllamaEmbedder",
}
provider_to_config_class = {
"azure_openai": "embedchain.config.embedder.base.BaseEmbedderConfig",
@@ -65,6 +66,7 @@ class EmbedderFactory:
"gpt4all": "embedchain.config.embedder.base.BaseEmbedderConfig",
"huggingface": "embedchain.config.embedder.base.BaseEmbedderConfig",
"openai": "embedchain.config.embedder.base.BaseEmbedderConfig",
"ollama": "embedchain.config.embedder.ollama.OllamaEmbedderConfig",
}
@classmethod