1.找到langchain-chatchat中的knowledge_base_chat.py
2.knowledge_base_chat.py的api内容加上一个flie_name参数,即传过来你需要指定一个文件名称,或多个文件名称,同时也可以不指定,加上以下代码:
flie_name: List = Body(
None,
description=""
),
3.knowledge_base_chat.py文件中再加上以下代码:
if flie_name:
docs1 = []
for doc in docs:
if doc.metadata.get("source") in flie_name:
docs1.append(doc)
else:
pass
else:
docs1=docs
4.全部加上后,运行即可指定一个或多个文件进行知识库回答