fix: add custom disallowed elements to Markdown component and restore the default disallowed elements (#15057)

This commit is contained in:
Wu Tianwei 2025-03-06 10:57:49 +08:00 committed by GitHub
parent 9962118dbd
commit 6c9d6a4d57
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 10 additions and 3 deletions

View File

@ -239,7 +239,7 @@ const Link = ({ node, ...props }: any) => {
} }
} }
export function Markdown(props: { content: string; className?: string }) { export function Markdown(props: { content: string; className?: string; customDisallowedElements?: string[] }) {
const latexContent = flow([ const latexContent = flow([
preprocessThinkTag, preprocessThinkTag,
preprocessLaTeX, preprocessLaTeX,
@ -274,7 +274,7 @@ export function Markdown(props: { content: string; className?: string }) {
} }
}, },
]} ]}
disallowedElements={['iframe', 'head', 'html', 'meta', 'link', 'style', 'body', 'input']} disallowedElements={['iframe', 'head', 'html', 'meta', 'link', 'style', 'body', ...(props.customDisallowedElements || [])]}
components={{ components={{
code: CodeBlock, code: CodeBlock,
img: Img, img: Img,

View File

@ -181,6 +181,7 @@ const ChunkContent: FC<IChunkContentProps> = ({
<Markdown <Markdown
className='h-full w-full !text-text-secondary' className='h-full w-full !text-text-secondary'
content={question} content={question}
customDisallowedElements={['input']}
/> />
) )
} }

View File

@ -50,6 +50,7 @@ const ChunkContent: FC<ChunkContentProps> = ({
className, className,
)} )}
content={sign_content || content || ''} content={sign_content || content || ''}
customDisallowedElements={['input']}
/> />
} }

View File

@ -60,6 +60,7 @@ const ChunkDetailModal: FC<Props> = ({
<Markdown <Markdown
className={cn('!mt-2 !text-text-secondary', heighClassName)} className={cn('!mt-2 !text-text-secondary', heighClassName)}
content={sign_content || content} content={sign_content || content}
customDisallowedElements={['input']}
/> />
{!isParentChildRetrieval && keywords && keywords.length > 0 && ( {!isParentChildRetrieval && keywords && keywords.length > 0 && (
<div className='mt-6'> <div className='mt-6'>

View File

@ -47,7 +47,11 @@ const ResultItem: FC<Props> = ({
{/* Main */} {/* Main */}
<div className='mt-1 px-3'> <div className='mt-1 px-3'>
<Markdown className='line-clamp-2' content={sign_content || content} /> <Markdown
className='line-clamp-2'
content={sign_content || content}
customDisallowedElements={['input']}
/>
{isParentChildRetrieval && ( {isParentChildRetrieval && (
<div className='mt-1'> <div className='mt-1'>
<div <div