Fix arg name (#248)

* Fix arg name

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: tellet-q

.github/workflows/continuous-benchmark.yaml

@@ -123,7 +123,7 @@ jobs:
         uses: ./.github/workflows/actions/send-slack-msg
         if: failure() || cancelled()
         with:
-          source_name: "runTenantsBenchmark"
+          bench_name: "runTenantsBenchmark"
           job_status: ${{ job.status }}
           failed_outputs: ${{ steps.benches.outputs.failed }}
           qdrant_version: ${{ steps.benches.outputs.qdrant_version }}
@@ -179,7 +179,7 @@ jobs:
         uses: ./.github/workflows/actions/send-slack-msg
         if: failure() || cancelled()
         with:
-          source_name: "runParallelBenchmark"
+          bench_name: "runParallelBenchmark"
           job_status: ${{ job.status }}
           failed_outputs: ${{ steps.benches.outputs.failed }}
           qdrant_version: ${{ steps.benches.outputs.qdrant_version }}

engine/clients/qdrant/configure.py

@@ -4,7 +4,7 @@
 from benchmark.dataset import Dataset
 from engine.base_client.configure import BaseConfigurator
 from engine.base_client.distances import Distance
-from engine.clients.qdrant.config import QDRANT_COLLECTION_NAME, QDRANT_API_KEY
+from engine.clients.qdrant.config import QDRANT_API_KEY, QDRANT_COLLECTION_NAME
 
 
 class QdrantConfigurator(BaseConfigurator):
@@ -32,7 +32,9 @@ class QdrantConfigurator(BaseConfigurator):
     def __init__(self, host, collection_params: dict, connection_params: dict):
         super().__init__(host, collection_params, connection_params)
 
-        self.client = QdrantClient(url=host, api_key=QDRANT_API_KEY, **connection_params)
+        self.client = QdrantClient(
+            url=host, api_key=QDRANT_API_KEY, **connection_params
+        )
 
     def clean(self):
         self.client.delete_collection(collection_name=QDRANT_COLLECTION_NAME)

engine/clients/qdrant/search.py

@@ -2,13 +2,12 @@
 from typing import List, Tuple
 
 import httpx
-from qdrant_client import QdrantClient
+from qdrant_client import QdrantClient, models
 from qdrant_client._pydantic_compat import construct
-from qdrant_client import models
 
 from dataset_reader.base_reader import Query
 from engine.base_client.search import BaseSearcher
-from engine.clients.qdrant.config import QDRANT_COLLECTION_NAME, QDRANT_API_KEY
+from engine.clients.qdrant.config import QDRANT_API_KEY, QDRANT_COLLECTION_NAME
 from engine.clients.qdrant.parser import QdrantConditionParser
 
 
@@ -48,7 +47,6 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]:
                 values=query.sparse_vector.values,
             )
 
-
         prefetch = cls.search_params.get("prefetch")
 
         if prefetch:
@@ -65,7 +63,9 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]:
                 query=query_vector,
                 query_filter=cls.parser.parse(query.meta_conditions),
                 limit=top,
-                search_params=models.SearchParams(**cls.search_params.get("config", {})),
+                search_params=models.SearchParams(
+                    **cls.search_params.get("config", {})
+                ),
                 with_payload=cls.search_params.get("with_payload", False),
             )
         except Exception as ex:

engine/clients/qdrant/upload.py

@@ -13,7 +13,7 @@
 
 from dataset_reader.base_reader import Record
 from engine.base_client.upload import BaseUploader
-from engine.clients.qdrant.config import QDRANT_COLLECTION_NAME, QDRANT_API_KEY
+from engine.clients.qdrant.config import QDRANT_API_KEY, QDRANT_COLLECTION_NAME
 
 
 class QdrantUploader(BaseUploader):
@@ -24,7 +24,9 @@ class QdrantUploader(BaseUploader):
     def init_client(cls, host, distance, connection_params, upload_params):
         os.environ["GRPC_ENABLE_FORK_SUPPORT"] = "true"
         os.environ["GRPC_POLL_STRATEGY"] = "epoll,poll"
-        cls.client = QdrantClient(url=host, prefer_grpc=True, api_key=QDRANT_API_KEY, **connection_params)
+        cls.client = QdrantClient(
+            url=host, prefer_grpc=True, api_key=QDRANT_API_KEY, **connection_params
+        )
         cls.upload_params = upload_params
 
     @classmethod