@juspay/neurolink
Version:
Universal AI Development Platform with working MCP integration, multi-provider support, voice (TTS/STT/realtime), and professional CLI. 58+ external MCP servers discoverable, multimodal file processing, RAG pipelines. Build, test, and deploy AI applicatio
340 lines • 14.6 kB
JavaScript
#!/usr/bin/env node
/**
* Google AI Studio Setup Command
*
* Simple setup for Google AI Studio (Google AI) integration:
* - GOOGLE_AI_API_KEY (required)
* - GOOGLE_AI_MODEL (optional, with recommended choices)
*
* Follows the same UX patterns as setup-openai and setup-gcp
*/
import path from "path";
import inquirer from "inquirer";
import chalk from "chalk";
import ora from "ora";
import { logger } from "../../lib/utils/logger.js";
import { GoogleAIModels } from "../../lib/constants/enums.js";
import { updateEnvFile as updateEnvFileManager, displayEnvUpdateSummary, } from "../utils/envManager.js";
import { getTopModelChoices } from "../../lib/utils/modelChoices.js";
import { AIProviderName, } from "../../lib/types/index.js";
import { maskCredential } from "../utils/maskCredential.js";
/**
* Get the runtime default model that matches the provider implementation
*/
function getRuntimeDefaultModel() {
return process.env.GOOGLE_AI_MODEL || GoogleAIModels.GEMINI_2_5_FLASH;
}
export async function handleGoogleAISetup(argv) {
try {
const options = {
checkOnly: argv.check || false,
interactive: !argv.nonInteractive,
};
logger.always(chalk.blue("🔍 Checking Google AI Studio configuration..."));
// Step 1: Check for existing configuration
const hasApiKey = !!(process.env.GOOGLE_AI_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY);
const hasModel = !!process.env.GOOGLE_AI_MODEL;
const currentApiKey = process.env.GOOGLE_AI_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
// Display current status
displayCurrentStatus(hasApiKey, hasModel);
// Check-only mode - show status and exit
if (options.checkOnly) {
if (hasApiKey && currentApiKey) {
logger.always(chalk.green("✅ Google AI Studio setup complete"));
logger.always(` API Key: ${maskCredential(currentApiKey)}`);
if (hasModel) {
logger.always(` Model: ${process.env.GOOGLE_AI_MODEL}`);
}
else {
logger.always(` Model: ${getRuntimeDefaultModel()} (default)`);
}
}
else {
logger.always(chalk.yellow("⚠️ Google AI Studio setup incomplete"));
}
return;
}
const config = {};
// Step 2: Handle existing configuration
if (hasApiKey && currentApiKey) {
logger.always(chalk.green("✅ Google AI Studio API key found in environment"));
logger.always(` API Key: ${maskCredential(currentApiKey)}`);
if (hasModel) {
logger.always(` Model: ${process.env.GOOGLE_AI_MODEL}`);
}
else {
logger.always(` Model: ${getRuntimeDefaultModel()} (default)`);
}
if (options.interactive) {
const { reconfigure } = await inquirer.prompt([
{
type: "confirm",
name: "reconfigure",
message: "Google AI Studio is already configured. Do you want to reconfigure?",
default: false,
},
]);
if (!reconfigure) {
// Still offer model selection if no model is set
if (!hasModel) {
const { wantsCustomModel } = await inquirer.prompt([
{
type: "confirm",
name: "wantsCustomModel",
message: "Do you want to specify a Google AI model? (optional)",
default: false,
},
]);
if (wantsCustomModel) {
config.model = await promptForModel();
}
}
else {
// Offer to change existing model
const { wantsChangeModel } = await inquirer.prompt([
{
type: "confirm",
name: "wantsChangeModel",
message: `Do you want to change the Google AI model? (current: ${process.env.GOOGLE_AI_MODEL})`,
default: false,
},
]);
if (wantsChangeModel) {
config.model = await promptForModel();
}
}
if (config.model) {
await updateEnvFile(config);
logger.always(chalk.green("✅ Model configuration updated!"));
logger.always(` GOOGLE_AI_MODEL=${config.model}`);
}
else {
logger.always(chalk.blue("👍 Keeping existing configuration."));
}
// Show usage example
showUsageExample();
return;
}
else {
// User chose to reconfigure - mark this for proper handling
logger.always(chalk.blue("📝 Reconfiguring Google AI Studio setup..."));
config.isReconfiguring = true;
}
}
else {
// Non-interactive mode - just use existing credentials
logger.always(chalk.green("✅ Setup complete! Using existing Google AI Studio configuration."));
return;
}
}
// Step 3: Interactive setup for missing or reconfiguring credentials
if (options.interactive) {
const isReconfiguring = config.isReconfiguring === true;
// Handle API key setup/reconfiguration
if (!hasApiKey) {
// No API key exists - prompt for it
logger.always("");
logger.always(chalk.yellow("📋 To get your Google AI Studio API key:"));
logger.always("1. Visit: https://aistudio.google.com/app/apikey");
logger.always("2. Sign in with your Google account");
logger.always("3. Click 'Create API key' (free tier available)");
logger.always("4. Copy the API key (starts with AIza...)");
logger.always("");
const { apiKey } = await inquirer.prompt([
{
type: "password",
name: "apiKey",
message: "Enter your Google AI Studio API key:",
validate: validateApiKey,
},
]);
config.apiKey = apiKey.trim();
}
else if (isReconfiguring) {
// API key exists and user is reconfiguring - ask if they want to change it
const { wantsChangeApiKey } = await inquirer.prompt([
{
type: "confirm",
name: "wantsChangeApiKey",
message: `Do you want to change the Google AI Studio API key? (current: ${currentApiKey ? maskCredential(currentApiKey) : "****"})`,
default: false,
},
]);
if (wantsChangeApiKey) {
logger.always("");
logger.always(chalk.yellow("📋 To get your Google AI Studio API key:"));
logger.always("1. Visit: https://aistudio.google.com/app/apikey");
logger.always("2. Sign in with your Google account");
logger.always("3. Click 'Create API key' (free tier available)");
logger.always("4. Copy the API key (starts with AIza...)");
logger.always("");
const { apiKey } = await inquirer.prompt([
{
type: "password",
name: "apiKey",
message: "Enter your new Google AI Studio API key (replacing existing):",
validate: validateApiKey,
},
]);
config.apiKey = apiKey.trim();
}
}
// Prompt for model selection
const { wantsCustomModel } = await inquirer.prompt([
{
type: "confirm",
name: "wantsCustomModel",
message: hasModel
? `Do you want to change the Google AI model? (current: ${process.env.GOOGLE_AI_MODEL})`
: "Do you want to specify a Google AI model? (optional)",
default: false,
},
]);
if (wantsCustomModel) {
config.model = await promptForModel();
}
}
else {
// Non-interactive mode
logger.always(chalk.yellow("⚠️ Non-interactive mode: setup incomplete"));
logger.always(chalk.yellow("💡 Run without --non-interactive to configure Google AI Studio"));
return;
}
// Step 4: Update .env file
if (config.apiKey || config.model) {
await updateEnvFile(config);
logger.always(chalk.green("✅ Google AI Studio setup complete!"));
if (config.apiKey) {
logger.always(` API Key: ${maskCredential(config.apiKey)}`);
}
if (config.model) {
logger.always(` Model: ${config.model}`);
}
// Show usage example
showUsageExample();
}
else if (options.interactive && !options.checkOnly) {
logger.always(chalk.green("✅ Setup complete!"));
showUsageExample();
}
}
catch (error) {
logger.error(chalk.red("❌ Google AI Studio setup failed:"));
logger.error(chalk.red(error instanceof Error ? error.message : "Unknown error"));
process.exit(1);
}
}
/**
* Display current configuration status
*/
function displayCurrentStatus(hasApiKey, hasModel) {
if (hasApiKey) {
logger.always(chalk.green("✔ Google AI Studio API key found in environment"));
}
else {
logger.always(chalk.red("✘ Google AI Studio API key not found"));
}
if (hasModel) {
logger.always(chalk.green(`✔ GOOGLE_AI_MODEL found: ${process.env.GOOGLE_AI_MODEL}`));
}
else {
logger.always(chalk.yellow(`⚠ GOOGLE_AI_MODEL not set (will use ${getRuntimeDefaultModel()} default)`));
}
}
/**
* Validate Google AI Studio API key format
*/
function validateApiKey(input) {
if (!input.trim()) {
return "Google AI Studio API key is required";
}
const trimmed = input.trim();
if (!trimmed.startsWith("AIza")) {
return "Google AI Studio API key should start with 'AIza'";
}
if (trimmed.length < 20) {
return "Google AI Studio API key seems too short";
}
// Basic format check: AIza[32+ char random string]
if (!/^AIza[a-zA-Z0-9_-]{20,}$/.test(trimmed)) {
return "Invalid Google AI Studio API key format";
}
return true;
}
/**
* Prompt user for model selection
*/
async function promptForModel() {
const { modelChoice } = await inquirer.prompt([
{
type: "select",
name: "modelChoice",
message: "Select a Google AI model:",
choices: getTopModelChoices(AIProviderName.GOOGLE_AI, 5),
},
]);
if (modelChoice === "custom") {
const { customModel } = await inquirer.prompt([
{
type: "input",
name: "customModel",
message: "Enter your custom Google AI model name:",
validate: (input) => {
if (!input.trim()) {
return "Model name is required";
}
// Basic validation - Google AI models typically follow certain patterns
const trimmed = input.trim();
if (!/^[a-z0-9-._]+$/i.test(trimmed)) {
return "Model name should contain only letters, numbers, hyphens, dots, and underscores";
}
return true;
},
},
]);
return customModel.trim();
}
return modelChoice;
}
/**
* Update .env file with Google AI Studio configuration
*/
async function updateEnvFile(config) {
const envPath = path.join(process.cwd(), ".env");
const spinner = ora("💾 Updating .env file...").start();
try {
// Prepare environment variables to update
const envVars = {};
const keysToDelete = [];
if (config.apiKey) {
// Use GOOGLE_AI_API_KEY as the primary key
envVars.GOOGLE_AI_API_KEY = config.apiKey;
// Remove alternative key if it exists
keysToDelete.push("GOOGLE_GENERATIVE_AI_API_KEY");
}
if (config.model) {
envVars.GOOGLE_AI_MODEL = config.model;
}
// Update .env file using centralized envManager
const result = updateEnvFileManager(envVars, envPath, true, keysToDelete);
spinner.succeed(chalk.green("✔ .env file updated successfully"));
// Display summary of changes (quietly, since we have our own success message)
displayEnvUpdateSummary(result, true);
}
catch (error) {
spinner.fail(chalk.red("❌ Failed to update .env file"));
logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : "Unknown error"}`));
throw error;
}
}
/**
* Show usage example
*/
function showUsageExample() {
logger.always("");
logger.always(chalk.green("🚀 You can now use Google AI Studio with the NeuroLink CLI:"));
logger.always(chalk.cyan(" pnpm cli generate 'Hello from Google AI!' --provider google-ai"));
logger.always(chalk.cyan(" pnpm cli generate 'Explain quantum computing' --provider google-ai"));
logger.always(chalk.cyan(" pnpm cli generate 'Analyze this data' --provider google-ai --enable-analytics"));
}
//# sourceMappingURL=setup-google-ai.js.map