ask-my-llm 1.0.7 → 1.0.8

package/bin/ask-ai

@@ -10,36 +10,113 @@ function loadConfig() {
   try {
     return JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf8'));
   } catch (e) {
-    return { baseApi: 'https://api.openai.com/v1', apiKey: '', model: 'gpt-3.5-turbo' };
+    return { current: 'default', configs: { default: { baseApi: 'https://api.openai.com/v1', apiKey: '', model: 'gpt-3.5-turbo' } } };
   }
 }
 
-async function setup() {
-  const existing = loadConfig();
+function saveConfig(config) {
+  fs.writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2));
+}
+
+async function mainMenu() {
+  const config = loadConfig();
+  const choices = ['Add new config', 'Select config', 'Remove config'];
   
+  if (config.current) {
+    choices.unshift(`Use current (${config.current})`);
+  }
+
+  const { action } = await inquirer.prompt([{
+    type: 'list',
+    name: 'action',
+    message: 'Manage configs:',
+    choices
+  }]);
+
+  if (action === 'Add new config') {
+    await addConfig(config);
+  } else if (action === 'Select config') {
+    await selectConfig(config);
+  } else if (action === 'Remove config') {
+    await removeConfig(config);
+  } else {
+    console.log(`Current: ${config.current}`);
+  }
+}
+
+async function addConfig(config) {
+  const { name } = await inquirer.prompt([{
+    type: 'input',
+    name: 'name',
+    message: 'Config name:'
+  }]);
+
+  if (config.configs[name]) {
+    console.log('Config already exists. Edit it instead.');
+    return;
+  }
+
   const answers = await inquirer.prompt([
     {
       type: 'input',
       name: 'baseApi',
-      message: 'Base API URL (e.g., https://api.openai.com/v1):',
-      default: existing.baseApi
+      message: 'Base API URL:',
+      default: 'https://api.openai.com/v1'
     },
     {
       type: 'input',
       name: 'apiKey',
       message: 'API Key (leave empty for none):',
-      default: existing.apiKey || ''
+      default: ''
     },
     {
       type: 'input',
       name: 'model',
-      message: 'Model (e.g., gpt-3.5-turbo):',
-      default: existing.model
+      message: 'Model:',
+      default: 'gpt-3.5-turbo'
     }
   ]);
 
-  fs.writeFileSync(CONFIG_PATH, JSON.stringify(answers, null, 2));
-  console.log('Configuration saved to ~/.askairc');
+  config.configs[name] = answers;
+  config.current = name;
+  saveConfig(config);
+  console.log(`Config '${name}' saved.`);
+}
+
+async function selectConfig(config) {
+  const names = Object.keys(config.configs);
+  const { name } = await inquirer.prompt([{
+    type: 'list',
+    name: 'name',
+    message: 'Select config:',
+    choices: names
+  }]);
+
+  config.current = name;
+  saveConfig(config);
+  console.log(`Using '${name}'.`);
+}
+
+async function removeConfig(config) {
+  const names = Object.keys(config.configs);
+  if (names.length <= 1) {
+    console.log('Cannot remove last config.');
+    return;
+  }
+
+  const { name } = await inquirer.prompt([{
+    type: 'list',
+    name: 'name',
+    message: 'Remove config:',
+    choices: names
+  }]);
+
+  delete config.configs[name];
+  if (config.current === name) {
+    config.current = Object.keys(config.configs)[0];
+  }
+  saveConfig(config);
+  console.log(`Config '${name}' removed.`);
 }
 
-setup();
+mainMenu();

package/index.js

@@ -6,9 +6,14 @@ const CONFIG_PATH = path.join(process.env.HOME || process.env.USERPROFILE, '.ask
 
 function loadConfig() {
   try {
-    return JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf8'));
+    const data = JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf8'));
+    const current = data.current || 'default';
+    if (!data.configs || !data.configs[current]) {
+      throw new Error('No config found. Run `npx ask-my-llm` first.');
+    }
+    return data.configs[current];
   } catch (e) {
-    throw new Error('Not configured. Run `npx ask-ai` first.');
+    throw new Error('Not configured. Run `npx ask-my-llm` first.');
   }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ask-my-llm",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "Oversimplified AI usage npm module",
   "main": "index.js",
   "bin": {