Merge branch 'master' into pre-release
This commit is contained in:
commit
30c00ffac6
|
|
@ -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. 知识库初始化与迁移
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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":
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue