viho 0.0.9 → 0.1.1

package/src/experts/experts.js

@@ -0,0 +1,10 @@
+exports.experts = [
+  {
+    name: 'antd',
+    url: 'https://ant.design/llms-full.txt',
+  },
+  {
+    name: 'daisyui',
+    url: 'https://daisyui.com/llms.txt',
+  },
+];

package/src/llm.js

@@ -1,12 +1,20 @@
+// path
+const path = require('path');
+
 // qiao
 const cli = require('qiao-cli');
+const { readFile } = require('qiao-file');
+
+// prompt
+const defaultSystemPrompt = 'You are a helpful AI assistant';
 
 /**
  * ask
  * @param {*} llm
  * @param {*} model
+ * @param {*} systemPrompt
  */
-exports.ask = async (llm, model) => {
+exports.ask = async (llm, model, systemPrompt) => {
   // ask
   const questions = [
     {
@@ -27,7 +35,7 @@ exports.ask = async (llm, model) => {
   const chatOptions = {
     model: model.modelID,
     messages: [
-      { role: 'system', content: 'You are a helpful AI assistant' },
+      { role: 'system', content: systemPrompt || defaultSystemPrompt },
       { role: 'user', content: answers.content },
     ],
     thinking: {
@@ -67,3 +75,18 @@ exports.ask = async (llm, model) => {
   // go
   await llm.chatWithStreaming(chatOptions, callbackOptions);
 };
+
+/**
+ * expertAsk
+ * @param {*} llm
+ * @param {*} model
+ * @param {*} expertName
+ */
+exports.expertAsk = async (llm, model, expertName) => {
+  // llms.txt
+  const txtPath = path.resolve(__dirname, `./experts/${expertName}.md`);
+  const txtContent = await readFile(txtPath);
+
+  const systemPrompt = `${defaultSystemPrompt}. You are an expert on ${expertName}. Use the following documentation to answer questions:\n\n${txtContent}`;
+  await exports.ask(llm, model, systemPrompt);
+};

package/src/util.js

@@ -10,6 +10,9 @@ const cli = require('qiao-cli');
 // db
 const DB = require('qiao-config');
 
+// model
+const { getModelByName } = require('./model.js');
+
 /**
  * getDB
  * @returns
@@ -34,3 +37,24 @@ exports.printLogo = () => {
     `),
   );
 };
+
+/**
+ * preLLMAsk
+ * @param {*} type
+ * @param {*} db
+ * @param {*} modelName
+ * @returns
+ */
+exports.preLLMAsk = async (type, db, modelName) => {
+  // model
+  const model = await getModelByName(db, modelName);
+  if (!model) return;
+
+  // logo
+  exports.printLogo();
+  console.log(cli.colors.cyan(`Welcome to viho ${type}! Using model: ${model.modelName}`));
+  console.log(cli.colors.gray('Press Ctrl+C to exit\n'));
+
+  // r
+  return model;
+};