diff --git a/desktop/src/features/agents/ui/AgentIdentityCard.tsx b/desktop/src/features/agents/ui/AgentIdentityCard.tsx
index 1f782f6d8..90630a374 100644
--- a/desktop/src/features/agents/ui/AgentIdentityCard.tsx
+++ b/desktop/src/features/agents/ui/AgentIdentityCard.tsx
@@ -3,6 +3,7 @@ import { AgentProviderCollage } from "./AgentProviderCollage";
type AgentIdentityCardProps = {
ariaLabel: string;
+ avatarUrl?: string | null;
dataTestId: string;
label: string;
modelLabel: string;
@@ -11,6 +12,7 @@ type AgentIdentityCardProps = {
export function AgentIdentityCard({
ariaLabel,
+ avatarUrl,
dataTestId,
label,
modelLabel,
@@ -26,7 +28,7 @@ export function AgentIdentityCard({
onClick={onClick}
type="button"
>
-
+
diff --git a/desktop/src/features/agents/ui/AgentProviderCollage.tsx b/desktop/src/features/agents/ui/AgentProviderCollage.tsx
index 34166dad7..9fbd0d025 100644
--- a/desktop/src/features/agents/ui/AgentProviderCollage.tsx
+++ b/desktop/src/features/agents/ui/AgentProviderCollage.tsx
@@ -1,5 +1,6 @@
import type { CSSProperties } from "react";
+import { ProfileAvatar } from "@/features/profile/ui/ProfileAvatar";
import { IdentityInitialsAvatar } from "./IdentityInitialsAvatar";
type ClusterLayout = {
@@ -7,13 +8,17 @@ type ClusterLayout = {
};
type AgentProviderCollageProps = {
+ avatarUrl?: string | null;
label: string;
};
const CLUSTER_CENTER_TOP_PERCENT = 50;
const AGENT_INITIAL_AVATAR_SIZE = 152;
-export function AgentProviderCollage({ label }: AgentProviderCollageProps) {
+export function AgentProviderCollage({
+ avatarUrl,
+ label,
+}: AgentProviderCollageProps) {
const { avatarStyle } = buildClusterPoints();
return (
@@ -24,11 +29,21 @@ export function AgentProviderCollage({ label }: AgentProviderCollageProps) {
data-agent-cluster-item="avatar"
style={avatarStyle}
>
-
+ {avatarUrl ? (
+
+ ) : (
+
+ )}
diff --git a/desktop/src/features/agents/ui/UnifiedAgentsSection.tsx b/desktop/src/features/agents/ui/UnifiedAgentsSection.tsx
index cad96d62f..d52318898 100644
--- a/desktop/src/features/agents/ui/UnifiedAgentsSection.tsx
+++ b/desktop/src/features/agents/ui/UnifiedAgentsSection.tsx
@@ -231,6 +231,7 @@ function AgentPersonaCard({
return (