diff --git a/frontend/src/core/citations/utils.ts b/frontend/src/core/citations/utils.ts index f83feba..aadd0e1 100644 --- a/frontend/src/core/citations/utils.ts +++ b/frontend/src/core/citations/utils.ts @@ -67,8 +67,14 @@ export function parseCitations(content: string): ParseCitationsResult { } } - // Remove ALL citations blocks from content + // Remove ALL citations blocks from content (both complete and incomplete) cleanContent = content.replace(/[\s\S]*?<\/citations>/g, "").trim(); + + // Also remove incomplete citations blocks (during streaming) + // Match without closing tag or followed by anything until end of string + if (cleanContent.includes("")) { + cleanContent = cleanContent.replace(/[\s\S]*$/g, "").trim(); + } return { citations, cleanContent }; }