diff --git a/webview-ui/src/components/chat/TaskHeader.tsx b/webview-ui/src/components/chat/TaskHeader.tsx index 1896df486b..6fed1d741b 100644 --- a/webview-ui/src/components/chat/TaskHeader.tsx +++ b/webview-ui/src/components/chat/TaskHeader.tsx @@ -15,6 +15,7 @@ import { useExtensionState } from "@src/context/ExtensionStateContext" import { useSelectedModel } from "@/components/ui/hooks/useSelectedModel" import Thumbnails from "../common/Thumbnails" +import { ModeBadge } from "../common/ModeBadge" import { TaskActions } from "./TaskActions" import { ShareButton } from "./ShareButton" @@ -101,6 +102,11 @@ const TaskHeader = ({ )} + {currentTaskItem?.mode && ( +
+ +
+ )}