running query gives content

Author: emrberk

src/components/AIStatusIndicator/AssistantModes.tsx

@@ -82,6 +82,7 @@ export const getIsExpandableSection = (section: OperationSection) => {
     AIOperationStatus.InvestigatingTable,
     AIOperationStatus.InvestigatingDocs,
     AIOperationStatus.Thinking,
+    AIOperationStatus.RunningQuery,
   ].includes(section.type)
 }
 

src/components/AIStatusIndicator/AssistantModesCompact.tsx

@@ -361,10 +361,12 @@ export const AssistantModesCompact: React.FC<AssistantModesCompactProps> = ({
       nextSection,
       !nextSection ? endTimestamp : undefined,
     )
-    const thinkingSegmentText =
-      section.type === AIOperationStatus.Thinking
-        ? (section.operations[0]?.content ?? "")
-        : ""
+    const isContentSection =
+      section.type === AIOperationStatus.Thinking ||
+      section.type === AIOperationStatus.RunningQuery
+    const sectionContentText = isContentSection
+      ? (section.operations[0]?.content ?? "")
+      : ""
 
     return (
       <ModeHeader
@@ -414,16 +416,16 @@ export const AssistantModesCompact: React.FC<AssistantModesCompactProps> = ({
             )}
           </Box>
         </ModeHeaderTop>
-        {isExpandable && section.type === AIOperationStatus.Thinking && (
+        {isExpandable && isContentSection && (
           <ExpandableWrapper $expanded={isExpanded}>
             <ExpandableContent>
-              {thinkingSegmentText ? (
-                <ThinkingContent>{thinkingSegmentText}</ThinkingContent>
+              {sectionContentText ? (
+                <ThinkingContent>{sectionContentText}</ThinkingContent>
               ) : null}
             </ExpandableContent>
           </ExpandableWrapper>
         )}
-        {isExpandable && section.type !== AIOperationStatus.Thinking && (
+        {isExpandable && !isContentSection && (
           <ExpandableWrapper $expanded={isExpanded}>
             <ExpandableContent>
               <ReasoningThread>

src/providers/AIStatusProvider/index.tsx

@@ -70,6 +70,7 @@ export type StatusArgs = {
   items?: Array<{ name: string; section?: string }>
   label?: string
   cellId?: string
+  content?: string
 }
 
 export type StatusEntry = {
@@ -186,6 +187,9 @@ export const AIStatusProvider: React.FC<AIStatusProviderProps> = ({
           type: newStatus,
           args: normalizedArgs ?? undefined,
           timestamp: Date.now(),
+          ...(normalizedArgs?.content !== undefined && {
+            content: normalizedArgs.content,
+          }),
         }
         if (
           statusRef.current === null ||

src/utils/tools/dispatch.ts

@@ -31,6 +31,7 @@ import { buildRunQueryPayload, RUN_QUERY_DEFAULT_LIMIT } from "./runQuery"
 import { eventBus } from "../../modules/EventBus"
 import { EventType } from "../../modules/EventBus/types"
 import type { ToolExecutionContext } from "../ai/shared"
+import { formatSql } from "../formatSql"
 
 const notebookErrorHint = (code: NotebookToolErrorCode): string => {
   switch (code) {
@@ -752,7 +753,13 @@ export const dispatchTool = async (
             return { content: decision.reason, is_error: true }
           }
         }
-        setStatus(AIOperationStatus.RunningQuery)
+        let formattedSql = sql
+        try {
+          formattedSql = formatSql(sql)
+        } catch {
+          // Fall back to raw SQL if the formatter can't parse it.
+        }
+        setStatus(AIOperationStatus.RunningQuery, { content: formattedSql })
         try {
           const requestedLimit =
             typeof limit === "number" && Number.isFinite(limit)