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 ? (