diff --git a/ui/src/components/task-chat/TaskChatBubble.test.tsx b/ui/src/components/task-chat/TaskChatBubble.test.tsx index 00d8746155..70fe754ae9 100644 --- a/ui/src/components/task-chat/TaskChatBubble.test.tsx +++ b/ui/src/components/task-chat/TaskChatBubble.test.tsx @@ -285,6 +285,29 @@ describe("TaskChatBubble accent-bubble text color", () => { }); describe("TaskChatBubble agent page-surface treatment", () => { + it("does not show an on-behalf-of badge in the new task view", () => { + const container = document.createElement("div"); + document.body.appendChild(container); + const root = createRoot(container); + + flushSync(() => + root.render( + + + , + ), + ); + + expect(container.textContent).toContain("Fable"); + expect(container.textContent).not.toContain("for Dotta"); + expect(container.querySelector('[data-testid="comment-attribution-chip"]')).toBeNull(); + + flushSync(() => root.unmount()); + container.remove(); + }); + it("renders agent prose without a card background or constrained width", () => { const container = document.createElement("div"); document.body.appendChild(container); diff --git a/ui/src/components/task-chat/TaskChatBubble.tsx b/ui/src/components/task-chat/TaskChatBubble.tsx index 3caae25934..bc51a51133 100644 --- a/ui/src/components/task-chat/TaskChatBubble.tsx +++ b/ui/src/components/task-chat/TaskChatBubble.tsx @@ -11,7 +11,6 @@ import { } from "@/components/ImageGalleryModal"; import { Avatar, AvatarFallback } from "@/components/ui/avatar"; import { AgentIcon } from "@/components/AgentIconPicker"; -import { CommentAttributionChip } from "@/components/CommentAttributionChip"; import { Attachment, AttachmentContent, @@ -75,11 +74,9 @@ function initialsForName(name: string) { export function TaskChatAgentIdentity({ agentName, agentIcon, - onBehalfOfUserName, }: { agentName: string; agentIcon?: string | null; - onBehalfOfUserName?: string; }) { return ( {agentName} - {onBehalfOfUserName ? ( - - ) : null} ); } @@ -241,7 +232,6 @@ function TaskChatBubbleContent({ ) : null} {bodyText.length > 0 ? (