Update with h2oGPT hash 1c8a0cdaca009bbce07e1f9eda2372252bdf57dd

gradio_runner.py

@@ -499,26 +499,27 @@ def go_gradio(**kwargs):
                             pass
                     with gr.Row():
                         with gr.Column(scale=1):
+                            visible_add_remove_collection = (allow_upload_to_user_data or
+                                                             allow_upload_to_my_data) and \
+                                                            kwargs['langchain_mode'] != 'Disabled'
                             add_placeholder = "e.g. UserData2, user_path2 (optional)" \
                                 if not is_public else "e.g. MyData2"
                             remove_placeholder = "e.g. UserData2" if not is_public else "e.g. MyData2"
-                            new_langchain_mode_text = gr.Textbox(value="", visible=allow_upload_to_user_data or
-                                                                                   allow_upload_to_my_data,
+                            new_langchain_mode_text = gr.Textbox(value="", visible=visible_add_remove_collection,
                                                                  label='Add Collection',
                                                                  placeholder=add_placeholder,
                                                                  interactive=True)
-                            remove_langchain_mode_text = gr.Textbox(value="", visible=allow_upload_to_user_data or
-                                                                                      allow_upload_to_my_data,
+                            remove_langchain_mode_text = gr.Textbox(value="", visible=visible_add_remove_collection,
                                                                     label='Remove Collection',
                                                                     placeholder=remove_placeholder,
                                                                     interactive=True)
                             load_langchain = gr.Button(value="Load LangChain State", scale=0, size='sm',
-                                                       visible=allow_upload_to_user_data)
+                                                       visible=allow_upload_to_user_data and
+                                                               kwargs['langchain_mode'] != 'Disabled')
                         with gr.Column(scale=1):
                             df0 = get_df_langchain_mode_paths(selection_docs_state0)
                             langchain_mode_path_text = gr.Dataframe(value=df0,
-                                                                    visible=allow_upload_to_user_data or
-                                                                            allow_upload_to_my_data,
+                                                                    visible=visible_add_remove_collection,
                                                                     label='LangChain Mode-Path',
                                                                     show_label=False,
                                                                     interactive=False)
@@ -2543,7 +2544,7 @@ def get_sources(db1s, langchain_mode, dbs=None, docs_state0=None):
     for k in db1s:
         set_userid(db1s[k])
 
-    if langchain_mode in ['ChatLLM', 'LLM']:
+    if langchain_mode in ['LLM']:
         source_files_added = "NA"
         source_list = []
     elif langchain_mode in ['wiki_full']:
@@ -2787,7 +2788,7 @@ def get_db(db1s, langchain_mode, dbs=None):
 
 def get_source_files_given_langchain_mode(db1s, langchain_mode='UserData', dbs=None):
     db = get_db(db1s, langchain_mode, dbs=dbs)
-    if langchain_mode in ['ChatLLM', 'LLM'] or db is None:
+    if langchain_mode in ['LLM'] or db is None:
         return "Sources: N/A"
     return get_source_files(db=db, exceptions=None)
 

utils.py

@@ -1023,7 +1023,7 @@ def save_collection_names(langchain_modes, visible_langchain_modes, langchain_mo
     scratch_collection_names = list(db1s.keys())
     user_hash = db1s.get(LangChainMode.MY_DATA.value, '')[1]
 
-    llms = ['ChatLLM', 'LLM', 'Disabled']
+    llms = ['LLM', 'Disabled']
 
     scratch_langchain_modes = [x for x in langchain_modes if x in scratch_collection_names]
     scratch_visible_langchain_modes = [x for x in visible_langchain_modes if x in scratch_collection_names]