jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: adds handling for <think> tags (#7109)

Browse source 
* Adds handling for <think> tags

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Lucas Oliveira 2025-03-17 19:50:42 -03:00 • committed by GitHub
commit 5c6ebecdb5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/frontend/src/modals/IOModal/components/chatView/chatMessage/components/edit-message.tsx
View file

@ -13,12 +13,23 @@ type MarkdownFieldProps = {
editedFlag: React.ReactNode;
};
// Function to replace <think> tags with a placeholder before markdown processing
const preprocessChatMessage = (text: string): string => {
// Replace <think> tags with `<span class="think-tag">think:</span>`
return text
.replace(/<think>/g, "`<think>`")
.replace(/<\/think>/g, "`</think>`");
};
export const MarkdownField = ({
chat,
isEmpty,
chatMessage,
editedFlag,
}: MarkdownFieldProps) => {
// Process the chat message to handle <think> tags
const processedChatMessage = preprocessChatMessage(chatMessage);
return (
<div className="w-full items-baseline gap-2">
<Markdown
@ -56,6 +67,11 @@ export const MarkdownField = ({
if (content[0] === "▍") {
return <span className="form-modal-markdown-span"></span>;
}
// Specifically handle <think> tags that were wrapped in backticks
if (content === "<think>" || content === "</think>") {
return <span>{content}</span>;
}
}
const match = /language-(\w+)/.exec(className || "");
@ -74,7 +90,9 @@ export const MarkdownField = ({
},
}}
>
{isEmpty && !chat.stream_url ? EMPTY_OUTPUT_SEND_MESSAGE : chatMessage}
{isEmpty && !chat.stream_url
? EMPTY_OUTPUT_SEND_MESSAGE
: processedChatMessage}
</Markdown>
{editedFlag}
</div>