Fix docs and config for vector store

This commit is contained in:
Dev-Khant
2024-08-04 21:10:51 +05:30
parent b6cfd960d1
commit b10ec8c34a
3 changed files with 9 additions and 7 deletions

View File

@@ -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",

View File

@@ -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):

View File

@@ -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)