(bug-fix) : fix VertexAI missing configurations (#1926)

2024-10-03 21:34:14 +05:30
parent c09c4926a7
commit d107b639b3
5 changed files with 15 additions and 8 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -42,6 +42,7 @@ class EmbedderFactory:
        "huggingface": "mem0.embeddings.huggingface.HuggingFaceEmbedding",
        "azure_openai": "mem0.embeddings.azure_openai.AzureOpenAIEmbedding",
        "gemini": "mem0.embeddings.gemini.GoogleGenAIEmbedding",
+        "vertexai": "mem0.embeddings.vertexai.VertexAIEmbedding",
    }

    @classmethod