From 5ca4c4a44d45b1cf8ece1874be2f55750bb04659 Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Wed, 3 Jan 2024 22:23:04 +0800 Subject: [PATCH] add qdrant client timeout limit (#1894) Co-authored-by: jyong --- api/config.py | 1 + api/core/index/vector_index/qdrant_vector_index.py | 2 ++ api/core/index/vector_index/vector_index.py | 3 ++- 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/api/config.py b/api/config.py index 290c9c7b7..e14e8abfe 100644 --- a/api/config.py +++ b/api/config.py @@ -197,6 +197,7 @@ class Config: # qdrant settings self.QDRANT_URL = get_env('QDRANT_URL') self.QDRANT_API_KEY = get_env('QDRANT_API_KEY') + self.QDRANT_CLIENT_TIMEOUT = get_env('QDRANT_CLIENT_TIMEOUT') # milvus / zilliz setting self.MILVUS_HOST = get_env('MILVUS_HOST') diff --git a/api/core/index/vector_index/qdrant_vector_index.py b/api/core/index/vector_index/qdrant_vector_index.py index fdb0b49bb..e79713403 100644 --- a/api/core/index/vector_index/qdrant_vector_index.py +++ b/api/core/index/vector_index/qdrant_vector_index.py @@ -18,6 +18,7 @@ from models.dataset import Dataset, DatasetCollectionBinding class QdrantConfig(BaseModel): endpoint: str api_key: Optional[str] + timeout: float = 20 root_path: Optional[str] def to_qdrant_params(self): @@ -33,6 +34,7 @@ class QdrantConfig(BaseModel): return { 'url': self.endpoint, 'api_key': self.api_key, + 'timeout': self.timeout } diff --git a/api/core/index/vector_index/vector_index.py b/api/core/index/vector_index/vector_index.py index 614f23a29..fe93fad11 100644 --- a/api/core/index/vector_index/vector_index.py +++ b/api/core/index/vector_index/vector_index.py @@ -49,7 +49,8 @@ class VectorIndex: config=QdrantConfig( endpoint=config.get('QDRANT_URL'), api_key=config.get('QDRANT_API_KEY'), - root_path=current_app.root_path + root_path=current_app.root_path, + timeout=config.get('QDRANT_CLIENT_TIMEOUT') ), embeddings=embeddings )