genaicode 0.0.37 → 0.2.0

package/dist/ai-service/common.test.js

@@ -0,0 +1,85 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { printTokenUsageAndCost, processFunctionCalls } from './common.js';
+import { functionDefs } from '../prompt/function-calling.js';
+// Mock cli-params.js
+vi.mock('../cli/cli-params.js', () => ({
+    chatGpt: true,
+    anthropic: false,
+    vertexAi: false,
+    vertexAiClaude: false,
+    temperature: 0.7,
+    cheap: false,
+    geminiBlockNone: false,
+    requireExplanations: false,
+    vision: false,
+}));
+describe('printTokenUsageAndCost', () => {
+    let consoleLogSpy;
+    beforeEach(() => {
+        consoleLogSpy = vi.spyOn(console, 'log').mockImplementation(() => { });
+    });
+    afterEach(() => {
+        consoleLogSpy.mockRestore();
+    });
+    it('should correctly log token usage and estimated cost', () => {
+        const usage = {
+            inputTokens: 100,
+            outputTokens: 50,
+            totalTokens: 150,
+        };
+        const inputCostPerToken = 0.0001;
+        const outputCostPerToken = 0.0002;
+        printTokenUsageAndCost(usage, inputCostPerToken, outputCostPerToken);
+        expect(consoleLogSpy).toHaveBeenCalledTimes(5);
+        expect(consoleLogSpy).toHaveBeenCalledWith('Token Usage:');
+        expect(consoleLogSpy).toHaveBeenCalledWith('  - Input tokens: ', 100);
+        expect(consoleLogSpy).toHaveBeenCalledWith('  - Output tokens: ', 50);
+        expect(consoleLogSpy).toHaveBeenCalledWith('  - Total tokens: ', 150);
+        expect(consoleLogSpy).toHaveBeenCalledWith('  - Estimated cost: ', '0.020000', ' USD');
+    });
+    it('should handle zero tokens correctly', () => {
+        const usage = {
+            inputTokens: 0,
+            outputTokens: 0,
+            totalTokens: 0,
+        };
+        const inputCostPerToken = 0.0001;
+        const outputCostPerToken = 0.0002;
+        printTokenUsageAndCost(usage, inputCostPerToken, outputCostPerToken);
+        expect(consoleLogSpy).toHaveBeenCalledWith('  - Estimated cost: ', '0.000000', ' USD');
+    });
+});
+describe('processFunctionCalls', () => {
+    it('should process valid function calls correctly', () => {
+        const validFunctionCalls = [
+            { name: 'explanation', args: { text: 'This is an explanation' } },
+            { name: functionDefs[0].name, args: {} },
+        ];
+        const result = processFunctionCalls(validFunctionCalls);
+        expect(result).toEqual(validFunctionCalls);
+    });
+    it('should throw an error for unknown function names', () => {
+        const invalidFunctionCalls = [{ name: 'unknownFunction', args: {} }];
+        expect(() => processFunctionCalls(invalidFunctionCalls)).toThrow('Unknown function name: unknownFunction');
+    });
+    it('should correctly handle explanations', () => {
+        const consoleLogSpy = vi.spyOn(console, 'log').mockImplementation(() => { });
+        const functionCallsWithExplanation = [
+            { name: 'explanation', args: { text: 'This is an explanation' } },
+            { name: functionDefs[0].name, args: {} },
+        ];
+        processFunctionCalls(functionCallsWithExplanation);
+        expect(consoleLogSpy).toHaveBeenCalledWith('Explanations:', ['This is an explanation']);
+        consoleLogSpy.mockRestore();
+    });
+    it('should return all function calls including explanations', () => {
+        const functionCalls = [
+            { name: 'explanation', args: { text: 'This is an explanation' } },
+            { name: functionDefs[0].name, args: {} },
+        ];
+        const result = processFunctionCalls(functionCalls);
+        expect(result).toEqual(functionCalls);
+        expect(result.length).toBe(2);
+    });
+});
+//# sourceMappingURL=common.test.js.map

package/dist/ai-service/common.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"common.test.js","sourceRoot":"","sources":["../../src/ai-service/common.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAgB,MAAM,QAAQ,CAAC;AACvF,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAC3E,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAE7D,qBAAqB;AACrB,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,OAAO,EAAE,IAAI;IACb,SAAS,EAAE,KAAK;IAChB,QAAQ,EAAE,KAAK;IACf,cAAc,EAAE,KAAK;IACrB,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,eAAe,EAAE,KAAK;IACtB,mBAAmB,EAAE,KAAK;IAC1B,MAAM,EAAE,KAAK;CACd,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,IAAI,aAGF,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACd,aAAa,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,aAAa,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG;YACZ,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,WAAW,EAAE,GAAG;SACjB,CAAC;QACF,MAAM,iBAAiB,GAAG,MAAM,CAAC;QACjC,MAAM,kBAAkB,GAAG,MAAM,CAAC;QAElC,sBAAsB,CAAC,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,CAAC,CAAC;QAErE,MAAM,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAC3D,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE,GAAG,CAAC,CAAC;QACtE,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE,GAAG,CAAC,CAAC;QACtE,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,sBAAsB,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IACzF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,KAAK,GAAG;YACZ,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;SACf,CAAC;QACF,MAAM,iBAAiB,GAAG,MAAM,CAAC;QACjC,MAAM,kBAAkB,GAAG,MAAM,CAAC;QAElC,sBAAsB,CAAC,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,CAAC,CAAC;QAErE,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,sBAAsB,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IACzF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,kBAAkB,GAAG;YACzB,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,EAAE;YACjE,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE;SACzC,CAAC;QAEF,MAAM,MAAM,GAAG,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;QAExD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,oBAAoB,GAAG,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,CAAC,CAAC,OAAO,CAAC,wCAAwC,CAAC,CAAC;IAC7G,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,aAAa,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAE5E,MAAM,4BAA4B,GAAG;YACnC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,EAAE;YACjE,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE;SACzC,CAAC;QAEF,oBAAoB,CAAC,4BAA4B,CAAC,CAAC;QAEnD,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,eAAe,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC;QAExF,aAAa,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,aAAa,GAAG;YACpB,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,EAAE;YACjE,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE;SACzC,CAAC;QAEF,MAAM,MAAM,GAAG,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/ai-service/dall-e.d.ts

@@ -0,0 +1,14 @@
+interface ImageSize {
+    width: number;
+    height: number;
+}
+/**
+ * Generate an image using OpenAI's DALL-E model and save it to a file
+ * @param {string} prompt - The description of the image to generate
+ * @param {string|undefined} contextImagePath - The image to be used as a context
+ * @param {ImageSize} size - The size of the image to generate
+ * @param {boolean} cheap - Whether to use a cheaper model
+ * @returns {Promise<string>} - The url of the image
+ */
+export declare function generateImage(prompt: string, contextImagePath: string | undefined, size: ImageSize, cheap?: boolean): Promise<string>;
+export {};

package/dist/ai-service/dall-e.js

@@ -0,0 +1,48 @@
+import OpenAI, { toFile } from 'openai';
+import { setTempBuffer } from '../files/temp-buffer.js';
+import { resizeImageBuffer } from '../images/resize-image.js';
+import { ensureAlpha } from '../images/ensure-alpha.js';
+import { abortController } from '../main/interactive/codegen-worker.js';
+/**
+ * Generate an image using OpenAI's DALL-E model and save it to a file
+ * @param {string} prompt - The description of the image to generate
+ * @param {string|undefined} contextImagePath - The image to be used as a context
+ * @param {ImageSize} size - The size of the image to generate
+ * @param {boolean} cheap - Whether to use a cheaper model
+ * @returns {Promise<string>} - The url of the image
+ */
+export async function generateImage(prompt, contextImagePath, size, cheap = false) {
+    const openai = new OpenAI();
+    try {
+        const model = contextImagePath ? 'dall-e-2' : cheap ? 'dall-e-2' : 'dall-e-3';
+        console.log(`Using DALL-E model: ${model}`);
+        const options = {
+            model: model,
+            prompt: prompt,
+            n: 1,
+            size: '1024x1024',
+            response_format: 'url',
+        };
+        const response = contextImagePath
+            ? await openai.images.edit({
+                ...options,
+                image: await toFile(await ensureAlpha(contextImagePath)),
+            }, { signal: abortController?.signal })
+            : await openai.images.generate(options, { signal: abortController?.signal });
+        let imageUrl = response.data[0].url;
+        if (size.width !== 1024 || size.height !== 1024) {
+            console.log('Resizing image to desired size', size);
+            const imageResponse = await fetch(imageUrl);
+            const arrayBuffer = await imageResponse.arrayBuffer();
+            const buffer = Buffer.from(arrayBuffer);
+            imageUrl = setTempBuffer(await resizeImageBuffer(buffer, size));
+        }
+        console.log(`Image generated, url: ${imageUrl}`);
+        return imageUrl;
+    }
+    catch (error) {
+        console.error('Error generating image:', error);
+        throw error;
+    }
+}
+//# sourceMappingURL=dall-e.js.map

package/dist/ai-service/dall-e.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dall-e.js","sourceRoot":"","sources":["../../src/ai-service/dall-e.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAExC,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAOxE;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAc,EACd,gBAAoC,EACpC,IAAe,EACf,KAAK,GAAG,KAAK;IAEb,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;IAE5B,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,gBAAgB,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC;QAC9E,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;QAE5C,MAAM,OAAO,GAAsC;YACjD,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,MAAM;YACd,CAAC,EAAE,CAAC;YACJ,IAAI,EAAE,WAAW;YACjB,eAAe,EAAE,KAAK;SACvB,CAAC;QAEF,MAAM,QAAQ,GAAG,gBAAgB;YAC/B,CAAC,CAAC,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CACtB;gBACE,GAAG,OAAO;gBACV,KAAK,EAAE,MAAM,MAAM,CAAC,MAAM,WAAW,CAAC,gBAAgB,CAAC,CAAC;aACxB,EAClC,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,EAAE,CACpC;YACH,CAAC,CAAC,MAAM,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,EAAE,CAAC,CAAC;QAE/E,IAAI,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAI,CAAC;QAErC,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE,IAAI,CAAC,CAAC;YACpD,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,WAAW,EAAE,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAExC,QAAQ,GAAG,aAAa,CAAC,MAAM,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;QAClE,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;QACjD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QAChD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/ai-service/vertex-ai-claude.d.ts

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

package/dist/ai-service/vertex-ai-claude.js

@@ -0,0 +1,94 @@
+import { AnthropicVertex } from '@anthropic-ai/vertex-sdk';
+import { printTokenUsageAndCost, processFunctionCalls } from './common.js';
+import { abortController } from '../main/interactive/codegen-worker.js';
+/**
+ * This function generates content using the Anthropic Claude model via Vertex AI.
+ */
+export async function generateContent(prompt, functionDefs, requiredFunctionName, temperature, cheap = false) {
+    const projectId = process.env.GOOGLE_CLOUD_PROJECT;
+    const region = process.env.GOOGLE_CLOUD_REGION;
+    if (!projectId || !region) {
+        throw new Error('GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_REGION environment variables must be set');
+    }
+    const client = new AnthropicVertex({
+        projectId,
+        region,
+    });
+    const messages = prompt
+        .filter((item) => item.type !== 'systemPrompt')
+        .map((item) => {
+        if (item.type === 'user') {
+            return {
+                role: 'user',
+                content: [
+                    ...(item.functionResponses ?? []).map((response) => ({
+                        type: 'tool_result',
+                        tool_use_id: response.call_id ?? response.name,
+                        content: response.content,
+                        is_error: response.isError === true,
+                    })),
+                    ...(item.images ?? []).map((image) => ({
+                        type: 'image',
+                        source: {
+                            type: 'base64',
+                            media_type: image.mediaType,
+                            data: image.base64url,
+                        },
+                    })),
+                    { type: 'text', text: item.text },
+                ],
+            };
+        }
+        else if (item.type === 'assistant') {
+            return {
+                role: 'assistant',
+                content: [
+                    ...(item.text ? [{ type: 'text', text: item.text }] : []),
+                    ...item.functionCalls.map((call) => ({
+                        type: 'tool_use',
+                        id: call.id ?? call.name,
+                        name: call.name,
+                        input: call.args ?? {},
+                    })),
+                ],
+            };
+        }
+    });
+    const model = cheap ? 'claude-3-haiku@20240307' : 'claude-3-5-sonnet@20240620';
+    console.log(`Using Vertex AI Claude model: ${model}`);
+    const response = await client.messages.create({
+        model: model,
+        max_tokens: cheap ? 4096 : 8192,
+        temperature: temperature,
+        system: prompt.find((item) => item.type === 'systemPrompt').systemPrompt,
+        messages: messages,
+        tools: functionDefs.map((fd) => ({
+            name: fd.name,
+            description: fd.description,
+            input_schema: fd.parameters,
+        })),
+        tool_choice: requiredFunctionName
+            ? { type: 'tool', name: requiredFunctionName }
+            : { type: 'any' },
+    }, { signal: abortController?.signal });
+    // Print token usage for Anthropic Vertex AI
+    const usage = {
+        inputTokens: response.usage.input_tokens,
+        outputTokens: response.usage.output_tokens,
+        totalTokens: response.usage.input_tokens + response.usage.output_tokens,
+    };
+    printTokenUsageAndCost(usage, 3 / 1000 / 1000, 15 / 1000 / 1000);
+    const responseMessages = response.content.filter((item) => item.type !== 'tool_use');
+    if (responseMessages.length > 0) {
+        console.log('Response messages', responseMessages);
+    }
+    const functionCalls = response.content
+        .filter((item) => item.type === 'tool_use')
+        .map((item) => ({
+        id: item.id,
+        name: item.name,
+        args: item.input,
+    }));
+    return processFunctionCalls(functionCalls);
+}
+//# sourceMappingURL=vertex-ai-claude.js.map

package/dist/ai-service/vertex-ai-claude.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex-ai-claude.js","sourceRoot":"","sources":["../../src/ai-service/vertex-ai-claude.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAyC,MAAM,aAAa,CAAC;AAElH,OAAO,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAExE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,MAAoB,EACpB,YAA2B,EAC3B,oBAAmC,EACnC,WAAmB,EACnB,KAAK,GAAG,KAAK;IAEb,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;IACnD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;IAE/C,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,gFAAgF,CAAC,CAAC;IACpG,CAAC;IAED,MAAM,MAAM,GAAoB,IAAI,eAAe,CAAC;QAClD,SAAS;QACT,MAAM;KACP,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,MAAM;SACpB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,OAAO;gBACL,IAAI,EAAE,MAAe;gBACrB,OAAO,EAAE;oBACP,GAAG,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;wBACnD,IAAI,EAAE,aAAsB;wBAC5B,WAAW,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,IAAI;wBAC9C,OAAO,EAAE,QAAQ,CAAC,OAAO;wBACzB,QAAQ,EAAE,QAAQ,CAAC,OAAO,KAAK,IAAI;qBACpC,CAAC,CAAC;oBACH,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBACrC,IAAI,EAAE,OAAgB;wBACtB,MAAM,EAAE;4BACN,IAAI,EAAE,QAAiB;4BACvB,UAAU,EAAE,KAAK,CAAC,SAAS;4BAC3B,IAAI,EAAE,KAAK,CAAC,SAAS;yBACtB;qBACF,CAAC,CAAC;oBACH,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;iBAC3C;aACF,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,OAAO;gBACL,IAAI,EAAE,WAAoB;gBAC1B,OAAO,EAAE;oBACP,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAClE,GAAG,IAAI,CAAC,aAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;wBACpC,IAAI,EAAE,UAAmB;wBACzB,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,IAAI;wBACxB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,KAAK,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;qBACvB,CAAC,CAAC;iBACJ;aACF,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,4BAA4B,CAAC;IAC/E,OAAO,CAAC,GAAG,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;IAEtD,MAAM,QAAQ,GAAY,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,CACpD;QACE,KAAK,EAAE,KAAK;QACZ,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI;QAC/B,WAAW,EAAE,WAAW;QACxB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAE,CAAC,YAAa;QAC1E,QAAQ,EAAE,QAAQ;QAClB,KAAK,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC/B,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,WAAW,EAAE,EAAE,CAAC,WAAW;YAC3B,YAAY,EAAE,EAAE,CAAC,UAAU;SAC5B,CAAC,CAAC;QACH,WAAW,EAAE,oBAAoB;YAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,oBAAoB,EAAE;YACvD,CAAC,CAAC,EAAE,IAAI,EAAE,KAAc,EAAE;KAC7B,EACD,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,EAAE,CACpC,CAAC;IAEF,4CAA4C;IAC5C,MAAM,KAAK,GAAG;QACZ,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;QACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;QAC1C,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa;KACxE,CAAC;IACF,sBAAsB,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI,GAAG,IAAI,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;IAEjE,MAAM,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;IACrF,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,gBAAgB,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO;SACnC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,UAAU,CAAC;SAC1C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACd,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,IAAI,EAAE,IAAI,CAAC,KAAgC;KAC5C,CAAC,CAAC,CAAC;IAEN,OAAO,oBAAoB,CAAC,aAAa,CAAC,CAAC;AAC7C,CAAC"}

package/dist/ai-service/vertex-ai-imagen.d.ts

@@ -0,0 +1,14 @@
+interface ImageSize {
+    width: number;
+    height: number;
+}
+/**
+ * Generate an image using Vertex AI's Imagen model and return the image URL
+ * @param {string} prompt - The description of the image to generate
+ * @param {string|undefined} contextImagePath - The image to be used as a context
+ * @param {ImageSize} size - The size of the image to generate
+ * @param {boolean} cheap - Whether to use a cheaper model
+ * @returns {Promise<string>} - The url of the generated image
+ */
+export declare function generateImage(prompt: string, contextImagePath: string | undefined, size: ImageSize, cheap?: boolean): Promise<string>;
+export {};

package/dist/ai-service/vertex-ai-imagen.js

@@ -0,0 +1,77 @@
+import fs from 'fs';
+import mime from 'mime-types';
+import { PredictionServiceClient, helpers } from '@google-cloud/aiplatform';
+import { setTempBuffer } from '../files/temp-buffer.js';
+import { resizeImageBuffer } from '../images/resize-image.js';
+import { abortController } from '../main/interactive/codegen-worker.js';
+/**
+ * Generate an image using Vertex AI's Imagen model and return the image URL
+ * @param {string} prompt - The description of the image to generate
+ * @param {string|undefined} contextImagePath - The image to be used as a context
+ * @param {ImageSize} size - The size of the image to generate
+ * @param {boolean} cheap - Whether to use a cheaper model
+ * @returns {Promise<string>} - The url of the generated image
+ */
+export async function generateImage(prompt, contextImagePath, size, cheap = false) {
+    // Limitation in @google-cloud/aiplatform
+    abortController?.signal.throwIfAborted();
+    // Initialize the PredictionServiceClient
+    const client = new PredictionServiceClient({
+        apiEndpoint: `${process.env.GOOGLE_CLOUD_REGION}-aiplatform.googleapis.com`,
+    });
+    // Set the project and location
+    const projectId = process.env.GOOGLE_CLOUD_PROJECT;
+    const location = process.env.GOOGLE_CLOUD_REGION;
+    if (!projectId || !location) {
+        throw new Error('GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_REGION environment variables must be set');
+    }
+    // Set the model name based on the cheap parameter
+    const modelName = contextImagePath
+        ? 'imagegeneration@002'
+        : cheap
+            ? 'imagen-3.0-fast-generate-001'
+            : 'imagen-3.0-generate-001';
+    console.log(`Using Vertex AI Imagen model: ${modelName}`);
+    try {
+        // Prepare the request
+        const instance = contextImagePath
+            ? helpers.toValue({
+                prompt,
+                image: {
+                    bytesBase64Encoded: fs.readFileSync(contextImagePath, 'base64'),
+                    mediaType: mime.lookup(contextImagePath) ?? 'application/octet-stream',
+                },
+            })
+            : helpers.toValue({ prompt });
+        const request = {
+            endpoint: `projects/${projectId}/locations/${location}/publishers/google/models/${modelName}`,
+            instances: [instance],
+            parameters: helpers.toValue({
+                sampleCount: 1,
+                safetySetting: 'block_most',
+                personGeneration: 'allow_adult',
+                includeRaiReason: true,
+                language: 'auto',
+                aspectRatio: '1:1',
+                addWatermark: false,
+            }),
+        };
+        // Make the prediction request
+        const [response] = (await client.predict(request));
+        if (response.predictions && response.predictions.length > 0 && response.predictions[0]) {
+            const value = response.predictions[0];
+            const prediction = helpers.fromValue(value);
+            if (prediction.bytesBase64Encoded) {
+                console.log(`Image generated successfully, resizing to desired dimension`, size);
+                const buffer = Buffer.from(prediction.bytesBase64Encoded, 'base64');
+                return setTempBuffer(await resizeImageBuffer(buffer, size));
+            }
+        }
+        throw new Error('No image generated in the response');
+    }
+    catch (error) {
+        console.error('Error generating image:', error);
+        throw error;
+    }
+}
+//# sourceMappingURL=vertex-ai-imagen.js.map

package/dist/ai-service/vertex-ai-imagen.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex-ai-imagen.js","sourceRoot":"","sources":["../../src/ai-service/vertex-ai-imagen.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,EAAE,uBAAuB,EAAE,OAAO,EAAU,MAAM,0BAA0B,CAAC;AACpF,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAOxE;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAc,EACd,gBAAoC,EACpC,IAAe,EACf,KAAK,GAAG,KAAK;IAEb,yCAAyC;IACzC,eAAe,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC;IAEzC,yCAAyC;IACzC,MAAM,MAAM,GAAG,IAAI,uBAAuB,CAAC;QACzC,WAAW,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,4BAA4B;KAC5E,CAAC,CAAC;IAEH,+BAA+B;IAC/B,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;IACnD,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;IAEjD,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,gFAAgF,CAAC,CAAC;IACpG,CAAC;IAED,kDAAkD;IAClD,MAAM,SAAS,GAAG,gBAAgB;QAChC,CAAC,CAAC,qBAAqB;QACvB,CAAC,CAAC,KAAK;YACL,CAAC,CAAC,8BAA8B;YAChC,CAAC,CAAC,yBAAyB,CAAC;IAChC,OAAO,CAAC,GAAG,CAAC,iCAAiC,SAAS,EAAE,CAAC,CAAC;IAE1D,IAAI,CAAC;QACH,sBAAsB;QACtB,MAAM,QAAQ,GAAG,gBAAgB;YAC/B,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;gBACd,MAAM;gBACN,KAAK,EAAE;oBACL,kBAAkB,EAAE,EAAE,CAAC,YAAY,CAAC,gBAAgB,EAAE,QAAQ,CAAC;oBAC/D,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,0BAA0B;iBACvE;aACF,CAAC;YACJ,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAChC,MAAM,OAAO,GAAsD;YACjE,QAAQ,EAAE,YAAY,SAAS,cAAc,QAAQ,6BAA6B,SAAS,EAAE;YAC7F,SAAS,EAAE,CAAC,QAAS,CAAC;YACtB,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC;gBAC1B,WAAW,EAAE,CAAC;gBACd,aAAa,EAAE,YAAY;gBAC3B,gBAAgB,EAAE,aAAa;gBAC/B,gBAAgB,EAAE,IAAI;gBACtB,QAAQ,EAAE,MAAM;gBAChB,WAAW,EAAE,KAAK;gBAClB,YAAY,EAAE,KAAK;aACpB,CAAC;SACH,CAAC;QAEF,8BAA8B;QAC9B,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAE,CAAC;QAEpD,IAAI,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;YACvF,MAAM,KAAK,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC,CAA2B,CAAC;YAChE,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAA+C,CAAC;YAC1F,IAAI,UAAU,CAAC,kBAAkB,EAAE,CAAC;gBAClC,OAAO,CAAC,GAAG,CAAC,6DAA6D,EAAE,IAAI,CAAC,CAAC;gBACjF,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;gBACpE,OAAO,aAAa,CAAC,MAAM,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACxD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QAChD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/ai-service/vertex-ai.d.ts

@@ -0,0 +1,7 @@
+import { FunctionCall, PromptItem, FunctionDef } from './common.js';
+import { CodegenOptions } from '../main/codegen-types.js';
+/**
+ * This function generates content using the Gemini Pro model.
+ */
+export declare function generateContent(prompt: PromptItem[], functionDefs: FunctionDef[], requiredFunctionName: string | null, temperature: number, cheap: boolean | undefined, options: CodegenOptions): Promise<FunctionCall[]>;
+export declare function getGenModel(systemPrompt: string, temperature: number, functionDefs: FunctionDef[], geminiBlockNone: boolean | undefined, requiredFunctionName: string | null, cheap?: boolean): import("@google-cloud/vertexai").GenerativeModelPreview;

package/dist/ai-service/vertex-ai.js

@@ -0,0 +1,142 @@
+import assert from 'node:assert';
+import { VertexAI, HarmCategory, HarmBlockThreshold, FunctionCallingMode, } from '@google-cloud/vertexai';
+import { printTokenUsageAndCost, processFunctionCalls } from './common.js';
+import { abortController } from '../main/interactive/codegen-worker.js';
+/**
+ * This function generates content using the Gemini Pro model.
+ */
+export async function generateContent(prompt, functionDefs, requiredFunctionName, temperature, cheap = false, options) {
+    // Limitation: https://github.com/googleapis/nodejs-vertexai/issues/143
+    abortController?.signal.throwIfAborted();
+    const messages = prompt
+        .filter((item) => item.type === 'user' || item.type === 'assistant')
+        .map((item) => {
+        if (item.type === 'user') {
+            const content = {
+                role: 'user',
+                parts: [
+                    ...(item.functionResponses ?? []).map((response) => ({
+                        functionResponse: {
+                            name: response.name,
+                            response: { name: response.name, content: response.content },
+                        },
+                    })),
+                    ...(item.images ?? []).map((image) => ({
+                        inlineData: {
+                            mimeType: image.mediaType,
+                            data: image.base64url,
+                        },
+                    })),
+                    { text: item.text },
+                ],
+            };
+            return content;
+        }
+        else {
+            assert(item.type === 'assistant');
+            const content = {
+                role: 'model',
+                parts: [
+                    ...(item.text ? [{ text: item.text }] : []),
+                    ...item.functionCalls.map((call) => ({
+                        functionCall: {
+                            name: call.name,
+                            args: call.args ?? {},
+                        },
+                    })),
+                ],
+            };
+            return content;
+        }
+    });
+    const req = {
+        contents: messages,
+    };
+    const model = await getGenModel(prompt.find((item) => item.type === 'systemPrompt').systemPrompt, temperature, functionDefs, options.geminiBlockNone, requiredFunctionName, cheap);
+    const result = await model.generateContent(req);
+    // Print token usage
+    const usageMetadata = result.response.usageMetadata;
+    const usage = {
+        inputTokens: usageMetadata.promptTokenCount,
+        outputTokens: usageMetadata.candidatesTokenCount,
+        totalTokens: usageMetadata.totalTokenCount,
+    };
+    printTokenUsageAndCost(usage, 0.000125 / 1000, 0.000375 / 1000);
+    if (result.response.promptFeedback) {
+        console.log('Prompt feedback:');
+        console.log(JSON.stringify(result.response.promptFeedback, null, 2));
+    }
+    if (!result.response.candidates?.length) {
+        console.log('Response:', result);
+        throw new Error('No candidates found');
+    }
+    const functionCalls = result.response.candidates
+        .map((candidate) => candidate.content.parts?.map((part) => part.functionCall))
+        .flat()
+        .filter((functionCall) => !!functionCall)
+        .map((call) => ({ name: call.name, args: call.args }));
+    if (functionCalls.length === 0) {
+        const textResponse = result.response.candidates
+            .map((candidate) => candidate.content.parts?.map((part) => part.text))
+            .flat()
+            .filter((text) => !!text)
+            .join('\n');
+        console.log('No function calls, output text response if it exists:', textResponse);
+    }
+    return processFunctionCalls(functionCalls);
+}
+// A function to get the generative model
+// Modified to accept temperature parameter and cheap flag
+export function getGenModel(systemPrompt, temperature, functionDefs, geminiBlockNone, requiredFunctionName, cheap = false) {
+    // Initialize Vertex with your Cloud project and location
+    const vertex_ai = new VertexAI({});
+    const model = cheap ? 'gemini-1.5-flash-001' : 'gemini-1.5-pro-001';
+    console.log(`Using Vertex AI model: ${model}`);
+    // Instantiate the models
+    return vertex_ai.preview.getGenerativeModel({
+        model: model,
+        generationConfig: {
+            maxOutputTokens: 8192,
+            temperature: temperature,
+            topP: 0.95,
+        },
+        safetySettings: [
+            {
+                category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+            },
+            {
+                category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+            },
+            {
+                category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+            },
+            {
+                category: HarmCategory.HARM_CATEGORY_HARASSMENT,
+                threshold: geminiBlockNone ? HarmBlockThreshold.BLOCK_NONE : HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+            },
+        ],
+        systemInstruction: {
+            role: 'system',
+            parts: [
+                {
+                    text: systemPrompt,
+                },
+            ],
+        },
+        tools: [
+            {
+                functionDeclarations: functionDefs,
+            },
+        ],
+        toolConfig: {
+            functionCallingConfig: {
+                mode: cheap ? undefined : FunctionCallingMode.ANY,
+                ...(!cheap && requiredFunctionName ? { allowedFunctionNames: [requiredFunctionName] } : {}),
+            },
+        },
+    });
+}
+//# sourceMappingURL=vertex-ai.js.map

package/dist/ai-service/vertex-ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex-ai.js","sourceRoot":"","sources":["../../src/ai-service/vertex-ai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EACL,QAAQ,EAGR,YAAY,EACZ,kBAAkB,EAElB,mBAAmB,GACpB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAyC,MAAM,aAAa,CAAC;AAElH,OAAO,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAExE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,MAAoB,EACpB,YAA2B,EAC3B,oBAAmC,EACnC,WAAmB,EACnB,KAAK,GAAG,KAAK,EACb,OAAuB;IAEvB,uEAAuE;IACvE,eAAe,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC;IAEzC,MAAM,QAAQ,GAAc,MAAM;SAC/B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC;SACnE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,MAAM,OAAO,GAAY;gBACvB,IAAI,EAAE,MAAe;gBACrB,KAAK,EAAE;oBACL,GAAG,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;wBACnD,gBAAgB,EAAE;4BAChB,IAAI,EAAE,QAAQ,CAAC,IAAI;4BACnB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE;yBAC7D;qBACF,CAAC,CAAC;oBACH,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBACrC,UAAU,EAAE;4BACV,QAAQ,EAAE,KAAK,CAAC,SAAS;4BACzB,IAAI,EAAE,KAAK,CAAC,SAAS;yBACtB;qBACF,CAAC,CAAC;oBACH,EAAE,IAAI,EAAE,IAAI,CAAC,IAAK,EAAE;iBACrB;aACF,CAAC;YACF,OAAO,OAAO,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;YAClC,MAAM,OAAO,GAAY;gBACvB,IAAI,EAAE,OAAgB;gBACtB,KAAK,EAAE;oBACL,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC3C,GAAG,IAAI,CAAC,aAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;wBACpC,YAAY,EAAE;4BACZ,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;yBACtB;qBACF,CAAC,CAAC;iBACJ;aACF,CAAC;YACF,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,MAAM,GAAG,GAA2B;QAClC,QAAQ,EAAE,QAAQ;KACnB,CAAC;IAEF,MAAM,KAAK,GAAG,MAAM,WAAW,CAC7B,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAE,CAAC,YAAa,EAClE,WAAW,EACX,YAAY,EACZ,OAAO,CAAC,eAAe,EACvB,oBAAoB,EACpB,KAAK,CACN,CAAC;IAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAEhD,oBAAoB;IACpB,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,aAAc,CAAC;IACrD,MAAM,KAAK,GAAG;QACZ,WAAW,EAAE,aAAa,CAAC,gBAAgB;QAC3C,YAAY,EAAE,aAAa,CAAC,oBAAoB;QAChD,WAAW,EAAE,aAAa,CAAC,eAAe;KAC3C,CAAC;IACF,sBAAsB,CAAC,KAAK,EAAE,QAAQ,GAAG,IAAI,EAAE,QAAQ,GAAG,IAAI,CAAC,CAAC;IAEhE,IAAI,MAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;IAED,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU;SAC7C,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;SAC7E,IAAI,EAAE;SACN,MAAM,CAAC,CAAC,YAAY,EAAoD,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC;SAC1F,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAA+B,EAAE,CAAC,CAAC,CAAC;IAEpF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU;aAC5C,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aACrE,IAAI,EAAE;aACN,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aACxC,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,OAAO,CAAC,GAAG,CAAC,uDAAuD,EAAE,YAAY,CAAC,CAAC;IACrF,CAAC;IAED,OAAO,oBAAoB,CAAC,aAAa,CAAC,CAAC;AAC7C,CAAC;AAED,yCAAyC;AACzC,0DAA0D;AAC1D,MAAM,UAAU,WAAW,CACzB,YAAoB,EACpB,WAAmB,EACnB,YAA2B,EAC3B,eAAoC,EACpC,oBAAmC,EACnC,KAAK,GAAG,KAAK;IAEb,yDAAyD;IACzD,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;IAEpE,OAAO,CAAC,GAAG,CAAC,0BAA0B,KAAK,EAAE,CAAC,CAAC;IAE/C,yBAAyB;IACzB,OAAO,SAAS,CAAC,OAAO,CAAC,kBAAkB,CAAC;QAC1C,KAAK,EAAE,KAAK;QACZ,gBAAgB,EAAE;YAChB,eAAe,EAAE,IAAI;YACrB,WAAW,EAAE,WAAW;YACxB,IAAI,EAAE,IAAI;SACX;QACD,cAAc,EAAE;YACd;gBACE,QAAQ,EAAE,YAAY,CAAC,yBAAyB;gBAChD,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,mBAAmB;aACpG;YACD;gBACE,QAAQ,EAAE,YAAY,CAAC,+BAA+B;gBACtD,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,mBAAmB;aACpG;YACD;gBACE,QAAQ,EAAE,YAAY,CAAC,+BAA+B;gBACtD,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,mBAAmB;aACpG;YACD;gBACE,QAAQ,EAAE,YAAY,CAAC,wBAAwB;gBAC/C,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,mBAAmB;aACpG;SACF;QACD,iBAAiB,EAAE;YACjB,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,YAAY;iBACnB;aACF;SACF;QACD,KAAK,EAAE;YACL;gBACE,oBAAoB,EAAE,YAAgD;aACvE;SACF;QACD,UAAU,EAAE;YACV,qBAAqB,EAAE;gBACrB,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG;gBACjD,GAAG,CAAC,CAAC,KAAK,IAAI,oBAAoB,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5F;SACF;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/cli/cli-options.d.ts

@@ -0,0 +1,10 @@
+interface CliOption {
+    name: string;
+    description: string;
+}
+declare const cliOptions: CliOption[];
+/**
+ * Print the help message with all available CLI options
+ */
+export declare function printHelpMessage(): void;
+export { cliOptions };