diff --git a/ui/src/components/CommentThread.tsx b/ui/src/components/CommentThread.tsx
index b96c1af0..f529382b 100644
--- a/ui/src/components/CommentThread.tsx
+++ b/ui/src/components/CommentThread.tsx
@@ -8,6 +8,7 @@ import { InlineEntitySelector, type InlineEntityOption } from "./InlineEntitySel
import { MarkdownBody } from "./MarkdownBody";
import { MarkdownEditor, type MarkdownEditorRef, type MentionOption } from "./MarkdownEditor";
import { StatusBadge } from "./StatusBadge";
+import { AgentIcon } from "./AgentIconPicker";
import { formatDateTime } from "../lib/utils";
interface CommentWithRunMeta extends IssueComment {
@@ -385,6 +386,32 @@ export function CommentThread({
emptyMessage="No assignees found."
onChange={setReassignTarget}
className="text-xs h-8"
+ renderTriggerValue={(option) => {
+ if (!option) return Assignee;
+ const agentId = option.id.startsWith("agent:") ? option.id.slice("agent:".length) : null;
+ const agent = agentId ? agentMap?.get(agentId) : null;
+ return (
+ <>
+ {agent ? (
+
+ ) : null}
+ {option.label}
+ >
+ );
+ }}
+ renderOption={(option) => {
+ if (!option.id) return {option.label};
+ const agentId = option.id.startsWith("agent:") ? option.id.slice("agent:".length) : null;
+ const agent = agentId ? agentMap?.get(agentId) : null;
+ return (
+ <>
+ {agent ? (
+
+ ) : null}
+ {option.label}
+ >
+ );
+ }}
/>
)}