genaicode 0.8.6 → 0.9.0

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

@@ -1,326 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { executeStepEnsureContext } from './step-ensure-context';
-import { getSourceCode } from '../../files/read-files';
-import { putSystemMessage } from '../../main/common/content-bus';
-import { StepResult } from './steps-types';
-import '../../files/find-files.js';
-import { importantContext } from '../../main/config.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('../../main/config.js', () => ({
-    rootDir: '/test',
-    rcConfig: {
-        rootDir: '/test',
-        extensions: ['.js', '.ts', '.tsx', '.jsx'],
-    },
-    importantContext: {
-        files: ['/test/important.ts'],
-    },
-}));
-describe('executeStepEnsureContext', () => {
-    const mockOptions = {
-        aiService: 'vertex-ai',
-        askQuestion: false,
-    };
-    beforeEach(() => {
-        vi.clearAllMocks();
-        importantContext.files = ['/test/important.ts'];
-    });
-    describe('Path extraction from codegenSummary', () => {
-        it('should extract paths from fileUpdates and contextPaths', async () => {
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: {
-                    fileUpdates: [{ filePath: '/test/file1.ts' }, { filePath: '/test/file2.ts' }],
-                    contextPaths: ['/test/context1.ts', '/test/context2.ts'],
-                    explanation: 'test',
-                },
-            };
-            vi.mocked(getSourceCode).mockReturnValue({
-                '/test/file1.ts': { content: 'content1' },
-                '/test/file2.ts': { content: 'content2' },
-                '/test/context1.ts': { content: 'context1' },
-                '/test/context2.ts': { content: 'context2' },
-                '/test/important.ts': { content: 'important' },
-            });
-            const prompt = [];
-            const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(getSourceCode).toHaveBeenCalledWith(expect.objectContaining({
-                filterPaths: expect.arrayContaining([
-                    '/test/file1.ts',
-                    '/test/file2.ts',
-                    '/test/context1.ts',
-                    '/test/context2.ts',
-                    '/test/important.ts',
-                ]),
-            }), mockOptions);
-        });
-        it('should handle empty fileUpdates and contextPaths', async () => {
-            importantContext.files = [];
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: {
-                    fileUpdates: [],
-                    contextPaths: [],
-                    explanation: 'test',
-                },
-            };
-            vi.mocked(getSourceCode).mockReturnValue({});
-            const prompt = [];
-            const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');
-            expect(getSourceCode).not.toHaveBeenCalled();
-        });
-        it('should handle missing args in codegenSummary', async () => {
-            importantContext.files = [];
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: undefined,
-            };
-            const prompt = [];
-            const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');
-            expect(getSourceCode).not.toHaveBeenCalled();
-        });
-    });
-    describe('Path extraction from codegenPlanning', () => {
-        it('should extract paths from affectedFiles including dependencies', async () => {
-            const mockCodegenPlanning = {
-                name: 'codegenPlanning',
-                args: {
-                    affectedFiles: [
-                        {
-                            filePath: '/test/main.ts',
-                            dependencies: ['/test/util1.ts', '/test/util2.ts'],
-                        },
-                        {
-                            filePath: '/test/component.ts',
-                            dependencies: ['/test/shared.ts'],
-                        },
-                    ],
-                },
-            };
-            vi.mocked(getSourceCode).mockReturnValue({
-                '/test/main.ts': { content: 'main content' },
-                '/test/util1.ts': { content: 'util1 content' },
-                '/test/util2.ts': { content: 'util2 content' },
-                '/test/component.ts': { content: 'component content' },
-                '/test/shared.ts': { content: 'shared content' },
-                '/test/important.ts': { content: 'important content' },
-            });
-            const prompt = [];
-            const result = await executeStepEnsureContext(prompt, mockCodegenPlanning, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(getSourceCode).toHaveBeenCalledWith(expect.objectContaining({
-                filterPaths: expect.arrayContaining([
-                    '/test/main.ts',
-                    '/test/util1.ts',
-                    '/test/util2.ts',
-                    '/test/component.ts',
-                    '/test/shared.ts',
-                    '/test/important.ts',
-                ]),
-            }), mockOptions);
-            // Verify that all paths are included in the getSourceCode function call
-            expect(prompt[0].functionCalls?.[0].args?.filePaths).toEqual(expect.arrayContaining([
-                '/test/main.ts',
-                '/test/util1.ts',
-                '/test/util2.ts',
-                '/test/component.ts',
-                '/test/shared.ts',
-                '/test/important.ts',
-            ]));
-            // Verify the source code response is properly formatted
-            expect(prompt[1].functionResponses?.[0].name).toBe('getSourceCode');
-            const responseContent = JSON.parse(prompt[1].functionResponses?.[0].content ?? '');
-            expect(responseContent['/test']).toBeDefined();
-            expect(Object.keys(responseContent['/test'])).toHaveLength(6); // All files should be included
-        });
-        it('should handle empty affectedFiles in codegenPlanning', async () => {
-            importantContext.files = [];
-            const mockCodegenPlanning = {
-                name: 'codegenPlanning',
-                args: {
-                    affectedFiles: [],
-                },
-            };
-            const prompt = [];
-            const result = await executeStepEnsureContext(prompt, mockCodegenPlanning, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');
-            expect(getSourceCode).not.toHaveBeenCalled();
-        });
-        it('should handle missing affectedFiles in codegenPlanning', async () => {
-            importantContext.files = [];
-            const mockCodegenPlanning = {
-                name: 'codegenPlanning',
-                args: {},
-            };
-            const prompt = [];
-            const result = await executeStepEnsureContext(prompt, mockCodegenPlanning, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');
-            expect(getSourceCode).not.toHaveBeenCalled();
-        });
-    });
-    describe('Context checking in conversation', () => {
-        it('should identify files already in context', async () => {
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: {
-                    fileUpdates: [{ filePath: '/test/file1.ts' }],
-                    contextPaths: ['/test/context1.ts'],
-                    explanation: 'test',
-                },
-            };
-            const prompt = [
-                {
-                    type: 'user',
-                    functionResponses: [
-                        {
-                            name: 'getSourceCode',
-                            content: JSON.stringify({
-                                '/test': {
-                                    'file1.ts': { content: 'existing content' },
-                                },
-                            }),
-                        },
-                    ],
-                },
-            ];
-            vi.mocked(getSourceCode).mockReturnValue({
-                '/test/context1.ts': { content: 'context1' },
-                '/test/important.ts': { content: 'important' },
-            });
-            const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(getSourceCode).toHaveBeenCalledWith(expect.objectContaining({
-                filterPaths: expect.arrayContaining(['/test/context1.ts', '/test/important.ts']),
-            }), mockOptions);
-        });
-        it('should handle both content and summary in context', async () => {
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: {
-                    fileUpdates: [{ filePath: '/test/file1.ts' }],
-                    contextPaths: ['/test/context1.ts'],
-                    explanation: 'test',
-                },
-            };
-            const prompt = [
-                {
-                    type: 'user',
-                    functionResponses: [
-                        {
-                            name: 'getSourceCode',
-                            content: JSON.stringify({
-                                '/test': {
-                                    'file1.ts': { content: 'existing content' },
-                                    'context1.ts': { summary: 'summary only' },
-                                },
-                            }),
-                        },
-                    ],
-                },
-            ];
-            vi.mocked(getSourceCode).mockReturnValue({
-                '/test/context1.ts': { content: 'context1' },
-                '/test/important.ts': { content: 'important' },
-            });
-            const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(getSourceCode).toHaveBeenCalledWith(expect.objectContaining({
-                filterPaths: expect.arrayContaining(['/test/context1.ts', '/test/important.ts']),
-            }), mockOptions);
-        });
-    });
-    describe('getSourceCode appending', () => {
-        it('should append getSourceCode call/response for missing files', async () => {
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: {
-                    fileUpdates: [{ filePath: '/test/new1.ts' }],
-                    contextPaths: ['/test/new2.ts'],
-                    explanation: 'test',
-                },
-            };
-            const prompt = [];
-            vi.mocked(getSourceCode).mockReturnValue({
-                '/test/new1.ts': { content: 'new1' },
-                '/test/new2.ts': { content: 'new2' },
-                '/test/important.ts': { content: 'important' },
-            });
-            await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(prompt).toHaveLength(2);
-            expect(prompt[0].type).toBe('assistant');
-            expect(prompt[0].functionCalls?.[0].name).toBe('getSourceCode');
-            expect(prompt[1].type).toBe('user');
-            expect(prompt[1].functionResponses?.[0].name).toBe('getSourceCode');
-            expect(JSON.parse(prompt[1].functionResponses?.[0].content ?? '')).toEqual({
-                '/test': {
-                    'new1.ts': { content: 'new1' },
-                    'new2.ts': { content: 'new2' },
-                    'important.ts': { content: 'important' },
-                },
-            });
-        });
-    });
-    describe('Edge cases and error handling', () => {
-        it('should handle invalid getSourceCode responses in context', async () => {
-            console.warn = vi.fn();
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: {
-                    fileUpdates: [{ filePath: '/test/file1.ts' }],
-                    contextPaths: [],
-                    explanation: 'test',
-                },
-            };
-            const prompt = [
-                {
-                    type: 'user',
-                    functionResponses: [
-                        {
-                            name: 'getSourceCode',
-                            content: 'invalid json',
-                        },
-                    ],
-                },
-            ];
-            vi.mocked(getSourceCode).mockReturnValue({
-                '/test/file1.ts': { content: 'content1' },
-                '/test/important.ts': { content: 'important' },
-            });
-            const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(result).toBe(StepResult.CONTINUE);
-            expect(console.warn).toHaveBeenCalledWith('Failed to parse getSourceCode response:', expect.any(Error));
-        });
-        it('should handle getSourceCode errors', async () => {
-            const mockCodegenSummary = {
-                name: 'codegenSummary',
-                args: {
-                    fileUpdates: [{ filePath: '/test/file1.ts' }],
-                    contextPaths: [],
-                    explanation: 'test',
-                },
-            };
-            vi.mocked(getSourceCode).mockImplementation(() => {
-                throw new Error('Mock error');
-            });
-            const prompt = [];
-            const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);
-            expect(result).toBe(StepResult.BREAK);
-            expect(putSystemMessage).toHaveBeenCalledWith('Error: Context completeness check failed. This is unexpected.');
-        });
-    });
-});
-//# sourceMappingURL=step-ensure-context.test.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"step-ensure-context.test.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-ensure-context.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAE3C,OAAO,2BAA2B,CAAC;AAGnC,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAExD,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,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;QAChB,KAAK,EAAE,CAAC,oBAAoB,CAAC;KAC9B;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,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,gBAAgB,CAAC,KAAK,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qCAAqC,EAAE,GAAG,EAAE;QACnD,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC;oBAC7E,YAAY,EAAE,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;oBACxD,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC;gBACvC,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;gBACzC,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;gBACzC,mBAAmB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;gBAC5C,mBAAmB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;gBAC5C,oBAAoB,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;aAC/C,CAAC,CAAC;YAEH,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,MAAM,CAAC,eAAe,CAAC;oBAClC,gBAAgB;oBAChB,gBAAgB;oBAChB,mBAAmB;oBACnB,mBAAmB;oBACnB,oBAAoB;iBACrB,CAAC;aACH,CAAC,EACF,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,gBAAgB,CAAC,KAAK,GAAG,EAAE,CAAC;YAE5B,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,EAAE;oBACf,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YAE7C,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YAChF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,gBAAgB,CAAC,KAAK,GAAG,EAAE,CAAC;YAC5B,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,SAAS;aAChB,CAAC;YAEF,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YAChF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;QACpD,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,mBAAmB,GAAG;gBAC1B,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE;oBACJ,aAAa,EAAE;wBACb;4BACE,QAAQ,EAAE,eAAe;4BACzB,YAAY,EAAE,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;yBACnD;wBACD;4BACE,QAAQ,EAAE,oBAAoB;4BAC9B,YAAY,EAAE,CAAC,iBAAiB,CAAC;yBAClC;qBACF;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC;gBACvC,eAAe,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE;gBAC5C,gBAAgB,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;gBAC9C,gBAAgB,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;gBAC9C,oBAAoB,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE;gBACtD,iBAAiB,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE;gBAChD,oBAAoB,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE;aACvD,CAAC,CAAC;YAEH,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,mBAAmB,EAAE,WAAW,CAAC,CAAC;YAExF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,MAAM,CAAC,eAAe,CAAC;oBAClC,eAAe;oBACf,gBAAgB;oBAChB,gBAAgB;oBAChB,oBAAoB;oBACpB,iBAAiB;oBACjB,oBAAoB;iBACrB,CAAC;aACH,CAAC,EACF,WAAW,CACZ,CAAC;YAEF,wEAAwE;YACxE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,OAAO,CAC1D,MAAM,CAAC,eAAe,CAAC;gBACrB,eAAe;gBACf,gBAAgB;gBAChB,gBAAgB;gBAChB,oBAAoB;gBACpB,iBAAiB;gBACjB,oBAAoB;aACrB,CAAC,CACH,CAAC;YAEF,wDAAwD;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;YACnF,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,+BAA+B;QAChG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,gBAAgB,CAAC,KAAK,GAAG,EAAE,CAAC;YAC5B,MAAM,mBAAmB,GAAG;gBAC1B,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE;oBACJ,aAAa,EAAE,EAAE;iBAClB;aACF,CAAC;YAEF,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,mBAAmB,EAAE,WAAW,CAAC,CAAC;YAExF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YAChF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,gBAAgB,CAAC,KAAK,GAAG,EAAE,CAAC;YAC5B,MAAM,mBAAmB,GAAG;gBAC1B,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE,EAAE;aACT,CAAC;YAEF,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,mBAAmB,EAAE,WAAW,CAAC,CAAC;YAExF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YAChF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC;oBAC7C,YAAY,EAAE,CAAC,mBAAmB,CAAC;oBACnC,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG;gBACb;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;gCACtB,OAAO,EAAE;oCACP,UAAU,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE;iCAC5C;6BACF,CAAC;yBACH;qBACF;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC;gBACvC,mBAAmB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;gBAC5C,oBAAoB,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;aAC/C,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,mBAAmB,EAAE,oBAAoB,CAAC,CAAC;aACjF,CAAC,EACF,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC;oBAC7C,YAAY,EAAE,CAAC,mBAAmB,CAAC;oBACnC,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG;gBACb;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;gCACtB,OAAO,EAAE;oCACP,UAAU,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE;oCAC3C,aAAa,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE;iCAC3C;6BACF,CAAC;yBACH;qBACF;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC;gBACvC,mBAAmB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;gBAC5C,oBAAoB,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;aAC/C,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,mBAAmB,EAAE,oBAAoB,CAAC,CAAC;aACjF,CAAC,EACF,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;oBAC5C,YAAY,EAAE,CAAC,eAAe,CAAC;oBAC/B,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC;gBACvC,eAAe,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;gBACpC,eAAe,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;gBACpC,oBAAoB,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;aAC/C,CAAC,CAAC;YAEH,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAExE,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAChE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;gBACzE,OAAO,EAAE;oBACP,SAAS,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;oBAC9B,SAAS,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;oBAC9B,cAAc,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;iBACzC;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,OAAO,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEvB,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC;oBAC7C,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG;gBACb;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,cAAc;yBACxB;qBACF;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC;gBACvC,gBAAgB,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE;gBACzC,oBAAoB,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;aAC/C,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,yCAAyC,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,kBAAkB,GAAG;gBACzB,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC;oBAC7C,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC/C,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAiB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,WAAW,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACtC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,+DAA+D,CAAC,CAAC;QACjH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/prompt/steps/step-generate-image.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"step-generate-image.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/step-generate-image.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEjF,wBAAsB,wBAAwB,CAC5C,eAAe,EAAE,qBAAqB,EACtC,iBAAiB,EAAE,YAAY,GAC9B,OAAO,CACR,YAAY,CACR;IACE,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;CACrB,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,SAAS,CAAC;CACxB,CACJ,CACF,CA2CA"}

package/dist/prompt/steps/step-generate-summary.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"step-generate-summary.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/step-generate-summary.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,uBAAuB,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACtG,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAK7D,wBAAsB,0BAA0B,CAC9C,iBAAiB,EAAE,uBAAuB,EAC1C,MAAM,EAAE,UAAU,EAAE,EACpB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,IAAI,CAAC,CAqCf"}

package/dist/prompt/steps/step-history-update.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"step-history-update.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/step-history-update.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAuB,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAEtG,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAG7D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAG9C,wBAAsB,wBAAwB,CAC5C,iBAAiB,EAAE,uBAAuB,EAC1C,MAAM,EAAE,UAAU,EAAE,EACpB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,UAAU,CAAC,CAgDrB;AAED,wBAAgB,iBAAiB,WAEhC"}

package/dist/prompt/steps/step-summarization.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"step-summarization.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/step-summarization.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAiD,MAAM,4BAA4B,CAAC;AACpH,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAiD1E,wBAAsB,mBAAmB,CACvC,iBAAiB,EAAE,uBAAuB,EAC1C,UAAU,EAAE,aAAa,EACzB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,IAAI,CAAC,CAQf;AAsED,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM;;;cAQ1C"}

package/dist/prompt/steps/step-validate-recover.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"step-validate-recover.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/step-validate-recover.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAGxG,wBAAsB,8BAA8B,CAClD,CAAC,MAAM,EAAE,YAAY,EAAE,oBAAoB,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,mBAAmB,EAC9F,MAAM,EAAE,YAAY,EAAE,EACtB,iBAAiB,EAAE,uBAAuB,GACzC,OAAO,CAAC,YAAY,EAAE,CAAC,CA2DzB"}

package/dist/prompt/steps/step-verify-patch.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"step-verify-patch.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/step-verify-patch.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,UAAU,EACV,WAAW,EACX,YAAY,EACZ,uBAAuB,EAExB,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAE7D,wBAAsB,sBAAsB,CAC1C,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,EACxD,iBAAiB,EAAE,uBAAuB,EAC1C,MAAM,EAAE,UAAU,EAAE,EACpB,YAAY,EAAE,WAAW,EAAE,EAC3B,WAAW,EAAE,MAAM,EACnB,KAAK,EAAE,OAAO,EACd,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,YAAY,EAAE,CAAC,CAkCzB"}

package/dist/prompt/steps/steps-types.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"steps-types.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/steps-types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAExD,oBAAY,UAAU;IACpB,QAAQ,IAAA;IACR,KAAK,IAAA;CACN;AAED,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,YAAY,GAAG,MAAM,CAC/B,MAAM,EACN;IACE,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,cAAc,EAAE,CAAC;CAChC,CACF,GAAG;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC"}

package/dist/prompt/steps/steps-utils.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"steps-utils.d.ts","sourceRoot":"","sources":["../../../src/prompt/steps/steps-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAExD,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,UAAU,EAAE;;;;;SAezD"}

package/dist/prompt/systemprompt.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"systemprompt.d.ts","sourceRoot":"","sources":["../../src/prompt/systemprompt.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAG1D,wBAAgB,eAAe,CAAC,OAAO,EAAE,cAAc,UA0HtD"}

package/dist/prompt/systemprompt.test.d.ts

@@ -1,3 +0,0 @@
-import '../cli/cli-params.js';
-import '../files/find-files.js';
-import '../main/config.js';

package/dist/prompt/systemprompt.test.js

@@ -1,42 +0,0 @@
-import { describe, it, expect, beforeEach, vi } from 'vitest';
-import { getSystemPrompt } from './systemprompt.js';
-import '../cli/cli-params.js';
-import '../files/find-files.js';
-import '../main/config.js';
-vi.mock('../cli/cli-params.js', () => ({
-    disableExplanations: true,
-    explicitPrompt: false,
-    allowFileCreate: false,
-    allowFileDelete: false,
-    allowDirectoryCreate: false,
-    allowFileMove: false,
-    verbosePrompt: false,
-    askQuestion: false,
-}));
-vi.mock('../files/find-files.js', () => ({
-    getSourceFiles: vi.fn(),
-}));
-vi.mock('../main/config.js', () => ({
-    rcConfig: { rootDir: '/mocked/root/dir' },
-    importantContext: {},
-}));
-describe('getSystemPrompt', () => {
-    beforeEach(() => {
-        vi.clearAllMocks();
-    });
-    it('generates correct system prompt', () => {
-        const systemPrompt = getSystemPrompt({ aiService: 'vertex-ai', askQuestion: false });
-        expect(systemPrompt).toContain('You are GenAIcode, a code generation assistant');
-        expect(systemPrompt).toContain('You should parse my application source code');
-        expect(systemPrompt).toContain('Please limit any changes to the root directory of my application, which is `/mocked/root/dir`');
-        expect(systemPrompt).toContain('absolute file paths exactly');
-    });
-    it('verifies system prompt limit', () => {
-        const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => { });
-        getSystemPrompt({ aiService: 'vertex-ai', verbose: true, askQuestion: false });
-        expect(consoleSpy).toHaveBeenCalledWith('System prompt:');
-        expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining('You are GenAIcode, a code generation assistant'));
-        consoleSpy.mockRestore();
-    });
-});
-//# sourceMappingURL=systemprompt.test.js.map

package/dist/prompt/systemprompt.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"systemprompt.test.js","sourceRoot":"","sources":["../../src/prompt/systemprompt.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,sBAAsB,CAAC;AAC9B,OAAO,wBAAwB,CAAC;AAChC,OAAO,mBAAmB,CAAC;AAE3B,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,mBAAmB,EAAE,IAAI;IACzB,cAAc,EAAE,KAAK;IACrB,eAAe,EAAE,KAAK;IACtB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,KAAK;IAC3B,aAAa,EAAE,KAAK;IACpB,aAAa,EAAE,KAAK;IACpB,WAAW,EAAE,KAAK;CACnB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;CACxB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE;IACzC,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,YAAY,GAAG,eAAe,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;QAErF,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,gDAAgD,CAAC,CAAC;QACjF,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,6CAA6C,CAAC,CAAC;QAC9E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAC5B,+FAA+F,CAChG,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAEzE,eAAe,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;QAE/E,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAC1D,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,gDAAgD,CAAC,CAAC,CAAC;QAEnH,UAAU,CAAC,WAAW,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/prompt/token-estimator.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"token-estimator.d.ts","sourceRoot":"","sources":["../../src/prompt/token-estimator.ts"],"names":[],"mappings":"AAOA,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAkBxD"}

package/dist/src/ai-service/ai-studio.d.ts

@@ -1,6 +0,0 @@
-import { FunctionCall, FunctionDef, PromptItem } from './common.js';
-import { CodegenOptions } from '../main/codegen-types.js';
-/**
- * This function generates content using the Anthropic Claude model via Vertex AI.
- */
-export declare function generateContent(prompt: PromptItem[], functionDefs: FunctionDef[], requiredFunctionName: string | null, temperature: number, cheap: boolean | undefined, options: CodegenOptions): Promise<FunctionCall[]>;

package/dist/src/ai-service/ai-studio.js

@@ -1,190 +0,0 @@
-import { FunctionCallingMode, GoogleGenerativeAI, HarmBlockThreshold, HarmCategory, } from '@google/generative-ai';
-import assert from 'node:assert';
-import { printTokenUsageAndCost, processFunctionCalls } from './common.js';
-import { abortController } from '../main/interactive/codegen-worker.js';
-import { modelOverrides } from '../main/config.js';
-import { unescapeFunctionCall } from './unescape-function-call.js';
-import { disableVertexUnescape } from '../cli/cli-params.js';
-/**
- * This function generates content using the Anthropic Claude model via Vertex AI.
- */
-export async function generateContent(prompt, functionDefs, requiredFunctionName, temperature, cheap = false, options) {
-    const messages = prompt
-        .filter((item) => item.type === 'user' || item.type === 'assistant')
-        .map((item) => {
-        if (item.type === 'user') {
-            const content = {
-                role: 'user',
-                parts: [
-                    ...(item.functionResponses ?? []).map((response) => ({
-                        functionResponse: {
-                            name: response.name,
-                            response: { name: response.name, content: response.content },
-                        },
-                    })),
-                    ...(item.images ?? []).map((image) => ({
-                        inlineData: {
-                            mimeType: image.mediaType,
-                            data: image.base64url,
-                        },
-                    })),
-                    { text: item.text },
-                ],
-            };
-            return content;
-        }
-        else {
-            assert(item.type === 'assistant');
-            const content = {
-                role: 'model',
-                parts: [
-                    ...(item.text ? [{ text: item.text }] : []),
-                    ...(item.functionCalls ?? []).map((call) => ({
-                        functionCall: {
-                            name: call.name,
-                            args: call.args ?? {},
-                        },
-                    })),
-                ],
-            };
-            return content;
-        }
-    });
-    const req = {
-        contents: messages,
-        tools: [
-            {
-                functionDeclarations: functionDefs,
-            },
-        ],
-        toolConfig: {
-            functionCallingConfig: {
-                mode: FunctionCallingMode.ANY,
-                ...(requiredFunctionName ? { allowedFunctionNames: [requiredFunctionName] } : {}),
-            },
-        },
-    };
-    const model = getModel(cheap ? 'gemini-1.5-flash-002' : 'gemini-1.5-pro-002', temperature, prompt.find((item) => item.type === 'systemPrompt').systemPrompt, options.geminiBlockNone);
-    const result = await model.generateContent(req, { signal: abortController?.signal });
-    // Print token usage
-    const usageMetadata = result.response.usageMetadata;
-    const usage = {
-        inputTokens: usageMetadata.promptTokenCount,
-        outputTokens: usageMetadata.candidatesTokenCount,
-        totalTokens: usageMetadata.totalTokenCount,
-    };
-    printTokenUsageAndCost({
-        aiService: 'ai-studio',
-        usage,
-        inputCostPerToken: 0.000125 / 1000,
-        outputCostPerToken: 0.000375 / 1000,
-        cheap,
-    });
-    if (result.response.promptFeedback) {
-        console.log('Prompt feedback:');
-        console.log(JSON.stringify(result.response.promptFeedback, null, 2));
-    }
-    if (!result.response.candidates?.length) {
-        console.log('Response:', result);
-        throw new Error('No candidates found');
-    }
-    const functionCalls = result.response.candidates
-        .map((candidate) => candidate.content.parts?.map((part) => part.functionCall))
-        .flat()
-        .filter((functionCall) => !!functionCall)
-        .map((call) => ({ name: call.name, args: call.args }))
-        .map(disableVertexUnescape ? (call) => call : unescapeFunctionCall);
-    if (functionCalls.length === 0) {
-        const textResponse = result.response.candidates
-            .map((candidate) => candidate.content.parts?.map((part) => part.text))
-            .flat()
-            .filter((text) => !!text)[0] ??
-            result.response.candidates.map((candidate) => 
-            // @ts-expect-error: FinishReason type does not have this error
-            candidate.finishReason === 'MALFORMED_FUNCTION_CALL' ? candidate.finishMessage : '')[0];
-        const recoveredFunctionCall = textResponse &&
-            (await recoverFunctionCall(textResponse, functionCalls, functionDefs.find((def) => def.name === requiredFunctionName)));
-        if (!recoveredFunctionCall) {
-            console.log('No function calls, output text response if it exists:', textResponse);
-        }
-        else {
-            console.log('Recovered function call.');
-        }
-    }
-    return processFunctionCalls(functionCalls, functionDefs);
-}
-function getModel(defaultModel, temperature, systemPrompt, geminiBlockNone) {
-    assert(process.env.API_KEY, 'API_KEY environment variable is not set');
-    const genAI = new GoogleGenerativeAI(process.env.API_KEY);
-    const cheap = defaultModel === 'gemini-1.5-flash-001';
-    const model = cheap
-        ? (modelOverrides.aiStudio?.cheap ?? defaultModel)
-        : (modelOverrides.aiStudio?.default ?? defaultModel);
-    console.log(`Using AI Studio model: ${model}`);
-    return genAI.getGenerativeModel({
-        model,
-        generationConfig: {
-            maxOutputTokens: 8192,
-            temperature,
-            topP: 0.95,
-        },
-        safetySettings: [
-            {
-                category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
-                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
-            },
-            {
-                category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
-            },
-            {
-                category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
-            },
-            {
-                category: HarmCategory.HARM_CATEGORY_HARASSMENT,
-                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
-            },
-        ],
-        systemInstruction: {
-            role: 'system',
-            parts: [
-                {
-                    text: systemPrompt,
-                },
-            ],
-        },
-    });
-}
-async function recoverFunctionCall(textResponse, functionCalls, functionDef) {
-    console.log('Recovering function call');
-    // @ts-expect-error: "object" !== "OBJECT"
-    const schema = functionDef.parameters;
-    const result = await getModel('gemini-1.5-pro-002', 0.2, 'Your role is read the text below and if possible, return it in the desired format.', undefined).generateContent({
-        contents: [
-            {
-                role: 'user',
-                parts: [
-                    {
-                        text: textResponse,
-                    },
-                ],
-            },
-        ],
-        generationConfig: {
-            responseMimeType: 'application/json',
-            responseSchema: schema,
-        },
-    }, {
-        signal: abortController?.signal,
-    });
-    try {
-        const parsed = JSON.parse(result.response.text());
-        functionCalls.push(unescapeFunctionCall({ name: functionDef.name, args: parsed }));
-        return true;
-    }
-    catch {
-        return false;
-    }
-}
-//# sourceMappingURL=ai-studio.js.map