family-ai-agent 1.0.2 → 1.0.3

package/src/cli/commands/config.ts

@@ -0,0 +1,358 @@
+import { Command } from 'commander';
+import inquirer from 'inquirer';
+import chalk from 'chalk';
+import ora from 'ora';
+
+import {
+  loadUserConfig,
+  saveUserConfig,
+  deleteUserConfig,
+  updateUserConfig,
+  addCustomModel,
+  removeCustomModel,
+  getAllModels,
+  getDisplayConfig,
+  getConfigPath,
+  configExists,
+  type CustomModel,
+} from '../../config/user-config.js';
+import { runSetupWizard } from '../setup-wizard.js';
+
+/**
+ * Create config command group
+ */
+export function createConfigCommand(): Command {
+  const configCmd = new Command('config')
+    .description('Manage Family AI Agent configuration');
+
+  // config setup - Run interactive setup wizard
+  configCmd
+    .command('setup')
+    .description('Run interactive setup wizard')
+    .action(async () => {
+      await runSetupWizard();
+    });
+
+  // config show - Display current configuration
+  configCmd
+    .command('show')
+    .description('Display current configuration')
+    .action(() => {
+      const config = getDisplayConfig();
+
+      if (!config) {
+        console.log(chalk.yellow('\nNo configuration found.'));
+        console.log(chalk.gray('Run `family-ai-agent config setup` to create one.\n'));
+        return;
+      }
+
+      console.log(chalk.bold('\n📋 Current Configuration\n'));
+      console.log(chalk.white('Location: ') + chalk.gray(getConfigPath()));
+      console.log();
+
+      // API Settings
+      console.log(chalk.bold.cyan('API Settings'));
+      console.log(`  API Key:    ${chalk.gray(config.apiKey as string)}`);
+      console.log(`  Base URL:   ${chalk.gray(config.baseUrl as string)}`);
+      console.log();
+
+      // Model Settings
+      console.log(chalk.bold.cyan('Models'));
+      console.log(`  Default:    ${chalk.green(config.defaultModel as string)}`);
+      console.log(`  Fast:       ${chalk.green(config.fastModel as string)}`);
+      console.log(`  Embedding:  ${chalk.green(config.embeddingModel as string)}`);
+      console.log();
+
+      // Safety Settings
+      console.log(chalk.bold.cyan('Safety'));
+      console.log(
+        `  Filters:    ${(config.enableSafetyFilters as boolean) ? chalk.green('Enabled') : chalk.yellow('Disabled')}`
+      );
+      console.log(
+        `  Logging:    ${(config.enableAuditLogging as boolean) ? chalk.green('Enabled') : chalk.yellow('Disabled')}`
+      );
+      console.log();
+
+      // Custom Models
+      const customModels = config.customModels as CustomModel[];
+      if (customModels && customModels.length > 0) {
+        console.log(chalk.bold.cyan('Custom Models'));
+        for (const model of customModels) {
+          console.log(`  - ${chalk.white(model.name)} (${chalk.gray(model.id)})`);
+        }
+        console.log();
+      }
+
+      // Timestamps
+      if (config.createdAt || config.updatedAt) {
+        console.log(chalk.bold.cyan('Metadata'));
+        if (config.createdAt) {
+          console.log(`  Created:    ${chalk.gray(new Date(config.createdAt as string).toLocaleString())}`);
+        }
+        if (config.updatedAt) {
+          console.log(`  Updated:    ${chalk.gray(new Date(config.updatedAt as string).toLocaleString())}`);
+        }
+        console.log();
+      }
+    });
+
+  // config set <key> <value> - Set a configuration value
+  configCmd
+    .command('set <key> <value>')
+    .description('Set a configuration value')
+    .action(async (key: string, value: string) => {
+      if (!configExists()) {
+        console.log(chalk.yellow('\nNo configuration found.'));
+        console.log(chalk.gray('Run `family-ai-agent config setup` first.\n'));
+        return;
+      }
+
+      const validKeys = [
+        'apiKey',
+        'baseUrl',
+        'defaultModel',
+        'fastModel',
+        'embeddingModel',
+        'enableSafetyFilters',
+        'enableAuditLogging',
+      ];
+
+      // Map common aliases
+      const keyAliases: Record<string, string> = {
+        'api-key': 'apiKey',
+        'api_key': 'apiKey',
+        'key': 'apiKey',
+        'base-url': 'baseUrl',
+        'base_url': 'baseUrl',
+        'url': 'baseUrl',
+        'model': 'defaultModel',
+        'default-model': 'defaultModel',
+        'default_model': 'defaultModel',
+        'fast-model': 'fastModel',
+        'fast_model': 'fastModel',
+        'fast': 'fastModel',
+        'embedding': 'embeddingModel',
+        'embedding-model': 'embeddingModel',
+        'safety': 'enableSafetyFilters',
+        'filters': 'enableSafetyFilters',
+        'logging': 'enableAuditLogging',
+        'audit': 'enableAuditLogging',
+      };
+
+      const normalizedKey = keyAliases[key.toLowerCase()] || key;
+
+      if (!validKeys.includes(normalizedKey)) {
+        console.log(chalk.red(`\nUnknown configuration key: ${key}`));
+        console.log(chalk.gray(`Valid keys: ${validKeys.join(', ')}\n`));
+        return;
+      }
+
+      const spinner = ora('Updating configuration...').start();
+
+      try {
+        // Handle boolean values
+        let parsedValue: unknown = value;
+        if (normalizedKey.startsWith('enable')) {
+          parsedValue = ['true', '1', 'yes', 'on'].includes(value.toLowerCase());
+        }
+
+        updateUserConfig({ [normalizedKey]: parsedValue });
+        spinner.succeed(`Set ${normalizedKey} = ${value}`);
+      } catch (error) {
+        spinner.fail('Failed to update configuration');
+        console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+      }
+    });
+
+  // config models - List available models
+  configCmd
+    .command('models')
+    .description('List available models')
+    .action(() => {
+      const models = getAllModels();
+
+      console.log(chalk.bold('\n📦 Available Models\n'));
+
+      // Group by type
+      const builtIn = models.filter((m) => !m.isCustom);
+      const custom = models.filter((m) => m.isCustom);
+
+      console.log(chalk.bold.cyan('Built-in Models'));
+      for (const model of builtIn) {
+        console.log(`  ${chalk.white(model.name)}`);
+        console.log(`    ${chalk.gray(model.id)}`);
+      }
+
+      if (custom.length > 0) {
+        console.log();
+        console.log(chalk.bold.cyan('Custom Models'));
+        for (const model of custom) {
+          console.log(`  ${chalk.white(model.name)}`);
+          console.log(`    ${chalk.gray(model.id)}`);
+        }
+      }
+
+      console.log();
+      console.log(chalk.gray('Tip: Use `config add-model` to add a custom model.'));
+      console.log();
+    });
+
+  // config add-model - Add a custom model
+  configCmd
+    .command('add-model')
+    .description('Add a custom model')
+    .action(async () => {
+      if (!configExists()) {
+        console.log(chalk.yellow('\nNo configuration found.'));
+        console.log(chalk.gray('Run `family-ai-agent config setup` first.\n'));
+        return;
+      }
+
+      console.log(chalk.bold('\n➕ Add Custom Model\n'));
+
+      try {
+        const answers = await inquirer.prompt<{
+          id: string;
+          name: string;
+          contextWindow: number;
+          maxOutput: number;
+          description: string;
+        }>([
+          {
+            type: 'input',
+            name: 'id',
+            message: 'Model ID (e.g., provider/model-name):',
+            validate: (input: string) => {
+              if (!input || input.trim().length === 0) {
+                return 'Model ID is required';
+              }
+              return true;
+            },
+          },
+          {
+            type: 'input',
+            name: 'name',
+            message: 'Display name:',
+            validate: (input: string) => {
+              if (!input || input.trim().length === 0) {
+                return 'Display name is required';
+              }
+              return true;
+            },
+          },
+          {
+            type: 'number',
+            name: 'contextWindow',
+            message: 'Context window size:',
+            default: 8192,
+          },
+          {
+            type: 'number',
+            name: 'maxOutput',
+            message: 'Max output tokens:',
+            default: 4096,
+          },
+          {
+            type: 'input',
+            name: 'description',
+            message: 'Description (optional):',
+            default: '',
+          },
+        ]);
+
+        const model: CustomModel = {
+          id: answers.id,
+          name: answers.name,
+          contextWindow: answers.contextWindow,
+          maxOutput: answers.maxOutput,
+          description: answers.description || undefined,
+        };
+
+        addCustomModel(model);
+        console.log(chalk.green(`\n✓ Added custom model: ${model.name}\n`));
+      } catch (error) {
+        if ((error as { name?: string }).name === 'ExitPromptError') {
+          console.log(chalk.yellow('\nCancelled.\n'));
+          return;
+        }
+        throw error;
+      }
+    });
+
+  // config remove-model <id> - Remove a custom model
+  configCmd
+    .command('remove-model <id>')
+    .description('Remove a custom model')
+    .action(async (id: string) => {
+      if (!configExists()) {
+        console.log(chalk.yellow('\nNo configuration found.\n'));
+        return;
+      }
+
+      const config = loadUserConfig();
+      const model = config?.customModels.find((m) => m.id === id);
+
+      if (!model) {
+        console.log(chalk.yellow(`\nModel not found: ${id}\n`));
+        return;
+      }
+
+      const { confirm } = await inquirer.prompt<{ confirm: boolean }>([
+        {
+          type: 'confirm',
+          name: 'confirm',
+          message: `Remove custom model "${model.name}"?`,
+          default: false,
+        },
+      ]);
+
+      if (confirm) {
+        removeCustomModel(id);
+        console.log(chalk.green(`\n✓ Removed model: ${model.name}\n`));
+      } else {
+        console.log(chalk.gray('\nCancelled.\n'));
+      }
+    });
+
+  // config reset - Reset configuration
+  configCmd
+    .command('reset')
+    .description('Reset configuration to defaults')
+    .action(async () => {
+      if (!configExists()) {
+        console.log(chalk.yellow('\nNo configuration to reset.\n'));
+        return;
+      }
+
+      const { confirm } = await inquirer.prompt<{ confirm: boolean }>([
+        {
+          type: 'confirm',
+          name: 'confirm',
+          message: chalk.red('Are you sure you want to reset all configuration?'),
+          default: false,
+        },
+      ]);
+
+      if (confirm) {
+        deleteUserConfig();
+        console.log(chalk.green('\n✓ Configuration reset.\n'));
+        console.log(chalk.gray('Run `family-ai-agent config setup` to reconfigure.\n'));
+      } else {
+        console.log(chalk.gray('\nCancelled.\n'));
+      }
+    });
+
+  // config path - Show config file path
+  configCmd
+    .command('path')
+    .description('Show configuration file path')
+    .action(() => {
+      console.log(chalk.white('\nConfiguration file:'));
+      console.log(chalk.cyan(getConfigPath()));
+      console.log();
+    });
+
+  return configCmd;
+}
+
+export default createConfigCommand;

package/src/cli/index.ts

@@ -6,17 +6,20 @@ import chalk from 'chalk';
 import ora from 'ora';
 import boxen from 'boxen';
 
-import { runOrchestrator, streamOrchestrator } from '../core/orchestrator/graph.js';
+import { runOrchestrator } from '../core/orchestrator/graph.js';
 import { getVectorStore } from '../memory/longterm/vector-store.js';
 import { getConversationMemory } from '../memory/conversation/index.js';
 import { getKnowledgeBase } from '../memory/knowledge-base/index.js';
 import { validateInput } from '../safety/guardrails/input-guardrail.js';
 import { validateOutput } from '../safety/guardrails/output-guardrail.js';
 import { initDatabase, closePool } from '../database/client.js';
-import { config, isDevelopment } from '../config/index.js';
+import { config, isConfigured, hasUserConfig, getUserConfigPath } from '../config/index.js';
 import { createLogger } from '../utils/logger.js';
 import { nanoid } from 'nanoid';
 
+import { runSetupWizard, checkAndRunSetup } from './setup-wizard.js';
+import { createConfigCommand } from './commands/config.js';
+
 const logger = createLogger('CLI');
 const program = new Command();
 
@@ -38,15 +41,80 @@ function showBanner(): void {
   console.log(banner);
 }
 
+// Check if configured and prompt setup if needed
+async function ensureConfigured(): Promise<boolean> {
+  if (isConfigured()) {
+    return true;
+  }
+
+  console.log(chalk.yellow('\n⚠️  No API key configured.\n'));
+
+  const { action } = await inquirer.prompt<{ action: string }>([
+    {
+      type: 'list',
+      name: 'action',
+      message: 'What would you like to do?',
+      choices: [
+        { name: 'Run setup wizard', value: 'setup' },
+        { name: 'Enter API key directly', value: 'quick' },
+        { name: 'Exit', value: 'exit' },
+      ],
+    },
+  ]);
+
+  if (action === 'exit') {
+    console.log(chalk.gray('\nRun `family-ai-agent config setup` when ready.\n'));
+    return false;
+  }
+
+  if (action === 'quick') {
+    const { apiKey } = await inquirer.prompt<{ apiKey: string }>([
+      {
+        type: 'password',
+        name: 'apiKey',
+        message: 'Enter your OpenRouter API Key:',
+        mask: '*',
+        validate: (input: string) => {
+          if (!input || input.trim().length === 0) {
+            return 'API key is required';
+          }
+          return true;
+        },
+      },
+    ]);
+
+    const { runQuickSetup } = await import('./setup-wizard.js');
+    const success = await runQuickSetup(apiKey);
+
+    if (success) {
+      console.log(chalk.green('\n✓ Configuration saved! Please restart the application.\n'));
+    }
+    return false; // Need restart to pick up new config
+  }
+
+  // Full setup wizard
+  const success = await runSetupWizard();
+  if (success) {
+    console.log(chalk.gray('\nPlease restart the application to apply the new configuration.\n'));
+  }
+  return false; // Need restart to pick up new config
+}
+
 // Interactive chat mode
 async function startInteractiveChat(): Promise<void> {
   showBanner();
 
+  // Show config status
+  if (hasUserConfig()) {
+    console.log(chalk.gray(`\nConfig: ${getUserConfigPath()}`));
+  }
+  console.log(chalk.gray(`Model: ${config.DEFAULT_MODEL}`));
+
   const threadId = `thread-${nanoid(8)}`;
   const conversationMemory = getConversationMemory();
   const vectorStore = getVectorStore();
 
-  console.log(chalk.gray(`\nSession: ${threadId}`));
+  console.log(chalk.gray(`Session: ${threadId}`));
   console.log(chalk.gray('Type "exit" or "quit" to end the session'));
   console.log(chalk.gray('Type "clear" to start a new conversation'));
   console.log(chalk.gray('Type "memory" to view recent memories'));
@@ -250,15 +318,21 @@ async function runQuery(query: string): Promise<void> {
 
 // Main program setup
 program
-  .name('family-ai')
+  .name('family-ai-agent')
   .description('Family AI Agent - Your AI Family for All Tasks')
-  .version('1.0.0');
+  .version('1.0.2');
+
+// Add config command group
+program.addCommand(createConfigCommand());
 
 program
   .command('chat')
   .description('Start interactive chat mode')
   .action(async () => {
     try {
+      if (!(await ensureConfigured())) {
+        process.exit(0);
+      }
       await initDatabase();
       await startInteractiveChat();
     } catch (error) {
@@ -274,6 +348,9 @@ program
   .description('Ask a single question')
   .action(async (query: string) => {
     try {
+      if (!(await ensureConfigured())) {
+        process.exit(0);
+      }
       await initDatabase();
       await runQuery(query);
     } catch (error) {
@@ -288,6 +365,10 @@ program
   .command('upload <filepath>')
   .description('Upload a document to the knowledge base')
   .action(async (filepath: string) => {
+    if (!(await ensureConfigured())) {
+      process.exit(0);
+    }
+
     const spinner = ora('Uploading document...').start();
 
     try {
@@ -321,6 +402,10 @@ program
   .description('Search the knowledge base')
   .option('-l, --limit <number>', 'Maximum results', '5')
   .action(async (query: string, options: { limit: string }) => {
+    if (!(await ensureConfigured())) {
+      process.exit(0);
+    }
+
     const spinner = ora('Searching...').start();
 
     try {
@@ -357,22 +442,36 @@ program
   .command('status')
   .description('Show system status')
   .action(async () => {
+    console.log(chalk.bold('\n🤖 Family AI Agent Status\n'));
+
+    // Config status
+    if (hasUserConfig()) {
+      console.log(chalk.white('Config:'), chalk.green('User config loaded'));
+      console.log(chalk.gray(`         ${getUserConfigPath()}`));
+    } else if (isConfigured()) {
+      console.log(chalk.white('Config:'), chalk.yellow('Using environment variables'));
+    } else {
+      console.log(chalk.white('Config:'), chalk.red('Not configured'));
+      console.log(chalk.gray('         Run `family-ai-agent config setup` to configure'));
+    }
+
+    console.log(chalk.white('API Key:'), isConfigured() ? chalk.green('Set') : chalk.red('Missing'));
+    console.log(chalk.white('Default Model:'), chalk.cyan(config.DEFAULT_MODEL));
+    console.log(chalk.white('Fast Model:'), chalk.cyan(config.FAST_MODEL));
+    console.log(chalk.white('Safety Filters:'), config.ENABLE_CONTENT_FILTER ? chalk.green('Enabled') : chalk.yellow('Disabled'));
+    console.log(chalk.white('Audit Logging:'), config.ENABLE_AUDIT_LOGGING ? chalk.green('Enabled') : chalk.yellow('Disabled'));
+
+    // Database status
     try {
       await initDatabase();
-
-      console.log(chalk.bold('\n🤖 Family AI Agent Status\n'));
       console.log(chalk.white('Database:'), chalk.green('Connected'));
-      console.log(chalk.white('Environment:'), chalk.cyan(config.NODE_ENV));
-      console.log(chalk.white('Default Model:'), chalk.cyan(config.DEFAULT_MODEL));
-      console.log(chalk.white('Safety Filters:'), config.ENABLE_CONTENT_FILTER ? chalk.green('Enabled') : chalk.yellow('Disabled'));
-      console.log(chalk.white('Audit Logging:'), config.ENABLE_AUDIT_LOGGING ? chalk.green('Enabled') : chalk.yellow('Disabled'));
-      console.log();
-    } catch (error) {
+    } catch {
       console.log(chalk.white('Database:'), chalk.red('Disconnected'));
-      console.error(error instanceof Error ? error.message : 'Unknown error');
     } finally {
       await closePool();
     }
+
+    console.log();
   });
 
 // Handle unhandled rejections
@@ -386,7 +485,18 @@ program.parse();
 
 // If no command specified, start interactive chat
 if (!process.argv.slice(2).length) {
-  initDatabase()
-    .then(() => startInteractiveChat())
-    .finally(() => closePool());
+  (async () => {
+    try {
+      if (!(await ensureConfigured())) {
+        process.exit(0);
+      }
+      await initDatabase();
+      await startInteractiveChat();
+    } catch (error) {
+      console.error(chalk.red('Failed to start:'), error);
+      process.exit(1);
+    } finally {
+      await closePool();
+    }
+  })();
 }