Distance metric change and PGVectorScale support (#1703)

2024-08-27 13:26:01 +02:00
parent e8004537c1
commit a94bd11a76
4 changed files with 24 additions and 5 deletions
--- a/mem0/configs/vector_stores/pgvector.py
+++ b/mem0/configs/vector_stores/pgvector.py
@@ -14,6 +14,7 @@ class PGVectorConfig(BaseModel):
    password: Optional[str] = Field(None, description="Database password")
    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")

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