Add support for pgvector (#1675)

2024-08-13 00:15:08 +05:30
parent 629bb5bb63
commit 6cc4a31e91
8 changed files with 352 additions and 81 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -52,6 +52,7 @@ class VectorStoreFactory:
    provider_to_class = {
        "qdrant": "mem0.vector_stores.qdrant.Qdrant",
        "chroma": "mem0.vector_stores.chroma.ChromaDB",
+        "pgvector": "mem0.vector_stores.pgvector.PGVector"
    }

    @classmethod