diff --git a/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.test.tsx
index c218db5b914..a83c11d1444 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.test.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.test.tsx
@@ -87,6 +87,21 @@ describe("ChatMessageBubble", () => {
expect(screen.getByText("Hi there")).toBeInTheDocument();
});
+ it.each([
+ { role: "user" as const, bubble: ["bg-info/10", "border-info/20"], avatar: "bg-info/20" },
+ { role: "assistant" as const, bubble: ["bg-card", "border-border"], avatar: "bg-muted" },
+ ])("should paint the $role surface from theme tokens, not fixed colours", ({ role, bubble, avatar }) => {
+ render();
+
+ const header = screen.getByText(role).closest("div") as HTMLElement;
+ const surface = header.parentElement as HTMLElement;
+
+ expect(surface).toHaveClass(...bubble);
+ expect(surface).not.toHaveAttribute("style");
+ expect(header.firstElementChild).toHaveClass(avatar);
+ expect(header.firstElementChild).not.toHaveAttribute("style");
+ });
+
it("should show model badge for assistant messages when model is provided", () => {
render();
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.tsx b/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.tsx
index eb24463cfd3..8c54d9e89fa 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatMessageBubble.tsx
@@ -46,20 +46,16 @@ function ChatMessageBubble({
return (
{/* Header: role icon + name + model badge */}
{isUser ? (
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatUI.tsx b/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatUI.tsx
index 2adc7fc30ea..e6257907918 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatUI.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/playground/components/chat_ui/ChatUI.tsx
@@ -1784,19 +1784,9 @@ const ChatUI: React.FC
= ({
chatHistory.length > 0 &&
chatHistory[chatHistory.length - 1].role === "user" && (