mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-27 17:08:47 +00:00
minor updates to regex
This commit is contained in:
parent
dd10cf945c
commit
b31475c622
@ -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:
|
||||
|
Loading…
Reference in New Issue
Block a user