fix: fix async grpc info (#800)

qdrant · Oct 7, 2024 · a204d36 · a204d36
1 parent 7225ee8
commit a204d36
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/qdrant_client/async_qdrant_remote.py b/qdrant_client/async_qdrant_remote.py
@@ -2989,7 +2989,7 @@ async def delete_shard_key(
 
     async def info(self) -> types.VersionInfo:
         if self._prefer_grpc:
-            version_info = self.grpc_root.HealthCheck(grpc.HealthCheckRequest())
+            version_info = await self.grpc_root.HealthCheck(grpc.HealthCheckRequest())
             return GrpcToRest.convert_health_check_reply(version_info)
         version_info = await self.rest.service_api.root()
         assert version_info is not None, "Healthcheck returned None"

diff --git a/tools/async_client_generator/remote_generator.py b/tools/async_client_generator/remote_generator.py
@@ -2,7 +2,7 @@
 import inspect
 from typing import Dict, List, Optional
 
-from qdrant_client.grpc import CollectionsStub, PointsStub, SnapshotsStub
+from qdrant_client.grpc import CollectionsStub, PointsStub, SnapshotsStub, QdrantStub
 from qdrant_client.http import AsyncApiClient
 from qdrant_client.http.api.cluster_api import AsyncClusterApi
 from qdrant_client.http.api.collections_api import AsyncCollectionsApi
@@ -94,7 +94,7 @@ def async_methods(self) -> List[str]:
             ):
                 self._async_methods.extend(self.get_async_methods(cls_))
 
-            for cls_ in (PointsStub, SnapshotsStub, CollectionsStub):
+            for cls_ in (PointsStub, SnapshotsStub, CollectionsStub, QdrantStub):
                 self._async_methods.extend(self._get_grpc_methods(cls_))
 
         return self._async_methods