Skip to content

Commit 1337461

Browse files
committed
fix: Need to download qdrant for default install
Signed-off-by: yangxuan <xuan.yang@zilliz.com>
1 parent cb4712c commit 1337461

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

  • vectordb_bench/backend/clients/qdrant_cloud

vectordb_bench/backend/clients/qdrant_cloud/config.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,12 @@
1+
from typing import TypeVar
2+
13
from pydantic import BaseModel, SecretStr, validator
2-
from qdrant_client.http.models import QuantizationSearchParams, SearchParams
34

45
from ..api import DBCaseConfig, DBConfig, MetricType
56

7+
# define type "SearchParams"
8+
SearchParams = TypeVar("SearchParams")
9+
610

711
# Allowing `api_key` to be left empty, to ensure compatibility with the open-source Qdrant.
812
class QdrantConfig(DBConfig):
@@ -72,6 +76,9 @@ def index_param(self) -> dict:
7276
return {"distance": self.parse_metric()}
7377

7478
def search_param(self) -> SearchParams:
79+
# Import while in use
80+
from qdrant_client.http.models import QuantizationSearchParams, SearchParams
81+
7582
quantization = (
7683
QuantizationSearchParams(
7784
ignore=False,

0 commit comments

Comments
 (0)