HNSW support for pgvector (#2139)

2025-01-11 23:46:42 +05:30
parent 6851fac327
commit 7397279872
3 changed files with 17 additions and 5 deletions
--- a/mem0/configs/vector_stores/pgvector.py
+++ b/mem0/configs/vector_stores/pgvector.py
@@ -12,6 +12,7 @@ class PGVectorConfig(BaseModel):
    host: Optional[str] = Field(None, description="Database host. Default is localhost")
    port: Optional[int] = Field(None, description="Database port. Default is 1536")
    diskann: Optional[bool] = Field(True, description="Use diskann for approximate nearest neighbors search")
+    hnsw: Optional[bool] = Field(False, description="Use hnsw for faster search")

    @model_validator(mode="before")
    def check_auth_and_connection(cls, values):