add removeDuplicateURL

This commit is contained in:
chungyau97
2023-08-03 18:21:45 +08:00
parent 1b5f638e49
commit 64c0ecb075
@@ -64,6 +64,20 @@ export const ChatMessage = ({ open, chatflowid, isDialog }) => {
window.open(data, '_blank') window.open(data, '_blank')
} }
const removeDuplicateURL = (message) => {
const visitedURLs = []
const newSourceDocuments = []
message.sourceDocuments.forEach((source) => {
if (isValidURL(source.metadata.source) && !visitedURLs.includes(source.metadata.source)) {
visitedURLs.push(source.metadata.source)
newSourceDocuments.push(source)
} else if (!isValidURL(source.metadata.source)) {
newSourceDocuments.push(source)
}
})
return newSourceDocuments
}
const scrollToBottom = () => { const scrollToBottom = () => {
if (ps.current) { if (ps.current) {
ps.current.scrollTo({ top: maxScroll }) ps.current.scrollTo({ top: maxScroll })
@@ -324,7 +338,7 @@ export const ChatMessage = ({ open, chatflowid, isDialog }) => {
</div> </div>
{message.sourceDocuments && ( {message.sourceDocuments && (
<div style={{ display: 'block', flexDirection: 'row', width: '100%' }}> <div style={{ display: 'block', flexDirection: 'row', width: '100%' }}>
{message.sourceDocuments.map((source, index) => { {removeDuplicateURL(message).map((source, index) => {
const URL = isValidURL(source.metadata.source) const URL = isValidURL(source.metadata.source)
return ( return (
<Chip <Chip