From 8cfd185c7584e7a792811833c77b95bc62a5a5cf Mon Sep 17 00:00:00 2001 From: choccccy Date: Sat, 22 Jul 2023 15:56:53 -0600 Subject: [PATCH] seach with an embedded_query, not an embedded_text --- pylib/embedding_helper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pylib/embedding_helper.py b/pylib/embedding_helper.py index 65e9567..12c264d 100644 --- 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): '''