github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-23 16:08:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed issue with white space and added unit tests

Browse Source
This commit is contained in:
Nikhil Suresh 2023-08-26 00:10:52 +00:00
parent 6d3485e798
commit b1f649bca5
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/chains/qa_with_sources/base.py
View File

@ -121,10 +121,8 @@ class BaseQAWithSourcesChain(Chain, ABC):
def _split_sources(self, answer: str) -> Tuple[str, str]: def _split_sources(self, answer: str) -> Tuple[str, str]:
"""Split sources from answer.""" """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( answer, sources = re.split(r"SOURCES?[:\s]|QUESTION:\s", answer, flags=re.IGNORECASE)[:2]
r"SOURCES?[:\s]|QUESTION:\s", answer, flags=re.IGNORECASE sources = re.split(r"\n", sources)[0].strip()
)[:2]
sources = re.split(r"\n", sources)[0]
else: else:
sources = "" sources = ""
return answer, sources return answer, sources