From b733a479022c3c8d9a9d15330bc3bf64ae8fb381 Mon Sep 17 00:00:00 2001 From: liunux4odoo Date: Thu, 10 Aug 2023 15:51:22 +0800 Subject: [PATCH] split requirements to 2 parts: basic with api; webui --- requirements.txt | 21 +++++++-------------- requirements_webui.txt | 6 ++++++ webui_pages/utils.py | 2 +- 3 files changed, 14 insertions(+), 15 deletions(-) create mode 100644 requirements_webui.txt diff --git a/requirements.txt b/requirements.txt index c187f88..5b366c0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,32 +1,25 @@ langchain==0.0.257 openai sentence_transformers -chromadb fschat==0.2.20 transformers transformers_stream_generator torch~=2.0.0 fastapi~=0.99.1 -loguru fastapi-offline nltk~=3.8.1 uvicorn~=0.23.1 starlette~=0.27.0 numpy~=1.24.4 +pandas~=2.0.3 pydantic~=1.10.11 unstructured[all-docs] python-magic-bin; sys_platform == 'win32' - -streamlit>=1.25.0 -streamlit-option-menu -streamlit-antd-components -streamlit-chatbox>=1.1.6 -streamlit-aggrid -httpx - -faiss-cpu -pymilvus==2.1.3 # requires milvus==2.1.3 - 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 +# chromadb # psycopg2 -# pgvector \ No newline at end of file +# pgvector diff --git a/requirements_webui.txt b/requirements_webui.txt new file mode 100644 index 0000000..b4a954e --- /dev/null +++ b/requirements_webui.txt @@ -0,0 +1,6 @@ +streamlit>=1.25.0 +streamlit-option-menu +streamlit-antd-components +streamlit-chatbox>=1.1.6 +streamlit-aggrid +httpx diff --git a/webui_pages/utils.py b/webui_pages/utils.py index b4c2feb..6dac5a9 100644 --- a/webui_pages/utils.py +++ b/webui_pages/utils.py @@ -443,7 +443,7 @@ class ApiRequest: temp_file.write(file.read()) temp_file.seek(0) response = run_async(upload_doc( - UploadFile(temp_file, filename=filename), + UploadFile(file=temp_file, filename=filename), knowledge_base_name, override, ))