chore: Change embed to query_embed in fastembed mixin for sparse models

qdrant · Oct 2, 2024 · 5a201a0 · 5a201a0
1 parent 2c63890
commit 5a201a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/qdrant_client/async_qdrant_fastembed.py b/qdrant_client/async_qdrant_fastembed.py
@@ -631,7 +631,7 @@ async def query(
         sparse_embedding_model_inst = self._get_or_init_sparse_model(
             model_name=self.sparse_embedding_model_name
         )
-        sparse_vector = list(sparse_embedding_model_inst.embed(documents=query_text))[0]
+        sparse_vector = list(sparse_embedding_model_inst.query_embed(query=query_text))[0]
         sparse_query_vector = models.SparseVector(
             indices=sparse_vector.indices.tolist(), values=sparse_vector.values.tolist()
         )

diff --git a/qdrant_client/qdrant_fastembed.py b/qdrant_client/qdrant_fastembed.py
@@ -695,7 +695,7 @@ def query(
         sparse_embedding_model_inst = self._get_or_init_sparse_model(
             model_name=self.sparse_embedding_model_name
         )
-        sparse_vector = list(sparse_embedding_model_inst.embed(documents=query_text))[0]
+        sparse_vector = list(sparse_embedding_model_inst.query_embed(query=query_text))[0]
         sparse_query_vector = models.SparseVector(
             indices=sparse_vector.indices.tolist(),
             values=sparse_vector.values.tolist(),