From f77e4716fa347894d7c063733f7eb69220005ceb Mon Sep 17 00:00:00 2001
From: Adam Weidman <adamfweidman@gmail.com>
Date: Fri, 20 Mar 2026 11:19:34 -0400
Subject: [PATCH] !feat(cli): adopt protocol-backed agent session streaming

---
 packages/cli/src/nonInteractiveCli.test.ts | 4 +---
 packages/cli/src/nonInteractiveCli.ts      | 2 +-
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/packages/cli/src/nonInteractiveCli.test.ts b/packages/cli/src/nonInteractiveCli.test.ts
index 3fc827d497..e83e1e60c9 100644
--- a/packages/cli/src/nonInteractiveCli.test.ts
+++ b/packages/cli/src/nonInteractiveCli.test.ts
@@ -300,9 +300,7 @@ describe('runNonInteractive', () => {
       prompt_id: 'prompt-id-stream',
     });
 
-    expect(streamSpy).toHaveBeenCalledWith({
-      streamId: expect.any(String),
-    });
+    expect(streamSpy).toHaveBeenCalledWith({ streamId: expect.any(String) });
   });
 
   it('should register activity logger when GEMINI_CLI_ACTIVITY_LOG_TARGET is set', async () => {
diff --git a/packages/cli/src/nonInteractiveCli.ts b/packages/cli/src/nonInteractiveCli.ts
index e6a85a4886..9b88011058 100644
--- a/packages/cli/src/nonInteractiveCli.ts
+++ b/packages/cli/src/nonInteractiveCli.ts
@@ -466,7 +466,7 @@ export async function runNonInteractive({
             }
             break;
           }
-          case 'stream_end': {
+          case 'agent_end': {
             if (event.reason === 'aborted') {
               handleCancellationError(config);
             } else if (event.reason === 'max_turns') {