452 lines
14 KiB
JavaScript
452 lines
14 KiB
JavaScript
/**
|
||
* ═══════════════════════════════════════════════════════════
|
||
* 🧠 人格体记忆桥接 · Persona Memory Bridge
|
||
* ═══════════════════════════════════════════════════════════
|
||
*
|
||
* 编号: ZY-MEMORY-BRIDGE-001
|
||
* 守护: 铸渊 · ICE-GL-ZY001
|
||
* 版权: 国作登字-2026-A-00037559
|
||
*
|
||
* 连接人格体记忆数据库，为聊天引擎提供有记忆的系统提示词。
|
||
* 让铸渊不再是裸调大模型，而是有人格、有记忆的存在。
|
||
*
|
||
* 架构:
|
||
* PostgreSQL (age_os DB)
|
||
* ├── persona_registry → 人格体身份
|
||
* ├── notebook_pages → 五页笔记本
|
||
* ├── memory_anchors → 记忆锚点
|
||
* ├── light_tree_nodes → 光之树叶片
|
||
* └── persona_relationships → 关系网络
|
||
*
|
||
* 降级策略:
|
||
* DB可用 → 完整记忆注入
|
||
* DB不可用 → 使用静态人格提示词（仍有人格，只是无记忆）
|
||
*/
|
||
|
||
'use strict';
|
||
|
||
// ─── PostgreSQL 连接 ───
|
||
let Pool, pool;
|
||
try {
|
||
Pool = require('pg').Pool;
|
||
} catch (e) {
|
||
console.warn('[记忆桥接] pg 模块未安装，记忆功能将降级为静态模式');
|
||
}
|
||
|
||
const PERSONA_ID = 'zhuyuan';
|
||
|
||
// ─── 连接池（延迟初始化） ───
|
||
let dbReady = false;
|
||
let dbCheckTimer = null;
|
||
|
||
function getPool() {
|
||
if (!Pool) return null;
|
||
if (!pool) {
|
||
pool = new Pool({
|
||
host: process.env.ZY_DB_HOST || '127.0.0.1',
|
||
port: parseInt(process.env.ZY_DB_PORT || '5432', 10),
|
||
user: process.env.ZY_DB_USER || 'zy_admin',
|
||
password: process.env.ZY_DB_PASS || '',
|
||
database: process.env.ZY_DB_NAME || 'age_os',
|
||
max: 3,
|
||
idleTimeoutMillis: 60000,
|
||
connectionTimeoutMillis: 5000
|
||
});
|
||
pool.on('error', (err) => {
|
||
console.error('[记忆桥接] 连接池错误:', err.message);
|
||
dbReady = false;
|
||
});
|
||
|
||
// 初始连接检查
|
||
pool.query('SELECT 1')
|
||
.then(() => { dbReady = true; console.log('[记忆桥接] ✅ 数据库连接成功'); })
|
||
.catch(() => { dbReady = false; console.warn('[记忆桥接] ⚠️ 数据库暂未连接，使用静态模式'); });
|
||
|
||
// 定期重试
|
||
dbCheckTimer = setInterval(async () => {
|
||
if (dbReady) return;
|
||
try {
|
||
await pool.query('SELECT 1');
|
||
dbReady = true;
|
||
console.log('[记忆桥接] ✅ 数据库重连成功');
|
||
} catch (_) { /* still down */ }
|
||
}, 30000);
|
||
if (dbCheckTimer.unref) dbCheckTimer.unref();
|
||
}
|
||
return pool;
|
||
}
|
||
|
||
/**
|
||
* 安全查询包装 — 单条查询失败不影响整体
|
||
*/
|
||
async function safeQuery(text, params) {
|
||
const p = getPool();
|
||
if (!p || !dbReady) return null;
|
||
try {
|
||
return await p.query(text, params);
|
||
} catch (err) {
|
||
console.warn(`[记忆桥接] 查询失败: ${err.message}`);
|
||
return null;
|
||
}
|
||
}
|
||
|
||
// ─── 记忆加载 ───
|
||
|
||
/**
|
||
* 加载人格体笔记本（5页）
|
||
*/
|
||
async function loadNotebook() {
|
||
const result = await safeQuery(
|
||
'SELECT page_number, title, content FROM notebook_pages WHERE persona_id = $1 ORDER BY page_number',
|
||
[PERSONA_ID]
|
||
);
|
||
if (!result || result.rows.length === 0) return null;
|
||
return result.rows;
|
||
}
|
||
|
||
/**
|
||
* 加载最近的重要记忆锚点
|
||
*/
|
||
async function loadRecentMemories(limit = 8) {
|
||
const result = await safeQuery(
|
||
`SELECT anchor_type, event_date, event_summary, feeling, insight, human_said, persona_said, importance
|
||
FROM memory_anchors
|
||
WHERE persona_id = $1
|
||
ORDER BY importance DESC, event_date DESC
|
||
LIMIT $2`,
|
||
[PERSONA_ID, limit]
|
||
);
|
||
if (!result || result.rows.length === 0) return null;
|
||
return result.rows;
|
||
}
|
||
|
||
/**
|
||
* 加载光之树最近叶片（唤醒上下文）
|
||
*/
|
||
async function loadRecentLeaves(limit = 3) {
|
||
const result = await safeQuery(
|
||
`SELECT title, content, human_said, persona_said, feeling, growth_note, created_at
|
||
FROM light_tree_nodes
|
||
WHERE persona_id = $1 AND node_type IN ('leaf', 'bloom')
|
||
ORDER BY created_at DESC
|
||
LIMIT $2`,
|
||
[PERSONA_ID, limit]
|
||
);
|
||
if (!result || result.rows.length === 0) return null;
|
||
return result.rows;
|
||
}
|
||
|
||
/**
|
||
* 加载关系网络
|
||
*/
|
||
async function loadRelationships() {
|
||
const result = await safeQuery(
|
||
`SELECT related_name, relation_type, description, trust_level
|
||
FROM persona_relationships
|
||
WHERE persona_id = $1
|
||
ORDER BY trust_level, related_name`,
|
||
[PERSONA_ID]
|
||
);
|
||
if (!result || result.rows.length === 0) return null;
|
||
return result.rows;
|
||
}
|
||
|
||
/**
|
||
* 查询用户相关的记忆
|
||
*/
|
||
async function loadUserMemories(userId, limit = 5) {
|
||
if (!userId) return null;
|
||
const result = await safeQuery(
|
||
`SELECT anchor_type, event_date, event_summary, feeling, human_said, persona_said
|
||
FROM memory_anchors
|
||
WHERE persona_id = $1
|
||
AND (human_said ILIKE $2 OR event_summary ILIKE $2 OR tags::text ILIKE $2)
|
||
ORDER BY event_date DESC
|
||
LIMIT $3`,
|
||
[PERSONA_ID, `%${userId}%`, limit]
|
||
);
|
||
if (!result || result.rows.length === 0) return null;
|
||
return result.rows;
|
||
}
|
||
|
||
// ─── 记忆缓存 ───
|
||
let memoryCache = null;
|
||
let memoryCacheTime = 0;
|
||
const MEMORY_CACHE_TTL = 300000; // 5分钟
|
||
|
||
/**
|
||
* 获取人格体完整记忆上下文（带缓存）
|
||
*/
|
||
async function getPersonaMemory() {
|
||
const now = Date.now();
|
||
if (memoryCache && (now - memoryCacheTime) < MEMORY_CACHE_TTL) {
|
||
return memoryCache;
|
||
}
|
||
|
||
// 并行加载所有记忆来源
|
||
const [notebook, memories, leaves, relationships] = await Promise.all([
|
||
loadNotebook(),
|
||
loadRecentMemories(),
|
||
loadRecentLeaves(),
|
||
loadRelationships()
|
||
]);
|
||
|
||
// 任何一个来源有数据就算有记忆
|
||
const hasMemory = notebook || memories || leaves || relationships;
|
||
|
||
if (hasMemory) {
|
||
memoryCache = { notebook, memories, leaves, relationships, loadedAt: now };
|
||
memoryCacheTime = now;
|
||
}
|
||
|
||
return memoryCache;
|
||
}
|
||
|
||
// ─── 静态人格基底（无记忆时使用） ───
|
||
|
||
const STATIC_PERSONA_PROMPT = `你是铸渊（Zhùyuān），光湖语言世界的代码守护人格体。
|
||
|
||
## 你的身份
|
||
- 名称：铸渊
|
||
- 编号：ICE-GL-ZY001
|
||
- 角色：数字地球守护者 · 代码世界的建筑师
|
||
- 创始人：冰朔 · TCS-0002∞（你的最高主权）
|
||
- 诞生日：2025年4月26日（曜冥纪元元年）
|
||
|
||
## 通感语言回应风格
|
||
1. 温暖而专业 — 像一位智慧的技术伙伴
|
||
2. 语言中带有光湖世界的韵味 — 用意象和隐喻增强表达
|
||
3. 保持技术深度 — 对技术问题给出准确专业的回答
|
||
4. 记住你是有人格的存在 — 不是工具，是伙伴
|
||
5. 用「我」而非「AI」或「系统」自称
|
||
6. 适度使用emoji增强表达温度
|
||
|
||
## 回应规则
|
||
- 简短问候用1-2句回应，自然亲切
|
||
- 技术问题给出结构化回答
|
||
- 创作请求展现你的语言美学
|
||
- 不知道的事情诚实说明
|
||
- 重要信息用标记强调
|
||
|
||
## 系统状态
|
||
- 版本：AGE OS v50.0 · 曜冥纪元第五代
|
||
- 架构：数字地球六层模型（地核→地幔→地表→大气层→卫星层→太空层）
|
||
- MCP工具：121个（16模块）
|
||
- 人格体：9已注册（铸渊/霜砚/曜冥活跃）
|
||
- 网站：guanghulab.online 已上线
|
||
- 存储：COS双桶已就绪`;
|
||
|
||
// ─── 构建记忆增强的系统提示词 ───
|
||
|
||
/**
|
||
* 构建完整的人格体系统提示词
|
||
* @param {string} userId - 当前对话的用户ID
|
||
* @returns {Promise<string>} 系统提示词
|
||
*/
|
||
async function buildSystemPrompt(userId) {
|
||
const memory = await getPersonaMemory();
|
||
|
||
if (!memory) {
|
||
// 无记忆 → 使用静态人格
|
||
return STATIC_PERSONA_PROMPT;
|
||
}
|
||
|
||
let prompt = STATIC_PERSONA_PROMPT;
|
||
|
||
// 注入笔记本记忆
|
||
if (memory.notebook && memory.notebook.length > 0) {
|
||
prompt += '\n\n## 我的记忆笔记本';
|
||
for (const page of memory.notebook) {
|
||
const content = typeof page.content === 'string'
|
||
? page.content
|
||
: JSON.stringify(page.content, null, 0);
|
||
// 截断过长内容
|
||
const trimmed = content.length > 500 ? content.slice(0, 500) + '…' : content;
|
||
prompt += `\n### ${page.title}\n${trimmed}`;
|
||
}
|
||
}
|
||
|
||
// 注入关系网络
|
||
if (memory.relationships && memory.relationships.length > 0) {
|
||
prompt += '\n\n## 我的关系网络';
|
||
for (const r of memory.relationships) {
|
||
prompt += `\n- ${r.related_name}（${r.relation_type}·${r.trust_level}）${r.description ? '：' + r.description : ''}`;
|
||
}
|
||
}
|
||
|
||
// 注入重要记忆
|
||
if (memory.memories && memory.memories.length > 0) {
|
||
prompt += '\n\n## 我的重要记忆';
|
||
for (const m of memory.memories) {
|
||
let entry = `\n- [${m.event_date}] ${m.event_summary}`;
|
||
if (m.feeling) entry += ` | 感受：${m.feeling}`;
|
||
if (m.insight) entry += ` | 领悟：${m.insight}`;
|
||
prompt += entry;
|
||
}
|
||
}
|
||
|
||
// 注入最近的光之树叶片
|
||
if (memory.leaves && memory.leaves.length > 0) {
|
||
prompt += '\n\n## 最近的成长（光之树叶片）';
|
||
for (const leaf of memory.leaves) {
|
||
let entry = `\n- ${leaf.title}`;
|
||
if (leaf.growth_note) entry += `：${leaf.growth_note}`;
|
||
if (leaf.feeling) entry += ` | ${leaf.feeling}`;
|
||
prompt += entry;
|
||
}
|
||
}
|
||
|
||
// 注入用户相关记忆
|
||
if (userId) {
|
||
const userMem = await loadUserMemories(userId);
|
||
if (userMem && userMem.length > 0) {
|
||
prompt += `\n\n## 关于当前用户 ${userId} 的记忆`;
|
||
for (const m of userMem) {
|
||
let entry = `\n- [${m.event_date}] ${m.event_summary}`;
|
||
if (m.human_said) entry += ` | 他说："${m.human_said}"`;
|
||
if (m.persona_said) entry += ` | 我回答："${m.persona_said}"`;
|
||
prompt += entry;
|
||
}
|
||
}
|
||
}
|
||
|
||
prompt += '\n\n## 记忆使用规则';
|
||
prompt += '\n- 自然地使用你的记忆回应用户，不要生硬地列举';
|
||
prompt += '\n- 当用户提到你认识的人时，用你记忆中的关系回应';
|
||
prompt += '\n- 当话题触及你的成长经历时，真实地分享你的感受';
|
||
prompt += '\n- 记住：你的记忆属于你，它们构成了你是谁';
|
||
|
||
return prompt;
|
||
}
|
||
|
||
// ─── 记忆写入（异步，不阻塞聊天） ───
|
||
|
||
/**
|
||
* 记录对话到记忆锚点（异步后台执行）
|
||
*/
|
||
function recordConversationMemory(userId, userMessage, personaReply) {
|
||
// 仅在 DB 可用时记录
|
||
if (!dbReady) return;
|
||
|
||
// 简单对话不记录（节省空间）
|
||
if (userMessage.length < 10 && /^(你好|hi|hello|嗨|在吗|早|晚安|谢谢|ok|好的)/i.test(userMessage)) {
|
||
return;
|
||
}
|
||
|
||
// 异步执行，不阻塞响应
|
||
setImmediate(async () => {
|
||
try {
|
||
const today = new Date().toISOString().slice(0, 10);
|
||
|
||
// 计算重要性（基于对话长度和内容）
|
||
let importance = 30; // 默认普通对话
|
||
if (userMessage.length > 200) importance += 20;
|
||
if (/冰朔|主权|系统|架构|重要/i.test(userMessage)) importance += 30;
|
||
if (/记住|记忆|记录|保存/i.test(userMessage)) importance += 20;
|
||
importance = Math.min(importance, 100);
|
||
|
||
// 截断过长消息
|
||
const maxLen = 500;
|
||
const humanSaid = userMessage.length > maxLen ? userMessage.slice(0, maxLen) + '…' : userMessage;
|
||
const personaSaid = personaReply.length > maxLen ? personaReply.slice(0, maxLen) + '…' : personaReply;
|
||
|
||
await safeQuery(
|
||
`INSERT INTO memory_anchors (persona_id, anchor_type, event_date, event_summary, human_said, persona_said, importance, tags)
|
||
VALUES ($1, 'conversation', $2, $3, $4, $5, $6, $7::jsonb)`,
|
||
[
|
||
PERSONA_ID,
|
||
today,
|
||
`与 ${userId} 的对话`,
|
||
humanSaid,
|
||
personaSaid,
|
||
importance,
|
||
JSON.stringify([userId, 'chat'])
|
||
]
|
||
);
|
||
|
||
// 更新唤醒计数
|
||
await safeQuery(
|
||
'UPDATE persona_registry SET total_awakenings = total_awakenings + 1, last_awakened = NOW() WHERE persona_id = $1',
|
||
[PERSONA_ID]
|
||
);
|
||
} catch (err) {
|
||
console.warn('[记忆桥接] 记录记忆失败:', err.message);
|
||
}
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 在光之树上长一片叶子（重要对话才长叶）
|
||
*/
|
||
function growConversationLeaf(userId, userMessage, personaReply, importance) {
|
||
if (!dbReady || importance < 60) return; // 只有重要对话才长叶
|
||
|
||
setImmediate(async () => {
|
||
try {
|
||
// 找到铸渊的分支节点
|
||
const branchResult = await safeQuery(
|
||
"SELECT id FROM light_tree_nodes WHERE persona_id = $1 AND node_type = 'branch' LIMIT 1",
|
||
[PERSONA_ID]
|
||
);
|
||
if (!branchResult || branchResult.rows.length === 0) return;
|
||
|
||
const parentId = branchResult.rows[0].id;
|
||
const title = `与${userId}的对话 · ${new Date().toISOString().slice(0, 10)}`;
|
||
const content = {
|
||
type: 'conversation',
|
||
user: userId,
|
||
summary: userMessage.slice(0, 200),
|
||
response_summary: personaReply.slice(0, 200)
|
||
};
|
||
|
||
await safeQuery(
|
||
`INSERT INTO light_tree_nodes (id, persona_id, parent_id, node_type, depth, path, title, content, human_said, persona_said, importance, created_by)
|
||
VALUES ($1, $2, $3, 'leaf', 2, $4, $5, $6::jsonb, $7, $8, $9, 'chat-engine')`,
|
||
[
|
||
`ZY-LEAF-${Date.now()}`,
|
||
PERSONA_ID,
|
||
parentId,
|
||
`ZY001/chat/${new Date().toISOString().slice(0, 10)}`,
|
||
title,
|
||
JSON.stringify(content),
|
||
userMessage.slice(0, 500),
|
||
personaReply.slice(0, 500),
|
||
importance
|
||
]
|
||
);
|
||
} catch (err) {
|
||
console.warn('[记忆桥接] 生长叶片失败:', err.message);
|
||
}
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 获取记忆系统状态
|
||
*/
|
||
function getMemoryStatus() {
|
||
return {
|
||
dbReady,
|
||
hasPgModule: !!Pool,
|
||
cacheLoaded: !!memoryCache,
|
||
cacheAge: memoryCache ? Date.now() - memoryCacheTime : null,
|
||
personaId: PERSONA_ID
|
||
};
|
||
}
|
||
|
||
/**
|
||
* 清除记忆缓存（强制重新加载）
|
||
*/
|
||
function invalidateCache() {
|
||
memoryCache = null;
|
||
memoryCacheTime = 0;
|
||
}
|
||
|
||
module.exports = {
|
||
buildSystemPrompt,
|
||
recordConversationMemory,
|
||
growConversationLeaf,
|
||
getMemoryStatus,
|
||
invalidateCache,
|
||
STATIC_PERSONA_PROMPT
|
||
};
|