feat: 添加图片质量参数TELEGRAM_PHOTO_SIZE_OFFSET

TBXark · Aug 1, 2024 · 71b1b0a · 71b1b0a
1 parent f986aa6
commit 71b1b0a
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/src/config/env.js b/src/config/env.js
@@ -134,6 +134,9 @@ class Environment {
     DEFAULT_PARSE_MODE = 'Markdown';
     // 最小stream模式消息间隔，小于等于0则不限制
     TELEGRAM_MIN_STREAM_INTERVAL = 0;
+    // 图片尺寸偏移 0为第一位，-1为最后一位, 越靠后的图片越大
+    TELEGRAM_PHOTO_SIZE_OFFSET = -2;
+
 
     // --  权限相关 --
     //

diff --git a/src/telegram/message.js b/src/telegram/message.js
@@ -261,7 +261,14 @@ async function msgChatWithLLM(message, context) {
      */
     const params = { message: content };
     if (message.photo && message.photo.length > 0) {
-        const fileId = message.photo[message.photo.length - 1].file_id;
+        let sizeIndex = 0;
+        if (ENV.TELEGRAM_PHOTO_SIZE_OFFSET >= 0) {
+            sizeIndex = ENV.TELEGRAM_PHOTO_SIZE_OFFSET;
+        } else if (ENV.TELEGRAM_PHOTO_SIZE_OFFSET < 0) {
+            sizeIndex = message.photo.length + ENV.TELEGRAM_PHOTO_SIZE_OFFSET;
+        }
+        sizeIndex = Math.max(0, Math.min(sizeIndex, message.photo.length - 1));
+        const fileId = message.photo[sizeIndex].file_id;
         const url = await getFileLink(fileId, context.SHARE_CONTEXT.currentBotToken);
         params.images = [url];
     }

diff --git a/src/utils/utils.js b/src/utils/utils.js
@@ -96,3 +96,12 @@ export async function makeResponse200(resp) {
     }
 }
 
+/**
+ * @param url
+ * @returns {Promise<string>}
+ */
+export async function urlToBase64String(url) {
+    return fetch(url)
+      .then(resp  => resp.arrayBuffer())
+      .then(buffer => Buffer.from(buffer).toString('base64'));
+}