diff --git a/docs/components/vectordb.mdx b/docs/components/vectordb.mdx index 051177b2..dd17e1f8 100644 --- a/docs/components/vectordb.mdx +++ b/docs/components/vectordb.mdx @@ -24,7 +24,7 @@ from mem0 import Memory config = { - "vectordb": { + "vector_store": { "provider": "qdrant", "config": { "collection_name": "test", @@ -50,7 +50,7 @@ from mem0 import Memory config = { - "vectordb": { + "vector_store": { "provider": "chroma", "config": { "collection_name": "test", diff --git a/mem0/configs/vector_stores/chroma.py b/mem0/configs/vector_stores/chroma.py index ad4d7a29..625d0fbf 100644 --- a/mem0/configs/vector_stores/chroma.py +++ b/mem0/configs/vector_stores/chroma.py @@ -13,7 +13,7 @@ class ChromaDbConfig(BaseModel): client: Optional[Client] = Field(None, description="Existing ChromaDB client instance") path: Optional[str] = Field(None, description="Path to the database directory") host: Optional[str] = Field(None, description="Database connection remote host") - port: Optional[str] = Field(None, description="Database connection remote port") + port: Optional[int] = Field(None, description="Database connection remote port") @model_validator(mode="before") def check_host_port_or_path(cls, values): diff --git a/mem0/vector_stores/qdrant.py b/mem0/vector_stores/qdrant.py index fff49252..d4a32adf 100644 --- a/mem0/vector_stores/qdrant.py +++ b/mem0/vector_stores/qdrant.py @@ -44,10 +44,6 @@ class Qdrant(VectorStoreBase): self.client = client else: params = {} - if path: - params["path"] = path - if os.path.exists(path) and os.path.isdir(path): - shutil.rmtree(path) if api_key: params["api_key"] = api_key if url: @@ -55,6 +51,12 @@ class Qdrant(VectorStoreBase): if host and port: params["host"] = host params["port"] = port + + if not params: + params["path"] = path + if os.path.exists(path) and os.path.isdir(path): + shutil.rmtree(path) + self.client = QdrantClient(**params) self.create_col(collection_name, embedding_model_dims)