diff --git a/src/components/message/content-parts-renderer.tsx b/src/components/message/content-parts-renderer.tsx index 23a40e5..3dd15f2 100644 --- a/src/components/message/content-parts-renderer.tsx +++ b/src/components/message/content-parts-renderer.tsx @@ -31,7 +31,6 @@ import { GlobeIcon, ListTodoIcon, SparklesIcon, - BrainIcon, CircleIcon, CircleDotIcon, CircleCheckIcon, @@ -873,9 +872,12 @@ function getToolIcon( return if (name === "agent") return getTaskToolIcon(input ?? null) if (name === "skill") return - if (name === "enterplanmode" || name === "exitplanmode") - return - if (name === "switch_mode") return + if ( + name === "enterplanmode" || + name === "exitplanmode" || + name === "switch_mode" + ) + return return undefined } @@ -1056,8 +1058,16 @@ function deriveToolTitle( name === "exitplanmode" || name === "switch_mode" ) { + const plan = getField("plan") + if (plan) { + const firstLine = plan + .split("\n") + .map((l) => l.replace(/^#+\s*/, "").trim()) + .find((l) => l.length > 0) + if (firstLine) return `Plan · ${ellipsis(firstLine, 60)}` + } const title = getField("title") - if (title) return title + if (title) return `Plan · ${title}` return "Plan" }