Add ollama embeddings (#1634)

2024-08-03 10:55:40 +05:30
parent 1c46fddce1
commit 024089d33e
7 changed files with 57 additions and 16 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -33,8 +33,7 @@ class LlmFactory:
 class EmbedderFactory:
    provider_to_class = {
        "openai": "mem0.embeddings.openai.OpenAIEmbedding",
-        "ollama": "mem0.embeddings.ollama.OllamaEmbedding",
-        "huggingface": "mem0.embeddings.huggingface.HuggingFaceEmbedding"
+        "ollama": "mem0.embeddings.ollama.OllamaEmbedding"
    }

    @classmethod