diff --git a/packages/ui/src/hooks/useAssistantStatus.ts b/packages/ui/src/hooks/useAssistantStatus.ts index 3926f305..a952fa02 100644 --- a/packages/ui/src/hooks/useAssistantStatus.ts +++ b/packages/ui/src/hooks/useAssistantStatus.ts @@ -224,7 +224,7 @@ export function useAssistantStatus(): AssistantStatusSnapshot { let activePartType: 'text' | 'tool' | 'reasoning' | 'editing' | undefined = undefined; let activeToolName: string | undefined = undefined; - const editingTools = new Set(['edit', 'write', 'apply_patch']); + const editingTools = new Set(['edit', 'write', 'multiedit', 'apply_patch']); for (let i = (lastAssistant.parts ?? []).length - 1; i >= 0; i -= 1) { const part = lastAssistant.parts?.[i]; @@ -245,6 +245,7 @@ export function useAssistantStatus(): AssistantStatusSnapshot { const toolName = getToolDisplayName(part); if (editingTools.has(toolName)) { activePartType = 'editing'; + activeToolName = toolName; } else { activePartType = 'tool'; activeToolName = toolName; @@ -317,7 +318,7 @@ export function useAssistantStatus(): AssistantStatusSnapshot { const isGenericStatus = activePartType === undefined; const statusText = (() => { - if (activePartType === 'editing') return 'editing file'; + if (activePartType === 'editing') return activeToolName === 'multiedit' ? getToolStatusPhrase(activeToolName) : 'editing file'; if (activePartType === 'tool' && activeToolName) return getToolStatusPhrase(activeToolName); if (activePartType === 'reasoning') return 'thinking'; if (activePartType === 'text') return 'composing';