axon-code 2.4.0

package/dist/memory/memory-sync.js

@@ -0,0 +1,429 @@
+/**
+ * 记忆增量同步引擎
+ * 扫描 memory、session summary、session transcript 文件，基于 hash 对比增量更新
+ */
+import * as fs from 'fs/promises';
+import * as path from 'path';
+import * as crypto from 'crypto';
+import { glob } from 'glob';
+/**
+ * 递归列出目录下的所有 .md 文件
+ */
+async function listMarkdownFiles(dir) {
+    try {
+        // 检查目录是否存在
+        const stats = await fs.stat(dir);
+        if (!stats.isDirectory()) {
+            return [];
+        }
+        // 使用 glob 查找所有 .md 文件
+        const pattern = path.join(dir, '**/*.md').replace(/\\/g, '/');
+        const files = await glob(pattern, {
+            nodir: true,
+            absolute: true,
+        });
+        return files;
+    }
+    catch (error) {
+        // 目录不存在或无权限，返回空数组
+        return [];
+    }
+}
+/**
+ * 计算文件 hash (SHA-256)
+ */
+async function computeFileHash(filePath) {
+    const content = await fs.readFile(filePath, 'utf-8');
+    return crypto.createHash('sha256').update(content).digest('hex');
+}
+/**
+ * 构建文件条目
+ */
+async function buildFileEntry(absPath, baseDir, source) {
+    const stats = await fs.stat(absPath);
+    const hash = await computeFileHash(absPath);
+    // 计算相对路径
+    let relativePath = path.relative(baseDir, absPath);
+    // 规范化路径（反斜杠转正斜杠）
+    relativePath = relativePath.replace(/\\/g, '/');
+    return {
+        path: relativePath,
+        absPath,
+        source,
+        hash,
+        mtime: stats.mtimeMs,
+        size: stats.size,
+    };
+}
+/**
+ * 记忆同步引擎
+ */
+export class MemorySyncEngine {
+    store;
+    projectDir;
+    constructor(store, opts) {
+        this.store = store;
+        this.projectDir = opts?.projectDir;
+    }
+    /**
+     * 同步 memory 目录下的 .md 文件
+     */
+    async syncMemoryFiles(memoryDir) {
+        const result = {
+            added: 0,
+            updated: 0,
+            removed: 0,
+            unchanged: 0,
+        };
+        try {
+            // 列出所有 .md 文件
+            const files = await listMarkdownFiles(memoryDir);
+            const processedPaths = new Set();
+            for (const absPath of files) {
+                try {
+                    // 跳过符号链接
+                    const stats = await fs.lstat(absPath);
+                    if (stats.isSymbolicLink()) {
+                        continue;
+                    }
+                    // 构建文件条目
+                    const entry = await buildFileEntry(absPath, memoryDir, 'memory');
+                    processedPaths.add(entry.path);
+                    // 检查是否需要更新
+                    const existingHash = this.store.getFileHash(entry.path);
+                    if (!existingHash) {
+                        // 新文件
+                        const content = await fs.readFile(absPath, 'utf-8');
+                        this.store.indexFile(entry, content);
+                        result.added++;
+                    }
+                    else if (existingHash !== entry.hash) {
+                        // 文件已修改
+                        const content = await fs.readFile(absPath, 'utf-8');
+                        this.store.indexFile(entry, content);
+                        result.updated++;
+                    }
+                    else {
+                        // 未修改
+                        result.unchanged++;
+                    }
+                }
+                catch (error) {
+                    // 单个文件处理失败不影响其他文件
+                    console.warn(`[MemorySync] Failed to process file ${absPath}:`, error);
+                }
+            }
+            // Delete indexed paths that no longer exist on disk (source: 'memory')
+            const indexedPaths = this.store.listFilePaths('memory');
+            for (const indexedPath of indexedPaths) {
+                if (!processedPaths.has(indexedPath)) {
+                    this.store.removeFile(indexedPath);
+                    result.removed++;
+                }
+            }
+        }
+        catch (error) {
+            console.error('[MemorySync] Failed to sync memory files:', error);
+        }
+        return result;
+    }
+    /**
+     * 同步 session 目录下的 summary.md 文件
+     */
+    async syncSessionFiles(sessionsDir) {
+        const result = {
+            added: 0,
+            updated: 0,
+            removed: 0,
+            unchanged: 0,
+        };
+        try {
+            // 查找所有 session-memory/summary.md 文件
+            const pattern = path.join(sessionsDir, '*/session-memory/summary.md').replace(/\\/g, '/');
+            const files = await glob(pattern, {
+                nodir: true,
+                absolute: true,
+            });
+            const processedPaths = new Set();
+            for (const absPath of files) {
+                try {
+                    // 跳过符号链接
+                    const stats = await fs.lstat(absPath);
+                    if (stats.isSymbolicLink()) {
+                        continue;
+                    }
+                    // 构建文件条目
+                    const entry = await buildFileEntry(absPath, sessionsDir, 'session');
+                    processedPaths.add(entry.path);
+                    // 检查是否需要更新
+                    const existingHash = this.store.getFileHash(entry.path);
+                    if (!existingHash) {
+                        // 新文件
+                        const content = await fs.readFile(absPath, 'utf-8');
+                        this.store.indexFile(entry, content);
+                        result.added++;
+                    }
+                    else if (existingHash !== entry.hash) {
+                        // 文件已修改
+                        const content = await fs.readFile(absPath, 'utf-8');
+                        this.store.indexFile(entry, content);
+                        result.updated++;
+                    }
+                    else {
+                        // 未修改
+                        result.unchanged++;
+                    }
+                }
+                catch (error) {
+                    console.warn(`[MemorySync] Failed to process session file ${absPath}:`, error);
+                }
+            }
+            // Delete indexed paths for 'session' source that no longer exist on disk
+            const indexedSessionPaths = this.store.listFilePaths('session');
+            for (const indexedPath of indexedSessionPaths) {
+                if (!processedPaths.has(indexedPath)) {
+                    this.store.removeFile(indexedPath);
+                    result.removed++;
+                }
+            }
+        }
+        catch (error) {
+            console.error('[MemorySync] Failed to sync session files:', error);
+        }
+        return result;
+    }
+    /**
+     * 同步 sessions 目录下的会话 transcript（.json 文件）
+     * 从 messages/chatHistory 中提取用户和助手的纯文本，转为 Markdown 索引
+     */
+    async syncTranscriptFiles(transcriptsDir) {
+        const result = {
+            added: 0,
+            updated: 0,
+            removed: 0,
+            unchanged: 0,
+        };
+        try {
+            const dirStat = await fs.stat(transcriptsDir).catch(() => null);
+            if (!dirStat?.isDirectory())
+                return result;
+            const entries = await fs.readdir(transcriptsDir, { withFileTypes: true });
+            const jsonFiles = entries.filter(e => e.isFile() && e.name.endsWith('.json'));
+            // 用 "transcript:" 前缀区分，避免和 summary.md 路径冲突
+            const processedPaths = new Set();
+            for (const dirent of jsonFiles) {
+                const absPath = path.join(transcriptsDir, dirent.name);
+                const indexPath = `transcript:${dirent.name}`;
+                processedPaths.add(indexPath);
+                try {
+                    const stat = await fs.stat(absPath);
+                    // 用 mtime+size 做轻量 hash，避免读取大文件
+                    const quickHash = crypto.createHash('sha256')
+                        .update(`${stat.mtimeMs}:${stat.size}`)
+                        .digest('hex');
+                    const existingHash = this.store.getFileHash(indexPath);
+                    if (existingHash === quickHash) {
+                        result.unchanged++;
+                        continue;
+                    }
+                    // 读取并解析 JSON
+                    const raw = await fs.readFile(absPath, 'utf-8');
+                    let data;
+                    try {
+                        data = JSON.parse(raw);
+                    }
+                    catch {
+                        continue; // 跳过损坏的 JSON
+                    }
+                    // 提取对话文本
+                    const markdown = this.extractTranscriptMarkdown(data);
+                    if (!markdown) {
+                        result.unchanged++;
+                        continue;
+                    }
+                    // 构建 FileEntry 并索引
+                    const entry = {
+                        path: indexPath,
+                        absPath,
+                        source: 'session',
+                        hash: quickHash,
+                        mtime: stat.mtimeMs,
+                        size: markdown.length,
+                    };
+                    this.store.indexFile(entry, markdown);
+                    result[existingHash ? 'updated' : 'added']++;
+                }
+                catch (error) {
+                    console.warn(`[MemorySync] Failed to process transcript ${dirent.name}:`, error);
+                }
+            }
+            // 清理已删除的 transcript 索引
+            const indexedPaths = this.store.listFilePaths('session');
+            for (const p of indexedPaths) {
+                if (p.startsWith('transcript:') && !processedPaths.has(p)) {
+                    this.store.removeFile(p);
+                    result.removed++;
+                }
+            }
+        }
+        catch (error) {
+            console.error('[MemorySync] Failed to sync transcript files:', error);
+        }
+        return result;
+    }
+    /**
+     * 从会话 JSON 中提取对话文本，转为 Markdown 格式
+     * 只提取 user/assistant 的纯文本内容，跳过 tool_use/tool_result 噪音
+     */
+    extractTranscriptMarkdown(data) {
+        const metadata = data?.metadata;
+        const messages = data?.messages || [];
+        const chatHistory = data?.chatHistory || [];
+        // 优先用 chatHistory（已经是面向展示的格式），fallback 到 messages
+        const source = chatHistory.length > 0 ? chatHistory : messages;
+        if (source.length === 0)
+            return null;
+        const lines = [];
+        // 标题
+        const name = metadata?.name || metadata?.id || 'Untitled';
+        const date = metadata?.createdAt
+            ? new Date(metadata.createdAt).toISOString().split('T')[0]
+            : '';
+        lines.push(`# ${name}`);
+        if (date)
+            lines.push(`Date: ${date}`);
+        if (metadata?.model)
+            lines.push(`Model: ${metadata.model}`);
+        if (metadata?.workingDirectory)
+            lines.push(`Project: ${metadata.workingDirectory}`);
+        lines.push('');
+        // 提取对话（限制总长度，避免超大会话占太多索引空间）
+        const MAX_CHARS = 20000;
+        let totalChars = 0;
+        for (const msg of source) {
+            if (totalChars >= MAX_CHARS) {
+                lines.push('\n[...truncated...]');
+                break;
+            }
+            const role = msg.role;
+            if (role !== 'user' && role !== 'assistant')
+                continue;
+            const texts = this.extractTextFromContent(msg.content);
+            if (!texts)
+                continue;
+            const prefix = role === 'user' ? '## User' : '## Assistant';
+            lines.push(prefix);
+            const text = texts.substring(0, MAX_CHARS - totalChars);
+            lines.push(text);
+            lines.push('');
+            totalChars += text.length;
+        }
+        // 至少要有一条消息才有索引价值
+        if (totalChars === 0)
+            return null;
+        return lines.join('\n');
+    }
+    /**
+     * 从 message.content 中提取纯文本
+     */
+    extractTextFromContent(content) {
+        if (!content)
+            return null;
+        if (typeof content === 'string')
+            return content;
+        if (Array.isArray(content)) {
+            const texts = [];
+            for (const block of content) {
+                if (block.type === 'text' && block.text) {
+                    texts.push(block.text);
+                }
+                // 跳过 tool_use, tool_result, image 等
+            }
+            return texts.length > 0 ? texts.join('\n') : null;
+        }
+        return null;
+    }
+    /**
+     * 同步所有文件
+     */
+    async syncAll(opts) {
+        const memoryDir = opts?.memoryDir;
+        const sessionsDir = opts?.sessionsDir;
+        const transcriptsDir = opts?.transcriptsDir;
+        const memoryResult = memoryDir
+            ? await this.syncMemoryFiles(memoryDir)
+            : { added: 0, updated: 0, removed: 0, unchanged: 0 };
+        const sessionsResult = sessionsDir
+            ? await this.syncSessionFiles(sessionsDir)
+            : { added: 0, updated: 0, removed: 0, unchanged: 0 };
+        const transcriptsResult = transcriptsDir
+            ? await this.syncTranscriptFiles(transcriptsDir)
+            : { added: 0, updated: 0, removed: 0, unchanged: 0 };
+        return {
+            memory: memoryResult,
+            sessions: sessionsResult,
+            transcripts: transcriptsResult,
+        };
+    }
+    /**
+     * 批量补齐缺失的 embedding（在 sync 完成后异步调用）
+     * 每次最多处理 batchSize 个 chunk，避免单次 API 调用量过大
+     */
+    async backfillEmbeddings(provider, cache, batchSize = 100) {
+        const chunks = this.store.getChunksWithoutEmbedding(batchSize);
+        if (chunks.length === 0)
+            return 0;
+        const model = provider.model;
+        const texts = chunks.map(c => c.text);
+        // 先查缓存
+        let embeddings;
+        if (cache) {
+            embeddings = cache.getBatch(model, texts);
+        }
+        else {
+            embeddings = texts.map(() => null);
+        }
+        // 收集未命中缓存的 indices
+        const missIndices = [];
+        for (let i = 0; i < embeddings.length; i++) {
+            if (!embeddings[i])
+                missIndices.push(i);
+        }
+        // 批量调用 API
+        if (missIndices.length > 0) {
+            const missTexts = missIndices.map(i => texts[i]);
+            try {
+                const apiResults = await provider.embedBatch(missTexts);
+                for (let j = 0; j < missIndices.length; j++) {
+                    const idx = missIndices[j];
+                    embeddings[idx] = apiResults[j];
+                }
+                // 写入缓存
+                if (cache) {
+                    const cacheEntries = missIndices.map((idx, j) => ({
+                        text: texts[idx],
+                        embedding: apiResults[j],
+                    }));
+                    cache.setBatch(model, cacheEntries);
+                }
+            }
+            catch (e) {
+                console.warn('[MemorySync] Embedding API failed:', e);
+                return 0;
+            }
+        }
+        // 写入数据库
+        const updates = [];
+        for (let i = 0; i < chunks.length; i++) {
+            const emb = embeddings[i];
+            if (emb) {
+                updates.push({ id: chunks[i].id, embedding: emb });
+            }
+        }
+        if (updates.length > 0) {
+            this.store.updateEmbeddings(updates);
+        }
+        return updates.length;
+    }
+}
+//# sourceMappingURL=memory-sync.js.map

package/dist/memory/memory-sync.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-sync.js","sourceRoot":"","sources":["../../src/memory/memory-sync.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAElC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAgB5B;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAAC,GAAW;IAC1C,IAAI,CAAC;QACH,WAAW;QACX,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,sBAAsB;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE;YAChC,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,IAAI;SACf,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,kBAAkB;QAClB,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,eAAe,CAAC,QAAgB;IAC7C,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACrD,OAAO,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACnE,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,cAAc,CAC3B,OAAe,EACf,OAAe,EACf,MAAoB;IAEpB,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,CAAC;IAE5C,SAAS;IACT,IAAI,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACnD,iBAAiB;IACjB,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAEhD,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,OAAO;QACP,MAAM;QACN,IAAI;QACJ,KAAK,EAAE,KAAK,CAAC,OAAO;QACpB,IAAI,EAAE,KAAK,CAAC,IAAI;KACjB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,gBAAgB;IACnB,KAAK,CAAgB;IACrB,UAAU,CAAU;IAE5B,YAAY,KAAoB,EAAE,IAA8B;QAC9D,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,IAAI,EAAE,UAAU,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,SAAiB;QACrC,MAAM,MAAM,GAAe;YACzB,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC;SACb,CAAC;QAEF,IAAI,CAAC;YACH,cAAc;YACd,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;YAEzC,KAAK,MAAM,OAAO,IAAI,KAAK,EAAE,CAAC;gBAC5B,IAAI,CAAC;oBACH,SAAS;oBACT,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACtC,IAAI,KAAK,CAAC,cAAc,EAAE,EAAE,CAAC;wBAC3B,SAAS;oBACX,CAAC;oBAED,SAAS;oBACT,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;oBACjE,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAE/B,WAAW;oBACX,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAExD,IAAI,CAAC,YAAY,EAAE,CAAC;wBAClB,MAAM;wBACN,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBACpD,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;wBACrC,MAAM,CAAC,KAAK,EAAE,CAAC;oBACjB,CAAC;yBAAM,IAAI,YAAY,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC;wBACvC,QAAQ;wBACR,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBACpD,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;wBACrC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,CAAC;yBAAM,CAAC;wBACN,MAAM;wBACN,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,kBAAkB;oBAClB,OAAO,CAAC,IAAI,CAAC,uCAAuC,OAAO,GAAG,EAAE,KAAK,CAAC,CAAC;gBACzE,CAAC;YACH,CAAC;YAED,uEAAuE;YACvE,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;YACxD,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;gBACvC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;oBACrC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;oBACnC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;QAEH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2CAA2C,EAAE,KAAK,CAAC,CAAC;QACpE,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,WAAmB;QACxC,MAAM,MAAM,GAAe;YACzB,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC;SACb,CAAC;QAEF,IAAI,CAAC;YACH,oCAAoC;YACpC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,6BAA6B,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAC1F,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE;gBAChC,KAAK,EAAE,IAAI;gBACX,QAAQ,EAAE,IAAI;aACf,CAAC,CAAC;YACH,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;YAGzC,KAAK,MAAM,OAAO,IAAI,KAAK,EAAE,CAAC;gBAC5B,IAAI,CAAC;oBACH,SAAS;oBACT,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACtC,IAAI,KAAK,CAAC,cAAc,EAAE,EAAE,CAAC;wBAC3B,SAAS;oBACX,CAAC;oBAED,SAAS;oBACT,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;oBACpE,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAE/B,WAAW;oBACX,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAExD,IAAI,CAAC,YAAY,EAAE,CAAC;wBAClB,MAAM;wBACN,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBACpD,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;wBACrC,MAAM,CAAC,KAAK,EAAE,CAAC;oBACjB,CAAC;yBAAM,IAAI,YAAY,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC;wBACvC,QAAQ;wBACR,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBACpD,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;wBACrC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,CAAC;yBAAM,CAAC;wBACN,MAAM;wBACN,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,IAAI,CAAC,+CAA+C,OAAO,GAAG,EAAE,KAAK,CAAC,CAAC;gBACjF,CAAC;YACH,CAAC;YACD,yEAAyE;YACzE,MAAM,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAChE,KAAK,MAAM,WAAW,IAAI,mBAAmB,EAAE,CAAC;gBAC9C,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;oBACrC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;oBACnC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;QAGH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,4CAA4C,EAAE,KAAK,CAAC,CAAC;QACrE,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,mBAAmB,CAAC,cAAsB;QAC9C,MAAM,MAAM,GAAe;YACzB,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC;SACb,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YAChE,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE;gBAAE,OAAO,MAAM,CAAC;YAE3C,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1E,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YAE9E,2CAA2C;YAC3C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;YAEzC,KAAK,MAAM,MAAM,IAAI,SAAS,EAAE,CAAC;gBAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;gBACvD,MAAM,SAAS,GAAG,cAAc,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC9C,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAE9B,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACpC,gCAAgC;oBAChC,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC;yBAC1C,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;yBACtC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAEjB,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;oBACvD,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;wBAC/B,MAAM,CAAC,SAAS,EAAE,CAAC;wBACnB,SAAS;oBACX,CAAC;oBAED,aAAa;oBACb,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBAChD,IAAI,IAAS,CAAC;oBACd,IAAI,CAAC;wBACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBACzB,CAAC;oBAAC,MAAM,CAAC;wBACP,SAAS,CAAC,aAAa;oBACzB,CAAC;oBAED,SAAS;oBACT,MAAM,QAAQ,GAAG,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;oBACtD,IAAI,CAAC,QAAQ,EAAE,CAAC;wBACd,MAAM,CAAC,SAAS,EAAE,CAAC;wBACnB,SAAS;oBACX,CAAC;oBAED,mBAAmB;oBACnB,MAAM,KAAK,GAAc;wBACvB,IAAI,EAAE,SAAS;wBACf,OAAO;wBACP,MAAM,EAAE,SAAS;wBACjB,IAAI,EAAE,SAAS;wBACf,KAAK,EAAE,IAAI,CAAC,OAAO;wBACnB,IAAI,EAAE,QAAQ,CAAC,MAAM;qBACtB,CAAC;oBAEF,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;oBACtC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC/C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,IAAI,CAAC,6CAA6C,MAAM,CAAC,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;gBACnF,CAAC;YACH,CAAC;YAED,uBAAuB;YACvB,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YACzD,KAAK,MAAM,CAAC,IAAI,YAAY,EAAE,CAAC;gBAC7B,IAAI,CAAC,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC1D,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;oBACzB,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+CAA+C,EAAE,KAAK,CAAC,CAAC;QACxE,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACK,yBAAyB,CAAC,IAAS;QACzC,MAAM,QAAQ,GAAG,IAAI,EAAE,QAAQ,CAAC;QAChC,MAAM,QAAQ,GAAU,IAAI,EAAE,QAAQ,IAAI,EAAE,CAAC;QAC7C,MAAM,WAAW,GAAU,IAAI,EAAE,WAAW,IAAI,EAAE,CAAC;QAEnD,kDAAkD;QAClD,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC/D,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAErC,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK;QACL,MAAM,IAAI,GAAG,QAAQ,EAAE,IAAI,IAAI,QAAQ,EAAE,EAAE,IAAI,UAAU,CAAC;QAC1D,MAAM,IAAI,GAAG,QAAQ,EAAE,SAAS;YAC9B,CAAC,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC,CAAC,EAAE,CAAC;QACP,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QACxB,IAAI,IAAI;YAAE,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;QACtC,IAAI,QAAQ,EAAE,KAAK;YAAE,KAAK,CAAC,IAAI,CAAC,UAAU,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;QAC5D,IAAI,QAAQ,EAAE,gBAAgB;YAAE,KAAK,CAAC,IAAI,CAAC,YAAY,QAAQ,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACpF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,4BAA4B;QAC5B,MAAM,SAAS,GAAG,KAAK,CAAC;QACxB,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,UAAU,IAAI,SAAS,EAAE,CAAC;gBAC5B,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;gBAClC,MAAM;YACR,CAAC;YAED,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;YACtB,IAAI,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,WAAW;gBAAE,SAAS;YAEtD,MAAM,KAAK,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACvD,IAAI,CAAC,KAAK;gBAAE,SAAS;YAErB,MAAM,MAAM,GAAG,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC;YAC5D,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnB,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC,CAAC;YACxD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC;QAC5B,CAAC;QAED,iBAAiB;QACjB,IAAI,UAAU,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAElC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,OAAY;QACzC,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAE1B,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,OAAO,OAAO,CAAC;QAEhD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;oBACxC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACzB,CAAC;gBACD,oCAAoC;YACtC,CAAC;YACD,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACpD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,IAIb;QAKC,MAAM,SAAS,GAAG,IAAI,EAAE,SAAS,CAAC;QAClC,MAAM,WAAW,GAAG,IAAI,EAAE,WAAW,CAAC;QACtC,MAAM,cAAc,GAAG,IAAI,EAAE,cAAc,CAAC;QAE5C,MAAM,YAAY,GAAe,SAAS;YACxC,CAAC,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC;YACvC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;QAEvD,MAAM,cAAc,GAAe,WAAW;YAC5C,CAAC,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC;YAC1C,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;QAEvD,MAAM,iBAAiB,GAAe,cAAc;YAClD,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC;YAChD,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;QAEvD,OAAO;YACL,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE,cAAc;YACxB,WAAW,EAAE,iBAAiB;SAC/B,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,kBAAkB,CACtB,QAA2B,EAC3B,KAA6B,EAC7B,YAAoB,GAAG;QAEvB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC;QAC/D,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAElC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEtC,OAAO;QACP,IAAI,UAA+B,CAAC;QACpC,IAAI,KAAK,EAAE,CAAC;YACV,UAAU,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,UAAU,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;QAED,mBAAmB;QACnB,MAAM,WAAW,GAAa,EAAE,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;gBAAE,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC;QAED,WAAW;QACX,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACjD,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;gBACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC5C,MAAM,GAAG,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;oBAC3B,UAAU,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;gBAClC,CAAC;gBAED,OAAO;gBACP,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;wBAChD,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC;wBAChB,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC;qBACzB,CAAC,CAAC,CAAC;oBACJ,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;gBACtC,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;gBACtD,OAAO,CAAC,CAAC;YACX,CAAC;QACH,CAAC;QAED,QAAQ;QACR,MAAM,OAAO,GAA+C,EAAE,CAAC;QAC/D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,GAAG,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,GAAG,EAAE,CAAC;gBACR,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QAED,OAAO,OAAO,CAAC,MAAM,CAAC;IACxB,CAAC;CACF"}

package/dist/memory/mmr.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Maximal Marginal Relevance (MMR) 多样性重排算法
+ *
+ * MMR 平衡相关性与多样性，迭代选择最大化目标函数的结果：
+ *   MMR = λ * relevance - (1-λ) * max_similarity_to_selected
+ *
+ * 参考: Carbonell & Goldstein (1998)
+ * 移植自 moltbot/src/memory/mmr.ts，简化版
+ */
+export interface MMRItem {
+    id: string;
+    score: number;
+    content: string;
+}
+export interface MMRConfig {
+    enabled: boolean;
+    /** Lambda: 0=最大多样性, 1=最大相关性。默认 0.7 */
+    lambda: number;
+}
+export declare const DEFAULT_MMR_CONFIG: MMRConfig;
+/**
+ * MMR 重排
+ */
+export declare function mmrRerank<T extends MMRItem>(items: T[], config?: Partial<MMRConfig>): T[];
+/**
+ * 将混合搜索结果适配为 MMR 格式并重排
+ */
+export declare function applyMMRToResults<T extends {
+    score: number;
+    text: string;
+    path: string;
+    startLine: number;
+}>(results: T[], config?: Partial<MMRConfig>): T[];
+//# sourceMappingURL=mmr.d.ts.map

package/dist/memory/mmr.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mmr.d.ts","sourceRoot":"","sources":["../../src/memory/mmr.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,OAAO,CAAC;IACjB,sCAAsC;IACtC,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,eAAO,MAAM,kBAAkB,EAAE,SAGhC,CAAC;AA6BF;;GAEG;AACH,wBAAgB,SAAS,CAAC,CAAC,SAAS,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,MAAM,GAAE,OAAO,CAAC,SAAS,CAAM,GAAG,CAAC,EAAE,CAiD7F;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,SAAS;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,EAC1G,OAAO,EAAE,CAAC,EAAE,EACZ,MAAM,GAAE,OAAO,CAAC,SAAS,CAAM,GAC9B,CAAC,EAAE,CAYL"}

package/dist/memory/mmr.js

@@ -0,0 +1,101 @@
+/**
+ * Maximal Marginal Relevance (MMR) 多样性重排算法
+ *
+ * MMR 平衡相关性与多样性，迭代选择最大化目标函数的结果：
+ *   MMR = λ * relevance - (1-λ) * max_similarity_to_selected
+ *
+ * 参考: Carbonell & Goldstein (1998)
+ * 移植自 moltbot/src/memory/mmr.ts，简化版
+ */
+export const DEFAULT_MMR_CONFIG = {
+    enabled: false,
+    lambda: 0.7,
+};
+/**
+ * 分词（用于 Jaccard 相似度计算）
+ */
+function tokenize(text) {
+    const tokens = text.toLowerCase().match(/[a-z0-9\u4e00-\u9fff]+/g) ?? [];
+    return new Set(tokens);
+}
+/**
+ * Jaccard 相似度
+ */
+function jaccardSimilarity(setA, setB) {
+    if (setA.size === 0 && setB.size === 0)
+        return 1;
+    if (setA.size === 0 || setB.size === 0)
+        return 0;
+    let intersection = 0;
+    const smaller = setA.size <= setB.size ? setA : setB;
+    const larger = setA.size <= setB.size ? setB : setA;
+    for (const token of smaller) {
+        if (larger.has(token))
+            intersection++;
+    }
+    const union = setA.size + setB.size - intersection;
+    return union === 0 ? 0 : intersection / union;
+}
+/**
+ * MMR 重排
+ */
+export function mmrRerank(items, config = {}) {
+    const lambda = Math.max(0, Math.min(1, config.lambda ?? DEFAULT_MMR_CONFIG.lambda));
+    if (!config.enabled || items.length <= 1)
+        return [...items];
+    if (lambda === 1)
+        return [...items].sort((a, b) => b.score - a.score);
+    // 预计算 token 集合
+    const tokenCache = new Map();
+    for (const item of items) {
+        tokenCache.set(item.id, tokenize(item.content));
+    }
+    // 归一化分数到 [0, 1]
+    const maxScore = Math.max(...items.map(i => i.score));
+    const minScore = Math.min(...items.map(i => i.score));
+    const range = maxScore - minScore;
+    const normalize = (s) => range === 0 ? 1 : (s - minScore) / range;
+    const selected = [];
+    const remaining = new Set(items);
+    while (remaining.size > 0) {
+        let best = null;
+        let bestMMR = -Infinity;
+        for (const candidate of remaining) {
+            const relevance = normalize(candidate.score);
+            // 计算与已选择项的最大相似度
+            let maxSim = 0;
+            const candidateTokens = tokenCache.get(candidate.id);
+            for (const sel of selected) {
+                const sim = jaccardSimilarity(candidateTokens, tokenCache.get(sel.id));
+                if (sim > maxSim)
+                    maxSim = sim;
+            }
+            const mmrScore = lambda * relevance - (1 - lambda) * maxSim;
+            if (mmrScore > bestMMR || (mmrScore === bestMMR && candidate.score > (best?.score ?? -Infinity))) {
+                bestMMR = mmrScore;
+                best = candidate;
+            }
+        }
+        if (!best)
+            break;
+        selected.push(best);
+        remaining.delete(best);
+    }
+    return selected;
+}
+/**
+ * 将混合搜索结果适配为 MMR 格式并重排
+ */
+export function applyMMRToResults(results, config = {}) {
+    if (results.length === 0 || !config.enabled)
+        return results;
+    const itemMap = new Map();
+    const mmrItems = results.map((r, i) => {
+        const id = `${r.path}:${r.startLine}:${i}`;
+        itemMap.set(id, r);
+        return { id, score: r.score, content: r.text };
+    });
+    const reranked = mmrRerank(mmrItems, config);
+    return reranked.map(item => itemMap.get(item.id));
+}
+//# sourceMappingURL=mmr.js.map

package/dist/memory/mmr.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mmr.js","sourceRoot":"","sources":["../../src/memory/mmr.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAcH,MAAM,CAAC,MAAM,kBAAkB,GAAc;IAC3C,OAAO,EAAE,KAAK;IACd,MAAM,EAAE,GAAG;CACZ,CAAC;AAEF;;GAEG;AACH,SAAS,QAAQ,CAAC,IAAY;IAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,yBAAyB,CAAC,IAAI,EAAE,CAAC;IACzE,OAAO,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;AACzB,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAiB,EAAE,IAAiB;IAC7D,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACjD,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAEjD,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IACrD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAEpD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,YAAY,EAAE,CAAC;IACxC,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;IACnD,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,KAAK,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,SAAS,CAAoB,KAAU,EAAE,SAA6B,EAAE;IACtF,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC;IAEpF,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;IAC5D,IAAI,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IAEtE,eAAe;IACf,MAAM,UAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IAClD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAClD,CAAC;IAED,gBAAgB;IAChB,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,QAAQ,GAAG,QAAQ,CAAC;IAClC,MAAM,SAAS,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,GAAG,KAAK,CAAC;IAE1E,MAAM,QAAQ,GAAQ,EAAE,CAAC;IACzB,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;IAEjC,OAAO,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QAC1B,IAAI,IAAI,GAAa,IAAI,CAAC;QAC1B,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC;QAExB,KAAK,MAAM,SAAS,IAAI,SAAS,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAE7C,gBAAgB;YAChB,IAAI,MAAM,GAAG,CAAC,CAAC;YACf,MAAM,eAAe,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAE,CAAC;YACtD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,MAAM,GAAG,GAAG,iBAAiB,CAAC,eAAe,EAAE,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC,CAAC;gBACxE,IAAI,GAAG,GAAG,MAAM;oBAAE,MAAM,GAAG,GAAG,CAAC;YACjC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,GAAG,SAAS,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,GAAG,MAAM,CAAC;YAC5D,IAAI,QAAQ,GAAG,OAAO,IAAI,CAAC,QAAQ,KAAK,OAAO,IAAI,SAAS,CAAC,KAAK,GAAG,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;gBACjG,OAAO,GAAG,QAAQ,CAAC;gBACnB,IAAI,GAAG,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QAED,IAAI,CAAC,IAAI;YAAE,MAAM;QACjB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAC/B,OAAY,EACZ,SAA6B,EAAE;IAE/B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO;QAAE,OAAO,OAAO,CAAC;IAE5D,MAAM,OAAO,GAAG,IAAI,GAAG,EAAa,CAAC;IACrC,MAAM,QAAQ,GAAc,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QAC/C,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,EAAE,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACnB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC7C,OAAO,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC,CAAC;AACrD,CAAC"}