genaicode 0.5.0 → 0.6.0

package/dist/src/prompt/prompt-service.test.js

@@ -0,0 +1,553 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { promptService } from './prompt-service.js';
+import * as aiStudio from '../ai-service/ai-studio.js';
+import * as vertexAi from '../ai-service/vertex-ai.js';
+import * as vertexAiClaude from '../ai-service/vertex-ai-claude.js';
+import * as chatGpt from '../ai-service/chat-gpt.js';
+import * as anthropic from '../ai-service/anthropic.js';
+import * as cliParams from '../cli/cli-params.js';
+import fs from 'fs';
+import * as diff from 'diff';
+import mime from 'mime-types';
+import '../files/cache-file.js';
+import { getImageAssets } from '../files/read-files.js';
+import '../files/find-files.js';
+import * as dalleService from '../ai-service/dall-e.js';
+import * as vertexAiImagen from '../ai-service/vertex-ai-imagen.js';
+import { getCodeGenPrompt } from './prompt-codegen.js';
+vi.mock('../ai-service/vertex-ai-claude.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/vertex-ai.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/chat-gpt.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/anthropic.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../cli/cli-params.js', () => ({
+    disableExplanations: true,
+    considerAllFiles: false,
+    dependencyTree: false,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    disableContextOptimization: true,
+    vision: false,
+    imagen: false,
+    temperature: 0.7,
+    cheap: false,
+    askQuestion: false,
+}));
+vi.mock('fs');
+vi.mock('diff');
+vi.mock('mime-types');
+vi.mock('../ai-service/dall-e.js', () => ({ generateImage: vi.fn() }));
+vi.mock('../ai-service/vertex-ai-imagen.js', () => ({ generateImage: vi.fn() }));
+vi.mock('../files/cache-file.js');
+// Mock find-files module
+vi.mock('../files/find-files.js', () => ({
+    getSourceFiles: () => [],
+    getImageAssetFiles: () => [],
+}));
+// Mock read-files module
+vi.mock('../files/read-files.js', () => ({
+    getSourceCode: () => ({}),
+    getImageAssets: vi.fn(() => ({})),
+}));
+vi.mock('../main/config.js', () => ({
+    rootDir: '/mocked/root/dir',
+    rcConfig: {
+        rootDir: '/mocked/root/dir',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+    importantContext: {},
+}));
+const GENERATE_CONTENT_FNS = {
+    'vertex-ai-claude': vertexAiClaude.generateContent,
+    'vertex-ai': vertexAi.generateContent,
+    'ai-studio': aiStudio.generateContent,
+    anthropic: anthropic.generateContent,
+    'chat-gpt': chatGpt.generateContent,
+};
+const GENERATE_IMAGE_FNS = {
+    'dall-e': dalleService.generateImage,
+    'vertex-ai': vertexAiImagen.generateImage,
+};
+describe('promptService', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+        vi.mocked(cliParams).dryRun = false;
+        vi.mocked(cliParams).vision = false;
+        vi.mocked(cliParams).imagen = undefined;
+        vi.mocked(cliParams).disableContextOptimization = false;
+    });
+    it('should process the codegen summary and return the result with Vertex AI', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Hello");' } },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+        expect(result).toEqual(mockFunctionCalls);
+        expect(vertexAi.generateContent).toHaveBeenCalled();
+    });
+    it('should process the codegen summary and return the result with ChatGPT', async () => {
+        vi.mocked(cliParams).aiService = 'chat-gpt';
+        const mockFunctionCalls = [{ name: 'createFile', args: { filePath: 'new.js', newContent: 'const x = 5;' } }];
+        vi.mocked(chatGpt.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'chat-gpt' }));
+        expect(result).toEqual(mockFunctionCalls);
+        expect(chatGpt.generateContent).toHaveBeenCalled();
+    });
+    it('should process the codegen summary and return the result with Anthropic', async () => {
+        vi.mocked(cliParams).aiService = 'anthropic';
+        const mockFunctionCalls = [{ name: 'deleteFile', args: { filePath: 'obsolete.js' } }];
+        vi.mocked(anthropic.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'anthropic' }));
+        expect(result).toEqual(mockFunctionCalls);
+        expect(anthropic.generateContent).toHaveBeenCalled();
+    });
+    it('should not update files in dry run mode', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        vi.mocked(cliParams).dryRun = true;
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Dry run");' } },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+        expect(result).toEqual(mockFunctionCalls);
+        expect(vertexAi.generateContent).toHaveBeenCalled();
+    });
+    it('should handle invalid patchFile call and retry without patchFile function', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [{ filePath: '/mocked/root/dir/test.js', updateToolName: 'patchFile', prompt: 'Generate file' }],
+                    contextPaths: [],
+                    explanation: 'Mock summary',
+                },
+            },
+        ];
+        const mockInvalidPatchCall = [
+            {
+                name: 'patchFile',
+                args: {
+                    filePath: '/mocked/root/dir/test.js',
+                    patch: 'invalid patch',
+                },
+            },
+        ];
+        const mockValidUpdateCall = [
+            {
+                name: 'updateFile',
+                args: {
+                    filePath: '/mocked/root/dir/test.js',
+                    newContent: 'console.log("Updated content");',
+                },
+            },
+        ];
+        // Mock the first call to return the codegen summary
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        // Mock the second call to return an invalid patch
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockInvalidPatchCall);
+        // Mock the third call (retry) to return a valid update
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockValidUpdateCall);
+        // Mock fs.readFileSync to return some content
+        vi.mocked(fs.readFileSync).mockReturnValue('Original content');
+        // Mock diff.applyPatch to fail for the invalid patch
+        vi.mocked(diff.applyPatch).mockReturnValue(false);
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+        expect(fs.readFileSync).toHaveBeenCalledWith('/mocked/root/dir/test.js', 'utf-8');
+        expect(diff.applyPatch).toHaveBeenCalledWith('Original content', 'invalid patch');
+        expect(result).toEqual(mockValidUpdateCall);
+    });
+    it('should include image assets when vision flag is true', async () => {
+        vi.mocked(cliParams).aiService = 'chat-gpt';
+        vi.mocked(cliParams).vision = true;
+        const mockImageAssets = {
+            '/path/to/image1.png': { width: 100, height: 100, mimeType: 'image/png' },
+            '/path/to/image2.jpg': { width: 200, height: 200, mimeType: 'image/png' },
+        };
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Vision test");' } },
+        ];
+        vi.mocked(getImageAssets).mockReturnValue(mockImageAssets);
+        vi.mocked(chatGpt.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'chat-gpt', vision: true }));
+        expect(chatGpt.generateContent).toHaveBeenCalled();
+        const calls = vi.mocked(chatGpt.generateContent).mock.calls[0];
+        expect(calls[0]).toEqual(expect.arrayContaining([
+            expect.objectContaining({
+                type: 'user',
+                text: expect.stringContaining('I should also provide you with a summary of application image assets'),
+            }),
+            expect.objectContaining({
+                type: 'assistant',
+                text: expect.stringContaining('Please provide summary of application image assets.'),
+            }),
+            expect.objectContaining({
+                type: 'user',
+                functionResponses: [{ name: 'getImageAssets', content: JSON.stringify(mockImageAssets) }],
+            }),
+        ]));
+    });
+    it('should include image data in the prompt when processing files with vision', async () => {
+        vi.mocked(cliParams).aiService = 'chat-gpt';
+        vi.mocked(cliParams).vision = true;
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        {
+                            filePath: '/mocked/root/dir/test.js',
+                            updateToolName: 'updateFile',
+                            contextImageAssets: ['/mocked/root/dir/image1.png', '/mocked/root/dir/image2.jpg'],
+                            prompt: 'Generate file update',
+                        },
+                    ],
+                    contextPaths: [],
+                    explanation: 'Mock summary',
+                },
+            },
+        ];
+        const mockUpdateCall = [
+            {
+                name: 'updateFile',
+                args: {
+                    filePath: '/mocked/root/dir/test.js',
+                    newContent: 'console.log("Updated with vision");',
+                },
+            },
+        ];
+        vi.mocked(chatGpt.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(chatGpt.generateContent).mockResolvedValueOnce(mockUpdateCall);
+        vi.mocked(fs.readFileSync).mockImplementation((path) => `mock-base64-data-for-${path}`);
+        vi.mocked(mime.lookup).mockImplementation((path) => (path.endsWith('.png') ? 'image/png' : 'image/jpeg'));
+        await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'chat-gpt', vision: true }));
+        expect(chatGpt.generateContent).toHaveBeenCalledTimes(2);
+        const secondCall = vi.mocked(chatGpt.generateContent).mock.calls[1];
+        expect(secondCall[0]).toEqual(expect.arrayContaining([
+            expect.objectContaining({
+                type: 'user',
+                text: expect.stringContaining('Generate file update'),
+                images: [
+                    {
+                        path: '/mocked/root/dir/image1.png',
+                        base64url: 'mock-base64-data-for-/mocked/root/dir/image1.png',
+                        mediaType: 'image/png',
+                    },
+                    {
+                        path: '/mocked/root/dir/image2.jpg',
+                        base64url: 'mock-base64-data-for-/mocked/root/dir/image2.jpg',
+                        mediaType: 'image/jpeg',
+                    },
+                ],
+            }),
+        ]));
+    });
+    it('should not include image assets when vision flag is false', async () => {
+        vi.mocked(cliParams).aiService = 'chat-gpt';
+        vi.mocked(cliParams).vision = false;
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("No vision test");' } },
+        ];
+        vi.mocked(chatGpt.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'chat-gpt' }));
+        expect(chatGpt.generateContent).toHaveBeenCalled();
+        const calls = vi.mocked(chatGpt.generateContent).mock.calls[0];
+        expect(calls[0]).not.toEqual(expect.arrayContaining([
+            expect.objectContaining({
+                type: 'user',
+                text: expect.stringContaining('I should also provide you with a summary of application image assets'),
+            }),
+        ]));
+    });
+    it('should handle context optimization', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        { filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile', prompt: 'Generate file' },
+                    ],
+                    contextPaths: ['/mocked/root/dir/context1.js', '/mocked/root/dir/context2.js'],
+                    explanation: 'Mock summary with context',
+                },
+            },
+        ];
+        const mockUpdateCall = [
+            {
+                name: 'updateFile',
+                args: {
+                    filePath: '/mocked/root/dir/test.js',
+                    newContent: 'console.log("Updated with context");',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockUpdateCall);
+        await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(2);
+        const firstCall = vi.mocked(vertexAi.generateContent).mock.calls[0];
+        expect(firstCall[0]).toEqual(expect.arrayContaining([
+            expect.objectContaining({
+                type: 'user',
+                functionResponses: [
+                    expect.objectContaining({
+                        name: 'getSourceCode',
+                        content: expect.any(String),
+                    }),
+                ],
+            }),
+        ]));
+    });
+    it('should handle disableContextOptimization flag', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        vi.mocked(cliParams).disableContextOptimization = true;
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        { filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile', prompt: 'Generate file' },
+                    ],
+                    contextPaths: ['/mocked/root/dir/context1.js', '/mocked/root/dir/context2.js'],
+                    explanation: 'Mock summary without context optimization',
+                },
+            },
+        ];
+        const mockUpdateCall = [
+            {
+                name: 'updateFile',
+                args: {
+                    filePath: '/mocked/root/dir/test.js',
+                    newContent: 'console.log("Updated without context optimization");',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockUpdateCall);
+        await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(2);
+        const firstCall = vi.mocked(vertexAi.generateContent).mock.calls[0];
+        expect(firstCall[0]).not.toEqual(expect.arrayContaining([
+            expect.objectContaining({
+                type: 'user',
+                functionResponses: [
+                    expect.objectContaining({
+                        name: 'getSourceCode',
+                        content: expect.stringContaining('context1.js'),
+                    }),
+                ],
+            }),
+        ]));
+    });
+    it('should handle image generation requests', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        vi.mocked(cliParams).imagen = 'dall-e';
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        { filePath: '/mocked/root/dir/image.png', updateToolName: 'generateImage', prompt: 'Generate file' },
+                    ],
+                    contextPaths: [],
+                    explanation: 'Mock summary with image generation',
+                },
+            },
+        ];
+        const mockGenerateImageCall = [
+            {
+                name: 'generateImage',
+                args: {
+                    prompt: 'A test image',
+                    filePath: '/mocked/root/dir/image.png',
+                    width: 256,
+                    height: 256,
+                },
+            },
+        ];
+        const mockDownloadFileCall = [
+            {
+                name: 'downloadFile',
+                args: {
+                    filePath: '/mocked/root/dir/image.png',
+                    explanation: 'Downloading generated image',
+                    downloadUrl: 'https://example.com/generated-image.png',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockGenerateImageCall);
+        vi.mocked(dalleService.generateImage).mockResolvedValue('https://example.com/generated-image.png');
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({
+            askQuestion: false,
+            disableContextOptimization: true,
+            aiService: 'vertex-ai',
+            imagen: 'dall-e',
+        }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(2);
+        expect(dalleService.generateImage).toHaveBeenCalledWith('A test image', undefined, { width: 256, height: 256 }, false);
+        expect(result).toEqual(expect.arrayContaining(mockDownloadFileCall));
+    });
+    it('should handle image generation failure', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        vi.mocked(cliParams).imagen = 'dall-e';
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        { filePath: '/mocked/root/dir/image.png', updateToolName: 'generateImage', prompt: 'Generate file' },
+                    ],
+                    contextPaths: [],
+                    explanation: 'Mock summary with image generation failure',
+                },
+            },
+        ];
+        const mockGenerateImageCall = [
+            {
+                name: 'generateImage',
+                args: {
+                    prompt: 'A test image',
+                    filePath: '/mocked/root/dir/image.png',
+                    width: 256,
+                    height: 256,
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockGenerateImageCall);
+        vi.mocked(dalleService.generateImage).mockRejectedValue(new Error('Image generation failed'));
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({
+            askQuestion: false,
+            disableContextOptimization: true,
+            aiService: 'vertex-ai',
+            imagen: 'dall-e',
+        }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(2);
+        expect(dalleService.generateImage).toHaveBeenCalledWith('A test image', undefined, { width: 256, height: 256 }, false);
+        expect(result).toEqual(expect.arrayContaining([
+            expect.objectContaining({
+                name: 'explanation',
+                args: {
+                    text: expect.stringContaining('Failed to generate image: Image generation failed'),
+                },
+            }),
+        ]));
+    });
+    it('should handle unexpected response without codegen summary', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        const mockUnexpectedResponse = [
+            {
+                name: 'updateFile',
+                args: {
+                    filePath: '/mocked/root/dir/test.js',
+                    newContent: 'console.log("Unexpected response");',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockUnexpectedResponse);
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(1);
+        expect(result).toEqual(mockUnexpectedResponse);
+    });
+    describe('validateAndRecoverSingleResult', () => {
+        it('should successfully recover from an invalid function call', async () => {
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockInvalidCall = [
+                {
+                    name: 'codegenSummary',
+                    args: {
+                        files: [{ filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile' }],
+                        contextPaths: [],
+                        explanation: 'Mock summary',
+                    },
+                },
+            ];
+            const mockValidCall = [
+                {
+                    name: 'codegenSummary',
+                    args: {
+                        fileUpdates: [
+                            { filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile', prompt: 'Generate file' },
+                        ],
+                        contextPaths: [],
+                        explanation: 'Mock summary',
+                    },
+                },
+            ];
+            vi.mocked(vertexAi.generateContent)
+                .mockResolvedValueOnce(mockInvalidCall)
+                .mockResolvedValueOnce(mockValidCall)
+                .mockResolvedValueOnce([
+                {
+                    name: 'updateFile',
+                    args: {
+                        filePath: '/mocked/root/dir/test.js',
+                        newContent: 'console.log("Unexpected response");',
+                    },
+                },
+            ]);
+            const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+            expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+            expect(result).toEqual([
+                {
+                    args: {
+                        filePath: '/mocked/root/dir/test.js',
+                        newContent: 'console.log("Unexpected response");',
+                    },
+                    name: 'updateFile',
+                },
+            ]);
+        });
+        it('should handle unsuccessful recovery', async () => {
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockInvalidCall = [
+                {
+                    name: 'updateFile',
+                    args: { filePath: '/mocked/root/dir/test.js', invalidArg: 'This should not be here' },
+                },
+            ];
+            vi.mocked(vertexAi.generateContent)
+                .mockResolvedValueOnce([
+                {
+                    name: 'codegenSummary',
+                    args: {
+                        fileUpdates: [
+                            { filePath: '/mocked/root/dir/test.js', updateToolName: 'patchFile', prompt: 'Generate file' },
+                        ],
+                        contextPaths: [],
+                        explanation: 'Mock summary',
+                    },
+                },
+            ])
+                .mockResolvedValueOnce(mockInvalidCall)
+                .mockResolvedValueOnce(mockInvalidCall); // Second call also returns invalid result
+            await expect(promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }))).rejects.toThrow('Recovery failed');
+            expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+        });
+        it('should not attempt recovery for multiple valid function calls', async () => {
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockValidCalls = [
+                {
+                    name: 'updateFile',
+                    args: { filePath: 'test1.js', newContent: 'console.log("File 1");' },
+                },
+                {
+                    name: 'updateFile',
+                    args: { filePath: 'test2.js', newContent: 'console.log("File 2");' },
+                },
+            ];
+            vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockValidCalls);
+            const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ askQuestion: false, disableContextOptimization: true, aiService: 'vertex-ai' }));
+            expect(vertexAi.generateContent).toHaveBeenCalledTimes(1);
+            expect(result).toEqual(mockValidCalls);
+        });
+    });
+});
+//# sourceMappingURL=prompt-service.test.js.map

package/dist/src/prompt/prompt-service.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-service.test.js","sourceRoot":"","sources":["../../../src/prompt/prompt-service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AACvD,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AACvD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,OAAO,MAAM,2BAA2B,CAAC;AACrD,OAAO,KAAK,SAAS,MAAM,4BAA4B,CAAC;AACxD,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,wBAAwB,CAAC;AAChC,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,wBAAwB,CAAC;AAChC,OAAO,KAAK,YAAY,MAAM,yBAAyB,CAAC;AACxD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAIvD,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnF,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3E,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,mBAAmB,EAAE,IAAI;IACzB,gBAAgB,EAAE,KAAK;IACvB,cAAc,EAAE,KAAK;IACrB,cAAc,EAAE,KAAK;IACrB,eAAe,EAAE,KAAK;IACtB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,KAAK;IAC3B,aAAa,EAAE,KAAK;IACpB,aAAa,EAAE,KAAK;IACpB,0BAA0B,EAAE,IAAI;IAChC,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,WAAW,EAAE,KAAK;CACnB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAChB,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACvE,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACjF,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAClC,yBAAyB;AACzB,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;CAC7B,CAAC,CAAC,CAAC;AAEJ,yBAAyB;AACzB,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;IACzB,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;CAClC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE;QACR,OAAO,EAAE,kBAAkB;QAC3B,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;IACD,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,MAAM,oBAAoB,GAAmD;IAC3E,kBAAkB,EAAE,cAAc,CAAC,eAAe;IAClD,WAAW,EAAE,QAAQ,CAAC,eAAe;IACrC,WAAW,EAAE,QAAQ,CAAC,eAAe;IACrC,SAAS,EAAE,SAAS,CAAC,eAAe;IACpC,UAAU,EAAE,OAAO,CAAC,eAAe;CAC3B,CAAC;AAEX,MAAM,kBAAkB,GAA8C;IACpE,QAAQ,EAAE,YAAY,CAAC,aAAa;IACpC,WAAW,EAAE,cAAc,CAAC,aAAa;CACjC,CAAC;AAEX,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC;QACxC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,0BAA0B,GAAG,KAAK,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,uBAAuB,EAAE,EAAE;SAC3F,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;QAC5C,MAAM,iBAAiB,GAAG,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,CAAC,CAAC;QAC7G,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAClG,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,MAAM,iBAAiB,GAAG,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;QACtF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE9E,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC1C,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QACnC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,yBAAyB,EAAE,EAAE;SAC7F,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC;oBAC7G,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,cAAc;iBAC5B;aACF;SACF,CAAC;QACF,MAAM,oBAAoB,GAAG;YAC3B;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE;oBACJ,QAAQ,EAAE,0BAA0B;oBACpC,KAAK,EAAE,eAAe;iBACvB;aACF;SACF,CAAC;QACF,MAAM,mBAAmB,GAAG;YAC1B;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE;oBACJ,QAAQ,EAAE,0BAA0B;oBACpC,UAAU,EAAE,iCAAiC;iBAC9C;aACF;SACF,CAAC;QAEF,oDAAoD;QACpD,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC9E,kDAAkD;QAClD,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,oBAAoB,CAAC,CAAC;QAChF,uDAAuD;QACvD,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;QAE/E,8CAA8C;QAC9C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAE/D,qDAAqD;QACrD,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC;QAClF,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,EAAE,eAAe,CAAC,CAAC;QAClF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;QAC5C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QACnC,MAAM,eAAe,GAAG;YACtB,qBAAqB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,WAAW,EAAE;YACzE,qBAAqB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,WAAW,EAAE;SAC1E,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,6BAA6B,EAAE,EAAE;SACjG,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QAC3D,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,aAAa,CACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAChH,CAAC;QAEF,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CACtB,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,sEAAsE,CAAC;aACtG,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,qDAAqD,CAAC;aACrF,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,EAAE,CAAC;aAC1F,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;QAC5C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QACnC,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX;4BACE,QAAQ,EAAE,0BAA0B;4BACpC,cAAc,EAAE,YAAY;4BAC5B,kBAAkB,EAAE,CAAC,6BAA6B,EAAE,6BAA6B,CAAC;4BAClF,MAAM,EAAE,sBAAsB;yBAC/B;qBACF;oBACD,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,cAAc;iBAC5B;aACF;SACF,CAAC;QACF,MAAM,cAAc,GAAG;YACrB;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE;oBACJ,QAAQ,EAAE,0BAA0B;oBACpC,UAAU,EAAE,qCAAqC;iBAClD;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC7E,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAEzE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,wBAAwB,IAAI,EAAE,CAAC,CAAC;QACxF,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QAE1G,MAAM,aAAa,CACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAChH,CAAC;QAEF,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzD,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC3B,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC;gBACrD,MAAM,EAAE;oBACN;wBACE,IAAI,EAAE,6BAA6B;wBACnC,SAAS,EAAE,kDAAkD;wBAC7D,SAAS,EAAE,WAAW;qBACvB;oBACD;wBACE,IAAI,EAAE,6BAA6B;wBACnC,SAAS,EAAE,kDAAkD;wBAC7D,SAAS,EAAE,YAAY;qBACxB;iBACF;aACF,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;QAC5C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,gCAAgC,EAAE,EAAE;SACpG,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,aAAa,CACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAClG,CAAC;QAEF,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAC1B,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,sEAAsE,CAAC;aACtG,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,eAAe,EAAE;qBAChG;oBACD,YAAY,EAAE,CAAC,8BAA8B,EAAE,8BAA8B,CAAC;oBAC9E,WAAW,EAAE,2BAA2B;iBACzC;aACF;SACF,CAAC;QACF,MAAM,cAAc,GAAG;YACrB;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE;oBACJ,QAAQ,EAAE,0BAA0B;oBACpC,UAAU,EAAE,sCAAsC;iBACnD;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC9E,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAE1E,MAAM,aAAa,CACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC1B,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB,MAAM,CAAC,gBAAgB,CAAC;wBACtB,IAAI,EAAE,eAAe;wBACrB,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;qBAC5B,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,0BAA0B,GAAG,IAAI,CAAC;QACvD,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,eAAe,EAAE;qBAChG;oBACD,YAAY,EAAE,CAAC,8BAA8B,EAAE,8BAA8B,CAAC;oBAC9E,WAAW,EAAE,2CAA2C;iBACzD;aACF;SACF,CAAC;QACF,MAAM,cAAc,GAAG;YACrB;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE;oBACJ,QAAQ,EAAE,0BAA0B;oBACpC,UAAU,EAAE,sDAAsD;iBACnE;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC9E,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAE1E,MAAM,aAAa,CACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAC9B,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB,MAAM,CAAC,gBAAgB,CAAC;wBACtB,IAAI,EAAE,eAAe;wBACrB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC;qBAChD,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,QAAQ,CAAC;QACvC,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX,EAAE,QAAQ,EAAE,4BAA4B,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,EAAE,eAAe,EAAE;qBACrG;oBACD,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,oCAAoC;iBAClD;aACF;SACF,CAAC;QACF,MAAM,qBAAqB,GAAG;YAC5B;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,cAAc;oBACtB,QAAQ,EAAE,4BAA4B;oBACtC,KAAK,EAAE,GAAG;oBACV,MAAM,EAAE,GAAG;iBACZ;aACF;SACF,CAAC;QACF,MAAM,oBAAoB,GAAG;YAC3B;gBACE,IAAI,EAAE,cAAc;gBACpB,IAAI,EAAE;oBACJ,QAAQ,EAAE,4BAA4B;oBACtC,WAAW,EAAE,6BAA6B;oBAC1C,WAAW,EAAE,yCAAyC;iBACvD;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC9E,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,qBAAqB,CAAC,CAAC;QACjF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,iBAAiB,CAAC,yCAAyC,CAAC,CAAC;QAEnG,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC;YACf,WAAW,EAAE,KAAK;YAClB,0BAA0B,EAAE,IAAI;YAChC,SAAS,EAAE,WAAW;YACtB,MAAM,EAAE,QAAQ;SACjB,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACrD,cAAc,EACd,SAAS,EACT,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAC3B,KAAK,CACN,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,oBAAoB,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,QAAQ,CAAC;QACvC,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX,EAAE,QAAQ,EAAE,4BAA4B,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,EAAE,eAAe,EAAE;qBACrG;oBACD,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,4CAA4C;iBAC1D;aACF;SACF,CAAC;QACF,MAAM,qBAAqB,GAAG;YAC5B;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,cAAc;oBACtB,QAAQ,EAAE,4BAA4B;oBACtC,KAAK,EAAE,GAAG;oBACV,MAAM,EAAE,GAAG;iBACZ;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC9E,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,qBAAqB,CAAC,CAAC;QACjF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAC;QAE9F,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC;YACf,WAAW,EAAE,KAAK;YAClB,0BAA0B,EAAE,IAAI;YAChC,SAAS,EAAE,WAAW;YACtB,MAAM,EAAE,QAAQ;SACjB,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACrD,cAAc,EACd,SAAS,EACT,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAC3B,KAAK,CACN,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,mDAAmD,CAAC;iBACnF;aACF,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,MAAM,sBAAsB,GAAG;YAC7B;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE;oBACJ,QAAQ,EAAE,0BAA0B;oBACpC,UAAU,EAAE,qCAAqC;iBAClD;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,sBAAsB,CAAC,CAAC;QAElF,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;QAC9C,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAC7C,MAAM,eAAe,GAAG;gBACtB;oBACE,IAAI,EAAE,gBAAgB;oBACtB,IAAI,EAAE;wBACJ,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC;wBAC/E,YAAY,EAAE,EAAE;wBAChB,WAAW,EAAE,cAAc;qBAC5B;iBACF;aACF,CAAC;YACF,MAAM,aAAa,GAAG;gBACpB;oBACE,IAAI,EAAE,gBAAgB;oBACtB,IAAI,EAAE;wBACJ,WAAW,EAAE;4BACX,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,eAAe,EAAE;yBAChG;wBACD,YAAY,EAAE,EAAE;wBAChB,WAAW,EAAE,cAAc;qBAC5B;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC;iBAChC,qBAAqB,CAAC,eAAe,CAAC;iBACtC,qBAAqB,CAAC,aAAa,CAAC;iBACpC,qBAAqB,CAAC;gBACrB;oBACE,IAAI,EAAE,YAAY;oBAClB,IAAI,EAAE;wBACJ,QAAQ,EAAE,0BAA0B;wBACpC,UAAU,EAAE,qCAAqC;qBAClD;iBACF;aACF,CAAC,CAAC;YAEL,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB;oBACE,IAAI,EAAE;wBACJ,QAAQ,EAAE,0BAA0B;wBACpC,UAAU,EAAE,qCAAqC;qBAClD;oBACD,IAAI,EAAE,YAAY;iBACnB;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAC7C,MAAM,eAAe,GAAG;gBACtB;oBACE,IAAI,EAAE,YAAY;oBAClB,IAAI,EAAE,EAAE,QAAQ,EAAE,0BAA0B,EAAE,UAAU,EAAE,yBAAyB,EAAE;iBACtF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC;iBAChC,qBAAqB,CAAC;gBACrB;oBACE,IAAI,EAAE,gBAAgB;oBACtB,IAAI,EAAE;wBACJ,WAAW,EAAE;4BACX,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,EAAE,eAAe,EAAE;yBAC/F;wBACD,YAAY,EAAE,EAAE;wBAChB,WAAW,EAAE,cAAc;qBAC5B;iBACF;aACF,CAAC;iBACD,qBAAqB,CAAC,eAAe,CAAC;iBACtC,qBAAqB,CAAC,eAAe,CAAC,CAAC,CAAC,0CAA0C;YAErF,MAAM,MAAM,CACV,aAAa,CACX,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YAErC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAC7C,MAAM,cAAc,GAAG;gBACrB;oBACE,IAAI,EAAE,YAAY;oBAClB,IAAI,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,wBAAwB,EAAE;iBACrE;gBACD;oBACE,IAAI,EAAE,YAAY;oBAClB,IAAI,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,wBAAwB,EAAE;iBACrE;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CACnG,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/cancel-code-generation.d.ts

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+export declare function handleCancelCodeGeneration({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult>;

package/dist/src/prompt/steps/step-ask-question/handlers/cancel-code-generation.js

@@ -0,0 +1,16 @@
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleCancelCodeGeneration({ askQuestionCall }) {
+    putSystemMessage('Assistant requested to stop code generation. Exiting...');
+    return {
+        breakLoop: true,
+        stepResult: StepResult.BREAK,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },
+                user: { type: 'user', text: 'Code generation cancelled.' },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=cancel-code-generation.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/cancel-code-generation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cancel-code-generation.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/cancel-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAE,eAAe,EAAsB;IACtF,gBAAgB,CAAC,yDAAyD,CAAC,CAAC;IAC5E,OAAO;QACL,SAAS,EAAE,IAAI;QACf,UAAU,EAAE,UAAU,CAAC,KAAK;QAC5B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9F,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,4BAA4B,EAAE;aAC3D;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/confirm-code-generation.d.ts

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+export declare function handleConfirmCodeGeneration({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult>;