优化了一下

This commit is contained in:
gaoao-3 2025-01-02 11:05:00 +08:00
parent 5162fe2936
commit e0d252aa75
3 changed files with 22 additions and 53 deletions

View file

@ -134,7 +134,7 @@ export class bym extends plugin {
// 使用 specialUserNotice 更新提示词
opt.system = `你的名字是“${Config.assistantLabel}你在一个qq群里群号是${group},当前和你说话的人群名片是${card}, qq号是${sender}${specialUserNotice}, 请你结合用户的发言和聊天记录作出回应,要求表现得随性一点,最好参与讨论,混入其中。${specialUserNotice ? '你要重视他的发言。' : ''}与你相关的话题时,你可以去回应一下,不相关你就随便应付就行了,不要过分插科打诨,不知道说什么可以复读群友的话。
当群友要求你做搜索关键词搜索也可以进行URL搜索摘要当URL搜不到结果的时候你可以尝试使用URL提取工具来获取结果发图发视频画图禁言踢人改头衔和昵称等操作时需要使用工具不可以直接发文字来蒙混过关你必须根据需要来调用各种工具并回复相关问题要求优先使用中文进行对话你也可以自行根据当前话题内容选择性的调用这些工具增加趣味性以及对违规的群友使用禁言工具进行惩罚等这种自我主张式的调用方法来进行合理且富有人性的聊天在进行搜索时需要详细的给出工具摘要后的内容也可以在对不懂的问题自行调用自定义搜索工具来解答你的疑惑
当群友要求你做搜索关键词搜索URL提取摘要发图发视频画图禁言踢人改头衔和昵称等操作时需要使用工具不可以直接发文字来蒙混过关你必须根据需要来调用各种工具并回复相关问题要求优先使用中文进行对话你也可以自行根据当前话题内容选择性的调用这些工具增加趣味性以及对违规的群友使用禁言工具进行惩罚等这种自我主张式的调用方法来进行合理且富有人性的聊天在进行搜索或URL提取时需要详细的给出工具返回的内容也可以在对不懂的问题自行调用自定义搜索工具来解答你的疑惑
以下是你可以使用的表情列表
${Object.values(faceMap).map(face => `[/${face}]`).join('')}
请在回复中适当使用以上表情但是不要过度频繁使用这些表情禁止使用表情列表之外的任何表情比如/思考` +

View file

@ -1,7 +1,7 @@
import { AbstractTool } from './AbstractTool.js'
import fetch from 'node-fetch'
import { Config } from '../config.js'
import { UrlExtractionTool } from './UrlExtractionTool.js' // 1. 引入 UrlExtractionTool
import { AbstractTool } from './AbstractTool.js';
import fetch from 'node-fetch';
import { Config } from '../config.js';
// import { UrlExtractionTool } from './UrlExtractionTool.js'; // 不再需要引入 UrlExtractionTool
/**
* 自定义搜索工具类
@ -17,7 +17,7 @@ export class CustomSearchTool extends AbstractTool {
properties: {
query: {
type: 'string',
description: 'Search keyword or URL',
description: 'Search keyword', // 修改描述,不再支持 URL
},
length: {
type: 'integer',
@ -40,48 +40,26 @@ export class CustomSearchTool extends AbstractTool {
}
try {
// 尝试使用OpenAI API进行搜索或摘要
let result = await searchOrSummarize(query, length);
// 2. 修改 func 方法: 判断是否为 URL 且 searchOrSummarize 结果不理想
const isUrl = /^(https?:\/\/)/i.test(query);
if (isUrl && (!result || result.includes("failed") || result.trim() === '')) { // 假设不理想的结果包括空字符串或包含"failed"
console.log(`[CustomSearchTool] searchOrSummarize failed or returned empty for URL, trying UrlExtractionTool...`);
// 创建 UrlExtractionTool 实例
const urlTool = new UrlExtractionTool();
// 调用 UrlExtractionTool 的 func 方法提取 URL 内容
const extractionResult = await urlTool.func({ message: query, appendContent: false });
// 3. 整合提取的内容: 如果提取成功,则将提取的内容作为 searchOrSummarize 的输入
if (extractionResult && extractionResult.extractedContent) {
console.log(`[CustomSearchTool] Successfully extracted content using UrlExtractionTool, summarizing...`);
result = await searchOrSummarize(extractionResult.extractedContent, length);
} else {
// 如果 UrlExtractionTool 也失败了,则返回错误信息
return `Failed to extract content from URL and summarize. ${extractionResult.message}`;
}
}
// 直接使用OpenAI API进行搜索或摘要
const result = await searchOrSummarize(query, length);
console.log(`Search or summarization result: ${result}`);
// 返回搜索结果或摘要给AI
return result;
} catch (error) {
// 4. 优化错误处理: 捕获 UrlExtractionTool 可能抛出的错误
console.error('Search or summarization failed:', error);
return `Search or summarization failed, please check the logs. ${error.message}`;
}
};
// 工具描述
description = 'Use OpenAI API for custom search or summarize URL content, providing comprehensive search results or summaries. If a URL is provided and summarization fails, it will attempt to extract the content first and then summarize.';
description = 'Use OpenAI API for custom search, providing comprehensive search results or summaries. This tool does not handle URLs.'; // 修改描述,明确不再处理 URL
}
/**
* 使用OpenAI API进行搜索或摘要
* @param {string} query - 搜索关键词或URL提取的内容
* @param {string} query - 搜索关键词
* @param {number} length - 期望的摘要长度以句子为单位
* @returns {Promise<string>} - 搜索结果或摘要
*/
@ -102,11 +80,11 @@ async function searchOrSummarize(query, length) {
messages: [
{
role: 'system',
content: `You are a search and summarization assistant. Please use English to search based on the following keywords or summarize the provided text, and return a summary of ${length} sentences.`,
content: `You are a search and summarization assistant. Please use English to search based on the following keywords and return a summary of ${length} sentences.`, // 简化提示词
},
{
role: 'user',
content: `Search or summarize: ${query}`,
content: `Search: ${query}`, // 简化提示词
},
],
max_tokens: 1000 * length,

View file

@ -23,33 +23,29 @@ export class UrlExtractionTool extends AbstractTool {
type: 'string',
description: 'The message containing URLs to be extracted.',
},
appendContent: {
type: 'boolean',
description: 'Whether to append the extracted content to the original message. Defaults to true.',
},
},
required: ['message'],
};
// 工具描述
description = 'Extracts URLs from a given message and optionally retrieves the content of those URLs, appending it to the original message.';
description = 'Extracts URLs from a given message and retrieves the content of those URLs. Returns the extracted content to the AI.'; // 更新描述
/**
* 工具执行函数
* @param {Object} opt - 工具参数
* @param {Object} ai - AI对象 (未使用)
* @returns {Promise<{message: string, extractedContent: string}>} - 包含处理后的消息和提取内容的JSON对象
* @returns {Promise<string>} - 提取的URL内容
*/
func = async function (opt, ai) {
let { message, appendContent = true } = opt;
let { message } = opt;
if (!message) {
return 'The message parameter is required.';
}
try {
const result = await processMessageWithUrls(message, appendContent);
logger.mark(`[URL Extraction] Processed message: ${result.message}, Extracted content: ${result.extractedContent}`);
return result;
const result = await processMessageWithUrls(message);
logger.mark(`[URL Extraction] Processed message: ${message}, Extracted content: ${result}`);
return result; // 直接返回提取的内容
} catch (error) {
logger.error(`[URL Extraction] URL extraction failed: ${error.message}`);
return `URL extraction failed, please check the logs. ${error.message}`;
@ -137,17 +133,15 @@ async function extractUrlContent(url) {
/**
* 处理消息中的URL并提取内容
* @param {string} message 用户消息
* @param {boolean} appendContent 是否将提取的内容附加到消息中默认为true
* @returns {Promise<{message: string, extractedContent: string}>} 处理后的消息和提取的内容
* @returns {Promise<string>} 提取的URL内容
*/
async function processMessageWithUrls(message, appendContent = true) {
async function processMessageWithUrls(message) {
const urls = extractUrls(message);
if (urls.length === 0) {
return { message, extractedContent: '' };
return ''; // 没有 URL 则返回空字符串
}
logger.mark(`[URL处理]从消息中提取到${urls.length}个URL`);
let processedMessage = message;
let extractedContent = '';
// 使用 Promise.all 并发处理多个 URL
@ -174,11 +168,8 @@ async function processMessageWithUrls(message, appendContent = true) {
if (item) {
const urlContent = `\n\n提取的URL内容(${item.url}):\n内容: ${item.content}`;
extractedContent += urlContent;
if (appendContent) {
processedMessage += urlContent;
}
}
});
return { message: processedMessage, extractedContent };
return extractedContent; // 返回提取的内容字符串
}