@iservu-inc/adf-cli 0.3.0 → 0.4.12

package/lib/ai/ai-config.js

@@ -0,0 +1,398 @@
+const inquirer = require('inquirer');
+const autocomplete = require('inquirer-autocomplete-prompt');
+const chalk = require('chalk');
+const fs = require('fs-extra');
+const path = require('path');
+const os = require('os');
+
+// Register autocomplete prompt
+inquirer.registerPrompt('autocomplete', autocomplete);
+
+/**
+ * AI Provider Configuration
+ * Handles setup and validation of AI providers for the interview system
+ * - Saves API keys to .env file for persistence
+ * - Fetches available models dynamically from providers
+ * - Autocomplete model selection with filtering
+ */
+
+const AI_PROVIDERS = {
+  ANTHROPIC: {
+    id: 'anthropic',
+    name: 'Anthropic Claude',
+    envVar: 'ANTHROPIC_API_KEY',
+    requiredFormat: 'sk-ant-',
+    website: 'https://console.anthropic.com/',
+    setup: 'Get your API key from https://console.anthropic.com/',
+    defaultModels: ['claude-sonnet-4-5-20250929', 'claude-3-5-sonnet-20241022', 'claude-3-opus-20240229']
+  },
+  OPENAI: {
+    id: 'openai',
+    name: 'OpenAI GPT',
+    envVar: 'OPENAI_API_KEY',
+    requiredFormat: 'sk-',
+    website: 'https://platform.openai.com/',
+    setup: 'Get your API key from https://platform.openai.com/api-keys',
+    defaultModels: ['gpt-4-turbo', 'gpt-4o', 'gpt-4', 'gpt-3.5-turbo']
+  },
+  GOOGLE: {
+    id: 'google',
+    name: 'Google Gemini',
+    envVar: 'GOOGLE_API_KEY',
+    requiredFormat: '',  // Google keys don't have consistent prefix
+    website: 'https://ai.google.dev/',
+    setup: 'Get your API key from https://aistudio.google.com/app/apikey',
+    defaultModels: ['gemini-2.0-flash-exp', 'gemini-1.5-pro', 'gemini-1.5-flash']
+  },
+  OPENROUTER: {
+    id: 'openrouter',
+    name: 'OpenRouter (Multi-Model)',
+    envVar: 'OPENROUTER_API_KEY',
+    requiredFormat: 'sk-or-',
+    website: 'https://openrouter.ai/',
+    setup: 'Get your API key from https://openrouter.ai/keys',
+    defaultModels: [
+      'anthropic/claude-sonnet-4-5',
+      'openai/gpt-4-turbo',
+      'google/gemini-pro-1.5',
+      'meta-llama/llama-3.1-70b-instruct'
+    ]
+  }
+};
+
+/**
+ * Get or create .env file path
+ */
+function getEnvFilePath(projectPath) {
+  const adfDir = path.join(projectPath, '.adf');
+  return path.join(adfDir, '.env');
+}
+
+/**
+ * Load existing .env file
+ */
+async function loadEnvFile(envPath) {
+  if (await fs.pathExists(envPath)) {
+    const content = await fs.readFile(envPath, 'utf-8');
+    const env = {};
+    content.split('\n').forEach(line => {
+      const trimmed = line.trim();
+      if (trimmed && !trimmed.startsWith('#')) {
+        const [key, ...valueParts] = trimmed.split('=');
+        if (key && valueParts.length > 0) {
+          env[key.trim()] = valueParts.join('=').trim().replace(/^["']|["']$/g, '');
+        }
+      }
+    });
+    return env;
+  }
+  return {};
+}
+
+/**
+ * Save API key to .env file
+ */
+async function saveToEnvFile(envPath, key, value) {
+  const env = await loadEnvFile(envPath);
+  env[key] = value;
+
+  const lines = [
+    '# AI Provider API Keys for adf-cli',
+    '# DO NOT commit this file to version control',
+    ''
+  ];
+
+  for (const [k, v] of Object.entries(env)) {
+    lines.push(`${k}="${v}"`);
+  }
+
+  await fs.ensureDir(path.dirname(envPath));
+  await fs.writeFile(envPath, lines.join('\n'), 'utf-8');
+}
+
+/**
+ * Load .env file into process.env
+ */
+function loadEnvIntoProcess(envPath) {
+  require('dotenv').config({ path: envPath });
+}
+
+/**
+ * Fetch available models from provider API
+ */
+async function fetchAvailableModels(provider, apiKey) {
+  const ora = require('ora');
+  const spinner = ora('Fetching available models...').start();
+
+  try {
+    switch (provider.id) {
+      case 'anthropic':
+        // Anthropic doesn't have a models list API, use defaults
+        spinner.succeed('Using known Anthropic models');
+        return provider.defaultModels;
+
+      case 'openai':
+        const OpenAI = require('openai');
+        const openai = new OpenAI({ apiKey });
+        const response = await openai.models.list();
+        const gptModels = response.data
+          .filter(m => m.id.startsWith('gpt-'))
+          .map(m => m.id)
+          .sort();
+        spinner.succeed(`Found ${gptModels.length} OpenAI models`);
+        return gptModels.length > 0 ? gptModels : provider.defaultModels;
+
+      case 'google':
+        // Google doesn't have a public models list API, use defaults
+        spinner.succeed('Using known Google Gemini models');
+        return provider.defaultModels;
+
+      case 'openrouter':
+        const fetch = require('node-fetch');
+        const orResponse = await fetch('https://openrouter.ai/api/v1/models', {
+          headers: {
+            'Authorization': `Bearer ${apiKey}`
+          }
+        });
+        const orData = await orResponse.json();
+        const orModels = orData.data.map(m => m.id).sort();
+        spinner.succeed(`Found ${orModels.length} OpenRouter models`);
+        return orModels.length > 0 ? orModels : provider.defaultModels;
+
+      default:
+        spinner.warn('Model fetching not supported, using defaults');
+        return provider.defaultModels;
+    }
+  } catch (error) {
+    spinner.fail(`Failed to fetch models: ${error.message}`);
+    console.log(chalk.yellow('  Using default model list\n'));
+    return provider.defaultModels;
+  }
+}
+
+/**
+ * Prompt user to select and configure AI provider
+ */
+async function configureAIProvider(projectPath = process.cwd()) {
+  console.log(chalk.cyan.bold('\n🤖 AI Provider Configuration\n'));
+  console.log(chalk.gray('This interview requires an AI assistant to analyze your answers and provide insights.\n'));
+
+  const envPath = getEnvFilePath(projectPath);
+
+  // Load existing .env file
+  const existingEnv = await loadEnvFile(envPath);
+
+  // Check for existing API keys (from process.env or .env file)
+  const availableProviders = [];
+  for (const [key, provider] of Object.entries(AI_PROVIDERS)) {
+    const apiKey = process.env[provider.envVar] || existingEnv[provider.envVar];
+    if (apiKey) {
+      availableProviders.push(provider);
+    }
+  }
+
+  if (availableProviders.length > 0) {
+    console.log(chalk.green('✓ Detected API keys for:'));
+    availableProviders.forEach(p => {
+      console.log(chalk.gray(`  • ${p.name} (${p.envVar})`));
+    });
+    console.log('');
+  }
+
+  // Provider selection
+  const providerChoices = [
+    {
+      name: `${AI_PROVIDERS.ANTHROPIC.name} ${availableProviders.find(p => p.id === 'anthropic') ? chalk.green('✓ Configured') : ''}`,
+      value: 'anthropic',
+      short: 'Anthropic'
+    },
+    {
+      name: `${AI_PROVIDERS.OPENAI.name} ${availableProviders.find(p => p.id === 'openai') ? chalk.green('✓ Configured') : ''}`,
+      value: 'openai',
+      short: 'OpenAI'
+    },
+    {
+      name: `${AI_PROVIDERS.GOOGLE.name} ${availableProviders.find(p => p.id === 'google') ? chalk.green('✓ Configured') : ''}`,
+      value: 'google',
+      short: 'Google'
+    },
+    {
+      name: `${AI_PROVIDERS.OPENROUTER.name} ${availableProviders.find(p => p.id === 'openrouter') ? chalk.green('✓ Configured') : ''}`,
+      value: 'openrouter',
+      short: 'OpenRouter'
+    }
+  ];
+
+  const { provider } = await inquirer.prompt([
+    {
+      type: 'list',
+      name: 'provider',
+      message: 'Select AI provider:',
+      choices: providerChoices
+    }
+  ]);
+
+  const selectedProvider = AI_PROVIDERS[provider.toUpperCase()];
+
+  // Check if API key exists
+  let apiKey = process.env[selectedProvider.envVar] || existingEnv[selectedProvider.envVar];
+
+  if (!apiKey) {
+    console.log(chalk.yellow(`\n⚠️  ${selectedProvider.envVar} not found\n`));
+    console.log(chalk.gray(`Setup instructions:`));
+    console.log(chalk.gray(`  ${selectedProvider.setup}\n`));
+
+    const { manualKey } = await inquirer.prompt([
+      {
+        type: 'password',
+        name: 'manualKey',
+        message: `Enter your ${selectedProvider.name} API key:`,
+        mask: '*',
+        validate: (input) => {
+          if (!input || input.trim().length === 0) {
+            return 'API key cannot be empty';
+          }
+          if (selectedProvider.requiredFormat && !input.startsWith(selectedProvider.requiredFormat)) {
+            return `API key should start with "${selectedProvider.requiredFormat}"`;
+          }
+          return true;
+        }
+      }
+    ]);
+
+    apiKey = manualKey.trim();
+
+    // Save to .env file
+    await saveToEnvFile(envPath, selectedProvider.envVar, apiKey);
+
+    console.log(chalk.green(`\n✓ API key saved to: ${path.relative(projectPath, envPath)}`));
+    console.log(chalk.gray('  This file is gitignored and will persist across sessions.\n'));
+
+    // Load into current process
+    process.env[selectedProvider.envVar] = apiKey;
+  } else {
+    const keySource = existingEnv[selectedProvider.envVar] ? '.env file' : 'environment';
+    console.log(chalk.green(`\n✓ Using API key from ${keySource}`));
+  }
+
+  // Fetch available models
+  const availableModels = await fetchAvailableModels(selectedProvider, apiKey);
+
+  // Model selection with autocomplete
+  console.log('');
+  const { model } = await inquirer.prompt([
+    {
+      type: 'autocomplete',
+      name: 'model',
+      message: `Select ${selectedProvider.name} model (type to filter):`,
+      source: async (answersSoFar, input) => {
+        const filtered = input
+          ? availableModels.filter(m => m.toLowerCase().includes(input.toLowerCase()))
+          : availableModels;
+        return filtered;
+      },
+      pageSize: 10
+    }
+  ]);
+
+  const config = {
+    provider: selectedProvider.id,
+    providerName: selectedProvider.name,
+    model,
+    apiKey,
+    envVar: selectedProvider.envVar,
+    envPath
+  };
+
+  // Test connection (optional but recommended)
+  const { testConnection } = await inquirer.prompt([
+    {
+      type: 'confirm',
+      name: 'testConnection',
+      message: 'Test AI connection before starting?',
+      default: true
+    }
+  ]);
+
+  if (testConnection) {
+    const ora = require('ora');
+    const spinner = ora('Testing AI connection...').start();
+
+    try {
+      const AIClient = require('./ai-client');
+      const client = new AIClient(config);
+      await client.test();
+      spinner.succeed(chalk.green('AI connection successful!'));
+    } catch (error) {
+      spinner.fail(chalk.red('AI connection failed'));
+      console.log(chalk.red(`\nError: ${error.message}\n`));
+
+      const { retry } = await inquirer.prompt([
+        {
+          type: 'confirm',
+          name: 'retry',
+          message: 'Try again with different configuration?',
+          default: true
+        }
+      ]);
+
+      if (retry) {
+        return configureAIProvider(projectPath); // Recursive retry
+      } else {
+        process.exit(1);
+      }
+    }
+  }
+
+  console.log(chalk.gray('\n' + '━'.repeat(60)) + '\n');
+
+  return config;
+}
+
+/**
+ * Detect which providers have API keys in environment or .env file
+ */
+async function detectAvailableProviders(projectPath = process.cwd()) {
+  const envPath = getEnvFilePath(projectPath);
+  const existingEnv = await loadEnvFile(envPath);
+
+  const available = [];
+
+  for (const [key, provider] of Object.entries(AI_PROVIDERS)) {
+    const apiKey = process.env[provider.envVar] || existingEnv[provider.envVar];
+    if (apiKey) {
+      available.push(provider);
+    }
+  }
+
+  return available;
+}
+
+/**
+ * Validate API key format
+ */
+function validateAPIKey(provider, apiKey) {
+  const config = AI_PROVIDERS[provider.toUpperCase()];
+
+  if (!apiKey || apiKey.trim().length === 0) {
+    return { valid: false, error: 'API key cannot be empty' };
+  }
+
+  if (config.requiredFormat && !apiKey.startsWith(config.requiredFormat)) {
+    return {
+      valid: false,
+      error: `API key should start with "${config.requiredFormat}"`
+    };
+  }
+
+  return { valid: true };
+}
+
+module.exports = {
+  configureAIProvider,
+  detectAvailableProviders,
+  validateAPIKey,
+  AI_PROVIDERS,
+  loadEnvIntoProcess,
+  getEnvFilePath
+};