fix: fix remaining async methods

qdrant · Apr 8, 2024 · 8cedf81 · 8cedf81
1 parent ad518ce
commit 8cedf81
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/qdrant_client/async_qdrant_fastembed.py b/qdrant_client/async_qdrant_fastembed.py
@@ -241,7 +241,7 @@ def get_vector_field_name(self) -> str:
         model_name = self.embedding_model_name.split("/")[-1].lower()
         return f"fast-{model_name}"
 
-    async def get_sparse_vector_field_name(self) -> Optional[str]:
+    def get_sparse_vector_field_name(self) -> Optional[str]:
         """
         Returns name of the vector field in qdrant collection, used by current fastembed model.
         Returns:
@@ -360,7 +360,7 @@ def get_fastembed_vector_params(
             )
         }
 
-    async def get_fastembed_sparse_vector_params(
+    def get_fastembed_sparse_vector_params(
         self, on_disk: Optional[bool] = None
     ) -> Optional[Dict[str, models.SparseVectorParams]]:
         """

diff --git a/tools/async_client_generator/fastembed_generator.py b/tools/async_client_generator/fastembed_generator.py
@@ -56,7 +56,9 @@ def get_async_methods(class_obj: type) -> List[str]:
             "set_model",
             "set_sparse_model",
             "get_vector_field_name",
+            "get_sparse_vector_field_name",
             "get_fastembed_vector_params",
+            "get_fastembed_sparse_vector_params",
             "embedding_model_name",
             "sparse_embedding_model_name",
         ],