瀏覽代碼

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

tags/1.0.1
Wu Tianwei 8 月之前
父節點
當前提交
6c9d6a4d57
沒有連結到貢獻者的電子郵件帳戶。

+ 2
- 2
web/app/components/base/markdown.tsx 查看文件

@@ -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([
preprocessThinkTag,
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={{
code: CodeBlock,
img: Img,

+ 1
- 0
web/app/components/datasets/documents/detail/completed/common/chunk-content.tsx 查看文件

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

+ 1
- 0
web/app/components/datasets/documents/detail/completed/segment-card/chunk-content.tsx 查看文件

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


+ 1
- 0
web/app/components/datasets/hit-testing/components/chunk-detail-modal.tsx 查看文件

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

+ 5
- 1
web/app/components/datasets/hit-testing/components/result-item.tsx 查看文件

@@ -47,7 +47,11 @@ const ResultItem: FC<Props> = ({

{/* Main */}
<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 && (
<div className='mt-1'>
<div

Loading…
取消
儲存