render tool output summarization to UI post summarization call (non-streaming)

Author: sidwan02

packages/cli/src/ui/components/messages/SubagentHistoryDisplay.tsx

@@ -11,6 +11,7 @@ import type {
   SubagentToolCallHistoryItem,
   SubagentToolResponseHistoryItem,
   SubagentThoughtHistoryItem,
+  SubagentToolSummaryHistoryItem,
 } from '../../types.js';
 import { MarkdownDisplay } from '../../utils/MarkdownDisplay.js';
 import { SubagentToolCallDisplay } from './SubagentToolCallDisplay.js';
@@ -25,6 +26,7 @@ type SubagentTurn = {
   thought?: SubagentThoughtHistoryItem;
   toolCall?: SubagentToolCallHistoryItem;
   toolResponse?: SubagentToolResponseHistoryItem;
+  toolSummary?: SubagentToolSummaryHistoryItem;
   subagentHistory?: SubagentHistoryItem[];
 };
 
@@ -77,9 +79,17 @@ export const SubagentHistoryDisplay: React.FC<SubagentHistoryDisplayProps> = ({
 
     switch (item.type) {
       case 'thought':
-        if (lastTurn.thought) {
+        // If the current turn already contains any tool-related activity,
+        // this new thought must start a new turn.
+        if (
+          lastTurn.toolCall ||
+          lastTurn.toolResponse ||
+          lastTurn.toolSummary
+        ) {
           acc.push({ thought: item });
         } else {
+          // Otherwise, it's either the first thought of a turn or a
+          // streaming update to the existing thought.
           lastTurn.thought = item;
         }
         break;
@@ -90,6 +100,13 @@ export const SubagentHistoryDisplay: React.FC<SubagentHistoryDisplayProps> = ({
           lastTurn.toolCall = item;
         }
         break;
+      case 'tool_summary':
+        if (lastTurn.toolSummary) {
+          acc.push({ toolSummary: item });
+        } else {
+          lastTurn.toolSummary = item;
+        }
+        break;
       // TODO: Shell tool output is very glitchy.
       // TODO: there's a bug when multiple shell tool calls are made in succession (for tool_output_chunk), they get merged into one response.
       case 'tool_response':
@@ -180,6 +197,24 @@ export const SubagentHistoryDisplay: React.FC<SubagentHistoryDisplayProps> = ({
                   />
                 </Box>
               )}
+              {turn.toolSummary && (
+                <Box
+                  flexDirection="column"
+                  marginBottom={1}
+                  width={availableWidth}
+                  borderStyle="single"
+                  borderColor="cyan"
+                  paddingX={1}
+                  paddingY={0}
+                >
+                  <Text color="cyan">Summarized Tool Output:</Text>
+                  <MarkdownDisplay
+                    text={turn.toolSummary.data.summary}
+                    isPending={false}
+                    terminalWidth={availableWidth - 2} // Adjust width for padding
+                  />
+                </Box>
+              )}
             </Box>
           );
         }

packages/cli/src/ui/hooks/useReactToolScheduler.ts

@@ -111,6 +111,7 @@ export function useReactToolScheduler(
           'GEMINI_SUBAGENT_THOUGHT::': 'thought',
           'GEMINI_SUBAGENT_TOOL_CALL::': 'tool_call',
           'GEMINI_SUBAGENT_TOOL_RESPONSE::': 'tool_response',
+          'GEMINI_SUBAGENT_TOOL_SUMMARY::': 'tool_summary',
           'GEMINI_SUBAGENT_TOOL_OUTPUT_CHUNK::': 'tool_output_chunk',
           'GEMINI_SUBAGENT_ERROR::': 'error',
         };

packages/cli/src/ui/types.ts

@@ -404,7 +404,13 @@ export type SubagentHistoryItem =
       data: { error: string; context?: string; name?: string };
     }
   | { type: 'interrupted'; data: { message: string } }
-  | { type: 'user_message'; data: { message: string } };
+  | { type: 'user_message'; data: { message: string } }
+  | SubagentToolSummaryHistoryItem;
+
+export type SubagentToolSummaryHistoryItem = {
+  type: 'tool_summary';
+  data: { summary: string };
+};
 
 export type SubagentThoughtHistoryItem = Extract<
   SubagentHistoryItem,

packages/core/src/agents/executor.ts

@@ -1362,6 +1362,9 @@ export class AgentExecutor<TOutput extends z.ZodTypeAny> {
             goal,
           );
           if (summary) {
+            this.emitActivity('TOOL_SUMMARY', {
+              summary,
+            });
             debugLogger.log(
               `[AgentExecutor] Summarized output for tool: ${functionCall.name} (ID: ${callId})`,
             );

packages/core/src/agents/invocation.ts

@@ -124,6 +124,11 @@ export class SubagentInvocation<
               )}`,
             );
             break;
+          case 'TOOL_SUMMARY':
+            updateOutput(
+              `GEMINI_SUBAGENT_TOOL_SUMMARY::${JSON.stringify(activity.data)}`,
+            );
+            break;
           case 'ERROR':
             updateOutput(
               `GEMINI_SUBAGENT_ERROR::${JSON.stringify(activity.data)}`,

packages/core/src/agents/types.ts

@@ -49,7 +49,8 @@ export interface SubagentActivityEvent {
     | 'TOOL_CALL_END'
     | 'THOUGHT_CHUNK'
     | 'ERROR'
-    | 'TOOL_OUTPUT_CHUNK';
+    | 'TOOL_OUTPUT_CHUNK'
+    | 'TOOL_SUMMARY';
   data: Record<string, unknown>;
 }
 

understand.md

@@ -225,6 +225,7 @@ Using the codebase investigator: tell me how the ollama inference works.
 
 - `use the build and test tool: run npm test -- src/tools/glob.test.ts`
 - `use the build and test tool: test src/tools/glob.test.ts`
+- `use the build and test tool: run npm test -- src/tools/glob.test.ts in the core workspace`
 
 # TODO: