Jerry/crystelf-plugin
1
0
Fork 0
mirror of https://github.com/Jerryplusy/crystelf-plugin.git synced 2025-12-05 15:41:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:完善at匹配

Browse Source
This commit is contained in:
Jerry 2025-10-18 00:49:20 +08:00
parent 92bfe8087c
commit 337c7e9f60
2 changed files with 283 additions and 260 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

542
apps/ai.js
View File

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

1
modules/ollama/ollamaChat.js
View File

@ -7,7 +7,6 @@ class OllamaChat {
} }
/** /**
*
* @param apiKey 密钥 * @param apiKey 密钥
* @param baseUrl ollamaAPI地址 * @param baseUrl ollamaAPI地址
*/ */