308 lines
8.4 KiB
JavaScript
308 lines
8.4 KiB
JavaScript
const BasePouchCommand = require('../BasePouchCommand')
|
||
const fs = require('fs-extra')
|
||
const path = require('path')
|
||
const { COMMANDS } = require('../../../../constants')
|
||
|
||
/**
|
||
* 记忆检索锦囊命令
|
||
* 负责从记忆库中检索相关知识和经验
|
||
*/
|
||
class RecallCommand extends BasePouchCommand {
|
||
constructor () {
|
||
super()
|
||
}
|
||
|
||
getPurpose () {
|
||
return 'AI主动检索记忆中的专业知识、最佳实践和历史经验'
|
||
}
|
||
|
||
async getContent (args) {
|
||
const [query] = args
|
||
|
||
try {
|
||
const memories = await this.getAllMemories(query)
|
||
|
||
if (memories.length === 0) {
|
||
return `🧠 AI记忆体系中暂无内容。
|
||
💡 建议：
|
||
1. 使用 MCP PromptX remember 工具内化新知识
|
||
2. 使用 MCP PromptX learn 工具学习后再内化
|
||
3. 开始构建AI的专业知识体系`
|
||
}
|
||
|
||
const formattedMemories = this.formatRetrievedKnowledge(memories, query)
|
||
|
||
return `🧠 AI记忆体系 ${query ? `检索"${query}"` : '全部记忆'} (${memories.length}条)：
|
||
${formattedMemories}
|
||
💡 记忆运用建议：
|
||
1. 结合当前任务场景灵活运用
|
||
2. 根据实际情况调整和变通
|
||
3. 持续学习和增强记忆能力`
|
||
} catch (error) {
|
||
return `❌ 检索记忆时出错：${error.message}`
|
||
}
|
||
}
|
||
|
||
getPATEOAS (args) {
|
||
const [query] = args
|
||
const currentState = query ? `recalled-${query}` : 'recall-waiting'
|
||
|
||
return {
|
||
currentState,
|
||
availableTransitions: ['hello', 'remember', 'learn', 'recall'],
|
||
nextActions: [
|
||
{
|
||
name: '选择角色',
|
||
description: '选择专业角色来应用检索到的知识',
|
||
method: 'MCP PromptX hello 工具'
|
||
},
|
||
{
|
||
name: '记忆新知识',
|
||
description: '继续内化更多专业知识',
|
||
method: 'MCP PromptX remember 工具'
|
||
},
|
||
{
|
||
name: '学习资源',
|
||
description: '学习相关专业资源',
|
||
method: 'MCP PromptX learn 工具'
|
||
},
|
||
{
|
||
name: '继续检索',
|
||
description: '检索其他相关记忆',
|
||
method: 'MCP PromptX recall 工具'
|
||
}
|
||
],
|
||
metadata: {
|
||
query: query || null,
|
||
resultCount: this.lastSearchCount || 0,
|
||
searchTime: new Date().toISOString(),
|
||
hasResults: (this.lastSearchCount || 0) > 0
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取所有记忆（支持多行格式）
|
||
*/
|
||
async getAllMemories (query) {
|
||
this.lastSearchCount = 0
|
||
const memories = []
|
||
|
||
// 读取单一记忆文件
|
||
const { getDirectoryService } = require('../../../utils/DirectoryService')
|
||
const directoryService = getDirectoryService()
|
||
const memoryDir = await directoryService.getMemoryDirectory()
|
||
const memoryFile = path.join(memoryDir, 'declarative.md')
|
||
|
||
try {
|
||
if (await fs.pathExists(memoryFile)) {
|
||
const content = await fs.readFile(memoryFile, 'utf-8')
|
||
const memoryBlocks = this.parseMemoryBlocks(content)
|
||
|
||
for (const memoryBlock of memoryBlocks) {
|
||
const memory = this.parseMemoryBlock(memoryBlock)
|
||
if (memory && (!query || this.matchesMemory(memory, query))) {
|
||
memories.push(memory)
|
||
}
|
||
}
|
||
}
|
||
} catch (error) {
|
||
console.error('Error reading memories:', error)
|
||
}
|
||
|
||
this.lastSearchCount = memories.length
|
||
return memories
|
||
}
|
||
|
||
/**
|
||
* 解析记忆块（新多行格式）
|
||
*/
|
||
parseMemoryBlocks (content) {
|
||
const blocks = []
|
||
const lines = content.split('\n')
|
||
let currentBlock = []
|
||
let inBlock = false
|
||
|
||
for (const line of lines) {
|
||
if (line.match(/^- \d{4}\/\d{2}\/\d{2} \d{2}:\d{2} START$/)) {
|
||
// 开始新的记忆块
|
||
if (inBlock && currentBlock.length > 0) {
|
||
blocks.push(currentBlock.join('\n'))
|
||
}
|
||
currentBlock = [line]
|
||
inBlock = true
|
||
} else if (line === '- END' && inBlock) {
|
||
// 结束当前记忆块
|
||
currentBlock.push(line)
|
||
blocks.push(currentBlock.join('\n'))
|
||
currentBlock = []
|
||
inBlock = false
|
||
} else if (inBlock) {
|
||
// 记忆块内容
|
||
currentBlock.push(line)
|
||
}
|
||
}
|
||
|
||
// 处理未结束的块
|
||
if (inBlock && currentBlock.length > 0) {
|
||
blocks.push(currentBlock.join('\n'))
|
||
}
|
||
|
||
return blocks
|
||
}
|
||
|
||
/**
|
||
* 解析单个记忆块
|
||
*/
|
||
parseMemoryBlock (blockContent) {
|
||
const lines = blockContent.split('\n')
|
||
|
||
// 解析开始行：- 2025/06/15 15:58 START
|
||
const startLine = lines[0]
|
||
const startMatch = startLine.match(/^- (\d{4}\/\d{2}\/\d{2} \d{2}:\d{2}) START$/)
|
||
if (!startMatch) return null
|
||
|
||
const timestamp = startMatch[1]
|
||
|
||
// 查找标签行：--tags xxx
|
||
let tagsLine = ''
|
||
let contentLines = []
|
||
|
||
for (let i = 1; i < lines.length; i++) {
|
||
const line = lines[i]
|
||
if (line.startsWith('--tags ')) {
|
||
tagsLine = line
|
||
} else if (line !== '- END') {
|
||
contentLines.push(line)
|
||
}
|
||
}
|
||
|
||
// 提取内容（去除空行）
|
||
const content = contentLines.join('\n').trim()
|
||
|
||
// 解析标签
|
||
let tags = []
|
||
if (tagsLine) {
|
||
const tagsContent = tagsLine.replace('--tags ', '')
|
||
const hashTags = tagsContent.match(/#[^\s]+/g) || []
|
||
const regularTags = tagsContent.replace(/#[^\s]+/g, '').trim().split(/\s+/).filter(t => t)
|
||
tags = [...regularTags, ...hashTags]
|
||
}
|
||
|
||
return {
|
||
timestamp,
|
||
content,
|
||
tags,
|
||
source: 'memory'
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 解析记忆行（向下兼容旧格式）
|
||
*/
|
||
parseMemoryLine (line) {
|
||
// 修复正则表达式，适配实际的记忆格式
|
||
// 格式：- 2025/05/31 14:30 内容 --tags 标签 ##分类 #评分:8 #有效期:长期
|
||
const match = line.match(/^- (\d{4}\/\d{2}\/\d{2} \d{2}:\d{2}) (.+)$/)
|
||
if (!match) return null
|
||
|
||
const [, timestamp, contentAndTags] = match
|
||
|
||
// 分离内容和标签
|
||
let content = contentAndTags
|
||
let tags = []
|
||
|
||
// 提取 --tags 后面的内容
|
||
const tagsMatch = contentAndTags.match(/--tags\s+(.*)/)
|
||
if (tagsMatch) {
|
||
const beforeTags = contentAndTags.substring(0, contentAndTags.indexOf('--tags')).trim()
|
||
content = beforeTags
|
||
|
||
// 解析标签部分，包括 --tags 后的内容和 # 开头的标签
|
||
const tagsContent = tagsMatch[1]
|
||
const hashTags = tagsContent.match(/#[^\s]+/g) || []
|
||
const regularTags = tagsContent.replace(/#[^\s]+/g, '').trim().split(/\s+/).filter(t => t)
|
||
|
||
tags = [...regularTags, ...hashTags]
|
||
} else {
|
||
// 如果没有 --tags，检查是否有直接的 # 标签
|
||
const hashTags = contentAndTags.match(/#[^\s]+/g) || []
|
||
if (hashTags.length > 0) {
|
||
content = contentAndTags.replace(/#[^\s]+/g, '').trim()
|
||
tags = hashTags
|
||
}
|
||
}
|
||
|
||
return {
|
||
timestamp,
|
||
content,
|
||
tags,
|
||
source: 'memory'
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 检查记忆是否匹配查询
|
||
*/
|
||
matchesMemory (memory, query) {
|
||
const lowerQuery = query.toLowerCase()
|
||
return memory.content.toLowerCase().includes(lowerQuery) ||
|
||
memory.tags.some(tag => tag.toLowerCase().includes(lowerQuery))
|
||
}
|
||
|
||
matchesQuery (content, query) {
|
||
const lowerContent = content.toLowerCase()
|
||
const lowerQuery = query.toLowerCase()
|
||
const keywords = lowerQuery.split(/\s+/)
|
||
|
||
return keywords.some(keyword => lowerContent.includes(keyword))
|
||
}
|
||
|
||
/**
|
||
* 格式化检索到的记忆（支持多行显示）
|
||
*/
|
||
formatRetrievedKnowledge (memories, query) {
|
||
return memories.map((memory, index) => {
|
||
// 保持完整的记忆内容，不进行截断
|
||
// 陈述性记忆的完整性对于系统价值至关重要
|
||
let content = memory.content
|
||
|
||
// 只对格式进行优化，但不截断内容
|
||
// 确保换行符正确显示
|
||
content = content.trim()
|
||
|
||
return `📝 ${index + 1}. **记忆** (${memory.timestamp})
|
||
${content}
|
||
${memory.tags.slice(0, 8).join(' ')}
|
||
---`
|
||
}).join('\n')
|
||
}
|
||
|
||
extractDomain (query) {
|
||
const domains = ['copywriter', 'scrum', 'developer', 'test', 'prompt']
|
||
const lowerQuery = query.toLowerCase()
|
||
|
||
return domains.find(domain => lowerQuery.includes(domain)) || null
|
||
}
|
||
|
||
getRelatedQuery (query) {
|
||
const relatedMap = {
|
||
copywriter: 'marketing',
|
||
scrum: 'agile',
|
||
frontend: 'ui',
|
||
backend: 'api',
|
||
test: 'qa'
|
||
}
|
||
|
||
for (const [key, value] of Object.entries(relatedMap)) {
|
||
if (query.toLowerCase().includes(key)) {
|
||
return value
|
||
}
|
||
}
|
||
|
||
return query + '-advanced'
|
||
}
|
||
}
|
||
|
||
module.exports = RecallCommand
|