From 2ec3c4ab205180d877de759d1f3b0c0b58642933 Mon Sep 17 00:00:00 2001 From: Prithvi Monangi Date: Fri, 17 Apr 2026 05:47:21 -0700 Subject: [PATCH] fix(embeddings): set FastEmbed embedding_dims from model metadata at init (#4711) --- mem0/embeddings/fastembed.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mem0/embeddings/fastembed.py b/mem0/embeddings/fastembed.py index 83868f283..fadaedb8c 100644 --- a/mem0/embeddings/fastembed.py +++ b/mem0/embeddings/fastembed.py @@ -13,7 +13,10 @@ class FastEmbedEmbedding(EmbeddingBase): super().__init__(config) 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): """