diff --git a/backend/danswer/llm/answering/stream_processing/quotes_processing.py b/backend/danswer/llm/answering/stream_processing/quotes_processing.py index 853e2d4ffbe1..3d81270e0d3a 100644 --- a/backend/danswer/llm/answering/stream_processing/quotes_processing.py +++ b/backend/danswer/llm/answering/stream_processing/quotes_processing.py @@ -26,9 +26,7 @@ from danswer.utils.text_processing import shared_precompare_cleanup logger = setup_logger() -answer_pattern = re.compile( - r'(?:```(?:json)?)?\s*{\s*"answer"\s*:\s*"', re.DOTALL | re.IGNORECASE -) +answer_pattern = re.compile(r'{\s*"answer"\s*:\s*"', re.IGNORECASE) def _extract_answer_quotes_freeform( @@ -232,7 +230,7 @@ def process_model_tokens( model_output += token if not found_answer_start: - m = answer_pattern.match(model_output) + m = answer_pattern.search(model_output) if m: found_answer_start = True