ask-my-llm 1.0.6 → 1.0.8

package/README.md

@@ -24,10 +24,14 @@ Config saved to `~/.askairc`.
 ## Usage
 
 ```js
-const { ask } = require("ask-my-llm");
+const { ask, askAsync } = require("ask-my-llm");
 
+// sync
 const response = ask("Hello");
 console.log(response);
+
+// async
+askAsync("Hello").then(console.log);
 ```
 
 ## API
@@ -38,3 +42,10 @@ Sends a prompt to the configured AI and returns the response synchronously.
 
 - `prompt` - The message to send
 - Returns: AI response as string
+
+### askAsync(prompt: string): Promise<string>
+
+Sends a prompt to the configured AI and returns a Promise.
+
+- `prompt` - The message to send
+- Returns: Promise that resolves to AI response as string

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

@@ -1,4 +1,4 @@
-const { spawnSync } = require('child_process');
+const { spawnSync, spawn } = require('child_process');
 const fs = require('fs');
 const path = require('path');
 
@@ -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.');
   }
 }
 
@@ -48,12 +53,64 @@ function ask(prompt) {
   if (result.error) throw result.error;
   if (result.status !== 0) throw new Error('curl failed: ' + (result.stderr || result.stdout));
   
-  const parsed = JSON.parse(result.stdout);
+  return parseResponse(result.stdout);
+}
+
+function askAsync(prompt) {
+  return new Promise((resolve, reject) => {
+    const config = loadConfig();
+    
+    const postData = JSON.stringify({
+      model: config.model,
+      messages: [{ role: 'user', content: prompt }],
+      temperature: 0.7
+    });
+
+    let baseUrl = config.baseApi.replace(/\/$/, '');
+    if (!baseUrl.endsWith('/v1')) {
+      baseUrl += '/v1';
+    }
+    const url = baseUrl + '/chat/completions';
+    
+    const headerArgs = [
+      '-s', '-S', '-X', 'POST',
+      '-H', 'Content-Type: application/json',
+      '-H', 'HTTP-Referer: https://cows.info.gf',
+      '-H', 'X-OpenRouter-Title: ask-my-llm'
+    ];
+    if (config.apiKey) {
+      headerArgs.push('-H', `Authorization: Bearer ${config.apiKey}`);
+    }
+    headerArgs.push('-d', postData, url);
+
+    const child = spawn('curl', headerArgs, { encoding: 'utf8' });
+    let data = '';
+    let error = '';
+
+    child.stdout.on('data', chunk => data += chunk);
+    child.stderr.on('data', chunk => error += chunk);
+    child.on('close', code => {
+      if (code !== 0) {
+        reject(new Error('curl failed: ' + (error || data)));
+      } else {
+        try {
+          resolve(parseResponse(data));
+        } catch (e) {
+          reject(e);
+        }
+      }
+    });
+    child.on('error', reject);
+  });
+}
+
+function parseResponse(stdout) {
+  const parsed = JSON.parse(stdout);
   if (parsed.choices && parsed.choices[0]) {
     return parsed.choices[0].message.content;
   } else {
-    throw new Error('API Error: ' + result.stdout);
+    throw new Error('API Error: ' + stdout);
   }
 }
 
-module.exports = { ask };
+module.exports = { ask, askAsync };

package/package.json

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