diff --git a/backend/tests/unit/danswer/llm/answering/stream_processing/test_citation_processing.py b/backend/tests/unit/danswer/llm/answering/stream_processing/test_citation_processing.py index e6a5fe1f027..13e6fd73b5a 100644 --- a/backend/tests/unit/danswer/llm/answering/stream_processing/test_citation_processing.py +++ b/backend/tests/unit/danswer/llm/answering/stream_processing/test_citation_processing.py @@ -374,6 +374,17 @@ def process_text( "The code demonstrates variable assignment.", [], ), + ( + "Citation as a single token", + [ + "Here is some text", + "[1]", + ".", + " Some other text", + ], + "Here is some text[[1]](https://0.com). Some other text", + ["doc_0"], + ), ], ) def test_citation_extraction(