fix: use st.rerun() for newer Streamlit versions

examples/webui/code.py

@@ -140,7 +140,11 @@ def format_selected_cb(cb_name: str) -> str:
                 )
                 st.toast(ret.get("msg", " "))
                 st.session_state["selected_cb_name"] = cb_name
-                st.experimental_rerun()
+        # Use st.rerun() for newer Streamlit versions
+        if hasattr(st, 'rerun'):
+            st.rerun()
+        else:
+            st.experimental_rerun()
     elif selected_cb:
         cb = selected_cb
 

examples/webui/dialogue.py

@@ -270,7 +270,11 @@ def on_cb_change():
                 upload2sandbox(interpreter_file, api)
                 st.session_state["interpreter_file_key"] += 1
                 interpreter_file = ""
-                st.experimental_rerun()
+                # Use st.rerun() for newer Streamlit versions
+        if hasattr(st, 'rerun'):
+            st.rerun()
+        else:
+            st.experimental_rerun()
 
             cols[1].download_button(webui_configs["sandbox"]["button_download_name"], 
                                     file_url, file_name)
@@ -520,7 +524,11 @@ def on_cb_change():
 
         # 将上传文件清空
         st.session_state["interpreter_file_key"] += 1
-        st.experimental_rerun()
+        # Use st.rerun() for newer Streamlit versions
+        if hasattr(st, 'rerun'):
+            st.rerun()
+        else:
+            st.experimental_rerun()
 
     now = datetime.now()
     with st.sidebar:
@@ -535,6 +543,10 @@ def on_cb_change():
             GLOBAL_EXE_CODE_TEXT = ""
             if 'history_node_list' in st.session_state:
                 st.session_state['history_node_list'] = []
+            # Use st.rerun() for newer Streamlit versions
+        if hasattr(st, 'rerun'):
+            st.rerun()
+        else:
             st.experimental_rerun()
 
     export_btn.download_button(

examples/webui/document.py

@@ -156,7 +156,11 @@ def format_selected_kb(kb_name: str) -> str:
                 )
                 st.toast(ret.get("msg", " "))
                 st.session_state["selected_kb_name"] = kb_name
-                st.experimental_rerun()
+                # Use st.rerun() for newer Streamlit versions
+                if hasattr(st, 'rerun'):
+                    st.rerun()
+                else:
+                    st.experimental_rerun()
 
     elif selected_kb:
         kb = selected_kb
@@ -322,7 +326,11 @@ def format_selected_kb(kb_name: str) -> str:
                                     #   api_key=llm_model_dict[LLM_MODEL]["api_key"],
                                     #   api_base_url=llm_model_dict[LLM_MODEL]["api_base_url"],
                                       )
-                st.experimental_rerun()
+                # Use st.rerun() for newer Streamlit versions
+                if hasattr(st, 'rerun'):
+                    st.rerun()
+                else:
+                    st.experimental_rerun()
 
             # 将文件从向量库中删除，但不删除文件本身。
             if cols[2].button(
@@ -338,7 +346,11 @@ def format_selected_kb(kb_name: str) -> str:
                                     #   model_device=EMBEDDING_DEVICE,
                                     #   api_key=llm_model_dict[LLM_MODEL]["api_key"],
                                     #   api_base_url=llm_model_dict[LLM_MODEL]["api_base_url"],)
-                st.experimental_rerun()
+                # Use st.rerun() for newer Streamlit versions
+                if hasattr(st, 'rerun'):
+                    st.rerun()
+                else:
+                    st.experimental_rerun()
 
             if cols[3].button(
                     "从知识库中删除",
@@ -354,7 +366,11 @@ def format_selected_kb(kb_name: str) -> str:
                                     #   api_key=llm_model_dict[LLM_MODEL]["api_key"],
                                     #   api_base_url=llm_model_dict[LLM_MODEL]["api_base_url"],)
                     st.toast(ret.get("msg", " "))
-                st.experimental_rerun()
+                # Use st.rerun() for newer Streamlit versions
+                if hasattr(st, 'rerun'):
+                    st.rerun()
+                else:
+                    st.experimental_rerun()
 
         st.divider()
 
@@ -380,7 +396,11 @@ def format_selected_kb(kb_name: str) -> str:
                         st.toast(msg)
                     else:
                         empty.progress(d["finished"] / d["total"], f"正在处理： {d['doc']}")
-                st.experimental_rerun()
+                # Use st.rerun() for newer Streamlit versions
+                if hasattr(st, 'rerun'):
+                    st.rerun()
+                else:
+                    st.experimental_rerun()
 
         if cols[2].button(
                 "删除知识库",

examples/webui/prompt.py

@@ -28,7 +28,11 @@ def prompt_page(api: ApiRequest):
                 "清空prompt",
                 use_container_width=True,
         ):
-            st.experimental_rerun()
+            # Use st.rerun() for newer Streamlit versions
+            if hasattr(st, 'rerun'):
+                st.rerun()
+            else:
+                st.experimental_rerun()
 
     export_btn.download_button(
         "导出记录",