Add ChromaDB support (#1612)

2024-08-01 22:16:35 +05:30
parent e585d3c1cc
commit 45ae1f0313
9 changed files with 452 additions and 148 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -44,4 +44,18 @@ class EmbedderFactory:
            return embedder_instance
        else:
            raise ValueError(f"Unsupported Embedder provider: {provider_name}")
-        
+        
+class VectorStoreFactory:
+    provider_to_class = {
+        "qdrant": "mem0.vector_stores.qdrant.Qdrant",
+        "chromadb": "mem0.vector_stores.chroma.ChromaDB",
+    }
+
+    @classmethod
+    def create(cls, provider_name, config):
+        class_type = cls.provider_to_class.get(provider_name)
+        if class_type:
+            vector_store_instance = load_class(class_type)
+            return vector_store_instance(**config)
+        else:
+            raise ValueError(f"Unsupported VectorStore provider: {provider_name}")