salvacybersec/CodeNomad
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use tool prefixes in timeline

Browse Source
This commit is contained in:
Shantur Rathore
2025-12-08 13:40:42 +00:00
parent 235b9338a7
commit 81a9c28971
1 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
packages/ui/src/components/message-timeline.tsx
View File

@@ -45,6 +45,7 @@ interface PendingSegment {
texts: string[] texts: string[]
reasoningTexts: string[] reasoningTexts: string[]
toolTitles: string[] toolTitles: string[]
toolTypeLabels: string[]
hasPrimaryText: boolean hasPrimaryText: boolean
} }
@@ -117,6 +118,13 @@ function getToolTitle(part: ToolCallPart): string {
return TOOL_FALLBACK_LABEL return TOOL_FALLBACK_LABEL
} }
function getToolTypeLabel(part: ToolCallPart): string {
if (typeof part.tool === "string" && part.tool.trim().length > 0) {
return part.tool.trim().slice(0, 4)
}
return TOOL_FALLBACK_LABEL.slice(0, 4)
}
function formatTextsTooltip(texts: string[], fallback: string): string { function formatTextsTooltip(texts: string[], fallback: string): string {
const combined = texts const combined = texts
.map((text) => text.trim()) .map((text) => text.trim())
@@ -151,7 +159,9 @@ export function buildTimelineSegments(instanceId: string, record: MessageRecord)
pending = null pending = null
return return
} }
const label = pending.type === "tool" ? (pending.toolTitles[0] || SEGMENT_LABELS[pending.type]) : SEGMENT_LABELS[pending.type] const label = pending.type === "tool"
? pending.toolTypeLabels[0] || TOOL_FALLBACK_LABEL.slice(0, 4)
: SEGMENT_LABELS[pending.type]
const tooltip = pending.type === "tool" const tooltip = pending.type === "tool"
? formatToolTooltip(pending.toolTitles) ? formatToolTooltip(pending.toolTitles)
: formatTextsTooltip( : formatTextsTooltip(
@@ -170,12 +180,12 @@ export function buildTimelineSegments(instanceId: string, record: MessageRecord)
pending = null pending = null
} }
const ensureSegment = (type: TimelineSegmentType) => { const ensureSegment = (type: TimelineSegmentType): PendingSegment => {
if (!pending || pending.type !== type) { if (!pending || pending.type !== type) {
flushPending() flushPending()
pending = { type, texts: [], reasoningTexts: [], toolTitles: [], hasPrimaryText: type !== "assistant" } pending = { type, texts: [], reasoningTexts: [], toolTitles: [], toolTypeLabels: [], hasPrimaryText: type !== "assistant" }
} }
return pending return pending!
} }
@@ -186,7 +196,9 @@ export function buildTimelineSegments(instanceId: string, record: MessageRecord)
if (part.type === "tool") { if (part.type === "tool") {
const target = ensureSegment("tool") const target = ensureSegment("tool")
target.toolTitles.push(getToolTitle(part as ToolCallPart)) const toolPart = part as ToolCallPart
target.toolTitles.push(getToolTitle(toolPart))
target.toolTypeLabels.push(getToolTypeLabel(toolPart))
continue continue
} }