Change upload points async (#730)

* made upload_points sync

* updated base_client_generator

* fix: remove excess file

* refactoring: fix timeout type, remove redundant import

---------

Co-authored-by: George Panchuk <george.panchuk@qdrant.tech>

qdrant_client/async_client_base.py

@@ -342,7 +342,7 @@ def upload_records(
     ) -> None:
         raise NotImplementedError()
 
-    async def upload_points(
+    def upload_points(
         self, collection_name: str, points: Iterable[types.PointStruct], **kwargs: Any
     ) -> None:
         raise NotImplementedError()

qdrant_client/async_qdrant_client.py

@@ -2162,7 +2162,7 @@ def upload_records(
             shard_key_selector=shard_key_selector,
         )
 
-    async def upload_points(
+    def upload_points(
         self,
         collection_name: str,
         points: Iterable[types.PointStruct],
@@ -2198,7 +2198,7 @@ async def upload_points(
 
         """
         assert len(kwargs) == 0, f"Unknown arguments: {list(kwargs.keys())}"
-        return await self._client.upload_points(
+        return self._client.upload_points(
             collection_name=collection_name,
             points=points,
             batch_size=batch_size,

qdrant_client/async_qdrant_fastembed.py

@@ -503,7 +503,7 @@ async def add(
             ids_accumulator=inserted_ids,
             sparse_vectors=encoded_sparse_docs,
         )
-        await self.upload_points(
+        self.upload_points(
             collection_name=collection_name,
             points=points,
             wait=True,

qdrant_client/async_qdrant_remote.py

@@ -2414,7 +2414,7 @@ def upload_records(
             wait=wait,
         )
 
-    async def upload_points(
+    def upload_points(
         self,
         collection_name: str,
         points: Iterable[types.PointStruct],

qdrant_client/local/async_qdrant_local.py

@@ -930,7 +930,7 @@ async def recreate_collection(
             collection_name, vectors_config, init_from, sparse_vectors_config
         )
 
-    async def upload_points(
+    def upload_points(
         self, collection_name: str, points: Iterable[types.PointStruct], **kwargs: Any
     ) -> None:
         self._upload_points(collection_name, points)

tests/test_async_qdrant_client.py

@@ -1,5 +1,4 @@
 import asyncio
-import os
 import random
 import time
 
@@ -35,7 +34,7 @@ async def test_async_grpc():
         for idx in range(NUM_VECTORS)
     )
 
-    client = QdrantClient(prefer_grpc=True, timeout=3.0)
+    client = QdrantClient(prefer_grpc=True, timeout=3)
 
     grpc_collections = client.async_grpc_collections
 

tools/async_client_generator/base_client_generator.py

@@ -30,7 +30,7 @@ def __init__(
 
     # Parse the code into an AST
     base_client_generator = BaseClientGenerator(
-        keep_sync=["__init__", "upload_records", "upload_collection", "migrate"],
+        keep_sync=["__init__", "upload_records", "upload_collection", "upload_points", "migrate"],
         class_replace_map={"QdrantBase": "AsyncQdrantBase"},
         constant_replace_map={"QdrantBase": "AsyncQdrantBase"},
     )