@umsai/ums-code 0.3.0-v2 → 0.6.0-v1

package/dist/src/acp-integration/session/HistoryReplayer.test.js

@@ -0,0 +1,374 @@
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { HistoryReplayer } from './HistoryReplayer.js';
+describe('HistoryReplayer', () => {
+    let mockContext;
+    let sendUpdateSpy;
+    let replayer;
+    beforeEach(() => {
+        sendUpdateSpy = vi.fn().mockResolvedValue(undefined);
+        const mockToolRegistry = {
+            getTool: vi.fn().mockReturnValue(null),
+        };
+        mockContext = {
+            sessionId: 'test-session-id',
+            config: {
+                getToolRegistry: () => mockToolRegistry,
+            },
+            sendUpdate: sendUpdateSpy,
+        };
+        replayer = new HistoryReplayer(mockContext);
+    });
+    const createUserRecord = (text) => ({
+        uuid: 'user-uuid',
+        parentUuid: null,
+        sessionId: 'test-session',
+        timestamp: new Date().toISOString(),
+        type: 'user',
+        cwd: '/test',
+        version: '1.0.0',
+        message: {
+            role: 'user',
+            parts: [{ text }],
+        },
+    });
+    const createAssistantRecord = (text, thought = false) => ({
+        uuid: 'assistant-uuid',
+        parentUuid: 'user-uuid',
+        sessionId: 'test-session',
+        timestamp: new Date().toISOString(),
+        type: 'assistant',
+        cwd: '/test',
+        version: '1.0.0',
+        message: {
+            role: 'model',
+            parts: [{ text, thought }],
+        },
+    });
+    const createToolResultRecord = (toolName, resultDisplay, hasError = false) => ({
+        uuid: 'tool-uuid',
+        parentUuid: 'assistant-uuid',
+        sessionId: 'test-session',
+        timestamp: new Date().toISOString(),
+        type: 'tool_result',
+        cwd: '/test',
+        version: '1.0.0',
+        message: {
+            role: 'user',
+            parts: [
+                {
+                    functionResponse: {
+                        name: toolName,
+                        response: { result: 'ok' },
+                    },
+                },
+            ],
+        },
+        toolCallResult: {
+            callId: 'call-123',
+            responseParts: [],
+            resultDisplay,
+            error: hasError ? new Error('Tool failed') : undefined,
+            errorType: undefined,
+        },
+    });
+    describe('replay', () => {
+        it('should replay empty records array', async () => {
+            await replayer.replay([]);
+            expect(sendUpdateSpy).not.toHaveBeenCalled();
+        });
+        it('should replay records in order', async () => {
+            const records = [
+                createUserRecord('Hello'),
+                createAssistantRecord('Hi there'),
+            ];
+            await replayer.replay(records);
+            expect(sendUpdateSpy).toHaveBeenCalledTimes(2);
+            expect(sendUpdateSpy.mock.calls[0][0].sessionUpdate).toBe('user_message_chunk');
+            expect(sendUpdateSpy.mock.calls[1][0].sessionUpdate).toBe('agent_message_chunk');
+        });
+    });
+    describe('user message replay', () => {
+        it('should emit user_message_chunk for user records', async () => {
+            const records = [createUserRecord('Hello, world!')];
+            await replayer.replay(records);
+            expect(sendUpdateSpy).toHaveBeenCalledWith({
+                sessionUpdate: 'user_message_chunk',
+                content: { type: 'text', text: 'Hello, world!' },
+            });
+        });
+        it('should skip user records without message', async () => {
+            const record = {
+                ...createUserRecord('test'),
+                message: undefined,
+            };
+            await replayer.replay([record]);
+            expect(sendUpdateSpy).not.toHaveBeenCalled();
+        });
+    });
+    describe('assistant message replay', () => {
+        it('should emit agent_message_chunk for assistant records', async () => {
+            const records = [createAssistantRecord('I can help with that.')];
+            await replayer.replay(records);
+            expect(sendUpdateSpy).toHaveBeenCalledWith({
+                sessionUpdate: 'agent_message_chunk',
+                content: { type: 'text', text: 'I can help with that.' },
+            });
+        });
+        it('should emit agent_thought_chunk for thought parts', async () => {
+            const records = [createAssistantRecord('Thinking about this...', true)];
+            await replayer.replay(records);
+            expect(sendUpdateSpy).toHaveBeenCalledWith({
+                sessionUpdate: 'agent_thought_chunk',
+                content: { type: 'text', text: 'Thinking about this...' },
+            });
+        });
+        it('should handle assistant records with multiple parts', async () => {
+            const record = {
+                ...createAssistantRecord('First'),
+                message: {
+                    role: 'model',
+                    parts: [
+                        { text: 'First part' },
+                        { text: 'Second part', thought: true },
+                        { text: 'Third part' },
+                    ],
+                },
+            };
+            await replayer.replay([record]);
+            expect(sendUpdateSpy).toHaveBeenCalledTimes(3);
+            expect(sendUpdateSpy.mock.calls[0][0]).toEqual({
+                sessionUpdate: 'agent_message_chunk',
+                content: { type: 'text', text: 'First part' },
+            });
+            expect(sendUpdateSpy.mock.calls[1][0]).toEqual({
+                sessionUpdate: 'agent_thought_chunk',
+                content: { type: 'text', text: 'Second part' },
+            });
+            expect(sendUpdateSpy.mock.calls[2][0]).toEqual({
+                sessionUpdate: 'agent_message_chunk',
+                content: { type: 'text', text: 'Third part' },
+            });
+        });
+    });
+    describe('function call replay', () => {
+        it('should emit tool_call for function call parts', async () => {
+            const record = {
+                ...createAssistantRecord(''),
+                message: {
+                    role: 'model',
+                    parts: [
+                        {
+                            functionCall: {
+                                name: 'read_file',
+                                args: { path: '/test.ts' },
+                            },
+                        },
+                    ],
+                },
+            };
+            await replayer.replay([record]);
+            expect(sendUpdateSpy).toHaveBeenCalledWith(expect.objectContaining({
+                sessionUpdate: 'tool_call',
+                status: 'in_progress',
+                title: 'read_file',
+                rawInput: { path: '/test.ts' },
+            }));
+        });
+        it('should use function call id as callId when available', async () => {
+            const record = {
+                ...createAssistantRecord(''),
+                message: {
+                    role: 'model',
+                    parts: [
+                        {
+                            functionCall: {
+                                id: 'custom-call-id',
+                                name: 'read_file',
+                                args: {},
+                            },
+                        },
+                    ],
+                },
+            };
+            await replayer.replay([record]);
+            expect(sendUpdateSpy).toHaveBeenCalledWith(expect.objectContaining({
+                toolCallId: 'custom-call-id',
+            }));
+        });
+    });
+    describe('tool result replay', () => {
+        it('should emit tool_call_update for tool result records', async () => {
+            const records = [
+                createToolResultRecord('read_file', 'File contents here'),
+            ];
+            await replayer.replay(records);
+            expect(sendUpdateSpy).toHaveBeenCalledWith({
+                sessionUpdate: 'tool_call_update',
+                toolCallId: 'call-123',
+                status: 'completed',
+                content: [
+                    {
+                        type: 'content',
+                        // Content comes from functionResponse.response (stringified)
+                        content: { type: 'text', text: '{"result":"ok"}' },
+                    },
+                ],
+                // resultDisplay is included as rawOutput
+                rawOutput: 'File contents here',
+            });
+        });
+        it('should emit failed status for tool results with errors', async () => {
+            const records = [createToolResultRecord('failing_tool', undefined, true)];
+            await replayer.replay(records);
+            expect(sendUpdateSpy).toHaveBeenCalledWith(expect.objectContaining({
+                sessionUpdate: 'tool_call_update',
+                status: 'failed',
+            }));
+        });
+        it('should emit plan update for TodoWriteTool results', async () => {
+            const todoDisplay = {
+                type: 'todo_list',
+                todos: [
+                    { id: '1', content: 'Task 1', status: 'pending' },
+                    { id: '2', content: 'Task 2', status: 'completed' },
+                ],
+            };
+            const record = createToolResultRecord('todo_write', todoDisplay);
+            // Override the function response name
+            record.message = {
+                role: 'user',
+                parts: [
+                    {
+                        functionResponse: {
+                            name: 'todo_write',
+                            response: { result: 'ok' },
+                        },
+                    },
+                ],
+            };
+            await replayer.replay([record]);
+            expect(sendUpdateSpy).toHaveBeenCalledWith({
+                sessionUpdate: 'plan',
+                entries: [
+                    { content: 'Task 1', priority: 'medium', status: 'pending' },
+                    { content: 'Task 2', priority: 'medium', status: 'completed' },
+                ],
+            });
+        });
+        it('should use record uuid as callId when toolCallResult.callId is missing', async () => {
+            const record = {
+                ...createToolResultRecord('test_tool'),
+                uuid: 'fallback-uuid',
+                toolCallResult: {
+                    callId: undefined,
+                    responseParts: [],
+                    resultDisplay: 'Result',
+                    error: undefined,
+                    errorType: undefined,
+                },
+            };
+            await replayer.replay([record]);
+            expect(sendUpdateSpy).toHaveBeenCalledWith(expect.objectContaining({
+                toolCallId: 'fallback-uuid',
+            }));
+        });
+    });
+    describe('system records', () => {
+        it('should skip system records', async () => {
+            const systemRecord = {
+                uuid: 'system-uuid',
+                parentUuid: null,
+                sessionId: 'test-session',
+                timestamp: new Date().toISOString(),
+                type: 'system',
+                subtype: 'chat_compression',
+                cwd: '/test',
+                version: '1.0.0',
+            };
+            await replayer.replay([systemRecord]);
+            expect(sendUpdateSpy).not.toHaveBeenCalled();
+        });
+    });
+    describe('mixed record types', () => {
+        it('should handle a complete conversation replay', async () => {
+            const records = [
+                createUserRecord('Read the file test.ts'),
+                {
+                    ...createAssistantRecord(''),
+                    message: {
+                        role: 'model',
+                        parts: [
+                            { text: "I'll read that file for you.", thought: true },
+                            {
+                                functionCall: {
+                                    id: 'call-read',
+                                    name: 'read_file',
+                                    args: { path: 'test.ts' },
+                                },
+                            },
+                        ],
+                    },
+                },
+                createToolResultRecord('read_file', 'export const x = 1;'),
+                createAssistantRecord('The file contains a simple export.'),
+            ];
+            await replayer.replay(records);
+            // Verify order and types of updates
+            const updateTypes = sendUpdateSpy.mock.calls.map((call) => call[0].sessionUpdate);
+            expect(updateTypes).toEqual([
+                'user_message_chunk',
+                'agent_thought_chunk',
+                'tool_call',
+                'tool_call_update',
+                'agent_message_chunk',
+            ]);
+        });
+    });
+    describe('usage metadata replay', () => {
+        it('should emit usage metadata after assistant message content', async () => {
+            const record = {
+                uuid: 'assistant-uuid',
+                parentUuid: 'user-uuid',
+                sessionId: 'test-session',
+                timestamp: new Date().toISOString(),
+                type: 'assistant',
+                cwd: '/test',
+                version: '1.0.0',
+                message: {
+                    role: 'model',
+                    parts: [{ text: 'Hello!' }],
+                },
+                usageMetadata: {
+                    promptTokenCount: 100,
+                    candidatesTokenCount: 50,
+                    totalTokenCount: 150,
+                },
+            };
+            await replayer.replay([record]);
+            expect(sendUpdateSpy).toHaveBeenCalledTimes(2);
+            expect(sendUpdateSpy).toHaveBeenNthCalledWith(1, {
+                sessionUpdate: 'agent_message_chunk',
+                content: { type: 'text', text: 'Hello!' },
+            });
+            expect(sendUpdateSpy).toHaveBeenNthCalledWith(2, {
+                sessionUpdate: 'agent_message_chunk',
+                content: { type: 'text', text: '' },
+                _meta: {
+                    usage: {
+                        promptTokens: 100,
+                        completionTokens: 50,
+                        thoughtsTokens: undefined,
+                        totalTokens: 150,
+                        cachedTokens: undefined,
+                    },
+                },
+            });
+        });
+    });
+});
+//# sourceMappingURL=HistoryReplayer.test.js.map

package/dist/src/acp-integration/session/HistoryReplayer.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"HistoryReplayer.test.js","sourceRoot":"","sources":["../../../../src/acp-integration/session/HistoryReplayer.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAUvD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,WAA2B,CAAC;IAChC,IAAI,aAAuC,CAAC;IAC5C,IAAI,QAAyB,CAAC;IAE9B,UAAU,CAAC,GAAG,EAAE;QACd,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,gBAAgB,GAAG;YACvB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;SACZ,CAAC;QAE7B,WAAW,GAAG;YACZ,SAAS,EAAE,iBAAiB;YAC5B,MAAM,EAAE;gBACN,eAAe,EAAE,GAAG,EAAE,CAAC,gBAAgB;aACnB;YACtB,UAAU,EAAE,aAAa;SAC1B,CAAC;QAEF,QAAQ,GAAG,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,CAAC,IAAY,EAAc,EAAE,CAAC,CAAC;QACtD,IAAI,EAAE,WAAW;QACjB,UAAU,EAAE,IAAI;QAChB,SAAS,EAAE,cAAc;QACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,IAAI,EAAE,MAAM;QACZ,GAAG,EAAE,OAAO;QACZ,OAAO,EAAE,OAAO;QAChB,OAAO,EAAE;YACP,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC;SAClB;KACF,CAAC,CAAC;IAEH,MAAM,qBAAqB,GAAG,CAC5B,IAAY,EACZ,OAAO,GAAG,KAAK,EACH,EAAE,CAAC,CAAC;QAChB,IAAI,EAAE,gBAAgB;QACtB,UAAU,EAAE,WAAW;QACvB,SAAS,EAAE,cAAc;QACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,IAAI,EAAE,WAAW;QACjB,GAAG,EAAE,OAAO;QACZ,OAAO,EAAE,OAAO;QAChB,OAAO,EAAE;YACP,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;SAC3B;KACF,CAAC,CAAC;IAEH,MAAM,sBAAsB,GAAG,CAC7B,QAAgB,EAChB,aAAiC,EACjC,QAAQ,GAAG,KAAK,EACJ,EAAE,CAAC,CAAC;QAChB,IAAI,EAAE,WAAW;QACjB,UAAU,EAAE,gBAAgB;QAC5B,SAAS,EAAE,cAAc;QACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,IAAI,EAAE,aAAa;QACnB,GAAG,EAAE,OAAO;QACZ,OAAO,EAAE,OAAO;QAChB,OAAO,EAAE;YACP,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL;oBACE,gBAAgB,EAAE;wBAChB,IAAI,EAAE,QAAQ;wBACd,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;qBAC3B;iBACF;aACF;SACF;QACD,cAAc,EAAE;YACd,MAAM,EAAE,UAAU;YAClB,aAAa,EAAE,EAAE;YACjB,aAAa;YACb,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS;YACtD,SAAS,EAAE,SAAS;SACrB;KACF,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAE1B,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,OAAO,GAAG;gBACd,gBAAgB,CAAC,OAAO,CAAC;gBACzB,qBAAqB,CAAC,UAAU,CAAC;aAClC,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/B,MAAM,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CACvD,oBAAoB,CACrB,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CACvD,qBAAqB,CACtB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC;YAEpD,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/B,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC;gBACzC,aAAa,EAAE,oBAAoB;gBACnC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE;aACjD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,MAAM,GAAe;gBACzB,GAAG,gBAAgB,CAAC,MAAM,CAAC;gBAC3B,OAAO,EAAE,SAAS;aACnB,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAEhC,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,OAAO,GAAG,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,CAAC,CAAC;YAEjE,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/B,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC;gBACzC,aAAa,EAAE,qBAAqB;gBACpC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uBAAuB,EAAE;aACzD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,OAAO,GAAG,CAAC,qBAAqB,CAAC,wBAAwB,EAAE,IAAI,CAAC,CAAC,CAAC;YAExE,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/B,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC;gBACzC,aAAa,EAAE,qBAAqB;gBACpC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,wBAAwB,EAAE;aAC1D,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAe;gBACzB,GAAG,qBAAqB,CAAC,OAAO,CAAC;gBACjC,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL,EAAE,IAAI,EAAE,YAAY,EAAE;wBACtB,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,EAAE;wBACtC,EAAE,IAAI,EAAE,YAAY,EAAE;qBACvB;iBACF;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAEhC,MAAM,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC7C,aAAa,EAAE,qBAAqB;gBACpC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE;aAC9C,CAAC,CAAC;YACH,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC7C,aAAa,EAAE,qBAAqB;gBACpC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE;aAC/C,CAAC,CAAC;YACH,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC7C,aAAa,EAAE,qBAAqB;gBACpC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE;aAC9C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,MAAM,GAAe;gBACzB,GAAG,qBAAqB,CAAC,EAAE,CAAC;gBAC5B,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL;4BACE,YAAY,EAAE;gCACZ,IAAI,EAAE,WAAW;gCACjB,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;6BAC3B;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAEhC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE,aAAa;gBACrB,KAAK,EAAE,WAAW;gBAClB,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;aAC/B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,MAAM,GAAe;gBACzB,GAAG,qBAAqB,CAAC,EAAE,CAAC;gBAC5B,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL;4BACE,YAAY,EAAE;gCACZ,EAAE,EAAE,gBAAgB;gCACpB,IAAI,EAAE,WAAW;gCACjB,IAAI,EAAE,EAAE;6BACT;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAEhC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,UAAU,EAAE,gBAAgB;aAC7B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,OAAO,GAAG;gBACd,sBAAsB,CAAC,WAAW,EAAE,oBAAoB,CAAC;aAC1D,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/B,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC;gBACzC,aAAa,EAAE,kBAAkB;gBACjC,UAAU,EAAE,UAAU;gBACtB,MAAM,EAAE,WAAW;gBACnB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,SAAS;wBACf,6DAA6D;wBAC7D,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE;qBACnD;iBACF;gBACD,yCAAyC;gBACzC,SAAS,EAAE,oBAAoB;aAChC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,OAAO,GAAG,CAAC,sBAAsB,CAAC,cAAc,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;YAE1E,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/B,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,aAAa,EAAE,kBAAkB;gBACjC,MAAM,EAAE,QAAQ;aACjB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,WAAW,GAAsB;gBACrC,IAAI,EAAE,WAAW;gBACjB,KAAK,EAAE;oBACL,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE;oBACjD,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE;iBACpD;aACF,CAAC;YACF,MAAM,MAAM,GAAG,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACjE,sCAAsC;YACtC,MAAM,CAAC,OAAO,GAAG;gBACf,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL;wBACE,gBAAgB,EAAE;4BAChB,IAAI,EAAE,YAAY;4BAClB,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;yBAC3B;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAEhC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC;gBACzC,aAAa,EAAE,MAAM;gBACrB,OAAO,EAAE;oBACP,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE;oBAC5D,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE;iBAC/D;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,MAAM,GAAe;gBACzB,GAAG,sBAAsB,CAAC,WAAW,CAAC;gBACtC,IAAI,EAAE,eAAe;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,SAA8B;oBACtC,aAAa,EAAE,EAAE;oBACjB,aAAa,EAAE,QAAQ;oBACvB,KAAK,EAAE,SAAS;oBAChB,SAAS,EAAE,SAAS;iBACrB;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAEhC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,UAAU,EAAE,eAAe;aAC5B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,YAAY,GAAe;gBAC/B,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,IAAI;gBAChB,SAAS,EAAE,cAAc;gBACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,kBAAkB;gBAC3B,GAAG,EAAE,OAAO;gBACZ,OAAO,EAAE,OAAO;aACjB,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;YAEtC,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,OAAO,GAAiB;gBAC5B,gBAAgB,CAAC,uBAAuB,CAAC;gBACzC;oBACE,GAAG,qBAAqB,CAAC,EAAE,CAAC;oBAC5B,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE;4BACL,EAAE,IAAI,EAAE,8BAA8B,EAAE,OAAO,EAAE,IAAI,EAAE;4BACvD;gCACE,YAAY,EAAE;oCACZ,EAAE,EAAE,WAAW;oCACf,IAAI,EAAE,WAAW;oCACjB,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;iCAC1B;6BACF;yBACF;qBACF;iBACF;gBACD,sBAAsB,CAAC,WAAW,EAAE,qBAAqB,CAAC;gBAC1D,qBAAqB,CAAC,oCAAoC,CAAC;aAC5D,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/B,oCAAoC;YACpC,MAAM,WAAW,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAC9C,CAAC,IAAe,EAAE,EAAE,CACjB,IAAI,CAAC,CAAC,CAA+B,CAAC,aAAa,CACvD,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC;gBAC1B,oBAAoB;gBACpB,qBAAqB;gBACrB,WAAW;gBACX,kBAAkB;gBAClB,qBAAqB;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAe;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,UAAU,EAAE,WAAW;gBACvB,SAAS,EAAE,cAAc;gBACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,IAAI,EAAE,WAAW;gBACjB,GAAG,EAAE,OAAO;gBACZ,OAAO,EAAE,OAAO;gBAChB,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;iBAC5B;gBACD,aAAa,EAAE;oBACb,gBAAgB,EAAE,GAAG;oBACrB,oBAAoB,EAAE,EAAE;oBACxB,eAAe,EAAE,GAAG;iBACrB;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAEhC,MAAM,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,aAAa,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE;gBAC/C,aAAa,EAAE,qBAAqB;gBACpC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC1C,CAAC,CAAC;YACH,MAAM,CAAC,aAAa,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE;gBAC/C,aAAa,EAAE,qBAAqB;gBACpC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;gBACnC,KAAK,EAAE;oBACL,KAAK,EAAE;wBACL,YAAY,EAAE,GAAG;wBACjB,gBAAgB,EAAE,EAAE;wBACpB,cAAc,EAAE,SAAS;wBACzB,WAAW,EAAE,GAAG;wBAChB,YAAY,EAAE,SAAS;qBACxB;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/acp-integration/session/Session.d.ts

@@ -0,0 +1,61 @@
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config, GeminiChat, ChatRecord } from '@umsai/ums-code-core';
+import * as acp from '../acp.js';
+import type { LoadedSettings } from '../../config/settings.js';
+import type { SetModeRequest, SetModeResponse } from '../schema.js';
+import type { SessionContext } from './types.js';
+/**
+ * Session represents an active conversation session with the AI model.
+ * It uses modular components for consistent event emission:
+ * - HistoryReplayer for replaying past conversations
+ * - ToolCallEmitter for tool-related session updates
+ * - PlanEmitter for todo/plan updates
+ * - SubAgentTracker for tracking sub-agent tool calls
+ */
+export declare class Session implements SessionContext {
+    #private;
+    private readonly chat;
+    readonly config: Config;
+    private readonly client;
+    private readonly settings;
+    private pendingPrompt;
+    private turn;
+    private readonly historyReplayer;
+    private readonly toolCallEmitter;
+    private readonly planEmitter;
+    private readonly messageEmitter;
+    readonly sessionId: string;
+    constructor(id: string, chat: GeminiChat, config: Config, client: acp.Client, settings: LoadedSettings);
+    getId(): string;
+    getConfig(): Config;
+    /**
+     * Replays conversation history to the client using modular components.
+     * Delegates to HistoryReplayer for consistent event emission.
+     */
+    replayHistory(records: ChatRecord[]): Promise<void>;
+    cancelPendingPrompt(): Promise<void>;
+    prompt(params: acp.PromptRequest): Promise<acp.PromptResponse>;
+    sendUpdate(update: acp.SessionUpdate): Promise<void>;
+    sendAvailableCommandsUpdate(): Promise<void>;
+    /**
+     * Requests permission from the client for a tool call.
+     * Used by SubAgentTracker for sub-agent approval requests.
+     */
+    requestPermission(params: acp.RequestPermissionRequest): Promise<acp.RequestPermissionResponse>;
+    /**
+     * Sets the approval mode for the current session.
+     * Maps ACP approval mode values to core ApprovalMode enum.
+     */
+    setMode(params: SetModeRequest): Promise<SetModeResponse>;
+    /**
+     * Sends a current_mode_update notification to the client.
+     * Called after the agent switches modes (e.g., from exit_plan_mode tool).
+     */
+    private sendCurrentModeUpdateNotification;
+    private runTool;
+    debug(msg: string): void;
+}