mirror of
https://github.com/Jerryplusy/crystelf-plugin.git
synced 2025-12-05 15:41:56 +00:00
feat:优化提示词
This commit is contained in:
parent
356233263c
commit
1c3ce32678
17
apps/ai.js
17
apps/ai.js
@ -102,7 +102,9 @@ function extractUserMessage(msg, nickname,e) {
|
||||
const regex = new RegExp(`^${nickname}\\s*([\\s\\S]*)?$`);
|
||||
const match = msg.match(regex);
|
||||
if (match && match[1]) {
|
||||
return match[1].trim();
|
||||
let message = match[1].trim();
|
||||
message = `[${e.sender?.nickname},id:${e.user_id}]说:${message}`;
|
||||
return message;
|
||||
} else {
|
||||
if(e.message){
|
||||
let text;
|
||||
@ -111,7 +113,7 @@ function extractUserMessage(msg, nickname,e) {
|
||||
text = message.text;
|
||||
}
|
||||
})
|
||||
if(text) return text;
|
||||
if(text) return `[${e.sender?.nickname},id:${e.user_id}]说:${text}`;
|
||||
}
|
||||
}
|
||||
logger.warn('[crystelf-ai] 字符串匹配失败,使用空字符串操作');
|
||||
@ -210,10 +212,11 @@ async function callAiForResponse(userMessage, e, aiConfig) {
|
||||
return null;
|
||||
}
|
||||
//搜索相关记忆
|
||||
const memories = await MemorySystem.searchMemories([userMessage], 5);
|
||||
const memories = await MemorySystem.searchMemories(e.user_id,[userMessage], 5);
|
||||
//构建聊天历史
|
||||
const chatHistory = session.chatHistory.slice(-10);
|
||||
const aiResult = await AiCaller.callAi(userMessage, chatHistory, memories);
|
||||
const historyLen = aiConfig.chatHistory;
|
||||
const chatHistory = session.chatHistory.slice(-historyLen|-10);
|
||||
const aiResult = await AiCaller.callAi(userMessage, chatHistory, memories,e);
|
||||
if (!aiResult.success) {
|
||||
logger.error(`[crystelf-ai] AI调用失败: ${aiResult.error}`);
|
||||
return [
|
||||
@ -227,8 +230,10 @@ async function callAiForResponse(userMessage, e, aiConfig) {
|
||||
const processedResponse = await ResponseHandler.processResponse(
|
||||
aiResult.response,
|
||||
userMessage,
|
||||
e.group_id
|
||||
e.group_id,
|
||||
e.user_id
|
||||
);
|
||||
|
||||
//更新session
|
||||
const newChatHistory = [
|
||||
...chatHistory,
|
||||
|
||||
@ -28,6 +28,8 @@
|
||||
"timeout": 30,
|
||||
"?maxSessions": "最大同时存在的sessions群聊数量",
|
||||
"maxSessions": 10,
|
||||
"?chatHistory": "聊天上下文最大长度",
|
||||
"chatHistory": 10,
|
||||
"?keywordCache": "是否缓存关键词到本地",
|
||||
"keywordCache": true,
|
||||
"?pinyinMatch": "是否启用拼音匹配",
|
||||
|
||||
@ -43,9 +43,10 @@ class AiCaller {
|
||||
* @param prompt 用户输入
|
||||
* @param chatHistory 聊天历史
|
||||
* @param memories 记忆
|
||||
* @param e
|
||||
* @returns {Promise<{success: boolean, response: (*|string), rawResponse: (*|string)}|{success: boolean, error: string}|{success: boolean, error}>}
|
||||
*/
|
||||
async callAi(prompt, chatHistory = [], memories = []) {
|
||||
async callAi(prompt, chatHistory = [], memories = [],e) {
|
||||
if (!this.isInitialized || !this.config) {
|
||||
logger.error('[crystelf-ai] 未初始化或配置无效');
|
||||
return { success: false, error: 'AI调用器未初始化' };
|
||||
@ -59,7 +60,7 @@ class AiCaller {
|
||||
chatHistory: chatHistory,
|
||||
model: this.config.modelType,
|
||||
temperature: this.config.temperature,
|
||||
customPrompt: await this.getSystemPrompt(),
|
||||
customPrompt: await this.getSystemPrompt(e),
|
||||
});
|
||||
|
||||
if (result.success) {
|
||||
@ -89,9 +90,10 @@ class AiCaller {
|
||||
* @param chatHistory 聊天记录
|
||||
* @param memories 记忆
|
||||
* @param onChunk 流式数据回调函数
|
||||
* @param e
|
||||
* @returns {Promise<Object|{success: boolean, error: string}|{success: boolean, error}>}
|
||||
*/
|
||||
async callAiStream(prompt, chatHistory = [], memories = [], onChunk = null) {
|
||||
async callAiStream(prompt, chatHistory = [], memories = [], onChunk = null,e) {
|
||||
if (!this.isInitialized || !this.config) {
|
||||
logger.error('[crystelf-ai] 未初始化或配置无效');
|
||||
return { success: false, error: 'AI调用器未初始化' };
|
||||
@ -99,7 +101,7 @@ class AiCaller {
|
||||
|
||||
if (!this.config.stream) {
|
||||
logger.warn('[crystelf-ai] 流式输出未启用,使用普通调用');
|
||||
return await this.callAi(prompt, chatHistory, memories);
|
||||
return await this.callAi(prompt, chatHistory, memories,e);
|
||||
}
|
||||
|
||||
try {
|
||||
@ -137,26 +139,59 @@ class AiCaller {
|
||||
*/
|
||||
buildPrompt(prompt, memories = []) {
|
||||
let fullPrompt = '';
|
||||
// TODO 加入标准信息
|
||||
if (memories && memories.length > 0) {
|
||||
fullPrompt += '相关记忆：\n';
|
||||
fullPrompt += '你可能会用到的记忆,请按情况使用,如果不合语境请忽略:\n';
|
||||
memories.forEach((memory, index) => {
|
||||
fullPrompt += `${index + 1}. ${memory.data}\n`;
|
||||
fullPrompt += `${index + 1}. 关键词:${memory.keywords},内容:${memory.data}\n`;
|
||||
});
|
||||
fullPrompt += '\n';
|
||||
}
|
||||
fullPrompt += `用户说: ${prompt}\n`;
|
||||
fullPrompt += '请根据以上信息进行回复：\n';
|
||||
fullPrompt += `以下是用户说的内容,会以[用户昵称,用户qq号]的形式给你,但是请注意,你回复message块的时候不需要带[]以及里面的内容,正常回复你想说的话即可:\n${prompt}\n`;
|
||||
return fullPrompt;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取系统提示词
|
||||
* @param {object} e 上下文事件对象
|
||||
* @returns {Promise<string>} 系统提示词
|
||||
*/
|
||||
async getSystemPrompt() {
|
||||
return this.config?.stream ? await getStreamSystemPrompt() : await getSystemPrompt();
|
||||
async getSystemPrompt(e) {
|
||||
try {
|
||||
const basePrompt = this.config?.stream
|
||||
? await getStreamSystemPrompt()
|
||||
: await getSystemPrompt();
|
||||
const config = await ConfigControl.get();
|
||||
const botInfo = {
|
||||
id: e.bot?.uin || '未知',
|
||||
name: config?.profile?.nickName || '晶灵'
|
||||
};
|
||||
|
||||
const userInfo = {
|
||||
id: e.user_id || e.sender?.user_id || '未知',
|
||||
name: e.sender?.card || e.sender?.nickname || '用户',
|
||||
isMaster: e.isMaster,
|
||||
};
|
||||
const contextIntro = [
|
||||
`以下是当前对话的上下文信息（仅供你理解对话背景，请勿泄露）：`,
|
||||
`[你的信息]`,
|
||||
`- 你的昵称：${botInfo.name}`,
|
||||
`- 你的qq号：${botInfo.id}`,
|
||||
``,
|
||||
`[跟你对话的用户的信息]`,
|
||||
`- 他的名字：${userInfo.name}`,
|
||||
`- 他的qq号(id)：${userInfo.id}`,
|
||||
`- 他${userInfo.isMaster ? '是':'不是'}你的主人`,
|
||||
``,
|
||||
`请基于以上上下文进行理解,这些信息是当你需要的时候使用的,绝对不能泄露这些信息,也不能主动提起`,
|
||||
``,
|
||||
].join('\n');
|
||||
return `${contextIntro}${basePrompt}`;
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 生成系统提示词失败: ${error.message}`);
|
||||
return await getSystemPrompt();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 将回复分割成多个块用于流式输出
|
||||
|
||||
@ -4,7 +4,7 @@ import path from 'path';
|
||||
|
||||
class MemorySystem {
|
||||
constructor() {
|
||||
this.memoryFile = path.join(process.cwd(), 'data', 'crystelf', 'ai_memory.json');
|
||||
this.baseDir = path.join(process.cwd(), 'data', 'crystelf', 'memories');
|
||||
this.memories = new Map(); // 内存中的记忆存储
|
||||
this.defaultTimeout = 30; // 默认超时时间(天)
|
||||
}
|
||||
@ -13,40 +13,60 @@ class MemorySystem {
|
||||
try {
|
||||
const config = await ConfigControl.get('ai');
|
||||
this.defaultTimeout = config?.timeout || 30;
|
||||
await this.loadMemories();
|
||||
await this.loadAllMemories();
|
||||
await this.cleanExpiredMemories();
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 记忆系统初始化失败: ${error.message}`);
|
||||
}
|
||||
} // TODO 群聊id/用户id分组保存
|
||||
}
|
||||
|
||||
async loadMemories() {
|
||||
async loadAllMemories() {
|
||||
try {
|
||||
if (fs.existsSync(this.memoryFile)) {
|
||||
const data = fs.readFileSync(this.memoryFile, 'utf8');
|
||||
const memoriesData = JSON.parse(data);
|
||||
if (!fs.existsSync(this.baseDir)) {
|
||||
fs.mkdirSync(this.baseDir, { recursive: true });
|
||||
}
|
||||
|
||||
const groupDirs = fs.readdirSync(this.baseDir);
|
||||
for (const groupId of groupDirs) {
|
||||
const groupPath = path.join(this.baseDir, groupId);
|
||||
if (!fs.statSync(groupPath).isDirectory()) continue;
|
||||
|
||||
const userFiles = fs.readdirSync(groupPath);
|
||||
for (const file of userFiles) {
|
||||
if (!file.endsWith('.json')) continue;
|
||||
const userId = path.basename(file, '.json');
|
||||
const filePath = path.join(groupPath, file);
|
||||
const data = fs.readFileSync(filePath, 'utf8');
|
||||
const memoriesData = JSON.parse(data || '{}');
|
||||
for (const [key, memory] of Object.entries(memoriesData)) {
|
||||
this.memories.set(key, memory);
|
||||
this.memories.set(`${groupId}_${userId}_${key}`, memory);
|
||||
}
|
||||
}
|
||||
}
|
||||
logger.info(`[crystelf-ai] 加载了 ${this.memories.size} 条记忆`);
|
||||
} else {
|
||||
const memoryDir = path.dirname(this.memoryFile);
|
||||
if (!fs.existsSync(memoryDir)) {
|
||||
fs.mkdirSync(memoryDir, { recursive: true });
|
||||
}
|
||||
fs.writeFileSync(this.memoryFile, '{}');
|
||||
logger.info('[crystelf-ai] 创建新的记忆文件');
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 加载记忆失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
async saveMemories() {
|
||||
async saveMemories(groupId, userId) {
|
||||
try {
|
||||
const memoriesData = Object.fromEntries(this.memories);
|
||||
fs.writeFileSync(this.memoryFile, JSON.stringify(memoriesData, null, 2));
|
||||
logger.info('[crystelf-ai] 记忆已保存到文件');
|
||||
const groupPath = path.join(this.baseDir, groupId);
|
||||
const filePath = path.join(groupPath, `${userId}.json`);
|
||||
if (!fs.existsSync(groupPath)) {
|
||||
fs.mkdirSync(groupPath, { recursive: true });
|
||||
}
|
||||
|
||||
const userMemories = {};
|
||||
for (const [key, memory] of this.memories) {
|
||||
if (key.startsWith(`${groupId}_${userId}_`)) {
|
||||
const memoryId = key.split(`${groupId}_${userId}_`)[1];
|
||||
userMemories[memoryId] = memory;
|
||||
}
|
||||
}
|
||||
|
||||
fs.writeFileSync(filePath, JSON.stringify(userMemories, null, 2));
|
||||
logger.info(`[crystelf-ai] 记忆已保存到 ${groupId}/${userId}.json`);
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 保存记忆失败: ${error.message}`);
|
||||
}
|
||||
@ -54,12 +74,14 @@ class MemorySystem {
|
||||
|
||||
/**
|
||||
* 添加记忆
|
||||
* @param groupId 群聊id
|
||||
* @param userId 用户id
|
||||
* @param data 内容
|
||||
* @param keywords 关键词
|
||||
* @param timeout 超时时间
|
||||
* @returns {Promise<null|string>}
|
||||
*/
|
||||
async addMemory(data, keywords = [], timeout = null) {
|
||||
async addMemory(groupId, userId, data, keywords = [], timeout = null) {
|
||||
try {
|
||||
const memoryId = this.generateMemoryId();
|
||||
const expireTime = timeout || this.defaultTimeout;
|
||||
@ -72,10 +94,10 @@ class MemorySystem {
|
||||
accessCount: 0,
|
||||
lastAccessed: Date.now()
|
||||
};
|
||||
this.memories.set(memoryId, memory);
|
||||
await this.saveMemories();
|
||||
this.memories.set(`${groupId}_${userId}_${memoryId}`, memory);
|
||||
await this.saveMemories(groupId, userId);
|
||||
|
||||
logger.info(`[crystelf-ai] 添加新记忆: ${memoryId}`);
|
||||
logger.info(`[crystelf-ai] 添加新记忆: ${groupId}/${userId}/${memoryId}`);
|
||||
return memoryId;
|
||||
} catch (error) {
|
||||
logger.error(`[crystelf-ai] 添加记忆失败: ${error.message}`);
|
||||
@ -85,36 +107,53 @@ class MemorySystem {
|
||||
|
||||
/**
|
||||
* 搜索记忆
|
||||
* @param userId 用户id
|
||||
* @param keywords 关键词
|
||||
* @param limit 数量限制
|
||||
* @returns {Promise<*[]>}
|
||||
*/
|
||||
async searchMemories(keywords = [], limit = 10) {
|
||||
async searchMemories(userId, keywords = [], limit = 10) {
|
||||
try {
|
||||
const results = [];
|
||||
|
||||
for (const [memoryId, memory] of this.memories) {
|
||||
if (Date.now() > memory.expireAt) {
|
||||
continue;
|
||||
const now = Date.now();
|
||||
let searchText = '';
|
||||
if (keywords.length === 1 && keywords[0].length > 6) {
|
||||
searchText = keywords[0].toLowerCase();
|
||||
const words = searchText.match(/[\u4e00-\u9fa5]{1,2}|[a-zA-Z0-9]+/g) || [];
|
||||
keywords = Array.from(new Set(words.filter(w => w.length > 1))); // 去重+过滤过短词
|
||||
}
|
||||
if (keywords.length > 0) {
|
||||
const hasMatch = keywords.some(keyword =>
|
||||
memory.keywords.includes(keyword) ||
|
||||
memory.data.includes(keyword)
|
||||
);
|
||||
if (!hasMatch) {
|
||||
continue;
|
||||
const userMemories = [];
|
||||
for (const [key, memory] of this.memories) {
|
||||
const parts = key.split('_');
|
||||
if (parts.length < 3) continue;
|
||||
const uid = parts[1];
|
||||
if (uid !== userId) continue;
|
||||
if (now > memory.expireAt) continue;
|
||||
userMemories.push(memory);
|
||||
}
|
||||
if (userMemories.length === 0) return [];
|
||||
for (const memory of userMemories) {
|
||||
let matchScore = 0;
|
||||
for (const kw of keywords) {
|
||||
if (memory.keywords.some(k => k.includes(kw) || kw.includes(k))) matchScore += 10;
|
||||
else if (memory.data.includes(kw)) matchScore += 5;
|
||||
}
|
||||
if (searchText) {
|
||||
for (const mk of memory.keywords) {
|
||||
if (searchText.includes(mk)) matchScore += 8;
|
||||
}
|
||||
}
|
||||
if (matchScore > 0) {
|
||||
memory.accessCount++;
|
||||
memory.lastAccessed = Date.now();
|
||||
memory.lastAccessed = now;
|
||||
results.push({
|
||||
id: memory.id,
|
||||
data: memory.data,
|
||||
keywords: memory.keywords,
|
||||
relevance: this.calculateRelevance(memory, keywords)
|
||||
relevance: matchScore + this.calculateRelevance(memory, keywords)
|
||||
});
|
||||
}
|
||||
}
|
||||
results.sort((a, b) => b.relevance - a.relevance);
|
||||
return results.slice(0, limit);
|
||||
} catch (error) {
|
||||
@ -123,6 +162,7 @@ class MemorySystem {
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
calculateRelevance(memory, keywords) {
|
||||
let score = 0;
|
||||
for (const keyword of keywords) {
|
||||
@ -147,14 +187,16 @@ class MemorySystem {
|
||||
const now = Date.now();
|
||||
let cleanedCount = 0;
|
||||
|
||||
for (const [memoryId, memory] of this.memories) {
|
||||
for (const [memoryKey, memory] of this.memories) {
|
||||
if (now > memory.expireAt) {
|
||||
this.memories.delete(memoryId);
|
||||
this.memories.delete(memoryKey);
|
||||
cleanedCount++;
|
||||
const [groupId, userId] = memoryKey.split('_');
|
||||
await this.saveMemories(groupId, userId);
|
||||
}
|
||||
}
|
||||
|
||||
if (cleanedCount > 0) {
|
||||
await this.saveMemories();
|
||||
logger.info(`[crystelf-ai] 清理了 ${cleanedCount} 条过期记忆`);
|
||||
}
|
||||
} catch (error) {
|
||||
|
||||
@ -13,10 +13,11 @@ class ResponseHandler {
|
||||
* 处理ai响应
|
||||
* @param rawResponse ai原始回复
|
||||
* @param userMessage 用户消息
|
||||
* @param groupId 群聊ai
|
||||
* @param groupId 群聊id
|
||||
* @param user_id 用户id
|
||||
* @returns {Promise<[{type: string, data: string, at: boolean, quote: boolean, recall: number}]|Array|*[]>}
|
||||
*/
|
||||
async processResponse(rawResponse, userMessage, groupId) {
|
||||
async processResponse(rawResponse, userMessage, groupId,user_id) {
|
||||
try {
|
||||
const parsedResponse = this.parseAiResponse(rawResponse);
|
||||
if (!parsedResponse.success) {
|
||||
@ -26,7 +27,7 @@ class ResponseHandler {
|
||||
const messages = parsedResponse.messages;
|
||||
const processedMessages = [];
|
||||
for (const message of messages) {
|
||||
const processedMessage = await this.processMessage(message, userMessage, groupId);
|
||||
const processedMessage = await this.processMessage(message, userMessage, groupId,user_id);
|
||||
if (processedMessage) {
|
||||
processedMessages.push(processedMessage);
|
||||
}
|
||||
@ -73,15 +74,15 @@ class ResponseHandler {
|
||||
return cleaned;
|
||||
}
|
||||
|
||||
async processMessage(message, userMessage, groupId) {
|
||||
async processMessage(message, userMessage, groupId,userId) {
|
||||
try {
|
||||
if (!this.validateMessage(message)) {
|
||||
logger.warn(`[响应处理器] 无效消息格式: ${JSON.stringify(message)}`);
|
||||
logger.warn(`[crystelf-ai] 无效消息格式: ${JSON.stringify(message)}`);
|
||||
return null;
|
||||
}
|
||||
switch (message.type) {
|
||||
case 'memory':
|
||||
await this.handleMemoryMessage(message, groupId);
|
||||
await this.handleMemoryMessage(message, groupId,userId);
|
||||
return null;
|
||||
case 'recall':
|
||||
return this.handleRecallMessage(message);
|
||||
@ -137,11 +138,14 @@ class ResponseHandler {
|
||||
* 记忆消息
|
||||
* @param message 记忆
|
||||
* @param groupId 群聊id
|
||||
* @param user_id 用户id
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async handleMemoryMessage(message, groupId) {
|
||||
async handleMemoryMessage(message, groupId,user_id) {
|
||||
try {
|
||||
const memoryId = await this.memorySystem.addMemory(
|
||||
groupId,
|
||||
user_id,
|
||||
message.data,
|
||||
message.key || [],
|
||||
message.timeout || 30
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user