Adding Gemini (#1862)

2024-09-27 11:46:40 -05:00
parent 699741c760
commit aaf8e6e7ff
9 changed files with 379 additions and 4 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -41,6 +41,7 @@ class EmbedderFactory:
        "ollama": "mem0.embeddings.ollama.OllamaEmbedding",
        "huggingface": "mem0.embeddings.huggingface.HuggingFaceEmbedding",
        "azure_openai": "mem0.embeddings.azure_openai.AzureOpenAIEmbedding",
+        "gemini": "mem0.embeddings.gemini.GoogleGenAIEmbedding",
    }

    @classmethod