ask-my-llm 1.1.0 → 1.1.2

package/bin/ask-ai

@@ -3,6 +3,7 @@
 const inquirer = require('inquirer');
 const fs = require('fs');
 const path = require('path');
+const { ask } = require('../index.js');
 
 const CONFIG_PATH = path.join(process.env.HOME || process.env.USERPROFILE, '.askairc');
 
@@ -18,6 +19,20 @@ function saveConfig(config) {
   fs.writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2));
 }
 
+async function testConfig(config) {
+  const { prompt } = await inquirer.prompt([{
+    type: 'input',
+    name: 'prompt',
+    message: 'Enter prompt:'
+  }]);
+
+  try {
+    console.log('Response:', ask(prompt));
+  } catch (e) {
+    console.log('Error:', e.message);
+  }
+}
+
 async function mainMenu() {
   let config = loadConfig();
   
@@ -27,7 +42,7 @@ async function mainMenu() {
     return;
   }
 
-  const choices = ['Add new config', 'Select config', 'Edit config', 'Remove config'];
+  const choices = ['Add new config', 'Select config', 'Edit config', 'Remove config', 'Test config'];
 
   const { action } = await inquirer.prompt([{
     type: 'list',
@@ -44,6 +59,8 @@ async function mainMenu() {
     await editConfig(config);
   } else if (action === 'Remove config') {
     await removeConfig(config);
+  } else if (action === 'Test config') {
+    await testConfig(config);
   }
 }
 

package/index.js

@@ -2,11 +2,26 @@ const { spawnSync, spawn } = require('child_process');
 const fs = require('fs');
 const path = require('path');
 
-const CONFIG_PATH = path.join(process.env.HOME || process.env.USERPROFILE, '.askairc');
+function getDefaultConfigPath() {
+  return path.join(process.env.HOME || process.env.USERPROFILE, '.askairc');
+}
 
-function loadConfig() {
+function loadConfig(options) {
+  let configPath;
+  let configName;
+  
+  if (typeof options === 'string') {
+    // String is config file path
+    configPath = options;
+  } else if (options && typeof options === 'object') {
+    configPath = options.configPath || getDefaultConfigPath();
+    configName = options.configName;
+  } else {
+    configPath = getDefaultConfigPath();
+  }
+  
   try {
-    const data = JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf8'));
+    const data = JSON.parse(fs.readFileSync(configPath, 'utf8'));
     
     // Old format: direct baseApi, apiKey, model
     if (!data.configs) {
@@ -14,9 +29,9 @@ function loadConfig() {
     }
     
     // New format with configs
-    const current = data.current || Object.keys(data.configs)[0];
+    const current = configName || data.current || Object.keys(data.configs)[0];
     if (!data.configs[current]) {
-      throw new Error('No config found. Run `npx ask-my-llm` first.');
+      throw new Error(`Config '${current}' not found.`);
     }
     return data.configs[current];
   } catch (e) {
@@ -24,8 +39,8 @@ function loadConfig() {
   }
 }
 
-function ask(prompt) {
-  const config = loadConfig();
+function ask(prompt, options) {
+  const config = loadConfig(options);
   
   const postData = JSON.stringify({
     model: config.model,
@@ -63,9 +78,9 @@ function ask(prompt) {
   return parseResponse(result.stdout);
 }
 
-function askAsync(prompt) {
+function askAsync(prompt, options) {
   return new Promise((resolve, reject) => {
-    const config = loadConfig();
+    const config = loadConfig(options);
     
     const postData = JSON.stringify({
       model: config.model,

package/package.json

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