diff --git a/libs/ktem/ktem/index/file/ui.py b/libs/ktem/ktem/index/file/ui.py index e973c21..834259e 100644 --- a/libs/ktem/ktem/index/file/ui.py +++ b/libs/ktem/ktem/index/file/ui.py @@ -187,6 +187,20 @@ class FileIndexPage(BasePage): ) with gr.Row(): self.is_zipped_state = gr.State(value=False) + + self.download_single_button = gr.DownloadButton( + "Download file", + visible=False, + ) + + with gr.Row() as self.selection_info: + self.selected_file_id = gr.State(value=None) + with gr.Column(scale=2): + self.selected_panel = gr.Markdown(self.selected_panel_false) + + self.chunks = gr.HTML(visible=False) + + with gr.Accordion("Advance Options", open=False): with gr.Row(): self.download_all_button = gr.DownloadButton( "Download all files", @@ -204,18 +218,6 @@ class FileIndexPage(BasePage): "Cancel", visible=False ) - self.download_single_button = gr.DownloadButton( - "Download file", - visible=False, - ) - - with gr.Row() as self.selection_info: - self.selected_file_id = gr.State(value=None) - with gr.Column(scale=2): - self.selected_panel = gr.Markdown(self.selected_panel_false) - - self.chunks = gr.HTML(visible=False) - def on_subscribe_public_events(self): """Subscribe to the declared public event of the app""" if self._app.f_user_management: diff --git a/libs/ktem/ktem/rerankings/ui.py b/libs/ktem/ktem/rerankings/ui.py index 311a794..6ac8fee 100644 --- a/libs/ktem/ktem/rerankings/ui.py +++ b/libs/ktem/ktem/rerankings/ui.py @@ -7,6 +7,8 @@ from ktem.app import BasePage from ktem.utils.file import YAMLNoDateSafeLoader from theflow.utils.modules import deserialize +from kotaemon.base import Document + from .manager import reranking_models_manager @@ -348,7 +350,7 @@ class RerankingManagement(BasePage): log_content += "- Sending a message ([`Hello`], `Hi`)
" yield log_content - _ = rerank(["Hello"], "Hi") + _ = rerank([Document(content="Hello")], "Hi") log_content += ( "- Connection success. "