tweak: show tool name in timeline chip

This commit is contained in:
Shantur Rathore
2025-12-08 10:54:50 +00:00
parent 67ff00d83e
commit 642d5e22e6

View File

@@ -32,7 +32,7 @@ const SEGMENT_LABELS: Record<TimelineSegmentType, string> = {
const SEGMENT_SHORT_LABELS: Record<TimelineSegmentType, string> = { const SEGMENT_SHORT_LABELS: Record<TimelineSegmentType, string> = {
user: "U", user: "U",
assistant: "A", assistant: "A",
tool: "T", tool: "Tool",
} }
const TOOL_FALLBACK_LABEL = "Tool Call" const TOOL_FALLBACK_LABEL = "Tool Call"
@@ -151,7 +151,7 @@ export function buildTimelineSegments(instanceId: string, record: MessageRecord)
pending = null pending = null
return return
} }
const label = SEGMENT_LABELS[pending.type] const label = pending.type === "tool" ? (pending.toolTitles[0] || SEGMENT_LABELS[pending.type]) : SEGMENT_LABELS[pending.type]
const tooltip = pending.type === "tool" const tooltip = pending.type === "tool"
? formatToolTooltip(pending.toolTitles) ? formatToolTooltip(pending.toolTitles)
: formatTextsTooltip( : formatTextsTooltip(
@@ -306,7 +306,9 @@ const MessageTimeline: Component<MessageTimelineProps> = (props) => {
onMouseLeave={handleMouseLeave} onMouseLeave={handleMouseLeave}
> >
<span class="message-timeline-label message-timeline-label-full">{segment.label}</span> <span class="message-timeline-label message-timeline-label-full">{segment.label}</span>
<span class="message-timeline-label message-timeline-label-short">{SEGMENT_SHORT_LABELS[segment.type]}</span> <span class="message-timeline-label message-timeline-label-short">
{segment.type === "tool" ? (segment.label.slice(0, 4).toUpperCase()) : SEGMENT_SHORT_LABELS[segment.type]}
</span>
</button> </button>
) )
}} }}