minor updates to regex

1 year ago · b31475c622
parent dd10cf945c
commit b31475c622
1 changed files with 2 additions and 2 deletions
--- 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?:", answer, re.IGNORECASE):
            answer, sources = re.split(
-                r"SOURCES?[:]\s|QUESTION:\s", answer, flags=re.IGNORECASE
+                r"SOURCES?:|QUESTION:\s", answer, flags=re.IGNORECASE
            )[:2]
            sources = re.split(r"\n", sources)[0].strip()
        else: