seach with an embedded_query, not an embedded_text

OoriData · Jul 22, 2023 · 8cfd185 · 8cfd185
1 parent 95d7a56
commit 8cfd185
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pylib/embedding_helper.py b/pylib/embedding_helper.py
@@ -173,8 +173,8 @@ def search(self, query, **kwargs):
             kwargs: other args to be passed to qdrant_client.QdrantClient.search(). Common ones:
                     limit - maximum number of results to return (useful for top-k query)
         '''
-        embedded_text = self._embedding_model.encode(query)
-        return self.db.search(collection_name=self.name, query_vector=embedded_text, **kwargs)
+        embedded_query = self._embedding_model.encode(query)
+        return self.db.search(collection_name=self.name, query_vector=embedded_query, **kwargs)
 
     def count(self):
         '''