RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-02 12:46:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix webui: 对话上下文控件变量被错误重置

Browse Source
This commit is contained in:
liunux4odoo 2024-06-11 16:52:34 +08:00
parent 35c2f596f6
commit 41411f3d84
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
libs/chatchat-server/chatchat/webui_pages/dialogue/dialogue.py
View File

@ -23,14 +23,14 @@ chat_box = ChatBox(
)
def save_session():
def save_session(conv_name:str=None):
"""save session state to chat context"""
chat_box.context_from_session(exclude=["selected_page", "prompt", "cur_conv_name"])
chat_box.context_from_session(conv_name, exclude=["selected_page", "prompt", "cur_conv_name"])
def restore_session():
def restore_session(conv_name:str=None):
"""restore sesstion state from chat context"""
chat_box.context_to_session(exclude=["selected_page", "prompt", "cur_conv_name"])
chat_box.context_to_session(conv_name, exclude=["selected_page", "prompt", "cur_conv_name"])
def rerun():
@ -121,7 +121,13 @@ def dialogue_page(
ctx.setdefault("llm_model", DEFAULT_LLM_MODEL)
ctx.setdefault("temperature", TEMPERATURE)
st.session_state.setdefault("cur_conv_name", chat_box.cur_chat_name)
restore_session()
st.session_state.setdefault("last_conv_name", chat_box.cur_chat_name)
# sac on_change callbacks not working since st>=1.34
if st.session_state.cur_conv_name != st.session_state.last_conv_name:
save_session(st.session_state.last_conv_name)
restore_session(st.session_state.cur_conv_name)
st.session_state.last_conv_name = st.session_state.cur_conv_name
# st.write(chat_box.cur_chat_name)
# st.write(st.session_state)
@ -195,7 +201,12 @@ def dialogue_page(
# 会话
cols = st.columns(3)
conv_names = chat_box.get_chat_names()
conversation_name = sac.buttons(conv_names, label="当前会话：", key="cur_conv_name")
def on_conv_change():
print(conversation_name, st.session_state.cur_conv_name)
save_session(conversation_name)
restore_session(st.session_state.cur_conv_name)
conversation_name = sac.buttons(conv_names, label="当前会话：", key="cur_conv_name", on_change=on_conv_change, )
chat_box.use_chat_name(conversation_name)
conversation_id = chat_box.context["uid"]
if cols[0].button("新建", on_click=add_conv):
@ -405,7 +416,4 @@ def dialogue_page(
use_container_width=True,
)
# st.write(chat_box.history)
save_session()
# st.write(chat_box.context)