From cdb8dcdb9e016093e3ea4709aa0ad9c46530642d Mon Sep 17 00:00:00 2001
From: Dev Khant <devkhant24@gmail.com>
Date: Mon, 7 Apr 2025 23:29:55 +0530
Subject: [PATCH] Add embeding_dims param to FAISS (#2513)

---
 mem0/configs/vector_stores/faiss.py | 1 +
 mem0/vector_stores/faiss.py         | 9 +++++----
 pyproject.toml                      | 2 +-
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/mem0/configs/vector_stores/faiss.py b/mem0/configs/vector_stores/faiss.py
index 064585c6..b77a8add 100644
--- a/mem0/configs/vector_stores/faiss.py
+++ b/mem0/configs/vector_stores/faiss.py
@@ -12,6 +12,7 @@ class FAISSConfig(BaseModel):
     normalize_L2: bool = Field(
         False, description="Whether to normalize L2 vectors (only applicable for euclidean distance)"
     )
+    embedding_model_dims: int = Field(1536, description="Dimension of the embedding vector")
 
     @model_validator(mode="before")
     @classmethod
diff --git a/mem0/vector_stores/faiss.py b/mem0/vector_stores/faiss.py
index 9a129f12..34405a84 100644
--- a/mem0/vector_stores/faiss.py
+++ b/mem0/vector_stores/faiss.py
@@ -35,6 +35,7 @@ class FAISS(VectorStoreBase):
         path: Optional[str] = None,
         distance_strategy: str = "euclidean",
         normalize_L2: bool = False,
+        embedding_model_dims: int = 1536,
     ):
         """
         Initialize the FAISS vector store.
@@ -51,6 +52,7 @@ class FAISS(VectorStoreBase):
         self.path = path or f"/tmp/faiss/{collection_name}"
         self.distance_strategy = distance_strategy
         self.normalize_L2 = normalize_L2
+        self.embedding_model_dims = embedding_model_dims
 
         # Initialize storage structures
         self.index = None
@@ -145,13 +147,12 @@ class FAISS(VectorStoreBase):
 
         return results
 
-    def create_col(self, name: str, vector_size: int = 1536, distance: str = None):
+    def create_col(self, name: str, distance: str = None):
         """
         Create a new collection.
 
         Args:
             name (str): Name of the collection.
-            vector_size (int, optional): Dimensionality of vectors. Defaults to 1536.
             distance (str, optional): Distance metric to use. Overrides the distance_strategy
                 passed during initialization. Defaults to None.
 
@@ -162,9 +163,9 @@ class FAISS(VectorStoreBase):
 
         # Create index based on distance strategy
         if distance_strategy.lower() == "inner_product" or distance_strategy.lower() == "cosine":
-            self.index = faiss.IndexFlatIP(vector_size)
+            self.index = faiss.IndexFlatIP(self.embedding_model_dims)
         else:
-            self.index = faiss.IndexFlatL2(vector_size)
+            self.index = faiss.IndexFlatL2(self.embedding_model_dims)
 
         self.collection_name = name
 
diff --git a/pyproject.toml b/pyproject.toml
index e2187d13..b6474a82 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mem0ai"
-version = "0.1.85"
+version = "0.1.86"
 description = "Long-term memory for AI Agents"
 authors = ["Mem0 <founders@mem0.ai>"]
 exclude = [