JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(embeddings): set FastEmbed embedding_dims from model metadata at init (#4711)

Browse Source
This commit is contained in:
Prithvi Monangi
2026-04-17 18:17:21 +05:30
committed by GitHub
parent 3fbc1c9aef
commit 2ec3c4ab20
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/embeddings/fastembed.py
+4 -1
View File
@@ -13,7 +13,10 @@ class FastEmbedEmbedding(EmbeddingBase):
super().__init__(config) super().__init__(config)
self.config.model = self.config.model or "thenlper/gte-large" self.config.model = self.config.model or "thenlper/gte-large"
self.dense_model = TextEmbedding(model_name = self.config.model) self.dense_model = TextEmbedding(model_name=self.config.model)
if not self.config.embedding_dims:
self.config.embedding_dims = self.dense_model.embedding_size
def embed(self, text, memory_action: Optional[Literal["add", "search", "update"]] = None): def embed(self, text, memory_action: Optional[Literal["add", "search", "update"]] = None):
""" """