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/vector_stores/configs.py
+++ b/mem0/vector_stores/configs.py
@@ -13,7 +13,8 @@ class VectorStoreConfig(BaseModel):

    _provider_configs: Dict[str, str] = {
        "qdrant": "QdrantConfig",
-        "chroma": "ChromaDbConfig"
+        "chroma": "ChromaDbConfig",
+        "pgvector": "PGVectorConfig"
    }

    @model_validator(mode="after")