From 91ec7ef8714d903f888dacc08dc13f75cd9dc461 Mon Sep 17 00:00:00 2001
From: GuoQing Zhang <zhangguoqing@dataelem.com>
Date: Thu, 11 Jul 2024 16:54:24 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E8=BF=94=E5=9B=9E=E6=9D=A5=E6=BA=90?=
 =?UTF-8?q?=E6=96=87=E6=A1=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../bisheng_langchain/rag/bisheng_rag_chain.py                  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/bisheng-langchain/bisheng_langchain/rag/bisheng_rag_chain.py b/src/bisheng-langchain/bisheng_langchain/rag/bisheng_rag_chain.py
index 39fd45d69..cab5b1dfa 100644
--- 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,
         )