如何在LangChain的Chroma VectorStore中按文档名列表过滤文档？

你可以通过在search_kwargs中添加filter参数，利用Chroma支持的$in操作符实现按文档名列表过滤，具体修改如下：

lst = ['doc1', 'doc2', 'doc3']
base_retriever = chroma_db.as_retriever(
    search_kwargs={
        'k': 10,
        'filter': {'doc_name': {'$in': lst}}
    }
)

说明 #

• $in是Chroma内置的查询操作符，用于筛选元数据中doc_name值存在于目标列表lst中的文档。
• 这样配置后，检索器只会从doc1、doc2、doc3这几个指定文档中返回匹配结果，同时保留原有的k=10返回数量限制。

内容的提问来源于stack exchange，提问作者Abhishek Sharma