matches colon and any number of white spaces after colon

This commit is contained in:
Nikhil Suresh 2023-08-29 04:18:33 +00:00
parent 64eb5a6082
commit 23ef836b48

View File

@ -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: