ask-my-llm 1.0.7 → 1.0.9

package/bin/ask-ai

@@ -10,18 +10,111 @@ 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 null;
   }
 }
 
-async function setup() {
-  const existing = loadConfig();
+function saveConfig(config) {
+  fs.writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2));
+}
+
+async function mainMenu() {
+  let config = loadConfig();
   
+  if (!config || !config.configs || Object.keys(config.configs).length === 0) {
+    console.log('No configs found. Creating first one...');
+    await addConfig({ current: null, configs: {} });
+    return;
+  }
+
+  const choices = ['Add new config', 'Select config', 'Edit config', 'Remove config'];
+
+  const { action } = await inquirer.prompt([{
+    type: 'list',
+    name: 'action',
+    message: `Current: ${config.current}`,
+    choices
+  }]);
+
+  if (action === 'Add new config') {
+    await addConfig(config);
+  } else if (action === 'Select config') {
+    await selectConfig(config);
+  } else if (action === 'Edit config') {
+    await editConfig(config);
+  } else if (action === 'Remove config') {
+    await removeConfig(config);
+  }
+}
+
+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:',
+      default: 'https://api.openai.com/v1'
+    },
+    {
+      type: 'input',
+      name: 'apiKey',
+      message: 'API Key (leave empty for none):',
+      default: ''
+    },
+    {
+      type: 'input',
+      name: 'model',
+      message: 'Model:',
+      default: 'gpt-3.5-turbo'
+    }
+  ]);
+
+  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 editConfig(config) {
+  const names = Object.keys(config.configs);
+  const { name } = await inquirer.prompt([{
+    type: 'list',
+    name: 'name',
+    message: 'Edit config:',
+    choices: names
+  }]);
+
+  const existing = config.configs[name];
   const answers = await inquirer.prompt([
     {
       type: 'input',
       name: 'baseApi',
-      message: 'Base API URL (e.g., https://api.openai.com/v1):',
+      message: 'Base API URL:',
       default: existing.baseApi
     },
     {
@@ -33,13 +126,36 @@ async function setup() {
     {
       type: 'input',
       name: 'model',
-      message: 'Model (e.g., gpt-3.5-turbo):',
+      message: 'Model:',
       default: existing.model
     }
   ]);
 
-  fs.writeFileSync(CONFIG_PATH, JSON.stringify(answers, null, 2));
-  console.log('Configuration saved to ~/.askairc');
+  config.configs[name] = answers;
+  saveConfig(config);
+  console.log(`Config '${name}' updated.`);
+}
+
+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,10 +1,10 @@
 {
   "name": "ask-my-llm",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "Oversimplified AI usage npm module",
   "main": "index.js",
   "bin": {
-    "ask-ai": "./bin/ask-ai"
+    "ask-my-llm": "./bin/ask-ai"
   },
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"