UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-25 05:18:15 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-19 15:26:35 -07:00
parent e3668a2f1c
commit 7b64b40270
2 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/lib/components/chat/MessageInput.svelte
View File

@ -214,7 +214,7 @@
}
} catch (e) {
// Remove the failed doc from the files array
files = files.filter((f) => f.name !== url);
// files = files.filter((f) => f.name !== url);
toast.error(e);
}
};

19
src/lib/utils/index.ts
View File

@ -6,7 +6,17 @@ import sha256 from 'js-sha256';
//////////////////////////
export const sanitizeResponseContent = (content: string) => {
return content
// First, temporarily replace valid <video> tags with a placeholder
const videoTagRegex = /<video\s+src="([^"]+)"\s+controls><\/video>/gi;
const placeholders: string[] = [];
content = content.replace(videoTagRegex, (_, src) => {
const placeholder = `{{VIDEO_${placeholders.length}}}`;
placeholders.push(`<video src="${src}" controls></video>`);
return placeholder;
});
// Now apply the sanitization to the rest of the content
content = content
.replace(/<\|[a-z]*$/, '')
.replace(/<\|[a-z]+\|$/, '')
.replace(/<$/, '')
@ -14,6 +24,13 @@ export const sanitizeResponseContent = (content: string) => {
.replaceAll('<', '&lt;')
.replaceAll('>', '&gt;')
.trim();
// Replace placeholders with original <video> tags
placeholders.forEach((placeholder, index) => {
content = content.replace(`{{VIDEO_${index}}}`, placeholder);
});
return content.trim();
};
export const replaceTokens = (content, char, user) => {