@morningljn/mnemo 0.1.2 → 0.1.4

package/src/server.ts

@@ -7,6 +7,7 @@ import { homedir } from 'node:os'
 import { z } from 'zod/v4'
 import { MemoryStore } from './store.js'
 import { FactRetriever } from './retriever.js'
+import { ResourceManager } from './resources.js'
 import { fullSecurityScan } from './security.js'
 import type { FactStoreArgs, FactFeedbackArgs, FactCategory } from './types.js'
 
@@ -27,11 +28,11 @@ const FACT_STORE_DESCRIPTION = `结构化事实记忆系统（SQLite+FTS5 索引
 
 const factStoreSchema = {
   action: z.enum(['add', 'search', 'probe', 'related', 'reason', 'contradict', 'update', 'remove', 'list']),
-  content: z.string().optional().describe("事实内容（'add' 必需）"),
+  content: z.union([z.string(), z.array(z.string())]).optional().describe("事实内容（'add' 必需，支持批量）"),
   query: z.string().optional().describe("搜索查询（'search' 必需）"),
   entity: z.string().optional().describe("实体名（'probe'/'related' 使用）"),
   entities: z.array(z.string()).optional().describe("实体列表（'reason' 使用）"),
-  fact_id: z.number().optional().describe("事实 ID（'update'/'remove' 使用）"),
+  fact_id: z.union([z.number(), z.array(z.number())]).optional().describe("事实 ID（'update'/'remove' 使用，支持批量）"),
   category: z.enum(['identity', 'coding_style', 'tool_pref', 'workflow', 'general']).optional(),
   tags: z.string().optional().describe('逗号分隔标签'),
   trust_delta: z.number().optional().describe("'update' 的信任调整值"),
@@ -64,6 +65,10 @@ store.auditContradictions()
 // -- MCP Server --
 const server = new McpServer({ name: 'mnemo-mcp', version: '0.1.0' })
 
+// -- MCP Resources: 会话预热注入 --
+const resourceManager = new ResourceManager(store)
+resourceManager.registerResources(server)
+
 server.tool(
   'fact_store',
   FACT_STORE_DESCRIPTION,
@@ -76,20 +81,34 @@ server.tool(
       switch (a.action) {
         case 'add': {
           if (!a.content) return { content: [{ type: 'text' as const, text: JSON.stringify({ error: 'Missing required argument: content' }) }] }
-          const similar = store.findSimilarFact(a.content, category) ?? store.findSimilarFact(a.content)
-          let warnings: string[] | undefined
-          const scan = fullSecurityScan(a.content)
-          if (scan.warnings.length > 0 || scan.hasPii) warnings = [...scan.warnings]
-
-          if (similar) {
-            store.updateFact(similar.factId, { content: a.content, tags: a.tags, trustDelta: 0.05 })
-            const demoted = store.demoteContradictingFacts(similar.factId, a.content, category)
-            return { content: [{ type: 'text' as const, text: JSON.stringify({ fact_id: similar.factId, status: 'updated', reason: 'similar_fact_merged', ...(demoted > 0 ? { contradicted_demoted: demoted } : {}), ...(warnings ? { warnings } : {}) }) }] }
+          const contents = Array.isArray(a.content) ? a.content : [a.content]
+          const results: Array<{ fact_id: number; status: string; reason?: string; category?: string; contradicted_demoted?: number; warnings?: string[] }> = []
+
+          for (const content of contents) {
+            if (!content || !content.trim()) {
+              results.push({ fact_id: -1, status: 'error', reason: 'empty content' })
+              continue
+            }
+            const similar = store.findSimilarFact(content, category) ?? store.findSimilarFact(content)
+            let warnings: string[] | undefined
+            const scan = fullSecurityScan(content)
+            if (scan.warnings.length > 0 || scan.hasPii) warnings = [...scan.warnings]
+
+            if (similar) {
+              store.updateFact(similar.factId, { content, tags: a.tags, trustDelta: 0.05 })
+              const demoted = store.demoteContradictingFacts(similar.factId, content, category)
+              results.push({ fact_id: similar.factId, status: 'updated', reason: 'similar_fact_merged', ...(demoted > 0 ? { contradicted_demoted: demoted } : {}), ...(warnings ? { warnings } : {}) })
+            } else {
+              const factId = store.addFact(content, category, a.tags ?? '')
+              const demoted = store.demoteContradictingFacts(factId, content, category)
+              results.push({ fact_id: factId, status: 'added', category, ...(demoted > 0 ? { contradicted_demoted: demoted } : {}), ...(warnings ? { warnings } : {}) })
+            }
           }
 
-          const factId = store.addFact(a.content, category, a.tags ?? '')
-          const demoted = store.demoteContradictingFacts(factId, a.content, category)
-          return { content: [{ type: 'text' as const, text: JSON.stringify({ fact_id: factId, status: 'added', category, ...(demoted > 0 ? { contradicted_demoted: demoted } : {}), ...(warnings ? { warnings } : {}) }) }] }
+          retriever.getCache().clear()
+          resourceManager.invalidate()
+          const response = Array.isArray(a.content) ? results : results[0]
+          return { content: [{ type: 'text' as const, text: JSON.stringify(response) }] }
         }
 
         case 'search': {
@@ -124,14 +143,20 @@ server.tool(
 
         case 'update': {
           if (!a.fact_id) return { content: [{ type: 'text' as const, text: JSON.stringify({ error: 'Missing required argument: fact_id' }) }] }
-          const updated = store.updateFact(a.fact_id, { content: a.content, tags: a.tags, category, trustDelta: a.trust_delta })
+          const updated = store.updateFact(a.fact_id as number, { content: a.content as string | undefined, tags: a.tags, category, trustDelta: a.trust_delta })
+          retriever.getCache().clear()
+          resourceManager.invalidate()
           return { content: [{ type: 'text' as const, text: JSON.stringify({ updated }) }] }
         }
 
         case 'remove': {
           if (!a.fact_id) return { content: [{ type: 'text' as const, text: JSON.stringify({ error: 'Missing required argument: fact_id' }) }] }
-          const removed = store.removeFact(a.fact_id)
-          return { content: [{ type: 'text' as const, text: JSON.stringify({ removed }) }] }
+          const ids = Array.isArray(a.fact_id) ? a.fact_id : [a.fact_id]
+          const results = ids.map(id => ({ fact_id: id, removed: store.removeFact(id) }))
+          retriever.getCache().clear()
+          resourceManager.invalidate()
+          const response = Array.isArray(a.fact_id) ? results : results[0]
+          return { content: [{ type: 'text' as const, text: JSON.stringify(response) }] }
         }
 
         case 'list': {

package/src/types.ts

@@ -54,11 +54,11 @@ export interface RetrieverOptions {
 /** fact_store 工具调用参数 */
 export interface FactStoreArgs {
   action: 'add' | 'search' | 'probe' | 'related' | 'reason' | 'contradict' | 'update' | 'remove' | 'list'
-  content?: string
+  content?: string | string[]
   query?: string
   entity?: string
   entities?: string[]
-  fact_id?: number
+  fact_id?: number | number[]
   category?: string
   tags?: string
   trust_delta?: number

package/tests/refine.test.ts

@@ -0,0 +1,52 @@
+import { describe, it, expect } from 'vitest'
+import { refineQuery } from '../src/refine.js'
+
+describe('refineQuery', () => {
+  it('filters action words from Chinese query', () => {
+    const result = refineQuery('帮我用 TypeScript 重构 auth 模块')
+    expect(result).not.toBeNull()
+    expect(result!.query).toContain('TypeScript')
+    expect(result!.query).toContain('auth')
+    expect(result!.query).not.toContain('帮我')
+    expect(result!.query).not.toContain('重构')
+  })
+
+  it('returns null for pure operation commands', () => {
+    expect(refineQuery('运行测试')).toBeNull()
+    expect(refineQuery('git status')).toBeNull()
+    expect(refineQuery('创建文件')).toBeNull()
+  })
+
+  it('extracts quoted Chinese entities', () => {
+    const result = refineQuery('我喜欢「深色主题」')
+    expect(result).not.toBeNull()
+    expect(result!.entityTokens).toContain('深色主题')
+    expect(result!.query).toContain('深色主题')
+  })
+
+  it('extracts book title entities', () => {
+    const result = refineQuery('读了《设计模式》这本书')
+    expect(result).not.toBeNull()
+    expect(result!.entityTokens).toContain('设计模式')
+  })
+
+  it('extracts capitalized English phrases', () => {
+    const result = refineQuery('使用 Visual Studio Code 编辑器')
+    expect(result).not.toBeNull()
+    expect(result!.entityTokens).toContain('Visual Studio Code')
+  })
+
+  it('returns null for empty string', () => {
+    expect(refineQuery('')).toBeNull()
+    expect(refineQuery('   ')).toBeNull()
+  })
+
+  it('preserves meaningful Chinese tokens', () => {
+    const result = refineQuery('用户偏好深色主题')
+    expect(result).not.toBeNull()
+    expect(result!.query).toContain('用户')
+    expect(result!.query).toContain('偏好')
+    expect(result!.query).toContain('深色')
+    expect(result!.query).toContain('主题')
+  })
+})

package/tests/resource.test.ts

@@ -0,0 +1,62 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest'
+import { MemoryStore } from '../src/store.js'
+import { ResourceManager } from '../src/resources.js'
+import { mkdtempSync, rmSync } from 'node:fs'
+import { join } from 'node:path'
+import { tmpdir } from 'node:os'
+
+let store: MemoryStore
+let manager: ResourceManager
+let tmpDir: string
+
+beforeEach(() => {
+  tmpDir = mkdtempSync(join(tmpdir(), 'mnemo-test-'))
+  store = new MemoryStore(join(tmpDir, 'test.db'))
+  manager = new ResourceManager(store)
+})
+
+afterEach(() => {
+  store.close()
+  rmSync(tmpDir, { recursive: true, force: true })
+})
+
+describe('ResourceManager', () => {
+  it('returns empty array for empty category', () => {
+    const facts = manager.getFacts('identity')
+    expect(facts).toEqual([])
+  })
+
+  it('returns facts ordered by trust score', () => {
+    store.addFact('用户偏好深色主题', 'tool_pref')
+    store.addFact('用户喜欢 VS Code', 'tool_pref')
+    const facts = manager.getFacts('tool_pref')
+    expect(facts.length).toBe(2)
+    // listFacts returns trust_score DESC, first fact added gets default trust
+    expect(facts.length).toBeGreaterThan(0)
+  })
+
+  it('caches results', () => {
+    store.addFact('测试事实', 'general')
+    manager.getFacts('general')
+    expect(manager.cacheSize()).toBe(1)
+    // Second call should hit cache
+    const facts2 = manager.getFacts('general')
+    expect(facts2.length).toBe(1)
+  })
+
+  it('invalidates cache on write', () => {
+    store.addFact('测试事实', 'general')
+    manager.getFacts('general')
+    expect(manager.cacheSize()).toBe(1)
+    manager.invalidate()
+    expect(manager.cacheSize()).toBe(0)
+  })
+
+  it('limits to top 10 facts', () => {
+    for (let i = 0; i < 15; i++) {
+      store.addFact(`事实 ${i}`, 'general')
+    }
+    const facts = manager.getFacts('general')
+    expect(facts.length).toBe(10)
+  })
+})