From f1317d7f71003c7c9c41963e99dcdeee933cb67e Mon Sep 17 00:00:00 2001
From: Isaac Sanchez-Hawkins <266845420+isanchez404@users.noreply.github.com>
Date: Tue, 12 May 2026 03:57:04 -0400
Subject: [PATCH] fix(status): classify multiedit as editing (#1209)

* fix(status): classify multiedit as editing

* fix(status): preserve multiedit status phrase

---------

Co-authored-by: Isaac Sanchez <isanchez-hawkins@arize.com>
---
 packages/ui/src/hooks/useAssistantStatus.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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';