update requirements.txt

This commit is contained in:
imClumsyPanda 2023-08-12 10:44:04 +08:00
parent 90d385cb92
commit e65458a506
3 changed files with 32 additions and 3 deletions

View File

@ -19,3 +19,12 @@ faiss-cpu
# pymilvus==2.1.3 # requires milvus==2.1.3 # pymilvus==2.1.3 # requires milvus==2.1.3
# psycopg2 # psycopg2
# pgvector # 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

21
requirements_api.txt Normal file
View File

@ -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

View File

@ -2,7 +2,6 @@ import streamlit as st
from webui_pages.utils import * from webui_pages.utils import *
from streamlit_chatbox import * from streamlit_chatbox import *
from datetime import datetime from datetime import datetime
import streamlit_antd_components as sac
from server.chat.search_engine_chat import SEARCH_ENGINES from server.chat.search_engine_chat import SEARCH_ENGINES
from typing import List, Dict from typing import List, Dict
@ -133,7 +132,7 @@ def dialogue_page(api: ApiRequest):
elif dialogue_mode == "知识库问答": elif dialogue_mode == "知识库问答":
history = get_messages_history(history_len) history = get_messages_history(history_len)
chat_box.ai_say([ chat_box.ai_say([
f"正在查询知识库 `{selected_kb}` ...", f"正在查询知识库 `{selected_kb}` ...",
Markdown("...", in_expander=True, title="知识库匹配结果"), Markdown("...", in_expander=True, title="知识库匹配结果"),
]) ])
text = "" text = ""
@ -144,7 +143,7 @@ def dialogue_page(api: ApiRequest):
chat_box.update_msg(text, 0, streaming=False) chat_box.update_msg(text, 0, streaming=False)
elif dialogue_mode == "搜索引擎问答": elif dialogue_mode == "搜索引擎问答":
chat_box.ai_say([ chat_box.ai_say([
f"正在执行{search_engine}搜索...", f"正在执行 `{search_engine}` 搜索...",
Markdown("...", in_expander=True, title="网络搜索结果"), Markdown("...", in_expander=True, title="网络搜索结果"),
]) ])
text = "" text = ""