搞个了发邮件的tool,我先玩玩

This commit is contained in:
gaoao-3 2025-01-02 14:14:28 +08:00
parent e0d252aa75
commit b9b758102b
4 changed files with 143 additions and 2 deletions

View file

@ -200,6 +200,8 @@ const defaultConfig = {
enableBYM: false,
// 触发对话的概率百分比。比如5%就写5
bymRate: 5,
// smtp
smtpAuthCode: '',
// 伪人模式触发骂人反击的关键词
bymFuckList: [
'骂',

View file

@ -0,0 +1,128 @@
import { AbstractTool } from './AbstractTool.js';
import fetch from 'node-fetch';
import { Config } from '../config.js';
/**
* 邮件发送工具类
* @class MailSenderTool
* @extends {AbstractTool}
*/
export class MailSenderTool extends AbstractTool {
// 工具名称
name = 'MailSenderTool';
// 工具参数
parameters = {
properties: {
aiTask: {
type: 'string',
description: '描述 AI 需要发送邮件的任务,例如:给用户发送一封问候邮件',
},
targetUserQQ: {
type: 'string',
description: '目标用户的 QQ 号码,用于拼接成 @qq.com 的邮箱地址',
},
to: {
type: 'string',
description: '收件人邮箱地址,如果提供了 targetUserQQ则此参数可省略',
},
},
required: ['aiTask'],
};
/**
* 工具执行函数
* @param {Object} opt - 工具参数
* @param {Object} ai - AI对象
* @returns {Promise<string>} - 邮件发送结果
*/
func = async function (opt, ai) {
const { aiTask, targetUserQQ, to } = opt;
// 检查必填参数
if (!aiTask) {
return '缺少必要的参数 aiTask。';
}
// 从 Config 中读取 SMTP 授权码
const smtpAuthCode = Config.smtpAuthCode;
// 检查配置信息
if (!smtpAuthCode) {
return 'Config 配置中缺少必要的参数 smtpAuthCode。';
}
// 确定收件人邮箱地址
const recipientEmail = targetUserQQ ? `${targetUserQQ}@qq.com` : to;
if (!recipientEmail) {
return '缺少收件人邮箱地址,请提供 targetUserQQ 或 to 参数。';
}
// 使用 AI 生成邮件所有内容,并修改提示词为“你”的视角
let from, aiName, title, text, aiQQNumber;
try {
const aiTaskPrompt = `你的任务是:根据“${aiTask}”这个主题,以“你”的视角,给邮箱为“${recipientEmail}”的用户写一封邮件。你需要自己决定邮件的标题、正文内容,以及你在邮件中使用的名字。同时,你需要提供你的 QQ 号码,用于构建发件人邮箱地址。请返回一个 JSON 对象包含以下字段title邮件标题、text邮件正文、senderName你在邮件中使用的名字、aiQQNumber你的 QQ 号码)。`;
const aiResponse = await ai.toolTask(aiTaskPrompt);
// 期望的 aiResponse 格式:
// {
// title: '邮件标题',
// text: '邮件正文',
// senderName: '你在邮件中使用的名字',
// aiQQNumber: '你的 QQ 号码'
// }
title = aiResponse.title;
text = aiResponse.text;
aiName = aiResponse.senderName;
aiQQNumber = aiResponse.aiQQNumber;
// 构建发件人邮箱
from = `${aiQQNumber}@qq.com`;
} catch (error) {
console.error('AI 生成邮件内容失败:', error);
return `AI 生成邮件内容失败: ${error.message}`;
}
// 构建请求 URL
const apiUrl = 'http://wswzh.ccccocccc.cc/api/mail/zdy.php';
const queryParams = new URLSearchParams({
api_key: 'free',
from: from,
code: smtpAuthCode,
host: 'smtp.qq.com',
to: recipientEmail,
title: title,
text: text,
sb: aiName, // 这里仍然使用 sb 参数,但其值是 AI 决定的 senderName
});
const fullApiUrl = `${apiUrl}?${queryParams.toString()}`;
try {
// 发送邮件请求
const response = await fetch(fullApiUrl);
// 检查响应状态
if (response.ok) {
// 解析响应数据
const data = await response.json();
// 根据 API 返回的数据判断邮件是否发送成功
if (data.success) {
console.log('邮件发送成功:', data.message);
return '邮件发送成功';
} else {
console.error('邮件发送失败:', data.error);
return `邮件发送失败: ${data.error}`;
}
} else {
console.error('邮件发送请求失败:', response.status, response.statusText);
return `邮件发送请求失败: ${response.status} ${response.statusText}`;
}
} catch (error) {
console.error('邮件发送出错:', error);
return `邮件发送出错: ${error.message}`;
}
};
// 工具描述
description = '用于发送自定义邮件的工具,由 AI 以“你”的视角完全自主地决定邮件标题、内容、发件人名字和 QQ 号码,支持配置目标用户的 QQ 邮箱。';
}