From 72c6a2e008a04f9d39351d206d101b540b3b3261 Mon Sep 17 00:00:00 2001
From: gaoao-3 <140887777+gaoao-3@users.noreply.github.com>
Date: Thu, 2 Jan 2025 19:02:25 +0800
Subject: [PATCH] =?UTF-8?q?=E2=80=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/tools/CustomSearchTool.js | 50 +++++++++++++++++----------------
 1 file changed, 26 insertions(+), 24 deletions(-)

diff --git a/utils/tools/CustomSearchTool.js b/utils/tools/CustomSearchTool.js
index d614e4e..495a165 100644
--- a/utils/tools/CustomSearchTool.js
+++ b/utils/tools/CustomSearchTool.js
@@ -20,34 +20,40 @@ export class CustomSearchTool extends AbstractTool {
         type: 'integer',
         description: '期望的摘要长度（句子数），默认为3',
       },
-      imageBase64: {
-        type: 'string',
-        description: '可选的图片base64数据',
+      sender: {
+        type: 'object',
+        description: '发送者信息',
       }
     },
-    required: ['query'],
+    required: ['query', 'sender'],
   };
 
-  description = '使用 Gemini API 进行智能搜索，根据输入的内容或关键词提供全面的搜索结果和摘要。支持自定义摘要长度和图片识别。';
+  description = '使用 Gemini API 进行智能搜索，根据输入的内容或关键词提供全面的搜索结果和摘要。支持自定义摘要长度。';
 
   /**
    * 工具执行函数
    * @param {Object} opt - 工具参数
    * @param {string} opt.query - 搜索内容或关键词
    * @param {number} [opt.length=3] - 摘要长度
-   * @param {string|string[]} [opt.imageBase64] - 图片base64数据
+   * @param {Object} opt.sender - 发送者信息
    * @returns {Promise<Object>} - 包含答案和来源的对象
    */
   func = async function (opt) {
-    const { query, length = 3, imageBase64 } = opt;
+    const { query, length = 3, sender } = opt;
 
     if (!query?.trim()) {
       throw new Error('搜索内容或关键词不能为空');
     }
 
     try {
-      const result = await this.searchWithGemini(query, length, imageBase64);
+      const result = await this.searchWithGemini(query, length);
       console.debug(`[CustomSearchTool] 搜索结果:`, result);
+      // 添加发送者信息
+      result.senderInfo = {
+        title: `${sender.card || sender.nickname || sender.user_id}的搜索结果`,
+        sender
+      };
+      
       return result;
     } catch (error) {
       console.error('[CustomSearchTool] 搜索失败:', error);
@@ -59,11 +65,10 @@ export class CustomSearchTool extends AbstractTool {
    * 使用 Gemini API 进行搜索
    * @param {string} query - 搜索内容或关键词
    * @param {number} length - 摘要长度
-   * @param {string|string[]} [imageBase64] - 图片base64数据
    * @returns {Promise<Object>} - 包含答案和来源的对象
    * @private
    */
-  async searchWithGemini(query, length, imageBase64) {
+  async searchWithGemini(query, length) {
     const apiKey = Config.geminiKey;
     const apiBaseUrl = Config.geminiBaseUrl;
     const apiUrl = `${apiBaseUrl}/v1beta/models/gemini-2.0-flash-exp:generateContent?key=${apiKey}`;
@@ -89,19 +94,6 @@ export class CustomSearchTool extends AbstractTool {
       }]
     };
 
-    // 处理图片数据
-    if (imageBase64) {
-      const imageArray = Array.isArray(imageBase64) ? imageBase64 : [imageBase64];
-      imageArray.forEach(image => {
-        requestBody.contents[0].parts.push({
-          "inline_data": {
-            "mime_type": "image/jpeg",
-            "data": image
-          }
-        });
-      });
-    }
-
     try {
       const response = await fetch(apiUrl, {
         method: 'POST',
@@ -180,9 +172,19 @@ export class CustomSearchTool extends AbstractTool {
 
     console.debug('[CustomSearchTool] 处理后的来源信息:', sources);
 
+    // 构建转发消息数组
+    const forwardMsg = [answer];
+    if (sources && sources.length > 0) {
+      forwardMsg.push('\n信息来源：');
+      sources.forEach((source, index) => {
+        forwardMsg.push(`${index + 1}. ${source.title}\n${source.url}`);
+      });
+    }
+
     return {
       answer,
-      sources
+      sources,
+      forwardMsg
     };
   }
 }
\ No newline at end of file