run-llama/chat-ui
1
0
Fork 0
mirror of https://github.com/run-llama/chat-ui.git synced 2026-07-21 11:25:22 -04:00
Code Issues 13 Packages Projects Releases 10 Wiki Activity

Always preprocess citations without sources (#58)

Browse Source
This commit is contained in:
Huu Le
2025-03-11 12:16:32 +07:00
committed by GitHub
parent e34f5bb41b
commit bd353f22aa
2 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.changeset/yellow-webs-hunt.md
+5
View File
@@ -0,0 +1,5 @@
---
'@llamaindex/chat-ui': minor
---
Remove citation text if there is no sources in chat message data
packages/chat-ui/src/widgets/markdown.tsx
+16 -18
View File
@@ -42,30 +42,28 @@ const preprocessMedia = (content: string) => {
/**
* Convert citation flags [citation:id] to markdown links [citation:id]()
*/
const preprocessCitations = (input: string, sources?: SourceData) => {
const preprocessCitations = (input: string) => {
let content = input
if (sources) {
// Match citation format [citation:node_id]
// Handle complete citations
const idToIndexRegex = /\[citation:([^\]]+)\]/g
content = content.replace(idToIndexRegex, (match, citationId) => {
const trimmedId = citationId.trim()
// Use a special format that doesn't get styled as a link by markdown-it
return `[citation:${trimmedId}](javascript:void(0))`
})
// Match citation format [citation:node_id]
// Handle complete citations
const idToIndexRegex = /\[citation:([^\]]+)\]/g
content = content.replace(idToIndexRegex, (match, citationId) => {
const trimmedId = citationId.trim()
// Use a special format that doesn't get styled as a link by markdown-it
return `[citation:${trimmedId}](javascript:void(0))`
})
// For incomplete citations - any [citation: pattern that isn't closed with ]
// Look for open bracket, citation text, then end of string or any char except closing bracket
const incompleteRegex = /\[citation:[^\]]*$/g
content = content.replace(incompleteRegex, '')
}
// For incomplete citations - any [citation: pattern that isn't closed with ]
// Look for open bracket, citation text, then end of string or any char except closing bracket
const incompleteRegex = /\[citation:[^\]]*$/g
content = content.replace(incompleteRegex, '')
return content
}
const preprocessContent = (content: string, sources?: SourceData) => {
return preprocessCitations(preprocessLaTeX(preprocessMedia(content)), sources)
const preprocessContent = (content: string) => {
return preprocessCitations(preprocessLaTeX(preprocessMedia(content)))
}
export function Markdown({
@@ -79,7 +77,7 @@ export function Markdown({
backend?: string
citationComponent?: ComponentType<CitationComponentProps>
}) {
const processedContent = preprocessContent(content, sources)
const processedContent = preprocessContent(content)
return (
<div>