feat: label pending and final agent messages

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
2026-07-29 07:58:47 +02:00 · 2026-03-22 09:46:13 +01:00
parent 16229b8b0a
commit 2f90a19736
3 changed files with 154 additions and 4 deletions
@@ -0,0 +1,87 @@
+import { describe, expect, test } from 'bun:test';
+
+import { getAssistantMessagePhase } from '@renderer/lib/messagePhase';
+import type { SessionRecord } from '@shared/domain/session';
+
+function createSession(
+  messages: SessionRecord['messages'],
+  status: SessionRecord['status'] = 'idle',
+): SessionRecord {
+  return {
+    id: 'session-1',
+    projectId: 'project-1',
+    patternId: 'pattern-1',
+    title: 'Test session',
+    createdAt: '2026-03-23T00:00:00.000Z',
+    updatedAt: '2026-03-23T00:00:00.000Z',
+    status,
+    messages,
+  };
+}
+
+describe('assistant message phase', () => {
+  test('marks pending assistant messages as thinking', () => {
+    const session = createSession([
+      {
+        id: 'msg-1',
+        role: 'assistant',
+        authorName: 'Triage',
+        content: 'Draft',
+        createdAt: '2026-03-23T00:00:00.000Z',
+        pending: true,
+      },
+    ], 'running');
+
+    expect(getAssistantMessagePhase(session, session.messages[0], 0)).toBe('thinking');
+  });
+
+  test('marks the last completed assistant message as final when the session is idle', () => {
+    const session = createSession([
+      {
+        id: 'msg-1',
+        role: 'assistant',
+        authorName: 'Triage',
+        content: 'Earlier',
+        createdAt: '2026-03-23T00:00:00.000Z',
+      },
+      {
+        id: 'msg-2',
+        role: 'assistant',
+        authorName: 'UX Specialist',
+        content: 'Final answer',
+        createdAt: '2026-03-23T00:00:01.000Z',
+      },
+    ]);
+
+    expect(getAssistantMessagePhase(session, session.messages[0], 0)).toBe('default');
+    expect(getAssistantMessagePhase(session, session.messages[1], 1)).toBe('final');
+  });
+
+  test('does not mark completed assistant messages as final while the session is still running', () => {
+    const session = createSession([
+      {
+        id: 'msg-1',
+        role: 'assistant',
+        authorName: 'Triage',
+        content: 'In progress',
+        createdAt: '2026-03-23T00:00:00.000Z',
+      },
+    ], 'running');
+
+    expect(getAssistantMessagePhase(session, session.messages[0], 0)).toBe('default');
+  });
+
+  test('ignores non-assistant messages', () => {
+    const session = createSession([
+      {
+        id: 'msg-1',
+        role: 'user',
+        authorName: 'You',
+        content: 'Hello',
+        createdAt: '2026-03-23T00:00:00.000Z',
+      },
+    ]);
+
+    expect(getAssistantMessagePhase(session, session.messages[0], 0)).toBe('default');
+  });
+});