fix(llm-markdown-code-rendering)

- Added replace for bad markdown code generated by LLM model

Signed-off-by: minicx <minicx@disroot.org>
This commit is contained in:
minicx 2024-12-13 01:42:22 +03:00
parent 2c5ca94b3c
commit 60d88a3514

View file

@ -51,11 +51,15 @@ const MessageBox = ({
message.sources.length > 0
) {
return setParsedMessage(
message.content.replace(
regex,
(_, number) =>
`<a href="${message.sources?.[number - 1]?.metadata?.url}" target="_blank" className="bg-light-secondary dark:bg-dark-secondary px-1 rounded ml-1 no-underline text-xs text-black/70 dark:text-white/70 relative">${number}</a>`,
),
message.content
.replace(
regex,
(_, number) =>
`<a href="${message.sources?.[number - 1]?.metadata?.url}" target="_blank" className="bg-light-secondary dark:bg-dark-secondary px-1 rounded ml-1 no-underline text-xs text-black/70 dark:text-white/70 relative">${number}</a>`,
)
.replace(/(\*\*.*?\*\*\s*)(\s*```[\s\S]*?```)/g, (match, p1, p2) => {
return `${p1}\n${p2.trim()}`;
}),
);
}