From 5aa6888a0ece71f0c595ca3d7be999b578e4bedb Mon Sep 17 00:00:00 2001
From: liangfung <jueliang@tabbyml.com>
Date: Wed, 5 Feb 2025 17:31:53 +0800
Subject: [PATCH] update: layout

---
 .../app/search/components/agent-steps.tsx      |  5 +++--
 .../components/assistant-message-section.tsx   | 18 +++++++++---------
 2 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/ee/tabby-ui/app/search/components/agent-steps.tsx b/ee/tabby-ui/app/search/components/agent-steps.tsx
index ff5812ac8faf..c971fa9220db 100644
--- a/ee/tabby-ui/app/search/components/agent-steps.tsx
+++ b/ee/tabby-ui/app/search/components/agent-steps.tsx
@@ -38,9 +38,10 @@ export function AgentSteps({
     <Accordion
       type="single"
       collapsible
-      className={cn('w-full border rounded-lg px-4 my-1.5', className)}
+      defaultValue="steps"
+      className={cn('w-full border rounded-lg px-4', className)}
     >
-      <AccordionItem value="item" className="border-0">
+      <AccordionItem value="steps" className="border-0">
         <AccordionTrigger className="w-full">
           <div className="font-semibold">Tabby Agent</div>
         </AccordionTrigger>
diff --git a/ee/tabby-ui/app/search/components/assistant-message-section.tsx b/ee/tabby-ui/app/search/components/assistant-message-section.tsx
index 0651ade4519d..9db63479e92a 100644
--- a/ee/tabby-ui/app/search/components/assistant-message-section.tsx
+++ b/ee/tabby-ui/app/search/components/assistant-message-section.tsx
@@ -266,6 +266,15 @@ export function AssistantMessageSection({
 
   return (
     <div className={cn('flex flex-col gap-y-5', className)}>
+      {/* expose steps */}
+      {!!message.readingCode && (
+        <AgentSteps
+          steps={message.readingCode}
+          codeSourceId={message.codeSourceId}
+          isReadingCode={message.isReadingCode}
+        />
+      )}
+
       {/* document search hits */}
       {messageAttachmentDocs && messageAttachmentDocs.length > 0 && (
         <div>
@@ -331,15 +340,6 @@ export function AssistantMessageSection({
           )}
         </div>
 
-        {/* expose step */}
-        {!!message.readingCode && (
-          <AgentSteps
-            steps={message.readingCode}
-            codeSourceId={message.codeSourceId}
-            isReadingCode={message.isReadingCode}
-          />
-        )}
-
         {/* attachment clientCode & code */}
         {messageAttachmentCodeLen > 0 && (
           <CodeReferences