ask-my-llm 1.0.9 → 1.1.1

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

@@ -7,8 +7,15 @@ const CONFIG_PATH = path.join(process.env.HOME || process.env.USERPROFILE, '.ask
 function loadConfig() {
   try {
     const data = JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf8'));
-    const current = data.current || 'default';
-    if (!data.configs || !data.configs[current]) {
+    
+    // Old format: direct baseApi, apiKey, model
+    if (!data.configs) {
+      return data;
+    }
+    
+    // New format with configs
+    const current = data.current || Object.keys(data.configs)[0];
+    if (!data.configs[current]) {
       throw new Error('No config found. Run `npx ask-my-llm` first.');
     }
     return data.configs[current];

package/package.json

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