Fix noise in headless mode on STDOUT (#1184)

1 files changed, 5 insertions, 0 deletions

diff --git a/packages/cli/src/nonInteractiveCli.ts b/packages/cli/src/nonInteractiveCli.ts
index 3efb8452..c5a89575 100644
--- a/packages/cli/src/nonInteractiveCli.ts
+++ b/packages/cli/src/nonInteractiveCli.ts
@@ -27,6 +27,11 @@ function getResponseText(response: GenerateContentResponse): string | null {
       candidate.content.parts &&
       candidate.content.parts.length > 0
     ) {
+      // We are running in headless mode so we don't need to return thoughts to STDOUT.
+      const thoughtPart = candidate.content.parts[0];
+      if (thoughtPart?.thought) {
+        return null;
+      }
       return candidate.content.parts
         .filter((part) => part.text)
         .map((part) => part.text)