Fix Embedding Dimension Parameter Not Being Passed (#2304)

2025-03-05 22:53:36 +08:00
parent dd1f2989bc
commit 8c6d16a6f0
4 changed files with 9 additions and 9 deletions
--- a/mem0/embeddings/gemini.py
+++ b/mem0/embeddings/gemini.py
@@ -28,5 +28,5 @@ class GoogleGenAIEmbedding(EmbeddingBase):
            list: The embedding vector.
        """
        text = text.replace("\n", " ")
-        response = genai.embed_content(model=self.config.model, content=text)
+        response = genai.embed_content(model=self.config.model, content=text, output_dimensionality=self.config.embedding_dims)
        return response["embedding"]