mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-22 11:00:37 +00:00
fixed regex to match sources for all cases, also includes source
This commit is contained in:
@@ -124,7 +124,7 @@ class BaseQAWithSourcesChain(Chain, ABC):
|
|||||||
answer, sources = re.split(
|
answer, sources = re.split(
|
||||||
r"SOURCES?[:\s]|QUESTION:\s", answer, flags=re.IGNORECASE
|
r"SOURCES?[:\s]|QUESTION:\s", answer, flags=re.IGNORECASE
|
||||||
)[:2]
|
)[:2]
|
||||||
sources = re.split(r"\n", sources)[0].strip()
|
sources = re.split(r"\n", sources)[0]
|
||||||
else:
|
else:
|
||||||
sources = ""
|
sources = ""
|
||||||
return answer, sources
|
return answer, sources
|
||||||
|
Reference in New Issue
Block a user