genaicode 0.5.1 → 0.6.1

package/dist/src/main/codegen.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen.js","sourceRoot":"","sources":["../../../src/main/codegen.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,EAAE,eAAe,IAAI,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACxF,OAAO,EAAE,eAAe,IAAI,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAClF,OAAO,EAAE,eAAe,IAAI,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACzF,OAAO,EAAE,eAAe,IAAI,6BAA6B,EAAE,MAAM,mCAAmC,CAAC;AACrG,OAAO,EAAE,eAAe,IAAI,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACxF,OAAO,EAAE,aAAa,IAAI,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC9E,OAAO,EAAE,aAAa,IAAI,qBAAqB,EAAE,MAAM,mCAAmC,CAAC;AAE3F,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAE/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EACL,gBAAgB,EAChB,cAAc,EACd,qBAAqB,EACrB,uBAAuB,GACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAEzD,uBAAuB;AACvB,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,KAAK,GAAG,KAAK;IAC5C,2CAA2C;IAC3C,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEvE,iBAAiB,EAAE,CAAC;IAEpB,uBAAuB;IACvB,IAAI,SAAS,CAAC,aAAa,EAAE,CAAC;QAC5B,gBAAgB,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAmB;QAC9B,cAAc,EAAE,SAAS,CAAC,cAAc;QACxC,QAAQ,EAAE,SAAS,CAAC,QAAQ;QAC5B,gBAAgB,EAAE,SAAS,CAAC,gBAAgB;QAC5C,cAAc,EAAE,SAAS,CAAC,cAAc;QAExC,eAAe,EAAE,SAAS,CAAC,eAAe;QAC1C,eAAe,EAAE,SAAS,CAAC,eAAe;QAC1C,oBAAoB,EAAE,SAAS,CAAC,oBAAoB;QACpD,aAAa,EAAE,SAAS,CAAC,aAAa;QAEtC,SAAS,EAAE,mBAAmB,EAAE;QAChC,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,MAAM,EAAE,SAAS,CAAC,MAAM;QAExB,0BAA0B,EAAE,SAAS,CAAC,0BAA0B;QAChE,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,KAAK,EAAE,SAAS,CAAC,KAAK;QACtB,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,OAAO,EAAE,SAAS,CAAC,aAAa;QAChC,mBAAmB,EAAE,CAAC,SAAS,CAAC,mBAAmB;QACnD,eAAe,EAAE,SAAS,CAAC,eAAe;QAC1C,kBAAkB,EAAE,SAAS,CAAC,kBAAkB;QAChD,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,cAAc,EAAE,SAAS,CAAC,cAAc;QACxC,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,YAAY,EAAE,SAAS,CAAC,YAAY;QACpC,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,EAAE,EAAE,SAAS,CAAC,EAAE;QAChB,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,wBAAwB,EAAE,SAAS,CAAC,wBAAwB;QAC5D,cAAc,EAAE,CAAC,SAAS,CAAC,cAAc;QACzC,qBAAqB,EAAE,CAAC,SAAS,CAAC,qBAAqB;QACvD,0BAA0B,EAAE,CAAC,SAAS,CAAC,0BAA0B;QAEjE,KAAK;KACN,CAAC;IAEF,IAAI,SAAS,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;SAAM,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC;QACjC,0BAA0B;QAC1B,MAAM,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;QACzD,MAAM,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,OAAuB,EACvB,WAAyB,EACzB,eAAoC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE;IAE3D,YAAY,EAAE,CAAC;IAEf,qBAAqB,EAAE,CAAC;IAExB,cAAc,CAAC,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,QAAQ,IAAI,gDAAgD,CAAC,CAAC;IAE/G,IAAI,QAAQ,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QACxD,IAAI,CAAC;YACH,gBAAgB,CAAC,2BAA2B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;YACpE,MAAM,WAAW,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,GAAG,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;YACnE,gBAAgB,CAAC,oCAAoC,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAA2C,CAAC;YACvE,gBAAgB,CACd,qHAAqH,CACtH,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YAC5C,uBAAuB,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;IACH,CAAC;SAAM,IAAI,QAAQ,CAAC,WAAW,IAAI,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;QACzB,uBAAuB,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,YAAY,EAAE,CAAC;IAErB,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;IACxC,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,aAAa,CACvC,2BAA2B,EAAE,EAC7B,kBAAkB,EAClB,gBAAgB,CAAC,OAAO,CAAC,EACzB,YAAY,CACb,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,0BAA0B,EAAE,aAAa,CAAC,CAAC;QAEvD,MAAM,YAAY,EAAE,CAAC;QAErB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;QAC1C,CAAC;aAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YAC1B,gBAAgB,CAAC,kCAAkC,CAAC,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,gBAAgB,CAAC,cAAc,CAAC,CAAC;YACjC,MAAM,WAAW,CACf,aAAa,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,EAC5F,OAAO,CACR,CAAC;YACF,gBAAgB,CAAC,yBAAyB,CAAC,CAAC;YAE5C,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACrE,CAAC;YAED,MAAM,YAAY,EAAE,CAAC;YAErB,oDAAoD;YACpD,IAAI,QAAQ,CAAC,WAAW,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrD,IAAI,CAAC;oBACH,gBAAgB,CAAC,2BAA2B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;oBACpE,MAAM,WAAW,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,GAAG,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;oBACnE,gBAAgB,CAAC,oCAAoC,CAAC,CAAC;gBACzD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,gBAAgB,CAAC,kDAAkD,CAAC,CAAC;oBAErE,oDAAoD;oBACpD,MAAM,cAAc,GAAG,KAA2C,CAAC;oBACnE,MAAM,eAAe,GAAG,gBAAgB,CACtC,QAAQ,CAAC,WAAW,EACpB,OAAO,EACP,cAAc,CAAC,MAAM,EACrB,cAAc,CAAC,MAAM,CACtB,CAAC;oBAEF,gBAAgB,CAAC,oCAAoC,CAAC,CAAC;oBACvD,MAAM,oBAAoB,GAAG,CAAC,MAAM,aAAa,CAC/C,2BAA2B,EAAE,EAC7B,kBAAkB,EAClB;wBACE,MAAM,EAAE,eAAe;wBACvB,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,gBAAgB,EAAE,IAAI,EAAE;qBAChD,EACD,YAAY,CACb,CAAmB,CAAC;oBAErB,OAAO,CAAC,GAAG,CAAC,yCAAyC,EAAE,oBAAoB,CAAC,CAAC;oBAE7E,MAAM,YAAY,EAAE,CAAC;oBAErB,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;oBACxC,WAAW,CACT,oBAAoB,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,EACnG,OAAO,CACR,CAAC;oBAEF,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;wBACzB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;oBAChE,CAAC;oBAED,yCAAyC;oBACzC,IAAI,CAAC;wBACH,gBAAgB,CAAC,4BAA4B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;wBACrE,MAAM,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;wBACxC,gBAAgB,CAAC,gDAAgD,CAAC,CAAC;oBACrE,CAAC;oBAAC,OAAO,eAAe,EAAE,CAAC;wBACzB,MAAM,KAAK,GAAG,eAAqD,CAAC;wBACpE,gBAAgB,CAAC,8EAA8E,CAAC,CAAC;wBACjG,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;oBAC1D,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzE,gBAAgB,CAAC,mCAAmC,CAAC,CAAC;YACxD,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBACzD,gBAAgB,CACd,+FAA+F,CAChG,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,gBAAgB,CAAC,qCAAqC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;aAAM,CAAC;YACN,gBAAgB,CAAC,0CAA0C,CAAC,CAAC;QAC/D,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;IACzC,CAAC;YAAS,CAAC;QACT,uBAAuB,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC;AAED,8BAA8B;AAE9B,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;AAEzC,SAAS,2BAA2B;IAClC,OAAO;QACL,kBAAkB,EAAE,6BAA6B;QACjD,WAAW,EAAE,uBAAuB;QACpC,WAAW,EAAE,uBAAuB;QACpC,SAAS,EAAE,wBAAwB;QACnC,UAAU,EAAE,kBAAkB;QAC9B,GAAG,MAAM,CAAC,WAAW,CAAC,uBAAuB,EAAE,CAAC,OAAO,EAAE,CAAC;KAC3D,CAAC;AACJ,CAAC;AAED,MAAM,kBAAkB,GAA8C;IACpE,QAAQ,EAAE,kBAAkB;IAC5B,WAAW,EAAE,qBAAqB;CAC1B,CAAC"}

package/dist/src/main/codegen.test.d.ts

@@ -0,0 +1,3 @@
+import '../files/find-files.js';
+import '../files/cache-file.js';
+import './config.js';

package/dist/src/main/codegen.test.js

@@ -0,0 +1,288 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { runCodegen } from './codegen.js';
+import * as vertexAi from '../ai-service/vertex-ai.js';
+import * as chatGpt from '../ai-service/chat-gpt.js';
+import * as anthropic from '../ai-service/anthropic.js';
+import * as vertexAiClaude from '../ai-service/vertex-ai-claude.js';
+import * as updateFiles from '../files/update-files.js';
+import '../files/find-files.js';
+import '../files/cache-file.js';
+import * as cliParams from '../cli/cli-params.js';
+import * as cliOptions from '../cli/cli-options.js';
+import * as vertexAiImagen from '../ai-service/vertex-ai-imagen.js';
+import * as dallE from '../ai-service/dall-e.js';
+import './config.js';
+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('../ai-service/vertex-ai-claude.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../files/update-files.js');
+vi.mock('../files/cache-file.js');
+vi.mock('../cli/cli-params.js', () => ({
+    interactive: false,
+    ui: false,
+    uiPort: 1337,
+    disableExplanations: true,
+    considerAllFiles: false,
+    dependencyTree: false,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    vertexAiClaude: false,
+    helpRequested: false,
+    aiStudio: false,
+    vision: false,
+    imagen: false,
+    temperature: 0.7,
+    cheap: false,
+    taskFile: undefined,
+    disableInitialLint: undefined,
+    askQuestion: false,
+    disableContextOptimization: true,
+    geminiBlockNone: undefined,
+    contentMask: undefined,
+    ignorePatterns: [],
+    disableCache: undefined,
+    disableAiServiceFallback: undefined,
+    disableHistory: true,
+    disableSelfReflection: true,
+    disableConversationSummary: true,
+}));
+vi.mock('../files/find-files.js', () => ({
+    getSourceFiles: () => [],
+    getImageAssetFiles: () => [],
+    refreshFiles: () => null,
+}));
+vi.mock('../cli/cli-options.js', () => ({
+    printHelpMessage: vi.fn(),
+}));
+vi.mock('../ai-service/vertex-ai-imagen.js', () => ({ generateImage: vi.fn() }));
+vi.mock('../ai-service/dall-e.js', () => ({ generateImage: vi.fn() }));
+vi.mock('./config.js', () => ({
+    rootDir: '/mocked/root/dir',
+    rcConfig: {
+        rootDir: '/mocked/root/dir',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+    importantContext: {},
+}));
+describe('runCodegen', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+        vi.mocked(cliParams).aiService = undefined;
+        vi.mocked(cliParams).dryRun = false;
+        vi.mocked(cliParams).helpRequested = false;
+        vi.mocked(cliParams).vision = false;
+        vi.mocked(cliParams).imagen = undefined;
+    });
+    it('should run codegen with Vertex AI by default', 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);
+        await runCodegen();
+        expect(vertexAi.generateContent).toHaveBeenCalled();
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    it('should run codegen with ChatGPT when chatGpt flag is true', 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);
+        await runCodegen();
+        expect(chatGpt.generateContent).toHaveBeenCalled();
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    it('should run codegen with Anthropic when anthropic flag is true', async () => {
+        vi.mocked(cliParams).aiService = 'anthropic';
+        const mockFunctionCalls = [{ name: 'deleteFile', args: { filePath: 'obsolete.js' } }];
+        vi.mocked(anthropic).generateContent.mockResolvedValueOnce(mockFunctionCalls);
+        await runCodegen();
+        expect(anthropic.generateContent).toHaveBeenCalled();
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    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);
+        await runCodegen();
+        expect(vertexAi.generateContent).toHaveBeenCalled();
+        expect(updateFiles.updateFiles).not.toHaveBeenCalled();
+    });
+    it('should run codegen with Vertex AI Claude when vertexAiClaude flag is true', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai-claude';
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Hello from Claude");' } },
+        ];
+        vi.mocked(vertexAiClaude).generateContent.mockResolvedValueOnce(mockFunctionCalls);
+        await runCodegen();
+        expect(vertexAiClaude.generateContent).toHaveBeenCalled();
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    it('should pass the temperature parameter to the AI service', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        vi.mocked(cliParams).temperature = 0.5;
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Temperature test");' } },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        await runCodegen();
+        expect(vertexAi.generateContent).toHaveBeenCalledWith(expect.anything(), expect.anything(), expect.anything(), 0.5, false, expect.anything());
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    it('should print help message and not run codegen when --help option is provided', async () => {
+        vi.mocked(cliParams).helpRequested = true;
+        await runCodegen();
+        expect(cliOptions.printHelpMessage).toHaveBeenCalled();
+        expect(vertexAi.generateContent).not.toHaveBeenCalled();
+        expect(chatGpt.generateContent).not.toHaveBeenCalled();
+        expect(anthropic.generateContent).not.toHaveBeenCalled();
+        expect(vertexAiClaude.generateContent).not.toHaveBeenCalled();
+        expect(updateFiles.updateFiles).not.toHaveBeenCalled();
+    });
+    it('should run codegen with vision when vision flag is true', async () => {
+        vi.mocked(cliParams).aiService = 'chat-gpt';
+        vi.mocked(cliParams).vision = true;
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Vision test");' } },
+        ];
+        vi.mocked(chatGpt).generateContent.mockResolvedValueOnce(mockFunctionCalls);
+        await runCodegen();
+        expect(chatGpt.generateContent).toHaveBeenCalled();
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+        // Check if the generateContent function was called with the correct parameters
+        expect(vi.mocked(chatGpt).generateContent.mock.calls[0][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: expect.any(String) }],
+            }),
+        ]));
+    });
+    it('should use Vertex AI Imagen when imagen flag is set to vertex-ai', async () => {
+        vi.mocked(cliParams).imagen = 'vertex-ai';
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        { filePath: '/mocked/root/dir/landscape.png', updateToolName: 'generateImage', prompt: 'Generate image' },
+                    ],
+                    contextPaths: [],
+                    explanation: 'Mock summary with image generation failure',
+                },
+            },
+        ];
+        const mockFunctionCalls = [
+            {
+                name: 'generateImage',
+                args: { prompt: 'A beautiful landscape', filePath: '/mocked/root/dir/landscape.png', width: 512, height: 512 },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        vi.mocked(vertexAiImagen).generateImage.mockResolvedValueOnce('mocked-image-data');
+        await runCodegen();
+        expect(vertexAi.generateContent).toHaveBeenCalled();
+        expect(vertexAiImagen.generateImage).toHaveBeenCalledWith('A beautiful landscape', undefined, {
+            height: 512,
+            width: 512,
+        }, false);
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    it('should use DALL-E when imagen flag is set to dall-e', async () => {
+        vi.mocked(cliParams).imagen = 'dall-e';
+        vi.mocked(cliParams).aiService = 'chat-gpt';
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        { filePath: '/mocked/root/dir/city.png', updateToolName: 'generateImage', prompt: 'Generate image' },
+                    ],
+                    contextPaths: [],
+                    explanation: 'Mock summary with image generation failure',
+                },
+            },
+        ];
+        const mockFunctionCalls = [
+            {
+                name: 'generateImage',
+                args: { prompt: 'A futuristic city', filePath: '/mocked/root/dir/city.png', width: 1024, height: 1024 },
+            },
+        ];
+        vi.mocked(chatGpt).generateContent.mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(chatGpt).generateContent.mockResolvedValueOnce(mockFunctionCalls);
+        vi.mocked(dallE).generateImage.mockResolvedValueOnce('mocked-image-data');
+        await runCodegen();
+        expect(chatGpt.generateContent).toHaveBeenCalled();
+        expect(dallE.generateImage).toHaveBeenCalledWith('A futuristic city', undefined, { width: 1024, height: 1024 }, false);
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    it('should throw an error when imagen flag is set but no AI service is specified', async () => {
+        vi.mocked(cliParams).imagen = 'vertex-ai';
+        await expect(runCodegen()).rejects.toThrow('Please specify which AI service should be used');
+    });
+    it('should pass the cheap parameter to the AI service when --cheap flag is true', async () => {
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        vi.mocked(cliParams).cheap = true;
+        const mockFunctionCalls = [
+            { name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Cheap test");' } },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        await runCodegen();
+        expect(vertexAi.generateContent).toHaveBeenCalledWith(expect.anything(), expect.anything(), expect.anything(), expect.anything(), true, expect.anything());
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+    it('should pass the cheap parameter to the image generation service when --cheap flag is true', async () => {
+        vi.mocked(cliParams).imagen = 'vertex-ai';
+        vi.mocked(cliParams).aiService = 'vertex-ai';
+        vi.mocked(cliParams).cheap = true;
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [
+                        { filePath: '/mocked/root/dir/landscape.png', updateToolName: 'generateImage', prompt: 'Generate image' },
+                    ],
+                    contextPaths: [],
+                    explanation: 'Mock summary with cheap image generation',
+                },
+            },
+        ];
+        const mockFunctionCalls = [
+            {
+                name: 'generateImage',
+                args: {
+                    prompt: 'A simple landscape',
+                    filePath: '/mocked/root/dir/landscape.png',
+                    width: 256,
+                    height: 256,
+                    cheap: true,
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
+        vi.mocked(vertexAiImagen).generateImage.mockResolvedValueOnce('mocked-cheap-image-data');
+        await runCodegen();
+        expect(vertexAi.generateContent).toHaveBeenCalledWith(expect.anything(), expect.anything(), expect.anything(), expect.anything(), true, expect.anything());
+        expect(vertexAiImagen.generateImage).toHaveBeenCalledWith('A simple landscape', undefined, { width: 256, height: 256 }, true);
+        expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
+    });
+});
+//# sourceMappingURL=codegen.test.js.map

package/dist/src/main/codegen.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen.test.js","sourceRoot":"","sources":["../../../src/main/codegen.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AACvD,OAAO,KAAK,OAAO,MAAM,2BAA2B,CAAC;AACrD,OAAO,KAAK,SAAS,MAAM,4BAA4B,CAAC;AACxD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,WAAW,MAAM,0BAA0B,CAAC;AACxD,OAAO,wBAAwB,CAAC;AAChC,OAAO,wBAAwB,CAAC;AAChC,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,KAAK,UAAU,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,KAAK,MAAM,yBAAyB,CAAC;AACjD,OAAO,aAAa,CAAC;AAErB,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,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnF,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAClC,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,WAAW,EAAE,KAAK;IAClB,EAAE,EAAE,KAAK;IACT,MAAM,EAAE,IAAI;IACZ,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,cAAc,EAAE,KAAK;IACrB,aAAa,EAAE,KAAK;IACpB,QAAQ,EAAE,KAAK;IACf,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,QAAQ,EAAE,SAAS;IACnB,kBAAkB,EAAE,SAAS;IAC7B,WAAW,EAAE,KAAK;IAClB,0BAA0B,EAAE,IAAI;IAChC,eAAe,EAAE,SAAS;IAC1B,WAAW,EAAE,SAAS;IACtB,cAAc,EAAE,EAAE;IAClB,YAAY,EAAE,SAAS;IACvB,wBAAwB,EAAE,SAAS;IACnC,cAAc,EAAE,IAAI;IACpB,qBAAqB,EAAE,IAAI;IAC3B,0BAA0B,EAAE,IAAI;CACjC,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;IAC5B,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;CACzB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AACJ,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,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACvE,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,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,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAE7C,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,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;QAE5C,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,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAE7C,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,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE9E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,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;QAEnC,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,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,kBAAkB,CAAC;QAEpD,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,mCAAmC,EAAE,EAAE;SACvG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAEnF,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1D,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC;QAEvC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,kCAAkC,EAAE,EAAE;SACtG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,GAAG,EACH,KAAK,EACL,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC5F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,IAAI,CAAC;QAE1C,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvD,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC9D,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;QAC5C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QAEnC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,6BAA6B,EAAE,EAAE;SACjG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC3F,+EAA+E;QAC/E,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CACjE,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,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;aAC7E,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;QAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAE7C,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX,EAAE,QAAQ,EAAE,gCAAgC,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,EAAE,gBAAgB,EAAE;qBAC1G;oBACD,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,4CAA4C;iBAC1D;aACF;SACF,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,EAAE,MAAM,EAAE,uBAAuB,EAAE,QAAQ,EAAE,gCAAgC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;aAC/G;SACF,CAAC;QACF,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,iBAAiB,CAAC,CAAC;QAC7E,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;QAEnF,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACvD,uBAAuB,EACvB,SAAS,EACT;YACE,MAAM,EAAE,GAAG;YACX,KAAK,EAAE,GAAG;SACX,EACD,KAAK,CACN,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,QAAQ,CAAC;QACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;QAE5C,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX,EAAE,QAAQ,EAAE,2BAA2B,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,EAAE,gBAAgB,EAAE;qBACrG;oBACD,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,4CAA4C;iBAC1D;aACF;SACF,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,EAAE,MAAM,EAAE,mBAAmB,EAAE,QAAQ,EAAE,2BAA2B,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;aACxG;SACF,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC7E,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAC5E,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;QAE1E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAC9C,mBAAmB,EACnB,SAAS,EACT,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAC7B,KAAK,CACN,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC5F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;QAE1C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;QAElC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,4BAA4B,EAAE,EAAE;SAChG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,IAAI,EACJ,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;QACzG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;QAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;QAElC,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE;wBACX,EAAE,QAAQ,EAAE,gCAAgC,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,EAAE,gBAAgB,EAAE;qBAC1G;oBACD,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,0CAA0C;iBACxD;aACF;SACF,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,oBAAoB;oBAC5B,QAAQ,EAAE,gCAAgC;oBAC1C,KAAK,EAAE,GAAG;oBACV,MAAM,EAAE,GAAG;oBACX,KAAK,EAAE,IAAI;iBACZ;aACF;SACF,CAAC;QACF,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,iBAAiB,CAAC,CAAC;QAC7E,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC,qBAAqB,CAAC,yBAAyB,CAAC,CAAC;QAEzF,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,IAAI,EACJ,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACvD,oBAAoB,EACpB,SAAS,EACT,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAC3B,IAAI,CACL,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/main/common/content-bus-types.d.ts

@@ -0,0 +1,21 @@
+export declare enum ChatMessageType {
+    USER = "user",
+    ASSISTANT = "assistant",
+    SYSTEM = "system"
+}
+export declare enum ChatMessageFlags {
+    CONVERSATION_SUMMARY = "conversation-summary"
+}
+export interface ChatMessage {
+    id: string;
+    iterationId: string;
+    type: ChatMessageType;
+    flags?: ChatMessageFlags[];
+    content: string;
+    timestamp: Date;
+    data?: unknown;
+}
+export type ContentProps = {
+    message?: ChatMessage;
+    data?: unknown;
+};

package/dist/src/main/common/content-bus-types.js

@@ -0,0 +1,11 @@
+export var ChatMessageType;
+(function (ChatMessageType) {
+    ChatMessageType["USER"] = "user";
+    ChatMessageType["ASSISTANT"] = "assistant";
+    ChatMessageType["SYSTEM"] = "system";
+})(ChatMessageType || (ChatMessageType = {}));
+export var ChatMessageFlags;
+(function (ChatMessageFlags) {
+    ChatMessageFlags["CONVERSATION_SUMMARY"] = "conversation-summary";
+})(ChatMessageFlags || (ChatMessageFlags = {}));
+//# sourceMappingURL=content-bus-types.js.map

package/dist/src/main/common/content-bus-types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"content-bus-types.js","sourceRoot":"","sources":["../../../../src/main/common/content-bus-types.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,eAIX;AAJD,WAAY,eAAe;IACzB,gCAAa,CAAA;IACb,0CAAuB,CAAA;IACvB,oCAAiB,CAAA;AACnB,CAAC,EAJW,eAAe,KAAf,eAAe,QAI1B;AAED,MAAM,CAAN,IAAY,gBAEX;AAFD,WAAY,gBAAgB;IAC1B,iEAA6C,CAAA;AAC/C,CAAC,EAFW,gBAAgB,KAAhB,gBAAgB,QAE3B"}

package/dist/src/main/common/content-bus.d.ts

@@ -0,0 +1,10 @@
+import { ChatMessageFlags, ContentProps } from './content-bus-types.js';
+type ContentHandler = (content: ContentProps) => void;
+export declare function setCurrentIterationId(): void;
+export declare function unsetCurrentIterationId(): void;
+type MessageArgs = [data?: unknown, flags?: ChatMessageFlags[]];
+export declare function putUserMessage(message: string, ...args: MessageArgs): void;
+export declare function putSystemMessage(message: string, ...args: MessageArgs): void;
+export declare function putAssistantMessage(message: string, ...args: MessageArgs): void;
+export declare function registerContentHandler(handler: ContentHandler): void;
+export {};

package/dist/src/main/common/content-bus.js

@@ -0,0 +1,39 @@
+import { ChatMessageType } from './content-bus-types.js';
+let contentHandler;
+let currentIterationId;
+export function setCurrentIterationId() {
+    currentIterationId = Date.now().toString();
+}
+export function unsetCurrentIterationId() {
+    currentIterationId = null;
+}
+export function putUserMessage(message, ...args) {
+    putMessage(message, ChatMessageType.USER, ...args);
+}
+export function putSystemMessage(message, ...args) {
+    putMessage(message, ChatMessageType.SYSTEM, ...args);
+}
+export function putAssistantMessage(message, ...args) {
+    putMessage(message, ChatMessageType.ASSISTANT, ...args);
+}
+function putMessage(message, type, data, flags) {
+    if (!currentIterationId) {
+        console.warn('No current iteration ID set');
+    }
+    console.log(message, data);
+    contentHandler?.({
+        message: {
+            id: (Date.now() + Math.random()).toString(),
+            iterationId: currentIterationId,
+            content: message,
+            type: type,
+            flags,
+            timestamp: new Date(),
+        },
+        data,
+    });
+}
+export function registerContentHandler(handler) {
+    contentHandler = handler;
+}
+//# sourceMappingURL=content-bus.js.map

package/dist/src/main/common/content-bus.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"content-bus.js","sourceRoot":"","sources":["../../../../src/main/common/content-bus.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,eAAe,EAAgB,MAAM,wBAAwB,CAAC;AAIzF,IAAI,cAA0C,CAAC;AAC/C,IAAI,kBAAiC,CAAC;AAEtC,MAAM,UAAU,qBAAqB;IACnC,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,uBAAuB;IACrC,kBAAkB,GAAG,IAAI,CAAC;AAC5B,CAAC;AAID,MAAM,UAAU,cAAc,CAAC,OAAe,EAAE,GAAG,IAAiB;IAClE,UAAU,CAAC,OAAO,EAAE,eAAe,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,OAAe,EAAE,GAAG,IAAiB;IACpE,UAAU,CAAC,OAAO,EAAE,eAAe,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,OAAe,EAAE,GAAG,IAAiB;IACvE,UAAU,CAAC,OAAO,EAAE,eAAe,CAAC,SAAS,EAAE,GAAG,IAAI,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,UAAU,CAAC,OAAe,EAAE,IAAqB,EAAE,IAAc,EAAE,KAA0B;IACpG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAE3B,cAAc,EAAE,CAAC;QACf,OAAO,EAAE;YACP,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;YAC3C,WAAW,EAAE,kBAAmB;YAChC,OAAO,EAAE,OAAO;YAChB,IAAI,EAAE,IAAI;YACV,KAAK;YACL,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB;QACD,IAAI;KACL,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,OAAuB;IAC5D,cAAc,GAAG,OAAO,CAAC;AAC3B,CAAC"}

package/dist/src/main/common/cost-collector.d.ts

@@ -0,0 +1,11 @@
+import { AiServiceType } from '../codegen-types.js';
+export declare function collectCost(cost: number, inputTokens: number, outputTokens: number, aiService: AiServiceType, cheap?: boolean): void;
+export interface UsageMetrics {
+    cost: number;
+    rpm: number;
+    rpd: number;
+    tpm: number;
+    tpd: number;
+    ipm: number;
+}
+export declare function getUsageMetrics(): Record<AiServiceType | 'total', UsageMetrics>;

package/dist/src/main/common/cost-collector.js

@@ -0,0 +1,90 @@
+import { writeCache, readCache } from '../../files/cache-file.js';
+export function collectCost(cost, inputTokens, outputTokens, aiService, cheap = false) {
+    addUsageTuple({
+        timestamp: Date.now(),
+        cost,
+        inputTokens,
+        outputTokens,
+        aiService,
+        cheap,
+    });
+}
+const USAGE_CACHE_KEY = 'usageData';
+const CLEANUP_INTERVAL = 60 * 60 * 1000; // 1 hour in milliseconds
+const DATA_RETENTION_PERIOD = 48 * 60 * 60 * 1000; // 48 hours in milliseconds
+// Function to get usage data from cache
+function getUsageDataFromCache() {
+    const defaultData = { tuples: [], lastCleanup: Date.now() };
+    return readCache(USAGE_CACHE_KEY, defaultData);
+}
+// Function to save usage data to cache
+function saveUsageDataToCache(data) {
+    writeCache(USAGE_CACHE_KEY, data);
+}
+// Function to add a new usage tuple
+function addUsageTuple(tuple) {
+    const data = getUsageDataFromCache();
+    data.tuples.push(tuple);
+    cleanupOldData(data);
+    saveUsageDataToCache(data);
+}
+// Function to clean up old data
+function cleanupOldData(data) {
+    const now = Date.now();
+    if (now - data.lastCleanup > CLEANUP_INTERVAL) {
+        const cutoffTime = now - DATA_RETENTION_PERIOD;
+        data.tuples = data.tuples.filter((tuple) => tuple.timestamp >= cutoffTime);
+        data.lastCleanup = now;
+        console.log(`Cleaned up usage data. Remaining entries: ${data.tuples.length}`);
+    }
+}
+export function getUsageMetrics() {
+    const data = getUsageDataFromCache();
+    const now = Date.now();
+    const oneMinuteAgo = now - 60 * 1000;
+    const oneDayAgo = now - 24 * 60 * 60 * 1000;
+    const tuples = data.tuples.filter((t) => t.timestamp >= oneDayAgo);
+    const result = {};
+    for (const tuple of tuples) {
+        if (!result[tuple.aiService]) {
+            result[tuple.aiService] = {
+                cost: 0,
+                rpm: 0,
+                rpd: 0,
+                tpm: 0,
+                tpd: 0,
+                ipm: 0,
+            };
+        }
+    }
+    for (const aiService of Object.keys(result)) {
+        const serviceTuples = tuples.filter((tuple) => tuple.aiService === aiService);
+        const requestsLastMinute = serviceTuples.filter((t) => t.timestamp >= oneMinuteAgo).length;
+        const requestsLastDay = serviceTuples.filter((t) => t.timestamp >= oneDayAgo).length;
+        const tokensLastMinute = serviceTuples
+            .filter((t) => t.timestamp >= oneMinuteAgo)
+            .reduce((sum, t) => sum + t.inputTokens + t.outputTokens, 0);
+        const tokensLastDay = serviceTuples
+            .filter((t) => t.timestamp >= oneDayAgo)
+            .reduce((sum, t) => sum + t.inputTokens + t.outputTokens, 0);
+        const cost = serviceTuples.filter((t) => t.timestamp >= oneDayAgo).reduce((sum, t) => sum + t.cost, 0);
+        result[aiService] = {
+            cost: cost,
+            rpm: requestsLastMinute,
+            rpd: requestsLastDay,
+            tpm: tokensLastMinute,
+            tpd: tokensLastDay,
+            ipm: 0, // TODO: Implement image request monitoring
+        };
+    }
+    result['total'] = {
+        cost: tuples.filter((t) => t.timestamp >= oneDayAgo).reduce((sum, t) => sum + t.cost, 0),
+        rpm: tuples.filter((t) => t.timestamp >= oneMinuteAgo).length,
+        rpd: tuples.filter((t) => t.timestamp >= oneDayAgo).length,
+        tpm: tuples.filter((t) => t.timestamp >= oneMinuteAgo).reduce((sum, t) => sum + t.inputTokens + t.outputTokens, 0),
+        tpd: tuples.filter((t) => t.timestamp >= oneDayAgo).reduce((sum, t) => sum + t.inputTokens + t.outputTokens, 0),
+        ipm: 0, // TODO: Implement image request monitoring
+    };
+    return result;
+}
+//# sourceMappingURL=cost-collector.js.map

package/dist/src/main/common/cost-collector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-collector.js","sourceRoot":"","sources":["../../../../src/main/common/cost-collector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAGlE,MAAM,UAAU,WAAW,CACzB,IAAY,EACZ,WAAmB,EACnB,YAAoB,EACpB,SAAwB,EACxB,QAAiB,KAAK;IAEtB,aAAa,CAAC;QACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,IAAI;QACJ,WAAW;QACX,YAAY;QACZ,SAAS;QACT,KAAK;KACN,CAAC,CAAC;AACL,CAAC;AA0BD,MAAM,eAAe,GAAG,WAAW,CAAC;AACpC,MAAM,gBAAgB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,yBAAyB;AAClE,MAAM,qBAAqB,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,2BAA2B;AAE9E,wCAAwC;AACxC,SAAS,qBAAqB;IAC5B,MAAM,WAAW,GAAc,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IACvE,OAAO,SAAS,CAAY,eAAe,EAAE,WAAW,CAAC,CAAC;AAC5D,CAAC;AAED,uCAAuC;AACvC,SAAS,oBAAoB,CAAC,IAAe;IAC3C,UAAU,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;AACpC,CAAC;AAED,oCAAoC;AACpC,SAAS,aAAa,CAAC,KAAiB;IACtC,MAAM,IAAI,GAAG,qBAAqB,EAAE,CAAC;IACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxB,cAAc,CAAC,IAAI,CAAC,CAAC;IACrB,oBAAoB,CAAC,IAAI,CAAC,CAAC;AAC7B,CAAC;AAED,gCAAgC;AAChC,SAAS,cAAc,CAAC,IAAe;IACrC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,IAAI,GAAG,GAAG,IAAI,CAAC,WAAW,GAAG,gBAAgB,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,GAAG,GAAG,qBAAqB,CAAC;QAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,UAAU,CAAC,CAAC;QAC3E,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;QACvB,OAAO,CAAC,GAAG,CAAC,6CAA6C,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACjF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,MAAM,IAAI,GAAG,qBAAqB,EAAE,CAAC;IACrC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,MAAM,YAAY,GAAG,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC;IACrC,MAAM,SAAS,GAAG,GAAG,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;IAE5C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC;IAEnE,MAAM,MAAM,GAA2D,EAAE,CAAC;IAE1E,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG;gBACxB,IAAI,EAAE,CAAC;gBACP,GAAG,EAAE,CAAC;gBACN,GAAG,EAAE,CAAC;gBACN,GAAG,EAAE,CAAC;gBACN,GAAG,EAAE,CAAC;gBACN,GAAG,EAAE,CAAC;aACP,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAoB,EAAE,CAAC;QAC/D,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;QAC9E,MAAM,kBAAkB,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,YAAY,CAAC,CAAC,MAAM,CAAC;QAC3F,MAAM,eAAe,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC,MAAM,CAAC;QACrF,MAAM,gBAAgB,GAAG,aAAa;aACnC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,YAAY,CAAC;aAC1C,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,aAAa,GAAG,aAAa;aAChC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC;aACvC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,IAAI,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAEvG,MAAM,CAAC,SAAS,CAAC,GAAG;YAClB,IAAI,EAAE,IAAI;YACV,GAAG,EAAE,kBAAkB;YACvB,GAAG,EAAE,eAAe;YACpB,GAAG,EAAE,gBAAgB;YACrB,GAAG,EAAE,aAAa;YAClB,GAAG,EAAE,CAAC,EAAE,2CAA2C;SACpD,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,GAAG;QAChB,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACxF,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,YAAY,CAAC,CAAC,MAAM;QAC7D,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC,MAAM;QAC1D,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC;QAClH,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC;QAC/G,GAAG,EAAE,CAAC,EAAE,2CAA2C;KACpD,CAAC;IAEF,OAAO,MAAuD,CAAC;AACjE,CAAC"}

package/dist/src/main/common/user-actions.d.ts

@@ -0,0 +1,19 @@
+type InputHandler = (prompt: string, message: string) => Promise<string>;
+type ConfirmHandlerResponse = {
+    confirmed: boolean | undefined;
+    answer?: string;
+};
+export type ConfirmHandlerProps = {
+    prompt: string;
+    confirmLabel: string;
+    declineLabel: string;
+    includeAnswer: boolean;
+    defaultValue: boolean;
+};
+type ConfirmHandler = (props: ConfirmHandlerProps) => Promise<ConfirmHandlerResponse>;
+export declare function askUserForInput(prompt: string, message: string): Promise<string>;
+export declare function askUserForConfirmation(prompt: string, defaultValue: boolean): Promise<ConfirmHandlerResponse>;
+export declare function askUserForConfirmationWithAnswer(prompt: string, confirmLabel: string, declineLabel: string, defaultValue: boolean): Promise<ConfirmHandlerResponse>;
+export declare function registerInputHandler(handler: InputHandler): void;
+export declare function registerConfirmHandler(handler: ConfirmHandler): void;
+export {};

package/dist/src/main/common/user-actions.js

@@ -0,0 +1,30 @@
+export async function askUserForInput(prompt, message) {
+    return await inputHandler(prompt, message);
+}
+export async function askUserForConfirmation(prompt, defaultValue) {
+    return await confirmHandler({
+        prompt,
+        confirmLabel: 'Yes',
+        declineLabel: 'No',
+        includeAnswer: false,
+        defaultValue,
+    });
+}
+export async function askUserForConfirmationWithAnswer(prompt, confirmLabel, declineLabel, defaultValue) {
+    return await confirmHandler({
+        prompt,
+        confirmLabel,
+        declineLabel,
+        includeAnswer: true,
+        defaultValue,
+    });
+}
+let inputHandler;
+let confirmHandler;
+export function registerInputHandler(handler) {
+    inputHandler = handler;
+}
+export function registerConfirmHandler(handler) {
+    confirmHandler = handler;
+}
+//# sourceMappingURL=user-actions.js.map

package/dist/src/main/common/user-actions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"user-actions.js","sourceRoot":"","sources":["../../../../src/main/common/user-actions.ts"],"names":[],"mappings":"AAiBA,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,MAAc,EAAE,OAAe;IACnE,OAAO,MAAM,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,MAAc,EAAE,YAAqB;IAChF,OAAO,MAAM,cAAc,CAAC;QAC1B,MAAM;QACN,YAAY,EAAE,KAAK;QACnB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,KAAK;QACpB,YAAY;KACb,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gCAAgC,CACpD,MAAc,EACd,YAAoB,EACpB,YAAoB,EACpB,YAAqB;IAErB,OAAO,MAAM,cAAc,CAAC;QAC1B,MAAM;QACN,YAAY;QACZ,YAAY;QACZ,aAAa,EAAE,IAAI;QACnB,YAAY;KACb,CAAC,CAAC;AACL,CAAC;AAED,IAAI,YAA0B,CAAC;AAC/B,IAAI,cAA8B,CAAC;AAEnC,MAAM,UAAU,oBAAoB,CAAC,OAAqB;IACxD,YAAY,GAAG,OAAO,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,OAAuB;IAC5D,cAAc,GAAG,OAAO,CAAC;AAC3B,CAAC"}

package/dist/src/main/config-lib.d.ts

@@ -0,0 +1,36 @@
+export interface TextPrompt {
+    content: string;
+}
+export interface ImportantContext {
+    textPrompts?: TextPrompt[];
+    files?: string[];
+}
+export interface ModelOverrides {
+    chatGpt?: {
+        cheap?: string;
+        default?: string;
+    };
+    anthropic?: {
+        cheap?: string;
+        default?: string;
+    };
+    vertexAi?: {
+        cheap?: string;
+        default?: string;
+    };
+    aiStudio?: {
+        cheap?: string;
+        default?: string;
+    };
+}
+export interface RcConfig {
+    rootDir: string;
+    lintCommand?: string;
+    extensions?: string[];
+    ignorePaths?: string[];
+    importantContext?: ImportantContext;
+    modelOverrides?: ModelOverrides;
+    plugins?: string[];
+}
+export declare function findRcFile(): Promise<string>;
+export declare function parseRcFile(rcFilePath: string): RcConfig;