diff --git a/flowsettings.py b/flowsettings.py index 5513f582..384d8f79 100644 --- a/flowsettings.py +++ b/flowsettings.py @@ -299,7 +299,7 @@ GRAPHRAG_INDEX = ( { - "name": "GraphRAG", + "name": "GraphRAG Collection", "config": { "supported_file_types": ( ".png, .jpeg, .jpg, .tiff, .tif, .pdf, .xls, .xlsx, .doc, .docx, " @@ -311,7 +311,7 @@ } if not USE_NANO_GRAPHRAG else { - "name": "NanoGraphRAG", + "name": "NanoGraphRAG Collection", "config": { "supported_file_types": ( ".png, .jpeg, .jpg, .tiff, .tif, .pdf, .xls, .xlsx, .doc, .docx, " @@ -325,7 +325,7 @@ KH_INDICES = [ { - "name": "File", + "name": "File Collection", "config": { "supported_file_types": ( ".png, .jpeg, .jpg, .tiff, .tif, .pdf, .xls, .xlsx, .doc, .docx, " diff --git a/libs/ktem/ktem/main.py b/libs/ktem/ktem/main.py index 00d23f20..deeb415d 100644 --- a/libs/ktem/ktem/main.py +++ b/libs/ktem/ktem/main.py @@ -84,7 +84,7 @@ def ui(self): ) as self._tabs["indices-tab"]: for index in self.index_manager.indices: with gr.Tab( - f"{index.name} Collection", + index.name, elem_id=f"{index.id}-tab", ) as self._tabs[f"{index.id}-tab"]: page = index.get_index_page_ui() diff --git a/libs/ktem/ktem/pages/chat/__init__.py b/libs/ktem/ktem/pages/chat/__init__.py index 02b0819d..57d8f665 100644 --- a/libs/ktem/ktem/pages/chat/__init__.py +++ b/libs/ktem/ktem/pages/chat/__init__.py @@ -101,7 +101,7 @@ def on_building_ui(self): index_ui.unrender() # need to rerender later within Accordion with gr.Accordion( - label=f"{index.name} Collection", open=index_id < 1 + label=index.name, open=index_id < 1 ): index_ui.render() gr_index = index_ui.as_gradio_component() diff --git a/libs/ktem/ktem/pages/settings.py b/libs/ktem/ktem/pages/settings.py index f60d8668..b74d641f 100644 --- a/libs/ktem/ktem/pages/settings.py +++ b/libs/ktem/ktem/pages/settings.py @@ -272,7 +272,7 @@ def index_tab(self): id2name = {k: v.name for k, v in self._app.index_manager.info().items()} with gr.Tab("Retrieval settings", visible=self._render_index_tab): for pn, sig in self._default_settings.index.options.items(): - name = "{} Collection".format(id2name.get(pn, f"")) + name = id2name.get(pn, f"") with gr.Tab(name): for n, si in sig.settings.items(): obj = render_setting_item(si, si.value)