quickfix - embedding return type

ArturOle · Nov 18, 2024 · 2b10d1d · 2b10d1d
1 parent 0930690
commit 2b10d1d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ContextSearch"
-version = "0.7.0"
+version = "0.7.1"
 description = "User friendly system for semantic search."
 authors = [
     "ArturOle"

diff --git a/src/context_search/data_manager.py b/src/context_search/data_manager.py
@@ -51,7 +51,7 @@ def retrieve_data(self, query, n):
             similarity score.
         """
         embedded_query = self.preprocessor.embedder.embed(query)
-        return self.communicator.search_n_records(embedded_query, n)
+        return self.communicator.search_n_records(embedded_query[0], n)
 
     def insert(self, directories: list) -> None:
         """Inserts data from the given directories to the database.