feat: add role-specific statistics to telemetry and UI (cont. #15234) (#18824)

Co-authored-by: Yuna Seol <yunaseol@google.com>

packages/cli/src/zed-integration/zedIntegration.test.ts

@@ -25,6 +25,7 @@ import {
   type GeminiChat,
   type Config,
   type MessageBus,
+  LlmRole,
 } from '@google/gemini-cli-core';
 import {
   SettingScope,
@@ -588,7 +589,8 @@ describe('Session', () => {
         }),
       ]),
       expect.anything(),
-      expect.anything(),
+      expect.any(AbortSignal),
+      LlmRole.MAIN,
     );
   });
 

packages/cli/src/zed-integration/zedIntegration.ts

@@ -35,6 +35,7 @@ import {
   startupProfiler,
   Kind,
   partListUnionToString,
+  LlmRole,
 } from '@google/gemini-cli-core';
 import * as acp from '@agentclientprotocol/sdk';
 import { AcpFileSystemService } from './fileSystemService.js';
@@ -493,6 +494,7 @@ export class Session {
           nextMessage?.parts ?? [],
           promptId,
           pendingSend.signal,
+          LlmRole.MAIN,
         );
         nextMessage = null;