diff --git a/configs/__init__.py b/configs/__init__.py index 64289ae..adec2c0 100644 --- a/configs/__init__.py +++ b/configs/__init__.py @@ -1,2 +1,4 @@ from .model_config import * -from .server_config import * \ No newline at end of file +from .server_config import * + +VERSION = "v0.2.1-preview" diff --git a/server/api.py b/server/api.py index c398f15..ecadd7c 100644 --- a/server/api.py +++ b/server/api.py @@ -6,6 +6,7 @@ sys.path.append(os.path.dirname(os.path.dirname(__file__))) from configs.model_config import NLTK_DATA_PATH from configs.server_config import OPEN_CROSS_DOMAIN +from configs import VERSION import argparse import uvicorn from fastapi.middleware.cors import CORSMiddleware @@ -15,11 +16,10 @@ from server.chat import (chat, knowledge_base_chat, openai_chat, from server.knowledge_base.kb_api import list_kbs, create_kb, delete_kb from server.knowledge_base.kb_doc_api import (list_docs, upload_doc, delete_doc, update_doc, download_doc, recreate_vector_store, - search_docs, DocumentWithScore) + search_docs, DocumentWithScore) from server.utils import BaseResponse, ListResponse, FastAPI, MakeFastAPIOffline from typing import List - nltk.data.path = [NLTK_DATA_PATH] + nltk.data.path @@ -28,7 +28,10 @@ async def document(): def create_app(): - app = FastAPI(title="Langchain-Chatchat API Server") + app = FastAPI( + title="Langchain-Chatchat API Server", + version=VERSION + ) MakeFastAPIOffline(app) # Add CORS middleware to allow all origins # 在config.py中设置OPEN_DOMAIN=True,允许跨域 @@ -76,10 +79,10 @@ def create_app(): )(create_kb) app.post("/knowledge_base/delete_knowledge_base", - tags=["Knowledge Base Management"], - response_model=BaseResponse, - summary="删除知识库" - )(delete_kb) + tags=["Knowledge Base Management"], + response_model=BaseResponse, + summary="删除知识库" + )(delete_kb) app.get("/knowledge_base/list_docs", tags=["Knowledge Base Management"], @@ -88,10 +91,10 @@ def create_app(): )(list_docs) app.post("/knowledge_base/search_docs", - tags=["Knowledge Base Management"], - response_model=List[DocumentWithScore], - summary="搜索知识库" - )(search_docs) + tags=["Knowledge Base Management"], + response_model=List[DocumentWithScore], + summary="搜索知识库" + )(search_docs) app.post("/knowledge_base/upload_doc", tags=["Knowledge Base Management"], @@ -100,10 +103,10 @@ def create_app(): )(upload_doc) app.post("/knowledge_base/delete_doc", - tags=["Knowledge Base Management"], - response_model=BaseResponse, - summary="删除知识库内指定文件" - )(delete_doc) + tags=["Knowledge Base Management"], + response_model=BaseResponse, + summary="删除知识库内指定文件" + )(delete_doc) app.post("/knowledge_base/update_doc", tags=["Knowledge Base Management"], diff --git a/webui.py b/webui.py index 99db3f6..58fc0e3 100644 --- a/webui.py +++ b/webui.py @@ -9,6 +9,7 @@ from webui_pages.utils import * from streamlit_option_menu import option_menu from webui_pages import * import os +from configs import VERSION api = ApiRequest(base_url="http://127.0.0.1:7861", no_remote_api=False) @@ -17,6 +18,11 @@ if __name__ == "__main__": "Langchain-Chatchat WebUI", os.path.join("img", "chatchat_icon_blue_square_v2.png"), initial_sidebar_state="expanded", + menu_items={ + 'Get Help': 'https://github.com/chatchat-space/Langchain-Chatchat', + 'Report a bug': "https://github.com/chatchat-space/Langchain-Chatchat/issues", + 'About': f"""欢迎使用 Langchain-Chatchat WebUI {VERSION}!""" + } ) if not chat_box.chat_inited: @@ -35,7 +41,7 @@ if __name__ == "__main__": "func": knowledge_base_page, }, } - + with st.sidebar: st.image( os.path.join( @@ -44,6 +50,10 @@ if __name__ == "__main__": ), use_column_width=True ) + st.caption( + f"""

当前版本:{VERSION}

""", + unsafe_allow_html=True, + ) options = list(pages) icons = [x["icon"] for x in pages.values()]