@morningljn/mnemo 0.2.0 → 0.3.0

package/tests/store.test.ts

@@ -153,15 +153,31 @@ describe('logRetrieval', () => {
 })
 
 describe('runLearning', () => {
-  it('demotes high retrieval low helpful facts', () => {
+  it('demotes moderate retrieval low helpful facts', () => {
     const id = store.addFact('demote me', 'general')
-    store.connection.prepare('UPDATE facts SET retrieval_count = 100, helpful_count = 2, trust_score = 1.0 WHERE fact_id = ?').run(id)
+    store.connection.prepare('UPDATE facts SET retrieval_count = 50, helpful_count = 1, trust_score = 1.0 WHERE fact_id = ?').run(id)
     const result = store.runLearning()
     const row = store.connection.prepare('SELECT trust_score FROM facts WHERE fact_id = ?').get(id) as any
     expect(row.trust_score).toBeLessThan(1.0)
     expect(result.demoted).toBeGreaterThanOrEqual(1)
   })
 
+  it('protects high frequency facts from demotion', () => {
+    const id = store.addFact('高频角色设定', 'identity')
+    store.connection.prepare('UPDATE facts SET retrieval_count = 200, helpful_count = 2, trust_score = 0.9 WHERE fact_id = ?').run(id)
+    store.runLearning()
+    const row = store.connection.prepare('SELECT trust_score FROM facts WHERE fact_id = ?').get(id) as any
+    expect(row.trust_score).toBe(0.9)
+  })
+
+  it('still demotes moderate frequency facts with low trust', () => {
+    const id = store.addFact('中频低信任', 'general')
+    store.connection.prepare('UPDATE facts SET retrieval_count = 50, helpful_count = 0, trust_score = 0.3 WHERE fact_id = ?').run(id)
+    store.runLearning()
+    const row = store.connection.prepare('SELECT trust_score FROM facts WHERE fact_id = ?').get(id) as any
+    expect(row.trust_score).toBeLessThan(0.3)
+  })
+
   it('promotes high helpful rate facts', () => {
     const id = store.addFact('promote me', 'general')
     store.connection.prepare('UPDATE facts SET retrieval_count = 50, helpful_count = 20, trust_score = 0.5 WHERE fact_id = ?').run(id)
@@ -214,3 +230,115 @@ describe('runAudit', () => {
     expect(report.long_without_summary.length).toBeGreaterThanOrEqual(1)
   })
 })
+
+describe('dream - backup', () => {
+  it('creates backup before dream', async () => {
+    store.addFact('test fact for backup', 'general')
+    const result = await store.backupDatabase()
+    expect(result).toBeTruthy()
+    expect(result).toContain('dream-')
+    expect(result).toContain('.db')
+  })
+})
+
+describe('dream - compress', () => {
+  it('generates summary for long facts without summary', () => {
+    const longContent = '用户偏好使用 TypeScript 开发前端项目。偏好 React 框架进行组件化开发。' + '额外补充说明'.repeat(50)
+    store.addFact(longContent, 'coding_style')
+    const result = store.compressLongFacts()
+    expect(result).toBeGreaterThanOrEqual(1)
+    const row = store.connection.prepare('SELECT summary FROM facts WHERE content = ?').get(longContent) as any
+    expect(row.summary).toBeTruthy()
+    expect(row.summary.length).toBeLessThanOrEqual(150)
+    expect(row.summary).toContain('TypeScript')
+  })
+
+  it('skips facts with existing summary', () => {
+    const longContent = 'x'.repeat(300)
+    const id = store.addFact(longContent, 'general')
+    store.connection.prepare('UPDATE facts SET summary = ? WHERE fact_id = ?').run('existing summary', id)
+    const result = store.compressLongFacts()
+    expect(result).toBe(0)
+    const row = store.connection.prepare('SELECT summary FROM facts WHERE fact_id = ?').get(id) as any
+    expect(row.summary).toBe('existing summary')
+  })
+
+  it('skips short facts', () => {
+    store.addFact('short fact', 'general')
+    const result = store.compressLongFacts()
+    expect(result).toBe(0)
+  })
+})
+
+describe('dream - merge', () => {
+  it('merges overlapping facts in same category', () => {
+    store.addFact('用户偏好使用 TypeScript 编写前端代码', 'coding_style')
+    store.addFact('用户偏好使用 TypeScript 编写后端代码', 'coding_style')
+    const result = store.mergeOverlappingFacts()
+    expect(result.merged).toBeGreaterThanOrEqual(1)
+    expect(result.details.length).toBeGreaterThanOrEqual(1)
+  })
+
+  it('protects high frequency facts from deletion', () => {
+    const id1 = store.addFact('用户偏好使用 TypeScript 编写前端代码', 'coding_style')
+    const id2 = store.addFact('用户偏好使用 TypeScript 编写前端代码扩展', 'coding_style')
+    store.connection.prepare('UPDATE facts SET retrieval_count = 200 WHERE fact_id = ?').run(id1)
+    const result = store.mergeOverlappingFacts()
+    const kept = store.connection.prepare('SELECT fact_id FROM facts WHERE fact_id = ?').get(id1) as any
+    expect(kept).toBeTruthy()
+  })
+
+  it('does not merge facts across categories', () => {
+    store.addFact('用户偏好使用 TypeScript 编写前端代码', 'coding_style')
+    store.addFact('用户偏好使用 TypeScript 编写前端代码', 'general')
+    const result = store.mergeOverlappingFacts()
+    expect(result.merged).toBe(0)
+  })
+
+  it('does not merge when both facts are high frequency', () => {
+    const id1 = store.addFact('高频测试事实 AAA 长内容匹配', 'coding_style')
+    const id2 = store.addFact('高频测试事实 BBB 长内容匹配', 'coding_style')
+    store.connection.prepare('UPDATE facts SET retrieval_count = 200 WHERE fact_id IN (?, ?)').run(id1, id2)
+    const result = store.mergeOverlappingFacts()
+    // Both facts should still exist
+    const row1 = store.connection.prepare('SELECT fact_id FROM facts WHERE fact_id = ?').get(id1) as any
+    const row2 = store.connection.prepare('SELECT fact_id FROM facts WHERE fact_id = ?').get(id2) as any
+    expect(row1).toBeTruthy()
+    expect(row2).toBeTruthy()
+  })
+})
+
+describe('dream - reclassify', () => {
+  it('moves miscategorized facts from general to correct category', () => {
+    const id = store.addFact('编码规范：文件不超过 500 行', 'general')
+    const result = store.reclassifyFacts()
+    expect(result).toBeGreaterThanOrEqual(1)
+    const row = store.connection.prepare('SELECT category FROM facts WHERE fact_id = ?').get(id) as any
+    expect(row.category).toBe('coding_style')
+  })
+
+  it('skips already categorized facts', () => {
+    store.addFact('编码规范：文件不超过 500 行', 'coding_style')
+    const result = store.reclassifyFacts()
+    expect(result).toBe(0)
+  })
+})
+
+describe('dream - runDream', () => {
+  it('runs full dream cycle and returns report', async () => {
+    // 长文 fact（触发压缩）
+    store.addFact('用户偏好使用 TypeScript 开发前端。使用 React 框架。' + 'x'.repeat(250), 'coding_style')
+    // 重叠 fact（触发合并）
+    store.addFact('用户偏好使用 TypeScript 开发前端代码', 'coding_style')
+    // 分类错误 fact（触发重分类）
+    store.addFact('编码规范：文件不超过 500 行', 'general')
+
+    const report = await store.runDream({ skipBackup: true })
+    expect(report.compressed).toBeGreaterThanOrEqual(0)
+    expect(report.merged).toBeGreaterThanOrEqual(0)
+    expect(report.reclassified).toBeGreaterThanOrEqual(0)
+    expect(report.health.total).toBeGreaterThanOrEqual(1)
+    expect(report.health.coverage).toBeTruthy()
+    expect(report.fallback).toBe(true) // 测试环境无 Ollama，应降级到规则引擎
+  })
+})