mirror of
https://github.com/Jerryplusy/crystelf-plugin.git
synced 2025-12-05 15:41:56 +00:00
feat:完善at匹配
This commit is contained in:
parent
92bfe8087c
commit
337c7e9f60
542
apps/ai.js
542
apps/ai.js
@ -22,7 +22,7 @@ export class crystelfAI extends plugin {
|
||||
rule: [
|
||||
{
|
||||
reg: `^${nickname}([\\s\\S]*)?$`,
|
||||
fnc: 'index',
|
||||
fnc: 'in',
|
||||
},
|
||||
],
|
||||
});
|
||||
@ -43,86 +43,143 @@ export class crystelfAI extends plugin {
|
||||
}
|
||||
}
|
||||
|
||||
async index(e) {
|
||||
try {
|
||||
//logger.info('111')
|
||||
const config = await ConfigControl.get();
|
||||
const aiConfig = config?.ai;
|
||||
if (!config?.config?.ai) {
|
||||
return;
|
||||
async in(e){
|
||||
return await index(e);
|
||||
}
|
||||
}
|
||||
|
||||
Bot.on("message.group",async(e)=>{
|
||||
let flag = false;
|
||||
if(e.message){
|
||||
e.message.forEach(message=>{
|
||||
if(message.type === 'at' && message.qq == e.bot.uin){
|
||||
flag = true;
|
||||
}
|
||||
if (aiConfig?.blockGroup?.includes(e.group_id)) {
|
||||
return;
|
||||
}
|
||||
if (aiConfig?.whiteGroup?.length > 0 && !aiConfig?.whiteGroup?.includes(e.group_id)) {
|
||||
return;
|
||||
}
|
||||
if (e.user_id === e.bot.uin) {
|
||||
return;
|
||||
}
|
||||
const userMessage = this.extractUserMessage(e.msg, nickname);
|
||||
if (!userMessage) {
|
||||
return;
|
||||
}
|
||||
logger.info(
|
||||
`[crystelf-ai] 收到消息: 群${e.group_id}, 用户${e.user_id}, 内容: ${userMessage}`
|
||||
);
|
||||
const result = await this.processMessage(userMessage, e, aiConfig);
|
||||
if (result && result.length > 0) {
|
||||
// TODO 优化流式输出
|
||||
await this.sendResponse(e, result);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理消息失败: ${error.message}`);
|
||||
const config = await ConfigControl.get();
|
||||
const aiConfig = config?.ai;
|
||||
return e.reply(segment.image(await Meme.getMeme(aiConfig.character, 'default')));
|
||||
})
|
||||
}
|
||||
if(!flag) return;
|
||||
return await index(e);
|
||||
})
|
||||
|
||||
async function index(e) {
|
||||
try {
|
||||
//logger.info('111')
|
||||
const config = await ConfigControl.get();
|
||||
const aiConfig = config?.ai;
|
||||
if (!config?.config?.ai) {
|
||||
return;
|
||||
}
|
||||
if (aiConfig?.blockGroup?.includes(e.group_id)) {
|
||||
return;
|
||||
}
|
||||
if (aiConfig?.whiteGroup?.length > 0 && !aiConfig?.whiteGroup?.includes(e.group_id)) {
|
||||
return;
|
||||
}
|
||||
if (e.user_id === e.bot.uin) {
|
||||
return;
|
||||
}
|
||||
const userMessage = extractUserMessage(e.msg, nickname,e);
|
||||
if (!userMessage) {
|
||||
return;
|
||||
}
|
||||
const result = await processMessage(userMessage, e, aiConfig);
|
||||
if (result && result.length > 0) {
|
||||
// TODO 优化流式输出
|
||||
await sendResponse(e, result);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理消息失败: ${error.message}`);
|
||||
const config = await ConfigControl.get();
|
||||
const aiConfig = config?.ai;
|
||||
return e.reply(segment.image(await Meme.getMeme(aiConfig.character, 'default')));
|
||||
}
|
||||
}
|
||||
|
||||
function extractUserMessage(msg, nickname,e) {
|
||||
if (!msg || !nickname) return '';
|
||||
const regex = new RegExp(`^${nickname}\\s*([\\s\\S]*)?$`);
|
||||
const match = msg.match(regex);
|
||||
if (match && match[1]) {
|
||||
return match[1].trim();
|
||||
} else {
|
||||
if(e.message){
|
||||
let text;
|
||||
e.message.forEach(message=>{
|
||||
if(message.type === 'text'){
|
||||
text = message.text;
|
||||
}
|
||||
})
|
||||
if(text) return text;
|
||||
}
|
||||
}
|
||||
logger.warn('[crystelf-ai] 字符串匹配失败,使用空字符串操作');
|
||||
return '';
|
||||
}
|
||||
|
||||
extractUserMessage(msg, nickname) {
|
||||
if (!msg || !nickname) return '';
|
||||
const regex = new RegExp(`^${nickname}\\s*([\\s\\S]*)?$`);
|
||||
const match = msg.match(regex);
|
||||
if (match && match[1]) {
|
||||
return match[1].trim();
|
||||
}
|
||||
logger.warn('[crystelf-ai] 字符串匹配失败,使用空字符串操作');
|
||||
return '';
|
||||
/**
|
||||
* 处理用户消息
|
||||
* @param userMessage
|
||||
* @param e
|
||||
* @param aiConfig
|
||||
* @returns {Promise<Array|null>}
|
||||
*/
|
||||
async function processMessage(userMessage, e, aiConfig) {
|
||||
const mode = aiConfig?.mode || 'mix';
|
||||
logger.info(`[crystelf-ai] 群${e.group_id} 用户${e.user_id}使用${mode}进行回复..`)
|
||||
switch (mode) {
|
||||
case 'keyword':
|
||||
return await handleKeywordMode(userMessage, e);
|
||||
case 'ai':
|
||||
return await handleAiMode(userMessage, e, aiConfig);
|
||||
case 'mix':
|
||||
return await handleMixMode(userMessage, e, aiConfig);
|
||||
default:
|
||||
logger.warn(`[crystelf-ai] 未知匹配模式: ${mode},将使用混合模式输出`);
|
||||
return await handleMixMode(userMessage, e, aiConfig);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理用户消息
|
||||
* @param userMessage
|
||||
* @param e
|
||||
* @param aiConfig
|
||||
* @returns {Promise<Array|null>}
|
||||
*/
|
||||
async processMessage(userMessage, e, aiConfig) {
|
||||
const mode = aiConfig?.mode || 'mix';
|
||||
logger.info(`[crystelf-ai] 群${e.group_id} 用户${e.user_id}使用${mode}进行回复..`)
|
||||
switch (mode) {
|
||||
case 'keyword':
|
||||
return await this.handleKeywordMode(userMessage, e);
|
||||
case 'ai':
|
||||
return await this.handleAiMode(userMessage, e, aiConfig);
|
||||
case 'mix':
|
||||
return await this.handleMixMode(userMessage, e, aiConfig);
|
||||
default:
|
||||
logger.warn(`[crystelf-ai] 未知匹配模式: ${mode},将使用混合模式输出`);
|
||||
return await this.handleMixMode(userMessage, e, aiConfig);
|
||||
}
|
||||
/**
|
||||
* 关键词模式
|
||||
* @param userMessage
|
||||
* @param e
|
||||
* @returns {Promise<[{type: string, data: string}]>}
|
||||
*/
|
||||
async function handleKeywordMode(userMessage, e) {
|
||||
const matchResult = await KeywordMatcher.matchKeywords(userMessage, 'ai');
|
||||
|
||||
if (matchResult && matchResult.matched) {
|
||||
return [
|
||||
{
|
||||
type: 'message',
|
||||
data: matchResult.text,
|
||||
at: false,
|
||||
quote: false,
|
||||
recall: 0,
|
||||
},
|
||||
];
|
||||
}
|
||||
logger.warn('[crystelf-ai] 关键词回复模式未查询到输出,将回复表情包');
|
||||
return [
|
||||
{
|
||||
type: 'meme',
|
||||
data: 'default',
|
||||
},
|
||||
];
|
||||
}
|
||||
|
||||
/**
|
||||
* 关键词模式
|
||||
* @param userMessage
|
||||
* @param e
|
||||
* @returns {Promise<[{type: string, data: string}]>}
|
||||
*/
|
||||
async handleKeywordMode(userMessage, e) {
|
||||
async function handleAiMode(userMessage, e, aiConfig) {
|
||||
return await callAiForResponse(userMessage, e, aiConfig);
|
||||
}
|
||||
|
||||
async function handleMixMode(userMessage, e, aiConfig) {
|
||||
const isTooLong = await KeywordMatcher.isMessageTooLong(userMessage);
|
||||
|
||||
if (isTooLong) {
|
||||
//消息太长,使用AI回复
|
||||
return await callAiForResponse(userMessage, e, aiConfig);
|
||||
} else {
|
||||
const matchResult = await KeywordMatcher.matchKeywords(userMessage, 'ai');
|
||||
|
||||
if (matchResult && matchResult.matched) {
|
||||
return [
|
||||
{
|
||||
@ -133,85 +190,30 @@ export class crystelfAI extends plugin {
|
||||
recall: 0,
|
||||
},
|
||||
];
|
||||
}
|
||||
logger.warn('[crystelf-ai] 关键词回复模式未查询到输出,将回复表情包');
|
||||
return [
|
||||
{
|
||||
type: 'meme',
|
||||
data: 'default',
|
||||
},
|
||||
];
|
||||
}
|
||||
|
||||
async handleAiMode(userMessage, e, aiConfig) {
|
||||
return await this.callAiForResponse(userMessage, e, aiConfig);
|
||||
}
|
||||
|
||||
async handleMixMode(userMessage, e, aiConfig) {
|
||||
const isTooLong = await KeywordMatcher.isMessageTooLong(userMessage);
|
||||
|
||||
if (isTooLong) {
|
||||
//消息太长,使用AI回复
|
||||
return await this.callAiForResponse(userMessage, e, aiConfig);
|
||||
} else {
|
||||
const matchResult = await KeywordMatcher.matchKeywords(userMessage, 'ai');
|
||||
if (matchResult && matchResult.matched) {
|
||||
return [
|
||||
{
|
||||
type: 'message',
|
||||
data: matchResult.text,
|
||||
at: false,
|
||||
quote: false,
|
||||
recall: 0,
|
||||
},
|
||||
];
|
||||
} else {
|
||||
//关键词匹配失败,使用AI回复
|
||||
return await this.callAiForResponse(userMessage, e, aiConfig);
|
||||
}
|
||||
//关键词匹配失败,使用AI回复
|
||||
return await callAiForResponse(userMessage, e, aiConfig);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async callAiForResponse(userMessage, e, aiConfig) {
|
||||
try {
|
||||
//创建session
|
||||
const session = SessionManager.createOrGetSession(e.group_id, e.user_id);
|
||||
if (!session) {
|
||||
logger.info(
|
||||
`[crystelf-ai] 群${e.group_id} , 用户${e.user_id}无法创建session,请检查是否聊天频繁`
|
||||
);
|
||||
return null;
|
||||
}
|
||||
//搜索相关记忆
|
||||
const memories = await MemorySystem.searchMemories([userMessage], 5);
|
||||
//构建聊天历史
|
||||
const chatHistory = session.chatHistory.slice(-10);
|
||||
const aiResult = await AiCaller.callAi(userMessage, chatHistory, memories);
|
||||
if (!aiResult.success) {
|
||||
logger.error(`[crystelf-ai] AI调用失败: ${aiResult.error}`);
|
||||
return [
|
||||
{
|
||||
type: 'meme',
|
||||
data: 'default',
|
||||
},
|
||||
];
|
||||
}
|
||||
//处理响应
|
||||
const processedResponse = await ResponseHandler.processResponse(
|
||||
aiResult.response,
|
||||
userMessage,
|
||||
e.group_id
|
||||
async function callAiForResponse(userMessage, e, aiConfig) {
|
||||
try {
|
||||
//创建session
|
||||
const session = SessionManager.createOrGetSession(e.group_id, e.user_id);
|
||||
if (!session) {
|
||||
logger.info(
|
||||
`[crystelf-ai] 群${e.group_id} , 用户${e.user_id}无法创建session,请检查是否聊天频繁`
|
||||
);
|
||||
//更新session
|
||||
const newChatHistory = [
|
||||
...chatHistory,
|
||||
{ role: 'user', content: userMessage },
|
||||
{ role: 'assistant', content: aiResult.response },
|
||||
];
|
||||
SessionManager.updateChatHistory(e.group_id, newChatHistory);
|
||||
return processedResponse;
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] AI调用失败: ${error.message}`);
|
||||
return null;
|
||||
}
|
||||
//搜索相关记忆
|
||||
const memories = await MemorySystem.searchMemories([userMessage], 5);
|
||||
//构建聊天历史
|
||||
const chatHistory = session.chatHistory.slice(-10);
|
||||
const aiResult = await AiCaller.callAi(userMessage, chatHistory, memories);
|
||||
if (!aiResult.success) {
|
||||
logger.error(`[crystelf-ai] AI调用失败: ${aiResult.error}`);
|
||||
return [
|
||||
{
|
||||
type: 'meme',
|
||||
@ -219,150 +221,172 @@ export class crystelfAI extends plugin {
|
||||
},
|
||||
];
|
||||
}
|
||||
//处理响应
|
||||
const processedResponse = await ResponseHandler.processResponse(
|
||||
aiResult.response,
|
||||
userMessage,
|
||||
e.group_id
|
||||
);
|
||||
//更新session
|
||||
const newChatHistory = [
|
||||
...chatHistory,
|
||||
{ role: 'user', content: userMessage },
|
||||
{ role: 'assistant', content: aiResult.response },
|
||||
];
|
||||
SessionManager.updateChatHistory(e.group_id, newChatHistory);
|
||||
return processedResponse;
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] AI调用失败: ${error.message}`);
|
||||
return [
|
||||
{
|
||||
type: 'meme',
|
||||
data: 'default',
|
||||
},
|
||||
];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送消息
|
||||
* @param e
|
||||
* @param messages 消息数组
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async sendResponse(e, messages) {
|
||||
try {
|
||||
for (const message of messages) {
|
||||
switch (message.type) {
|
||||
case 'message':
|
||||
if (message.recall > 0) {
|
||||
await e.reply(message.data, message.quote, {
|
||||
recallMsg: message.recall,
|
||||
at: message.at,
|
||||
});
|
||||
} else {
|
||||
await e.reply(message.data, message.quote, {
|
||||
at: message.at,
|
||||
});
|
||||
}
|
||||
break;
|
||||
/**
|
||||
* 发送消息
|
||||
* @param e
|
||||
* @param messages 消息数组
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async function sendResponse(e, messages) {
|
||||
try {
|
||||
for (const message of messages) {
|
||||
switch (message.type) {
|
||||
case 'message':
|
||||
if (message.recall > 0) {
|
||||
await e.reply(message.data, message.quote, {
|
||||
recallMsg: message.recall,
|
||||
at: message.at,
|
||||
});
|
||||
} else {
|
||||
await e.reply(message.data, message.quote, {
|
||||
at: message.at,
|
||||
});
|
||||
}
|
||||
break;
|
||||
|
||||
case 'code':
|
||||
await this.handleCodeMessage(e, message);
|
||||
break;
|
||||
case 'code':
|
||||
await handleCodeMessage(e, message);
|
||||
break;
|
||||
|
||||
case 'markdown':
|
||||
await this.handleMarkdownMessage(e, message);
|
||||
break;
|
||||
case 'markdown':
|
||||
await handleMarkdownMessage(e, message);
|
||||
break;
|
||||
|
||||
case 'meme':
|
||||
await this.handleMemeMessage(e, message);
|
||||
break;
|
||||
case 'meme':
|
||||
await handleMemeMessage(e, message);
|
||||
break;
|
||||
|
||||
case 'at':
|
||||
await e.reply(segment.at(message.id));
|
||||
break;
|
||||
case 'at':
|
||||
await e.reply(segment.at(message.id));
|
||||
break;
|
||||
|
||||
case 'poke':
|
||||
await this.handlePokeMessage(e, message);
|
||||
break;
|
||||
case 'poke':
|
||||
await handlePokeMessage(e, message);
|
||||
break;
|
||||
|
||||
case 'like':
|
||||
await this.handleLikeMessage(e, message);
|
||||
break;
|
||||
case 'like':
|
||||
await handleLikeMessage(e, message);
|
||||
break;
|
||||
|
||||
case 'recall':
|
||||
await this.handleRecallMessage(e, message);
|
||||
break;
|
||||
case 'recall':
|
||||
await handleRecallMessage(e, message);
|
||||
break;
|
||||
|
||||
default:
|
||||
logger.warn(`[crystelf-ai] 不支持的消息类型: ${message.type}`);
|
||||
}
|
||||
default:
|
||||
logger.warn(`[crystelf-ai] 不支持的消息类型: ${message.type}`);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 发送回复失败: ${error.message}`);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 发送回复失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
async handleCodeMessage(e, message) {
|
||||
try {
|
||||
//渲染代码为图片
|
||||
const imagePath = await Renderer.renderCode(message.data, message.language || 'text');
|
||||
if (imagePath) {
|
||||
await e.reply(segment.image(imagePath));
|
||||
} else {
|
||||
// 渲染失败 TODO 构造转发消息发送,避免刷屏
|
||||
await e.reply(segment.code(message.data));
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理代码消息失败: ${error.message}`);
|
||||
async function handleCodeMessage(e, message) {
|
||||
try {
|
||||
//渲染代码为图片
|
||||
const imagePath = await Renderer.renderCode(message.data, message.language || 'text');
|
||||
if (imagePath) {
|
||||
await e.reply(segment.image(imagePath));
|
||||
} else {
|
||||
// 渲染失败 TODO 构造转发消息发送,避免刷屏
|
||||
await e.reply(segment.code(message.data));
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理代码消息失败: ${error.message}`);
|
||||
await e.reply(segment.code(message.data));
|
||||
}
|
||||
}
|
||||
|
||||
async handleMarkdownMessage(e, message) {
|
||||
try {
|
||||
//渲染Markdown为图片
|
||||
const imagePath = await Renderer.renderMarkdown(message.data);
|
||||
if (imagePath) {
|
||||
await e.reply(segment.image(imagePath));
|
||||
} else {
|
||||
//渲染失败 TODO 构造转发消息发送,避免刷屏
|
||||
await e.reply(message.data);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理Markdown消息失败: ${error.message}`);
|
||||
async function handleMarkdownMessage(e, message) {
|
||||
try {
|
||||
//渲染Markdown为图片
|
||||
const imagePath = await Renderer.renderMarkdown(message.data);
|
||||
if (imagePath) {
|
||||
await e.reply(segment.image(imagePath));
|
||||
} else {
|
||||
//渲染失败 TODO 构造转发消息发送,避免刷屏
|
||||
await e.reply(message.data);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理Markdown消息失败: ${error.message}`);
|
||||
await e.reply(message.data);
|
||||
}
|
||||
}
|
||||
|
||||
async handleMemeMessage(e, message) {
|
||||
try {
|
||||
const config = await ConfigControl.get('ai');
|
||||
const memeConfig = config?.memeConfig || {};
|
||||
const availableEmotions = memeConfig.availableEmotions || [
|
||||
'happy',
|
||||
'sad',
|
||||
'angry',
|
||||
'confused',
|
||||
];
|
||||
//情绪是否有效
|
||||
const emotion = availableEmotions.includes(message.data) ? message.data : 'default';
|
||||
const character = memeConfig.character || 'default';
|
||||
const memeUrl = await Meme.getMeme(character, emotion);
|
||||
await e.reply(segment.image(memeUrl));
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理表情消息失败: ${error.message}`);
|
||||
e.reply(segment.image(await Meme.getMeme(aiConfig.character, 'default')));
|
||||
}
|
||||
async function handleMemeMessage(e, message) {
|
||||
try {
|
||||
const config = await ConfigControl.get('ai');
|
||||
const memeConfig = config?.memeConfig || {};
|
||||
const availableEmotions = memeConfig.availableEmotions || [
|
||||
'happy',
|
||||
'sad',
|
||||
'angry',
|
||||
'confused',
|
||||
];
|
||||
//情绪是否有效
|
||||
const emotion = availableEmotions.includes(message.data) ? message.data : 'default';
|
||||
const character = memeConfig.character || 'default';
|
||||
const memeUrl = await Meme.getMeme(character, emotion);
|
||||
await e.reply(segment.image(memeUrl));
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 处理表情消息失败: ${error.message}`);
|
||||
e.reply(segment.image(await Meme.getMeme(aiConfig.character, 'default')));
|
||||
}
|
||||
}
|
||||
|
||||
async handlePokeMessage(e, message) {
|
||||
try {
|
||||
await Group.groupPoke(e, message.id, e.group_id);
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 戳一戳失败: ${error.message}`);
|
||||
}
|
||||
async function handlePokeMessage(e, message) {
|
||||
try {
|
||||
await Group.groupPoke(e, message.id, e.group_id);
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 戳一戳失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
async handleLikeMessage(e, message) {
|
||||
try {
|
||||
// TODO 点赞逻辑
|
||||
const adapter = await YunzaiUtils.getAdapter(e);
|
||||
const messageId = e.message_id || e.source?.id;
|
||||
async function handleLikeMessage(e, message) {
|
||||
try {
|
||||
// TODO 点赞逻辑
|
||||
const adapter = await YunzaiUtils.getAdapter(e);
|
||||
const messageId = e.message_id || e.source?.id;
|
||||
|
||||
if (messageId) {
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 点赞失败: ${error.message}`);
|
||||
if (messageId) {
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 点赞失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
async handleRecallMessage(e, message) {
|
||||
try {
|
||||
if (message.seq) {
|
||||
await Message.deleteMsg(e, message.seq);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 撤回消息失败: ${error.message}`);
|
||||
async function handleRecallMessage(e, message) {
|
||||
try {
|
||||
if (message.seq) {
|
||||
await Message.deleteMsg(e, message.seq);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 撤回消息失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@ -7,7 +7,6 @@ class OllamaChat {
|
||||
}
|
||||
|
||||
/**
|
||||
*
|
||||
* @param apiKey 密钥
|
||||
* @param baseUrl ollamaAPI地址
|
||||
*/
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user