diff --git a/README.md b/README.md index 577e7f7..1d4bb8f 100644 --- a/README.md +++ b/README.md @@ -209,7 +209,7 @@ embedding_model_dict = { } ``` -如果你选择使用OpenAI的Embedding模型,请将模型的 ``key``写入 `embedding_model_dict`中。使用该模型,你需要鞥能够访问OpenAI官的API,或设置代理。 +如果你选择使用OpenAI的Embedding模型,请将模型的 ``key``写入 `embedding_model_dict`中。使用该模型,你需要能够访问OpenAI官的API,或设置代理。 ### 4. 知识库初始化与迁移 diff --git a/requirements.txt b/requirements.txt index 4271f3a..385747b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,6 @@ langchain==0.0.266 openai +zhipuai sentence_transformers fschat==0.2.24 transformers>=4.31.0 @@ -9,7 +10,7 @@ nltk~=3.8.1 uvicorn~=0.23.1 starlette~=0.27.0 pydantic~=1.10.11 -unstructured[all-docs] +unstructured[all-docs]>=0.10.4 python-magic-bin; sys_platform == 'win32' SQLAlchemy==2.0.19 faiss-cpu diff --git a/server/knowledge_base/migrate.py b/server/knowledge_base/migrate.py index 129dd53..c11c00d 100644 --- a/server/knowledge_base/migrate.py +++ b/server/knowledge_base/migrate.py @@ -75,7 +75,7 @@ def folder2db( files = list_files_from_folder(kb_name) 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) print(f"已将 {kb_name}/{kb_file.filename} 添加到数据库") elif mode == "update_in_db":