genaicode 0.6.1 → 0.8.0

package/dist/prompt/steps/step-context-optimization.test.js

@@ -0,0 +1,475 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { executeStepContextOptimization } from './step-context-optimization';
+import { getSourceCode } from '../../files/read-files';
+import { StepResult } from './steps-types';
+import { putSystemMessage } from '../../main/common/content-bus';
+import '../../main/config.js';
+import '../../files/find-files.js';
+// Mock dependencies
+vi.mock('../../files/read-files');
+vi.mock('../../files/find-files.js', () => ({
+    getSourceFiles: () => [],
+    getImageAssetFiles: () => [],
+    refreshFiles: () => null,
+}));
+vi.mock('../../main/common/content-bus');
+vi.mock('../token-estimator', () => ({
+    estimateTokenCount: vi.fn((content) => {
+        // Simple mock implementation that returns length of content as token count
+        return content ? content.length : 0;
+    }),
+}));
+vi.mock('./step-summarization', () => ({
+    getSummary: vi.fn().mockReturnValue({ summary: 'Test summary', dependencies: [{ path: 'test-dep', type: 'local' }] }),
+}));
+vi.mock('../../main/config.js', () => ({
+    rootDir: '/test',
+    rcConfig: {
+        rootDir: '/test',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+    importantContext: {},
+    sourceExtensions: ['.ts'],
+}));
+describe('executeStepContextOptimization', () => {
+    const mockGenerateContentFn = vi.fn();
+    const mockOptions = {
+        aiService: 'vertex-ai',
+        askQuestion: false,
+    };
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockGenerateContentFn.mockReset();
+    });
+    describe('Source code tree transformation', () => {
+        it('should correctly transform source code between flat and tree structures', async () => {
+            // Mock source code data
+            const mockSourceCode = {
+                '/test/file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+                '/test/file2.ts': { content: 'content2' },
+            };
+            const mockSourceCodeTree = {
+                '/test': {
+                    'file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+                    'file2.ts': { content: 'content2' },
+                },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            // Mock optimization response with high relevance scores
+            mockGenerateContentFn.mockResolvedValueOnce([
+                {
+                    name: 'optimizeContext',
+                    args: {
+                        userPrompt: 'test prompt',
+                        optimizedContext: [
+                            { filePath: '/test/file1.ts', relevance: 0.9 },
+                            { filePath: '/test/file2.ts', relevance: 0.8 },
+                        ],
+                    },
+                },
+            ]);
+            const result = await executeStepContextOptimization(mockGenerateContentFn, [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify(mockSourceCodeTree),
+                        },
+                    ],
+                },
+            ], mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            expect(putSystemMessage).toHaveBeenCalledWith('Context optimization in progress.', expect.arrayContaining([
+                ['/test/file1.ts', 0.9],
+                ['/test/file2.ts', 0.8],
+            ]));
+        });
+        it('should handle empty source code tree', async () => {
+            const emptySourceCode = {};
+            const emptySourceCodeTree = {};
+            vi.mocked(getSourceCode).mockReturnValue(emptySourceCode);
+            mockGenerateContentFn.mockResolvedValueOnce([
+                {
+                    name: 'optimizeContext',
+                    args: {
+                        userPrompt: 'test prompt',
+                        optimizedContext: [],
+                    },
+                },
+            ]);
+            const result = await executeStepContextOptimization(mockGenerateContentFn, [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify(emptySourceCodeTree),
+                        },
+                    ],
+                },
+            ], mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            expect(putSystemMessage).toHaveBeenCalledWith('Context optimization is not needed, because the code base is small.');
+        });
+    });
+    describe('Multiple getSourceCode responses handling', () => {
+        // This test verifies that when clearing previous getSourceCode responses:
+        // 1. Directory structure is preserved
+        // 2. File content is set to null
+        // 3. The overall structure of the conversation remains intact
+        it('should clear previous getSourceCode responses while preserving structure', async () => {
+            const mockSourceCode = {
+                '/test/file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+            };
+            const mockSourceCodeTree = {
+                '/test': {
+                    'file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+                },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            mockGenerateContentFn.mockResolvedValueOnce([
+                {
+                    name: 'optimizeContext',
+                    args: {
+                        userPrompt: 'test prompt',
+                        optimizedContext: [{ filePath: '/test/file1.ts', relevance: 0.9 }],
+                    },
+                },
+            ]);
+            const prompt = [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify({ '/test': { 'old-file.ts': { content: 'old content' } } }),
+                        },
+                    ],
+                },
+                {
+                    type: 'assistant',
+                    text: 'Some response',
+                },
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify(mockSourceCodeTree),
+                        },
+                    ],
+                },
+            ];
+            const result = await executeStepContextOptimization(mockGenerateContentFn, prompt, mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            // Verify that previous responses maintain structure but clear content
+            const expectedStructure = {
+                '/test': {
+                    'old-file.ts': { content: null },
+                },
+            };
+            expect(JSON.parse(prompt[0].functionResponses[0].content)).toEqual(expectedStructure);
+            // Verify that structure was preserved
+            expect(prompt[0].type).toBe('user');
+            expect(prompt[0].functionResponses[0].name).toBe('getSourceCode');
+            // Verify that non-getSourceCode items were preserved
+            expect(prompt[1].type).toBe('assistant');
+            expect(prompt[1].text).toBe('Some response');
+        });
+        // This test verifies that when handling multiple responses:
+        // 1. Each response maintains its directory structure
+        // 2. Content is cleared (set to null) for all previous responses
+        // 3. The latest response remains unchanged
+        it('should handle multiple responses with different content', async () => {
+            const mockSourceCode = {
+                '/test/file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+                '/test/file2.ts': { content: 'content2' },
+            };
+            const mockSourceCodeTree = {
+                '/test': {
+                    'file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+                    'file2.ts': { content: 'content2' },
+                },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            mockGenerateContentFn.mockResolvedValueOnce([
+                {
+                    name: 'optimizeContext',
+                    args: {
+                        userPrompt: 'test prompt',
+                        optimizedContext: [
+                            { filePath: '/test/file1.ts', relevance: 0.9 },
+                            { filePath: '/test/file2.ts', relevance: 0.8 },
+                        ],
+                    },
+                },
+            ]);
+            const prompt = [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify({ '/test': { 'file1.ts': { content: 'old content 1' } } }),
+                        },
+                    ],
+                },
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify({ '/test': { 'file2.ts': { content: 'old content 2' } } }),
+                        },
+                    ],
+                },
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify(mockSourceCodeTree),
+                        },
+                    ],
+                },
+            ];
+            const result = await executeStepContextOptimization(mockGenerateContentFn, prompt, mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            // Verify that previous responses maintain structure but clear content
+            prompt.slice(0, -1).forEach((item) => {
+                if (item.type === 'user' && item.functionResponses) {
+                    const response = item.functionResponses.find((r) => r.name === 'getSourceCode');
+                    const parsedContent = JSON.parse(response.content);
+                    // Check that structure is preserved but content is null
+                    expect(parsedContent).toHaveProperty('/test');
+                    Object.values(parsedContent['/test']).forEach((file) => {
+                        expect(file.content).toBe(null);
+                    });
+                }
+            });
+        });
+        it('should handle edge case with no getSourceCode responses', async () => {
+            const mockSourceCode = {
+                '/test/file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            const prompt = [
+                {
+                    type: 'user',
+                    text: 'Some user message',
+                },
+                {
+                    type: 'assistant',
+                    text: 'Some assistant response',
+                },
+            ];
+            const result = await executeStepContextOptimization(mockGenerateContentFn, prompt, mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            // Verify that the prompt structure remains unchanged
+            expect(prompt).toHaveLength(2);
+            expect(prompt[0].type).toBe('user');
+            expect(prompt[0].text).toBe('Some user message');
+            expect(prompt[1].type).toBe('assistant');
+            expect(prompt[1].text).toBe('Some assistant response');
+        });
+        it('should handle edge case with empty functionResponses', async () => {
+            const mockSourceCode = {
+                '/test/file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            const prompt = [
+                {
+                    type: 'user',
+                    functionResponses: [],
+                },
+                {
+                    type: 'user',
+                    functionResponses: undefined,
+                },
+            ];
+            const result = await executeStepContextOptimization(mockGenerateContentFn, prompt, mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            // Verify that the prompt structure remains unchanged
+            expect(prompt).toHaveLength(2);
+            expect(prompt[0].type).toBe('user');
+            expect(prompt[0].functionResponses).toEqual([]);
+            expect(prompt[1].type).toBe('user');
+            expect(prompt[1].functionResponses).toBeUndefined();
+        });
+    });
+    describe('Context optimization process', () => {
+        it('should optimize context based on relevance scores', async () => {
+            const mockSourceCode = {
+                '/test/high-relevance.ts': { content: 'important content' + Array.from(Array(10000).keys()).join(',') },
+                '/test/medium-relevance.ts': { content: 'somewhat important' },
+                '/test/low-relevance.ts': { content: 'not important' },
+            };
+            const mockSourceCodeTree = {
+                '/test': {
+                    'high-relevance.ts': { content: 'important content' },
+                    'medium-relevance.ts': { content: 'somewhat important' },
+                    'low-relevance.ts': { content: 'not important' },
+                },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            mockGenerateContentFn.mockResolvedValueOnce([
+                {
+                    name: 'optimizeContext',
+                    args: {
+                        userPrompt: 'test prompt',
+                        optimizedContext: [
+                            { filePath: '/test/high-relevance.ts', relevance: 0.9 },
+                            { filePath: '/test/medium-relevance.ts', relevance: 0.6 },
+                            { filePath: '/test/low-relevance.ts', relevance: 0.2 },
+                        ],
+                    },
+                },
+            ]);
+            const result = await executeStepContextOptimization(mockGenerateContentFn, [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify(mockSourceCodeTree),
+                        },
+                    ],
+                },
+            ], mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            expect(putSystemMessage).toHaveBeenCalledWith('Context optimization in progress.', expect.arrayContaining([
+                ['/test/high-relevance.ts', 0.9],
+                ['/test/medium-relevance.ts', 0.6],
+            ]));
+            // Low relevance file should not be included
+            expect(putSystemMessage).not.toHaveBeenCalledWith('Context optimization in progress.', expect.arrayContaining(['/test/low-relevance.ts', 0.2]));
+        });
+        it('should handle optimization failure gracefully', async () => {
+            const mockSourceCode = {
+                '/test/file.ts': { content: 'test content'.repeat(1000) }, // Make it large enough to trigger optimization
+            };
+            const mockSourceCodeTree = {
+                '/test': {
+                    'file.ts': { content: 'test content'.repeat(1000) },
+                },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            mockGenerateContentFn.mockResolvedValueOnce([
+                {
+                    name: 'optimizeContext',
+                    args: {
+                        userPrompt: 'test prompt',
+                        optimizedContext: [], // Empty context indicates optimization failure
+                    },
+                },
+            ]);
+            const result = await executeStepContextOptimization(mockGenerateContentFn, [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify(mockSourceCodeTree),
+                        },
+                    ],
+                },
+            ], mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            expect(putSystemMessage).toHaveBeenCalledWith('Warning: Context optimization failed to produce useful summaries for all batches. Proceeding with current context.');
+        });
+    });
+    describe('Edge cases and error handling', () => {
+        beforeEach(() => {
+            const mockSourceCode = {
+                '/test/file1.ts': { content: 'content1' + Array.from(Array(10000).keys()).join(',') },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+        });
+        it('should handle missing source code response', async () => {
+            const result = await executeStepContextOptimization(mockGenerateContentFn, [], // Empty prompt array
+            mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            expect(putSystemMessage).not.toHaveBeenCalledWith('Context optimization in progress.');
+        });
+        it('should handle invalid JSON in source code response', async () => {
+            await expect(executeStepContextOptimization(mockGenerateContentFn, [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: 'invalid json',
+                        },
+                    ],
+                },
+            ], mockOptions)).rejects.toThrow('Unexpected token');
+        });
+        it('should handle errors during optimization process', async () => {
+            vi.mocked(getSourceCode).mockImplementation(() => {
+                throw new Error('Unexpected error');
+            });
+            await expect(executeStepContextOptimization(mockGenerateContentFn, [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: '{}',
+                        },
+                    ],
+                },
+            ], mockOptions)).rejects.toThrow('Unexpected error');
+        });
+    });
+    describe('Token count handling', () => {
+        it('should respect MAX_TOTAL_TOKENS limit', async () => {
+            // Create source code that exceeds token limit
+            const largeContent = 'a'.repeat(12000); // Exceeds MAX_TOTAL_TOKENS (10000)
+            const smallContent = 'b'.repeat(3000);
+            const mockSourceCode = {
+                '/test/large-file.ts': { content: largeContent },
+                '/test/small-file.ts': { content: smallContent },
+            };
+            const mockSourceCodeTree = {
+                '/test': {
+                    'large-file.ts': { content: largeContent },
+                    'small-file.ts': { content: smallContent },
+                },
+            };
+            vi.mocked(getSourceCode).mockReturnValue(mockSourceCode);
+            mockGenerateContentFn.mockResolvedValueOnce([
+                {
+                    name: 'optimizeContext',
+                    args: {
+                        userPrompt: 'test prompt',
+                        optimizedContext: [
+                            { filePath: '/test/large-file.ts', relevance: 0.9 },
+                            { filePath: '/test/small-file.ts', relevance: 0.8 },
+                        ],
+                    },
+                },
+            ]);
+            const result = await executeStepContextOptimization(mockGenerateContentFn, [
+                {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'getSourceCode',
+                            content: JSON.stringify(mockSourceCodeTree),
+                        },
+                    ],
+                },
+            ], mockOptions);
+            expect(result).toBe(StepResult.CONTINUE);
+            expect(putSystemMessage).toHaveBeenCalledWith('Context optimization completed successfully.', expect.objectContaining({
+                tokensBefore: expect.any(Number),
+                tokensAfter: expect.any(Number),
+                contentTokenCount: expect.any(Number),
+                summaryTokenCount: expect.any(Number),
+                percentageReduced: expect.stringContaining('%'),
+            }));
+        });
+    });
+});
+//# sourceMappingURL=step-context-optimization.test.js.map

package/dist/prompt/steps/step-context-optimization.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-context-optimization.test.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-context-optimization.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,8BAA8B,EAAE,MAAM,6BAA6B,CAAC;AAC7E,OAAO,EAAe,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAEpE,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAE3C,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,sBAAsB,CAAC;AAC9B,OAAO,2BAA2B,CAAC;AAGnC,oBAAoB;AACpB,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAClC,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;IAC5B,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;CACzB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;AACzC,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,OAAe,EAAE,EAAE;QAC5C,2EAA2E;QAC3E,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC;CACH,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;CACtH,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,OAAO,EAAE,OAAO;IAChB,QAAQ,EAAE;QACR,OAAO,EAAE,OAAO;QAChB,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;IACD,gBAAgB,EAAE,EAAE;IACpB,gBAAgB,EAAE,CAAC,KAAK,CAAC;CAC1B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,MAAM,qBAAqB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACtC,MAAM,WAAW,GAAmB;QAClC,SAAS,EAAE,WAAW;QACtB,WAAW,EAAE,KAAK;KACnB,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,qBAAqB,CAAC,SAAS,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACvF,wBAAwB;YACxB,MAAM,cAAc,GAAG;gBACrB,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;gBACrF,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;aAC1C,CAAC;YAEF,MAAM,kBAAkB,GAAmB;gBACzC,OAAO,EAAE;oBACP,UAAU,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;oBAC/E,UAAU,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;iBACpC;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,wDAAwD;YACxD,qBAAqB,CAAC,qBAAqB,CAAC;gBAC1C;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE;wBACJ,UAAU,EAAE,aAAa;wBACzB,gBAAgB,EAAE;4BAChB,EAAE,QAAQ,EAAE,gBAAgB,EAAE,SAAS,EAAE,GAAG,EAAE;4BAC9C,EAAE,QAAQ,EAAE,gBAAgB,EAAE,SAAS,EAAE,GAAG,EAAE;yBAC/C;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,8BAA8B,CACjD,qBAAqB,EACrB;gBACE;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;yBAC5C;qBACF;iBACF;aACF,EACD,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,mCAAmC,EACnC,MAAM,CAAC,eAAe,CAAC;gBACrB,CAAC,gBAAgB,EAAE,GAAG,CAAC;gBACvB,CAAC,gBAAgB,EAAE,GAAG,CAAC;aACxB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,eAAe,GAAG,EAAE,CAAC;YAC3B,MAAM,mBAAmB,GAAG,EAAE,CAAC;YAE/B,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAE1D,qBAAqB,CAAC,qBAAqB,CAAC;gBAC1C;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE;wBACJ,UAAU,EAAE,aAAa;wBACzB,gBAAgB,EAAE,EAAE;qBACrB;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,8BAA8B,CACjD,qBAAqB,EACrB;gBACE;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;yBAC7C;qBACF;iBACF;aACF,EACD,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,qEAAqE,CACtE,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACzD,0EAA0E;QAC1E,sCAAsC;QACtC,iCAAiC;QACjC,8DAA8D;QAC9D,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,cAAc,GAAG;gBACrB,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;aACtF,CAAC;YAEF,MAAM,kBAAkB,GAAmB;gBACzC,OAAO,EAAE;oBACP,UAAU,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;iBAChF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,qBAAqB,CAAC,qBAAqB,CAAC;gBAC1C;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE;wBACJ,UAAU,EAAE,aAAa;wBACzB,gBAAgB,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC;qBACnE;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAiB;gBAC3B;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC;yBACpF;qBACF;iBACF;gBACD;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,eAAe;iBACtB;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;yBAC5C;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,8BAA8B,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAEhG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAEzC,sEAAsE;YACtE,MAAM,iBAAiB,GAAG;gBACxB,OAAO,EAAE;oBACP,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;iBACjC;aACF,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAkB,CAAC,CAAC,CAAC,CAAC,OAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YAExF,sCAAsC;YACtC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAkB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACnE,qDAAqD;YACrD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,4DAA4D;QAC5D,qDAAqD;QACrD,iEAAiE;QACjE,2CAA2C;QAC3C,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,cAAc,GAAG;gBACrB,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;gBACrF,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;aAC1C,CAAC;YAEF,MAAM,kBAAkB,GAAmB;gBACzC,OAAO,EAAE;oBACP,UAAU,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;oBAC/E,UAAU,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;iBACpC;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,qBAAqB,CAAC,qBAAqB,CAAC;gBAC1C;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE;wBACJ,UAAU,EAAE,aAAa;wBACzB,gBAAgB,EAAE;4BAChB,EAAE,QAAQ,EAAE,gBAAgB,EAAE,SAAS,EAAE,GAAG,EAAE;4BAC9C,EAAE,QAAQ,EAAE,gBAAgB,EAAE,SAAS,EAAE,GAAG,EAAE;yBAC/C;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAiB;gBAC3B;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;yBACnF;qBACF;iBACF;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;yBACnF;qBACF;iBACF;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;yBAC5C;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,8BAA8B,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAEhG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,sEAAsE;YACtE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACnC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;oBACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;oBAChF,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,QAAS,CAAC,OAAQ,CAAC,CAAC;oBACrD,wDAAwD;oBACxD,MAAM,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;oBAC9C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;wBACrD,MAAM,CAAE,IAAoB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACnD,CAAC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,cAAc,GAAG;gBACrB,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;aACtF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAiB;gBAC3B;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,mBAAmB;iBAC1B;gBACD;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,yBAAyB;iBAChC;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,8BAA8B,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAEhG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,qDAAqD;YACrD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,cAAc,GAAG;gBACrB,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;aACtF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAiB;gBAC3B;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE,EAAE;iBACtB;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE,SAAS;iBAC7B;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,8BAA8B,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAEhG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,qDAAqD;YACrD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,aAAa,EAAE,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,cAAc,GAAG;gBACrB,yBAAyB,EAAE,EAAE,OAAO,EAAE,mBAAmB,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;gBACvG,2BAA2B,EAAE,EAAE,OAAO,EAAE,oBAAoB,EAAE;gBAC9D,wBAAwB,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;aACvD,CAAC;YAEF,MAAM,kBAAkB,GAAmB;gBACzC,OAAO,EAAE;oBACP,mBAAmB,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE;oBACrD,qBAAqB,EAAE,EAAE,OAAO,EAAE,oBAAoB,EAAE;oBACxD,kBAAkB,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;iBACjD;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,qBAAqB,CAAC,qBAAqB,CAAC;gBAC1C;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE;wBACJ,UAAU,EAAE,aAAa;wBACzB,gBAAgB,EAAE;4BAChB,EAAE,QAAQ,EAAE,yBAAyB,EAAE,SAAS,EAAE,GAAG,EAAE;4BACvD,EAAE,QAAQ,EAAE,2BAA2B,EAAE,SAAS,EAAE,GAAG,EAAE;4BACzD,EAAE,QAAQ,EAAE,wBAAwB,EAAE,SAAS,EAAE,GAAG,EAAE;yBACvD;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,8BAA8B,CACjD,qBAAqB,EACrB;gBACE;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;yBAC5C;qBACF;iBACF;aACF,EACD,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,mCAAmC,EACnC,MAAM,CAAC,eAAe,CAAC;gBACrB,CAAC,yBAAyB,EAAE,GAAG,CAAC;gBAChC,CAAC,2BAA2B,EAAE,GAAG,CAAC;aACnC,CAAC,CACH,CAAC;YACF,4CAA4C;YAC5C,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAC/C,mCAAmC,EACnC,MAAM,CAAC,eAAe,CAAC,CAAC,wBAAwB,EAAE,GAAG,CAAC,CAAC,CACxD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,cAAc,GAAG;gBACrB,eAAe,EAAE,EAAE,OAAO,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,+CAA+C;aAC3G,CAAC;YAEF,MAAM,kBAAkB,GAAmB;gBACzC,OAAO,EAAE;oBACP,SAAS,EAAE,EAAE,OAAO,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;iBACpD;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,qBAAqB,CAAC,qBAAqB,CAAC;gBAC1C;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE;wBACJ,UAAU,EAAE,aAAa;wBACzB,gBAAgB,EAAE,EAAE,EAAE,+CAA+C;qBACtE;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,8BAA8B,CACjD,qBAAqB,EACrB;gBACE;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;yBAC5C;qBACF;iBACF;aACF,EACD,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,oHAAoH,CACrH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,cAAc,GAAG;gBACrB,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;aACtF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,MAAM,GAAG,MAAM,8BAA8B,CACjD,qBAAqB,EACrB,EAAE,EAAE,qBAAqB;YACzB,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAAC,mCAAmC,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,MAAM,CACV,8BAA8B,CAC5B,qBAAqB,EACrB;gBACE;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,cAAc;yBACxB;qBACF;iBACF;aACF,EACD,WAAW,CACZ,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC/C,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,CACV,8BAA8B,CAC5B,qBAAqB,EACrB;gBACE;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI;yBACd;qBACF;iBACF;aACF,EACD,WAAW,CACZ,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,8CAA8C;YAC9C,MAAM,YAAY,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,mCAAmC;YAC3E,MAAM,YAAY,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAEtC,MAAM,cAAc,GAAG;gBACrB,qBAAqB,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;gBAChD,qBAAqB,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;aACjD,CAAC;YAEF,MAAM,kBAAkB,GAAmB;gBACzC,OAAO,EAAE;oBACP,eAAe,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;oBAC1C,eAAe,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;iBAC3C;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAEzD,qBAAqB,CAAC,qBAAqB,CAAC;gBAC1C;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE;wBACJ,UAAU,EAAE,aAAa;wBACzB,gBAAgB,EAAE;4BAChB,EAAE,QAAQ,EAAE,qBAAqB,EAAE,SAAS,EAAE,GAAG,EAAE;4BACnD,EAAE,QAAQ,EAAE,qBAAqB,EAAE,SAAS,EAAE,GAAG,EAAE;yBACpD;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,8BAA8B,CACjD,qBAAqB,EACrB;gBACE;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;yBAC5C;qBACF;iBACF;aACF,EACD,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,8CAA8C,EAC9C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAChC,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC/B,iBAAiB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBACrC,iBAAiB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBACrC,iBAAiB,EAAE,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC;aAChD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/prompt/steps/step-ensure-context.d.ts

@@ -0,0 +1,15 @@
+import { FunctionCall } from '../../ai-service/common.js';
+import { CodegenOptions } from '../../main/codegen-types.js';
+import { StepResult } from './steps-types.js';
+/**
+ * Executes the context completeness check step.
+ * Ensures all necessary files are included in the conversation context.
+ */
+export declare function executeStepEnsureContext(prompt: {
+    type: string;
+    functionResponses?: {
+        name: string;
+        content?: string;
+    }[];
+    functionCalls?: FunctionCall[];
+}[], codegenSummaryCall: FunctionCall, options: CodegenOptions): Promise<StepResult>;

package/dist/prompt/steps/step-ensure-context.js

@@ -0,0 +1,102 @@
+import { importantContext } from '../../main/config.js';
+import { getSourceCode } from '../../files/read-files.js';
+import { putSystemMessage } from '../../main/common/content-bus.js';
+import { StepResult } from './steps-types.js';
+import { getSourceCodeTree } from '../../files/source-code-tree.js';
+/**
+ * Executes the context completeness check step.
+ * Ensures all necessary files are included in the conversation context.
+ */
+export async function executeStepEnsureContext(prompt, codegenSummaryCall, options) {
+    try {
+        // Extract all required paths
+        const requiredPaths = extractRequiredPaths(codegenSummaryCall);
+        if (requiredPaths.length === 0) {
+            putSystemMessage('No paths to ensure in context.');
+            return StepResult.CONTINUE;
+        }
+        // Get paths that are already in context
+        const existingPaths = getExistingContextPaths(prompt);
+        // Find paths that are missing from context
+        const missingPaths = requiredPaths.filter((path) => !existingPaths.has(path));
+        if (missingPaths.length === 0) {
+            putSystemMessage('All required paths are already in context.');
+            return StepResult.CONTINUE;
+        }
+        // Get source code for missing paths
+        putSystemMessage(`Ensuring context completeness. Adding ${missingPaths.length} missing files.`, { missingPaths });
+        const sourceCode = getSourceCode({ filterPaths: missingPaths }, options);
+        // Append getSourceCode function call/response to the prompt
+        prompt.push({
+            type: 'assistant',
+            functionCalls: [
+                {
+                    name: 'getSourceCode',
+                    args: {
+                        filePaths: missingPaths,
+                    },
+                },
+            ],
+        }, {
+            type: 'user',
+            functionResponses: [
+                {
+                    name: 'getSourceCode',
+                    content: JSON.stringify(getSourceCodeTree(sourceCode)),
+                },
+            ],
+        });
+        putSystemMessage('Context completeness check completed successfully.');
+        return StepResult.CONTINUE;
+    }
+    catch (error) {
+        putSystemMessage('Error: Context completeness check failed. This is unexpected.');
+        console.error('Context completeness check error:', error);
+        return StepResult.BREAK;
+    }
+}
+/**
+ * Extract all required file paths from codegenSummary function call
+ */
+function extractRequiredPaths(codegenSummaryCall) {
+    if (!codegenSummaryCall.args?.fileUpdates || !codegenSummaryCall.args?.contextPaths) {
+        throw new Error('Invalid codegenSummary call: missing fileUpdates or contextPaths');
+    }
+    // Ensure fileUpdates is an array and has the correct type
+    const fileUpdatePaths = Array.isArray(codegenSummaryCall.args.fileUpdates)
+        ? codegenSummaryCall.args.fileUpdates.map((update) => update.filePath)
+        : [];
+    const contextPaths = codegenSummaryCall.args.contextPaths;
+    const importantPaths = importantContext.files ?? [];
+    // Use Set to deduplicate paths
+    return Array.from(new Set([...fileUpdatePaths, ...contextPaths, ...importantPaths]));
+}
+/**
+ * Get paths that are already present in the conversation context
+ */
+function getExistingContextPaths(prompt) {
+    const existingPaths = new Set();
+    for (const item of prompt) {
+        if (!item.functionResponses)
+            continue;
+        const sourceCodeResponse = item.functionResponses.find((response) => response.name === 'getSourceCode');
+        if (!sourceCodeResponse?.content)
+            continue;
+        try {
+            const sourceCode = JSON.parse(sourceCodeResponse.content);
+            for (const [dirPath, files] of Object.entries(sourceCode)) {
+                for (const [fileName, fileInfo] of Object.entries(files)) {
+                    const fullPath = `${dirPath}/${fileName}`;
+                    if (fileInfo && typeof fileInfo === 'object' && 'content' in fileInfo && fileInfo.content !== null) {
+                        existingPaths.add(fullPath);
+                    }
+                }
+            }
+        }
+        catch (error) {
+            console.warn('Failed to parse getSourceCode response:', error);
+        }
+    }
+    return existingPaths;
+}
+//# sourceMappingURL=step-ensure-context.js.map

package/dist/prompt/steps/step-ensure-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-ensure-context.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-ensure-context.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AAEpE;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,MAAoH,EACpH,kBAAgC,EAChC,OAAuB;IAEvB,IAAI,CAAC;QACH,6BAA6B;QAC7B,MAAM,aAAa,GAAG,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;QAC/D,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,gBAAgB,CAAC,gCAAgC,CAAC,CAAC;YACnD,OAAO,UAAU,CAAC,QAAQ,CAAC;QAC7B,CAAC;QAED,wCAAwC;QACxC,MAAM,aAAa,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;QAEtD,2CAA2C;QAC3C,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9E,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,gBAAgB,CAAC,4CAA4C,CAAC,CAAC;YAC/D,OAAO,UAAU,CAAC,QAAQ,CAAC;QAC7B,CAAC;QAED,oCAAoC;QACpC,gBAAgB,CAAC,yCAAyC,YAAY,CAAC,MAAM,iBAAiB,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;QAClH,MAAM,UAAU,GAAG,aAAa,CAAC,EAAE,WAAW,EAAE,YAAY,EAAE,EAAE,OAAO,CAAC,CAAC;QAEzE,4DAA4D;QAC5D,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,aAAa,EAAE;gBACb;oBACE,IAAI,EAAE,eAAe;oBACrB,IAAI,EAAE;wBACJ,SAAS,EAAE,YAAY;qBACxB;iBACF;aACF;SACF,EACD;YACE,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,eAAe;oBACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;iBACvD;aACF;SACF,CACF,CAAC;QAEF,gBAAgB,CAAC,oDAAoD,CAAC,CAAC;QACvE,OAAO,UAAU,CAAC,QAAQ,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,gBAAgB,CAAC,+DAA+D,CAAC,CAAC;QAClF,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;QAC1D,OAAO,UAAU,CAAC,KAAK,CAAC;IAC1B,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAAC,kBAAgC;IAC5D,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,WAAW,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,YAAY,EAAE,CAAC;QACpF,MAAM,IAAI,KAAK,CAAC,kEAAkE,CAAC,CAAC;IACtF,CAAC;IAED,0DAA0D;IAC1D,MAAM,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,WAAW,CAAC;QACxE,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAA4B,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;QAC5F,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,YAAY,GAAG,kBAAkB,CAAC,IAAI,CAAC,YAAwB,CAAC;IACtE,MAAM,cAAc,GAAG,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC;IAEpD,+BAA+B;IAC/B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,eAAe,EAAE,GAAG,YAAY,EAAE,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;AACvF,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAAC,MAAsE;IACrG,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IAExC,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QAC1B,IAAI,CAAC,IAAI,CAAC,iBAAiB;YAAE,SAAS;QAEtC,MAAM,kBAAkB,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;QACxG,IAAI,CAAC,kBAAkB,EAAE,OAAO;YAAE,SAAS;QAE3C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC1D,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1D,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;oBACpF,MAAM,QAAQ,GAAG,GAAG,OAAO,IAAI,QAAQ,EAAE,CAAC;oBAC1C,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,SAAS,IAAI,QAAQ,IAAI,QAAQ,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;wBACnG,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBAC9B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,yCAAyC,EAAE,KAAK,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED,OAAO,aAAa,CAAC;AACvB,CAAC"}

package/dist/prompt/steps/step-ensure-context.test.d.ts

@@ -0,0 +1 @@
+import '../../files/find-files.js';