Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-08 07:10:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor updates to regex

Browse Source
This commit is contained in:
Nikhil Suresh 2023-08-29 23:13:31 +00:00
parent dd10cf945c
commit b31475c622
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -120,9 +120,9 @@ 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?:", answer, re.IGNORECASE):
answer, sources = re.split( answer, sources = re.split(
r"SOURCES?[:]\s|QUESTION:\s", answer, flags=re.IGNORECASE r"SOURCES?:|QUESTION:\s", answer, flags=re.IGNORECASE
)[:2] )[:2]
sources = re.split(r"\n", sources)[0].strip() sources = re.split(r"\n", sources)[0].strip()
else: else: