From 30aeec1c45495d38f962d2c5fd90dc2dbdd38a40 Mon Sep 17 00:00:00 2001 From: liunux4odoo <41217877+liunux4odoo@users.noreply.github.com> Date: Fri, 15 Sep 2023 01:48:02 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=99=BA=E8=B0=B1API?= =?UTF-8?q?=E9=BB=98=E8=AE=A4=E5=90=8D=E7=A7=B0=EF=BC=8C=E7=BB=99WEBUI?= =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E4=B8=BB=E9=A2=98=20(#1484)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * 給WEBUI设置主题 * chatglm-api更名为zhipu-api --- configs/model_config.py.example | 2 +- configs/server_config.py.example | 2 +- server/model_workers/zhipu.py | 2 +- startup.py | 7 ++++++- webui_pages/knowledge_base/knowledge_base.py | 2 +- 5 files changed, 10 insertions(+), 5 deletions(-) diff --git a/configs/model_config.py.example b/configs/model_config.py.example index 25763d6..5c92348 100644 --- a/configs/model_config.py.example +++ b/configs/model_config.py.example @@ -82,7 +82,7 @@ llm_model_dict = { # 如果没有设置有效的local_model_path，则认为是在线模型API。 # 请在server_config中为每个在线API设置不同的端口 # 具体注册及api key获取请前往 http://open.bigmodel.cn - "chatglm-api": { + "zhipu-api": { "api_base_url": "http://127.0.0.1:8888/v1", "api_key": "", "provider": "ChatGLMWorker", diff --git a/configs/server_config.py.example b/configs/server_config.py.example index 60ed32e..51f53dc 100644 --- a/configs/server_config.py.example +++ b/configs/server_config.py.example @@ -64,7 +64,7 @@ FSCHAT_MODEL_WORKERS = { "baichuan-7b": { # 使用default中的IP和端口 "device": "cpu", }, - "chatglm-api": { # 请为每个在线API设置不同的端口 + "zhipu-api": { # 请为每个在线API设置不同的端口 "port": 20003, }, "minimax-api": { # 请为每个在线API设置不同的端口 diff --git a/server/model_workers/zhipu.py b/server/model_workers/zhipu.py index 63fd3b4..f835ac0 100644 --- a/server/model_workers/zhipu.py +++ b/server/model_workers/zhipu.py @@ -12,7 +12,7 @@ class ChatGLMWorker(ApiModelWorker): def __init__( self, *, - model_names: List[str] = ["chatglm-api"], + model_names: List[str] = ["zhipu-api"], version: Literal["chatglm_pro", "chatglm_std", "chatglm_lite"] = "chatglm_std", controller_addr: str, worker_addr: str, diff --git a/startup.py b/startup.py index c3f1536..e53762d 100644 --- a/startup.py +++ b/startup.py @@ -360,7 +360,12 @@ def run_webui(started_event: mp.Event = None): p = subprocess.Popen(["streamlit", "run", "webui.py", "--server.address", host, - "--server.port", str(port)]) + "--server.port", str(port), + "--theme.base", "light", + "--theme.primaryColor", "#165dff", + "--theme.secondaryBackgroundColor", "#f5f5f5", + "--theme.textColor", "#000000", + ]) started_event.set() p.wait() diff --git a/webui_pages/knowledge_base/knowledge_base.py b/webui_pages/knowledge_base/knowledge_base.py index c4069ab..110b2e4 100644 --- a/webui_pages/knowledge_base/knowledge_base.py +++ b/webui_pages/knowledge_base/knowledge_base.py @@ -130,7 +130,7 @@ def knowledge_base_page(api: ApiRequest): with st.sidebar: chunk_size = st.number_input("单段文本最大长度：", 1, 1000, CHUNK_SIZE) chunk_overlap = st.number_input("相邻文本重合长度：", 0, 500, OVERLAP_SIZE) - zh_title_enhance = st.checkbox("开启中文标题加强：", ZH_TITLE_ENHANCE) + zh_title_enhance = st.checkbox("开启中文标题加强", ZH_TITLE_ENHANCE) # 上传文件 files = st.file_uploader("上传知识文件：",