genaicode 0.0.37 → 0.1.1

package/dist/prompt/prompt-codegen.test.js

@@ -0,0 +1,94 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { getCodeGenPrompt, getLintFixPrompt } from './prompt-codegen.js';
+import * as findFiles from '../files/find-files.js';
+import * as cliParams from '../cli/cli-params.js';
+import * as limits from './limits.js';
+import '../files/read-files.js';
+vi.mock('../files/find-files.js', () => ({
+    rcConfig: {},
+    getSourceFiles: vi.fn(),
+}));
+vi.mock('../files/read-files.js', () => ({
+    getSourceCode: () => ({}),
+}));
+vi.mock('../cli/cli-params.js', () => ({
+    requireExplanations: false,
+    considerAllFiles: false,
+    dependencyTree: false,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    imagen: false,
+    vision: false,
+}));
+vi.mock('./limits.js');
+describe('getCodeGenPrompt', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+    });
+    it('should generate prompt for all files when considerAllFiles is true', () => {
+        vi.spyOn(findFiles, 'getSourceFiles').mockReturnValue(['file1.js', 'file2.js']);
+        vi.mocked(cliParams).considerAllFiles = true;
+        vi.mocked(cliParams).explicitPrompt = undefined;
+        vi.mocked(cliParams).allowFileCreate = false;
+        vi.mocked(cliParams).allowFileDelete = false;
+        vi.mocked(cliParams).allowDirectoryCreate = false;
+        vi.mocked(cliParams).allowFileMove = false;
+        vi.spyOn(limits, 'verifyCodegenPromptLimit').mockImplementation(() => { });
+        const prompt = getCodeGenPrompt();
+        expect(prompt).toContain('You are allowed to modify all files in the application');
+        expect(prompt).toContain('Do not create new files.');
+        expect(prompt).toContain('Do not delete files.');
+        expect(prompt).toContain('Do not create new directories.');
+        expect(prompt).toContain('Do not move files.');
+    });
+    // Add more test cases for getCodeGenPrompt as needed
+});
+describe('getLintFixPrompt', () => {
+    beforeEach(() => {
+        vi.mocked(cliParams).verbosePrompt = false;
+        vi.spyOn(limits, 'verifyCodegenPromptLimit').mockImplementation(() => { });
+    });
+    it('should generate a lint fix prompt with provided command, stdout, and stderr', () => {
+        const command = 'eslint --fix';
+        const stdout = 'Fixed 2 errors';
+        const stderr = '';
+        const prompt = getLintFixPrompt(command, stdout, stderr);
+        expect(prompt).toContain('The following lint errors were encountered after the initial code generation:');
+        expect(prompt).toContain(`Lint command: ${command}`);
+        expect(prompt).toContain('Lint command stdout:');
+        expect(prompt).toContain(stdout);
+        expect(prompt).toContain('Lint command stderr:');
+        expect(prompt).toContain('Please suggest changes to fix these lint errors.');
+    });
+    it('should include stderr in the prompt when provided', () => {
+        const command = 'eslint --fix';
+        const stdout = '';
+        const stderr = 'Error: Unable to resolve path';
+        const prompt = getLintFixPrompt(command, stdout, stderr);
+        expect(prompt).toContain('Lint command stderr:');
+        expect(prompt).toContain(stderr);
+    });
+    it('should call verifyCodegenPromptLimit with the generated prompt', () => {
+        const command = 'eslint --fix';
+        const stdout = 'Fixed 1 error';
+        const stderr = '';
+        getLintFixPrompt(command, stdout, stderr);
+        expect(limits.verifyCodegenPromptLimit).toHaveBeenCalledWith(expect.any(String));
+    });
+    it('should log the prompt when verbosePrompt is true', () => {
+        vi.mocked(cliParams).verbosePrompt = true;
+        const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => { });
+        const command = 'eslint --fix';
+        const stdout = 'Fixed 1 error';
+        const stderr = '';
+        getLintFixPrompt(command, stdout, stderr);
+        expect(consoleSpy).toHaveBeenCalledWith('Lint fix prompt:');
+        expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining('The following lint errors were encountered'));
+        consoleSpy.mockRestore();
+    });
+});
+//# sourceMappingURL=prompt-codegen.test.js.map

package/dist/prompt/prompt-codegen.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-codegen.test.js","sourceRoot":"","sources":["../../src/prompt/prompt-codegen.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,KAAK,SAAS,MAAM,wBAAwB,CAAC;AACpD,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,wBAAwB,CAAC;AAEhC,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,QAAQ,EAAE,EAAE;IACZ,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;CACxB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;CAC1B,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,mBAAmB,EAAE,KAAK;IAC1B,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,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;CACd,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AAEvB,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC5E,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC;QAChF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,GAAG,SAAS,CAAC;QAChD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,eAAe,GAAG,KAAK,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,eAAe,GAAG,KAAK,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,0BAA0B,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAE1E,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAElC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wDAAwD,CAAC,CAAC;QACnF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,qDAAqD;AACvD,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,0BAA0B,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;QACrF,MAAM,OAAO,GAAG,cAAc,CAAC;QAC/B,MAAM,MAAM,GAAG,gBAAgB,CAAC;QAChC,MAAM,MAAM,GAAG,EAAE,CAAC;QAElB,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,+EAA+E,CAAC,CAAC;QAC1G,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,OAAO,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kDAAkD,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,OAAO,GAAG,cAAc,CAAC;QAC/B,MAAM,MAAM,GAAG,EAAE,CAAC;QAClB,MAAM,MAAM,GAAG,+BAA+B,CAAC;QAE/C,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,OAAO,GAAG,cAAc,CAAC;QAC/B,MAAM,MAAM,GAAG,eAAe,CAAC;QAC/B,MAAM,MAAM,GAAG,EAAE,CAAC;QAElB,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1C,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAG,cAAc,CAAC;QAC/B,MAAM,MAAM,GAAG,eAAe,CAAC;QAC/B,MAAM,MAAM,GAAG,EAAE,CAAC;QAElB,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,4CAA4C,CAAC,CAAC,CAAC;QAE/G,UAAU,CAAC,WAAW,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/prompt/prompt-consts.d.ts

@@ -0,0 +1 @@
+export declare const CODEGEN_TRIGGER: string;

package/{src → dist}/prompt/prompt-consts.js

@@ -1 +1,2 @@
 export const CODEGEN_TRIGGER = '@' + 'CODEGEN';
+//# sourceMappingURL=prompt-consts.js.map

package/dist/prompt/prompt-consts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-consts.js","sourceRoot":"","sources":["../../src/prompt/prompt-consts.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,eAAe,GAAW,GAAG,GAAG,SAAS,CAAC"}

package/dist/prompt/prompt-service-ask-question.test.d.ts

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

package/dist/prompt/prompt-service-ask-question.test.js

@@ -0,0 +1,146 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { promptService } from './prompt-service.js';
+import * as vertexAi from '../ai-service/vertex-ai.js';
+import * as readline from 'readline';
+import '../cli/cli-params.js';
+import '../files/read-files.js';
+import '../files/find-files.js';
+vi.mock('../ai-service/vertex-ai.js', () => ({ generateContent: vi.fn() }));
+vi.mock('readline', () => ({
+    createInterface: vi.fn(),
+}));
+vi.mock('../cli/cli-params.js', () => ({
+    requireExplanations: false,
+    considerAllFiles: false,
+    dependencyTree: false,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    disableContextOptimization: false,
+    vision: false,
+    imagen: false,
+    temperature: 0.7,
+    cheap: false,
+    askQuestion: true,
+}));
+// 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: '.',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+}));
+describe('promptService with askQuestion', () => {
+    let mockReadline;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        console.log = vi.fn();
+        mockReadline = {
+            question: vi.fn(),
+            close: vi.fn(),
+        };
+        // @ts-expect-error (avoid complex mock)
+        vi.spyOn(readline, 'createInterface').mockReturnValue(mockReadline);
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should handle askQuestion when enabled', async () => {
+        const mockAskQuestionCall = [
+            {
+                name: 'askQuestion',
+                args: {
+                    content: 'Do you want to proceed with code generation?',
+                    shouldPrompt: true,
+                    promptNecessity: 80,
+                    stopCodegen: false,
+                },
+            },
+        ];
+        const mockAskQuestionCall2 = [
+            {
+                name: 'askQuestion',
+                args: {
+                    content: 'Ok lets go',
+                    shouldPrompt: false,
+                    promptNecessity: 80,
+                    stopCodegen: false,
+                },
+            },
+        ];
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [],
+                    contextPaths: [],
+                    explanation: 'No updates needed',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent)
+            .mockResolvedValueOnce(mockAskQuestionCall)
+            .mockResolvedValueOnce(mockAskQuestionCall2)
+            .mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(mockReadline.question).mockImplementationOnce((_, callback) => {
+            callback('Yes');
+        });
+        await promptService(vertexAi.generateContent, undefined);
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+        expect(mockReadline.question).toHaveBeenCalledWith('Your answer: ', expect.any(Function));
+        expect(console.log).toHaveBeenCalledWith('Assistant asks:', expect.any(Object));
+        expect(console.log).toHaveBeenCalledWith('The question was answered');
+    });
+    it('should stop code generation when askQuestion returns stopCodegen: true', async () => {
+        const mockAskQuestionCall = [
+            {
+                name: 'askQuestion',
+                args: {
+                    content: 'Stopping code generation as requested.',
+                    shouldPrompt: false,
+                    promptNecessity: 100,
+                    stopCodegen: true,
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockAskQuestionCall);
+        const result = await promptService(vertexAi.generateContent, undefined);
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(1);
+        expect(result).toEqual([]);
+        expect(console.log).toHaveBeenCalledWith('Assistant requested to stop code generation. Exiting...');
+    });
+    it('should proceed with code generation when askQuestion returns shouldPrompt: false', async () => {
+        const mockAskQuestionCall = [];
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [],
+                    contextPaths: [],
+                    explanation: 'No updates needed',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent)
+            .mockResolvedValueOnce(mockAskQuestionCall)
+            .mockResolvedValueOnce(mockCodegenSummary);
+        await promptService(vertexAi.generateContent, undefined);
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(2);
+        expect(mockReadline.question).not.toHaveBeenCalled();
+        expect(console.log).toHaveBeenCalledWith('Assistant did not ask a question. Proceeding with code generation.');
+    });
+});
+//# sourceMappingURL=prompt-service-ask-question.test.js.map

package/dist/prompt/prompt-service-ask-question.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-service-ask-question.test.js","sourceRoot":"","sources":["../../src/prompt/prompt-service-ask-question.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AAEvD,OAAO,KAAK,QAAQ,MAAM,UAAU,CAAC;AACrC,OAAO,sBAAsB,CAAC;AAC9B,OAAO,wBAAwB,CAAC;AAChC,OAAO,wBAAwB,CAAC;AAEhC,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,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IACzB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;CACzB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,mBAAmB,EAAE,KAAK;IAC1B,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,KAAK;IACjC,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,WAAW,EAAE,IAAI;CAClB,CAAC,CAAC,CAAC;AAEJ,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,GAAG;QACZ,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,IAAI,YAAoG,CAAC;IAEzG,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEtB,YAAY,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf,CAAC;QACF,wCAAwC;QACxC,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,mBAAmB,GAAG;YAC1B;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,OAAO,EAAE,8CAA8C;oBACvD,YAAY,EAAE,IAAI;oBAClB,eAAe,EAAE,EAAE;oBACnB,WAAW,EAAE,KAAK;iBACnB;aACF;SACF,CAAC;QACF,MAAM,oBAAoB,GAAG;YAC3B;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,OAAO,EAAE,YAAY;oBACrB,YAAY,EAAE,KAAK;oBACnB,eAAe,EAAE,EAAE;oBACnB,WAAW,EAAE,KAAK;iBACnB;aACF;SACF,CAAC;QACF,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,EAAE;oBACf,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,mBAAmB;iBACjC;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC;aAChC,qBAAqB,CAAC,mBAAmB,CAAC;aAC1C,qBAAqB,CAAC,oBAAoB,CAAC;aAC3C,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAE7C,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAU,EAAE,QAA+B,EAAE,EAAE;YACtG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,CAAC,QAAQ,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;QAEzD,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1F,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAChF,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,mBAAmB,GAAG;YAC1B;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,OAAO,EAAE,wCAAwC;oBACjD,YAAY,EAAE,KAAK;oBACnB,eAAe,EAAE,GAAG;oBACpB,WAAW,EAAE,IAAI;iBAClB;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;QAE/E,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;QAExE,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,yDAAyD,CAAC,CAAC;IACtG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,mBAAmB,GAAmB,EAAE,CAAC;QAC/C,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,EAAE;oBACf,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,mBAAmB;iBACjC;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC;aAChC,qBAAqB,CAAC,mBAAmB,CAAC;aAC1C,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAE7C,MAAM,aAAa,CAAC,QAAQ,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;QAEzD,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,oEAAoE,CAAC,CAAC;IACjH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/prompt/prompt-service.d.ts

@@ -0,0 +1,13 @@
+import { PromptItem, FunctionDef, FunctionCall } from '../ai-service/common.js';
+interface GenerateContentFunction {
+    (prompt: PromptItem[], functionDefs: FunctionDef[], requiredFunctionName: string, temperature: number, cheap: boolean): Promise<FunctionCall[]>;
+}
+interface GenerateImageFunction {
+    (prompt: string, contextImagePath: string | undefined, size: {
+        width: number;
+        height: number;
+    }, cheap: boolean): Promise<string>;
+}
+/** A function that communicates with model using */
+export declare function promptService(generateContentFn: GenerateContentFunction, generateImageFn: GenerateImageFunction | undefined, codegenPrompt?: string): Promise<FunctionCall[]>;
+export {};

package/dist/prompt/prompt-service.js

@@ -0,0 +1,278 @@
+import assert from 'node:assert';
+import fs from 'fs';
+import * as diff from 'diff';
+import mime from 'mime-types';
+import { createInterface } from 'readline';
+import { getSystemPrompt } from './systemprompt.js';
+import { getCodeGenPrompt } from './prompt-codegen.js';
+import { functionDefs } from './function-calling.js';
+import { getSourceCode, getImageAssets } from '../files/read-files.js';
+import { disableContextOptimization, temperature, vision, cheap, askQuestion } from '../cli/cli-params.js';
+import { validateFunctionCall } from './function-calling-validate.js';
+/** A function that communicates with model using */
+export async function promptService(generateContentFn, generateImageFn, codegenPrompt = getCodeGenPrompt()) {
+    const messages = prepareMessages(codegenPrompt);
+    // First stage: generate code generation summary, which should not take a lot of output tokens
+    const getSourceCodeRequest = { name: 'getSourceCode' };
+    const prompt = [
+        { type: 'systemPrompt', systemPrompt: getSystemPrompt() },
+        { type: 'user', text: messages.suggestSourceCode },
+        { type: 'assistant', text: messages.requestSourceCode, functionCalls: [getSourceCodeRequest] },
+    ];
+    const getSourceCodeResponse = {
+        type: 'user',
+        functionResponses: [{ name: 'getSourceCode', content: messages.sourceCode }],
+        cache: true,
+    };
+    prompt.push(getSourceCodeResponse);
+    if (vision) {
+        prompt.slice(-1)[0].text = messages.suggestImageAssets;
+        prompt.push({ type: 'assistant', text: messages.requestImageAssets, functionCalls: [{ name: 'getImageAssets' }] }, { type: 'user', functionResponses: [{ name: 'getImageAssets', content: messages.imageAssets }] });
+    }
+    prompt.slice(-1)[0].text = messages.prompt;
+    // New step: Allow the assistant to ask a question if --ask-question is enabled
+    if (askQuestion) {
+        console.log('Allowing the assistant to ask a question...');
+        const questionAsked = false;
+        while (!questionAsked) {
+            const askQuestionResult = await generateContentFn(prompt, functionDefs, 'askQuestion', temperature, cheap);
+            const askQuestionCall = askQuestionResult.find((call) => call.name === 'askQuestion');
+            if (askQuestionCall) {
+                console.log('Assistant asks:', askQuestionCall.args);
+                const userAnswer = askQuestionCall.args?.shouldPrompt
+                    ? await getUserInput('Your answer: ')
+                    : 'Lets proceed with code generation.';
+                prompt.push({ type: 'assistant', functionCalls: [askQuestionCall] }, {
+                    type: 'user',
+                    text: userAnswer,
+                    functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id }],
+                });
+                console.log('The question was answered');
+                if (askQuestionCall.args?.stopCodegen) {
+                    console.log('Assistant requested to stop code generation. Exiting...');
+                    return [];
+                }
+                if (!askQuestionCall.args?.shouldPrompt) {
+                    console.log('Proceeding with code generation.');
+                    break;
+                }
+            }
+            else {
+                console.log('Assistant did not ask a question. Proceeding with code generation.');
+                break;
+            }
+        }
+    }
+    const baseRequest = [
+        prompt,
+        functionDefs,
+        'codegenSummary',
+        temperature,
+        cheap,
+    ];
+    let baseResult = await generateContentFn(...baseRequest);
+    let codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');
+    if (codegenSummaryRequest) {
+        // Second stage: for each file request the actual code updates
+        console.log('Received codegen summary, will collect partial updates', codegenSummaryRequest.args);
+        baseResult = await validateAndRecoverSingleResult(baseRequest, baseResult, messages, generateContentFn);
+        codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');
+        // Sometimes the result happens to be a string
+        assert(Array.isArray(codegenSummaryRequest?.args?.fileUpdates), 'fileUpdates is not an array');
+        assert(Array.isArray(codegenSummaryRequest?.args.contextPaths), 'contextPaths is not an array');
+        if (codegenSummaryRequest?.args.contextPaths.length > 0 && !disableContextOptimization) {
+            console.log('Optimize with context paths.');
+            // Monkey patch the initial getSourceCode, do not send parts of source code that are consider irrelevant
+            getSourceCodeRequest.args = {
+                filePaths: [
+                    ...codegenSummaryRequest.args.fileUpdates.map((file) => file.path),
+                    ...codegenSummaryRequest.args.contextPaths,
+                ],
+            };
+            getSourceCodeResponse.functionResponses.find((item) => item.name === 'getSourceCode').content =
+                messages.contextSourceCode(getSourceCodeRequest.args?.filePaths);
+        }
+        // Store the first stage response entirely in conversation history
+        prompt.push({ type: 'assistant', functionCalls: baseResult });
+        prompt.push({
+            type: 'user',
+            functionResponses: baseResult.map((call) => ({ name: call.name, call_id: call.id })),
+            cache: true,
+        });
+        const result = [];
+        for (const file of codegenSummaryRequest.args.fileUpdates) {
+            console.log('Collecting partial update for: ' + file.path + ' using tool: ' + file.updateToolName);
+            console.log('- Prompt:', file.prompt);
+            console.log('- Temperature', file.temperature);
+            console.log('- Cheap', file.cheap);
+            if (vision) {
+                console.log('- Context image assets', file.contextImageAssets);
+            }
+            // this is needed, otherwise we will get an error
+            if (prompt.slice(-1)[0].type === 'user') {
+                prompt.slice(-1)[0].text = file.prompt ?? messages.partialPromptTemplate(file.path);
+            }
+            else {
+                prompt.push({ type: 'user', text: file.prompt ?? messages.partialPromptTemplate(file.path) });
+            }
+            if (vision && file.contextImageAssets) {
+                prompt.slice(-1)[0].images = file.contextImageAssets.map((path) => ({
+                    path,
+                    base64url: fs.readFileSync(path, 'base64'),
+                    mediaType: mime.lookup(path) || '',
+                }));
+            }
+            let partialRequest = [
+                prompt,
+                functionDefs,
+                file.updateToolName,
+                file.temperature ?? temperature,
+                file.cheap === true,
+            ];
+            let partialResult = await generateContentFn(...partialRequest);
+            const getSourceCodeCall = partialResult.find((call) => call.name === 'getSourceCode');
+            assert(!getSourceCodeCall, 'Unexpected getSourceCode: ' + JSON.stringify(getSourceCodeCall));
+            // Handle image generation requests
+            const generateImageCall = partialResult.find((call) => call.name === 'generateImage');
+            if (generateImageCall) {
+                assert(!!generateImageFn, 'Image generation requested, but a image generation service was not provided');
+                console.log('Processing image generation request:', generateImageCall.args);
+                try {
+                    const { prompt: imagePrompt, filePath, contextImagePath, size, cheap, } = generateImageCall.args;
+                    const generatedImageUrl = await generateImageFn(imagePrompt, contextImagePath, size, cheap === true);
+                    // Add a createFile call to the result to ensure the generated image is tracked
+                    partialResult.push({
+                        name: 'downloadFile',
+                        args: {
+                            filePath: filePath,
+                            downloadUrl: generatedImageUrl,
+                            explanation: `Downloading generated image`,
+                        },
+                    });
+                }
+                catch (error) {
+                    console.error('Error generating image:', error);
+                    // Add an explanation about the failed image generation
+                    partialResult.push({
+                        name: 'explanation',
+                        args: {
+                            text: `Failed to generate image: ${error.message}`,
+                        },
+                    });
+                }
+            }
+            else {
+                partialResult = await validateAndRecoverSingleResult(partialRequest, partialResult, messages, generateContentFn);
+            }
+            // Verify if patchFile is one of the functions called, and test if patch is valid and can be applied successfully
+            const patchFileCall = partialResult.find((call) => call.name === 'patchFile');
+            if (patchFileCall) {
+                const { filePath, patch } = patchFileCall.args;
+                console.log('Verification of patch for file:', filePath);
+                let updatedContent = false;
+                const currentContent = fs.readFileSync(filePath, 'utf-8');
+                try {
+                    updatedContent = diff.applyPatch(currentContent, patch);
+                }
+                catch (e) {
+                    console.log('Error when applying patch', e);
+                }
+                if (!updatedContent) {
+                    console.log(`Patch could not be applied for ${filePath}. Retrying without patchFile function.`);
+                    // Rerun content generation without patchFile function
+                    partialRequest = [prompt, functionDefs, 'updateFile', file.temperature ?? temperature, file.cheap === true];
+                    partialResult = await generateContentFn(...partialRequest);
+                    partialResult = await validateAndRecoverSingleResult(partialRequest, partialResult, messages, generateContentFn);
+                    const getSourceCodeCall = partialResult.find((call) => call.name === 'getSourceCode');
+                    assert(!getSourceCodeCall, 'Unexpected getSourceCode: ' + JSON.stringify(getSourceCodeCall));
+                    assert(!partialResult.find((call) => call.name === 'patchFile'), 'Unexpected patchFile in retry response');
+                }
+                else {
+                    console.log('Patch verified successfully');
+                }
+            }
+            // add the code gen result to the context, as the subsequent code gen may depend on the result
+            prompt.push({ type: 'assistant', functionCalls: partialResult }, {
+                type: 'user',
+                functionResponses: partialResult.map((call) => ({ name: call.name, call_id: call.id })),
+            });
+            result.push(...partialResult);
+        }
+        return result;
+    }
+    else {
+        // This is unexpected, if happens probably means no code updates.
+        console.log('Did not receive codegen summary, returning result.');
+        return baseResult;
+    }
+}
+async function validateAndRecoverSingleResult([prompt, functionDefs, requiredFunctionName, temperature, cheap], result, messages, generateContentFn) {
+    if (result.length !== 1) {
+        return result;
+    }
+    const call = result[0];
+    const validatorError = validateFunctionCall(call);
+    if (validatorError) {
+        console.log('Invalid function call', call, validatorError);
+        prompt.push({ type: 'assistant', functionCalls: [call] }, {
+            type: 'user',
+            text: messages.invalidFunctionCall,
+            functionResponses: [
+                {
+                    name: call.name,
+                    call_id: call.id,
+                    content: JSON.stringify({ args: call.args, error: validatorError }),
+                    isError: true,
+                },
+            ],
+        });
+        console.log('Trying to recover...');
+        if (cheap) {
+            console.log('Disabling --cheap for recovery.');
+        }
+        result = await generateContentFn(prompt, functionDefs, requiredFunctionName, temperature, false);
+        console.log('Recover result:', result);
+        if (result.length === 1) {
+            const recoveryError = validateFunctionCall(result[0]);
+            assert(!recoveryError, 'Recovery failed');
+            console.log('Recovery was successful');
+        }
+        else {
+            console.log('Unexpected number of function calls', result);
+        }
+    }
+    return result;
+}
+/**
+ * Function to prepare messages for AI services
+ */
+function prepareMessages(prompt) {
+    return {
+        suggestSourceCode: 'I should provide you with application source code.',
+        requestSourceCode: 'Please provide application source code.',
+        suggestImageAssets: 'I should also provide you with a summary of application image assets',
+        requestImageAssets: 'Please provide summary of application image assets.',
+        prompt: prompt +
+            '\n Start from generating codegen summary, this summary will be used as a context to generate updates, so make sure that it contains useful information.',
+        sourceCode: JSON.stringify(getSourceCode()),
+        contextSourceCode: (paths) => JSON.stringify(getSourceCode(paths)),
+        imageAssets: JSON.stringify(getImageAssets()),
+        partialPromptTemplate(path) {
+            return `Thank you for providing the summary, now suggest changes for the \`${path}\` file using appropriate tools.`;
+        },
+        invalidFunctionCall: 'Function call was invalid, please analyze the error and respond with corrected function call.',
+    };
+}
+async function getUserInput(prompt) {
+    const rl = createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    return new Promise((resolve) => {
+        rl.question(prompt, (answer) => {
+            rl.close();
+            resolve(answer);
+        });
+    });
+}
+//# sourceMappingURL=prompt-service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-service.js","sourceRoot":"","sources":["../../src/prompt/prompt-service.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAE3C,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC3G,OAAO,EAAE,oBAAoB,EAAE,MAAM,gCAAgC,CAAC;AAsBtE,oDAAoD;AACpD,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,iBAA0C,EAC1C,eAAkD,EAClD,gBAAwB,gBAAgB,EAAE;IAE1C,MAAM,QAAQ,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;IAEhD,8FAA8F;IAC9F,MAAM,oBAAoB,GAAiB,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;IAErE,MAAM,MAAM,GAAiB;QAC3B,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,eAAe,EAAE,EAAE;QACzD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,iBAAiB,EAAE;QAClD,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,CAAC,iBAAiB,EAAE,aAAa,EAAE,CAAC,oBAAoB,CAAC,EAAE;KAC/F,CAAC;IAEF,MAAM,qBAAqB,GAAe;QACxC,IAAI,EAAE,MAAM;QACZ,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC;QAC5E,KAAK,EAAE,IAAI;KACZ,CAAC;IACF,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IAEnC,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,QAAQ,CAAC,kBAAkB,CAAC;QACvD,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,CAAC,kBAAkB,EAAE,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,EAAE,EACrG,EAAE,IAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,EAAE,CACjG,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC;IAE3C,+EAA+E;IAC/E,IAAI,WAAW,EAAE,CAAC;QAChB,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;QAC3D,MAAM,aAAa,GAAG,KAAK,CAAC;QAC5B,OAAO,CAAC,aAAa,EAAE,CAAC;YACtB,MAAM,iBAAiB,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;YAC3G,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;YACtF,IAAI,eAAe,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;gBACrD,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,EAAE,YAAY;oBACnD,CAAC,CAAC,MAAM,YAAY,CAAC,eAAe,CAAC;oBACrC,CAAC,CAAC,oCAAoC,CAAC;gBACzC,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE,EACvD;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,UAAU;oBAChB,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,EAAE,CAAC;iBAC1E,CACF,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;gBACzC,IAAI,eAAe,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC;oBACtC,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;oBACvE,OAAO,EAAE,CAAC;gBACZ,CAAC;gBACD,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,YAAY,EAAE,CAAC;oBACxC,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;oBAChD,MAAM;gBACR,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC;gBAClF,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAA2D;QAC1E,MAAM;QACN,YAAY;QACZ,gBAAgB;QAChB,WAAW;QACX,KAAK;KACN,CAAC;IACF,IAAI,UAAU,GAAG,MAAM,iBAAiB,CAAC,GAAG,WAAW,CAAC,CAAC;IAEzD,IAAI,qBAAqB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;IAEtF,IAAI,qBAAqB,EAAE,CAAC;QAC1B,8DAA8D;QAC9D,OAAO,CAAC,GAAG,CAAC,wDAAwD,EAAE,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAElG,UAAU,GAAG,MAAM,8BAA8B,CAAC,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QACxG,qBAAqB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;QAElF,8CAA8C;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,6BAA6B,CAAC,CAAC;QAC/F,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,CAAC,YAAY,CAAC,EAAE,8BAA8B,CAAC,CAAC;QAEhG,IAAI,qBAAqB,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,0BAA0B,EAAE,CAAC;YACvF,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;YAC5C,wGAAwG;YACxG,oBAAoB,CAAC,IAAI,GAAG;gBAC1B,SAAS,EAAE;oBACT,GAAG,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;oBACpF,GAAG,qBAAqB,CAAC,IAAI,CAAC,YAAY;iBAC3C;aACF,CAAC;YACF,qBAAqB,CAAC,iBAAkB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAE,CAAC,OAAO;gBAC7F,QAAQ,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,IAAI,EAAE,SAAqB,CAAC,CAAC;QACjF,CAAC;QAED,kEAAkE;QAClE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YACpF,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAmB,EAAE,CAAC;QAElC,KAAK,MAAM,IAAI,IAAI,qBAAsB,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YAC3D,OAAO,CAAC,GAAG,CAAC,iCAAiC,GAAG,IAAI,CAAC,IAAI,GAAG,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;YACnG,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACjE,CAAC;YAED,iDAAiD;YACjD,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtF,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChG,CAAC;YAED,IAAI,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACtC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC;oBAC1E,IAAI;oBACJ,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC;oBAC1C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE;iBACnC,CAAC,CAAC,CAAC;YACN,CAAC;YAED,IAAI,cAAc,GAA2D;gBAC3E,MAAM;gBACN,YAAY;gBACZ,IAAI,CAAC,cAAc;gBACnB,IAAI,CAAC,WAAW,IAAI,WAAW;gBAC/B,IAAI,CAAC,KAAK,KAAK,IAAI;aACpB,CAAC;YACF,IAAI,aAAa,GAAG,MAAM,iBAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;YAE/D,MAAM,iBAAiB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;YACtF,MAAM,CAAC,CAAC,iBAAiB,EAAE,4BAA4B,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAE7F,mCAAmC;YACnC,MAAM,iBAAiB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;YACtF,IAAI,iBAAiB,EAAE,CAAC;gBACtB,MAAM,CAAC,CAAC,CAAC,eAAe,EAAE,6EAA6E,CAAC,CAAC;gBAEzG,OAAO,CAAC,GAAG,CAAC,sCAAsC,EAAE,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBAC5E,IAAI,CAAC;oBACH,MAAM,EACJ,MAAM,EAAE,WAAW,EACnB,QAAQ,EACR,gBAAgB,EAChB,IAAI,EACJ,KAAK,GACN,GAAG,iBAAiB,CAAC,IAMrB,CAAC;oBACF,MAAM,iBAAiB,GAAG,MAAM,eAAe,CAAC,WAAW,EAAE,gBAAgB,EAAE,IAAI,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC;oBAErG,+EAA+E;oBAC/E,aAAa,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,cAAc;wBACpB,IAAI,EAAE;4BACJ,QAAQ,EAAE,QAAQ;4BAClB,WAAW,EAAE,iBAAiB;4BAC9B,WAAW,EAAE,6BAA6B;yBAC3C;qBACF,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;oBAChD,uDAAuD;oBACvD,aAAa,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,aAAa;wBACnB,IAAI,EAAE;4BACJ,IAAI,EAAE,6BAA8B,KAAe,CAAC,OAAO,EAAE;yBAC9D;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,aAAa,GAAG,MAAM,8BAA8B,CAClD,cAAc,EACd,aAAa,EACb,QAAQ,EACR,iBAAiB,CAClB,CAAC;YACJ,CAAC;YAED,iHAAiH;YACjH,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;YAC9E,IAAI,aAAa,EAAE,CAAC;gBAClB,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,aAAa,CAAC,IAA2C,CAAC;gBACtF,OAAO,CAAC,GAAG,CAAC,iCAAiC,EAAE,QAAQ,CAAC,CAAC;gBAEzD,IAAI,cAAc,GAAqB,KAAK,CAAC;gBAC7C,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAE1D,IAAI,CAAC;oBACH,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;gBAC1D,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;gBAC9C,CAAC;gBAED,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,OAAO,CAAC,GAAG,CAAC,kCAAkC,QAAQ,wCAAwC,CAAC,CAAC;oBAEhG,sDAAsD;oBACtD,cAAc,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE,YAAY,EAAE,IAAI,CAAC,WAAW,IAAI,WAAW,EAAE,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC;oBAC5G,aAAa,GAAG,MAAM,iBAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;oBAE3D,aAAa,GAAG,MAAM,8BAA8B,CAClD,cAAc,EACd,aAAa,EACb,QAAQ,EACR,iBAAiB,CAClB,CAAC;oBAEF,MAAM,iBAAiB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;oBACtF,MAAM,CAAC,CAAC,iBAAiB,EAAE,4BAA4B,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,CAAC;oBAC7F,MAAM,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,EAAE,wCAAwC,CAAC,CAAC;gBAC7G,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;gBAC7C,CAAC;YACH,CAAC;YAED,8FAA8F;YAC9F,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,EAAE,EACnD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;aACxF,CACF,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;SAAM,CAAC;QACN,iEAAiE;QACjE,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;QAClE,OAAO,UAAU,CAAC;IACpB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,8BAA8B,CAC3C,CAAC,MAAM,EAAE,YAAY,EAAE,oBAAoB,EAAE,WAAW,EAAE,KAAK,CAM9D,EACD,MAAsB,EACtB,QAA4C,EAC5C,iBAA0C;IAE1C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;IACvB,MAAM,cAAc,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;IAClD,IAAI,cAAc,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;QAE3D,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC,IAAI,CAAC,EAAE,EAC5C;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ,CAAC,mBAAmB;YAClC,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,OAAO,EAAE,IAAI,CAAC,EAAE;oBAChB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;oBACnE,OAAO,EAAE,IAAI;iBACd;aACF;SACF,CACF,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QACpC,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,YAAY,EAAE,oBAAoB,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;QACjG,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAEvC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,MAAM,CAAC,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QACzC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,qCAAqC,EAAE,MAAM,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,MAAc;IACrC,OAAO;QACL,iBAAiB,EAAE,oDAAoD;QACvE,iBAAiB,EAAE,yCAAyC;QAC5D,kBAAkB,EAAE,sEAAsE;QAC1F,kBAAkB,EAAE,qDAAqD;QACzE,MAAM,EACJ,MAAM;YACN,yJAAyJ;QAC3J,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC;QAC3C,iBAAiB,EAAE,CAAC,KAAe,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAC5E,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,CAAC;QAC7C,qBAAqB,CAAC,IAAY;YAChC,OAAO,sEAAsE,IAAI,kCAAkC,CAAC;QACtH,CAAC;QACD,mBAAmB,EACjB,+FAA+F;KAClG,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,MAAc;IACxC,MAAM,EAAE,GAAG,eAAe,CAAC;QACzB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;IAEH,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,EAAE;YAC7B,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,MAAM,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/prompt/prompt-service.test.d.ts

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