diff --git a/requirements.txt b/requirements.txt index 9e60611..fd1a0a1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,3 +19,12 @@ faiss-cpu # pymilvus==2.1.3 # requires milvus==2.1.3 # psycopg2 # pgvector + +numpy~=1.24.4 +pandas~=2.0.3 +streamlit>=1.25.0 +streamlit-option-menu>=0.3.6 +streamlit-antd-components>=0.1.11 +streamlit-chatbox>=1.1.6 +streamlit-aggrid>=0.3.4.post3 +httpx~=0.24.1 diff --git a/requirements_api.txt b/requirements_api.txt new file mode 100644 index 0000000..9e60611 --- /dev/null +++ b/requirements_api.txt @@ -0,0 +1,21 @@ +langchain==0.0.257 +openai +sentence_transformers +fschat==0.2.20 +transformers +torch~=2.0.0 +fastapi~=0.99.1 +fastapi-offline +nltk~=3.8.1 +uvicorn~=0.23.1 +starlette~=0.27.0 +pydantic~=1.10.11 +unstructured[all-docs] +python-magic-bin; sys_platform == 'win32' +SQLAlchemy==2.0.19 +faiss-cpu + +# uncomment libs if you want to use corresponding vector store +# pymilvus==2.1.3 # requires milvus==2.1.3 +# psycopg2 +# pgvector diff --git a/webui_pages/dialogue/dialogue.py b/webui_pages/dialogue/dialogue.py index abf7ab5..6369000 100644 --- a/webui_pages/dialogue/dialogue.py +++ b/webui_pages/dialogue/dialogue.py @@ -2,7 +2,6 @@ import streamlit as st from webui_pages.utils import * from streamlit_chatbox import * from datetime import datetime -import streamlit_antd_components as sac from server.chat.search_engine_chat import SEARCH_ENGINES from typing import List, Dict @@ -133,7 +132,7 @@ def dialogue_page(api: ApiRequest): elif dialogue_mode == "知识库问答": history = get_messages_history(history_len) chat_box.ai_say([ - f"正在查询知识库: `{selected_kb}` ...", + f"正在查询知识库 `{selected_kb}` ...", Markdown("...", in_expander=True, title="知识库匹配结果"), ]) text = "" @@ -144,7 +143,7 @@ def dialogue_page(api: ApiRequest): chat_box.update_msg(text, 0, streaming=False) elif dialogue_mode == "搜索引擎问答": chat_box.ai_say([ - f"正在执行{search_engine}搜索...", + f"正在执行 `{search_engine}` 搜索...", Markdown("...", in_expander=True, title="网络搜索结果"), ]) text = ""