wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-05 07:02:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: redesign step counter

Browse Source
This commit is contained in:
Henry Li
2026-01-17 17:45:13 +08:00
parent 63f3c9e2bb
commit 62921ec96a
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
frontend/src/components/workspace/messages/message-group.tsx
View File

@@ -42,6 +42,10 @@ export function MessageGroup({
isLoading?: boolean;
}) {
const steps = useMemo(() => convertToSteps(messages), [messages]);
const stepCount = useMemo(
() => steps.filter((step) => step.type !== "reasoning").length,
[steps],
);
const rehypePlugins = useRehypeSplitWordsIntoSpans(isLoading);
const [open, setOpen] = useState(false);
const lastStep = steps[steps.length - 1];
@@ -56,16 +60,16 @@ export function MessageGroup({
<ChainOfThoughtHeader
className="min-h-6"
icon={
open && steps.length > 1 ? <ListTreeIcon className="size-4" /> : icon
open && stepCount > 1 ? <ListTreeIcon className="size-4" /> : icon
}
>
<div className="flex w-full items-center justify-between">
<div>
<div>
{open && steps.length > 1 ? (
<div>{steps.length} steps</div>
{open && stepCount > 1 ? (
<div>{stepCount} steps</div>
) : (
<FlipDisplay uniqueKey={`step-${steps.length}`}>
<FlipDisplay uniqueKey={`step-${stepCount}`}>
<MessageResponse rehypePlugins={rehypePlugins}>
{label}
</MessageResponse>
@@ -74,11 +78,9 @@ export function MessageGroup({
</div>
</div>
<div>
{!open && steps.length > 1 && (
{!open && stepCount > 1 && (
<div>
{steps.length - 1 > 1
? `${steps.length - 1} more steps`
: `${steps.length - 1} more step`}
{stepCount > 1 ? `${stepCount} steps` : `${stepCount} step`}
</div>
)}
</div>