kimi-code-memory-mcp-server 0.1.1 → 0.1.2

package/dist/refine/store.js

@@ -0,0 +1,139 @@
+/**
+ * SQLite storage for refined turns.
+ */
+import fs from 'fs';
+import path from 'path';
+import Database from 'better-sqlite3';
+import { turnToRow } from './adapter.js';
+import { rowToTurn } from './adapter.js';
+import { LIMITS } from './constants.js';
+import { buildLikeConditions, scoreText } from '../utils/search.js';
+import { dateRangeSql } from '../utils/date.js';
+export class RefinedStore {
+    refinedRoot;
+    dbPath;
+    db;
+    constructor(refinedRoot) {
+        this.refinedRoot = refinedRoot;
+        this.dbPath = path.join(refinedRoot, 'refined.sqlite');
+        fs.mkdirSync(refinedRoot, { recursive: true });
+        this.db = new Database(this.dbPath);
+        this.initDb();
+    }
+    initDb() {
+        this.db.exec(`
+      CREATE TABLE IF NOT EXISTS refined_turns (
+        session_id TEXT NOT NULL,
+        turn_id INTEGER NOT NULL,
+        timestamp TEXT,
+        summary TEXT,
+        facts TEXT,
+        notes TEXT,
+        entities TEXT,
+        categories TEXT,
+        PRIMARY KEY (session_id, turn_id)
+      );
+      CREATE INDEX IF NOT EXISTS idx_refined_session ON refined_turns(session_id);
+      CREATE INDEX IF NOT EXISTS idx_refined_timestamp ON refined_turns(timestamp);
+    `);
+    }
+    getDbPath() {
+        return this.dbPath;
+    }
+    saveRefinedTurns(sessionId, refinedTurns) {
+        const insert = this.db.prepare(`INSERT OR REPLACE INTO refined_turns
+       (session_id, turn_id, timestamp, summary, facts, notes, entities, categories)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?)`);
+        const existing = this.loadRefinedTurns(sessionId);
+        const merged = new Map(existing.map((t) => [t.turnId, t]));
+        for (const turn of refinedTurns) {
+            merged.set(turn.turnId, turn);
+        }
+        const transaction = this.db.transaction(() => {
+            for (const turn of merged.values()) {
+                const row = turnToRow(turn);
+                insert.run(row.session_id, row.turn_id, row.timestamp, row.summary, row.facts, row.notes, row.entities, row.categories);
+            }
+        });
+        transaction();
+    }
+    loadRefinedTurns(sessionId) {
+        const stmt = this.db.prepare('SELECT * FROM refined_turns WHERE session_id = ? ORDER BY turn_id');
+        const rows = stmt.all(sessionId);
+        return rows.map((r) => rowToTurn(r));
+    }
+    loadRefinedTurn(sessionId, turnId) {
+        const stmt = this.db.prepare('SELECT * FROM refined_turns WHERE session_id = ? AND turn_id = ?');
+        const row = stmt.get(sessionId, turnId);
+        return row ? rowToTurn(row) : undefined;
+    }
+    searchRefinedTurns(options) {
+        const rawQuery = typeof options.query === 'string' ? options.query.trim() : '';
+        if (!rawQuery)
+            return [];
+        const terms = rawQuery
+            .toLowerCase()
+            .split(/\s+/)
+            .filter((t) => t.length > 0);
+        if (terms.length === 0)
+            return [];
+        const limit = typeof options.limit === 'number' ? Math.max(1, Math.floor(options.limit)) : LIMITS.searchResultLimit;
+        const dateFrom = options.dateFrom || null;
+        const dateTo = options.dateTo || null;
+        const conditions = buildLikeConditions(terms);
+        const params = [];
+        for (const term of terms) {
+            const escaped = term.split('%').join('\\%').split('_').join('\\_');
+            const like = `%${escaped}%`;
+            params.push(like, like, like);
+        }
+        const dateRange = dateRangeSql(dateFrom || undefined, dateTo || undefined);
+        if (dateRange.from) {
+            conditions.push('timestamp >= ?');
+            params.push(dateRange.from);
+        }
+        if (dateRange.to) {
+            conditions.push('timestamp <= ?');
+            params.push(dateRange.to);
+        }
+        const sql = `SELECT session_id, turn_id, timestamp, summary, facts, notes
+                 FROM refined_turns
+                 WHERE ${conditions.join(' AND ')}
+                 ORDER BY timestamp DESC
+                 LIMIT ?`;
+        params.push(limit * LIMITS.searchCandidateMultiplier);
+        const stmt = this.db.prepare(sql);
+        const rows = stmt.all(...params);
+        const matches = [];
+        for (const row of rows) {
+            const haystack = `${row.summary}\n${row.facts}\n${row.notes}`;
+            const score = scoreText(haystack, terms);
+            if (score === 0)
+                continue;
+            matches.push({
+                sessionId: row.session_id,
+                turnId: row.turn_id,
+                timestamp: row.timestamp ?? undefined,
+                summary: row.summary,
+                facts: JSON.parse(row.facts || '[]'),
+                notes: JSON.parse(row.notes || '[]'),
+                score,
+            });
+        }
+        matches.sort((a, b) => b.score - a.score);
+        return matches.slice(0, limit);
+    }
+    countAll() {
+        const row = this.db.prepare('SELECT COUNT(*) AS c FROM refined_turns').get();
+        return row?.c ?? 0;
+    }
+    listRecentTurns(limit) {
+        const stmt = this.db.prepare('SELECT * FROM refined_turns ORDER BY timestamp DESC LIMIT ?');
+        const rows = stmt.all(limit);
+        return rows.map((r) => rowToTurn(r));
+    }
+    close() {
+        this.db.close();
+    }
+}
+//# sourceMappingURL=store.js.map

package/dist/refine/store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../src/refine/store.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAEtC,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAE,mBAAmB,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpE,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAEhD,MAAM,OAAO,YAAY;IACvB,WAAW,CAAS;IACZ,MAAM,CAAS;IACf,EAAE,CAAoB;IAE9B,YAAY,WAAmB;QAC7B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC;QACvD,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAEO,MAAM;QACZ,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;KAcZ,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,gBAAgB,CAAC,SAAiB,EAAE,YAA2B;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC5B;;uCAEiC,CAClC,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;YAChC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAC3C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;gBACnC,MAAM,GAAG,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;gBAC5B,MAAM,CAAC,GAAG,CACR,GAAG,CAAC,UAAU,EACd,GAAG,CAAC,OAAO,EACX,GAAG,CAAC,SAAS,EACb,GAAG,CAAC,OAAO,EACX,GAAG,CAAC,KAAK,EACT,GAAG,CAAC,KAAK,EACT,GAAG,CAAC,QAAQ,EACZ,GAAG,CAAC,UAAU,CACf,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QACH,WAAW,EAAE,CAAC;IAChB,CAAC;IAED,gBAAgB,CAAC,SAAiB;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,mEAAmE,CACpE,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAS7B,CAAC;QACH,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,eAAe,CAAC,SAAiB,EAAE,MAAc;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,kEAAkE,CACnE,CAAC;QACF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAWzB,CAAC;QACd,OAAO,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1C,CAAC;IAED,kBAAkB,CAAC,OAA6B;QAC9C,MAAM,QAAQ,GAAG,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,IAAI,CAAC,QAAQ;YAAE,OAAO,EAAE,CAAC;QAEzB,MAAM,KAAK,GAAG,QAAQ;aACnB,WAAW,EAAE;aACb,KAAK,CAAC,KAAK,CAAC;aACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC/B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,MAAM,KAAK,GACT,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC;QACxG,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;QAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC;QAEtC,MAAM,UAAU,GAAa,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACxD,MAAM,MAAM,GAAwB,EAAE,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnE,MAAM,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,IAAI,SAAS,EAAE,MAAM,IAAI,SAAS,CAAC,CAAC;QAC3E,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC;YACnB,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAClC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;QACD,IAAI,SAAS,CAAC,EAAE,EAAE,CAAC;YACjB,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAClC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,GAAG,GAAG;;yBAES,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC;;yBAExB,CAAC;QACtB,MAAM,CAAC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,yBAAyB,CAAC,CAAC;QAEtD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAO7B,CAAC;QAEH,MAAM,OAAO,GAAyB,EAAE,CAAC;QACzC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,GAAG,GAAG,CAAC,OAAO,KAAK,GAAG,CAAC,KAAK,KAAK,GAAG,CAAC,KAAK,EAAE,CAAC;YAC9D,MAAM,KAAK,GAAG,SAAS,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YACzC,IAAI,KAAK,KAAK,CAAC;gBAAE,SAAS;YAE1B,OAAO,CAAC,IAAI,CAAC;gBACX,SAAS,EAAE,GAAG,CAAC,UAAU;gBACzB,MAAM,EAAE,GAAG,CAAC,OAAO;gBACnB,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,SAAS;gBACrC,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC;gBACpC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC;gBACpC,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAC1C,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,QAAQ;QACN,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,EAAmB,CAAC;QAC9F,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IAED,eAAe,CAAC,KAAa;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,6DAA6D,CAC9D,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CASzB,CAAC;QACH,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACF"}

package/dist/refine/types.d.ts

@@ -0,0 +1,56 @@
+/**
+ * Domain types for the turn refinement layer.
+ */
+export interface RawAction {
+    name?: string;
+    args?: unknown;
+    result?: unknown;
+}
+export interface RawTurn {
+    turnId?: string | number;
+    timestamp?: string;
+    user?: string;
+    agentText?: string;
+    agent?: string;
+    actions?: RawAction[];
+}
+export interface RefinedTurn {
+    sessionId: string;
+    turnId: number;
+    timestamp: string | undefined;
+    summary: string;
+    facts: string[];
+    notes: string[];
+    entities: {
+        files: string[];
+        tools: string[];
+        errors: string[];
+    };
+    categories: Record<string, string[]>;
+}
+export interface RefinedSearchOptions {
+    query: string;
+    dateFrom?: string;
+    dateTo?: string;
+    limit?: number;
+}
+export interface RefinedSearchMatch {
+    sessionId: string;
+    turnId: number;
+    timestamp: string | undefined;
+    summary: string;
+    facts: string[];
+    notes: string[];
+    score: number;
+}
+/** SQLite row shape for refined_turns. */
+export interface RefinedRow {
+    session_id: string;
+    turn_id: number;
+    timestamp: string | null;
+    summary: string;
+    facts: string;
+    notes: string;
+    entities: string;
+    categories: string;
+}

package/dist/refine/types.js

@@ -0,0 +1,5 @@
+/**
+ * Domain types for the turn refinement layer.
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/refine/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/refine/types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/refined-manager.d.ts

@@ -1,70 +1,24 @@
 /**
- * Refined turn storage and extraction backed by SQLite.
+ * Refined turn orchestrator.
  *
- * Refined turns are a local derived index / cache of conversation wires.
- * They are not user-facing memory assets; user memory stays in Markdown files.
- *
- * Extraction rules are intentionally local and deterministic: we look for
- * explicit structural cues (list items, action-keywords, headings) rather than
- * calling an LLM. This keeps the operation fast, free, and reproducible across
- * Chinese and English agent outputs.
+ * Delegates extraction, storage, and row mapping to focused modules under
+ * src/refine/. This file keeps the original public API and owns cross-cutting
+ * concerns such as the write mutex.
  */
-export interface RawAction {
-    name?: string;
-    args?: unknown;
-    result?: unknown;
-}
-export interface RawTurn {
-    turnId?: string | number;
-    timestamp?: string;
-    user?: string;
-    agentText?: string;
-    agent?: string;
-    actions?: RawAction[];
-}
-export interface RefinedTurn {
-    sessionId: string;
-    turnId: number;
-    timestamp: string | undefined;
-    summary: string;
-    facts: string[];
-    notes: string[];
-    entities: {
-        files: string[];
-        tools: string[];
-        errors: string[];
-    };
-    categories: Record<string, string[]>;
-}
-export interface RefinedSearchOptions {
-    query: string;
-    dateFrom?: string;
-    dateTo?: string;
-    limit?: number;
-}
-export interface RefinedSearchMatch {
-    sessionId: string;
-    turnId: number;
-    timestamp: string | undefined;
-    summary: string;
-    facts: string[];
-    notes: string[];
-    score: number;
-}
+import type { RawTurn, RefinedTurn, RefinedSearchOptions, RefinedSearchMatch } from './refine/types.js';
+export { type RawAction, type RawTurn, type RefinedTurn, type RefinedSearchOptions, type RefinedSearchMatch, } from './refine/types.js';
 export declare class RefinedManager {
     refinedRoot: string;
-    private dbPath;
-    private db;
+    private store;
     private mutex;
     constructor(refinedRoot: string);
-    private initDb;
     refineTurn(turn: RawTurn, sessionId: string): RefinedTurn;
-    private rowToTurn;
     saveRefinedTurns(sessionId: string, refinedTurns: RefinedTurn[]): Promise<void>;
     loadRefinedTurns(sessionId: string): RefinedTurn[];
-    private loadRefinedTurnsSync;
-    getDbPath(): string;
     loadRefinedTurn(sessionId: string, turnId: number): RefinedTurn | undefined;
+    getDbPath(): string;
     searchRefinedTurns(options: RefinedSearchOptions): RefinedSearchMatch[];
+    countAll(): number;
+    listRecentTurns(limit: number): RefinedTurn[];
     close(): void;
 }