viho 0.0.6 → 0.0.7

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>
@@ -156,13 +156,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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viho",
-  "version": "0.0.6",
+  "version": "0.0.7",
   "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": "b4d22881578a4da9c83d057f08b044bdd44c692a"
 }

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(`
+██╗   ██╗██╗██╗  ██╗ ██████╗
+██║   ██║██║██║  ██║██╔═══██╗
+██║   ██║██║███████║██║   ██║
+╚██╗ ██╔╝██║██╔══██║██║   ██║
+ ╚████╔╝ ██║██║  ██║╚██████╔╝
+  ╚═══╝  ╚═╝╚═╝  ╚═╝ ╚═════╝
+    `),
+  );
+};