fix: 返回来源文档

dataelement · Jul 11, 2024 · 91ec7ef · 91ec7ef
1 parent 1a97b12
commit 91ec7ef
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/bisheng-langchain/bisheng_langchain/rag/bisheng_rag_chain.py b/src/bisheng-langchain/bisheng_langchain/rag/bisheng_rag_chain.py
@@ -97,6 +97,7 @@ def from_llm(
         max_content: int = 15000,
         sort_by_source_and_index: bool = False,
         callbacks: Callbacks = None,
+        return_source_documents: bool = False,
         **kwargs: Any,
     ) -> BishengRetrievalQA:
         bisheng_rag_tool = BishengRAGTool(
@@ -111,6 +112,7 @@ def from_llm(
         return cls(
             bisheng_rag_tool=bisheng_rag_tool,
             callbacks=callbacks,
+            return_source_documents=return_source_documents,
             **kwargs,
         )