feat: add groq ai (#108)

Author: afc163

README.md

@@ -2,7 +2,7 @@
 
 # Fanyi
 
-A 🇨🇳 and 🇺🇸🇬🇧 translate tool in your command line, powered by iciba.
+A 🇨🇳 and 🇺🇸🇬🇧 translate tool in your command line, powered by iciba and groq.
 
 [![NPM version][npm-image]][npm-url]
 [![npm download][download-image]][download-url]
@@ -112,25 +112,16 @@ Also, you can use `list` command to see the history of your search.
 
 ## Configuration
 
-A configuration file can be put into ~/.config/fanyi/.fanyirc, in the user's home directory
+A configuration file can be put into `~/.config/fanyi/.fanyirc`, in the user's home directory.
 
-Use subcommand `fanyi config [options]`
+Use subcommand `fanyi config set <key> <value>` to set configuration options.
 
 Example:
 
 ```bash
-# or
-$ fanyi config --no-color // disable color globally
-$ fanyi config --color    // enable color globally
-$ fanyi config --no-iciba // disable iciba globally
-$ fanyi config --iciba // enable iciba globally
-```
-
-A sample `~/.config/fanyi/.fanyirc` file:
-
-```json
-{
-  "iciba": true,
-  "color": true,
-}
-```
+$ fanyi config list                       // list all configuration options
+$ fanyi config set iciba false            // disable iciba globally
+$ fanyi config set groq false             // disable groq globally
+$ fanyi config set color false            // disable color globally
+$ fanyi config set GROQ_API_KEY your-api-key // set GROQ_API_KEY
+```

bin/fanyi.js

@@ -1,42 +1,54 @@
 #!/usr/bin/env node
 
-const { program } = require('commander');
+const { Command } = require('commander');
 const chalk = require('chalk');
 const updateNotifier = require('update-notifier');
 const pkg = require('../package.json');
 const config = require('../lib/config');
 const { searchList } = require('../lib/searchHistory');
 
 updateNotifier({ pkg }).notify();
+const program = new Command();
 
 program
   .name(pkg.name)
   .description(pkg.description)
   .version(pkg.version)
   .action(() => {
-    // If the input is "fanyi", no parameters, ignore.
+    // 如果输入是 "fanyi"，没有参数，则忽略
     if (process.argv.length > 2) {
       return runFY();
     }
   });
 
 program
   .command('config')
-  .description('Set the global options')
-  .option('-c, --color', 'Output with color')
-  .option('-C, --no-color', 'Output without color')
-  .option('-i, --iciba', 'Enable the iciba translation engine')
-  .option('-I, --no-iciba', 'Disable the iciba translation engine')
-  .action((args) => {
-    // hack
-    // If the input is "fanyi config", then translate the word config.
-    if (process.argv.length === 3) {
-      return runFY();
-    }
-    const { color, iciba } = args;
-    const options = resolveOptions({ color, iciba });
-    return config.write(options);
-  });
+  .description('设置全局选项')
+  .addCommand(
+    new Command('list').description('查看配置项').action(async () => {
+      const options = await config.load();
+      console.log(`${chalk.gray(config.getConfigPath())}`);
+      console.log();
+      for (const [key, value] of Object.entries(options)) {
+        console.log(`${chalk.cyan(key)}: ${chalk.yellow(value)}`);
+      }
+    }),
+  )
+  .addCommand(
+    new Command('set')
+      .description('设置配置项')
+      .argument('<key>', '配置项键名')
+      .argument('<value>', '配置项值')
+      .action(async (key, value) => {
+        const options = {};
+        if (key === 'GROQ_API_KEY') {
+          options[key] = value;
+        } else {
+          options[key] = value === 'true' ? true : value === 'false' ? false : value;
+        }
+        await config.write(options);
+      }),
+  );
 
 program
   .command('list')
@@ -49,10 +61,15 @@ program
 
 program.on('--help', () => {
   console.log('');
-  console.log(chalk.gray('Examples:'));
+  console.log(chalk.gray('示例:'));
   console.log(`${chalk.cyan('  $ ')}fanyi word`);
   console.log(`${chalk.cyan('  $ ')}fanyi world peace`);
   console.log(`${chalk.cyan('  $ ')}fanyi chinglish`);
+  console.log(`${chalk.cyan('  $ ')}fanyi config set color true`);
+  console.log(`${chalk.cyan('  $ ')}fanyi config set iciba true`);
+  console.log(`${chalk.cyan('  $ ')}fanyi config set groq true`);
+  console.log(`${chalk.cyan('  $ ')}fanyi config set GROQ_API_KEY your_api_key_here`);
+  console.log(`${chalk.cyan('  $ ')}fanyi config list`);
   console.log('');
 });
 

bun.lockb

@@ -1,47 +1,114 @@
 const needle = require('needle');
-const chalk = require('chalk');
-const SOURCE = require('./lib/source');
+const { Groq } = require('groq-sdk');
 const print = require('./lib/print');
 const parseString = require('xml2js').parseString;
 const ora = require('ora');
 
-module.exports = (word, options, callback) => {
+module.exports = async (word, options) => {
   console.log('');
-  const { iciba } = options;
-  const requestCounts = [iciba].filter(isTrueOrUndefined).length;
-  const spinner = ora().start();
-
-  let count = 0;
-  const callbackAll = () => {
-    count += 1;
-    if (count >= requestCounts) {
-      spinner.stop();
-      spinner.clear();
-      callback?.();
-    }
-  };
-
+  const { iciba, groq, GROQ_API_KEY } = options;
   const endcodedWord = encodeURIComponent(word);
 
   // iciba
-  isTrueOrUndefined(iciba) &&
-    needle.get(
-      SOURCE.iciba.replace('${word}', endcodedWord),
-      { parse: false },
-      (error, response) => {
-        if (error) {
-          console.log(chalk.yellow('访问 iciba 失败，请检查网络'));
-        } else if (response.statusCode === 200) {
-          parseString(response.body, (err, result) => {
-            if (err) {
-              return;
-            }
-            print.iciba(result.dict, options);
+  if (isTrueOrUndefined(iciba)) {
+    const ICIBA_URL =
+      'http://dict-co.iciba.com/api/dictionary.php?key=D191EBD014295E913574E1EAF8E06666&w=';
+    const spinner = ora('正在查询 iciba...').start();
+    try {
+      const response = await needle('get', `${ICIBA_URL}${endcodedWord}`, { parse: false });
+      if (response.statusCode === 200) {
+        const result = await new Promise((resolve, reject) => {
+          parseString(response.body, (err, res) => {
+            if (err) reject(err);
+            else resolve(res);
           });
-        }
-        callbackAll();
-      },
-    );
+        });
+        spinner.stop();
+        print.iciba(result.dict, options);
+      }
+    } catch (error) {
+      spinner.fail('访问 iciba 失败，请检查网络');
+    }
+  }
+
+  // groq ai
+  if (isTrueOrUndefined(groq)) {
+    const groqClient = new Groq({
+      apiKey: GROQ_API_KEY || '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',
+    });
+    const model = 'llama3-groq-70b-8192-tool-use-preview';
+
+    const spinner = ora('正在查询 Groq AI...').start();
+    try {
+      const chatCompletion = await groqClient.chat.completions.create({
+        messages: [
+          {
+            role: 'system',
+            content: `
+你是一本专业的中英文双语词典。请按照以下要求提供翻译和解释：
+
+1. 格式要求：
+   [原词] [音标] ~ [翻译] [拼音]
+   
+   - [词性] [释义1]
+   - [词性] [释义2]
+   ...
+
+   例句：
+   1. [原文例句]
+      [翻译]
+   2. [原文例句]
+      [翻译]
+   ...
+
+   ❤️
+   [座右铭]
+   -----
+
+2. 翻译规则：
+   - 英文输入翻译为中文，中文输入翻译为英文
+   - 提供准确的音标（英文）或拼音（中文）
+   - 列出所有常见词性及其对应的释义
+   - 释义应简洁明了，涵盖词语的主要含义
+   - 提供2-3个地道的例句，体现词语的不同用法和语境
+
+3. 内容质量：
+   - 确保翻译和释义的准确性和权威性
+   - 例句应当实用、常见，并能体现词语的典型用法
+   - 注意词语的语体色彩，如正式、口语、书面语等
+   - 对于多义词，按照使用频率由高到低排列释义
+
+4. 特殊情况：
+   - 对于习语、谚语或特殊表达，提供对应的解释和等效表达
+   - 注明词语的使用范围，如地域、行业特定用语等
+   - 对于缩写词，提供完整形式和解释
+
+请基于以上要求，为用户提供简洁、专业、全面且易于理解的词语翻译和解释。
+
+---
+最后使用这个词写一句简短的积极向上令人深思的英文座右铭。
+`,
+          },
+          {
+            role: 'user',
+            content: `请翻译：${word}`,
+          },
+        ],
+        model,
+        temperature: 0.3,
+        max_tokens: 1024,
+        top_p: 0.8,
+        stream: true,
+        stop: null,
+      });
+      spinner.stop();
+      for await (const chunk of chatCompletion) {
+        process.stdout.write(chunk.choices[0]?.delta?.content || '');
+      }
+    } catch (error) {
+      spinner.fail('访问 Groq AI 失败，请检查网络或 API 密钥');
+    }
+  }
 };
 
 function isTrueOrUndefined(val) {

index.js

@@ -31,8 +31,12 @@ const config = {
     console.log(
       `${chalkInstance.bgGreen(JSON.stringify(options))} config saved at ${chalkInstance.gray(path)}:`,
     );
-    console.log();
-    console.log(chalkInstance.greenBright(content));
+    for (const [key, value] of Object.entries(options)) {
+      console.log(`${chalk.cyan(key)}: ${chalk.yellow(value)}`);
+    }
+  },
+  getConfigPath() {
+    return configPath;
   },
 };
 

lib/config.js

@@ -30,6 +30,10 @@ exports.iciba = (data, options = {}) => {
 
   log(firstLine + chalk.gray(' ~  iciba.com'));
 
+  if (data.pos?.length) {
+    log();
+  }
+
   // pos & acceptation
   data.pos?.forEach((item, i) => {
     if (typeof data.pos[i] !== 'string' || !data.pos[i]) {
@@ -64,7 +68,7 @@ exports.iciba = (data, options = {}) => {
 function log(message, indentNum = 1) {
   let indent = '';
   for (let i = 1; i < indentNum; i += 1) {
-    indent += '  ';
+    indent += ' ';
   }
   console.log(indent, message || '');
 }

lib/print.js

@@ -34,12 +34,13 @@
     "commander": "^12.1.0",
     "dayjs": "^1.11.5",
     "fs-extra": "^10.1.0",
+    "groq-sdk": "^0.7.0",
     "needle": "^3.1.0",
     "update-notifier": "^4.1.0",
     "xml2js": "^0.4.8"
   },
   "lint-staged": {
-    "*": [
+    "*.{js,ts,json,yml}": [
       "biome check --write --files-ignore-unknown=true",
       "biome format --write --files-ignore-unknown=true",
       "biome lint --write --files-ignore-unknown=true"