genaicode 0.5.1 → 0.6.1

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

@@ -0,0 +1,202 @@
+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

@@ -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;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

@@ -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 };

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

@@ -0,0 +1,132 @@
+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

@@ -0,0 +1 @@
+{"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

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

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

@@ -0,0 +1,34 @@
+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

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

@@ -0,0 +1 @@
+{"version":3,"file":"cli-options.test.js","sourceRoot":"","sources":["../../../src/cli/cli-options.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEhE,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEzE,gBAAgB,EAAE,CAAC;YAEnB,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,6CAA6C,CAAC,CAAC;YACvF,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,kCAAkC,CAAC,CAAC;YAC5E,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;YAEtD,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC5B,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC9E,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;YACvF,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,CAAC,CAAC;YAEzF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEzE,gBAAgB,EAAE,CAAC;YAEnB,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,CAAC,CAAC;YACvF,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CACrB,wGAAwG,CACzG,CACF,CAAC;YAEF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,YAAY,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,oBAAoB,CAAC,CAAC;YACvF,MAAM,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,CAAC,YAAa,CAAC,WAAW,CAAC,CAAC,IAAI,CACpC,wGAAwG,CACzG,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,32 @@
+export declare const dryRun: boolean;
+export declare const considerAllFiles: boolean;
+export declare const allowFileCreate: boolean;
+export declare const allowFileDelete: boolean;
+export declare const allowDirectoryCreate: boolean;
+export declare const allowFileMove: boolean;
+export declare let aiService: string | undefined;
+export declare const dependencyTree: boolean;
+export declare const verbosePrompt: boolean;
+export declare const disableCache: boolean;
+export declare const explicitPrompt: string | undefined;
+export declare const disableContextOptimization: boolean;
+export declare let taskFile: string | undefined;
+export declare const disableExplanations: boolean;
+export declare const geminiBlockNone: boolean;
+export declare const disableInitialLint: boolean;
+export declare const vision: boolean;
+export declare const imagen: "dall-e" | "vertex-ai" | undefined;
+export declare const cheap: boolean;
+export declare const askQuestion: boolean;
+export declare const interactive: boolean;
+export declare const ui: boolean;
+export declare const disableSelfReflection: boolean;
+export declare const disableConversationSummary: boolean;
+export declare const helpRequested: boolean;
+export declare const temperature: number;
+export declare const contentMask: string | undefined;
+export declare const ignorePatterns: string[];
+export declare const disableAiServiceFallback: boolean;
+export declare const disableVertexUnescape: boolean;
+export declare const disableHistory: boolean;
+export declare const uiPort: number;

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

@@ -0,0 +1,106 @@
+import fs from 'fs';
+import path from 'path';
+import { serviceAutoDetect } from './service-autodetect.js';
+const params = process.argv.slice(2);
+export const dryRun = params.includes('--dry-run');
+export const considerAllFiles = params.includes('--consider-all-files');
+export const allowFileCreate = !params.includes('--disallow-file-create');
+export const allowFileDelete = !params.includes('--disallow-file-delete');
+export const allowDirectoryCreate = !params.includes('--disallow-directory-create');
+export const allowFileMove = !params.includes('--disallow-file-move');
+export let aiService = params.find((param) => param.startsWith('--ai-service='))?.split('=')[1];
+export const dependencyTree = params.includes('--dependency-tree');
+export const verbosePrompt = params.includes('--verbose-prompt');
+export const disableCache = params.includes('--disable-cache');
+export const explicitPrompt = params.find((param) => param.startsWith('--explicit-prompt'))?.split('=')[1];
+export const disableContextOptimization = params.includes('--disable-context-optimization');
+export let taskFile = params.find((param) => param.startsWith('--task-file'))?.split('=')[1];
+export const disableExplanations = params.includes('--disable-explanations');
+export const geminiBlockNone = params.includes('--gemini-block-none');
+export const disableInitialLint = params.includes('--disable-initial-lint');
+export const vision = params.includes('--vision');
+export const imagen = params.find((param) => param.startsWith('--imagen'))?.split('=')[1];
+export const cheap = params.includes('--cheap');
+export const askQuestion = !params.includes('--disable-ask-question'); // Default enabled
+export const interactive = params.includes('--interactive');
+export const ui = params.includes('--ui');
+export const disableSelfReflection = params.includes('--disable-self-reflection');
+export const disableConversationSummary = params.includes('--disable-conversation-summary');
+// Add support for --help option
+export const helpRequested = params.includes('--help');
+// Temperature parameter
+export const temperature = parseFloat(params.find((param) => param.startsWith('--temperature='))?.split('=')[1] || '0.7'); // Default temperature value: 0.7
+export const contentMask = params.find((param) => param.startsWith('--content-mask='))?.split('=')[1] || undefined;
+export const ignorePatterns = params
+    .filter((param) => param.startsWith('--ignore-pattern='))
+    .map((param) => param.split('=')[1]);
+export const disableAiServiceFallback = params.includes('--disable-ai-service-fallback');
+export const disableVertexUnescape = params.includes('--disable-vertex-unescape');
+export const disableHistory = params.includes('--disable-history');
+// New variable to store the ui-port value
+export const uiPort = parseInt(params.find((param) => param.startsWith('--ui-port='))?.split('=')[1] || '1337', 10);
+if (taskFile) {
+    if (explicitPrompt) {
+        throw new Error('The --task-file option is exclusive with the --explicit-prompt option');
+    }
+    if (!fs.existsSync(taskFile)) {
+        throw new Error(`The task file ${taskFile} does not exist`);
+    }
+    if (!path.isAbsolute(taskFile)) {
+        taskFile = path.join(process.cwd(), taskFile);
+    }
+}
+if (considerAllFiles && dependencyTree) {
+    throw new Error('--consider-all-files and --dependency-tree are exclusive.');
+}
+if (!aiService && !helpRequested) {
+    const detected = serviceAutoDetect();
+    if (detected) {
+        console.log(`Autodetected --ai-service=${detected}`);
+        aiService = detected;
+    }
+    else {
+        console.warn('Missing --ai-service option. Please specify which AI service should be used.');
+    }
+}
+if (temperature) {
+    console.log(`Temperature value: ${temperature}`);
+}
+if (imagen) {
+    console.log('Image generation functionality enabled');
+}
+if (cheap) {
+    console.log('Cheaper AI models will be used for content and image generation');
+}
+if (contentMask) {
+    console.log(`Content mask: ${contentMask}`);
+}
+if (ignorePatterns.length > 0) {
+    console.log(`Ignore pattern: ${ignorePatterns.join(', ')}`);
+}
+if (interactive && ui) {
+    throw new Error('--ui and --interactive are exclusive.');
+}
+if (askQuestion && (interactive || ui)) {
+    console.log('Assistant can ask questions to the user');
+}
+else {
+    console.log('Assistant will not ask questions to the user');
+}
+if (interactive) {
+    console.log('Interactive mode enabled');
+}
+if (disableAiServiceFallback) {
+    console.log('AI service fallback is disabled');
+}
+else {
+    console.log('AI service fallback is enabled (default)');
+}
+if (disableSelfReflection) {
+    console.log('Self-reflection mechanism is disabled');
+}
+// Log the UI port
+if (ui) {
+    console.log(`UI mode enabled on port: ${uiPort}`);
+}
+//# sourceMappingURL=cli-params.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cli-params.js","sourceRoot":"","sources":["../../../src/cli/cli-params.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AAErC,MAAM,CAAC,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACnD,MAAM,CAAC,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;AACxE,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;AAC1E,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;AAC1E,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC;AACpF,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;AAEtE,MAAM,CAAC,IAAI,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AAEhG,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;AACnE,MAAM,CAAC,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;AACjE,MAAM,CAAC,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;AAC/D,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AAC3G,MAAM,CAAC,MAAM,0BAA0B,GAAG,MAAM,CAAC,QAAQ,CAAC,gCAAgC,CAAC,CAAC;AAC5F,MAAM,CAAC,IAAI,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AAC7F,MAAM,CAAC,MAAM,mBAAmB,GAAG,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;AAC7E,MAAM,CAAC,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;AACtE,MAAM,CAAC,MAAM,kBAAkB,GAAG,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;AAC5E,MAAM,CAAC,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;AAClD,MAAM,CAAC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAG3E,CAAC;AACd,MAAM,CAAC,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;AAChD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,kBAAkB;AACzF,MAAM,CAAC,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;AAC5D,MAAM,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AAC1C,MAAM,CAAC,MAAM,qBAAqB,GAAG,MAAM,CAAC,QAAQ,CAAC,2BAA2B,CAAC,CAAC;AAClF,MAAM,CAAC,MAAM,0BAA0B,GAAG,MAAM,CAAC,QAAQ,CAAC,gCAAgC,CAAC,CAAC;AAE5F,gCAAgC;AAChC,MAAM,CAAC,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AAEvD,wBAAwB;AACxB,MAAM,CAAC,MAAM,WAAW,GAAG,UAAU,CACnC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CACnF,CAAC,CAAC,iCAAiC;AAEpC,MAAM,CAAC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;AAEnH,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM;KACjC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;KACxD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAEvC,MAAM,CAAC,MAAM,wBAAwB,GAAG,MAAM,CAAC,QAAQ,CAAC,+BAA+B,CAAC,CAAC;AACzF,MAAM,CAAC,MAAM,qBAAqB,GAAG,MAAM,CAAC,QAAQ,CAAC,2BAA2B,CAAC,CAAC;AAClF,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;AAEnE,0CAA0C;AAC1C,MAAM,CAAC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC;AAEpH,IAAI,QAAQ,EAAE,CAAC;IACb,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAC;IAC3F,CAAC;IACD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,iBAAiB,QAAQ,iBAAiB,CAAC,CAAC;IAC9D,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/B,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC;AACH,CAAC;AAED,IAAI,gBAAgB,IAAI,cAAc,EAAE,CAAC;IACvC,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;AAC/E,CAAC;AAED,IAAI,CAAC,SAAS,IAAI,CAAC,aAAa,EAAE,CAAC;IACjC,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;IACrC,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,CAAC,GAAG,CAAC,6BAA6B,QAAQ,EAAE,CAAC,CAAC;QACrD,SAAS,GAAG,QAAQ,CAAC;IACvB,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,IAAI,CAAC,8EAA8E,CAAC,CAAC;IAC/F,CAAC;AACH,CAAC;AAED,IAAI,WAAW,EAAE,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,sBAAsB,WAAW,EAAE,CAAC,CAAC;AACnD,CAAC;AAED,IAAI,MAAM,EAAE,CAAC;IACX,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;AACxD,CAAC;AAED,IAAI,KAAK,EAAE,CAAC;IACV,OAAO,CAAC,GAAG,CAAC,iEAAiE,CAAC,CAAC;AACjF,CAAC;AAED,IAAI,WAAW,EAAE,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,iBAAiB,WAAW,EAAE,CAAC,CAAC;AAC9C,CAAC;AAED,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,mBAAmB,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC9D,CAAC;AAED,IAAI,WAAW,IAAI,EAAE,EAAE,CAAC;IACtB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;AAC3D,CAAC;AAED,IAAI,WAAW,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE,CAAC;IACvC,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;AACzD,CAAC;KAAM,CAAC;IACN,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;AAC9D,CAAC;AAED,IAAI,WAAW,EAAE,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;AAC1C,CAAC;AAED,IAAI,wBAAwB,EAAE,CAAC;IAC7B,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;AACjD,CAAC;KAAM,CAAC;IACN,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;AAC1D,CAAC;AAED,IAAI,qBAAqB,EAAE,CAAC;IAC1B,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;AACvD,CAAC;AAED,kBAAkB;AAClB,IAAI,EAAE,EAAE,CAAC;IACP,OAAO,CAAC,GAAG,CAAC,4BAA4B,MAAM,EAAE,CAAC,CAAC;AACpD,CAAC"}

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

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

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

@@ -0,0 +1,38 @@
+import { describe, it, expect, afterEach } from 'vitest';
+import { serviceAutoDetect } from './service-autodetect.js';
+describe('serviceAutoDetect', () => {
+    afterEach(() => {
+        // Clean up environment variables after each test
+        delete process.env.ANTHROPIC_API_KEY;
+        delete process.env.OPENAI_API_KEY;
+        delete process.env.GOOGLE_CLOUD_PROJECT;
+        delete process.env.API_KEY;
+    });
+    it('should return "anthropic" when ANTHROPIC_API_KEY is set', () => {
+        process.env.ANTHROPIC_API_KEY = 'test-key';
+        expect(serviceAutoDetect()).toBe('anthropic');
+    });
+    it('should return "chat-gpt" when OPENAI_API_KEY is set', () => {
+        process.env.OPENAI_API_KEY = 'test-key';
+        expect(serviceAutoDetect()).toBe('chat-gpt');
+    });
+    it('should return "vertex-ai" when GOOGLE_CLOUD_PROJECT is set', () => {
+        process.env.GOOGLE_CLOUD_PROJECT = 'test-project';
+        expect(serviceAutoDetect()).toBe('vertex-ai');
+    });
+    it('should return null when no service is configured', () => {
+        expect(serviceAutoDetect()).toBeNull();
+    });
+    it('should prioritize anthropic over other services', () => {
+        process.env.ANTHROPIC_API_KEY = 'test-key';
+        process.env.OPENAI_API_KEY = 'test-key';
+        process.env.GOOGLE_CLOUD_PROJECT = 'test-project';
+        expect(serviceAutoDetect()).toBe('anthropic');
+    });
+    it('should prioritize chat-gpt over vertex-ai', () => {
+        process.env.OPENAI_API_KEY = 'test-key';
+        process.env.GOOGLE_CLOUD_PROJECT = 'test-project';
+        expect(serviceAutoDetect()).toBe('chat-gpt');
+    });
+});
+//# sourceMappingURL=cli-params.test.js.map

package/dist/src/cli/cli-params.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli-params.test.js","sourceRoot":"","sources":["../../../src/cli/cli-params.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,SAAS,CAAC,GAAG,EAAE;QACb,iDAAiD;QACjD,OAAO,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QACrC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClC,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;QACxC,OAAO,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,UAAU,CAAC;QAC3C,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;QACxC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,cAAc,CAAC;QAClD,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,UAAU,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,cAAc,CAAC;QAClD,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,cAAc,CAAC;QAClD,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/cli/service-autodetect.d.ts

@@ -0,0 +1,2 @@
+/** Detects if one of ai services is configured */
+export declare function serviceAutoDetect(): string | null;

package/dist/src/cli/service-autodetect.js

@@ -0,0 +1,17 @@
+/** Detects if one of ai services is configured */
+export function serviceAutoDetect() {
+    if (process.env.ANTHROPIC_API_KEY) {
+        return 'anthropic';
+    }
+    else if (process.env.OPENAI_API_KEY) {
+        return 'chat-gpt';
+    }
+    else if (process.env.GOOGLE_CLOUD_PROJECT) {
+        return 'vertex-ai';
+    }
+    else if (process.env.API_KEY) {
+        return 'ai-studio';
+    }
+    return null;
+}
+//# sourceMappingURL=service-autodetect.js.map

package/dist/src/cli/service-autodetect.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service-autodetect.js","sourceRoot":"","sources":["../../../src/cli/service-autodetect.ts"],"names":[],"mappings":"AAAA,kDAAkD;AAClD,MAAM,UAAU,iBAAiB;IAC/B,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;QAClC,OAAO,WAAW,CAAC;IACrB,CAAC;SAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QACtC,OAAO,UAAU,CAAC;IACpB,CAAC;SAAM,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC;QAC5C,OAAO,WAAW,CAAC;IACrB,CAAC;SAAM,IAAI,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC/B,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/src/cli/service-autodetect.test.d.ts

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