viho 0.0.6 → 0.0.8

package/README.md

@@ -1,5 +1,5 @@
 <p align="center">
-  <img src="./logo.png" alt="viho logo" width="200"/>
+  <img src="https://static-small.vincentqiao.com/viho/logo.png" alt="viho logo" width="200"/>
 </p>
 
 <h1 align="center">viho</h1>
@@ -74,14 +74,39 @@ You'll be prompted to enter:
 viho model add
 ```
 
+After adding a model, it will be available for use with `viho ask` and `viho chat` commands.
+
 #### `viho model list`
 
-List all configured models:
+List all configured models with detailed information.
 
 ```bash
 viho model list
 ```
 
+This command displays:
+
+- Model name with a `(default)` tag for the default model
+- Model ID
+- Base URL
+- Thinking mode setting
+
+**Example output:**
+
+```
+Configured models:
+
+  • deepseek (default)
+    Model ID: ep-20250822181529-2gg27
+    Base URL: https://ark.cn-beijing.volces.com/api/v3
+    Thinking: auto
+
+  • kimi
+    Model ID: kimi-k2-thinking
+    Base URL: https://api.moonshot.cn/v1
+    Thinking: enabled
+```
+
 #### `viho model remove`
 
 Remove a model configuration:
@@ -90,14 +115,18 @@ Remove a model configuration:
 viho model remove
 ```
 
+You'll be prompted to enter the model name to remove. If the removed model was set as default, you'll need to set a new default model.
+
 #### `viho model default`
 
-Set a default model for chat sessions:
+Set a default model for chat and ask sessions:
 
 ```bash
 viho model default
 ```
 
+The default model will be used when you run `viho ask` or `viho chat` without specifying a model name.
+
 ### Ask
 
 #### `viho ask [modelName]`
@@ -156,13 +185,15 @@ Configuration is stored in `~/viho.json`. You can manage all settings through th
 
 ```json
 {
-  "mymodel": {
-    "modelName": "mymodel",
-    "apiKey": "your-api-key",
-    "baseURL": "https://api.openai.com/v1",
-    "modelID": "gpt-4",
-    "modelThinking": "auto"
-  },
+  "models": [
+    {
+      "modelName": "mymodel",
+      "apiKey": "your-api-key",
+      "baseURL": "https://api.openai.com/v1",
+      "modelID": "gpt-4",
+      "modelThinking": "auto"
+    }
+  ],
   "default": "mymodel"
 }
 ```

package/bin/viho-ask.js

@@ -6,7 +6,7 @@ const LLM = require('qiao-llm');
 
 // util
 const { ask } = require('../src/llm.js');
-const { getModels } = require('../src/model.js');
+const { getModelByName } = require('../src/model.js');
 const { getDB, printLogo } = require('../src/util.js');
 const db = getDB();
 
@@ -15,25 +15,11 @@ cli.cmd
   .command('ask [modelName]')
   .description('Ask a question to an AI model')
   .action(async (modelName) => {
-    if (!modelName) {
-      const defaultModel = await db.config('default');
-      if (!defaultModel) {
-        console.log(cli.colors.red('No default model set. Use: viho model default'));
-        return;
-      }
+    // model
+    const model = await getModelByName(db, modelName);
+    if (!model) return;
 
-      modelName = defaultModel;
-    }
-
-    // check
-    const models = await getModels(db);
-    const model = models.find((m) => m.modelName === modelName);
-    if (!model) {
-      console.log(cli.colors.red(`Model not found: ${modelName}`));
-      return;
-    }
-
-    // init
+    // llm
     const llm = LLM({
       apiKey: model.apiKey,
       baseURL: model.baseURL,

package/bin/viho-chat.js

@@ -6,7 +6,7 @@ const LLM = require('qiao-llm');
 
 // util
 const { ask } = require('../src/llm.js');
-const { getModels } = require('../src/model.js');
+const { getModelByName } = require('../src/model.js');
 const { getDB, printLogo } = require('../src/util.js');
 const db = getDB();
 
@@ -15,23 +15,9 @@ cli.cmd
   .command('chat [modelName]')
   .description('Start a continuous chat session with an AI model')
   .action(async (modelName) => {
-    if (!modelName) {
-      const defaultModel = await db.config('default');
-      if (!defaultModel) {
-        console.log(cli.colors.red('No default model set. Use: viho model default'));
-        return;
-      }
-
-      modelName = defaultModel;
-    }
-
-    // check
-    const models = await getModels(db);
-    const model = models.find((m) => m.modelName === modelName);
-    if (!model) {
-      console.log(cli.colors.red(`Model not found: ${modelName}`));
-      return;
-    }
+    // model
+    const model = await getModelByName(db, modelName);
+    if (!model) return;
 
     // init
     const llm = LLM({
@@ -41,7 +27,7 @@ cli.cmd
 
     // logo
     printLogo();
-    console.log(cli.colors.cyan(`Welcome to viho chat! Using model: ${modelName}`));
+    console.log(cli.colors.cyan(`Welcome to viho chat! Using model: ${model.modelName}`));
     console.log(cli.colors.gray('Press Ctrl+C to exit\n'));
 
     // chat

package/bin/viho-model.js

@@ -58,7 +58,7 @@ async function modelAdd() {
         type: 'list',
         name: 'modelThinking',
         message: 'Thinking mode:',
-        choices: ['enabled', 'disabled', 'auto'],
+        choices: ['enabled', 'disabled'],
       },
     ];
     const answers = await cli.ask(questions);
@@ -76,12 +76,8 @@ async function modelAdd() {
     // set
     models.push(answers);
     await setModels(db, models);
-    console.log(cli.colors.green('Model added'));
+    console.log(cli.colors.green('Model added successfully!'));
     console.log();
-
-    // list
-    const allModels = await getModels(db);
-    console.log(allModels);
   } catch (e) {
     console.log(cli.colors.red('Error: Failed to add model'));
     console.log();
@@ -99,9 +95,31 @@ async function modelList() {
 
     // list
     const models = await getModels(db);
+    const defaultModel = await db.config('default');
+
     console.log(cli.colors.cyan('Configured models:'));
     console.log();
-    console.log(models);
+
+    if (!models || models.length === 0) {
+      console.log(cli.colors.gray('No models configured. Use: viho model add'));
+      console.log();
+      return;
+    }
+
+    models.forEach((model) => {
+      const isDefault = model.modelName === defaultModel;
+      const defaultTag = isDefault ? cli.colors.green(' (default)') : '';
+      console.log(cli.colors.white(`  • ${model.modelName}${defaultTag}`));
+      console.log(cli.colors.gray(`    Model ID: ${model.modelID}`));
+      console.log(cli.colors.gray(`    Base URL: ${model.baseURL}`));
+      console.log(cli.colors.gray(`    Thinking: ${model.modelThinking}`));
+      console.log();
+    });
+
+    if (!defaultModel) {
+      console.log(cli.colors.yellow('No default model set. Use: viho model default'));
+      console.log();
+    }
   } catch (e) {
     console.log(cli.colors.red('Error: Failed to list models'));
     console.log();
@@ -149,12 +167,8 @@ async function modelRemove() {
       console.log();
     }
 
-    console.log(cli.colors.green('Model removed'));
+    console.log(cli.colors.green('Model removed successfully!'));
     console.log();
-
-    // list
-    const allModels = await getModels(db);
-    console.log(allModels);
   } catch (e) {
     console.log(cli.colors.red('Error: Failed to remove model'));
     console.log();
@@ -195,13 +209,16 @@ async function modelDefault() {
     if (!keys.includes(answers.modelName)) {
       console.log(cli.colors.red('Model not found. Available models:'));
       console.log();
-      console.log(models);
+      models.forEach((model) => {
+        console.log(cli.colors.gray(`  • ${model.modelName}`));
+      });
+      console.log();
       return;
     }
 
     // set
     await db.config('default', answers.modelName);
-    console.log(cli.colors.green(`Default model: ${answers.modelName}`));
+    console.log(cli.colors.green(`Default model set to: ${answers.modelName}`));
     console.log();
   } catch (e) {
     console.log(cli.colors.red('Error: Failed to set default model'));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viho",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "description": "A lightweight CLI tool for managing and interacting with AI models",
   "keywords": [
     "ai",
@@ -31,9 +31,9 @@
   },
   "files": [
     "bin",
+    "src",
     "LICENSE",
-    "README.md",
-    "logo.png"
+    "README.md"
   ],
   "engines": {
     "node": ">=18.0.0"
@@ -43,5 +43,5 @@
     "qiao-config": "^5.0.1",
     "qiao-llm": "^0.2.5"
   },
-  "gitHead": "2a10a0acfc92501a023e93f7f268e6f8ad57f9fc"
+  "gitHead": "6e3c00750d92bc6da1e84096ca4afe2f7f51624e"
 }

package/src/llm.js

@@ -0,0 +1,69 @@
+// qiao
+const cli = require('qiao-cli');
+
+/**
+ * ask
+ * @param {*} llm
+ * @param {*} model
+ */
+exports.ask = async (llm, model) => {
+  // ask
+  const questions = [
+    {
+      type: 'editor',
+      name: 'content',
+      message: 'Your question:',
+    },
+  ];
+  const answers = await cli.ask(questions);
+
+  // answers
+  console.log();
+  console.log(cli.colors.gray('Question:'));
+  console.log(cli.colors.gray(answers.content));
+  console.log();
+
+  // chat
+  const chatOptions = {
+    model: model.modelID,
+    messages: [
+      { role: 'system', content: 'You are a helpful AI assistant' },
+      { role: 'user', content: answers.content },
+    ],
+    thinking: {
+      type: model.modelThinking,
+    },
+  };
+
+  // callback options
+  const callbackOptions = {
+    firstThinkingCallback: () => {
+      console.log();
+      console.log(cli.colors.gray('[Thinking...]'));
+      console.log();
+    },
+    thinkingCallback: (msg) => {
+      process.stdout.write(cli.colors.gray(msg));
+    },
+    firstContentCallback: () => {
+      console.log();
+      console.log(cli.colors.cyan('[Response]'));
+      console.log();
+    },
+    contentCallback: (msg) => {
+      process.stdout.write(msg);
+    },
+    endCallback: () => {
+      console.log();
+      console.log();
+    },
+    errorCallback: (error) => {
+      console.log();
+      console.log(cli.colors.red('Error:'));
+      console.log(error);
+    },
+  };
+
+  // go
+  await llm.chatWithStreaming(chatOptions, callbackOptions);
+};

package/src/model.js

@@ -0,0 +1,51 @@
+// qiao
+const cli = require('qiao-cli');
+
+// models
+const modelsKey = 'models';
+
+/**
+ * getModels
+ * @param {*} db
+ * @returns
+ */
+exports.getModels = async (db) => {
+  return (await db.config(modelsKey)) || [];
+};
+
+/**
+ * setModels
+ * @param {*} db
+ * @returns
+ */
+exports.setModels = async (db, models) => {
+  await db.config(modelsKey, models);
+};
+
+/**
+ * getModelByName
+ * @param {*} db
+ * @returns
+ */
+exports.getModelByName = async (db, modelName) => {
+  if (!modelName) {
+    const defaultModel = await db.config('default');
+    if (!defaultModel) {
+      console.log(cli.colors.red('No default model set. Use: viho model default'));
+      return;
+    }
+
+    modelName = defaultModel;
+  }
+
+  // check
+  const models = await exports.getModels(db);
+  const model = models.find((m) => m.modelName === modelName);
+  if (!model) {
+    console.log(cli.colors.red(`Model not found: ${modelName}`));
+    return;
+  }
+
+  // r
+  return model;
+};

package/src/util.js

@@ -0,0 +1,36 @@
+// os
+const os = require('os');
+
+// path
+const path = require('path');
+
+// qiao
+const cli = require('qiao-cli');
+
+// db
+const DB = require('qiao-config');
+
+/**
+ * getDB
+ * @returns
+ */
+exports.getDB = () => {
+  const dbPath = path.resolve(os.homedir(), './viho.json');
+  return DB(dbPath);
+};
+
+/**
+ * printLogo
+ */
+exports.printLogo = () => {
+  console.log(
+    cli.colors.green(`
+██╗   ██╗██╗██╗  ██╗ ██████╗
+██║   ██║██║██║  ██║██╔═══██╗
+██║   ██║██║███████║██║   ██║
+╚██╗ ██╔╝██║██╔══██║██║   ██║
+ ╚████╔╝ ██║██║  ██║╚██████╔╝
+  ╚═══╝  ╚═╝╚═╝  ╚═╝ ╚═════╝
+    `),
+  );
+};