From 30bf989aa3965847769cfe2f1cdd3af613b91ebe Mon Sep 17 00:00:00 2001
From: shanmite <shanmite@hotmail.com>
Date: Tue, 19 Aug 2025 15:09:25 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20ai=E8=AF=84=E8=AE=BA=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?=E6=89=80=E6=9C=89=E5=85=BC=E5=AE=B9OpenAI=20API=E7=9A=84?=
 =?UTF-8?q?=E5=B9=B3=E5=8F=B0=20(#463)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Fixed #463
---
 README.md            |  2 +-
 env.example.js       | 29 +++++++++++++++++------------
 lib/core/monitor.js  | 10 +++++++---
 lib/data/config.js   |  6 ++++++
 lib/utils.js         | 34 +++++++++++++++++-----------------
 my_config.example.js | 40 +++++++++++++++++++++++++++++++++-------
 6 files changed, 81 insertions(+), 40 deletions(-)

diff --git a/README.md b/README.md
index 6e5fa2d..62b385b 100644
--- a/README.md
+++ b/README.md
@@ -33,7 +33,7 @@
 - 监控用户转发
 - 监控话题页面
 - 监控专栏合集
-- 自动点赞、评论、乱序转发、@好友、带话题、可选随机动态
+- 自动点赞、AI评论、乱序转发、@好友、带话题、可选随机动态
 - 直播预约抽奖
 - 检测是否中奖
   - 已读@
diff --git a/env.example.js b/env.example.js
index 5195544..c79c139 100644
--- a/env.example.js
+++ b/env.example.js
@@ -6,16 +6,19 @@ module.exports = Object.freeze({
      * - `NUMBER` 表示是第几个账号
      * - `CLEAR` 是否启用清理功能
      * - `ACCOUNT_UA` 账号UA, 可在浏览器控制台输入 navigator.userAgent 查看
+     * 
      * ## 高级功能
      * - `ENABLE_CHAT_CAPTCHA_OCR` 开启评论验证码识别 使用方法见README
      * - `CHAT_CAPTCHA_OCR_URL` 验证码识别接口 POST `url`->`code`
-     * - `ENABLE_MULTIPLE_ACCOUNT` 是否启用多账号
-     * - `MULTIPLE_ACCOUNT_PARM` 多账号参数(JSON格式) <不推荐使用
+     * - `ENABLE_AI_COMMENTS` 是否启用AI评论
+     * 
      * ## 调试相关
      * - `LOTTERY_LOG_LEVEL` 输出日志等级 Error<Warn<Notice<Info<Debug 0<1<2<3<4
      * - `NOT_GO_LOTTERY` 关闭抽奖行为
      *
      * ## 多账号
+     * - `ENABLE_MULTIPLE_ACCOUNT` 是否启用多账号
+     * - `MULTIPLE_ACCOUNT_PARM` 多账号参数(JSON格式) <不推荐使用
      * 1. 将 ENABLE_MULTIPLE_ACCOUNT 的值改为true
      * 2. 将账号信息依次填写于 multiple_account_parm 中, 参考例子类推
      * - `WAIT` 表示下一个账号运行等待时间(毫秒)
@@ -31,9 +34,11 @@ module.exports = Object.freeze({
 
         ENABLE_CHAT_CAPTCHA_OCR: false,
         CHAT_CAPTCHA_OCR_URL: 'http://127.0.0.1:9898/ocr/url/text',
-        ENABLE_MULTIPLE_ACCOUNT: false,
+        ENABLE_AI_COMMENTS: false,
 
+        ENABLE_MULTIPLE_ACCOUNT: false,
         MULTIPLE_ACCOUNT_PARM: '',
+
         LOTTERY_LOG_LEVEL: 3,
         NOT_GO_LOTTERY: ''
     },
@@ -65,10 +70,10 @@ module.exports = Object.freeze({
             CLEAR: true,
             WAIT: 60 * 1000,
             ACCOUNT_UA: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36',
-            PROXY_HOST:'',//代理ip
-            PROXY_PORT:'',//代理ip端口
-            PROXY_USER:'',//代理ip账号
-            PROXY_PASS:'',//代理ip密码
+            PROXY_HOST: '',//代理ip
+            PROXY_PORT: '',//代理ip端口
+            PROXY_USER: '',//代理ip账号
+            PROXY_PASS: '',//代理ip密码
         }
     ],
 
@@ -107,12 +112,12 @@ module.exports = Object.freeze({
     },
 
     /**
-     * ai相关参数
+     * AI Authentication(OpenAI 兼容的 API 格式)
+     * Chat completions
+     * 此处填写Key, 在my_config中的ai_comments_parm中填写API地址等信息
      */
     ai_parm: {
-        //[硅基流动](https://siliconflow.cn/) apikey
-        SILICON_FLOW_API_KEY:'',
-        //提示词
-        PROMPT:''
+        //apikey
+        AI_API_KEY: '',
     }
 });
diff --git a/lib/core/monitor.js b/lib/core/monitor.js
index 686d4bb..896ac7e 100644
--- a/lib/core/monitor.js
+++ b/lib/core/monitor.js
@@ -281,7 +281,7 @@ class Monitor extends Searcher {
                 reserve_lottery_wait, sneaktower, key_words,
                 model, chatmodel, chat: chats, relay: relays,
                 block_dynamic_type, max_create_time, is_imitator,
-                only_followed, at_users, blockword, blacklist, use_ai_comments
+                only_followed, at_users, blockword, blacklist, ai_comments_parm
             } = config,
             now_ts = Date.now() / 1000;
 
@@ -527,10 +527,14 @@ class Monitor extends Searcher {
                 /* 是否评论 */
                 if (isSendChat) {
                     onelotteryinfo.rid = rid;
-                    if (use_ai_comments) {
+                    if (hasEnv('ENABLE_AI_COMMENTS')) {
                         try {
                             log.info('开始获取Ai评论', `(https://t.bilibili.com/${dyid})`);
-                            onelotteryinfo.chat = await getAiContent(lottery_info.des) || '!!!';
+                            onelotteryinfo.chat = await getAiContent(
+                                ai_comments_parm.url,
+                                ai_comments_parm.body,
+                                ai_comments_parm.prompt,
+                                lottery_info.des) || '!!!';
                             log.info('获取到Ai评论内容', `${onelotteryinfo.chat}`);
                         } catch (e) {
                             log.error('获取AI评论失败，使用随机评论', e);
diff --git a/lib/data/config.js b/lib/data/config.js
index dc7e824..e92f185 100644
--- a/lib/data/config.js
+++ b/lib/data/config.js
@@ -332,6 +332,12 @@ const config = {
         '坚持不懈，迎难而上，开拓创新！', '[OK][OK]', '我来抽个奖', '中中中中中中', '[doge][doge][doge]', '我我我',
     ],
 
+    ai_comments_parm: {
+        url: '',
+        body: {},
+        prompt: ''
+    },
+
     /**
      * 是否抄热评
      */
diff --git a/lib/utils.js b/lib/utils.js
index c77bcd4..6fc2b36 100644
--- a/lib/utils.js
+++ b/lib/utils.js
@@ -511,36 +511,36 @@ const utils = {
     },
     /**
      * 获取ai评论
+     * @param {string} url
+     * @param {object} body
+     * @param {string} prompt
      * @param {string} content
      * @returns {Promise<string|null>}
      */
-    getAiContent(content) {
+    getAiContent(url, body, prompt, content) {
         return new Promise((resolve) => {
             send({
                 method: 'POST',
-                url: 'https://api.siliconflow.cn/v1/chat/completions',
+                url,
                 headers: {
-                    'authorization': 'Bearer ' + process.env.SILICON_FLOW_API_KEY,
+                    'authorization': 'Bearer ' + process.env.AI_API_KEY,
                     'content-type': 'application/json'
                 },
                 contents: {
-                    model: 'Qwen/Qwen3-32B',
+                    ...body,
                     'stream': false,
-                    'max_tokens': 512,
                     'enable_thinking': true,
-                    'thinking_budget': 4096,
-                    'min_p': 0.05,
-                    'temperature': 0.7,
-                    'top_p': 0.7,
-                    'top_k': 50,
-                    'frequency_penalty': 0.5,
-                    'n': 1,
-                    'stop': [],
                     'response_format': { 'type': 'text' },
-                    'messages': [{
-                        'role': 'system',
-                        'content': process.env.PROMPT
-                    }, { 'role': 'user', 'content': content }]
+                    'messages': [
+                        {
+                            'role': 'system',
+                            'content': prompt
+                        },
+                        {
+                            'role': 'user',
+                            'content': content
+                        }
+                    ]
                 },
                 success: res => {
                     const data = utils.strToJson(res.body);
diff --git a/my_config.example.js b/my_config.example.js
index 194411f..964ac76 100644
--- a/my_config.example.js
+++ b/my_config.example.js
@@ -341,6 +341,19 @@ module.exports = Object.freeze({
             '坚持不懈，迎难而上，开拓创新！', '[OK][OK]', '我来抽个奖', '中中中中中中', '[doge][doge][doge]', '我我我',
         ],
 
+        /**
+         * AI Chat completions参数
+         * https://learn.microsoft.com/en-us/azure/ai-foundry/openai/reference#chat-completions
+         */
+        ai_comments_parm: {
+            /**
+             * /chat/completions
+             */
+            url: '',
+            body: {},
+            prompt: ''
+        },
+
         /**
          * 是否抄热评
          */
@@ -460,13 +473,6 @@ module.exports = Object.freeze({
          * [1,2,4]
          */
         clear_dynamic_type: [1],
-        /**
-         * 是否使用ai评论。
-         * true:使用
-         * false:不使用
-         * 如需使用需要再env.js配置ai_parm
-         */
-        use_ai_comments: false
     },
 
     /**
@@ -493,6 +499,26 @@ module.exports = Object.freeze({
 
         APIs: [],
 
+        /**
+         * 默认为硅基流动，可以修改为其他AI服务
+         */
+        ai_comments_parm: {
+            url: 'https://api.siliconflow.cn/v1/chat/completions',
+            body: {
+                'model': 'Qwen/Qwen3-32B',
+                'max_tokens': 512,
+                'enable_thinking': true,
+                'thinking_budget': 4096,
+                'min_p': 0.05,
+                'temperature': 0.7,
+                'top_p': 0.7,
+                'top_k': 50,
+                'frequency_penalty': 0.5,
+                'n': 1,
+            },
+            prompt: '请根据以下内容直接生成一条简短评论，无需说明信息，且不包含任何敏感词汇。'
+        },
+
         save_lottery_info_to_file: true,
     },
     config_2: {},