Merge branch 'master' into pre-release

This commit is contained in:
imClumsyPanda 2023-09-05 14:47:28 +08:00
commit 30c00ffac6
3 changed files with 4 additions and 3 deletions

View File

@ -209,7 +209,7 @@ embedding_model_dict = {
} }
``` ```
如果你选择使用OpenAI的Embedding模型请将模型的 ``key``写入 `embedding_model_dict`中。使用该模型,你需要能够访问OpenAI官的API或设置代理。 如果你选择使用OpenAI的Embedding模型请将模型的 ``key``写入 `embedding_model_dict`中。使用该模型你需要能够访问OpenAI官的API或设置代理。
### 4. 知识库初始化与迁移 ### 4. 知识库初始化与迁移

View File

@ -1,5 +1,6 @@
langchain==0.0.266 langchain==0.0.266
openai openai
zhipuai
sentence_transformers sentence_transformers
fschat==0.2.24 fschat==0.2.24
transformers>=4.31.0 transformers>=4.31.0
@ -9,7 +10,7 @@ nltk~=3.8.1
uvicorn~=0.23.1 uvicorn~=0.23.1
starlette~=0.27.0 starlette~=0.27.0
pydantic~=1.10.11 pydantic~=1.10.11
unstructured[all-docs] unstructured[all-docs]>=0.10.4
python-magic-bin; sys_platform == 'win32' python-magic-bin; sys_platform == 'win32'
SQLAlchemy==2.0.19 SQLAlchemy==2.0.19
faiss-cpu faiss-cpu

View File

@ -75,7 +75,7 @@ def folder2db(
files = list_files_from_folder(kb_name) files = list_files_from_folder(kb_name)
kb_files = file_to_kbfile(kb_name, files) kb_files = file_to_kbfile(kb_name, files)
for kb_file in kb_file: for kb_file in kb_files:
add_file_to_db(kb_file) add_file_to_db(kb_file)
print(f"已将 {kb_name}/{kb_file.filename} 添加到数据库") print(f"已将 {kb_name}/{kb_file.filename} 添加到数据库")
elif mode == "update_in_db": elif mode == "update_in_db":