retrieve correction

ArturOle · Nov 18, 2024 · 9c86dd2 · 9c86dd2
1 parent 34ecb39
commit 9c86dd2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/ragger_example_retrieve.py b/examples/ragger_example_retrieve.py
@@ -4,7 +4,7 @@
 
 import os
 
-from src.data_manager import DataManager
+from context_search.data_manager import DataManager
 
 
 cwd = os.getcwd()

diff --git a/examples/ragger_example_submit.py b/examples/ragger_example_submit.py
@@ -4,7 +4,7 @@
 
 import os
 
-from src.data_manager import DataManager
+from context_search.data_manager import DataManager
 
 
 cwd = os.getcwd()

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.