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

Fix Embedding Dimension Parameter Not Being Passed (#2304)

Browse Source
This commit is contained in:
Rafael Nico T. Maniquiz
2025-03-05 22:53:36 +08:00
committed by GitHub
parent dd1f2989bc
commit 8c6d16a6f0
4 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mem0/embeddings/gemini.py
View File

@@ -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"]

2
mem0/embeddings/openai.py
View File

@@ -29,4 +29,4 @@ class OpenAIEmbedding(EmbeddingBase):
list: The embedding vector.
"""
text = text.replace("\n", " ")
return self.client.embeddings.create(input=[text], model=self.config.model).data[0].embedding
return self.client.embeddings.create(input=[text], model=self.config.model, dimensions = self.config.embedding_dims).data[0].embedding