From 848ede3d57167ae944ff4616eb0d56f174be87de Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E0=A4=B9=E0=A4=BF=E0=A4=AE=E0=A4=BE=E0=A4=82=E0=A4=B6?=
 =?UTF-8?q?=E0=A5=81?= <sharmahimanshu15082007@gmail.com>
Date: Wed, 6 May 2026 22:16:20 +0530
Subject: [PATCH] [studio]: Fix tool reasoning trace in UI  (#5314)

* fix thought for 1 second issue

* gemini suggesion
---
 .../src/components/assistant-ui/reasoning.tsx | 23 +++++++++++++++----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/studio/frontend/src/components/assistant-ui/reasoning.tsx b/studio/frontend/src/components/assistant-ui/reasoning.tsx
index 387f8cd458..fe913baf2a 100644
--- a/studio/frontend/src/components/assistant-ui/reasoning.tsx
+++ b/studio/frontend/src/components/assistant-ui/reasoning.tsx
@@ -316,15 +316,28 @@ const ReasoningGroupImpl: ReasoningGroupComponent = ({
     if (message.status?.type !== "running") {
       return false;
     }
-    const lastIndex = message.parts.length - 1;
-    if (lastIndex < 0) {
+    const parts = message.parts;
+    const len = parts.length;
+    if (len === 0) {
       return false;
     }
-    const lastType = message.parts[lastIndex]?.type;
-    if (lastType !== "reasoning") {
+
+    let groupHasReasoning = false;
+    for (let i = startIndex; i <= endIndex && i < len; i += 1) {
+      if (parts[i]?.type === "reasoning") {
+        groupHasReasoning = true;
+        break;
+      }
+    }
+    if (!groupHasReasoning) {
       return false;
     }
-    return lastIndex >= startIndex && lastIndex <= endIndex;
+    for (let i = endIndex + 1; i < len; i += 1) {
+      if (parts[i]?.type !== "tool-call") {
+        return false;
+      }
+    }
+    return true;
   });
 
   const persistedDuration = useAuiState(({ message }) => {