chore: poetry.lock

Signed-off-by: Anush008 <[email protected]>
mem0ai · Dec 11, 2024 · 396ed4e · 396ed4e
1 parent c8a555a
commit 396ed4e
Show file tree

Hide file tree

Showing 3 changed files with 82 additions and 40 deletions.
diff --git a/embedchain/poetry.lock b/embedchain/poetry.lock
diff --git a/embedchain/pyproject.toml b/embedchain/pyproject.toml
@@ -140,7 +140,7 @@ langchain-cohere = "^0.3.0"
 langchain-community = "^0.3.1"
 langchain-aws = {version = "^0.2.1", optional = true}
 langsmith = "^0.1.17"
-fastembed = { version = "^0.3.6", optional = true, python = ">=3.9,<3.13" }
+fastembed = { version = "^0.4.2", optional = true, python = "<3.13,>=3.8.0" }
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.3.0"

diff --git a/mem0/embeddings/fastembed.py b/mem0/embeddings/fastembed.py
@@ -1,3 +1,6 @@
+from typing import Optional
+
+from mem0.configs.embeddings.base import BaseEmbedderConfig
 from mem0.embeddings.base import EmbeddingBase
 
 try:
@@ -14,13 +17,10 @@ class FastEmbedEmbedding(EmbeddingBase):
     Find the list of supported models at https://qdrant.github.io/fastembed/examples/Supported_Models/.
     """
 
-    def __init__(
-        self,
-        model="BAAI/bge-small-en-v1.5",
-    ) -> None:
-        self.model = model
-        self.dims = 384
-        self._model = TextEmbedding(model_name=model)
+    def __init__(self, config: Optional[BaseEmbedderConfig] = None):
+        super().__init__(config)
+
+        self._model = TextEmbedding(model_name=self.config.model, **self.config.model_kwargs)
 
     def embed(self, text):
         return next(self._model.embed(text)).tolist()