github/deer-flow
1
0
Fork 0
mirror of https://github.com/bytedance/deer-flow.git synced 2026-04-25 19:28:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add reasoning check to message list item rendering

Browse Source
This commit is contained in:
Henry Li 2026-01-17 18:02:19 +08:00
parent cb54b5dffa
commit 0ea448a220
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frontend/src/components/workspace/messages/message-list-item.tsx
View File

@ -7,7 +7,11 @@ import {
MessageResponse as AIElementMessageResponse,
MessageToolbar,
} from "@/components/ai-elements/message";
import { extractContentFromMessage, hasToolCalls } from "@/core/messages/utils";
import {
extractContentFromMessage,
hasReasoning,
hasToolCalls,
} from "@/core/messages/utils";
import { useRehypeSplitWordsIntoSpans } from "@/core/rehype";
import { cn } from "@/lib/utils";
@ -66,6 +70,9 @@ function MessageContent_({
const rehypePlugins = useRehypeSplitWordsIntoSpans(isLoading);
return (
<AIElementMessageContent className={className}>
{hasReasoning(message) && (
<MessageGroup messages={messagesInGroup} isLoading={isLoading} />
)}
<AIElementMessageResponse rehypePlugins={rehypePlugins}>
{extractContentFromMessage(message)}
</AIElementMessageResponse>

3
frontend/src/core/messages/utils.ts
View File

@ -46,9 +46,6 @@ export function groupMessages<T>(
} else {
// Assistant messages with content (text or images) are shown as a group if they have content
// No matter whether it has tool calls or not
if (hasReasoning(message)) {
currentGroup.push(message);
}
yieldCurrentGroup();
currentGroup.push(message);
}