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

loom模型服务update_store更新逻辑

Browse Source
This commit is contained in:
glide-the 2024-01-25 15:30:23 +08:00 committed by liunux4odoo
parent 307b973f26
commit 802cfe8805
2 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
webui.py
View File

@ -1,5 +1,6 @@
import streamlit as st import streamlit as st
from webui_pages.loom_view_client import update_store
from webui_pages.openai_plugins import openai_plugins_page from webui_pages.openai_plugins import openai_plugins_page
from webui_pages.utils import * from webui_pages.utils import *
from streamlit_option_menu import option_menu from streamlit_option_menu import option_menu
@ -11,6 +12,11 @@ from configs import VERSION
from server.utils import api_address from server.utils import api_address
def on_change(key):
if key:
update_store()
api = ApiRequest(base_url=api_address()) api = ApiRequest(base_url=api_address())
if __name__ == "__main__": if __name__ == "__main__":
@ -58,6 +64,13 @@ if __name__ == "__main__":
"func": openai_plugins_page, "func": openai_plugins_page,
}, },
} }
# 更新状态
if "status" not in st.session_state \
or "run_plugins_list" not in st.session_state \
or "launch_subscribe_info" not in st.session_state \
or "list_running_models" not in st.session_state \
or "model_config" not in st.session_state:
update_store()
with st.sidebar: with st.sidebar:
st.image( st.image(
@ -76,11 +89,13 @@ if __name__ == "__main__":
default_index = 0 default_index = 0
selected_page = option_menu( selected_page = option_menu(
"", menu_title="",
key="selected_page",
options=options, options=options,
icons=icons, icons=icons,
# menu_icon="chat-quote", # menu_icon="chat-quote",
default_index=default_index, default_index=default_index,
on_change=on_change,
) )
if selected_page in pages: if selected_page in pages:

7
webui_pages/openai_plugins/base.py
View File

@ -11,12 +11,7 @@ from webui_pages.loom_view_client import (
def openai_plugins_page(api: ApiRequest, is_lite: bool = None): def openai_plugins_page(api: ApiRequest, is_lite: bool = None):
if "status" not in st.session_state \
or "run_plugins_list" not in st.session_state \
or "launch_subscribe_info" not in st.session_state \
or "list_running_models" not in st.session_state \
or "model_config" not in st.session_state:
update_store()
with (st.container()): with (st.container()):