genaicode 0.8.6 → 0.9.0

package/dist/src/ai-service/unescape-function-call.test.js

@@ -1,18 +0,0 @@
-import { describe, it, expect } from 'vitest';
-import { unescapeFunctionCall } from './unescape-function-call.js';
-describe('unescapeFunctionCall', () => {
-    it('unescapes double escaped code', () => {
-        expect(unescapeFunctionCall({
-            name: 'test',
-            args: {
-                content: `console.log(\\"hello world\\");\\nconsole.log(\\'helloooo\\');`,
-            },
-        })).toEqual({
-            name: 'test',
-            args: {
-                content: `console.log("hello world");\nconsole.log('helloooo');`,
-            },
-        });
-    });
-});
-//# sourceMappingURL=unescape-function-call.test.js.map

package/dist/src/ai-service/unescape-function-call.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"unescape-function-call.test.js","sourceRoot":"","sources":["../../../src/ai-service/unescape-function-call.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAEnE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,CACJ,oBAAoB,CAAC;YACnB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE;gBACJ,OAAO,EAAE,gEAAgE;aAC1E;SACF,CAAC,CACH,CAAC,OAAO,CAAC;YACR,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE;gBACJ,OAAO,EAAE,uDAAuD;aACjE;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -1,5 +0,0 @@
-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/src/ai-service/vertex-ai-claude.js

@@ -1,104 +0,0 @@
-import { AnthropicVertex } from '@anthropic-ai/vertex-sdk';
-import assert from 'node:assert';
-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 ?? {},
-                    })),
-                ],
-            };
-        }
-    })
-        .filter((message) => !!message);
-    const model = cheap ? 'claude-3-haiku@20240307' : 'claude-3-5-sonnet@20240620';
-    console.log(`Using Vertex AI Claude model: ${model}`);
-    assert(process.env.GOOGLE_CLOUD_PROJECT, 'GOOGLE_CLOUD_PROJECT environment variable is not set');
-    assert(process.env.GOOGLE_CLOUD_REGION, 'GOOGLE_CLOUD_REGION environment variable is not set');
-    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({
-        aiService: 'vertex-ai-claude',
-        usage,
-        inputCostPerToken: 3 / 1000 / 1000,
-        outputCostPerToken: 15 / 1000 / 1000,
-        cheap,
-    });
-    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, functionDefs);
-}
-//# sourceMappingURL=vertex-ai-claude.js.map

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

@@ -1 +0,0 @@
-{"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,MAAM,MAAM,aAAa,CAAC;AACjC,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,GAAmB,MAAM;SACpC,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,IAAI,EAAE,EAAE;iBACjD;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,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;wBAC3C,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;SACD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAElC,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,4BAA4B,CAAC;IAC/E,OAAO,CAAC,GAAG,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,sDAAsD,CAAC,CAAC;IACjG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,qDAAqD,CAAC,CAAC;IAE/F,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;QACrB,SAAS,EAAE,kBAAkB;QAC7B,KAAK;QACL,iBAAiB,EAAE,CAAC,GAAG,IAAI,GAAG,IAAI;QAClC,kBAAkB,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;QACpC,KAAK;KACN,CAAC,CAAC;IAEH,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,EAAE,YAAY,CAAC,CAAC;AAC3D,CAAC"}

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

@@ -1,14 +0,0 @@
-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/src/ai-service/vertex-ai-imagen.js

@@ -1,77 +0,0 @@
-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/src/ai-service/vertex-ai-imagen.js.map

@@ -1 +0,0 @@
-{"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/src/ai-service/vertex-ai.d.ts

@@ -1,6 +0,0 @@
-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[]>;

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

@@ -1,202 +0,0 @@
-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';
-import { modelOverrides } from '../main/config.js';
-import { unescapeFunctionCall } from './unescape-function-call.js';
-import { disableVertexUnescape } from '../cli/cli-params.js';
-/**
- * This function generates content using the 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({
-        aiService: 'vertex-ai',
-        usage,
-        inputCostPerToken: 0.000125 / 1000,
-        outputCostPerToken: 0.000375 / 1000,
-        cheap,
-    });
-    if (result.response.promptFeedback) {
-        console.log('Prompt feedback:');
-        console.log(JSON.stringify(result.response.promptFeedback, null, 2));
-    }
-    if (!result.response.candidates?.length) {
-        console.log('Response:', result);
-        throw new Error('No candidates found');
-    }
-    const functionCalls = result.response.candidates
-        .map((candidate) => candidate.content.parts?.map((part) => part.functionCall))
-        .flat()
-        .filter((functionCall) => !!functionCall)
-        .map((call) => ({ name: call.name, args: call.args }))
-        .map(disableVertexUnescape ? (call) => call : unescapeFunctionCall);
-    if (functionCalls.length === 0) {
-        const textResponse = result.response.candidates
-            .map((candidate) => candidate.content.parts?.map((part) => part.text))
-            .flat()
-            .filter((text) => !!text)[0] ??
-            result.response.candidates.map((candidate) => 
-            // @ts-expect-error: FinishReason type does not have this error
-            candidate.finishReason === 'MALFORMED_FUNCTION_CALL' ? candidate.finishMessage : '')[0];
-        const recoveredFunctionCall = textResponse &&
-            (await recoverFunctionCall(textResponse, functionCalls, functionDefs.find((def) => def.name === requiredFunctionName)));
-        if (!recoveredFunctionCall) {
-            console.log('No function calls, output text response if it exists:', textResponse);
-        }
-        else {
-            console.log('Recovered function call.');
-        }
-    }
-    return processFunctionCalls(functionCalls, functionDefs);
-}
-// A function to get the generative model
-// Modified to accept temperature parameter and cheap flag
-function getGenModel(systemPrompt, temperature, functionDefs, geminiBlockNone, requiredFunctionName, cheap = false) {
-    console.log('Recovering function call');
-    // Initialize Vertex with your Cloud project and location
-    const vertex_ai = new VertexAI({});
-    const defaultModel = cheap ? 'gemini-1.5-flash-002' : 'gemini-1.5-pro-002';
-    const model = cheap
-        ? (modelOverrides.vertexAi?.cheap ?? defaultModel)
-        : (modelOverrides.vertexAi?.default ?? defaultModel);
-    console.log(`Using Vertex AI model: ${model}`);
-    assert(process.env.GOOGLE_CLOUD_PROJECT, 'GOOGLE_CLOUD_PROJECT environment variable is not set');
-    // 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,
-                },
-            ],
-        },
-        ...((functionDefs?.length ?? 0) > 0
-            ? {
-                tools: [
-                    {
-                        functionDeclarations: functionDefs,
-                    },
-                ],
-                toolConfig: {
-                    functionCallingConfig: {
-                        mode: FunctionCallingMode.ANY,
-                        ...(requiredFunctionName ? { allowedFunctionNames: [requiredFunctionName] } : {}),
-                    },
-                },
-            }
-            : {}),
-    });
-}
-async function recoverFunctionCall(textResponse, functionCalls, functionDef) {
-    // @ts-expect-error: "object" !== "OBJECT"
-    const schema = functionDef.parameters;
-    const result = await getGenModel('Your role is read the text below and if possible, return it in the desired format.', 0.2, undefined, undefined, null, false).generateContent({
-        contents: [
-            {
-                role: 'user',
-                parts: [
-                    {
-                        text: textResponse,
-                    },
-                ],
-            },
-        ],
-        generationConfig: {
-            responseMimeType: 'application/json',
-            responseSchema: schema,
-        },
-    });
-    try {
-        const text = result.response.candidates?.[0].content.parts[0].text;
-        if (!text) {
-            return false;
-        }
-        const parsed = JSON.parse(text);
-        functionCalls.push(unescapeFunctionCall({ name: functionDef.name, args: parsed }));
-        return true;
-    }
-    catch {
-        return false;
-    }
-}
-//# sourceMappingURL=vertex-ai.js.map

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

@@ -1 +0,0 @@
-{"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;AACxE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAE7D;;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,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;wBAC3C,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;QACrB,SAAS,EAAE,WAAW;QACtB,KAAK;QACL,iBAAiB,EAAE,QAAQ,GAAG,IAAI;QAClC,kBAAkB,EAAE,QAAQ,GAAG,IAAI;QACnC,KAAK;KACN,CAAC,CAAC;IAEH,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;SAChF,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC;IAEtE,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,YAAY,GAChB,MAAM,CAAC,QAAQ,CAAC,UAAU;aACvB,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,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;YAC3C,+DAA+D;YAC/D,SAAS,CAAC,YAAY,KAAK,yBAAyB,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CACpF,CAAC,CAAC,CAAC,CAAC;QAEP,MAAM,qBAAqB,GACzB,YAAY;YACZ,CAAC,MAAM,mBAAmB,CACxB,YAAY,EACZ,aAAa,EACb,YAAY,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,oBAAoB,CAAE,CAC/D,CAAC,CAAC;QACL,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC3B,OAAO,CAAC,GAAG,CAAC,uDAAuD,EAAE,YAAY,CAAC,CAAC;QACrF,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,OAAO,oBAAoB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;AAC3D,CAAC;AAED,yCAAyC;AACzC,0DAA0D;AAC1D,SAAS,WAAW,CAClB,YAAoB,EACpB,WAAmB,EACnB,YAAuC,EACvC,eAAoC,EACpC,oBAAmC,EACnC,KAAK,GAAG,KAAK;IAEb,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IACxC,yDAAyD;IACzD,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;IACnC,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;IAC3E,MAAM,KAAK,GAAG,KAAK;QACjB,CAAC,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,IAAI,YAAY,CAAC;QAClD,CAAC,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,IAAI,YAAY,CAAC,CAAC;IAEvD,OAAO,CAAC,GAAG,CAAC,0BAA0B,KAAK,EAAE,CAAC,CAAC;IAC/C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,sDAAsD,CAAC,CAAC;IAEjG,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,GAAG,CAAC,CAAC,YAAY,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC;YACjC,CAAC,CAAC;gBACE,KAAK,EAAE;oBACL;wBACE,oBAAoB,EAAE,YAAgD;qBACvE;iBACF;gBACD,UAAU,EAAE;oBACV,qBAAqB,EAAE;wBACrB,IAAI,EAAE,mBAAmB,CAAC,GAAG;wBAC7B,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAClF;iBACF;aACF;YACH,CAAC,CAAC,EAAE,CAAC;KACR,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,mBAAmB,CAChC,YAAoB,EACpB,aAA6B,EAC7B,WAAwB;IAExB,0CAA0C;IAC1C,MAAM,MAAM,GAAmB,WAAW,CAAC,UAAU,CAAC;IACtD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,oFAAoF,EACpF,GAAG,EACH,SAAS,EACT,SAAS,EACT,IAAI,EACJ,KAAK,CACN,CAAC,eAAe,CAAC;QAChB,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,MAAe;gBACrB,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,YAAY;qBACnB;iBACF;aACF;SACF;QACD,gBAAgB,EAAE;YAChB,gBAAgB,EAAE,kBAAkB;YACpC,cAAc,EAAE,MAAM;SACvB;KACF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACnE,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChC,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACnF,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

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

@@ -1,10 +0,0 @@
-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 };

package/dist/src/cli/cli-options.js

@@ -1,132 +0,0 @@
-const cliOptions = [
-    {
-        name: '--help',
-        description: 'Display this help message.',
-    },
-    {
-        name: '--dry-run',
-        description: 'Run the codegen script without updating the source code.',
-    },
-    {
-        name: '--consider-all-files',
-        description: "Consider all files for code generation, even if they don't contain the @" + 'CODEGEN comments.',
-    },
-    {
-        name: '--disallow-file-create',
-        description: 'Disallow the codegen script to create new files (file creation is allowed by default).',
-    },
-    {
-        name: '--disallow-file-delete',
-        description: 'Disallow the codegen script to delete files (file deletion is allowed by default).',
-    },
-    {
-        name: '--disallow-directory-create',
-        description: 'Disallow codegen script to create directories (directory creation is allowed by default).',
-    },
-    {
-        name: '--disallow-file-move',
-        description: 'Disallow the codegen script to move files (file moving is allowed by default).',
-    },
-    {
-        name: '--ai-service=<service>',
-        description: 'Specify the AI service to use for code generation. Available options include vertex-ai, ai-studio, chat-gpt, anthropic, vertex-ai-claude, and any additional services loaded from plugins.',
-    },
-    {
-        name: '--explicit-prompt=<prompt>',
-        description: 'An explicit prompt to use for code generation.',
-    },
-    {
-        name: '--task-file=<file>',
-        description: 'Specifies a file with a task description for code generation.',
-    },
-    {
-        name: '--dependency-tree',
-        description: 'Limit the scope of codegen only to files marked with @' + 'CODEGEN and their dependencies.',
-    },
-    {
-        name: '--verbose-prompt',
-        description: 'Print the prompt used for code generation.',
-    },
-    {
-        name: '--disable-explanations',
-        description: 'Disable requirement of explanations for all code generation operations.',
-    },
-    {
-        name: '--disable-context-optimization',
-        description: 'Disable the optimization that uses context paths for more efficient code generation.',
-    },
-    {
-        name: '--gemini-block-none',
-        description: 'Disable safety settings for Gemini Pro model (requires whitelisted Cloud project).',
-    },
-    {
-        name: '--disable-initial-lint',
-        description: 'Skip the initial lint check before running the code generation process.',
-    },
-    {
-        name: '--temperature=<value>',
-        description: 'Set the temperature parameter for the AI model (default: 0.7).',
-    },
-    {
-        name: '--vision',
-        description: 'Enable vision capabilities for processing image inputs. This option allows the tool to analyze and generate code based on image content when used with compatible AI models.',
-    },
-    {
-        name: '--imagen=<service>',
-        description: 'Enable image generation functionality and specify the service to use (either "vertex-ai" or "dall-e").',
-    },
-    {
-        name: '--cheap',
-        description: 'Switch to cheaper models in AI services for content and image generation.',
-    },
-    {
-        name: '--content-mask=<path>',
-        description: 'Apply a content mask to limit the initial source code files included in the request. The value should be a prefix of the path relative to rootDir.',
-    },
-    {
-        name: '--ignore-pattern=<pattern>',
-        description: 'Specify a pattern of files to ignore during the initial source code fetching.',
-    },
-    {
-        name: '--disable-cache',
-        description: 'Disable caching for the application.',
-    },
-    {
-        name: '--disable-ask-question',
-        description: 'Disable the question-asking feature. By default, the assistant can ask questions for clarification during the code generation process.',
-    },
-    {
-        name: '--disable-vertex-unescape',
-        description: 'Disable the unescaping of special characters in Vertex AI responses.',
-    },
-    {
-        name: '--disable-self-reflection',
-        description: 'Disable the self-reflection feature of the AI model.',
-    },
-    {
-        name: '--disable-conversation-summary',
-        description: 'Disable the summary generation feature of the AI model.',
-    },
-    {
-        name: '--ui',
-        description: 'Run the tool as a web server, and use it via browser.',
-    },
-    {
-        name: '--ui-port=<port>',
-        description: 'Specify the port for the web server when using --ui (default: 1337).',
-    },
-];
-/**
- * Print the help message with all available CLI options
- */
-export function printHelpMessage() {
-    console.log('GenAIcode - AI-powered code generation tool');
-    console.log('\nUsage: npx genaicode [options]');
-    console.log('\nOptions:');
-    cliOptions.forEach((option) => {
-        console.log(`  ${option.name.padEnd(30)} ${option.description}`);
-    });
-    console.log('\nFor more information, please refer to the project home page: https://github.com/gtanczyk/genaicode/');
-}
-export { cliOptions };
-//# sourceMappingURL=cli-options.js.map

package/dist/src/cli/cli-options.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"cli-options.js","sourceRoot":"","sources":["../../../src/cli/cli-options.ts"],"names":[],"mappings":"AAMA,MAAM,UAAU,GAAgB;IAC9B;QACE,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,4BAA4B;KAC1C;IACD;QACE,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,0DAA0D;KACxE;IACD;QACE,IAAI,EAAE,sBAAsB;QAC5B,WAAW,EAAE,0EAA0E,GAAG,mBAAmB;KAC9G;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EAAE,wFAAwF;KACtG;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EAAE,oFAAoF;KAClG;IACD;QACE,IAAI,EAAE,6BAA6B;QACnC,WAAW,EAAE,2FAA2F;KACzG;IACD;QACE,IAAI,EAAE,sBAAsB;QAC5B,WAAW,EAAE,gFAAgF;KAC9F;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACT,4LAA4L;KAC/L;IACD;QACE,IAAI,EAAE,4BAA4B;QAClC,WAAW,EAAE,gDAAgD;KAC9D;IACD;QACE,IAAI,EAAE,oBAAoB;QAC1B,WAAW,EAAE,+DAA+D;KAC7E;IACD;QACE,IAAI,EAAE,mBAAmB;QACzB,WAAW,EAAE,wDAAwD,GAAG,iCAAiC;KAC1G;IACD;QACE,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE,4CAA4C;KAC1D;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EAAE,yEAAyE;KACvF;IACD;QACE,IAAI,EAAE,gCAAgC;QACtC,WAAW,EAAE,sFAAsF;KACpG;IACD;QACE,IAAI,EAAE,qBAAqB;QAC3B,WAAW,EAAE,oFAAoF;KAClG;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EAAE,yEAAyE;KACvF;IACD;QACE,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EAAE,gEAAgE;KAC9E;IACD;QACE,IAAI,EAAE,UAAU;QAChB,WAAW,EACT,8KAA8K;KACjL;IACD;QACE,IAAI,EAAE,oBAAoB;QAC1B,WAAW,EACT,wGAAwG;KAC3G;IACD;QACE,IAAI,EAAE,SAAS;QACf,WAAW,EAAE,2EAA2E;KACzF;IACD;QACE,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EACT,oJAAoJ;KACvJ;IACD;QACE,IAAI,EAAE,4BAA4B;QAClC,WAAW,EAAE,+EAA+E;KAC7F;IACD;QACE,IAAI,EAAE,iBAAiB;QACvB,WAAW,EAAE,sCAAsC;KACpD;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACT,wIAAwI;KAC3I;IACD;QACE,IAAI,EAAE,2BAA2B;QACjC,WAAW,EAAE,sEAAsE;KACpF;IACD;QACE,IAAI,EAAE,2BAA2B;QACjC,WAAW,EAAE,sDAAsD;KACpE;IACD;QACE,IAAI,EAAE,gCAAgC;QACtC,WAAW,EAAE,yDAAyD;KACvE;IACD;QACE,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,uDAAuD;KACrE;IACD;QACE,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE,sEAAsE;KACpF;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,gBAAgB;IAC9B,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAE1B,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QAC5B,OAAO,CAAC,GAAG,CAAC,KAAK,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,uGAAuG,CAAC,CAAC;AACvH,CAAC;AAED,OAAO,EAAE,UAAU,EAAE,CAAC"}

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

@@ -1 +0,0 @@
-export {};

package/dist/src/cli/cli-options.test.js

@@ -1,34 +0,0 @@
-import { describe, it, expect, vi } from 'vitest';
-import { printHelpMessage, cliOptions } from './cli-options.js';
-describe('CLI Options', () => {
-    describe('printHelpMessage', () => {
-        it('should print the help message with all CLI options', () => {
-            const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => { });
-            printHelpMessage();
-            expect(consoleSpy).toHaveBeenCalledWith('GenAIcode - AI-powered code generation tool');
-            expect(consoleSpy).toHaveBeenCalledWith('\nUsage: npx genaicode [options]');
-            expect(consoleSpy).toHaveBeenCalledWith('\nOptions:');
-            cliOptions.forEach((option) => {
-                expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining(option.name));
-                expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining(option.description));
-            });
-            expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining('For more information'));
-            consoleSpy.mockRestore();
-        });
-        it('should include the --imagen option in the help message', () => {
-            const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => { });
-            printHelpMessage();
-            expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining('--imagen=<service>'));
-            expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining('Enable image generation functionality and specify the service to use (either "vertex-ai" or "dall-e").'));
-            consoleSpy.mockRestore();
-        });
-    });
-    describe('cliOptions', () => {
-        it('should include the --imagen option', () => {
-            const imagenOption = cliOptions.find((option) => option.name === '--imagen=<service>');
-            expect(imagenOption).toBeDefined();
-            expect(imagenOption.description).toBe('Enable image generation functionality and specify the service to use (either "vertex-ai" or "dall-e").');
-        });
-    });
-});
-//# sourceMappingURL=cli-options.test.js.map