From 23ef836b48bbbc328813a0e48e64ac35bbaf5b5d Mon Sep 17 00:00:00 2001 From: Nikhil Suresh Date: Tue, 29 Aug 2023 04:18:33 +0000 Subject: [PATCH] matches colon and any number of white spaces after colon --- libs/langchain/langchain/chains/qa_with_sources/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langchain/langchain/chains/qa_with_sources/base.py b/libs/langchain/langchain/chains/qa_with_sources/base.py index e6a07c37630..79b083de0b7 100644 --- a/libs/langchain/langchain/chains/qa_with_sources/base.py +++ b/libs/langchain/langchain/chains/qa_with_sources/base.py @@ -120,9 +120,9 @@ class BaseQAWithSourcesChain(Chain, ABC): def _split_sources(self, answer: str) -> Tuple[str, str]: """Split sources from answer.""" - if re.search(r"SOURCES?[:\s]", answer, re.IGNORECASE): + if re.search(r"SOURCES?[:]\s", answer, re.IGNORECASE): answer, sources = re.split( - r"SOURCES?[:]|QUESTION:\s", answer, flags=re.IGNORECASE + r"SOURCES?[:]\s|QUESTION:\s", answer, flags=re.IGNORECASE )[:2] sources = re.split(r"\n", sources)[0].strip() else: