From 44a499e62cf7d51fcdfe9307d26f39d378090cfe Mon Sep 17 00:00:00 2001
From: matt <matthewkim20122012@gmail.com>
Date: Sun, 15 Feb 2026 14:32:32 +0900
Subject: [PATCH] feat(jsonl): enhance message counting logic for AIGroup
 interactions

- Added logic to await the first main-thread assistant message after a UserGroup to accurately count AIGroup messages.
- Updated tests to reflect the new message counting behavior, ensuring correct results in session file analysis.
---
 src/main/utils/jsonl.ts       | 11 +++++++++++
 test/main/utils/jsonl.test.ts |  2 +-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/main/utils/jsonl.ts b/src/main/utils/jsonl.ts
index 15b2cc1f..e09a922a 100644
--- a/src/main/utils/jsonl.ts
+++ b/src/main/utils/jsonl.ts
@@ -328,6 +328,8 @@ export async function analyzeSessionFileMetadata(
   let firstUserMessage: { text: string; timestamp: string } | null = null;
   let firstCommandMessage: { text: string; timestamp: string } | null = null;
   let messageCount = 0;
+  // After a UserGroup, await the first main-thread assistant message to count the AIGroup
+  let awaitingAIGroup = false;
   let gitBranch: string | null = null;
 
   let activityIndex = 0;
@@ -357,6 +359,15 @@ export async function analyzeSessionFileMetadata(
 
     if (isParsedUserChunkMessage(parsed)) {
       messageCount++;
+      awaitingAIGroup = true;
+    } else if (
+      awaitingAIGroup &&
+      parsed.type === 'assistant' &&
+      parsed.model !== '<synthetic>' &&
+      !parsed.isSidechain
+    ) {
+      messageCount++;
+      awaitingAIGroup = false;
     }
 
     if (!gitBranch && 'gitBranch' in entry && entry.gitBranch) {
diff --git a/test/main/utils/jsonl.test.ts b/test/main/utils/jsonl.test.ts
index e748b821..c7a1cf05 100644
--- a/test/main/utils/jsonl.test.ts
+++ b/test/main/utils/jsonl.test.ts
@@ -166,7 +166,7 @@ describe('jsonl', () => {
 
         expect(result.firstUserMessage?.text).toBe('hello world');
         expect(result.firstUserMessage?.timestamp).toBe('2026-01-01T00:00:00.000Z');
-        expect(result.messageCount).toBe(1);
+        expect(result.messageCount).toBe(2);
         expect(result.isOngoing).toBe(true);
         expect(result.gitBranch).toBe('feature/test');
       } finally {