UniqAI/bolt.diy
1
0
Fork 0
mirror of https://github.com/stackblitz-labs/bolt.diy synced 2025-03-10 06:00:19 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: support for <think></think> tags to allow reasoning tokens formatted in UI (#1205)
Some checks are pending
Docker Publish / docker-build-publish (push) Waiting to run
Details
Update Stable Branch / prepare-release (push) Waiting to run
Details

Browse Source
This commit is contained in:
Anirban Kar 2025-01-29 02:33:53 +05:30 committed by GitHub
parent 32bfdd9c24
commit a199295ad8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
app/utils/markdown.ts
View File

@ -54,8 +54,28 @@ export const allowedHTMLElements = [
'tr', 'tr',
'ul', 'ul',
'var', 'var',
'think',
]; ];
// Add custom rehype plugin
function remarkThinkRawContent() {
return (tree: any) => {
visit(tree, (node: any) => {
if (node.type === 'html' && node.value && node.value.startsWith('<think>')) {
const cleanedContent = node.value.slice(7);
node.value = `<div class="__boltThought__">${cleanedContent}`;
return;
}
if (node.type === 'html' && node.value && node.value.startsWith('</think>')) {
const cleanedContent = node.value.slice(8);
node.value = `</div>${cleanedContent}`;
}
});
};
}
const rehypeSanitizeOptions: RehypeSanitizeOptions = { const rehypeSanitizeOptions: RehypeSanitizeOptions = {
...defaultSchema, ...defaultSchema,
tagNames: allowedHTMLElements, tagNames: allowedHTMLElements,
@ -79,6 +99,8 @@ export function remarkPlugins(limitedMarkdown: boolean) {
plugins.unshift(limitedMarkdownPlugin); plugins.unshift(limitedMarkdownPlugin);
} }
plugins.unshift(remarkThinkRawContent);
return plugins; return plugins;
} }