langchain-chatchat指定一个或多个文件回答，不允许回答内容有其他文件内容，即屏蔽其他文件内容

萧鼎

于 2024-04-09 14:05:29 发布

阅读量416

点赞数 3

分类专栏：机器学习算法与实战文章标签： langchain

本文链接：https://blog.csdn.net/liaoqingjian/article/details/137549647

版权

机器学习算法与实战专栏收录该内容

58 篇文章

订阅专栏

文章介绍了如何在langchain-chatchat的knowledge_base_chat.py文件中添加一个file_name参数，以便按需指定一个或多个文件进行知识库查询。新代码逻辑会筛选出与指定文件名相关的文档进行回答。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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