1 files changed, 14 insertions, 6 deletions
diff --git a/packages/core/src/tools/shell.ts b/packages/core/src/tools/shell.ts
index 7e79c717..3dc3d0a6 100644
--- a/packages/core/src/tools/shell.ts
+++ b/packages/core/src/tools/shell.ts
@@ -489,14 +489,22 @@ Process Group PGID: Process group started or \`(none)\``,
       }
     }
 
-    const summary = await summarizeToolOutput(
-      llmContent,
-      this.config.getGeminiClient(),
-      abortSignal,
-    );
+    const summarizeConfig = this.config.getSummarizeToolOutputConfig();
+    if (summarizeConfig && summarizeConfig[this.name]) {
+      const summary = await summarizeToolOutput(
+        llmContent,
+        this.config.getGeminiClient(),
+        abortSignal,
+        summarizeConfig[this.name].tokenBudget,
+      );
+      return {
+        llmContent: summary,
+        returnDisplay: returnDisplayMessage,
+      };
+    }
 
     return {
-      llmContent: summary,
+      llmContent,
       returnDisplay: returnDisplayMessage,
     };
   }