fix: select all doesn't work

Cinnamon · Jan 13, 2025 · b454f37 · b454f37
1 parent 7551bdf
commit b454f37
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/libs/ktem/ktem/index/file/graph/graph_index.py b/libs/ktem/ktem/index/file/graph/graph_index.py
@@ -25,7 +25,7 @@ def get_indexing_pipeline(self, settings, user_id) -> BaseFileIndexIndexing:
     def get_retriever_pipelines(
         self, settings: dict, user_id: int, selected: Any = None
     ) -> list["BaseFileIndexRetriever"]:
-        _, file_ids, _ = selected
+        file_ids = self._selector_ui.get_selected_ids(selected)
         retrievers = [
             GraphRAGRetrieverPipeline(
                 file_ids=file_ids,

diff --git a/libs/ktem/ktem/index/file/graph/light_graph_index.py b/libs/ktem/ktem/index/file/graph/light_graph_index.py
@@ -28,7 +28,7 @@ def get_indexing_pipeline(self, settings, user_id) -> BaseFileIndexIndexing:
     def get_retriever_pipelines(
         self, settings: dict, user_id: int, selected: Any = None
     ) -> list["BaseFileIndexRetriever"]:
-        _, file_ids, _ = selected
+        file_ids = self._selector_ui.get_selected_ids(selected)
         # retrieval settings
         prefix = f"index.options.{self.id}."
         search_type = settings.get(prefix + "search_type", "local")

diff --git a/libs/ktem/ktem/index/file/graph/nano_graph_index.py b/libs/ktem/ktem/index/file/graph/nano_graph_index.py
@@ -28,7 +28,7 @@ def get_indexing_pipeline(self, settings, user_id) -> BaseFileIndexIndexing:
     def get_retriever_pipelines(
         self, settings: dict, user_id: int, selected: Any = None
     ) -> list["BaseFileIndexRetriever"]:
-        _, file_ids, _ = selected
+        file_ids = self._selector_ui.get_selected_ids(selected)
         # retrieval settings
         prefix = f"index.options.{self.id}."
         search_type = settings.get(prefix + "search_type", "local")