update configs.__init__, webui.py and api.py

configs/__init__.py

@@ -1,2 +1,4 @@
 from .model_config import *
-from .server_config import *
+from .server_config import *
+
+VERSION = "v0.2.1-preview"

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"],
+             tags=["Knowledge Base Management"],
-               response_model=BaseResponse,
+             response_model=BaseResponse,
-               summary="删除知识库"
+             summary="删除知识库"
-               )(delete_kb)
+             )(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"],
+             tags=["Knowledge Base Management"],
-            response_model=List[DocumentWithScore],
+             response_model=List[DocumentWithScore],
-            summary="搜索知识库"
+             summary="搜索知识库"
-            )(search_docs)
+             )(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"],
+             tags=["Knowledge Base Management"],
-               response_model=BaseResponse,
+             response_model=BaseResponse,
-               summary="删除知识库内指定文件"
+             summary="删除知识库内指定文件"
-               )(delete_doc)
+             )(delete_doc)
 
     app.post("/knowledge_base/update_doc",
              tags=["Knowledge Base Management"],

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"""<p align="right">当前版本：{VERSION}</p>""",
+            unsafe_allow_html=True,
+        )
         options = list(pages)
         icons = [x["icon"] for x in pages.values()]