@pcircle/memesh 2.10.0 → 2.11.0

package/dist/knowledge-graph.d.ts

@@ -0,0 +1,48 @@
+import Database from 'better-sqlite3';
+export interface Entity {
+    id: number;
+    name: string;
+    type: string;
+    created_at: string;
+    metadata?: any;
+    observations: string[];
+    tags: string[];
+    relations?: Relation[];
+}
+export interface Relation {
+    from: string;
+    to: string;
+    type: string;
+    metadata?: any;
+}
+export interface CreateEntityInput {
+    name: string;
+    type: string;
+    observations?: string[];
+    tags?: string[];
+    metadata?: any;
+}
+export interface SearchOptions {
+    tag?: string;
+    limit?: number;
+}
+export declare class KnowledgeGraph {
+    private db;
+    constructor(db: Database.Database);
+    createEntity(name: string, type: string, opts?: {
+        observations?: string[];
+        tags?: string[];
+        metadata?: any;
+    }): number;
+    createEntitiesBatch(entities: CreateEntityInput[]): void;
+    createRelation(fromName: string, toName: string, relationType: string, metadata?: any): void;
+    getEntity(name: string): Entity | null;
+    getRelations(entityName: string): Relation[];
+    search(query?: string, opts?: SearchOptions): Entity[];
+    listRecent(limit?: number): Entity[];
+    deleteEntity(name: string): {
+        deleted: boolean;
+    };
+    private rebuildFts;
+}
+//# sourceMappingURL=knowledge-graph.d.ts.map

package/dist/knowledge-graph.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-graph.d.ts","sourceRoot":"","sources":["../src/knowledge-graph.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAEtC,MAAM,WAAW,MAAM;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,GAAG,CAAC;IACf,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,GAAG,CAAC;CAChB;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,CAAC,EAAE,GAAG,CAAC;CAChB;AAED,MAAM,WAAW,aAAa;IAC5B,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,qBAAa,cAAc;IACb,OAAO,CAAC,EAAE;gBAAF,EAAE,EAAE,QAAQ,CAAC,QAAQ;IAEzC,YAAY,CACV,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,EACZ,IAAI,CAAC,EAAE;QAAE,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;QAAC,QAAQ,CAAC,EAAE,GAAG,CAAA;KAAE,GAClE,MAAM;IAmDT,mBAAmB,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,IAAI;IAaxD,cAAc,CACZ,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,YAAY,EAAE,MAAM,EACpB,QAAQ,CAAC,EAAE,GAAG,GACb,IAAI;IA2BP,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IA+BtC,YAAY,CAAC,UAAU,EAAE,MAAM,GAAG,QAAQ,EAAE;IAmB5C,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,aAAa,GAAG,MAAM,EAAE;IAgDtD,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAUpC,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE;IAyBhD,OAAO,CAAC,UAAU;CAsBnB"}

package/dist/knowledge-graph.js

@@ -0,0 +1,186 @@
+export class KnowledgeGraph {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    createEntity(name, type, opts) {
+        const insertResult = this.db
+            .prepare('INSERT OR IGNORE INTO entities (name, type, metadata) VALUES (?, ?, ?)')
+            .run(name, type, opts?.metadata ? JSON.stringify(opts.metadata) : null);
+        const isNewEntity = insertResult.changes > 0;
+        const row = this.db
+            .prepare('SELECT id FROM entities WHERE name = ?')
+            .get(name);
+        const entityId = row.id;
+        const prevObs = isNewEntity
+            ? []
+            : this.db
+                .prepare('SELECT content FROM observations WHERE entity_id = ?')
+                .all(entityId);
+        const prevObsText = isNewEntity
+            ? undefined
+            : prevObs.map((o) => o.content).join(' ');
+        if (opts?.observations?.length) {
+            const insertObs = this.db.prepare('INSERT INTO observations (entity_id, content) VALUES (?, ?)');
+            for (const obs of opts.observations) {
+                insertObs.run(entityId, obs);
+            }
+        }
+        this.rebuildFts(entityId, name, prevObsText);
+        if (opts?.tags?.length) {
+            const insertTag = this.db.prepare('INSERT INTO tags (entity_id, tag) VALUES (?, ?)');
+            for (const tag of opts.tags) {
+                insertTag.run(entityId, tag);
+            }
+        }
+        return entityId;
+    }
+    createEntitiesBatch(entities) {
+        const txn = this.db.transaction(() => {
+            for (const e of entities) {
+                this.createEntity(e.name, e.type, {
+                    observations: e.observations,
+                    tags: e.tags,
+                    metadata: e.metadata,
+                });
+            }
+        });
+        txn();
+    }
+    createRelation(fromName, toName, relationType, metadata) {
+        const fromRow = this.db
+            .prepare('SELECT id FROM entities WHERE name = ?')
+            .get(fromName);
+        const toRow = this.db
+            .prepare('SELECT id FROM entities WHERE name = ?')
+            .get(toName);
+        if (!fromRow) {
+            throw new Error(`Entity not found: ${fromName}`);
+        }
+        if (!toRow) {
+            throw new Error(`Entity not found: ${toName}`);
+        }
+        this.db
+            .prepare('INSERT OR IGNORE INTO relations (from_entity_id, to_entity_id, relation_type, metadata) VALUES (?, ?, ?, ?)')
+            .run(fromRow.id, toRow.id, relationType, metadata ? JSON.stringify(metadata) : null);
+    }
+    getEntity(name) {
+        const row = this.db
+            .prepare('SELECT id, name, type, created_at, metadata FROM entities WHERE name = ?')
+            .get(name);
+        if (!row)
+            return null;
+        const observations = this.db
+            .prepare('SELECT content FROM observations WHERE entity_id = ? ORDER BY id')
+            .all(row.id)
+            .map((o) => o.content);
+        const tags = this.db
+            .prepare('SELECT tag FROM tags WHERE entity_id = ?')
+            .all(row.id)
+            .map((t) => t.tag);
+        const relations = this.getRelations(name);
+        return {
+            id: row.id,
+            name: row.name,
+            type: row.type,
+            created_at: row.created_at,
+            metadata: row.metadata ? JSON.parse(row.metadata) : undefined,
+            observations,
+            tags,
+            relations: relations.length > 0 ? relations : undefined,
+        };
+    }
+    getRelations(entityName) {
+        const rows = this.db
+            .prepare(`SELECT e_from.name AS "from", e_to.name AS "to", r.relation_type AS type, r.metadata
+         FROM relations r
+         JOIN entities e_from ON r.from_entity_id = e_from.id
+         JOIN entities e_to ON r.to_entity_id = e_to.id
+         WHERE e_from.name = ?`)
+            .all(entityName);
+        return rows.map((r) => ({
+            from: r.from,
+            to: r.to,
+            type: r.type,
+            metadata: r.metadata ? JSON.parse(r.metadata) : undefined,
+        }));
+    }
+    search(query, opts) {
+        const limit = opts?.limit ?? 20;
+        if (!query || query.trim() === '') {
+            return this.listRecent(limit);
+        }
+        const sanitized = query.replace(/"/g, '""').trim();
+        const tokens = sanitized.split(/\s+/).filter((t) => t.length > 0);
+        if (tokens.length === 0)
+            return this.listRecent(limit);
+        const ftsQuery = tokens.map((token) => `"${token}"`).join(' ');
+        let ftsRows;
+        try {
+            ftsRows = this.db
+                .prepare(`SELECT e.name FROM entities_fts f
+           JOIN entities e ON e.id = f.rowid
+           WHERE entities_fts MATCH ?
+           LIMIT ?`)
+                .all(ftsQuery, limit);
+        }
+        catch (err) {
+            if (err.message?.includes('fts5'))
+                return [];
+            throw err;
+        }
+        if (ftsRows.length === 0)
+            return [];
+        const results = [];
+        for (const ftsRow of ftsRows) {
+            const entity = this.getEntity(ftsRow.name);
+            if (!entity)
+                continue;
+            if (opts?.tag) {
+                if (!entity.tags.includes(opts.tag))
+                    continue;
+            }
+            results.push(entity);
+        }
+        return results;
+    }
+    listRecent(limit) {
+        const rows = this.db
+            .prepare('SELECT name FROM entities ORDER BY id DESC LIMIT ?')
+            .all(limit ?? 20);
+        return rows
+            .map((r) => this.getEntity(r.name))
+            .filter((e) => e !== null);
+    }
+    deleteEntity(name) {
+        const row = this.db
+            .prepare('SELECT id FROM entities WHERE name = ?')
+            .get(name);
+        if (!row)
+            return { deleted: false };
+        const allObs = this.db
+            .prepare('SELECT content FROM observations WHERE entity_id = ?')
+            .all(row.id);
+        const obsText = allObs.map((o) => o.content).join(' ');
+        this.db
+            .prepare("INSERT INTO entities_fts (entities_fts, rowid, name, observations) VALUES('delete', ?, ?, ?)")
+            .run(row.id, name, obsText);
+        this.db.prepare('DELETE FROM entities WHERE id = ?').run(row.id);
+        return { deleted: true };
+    }
+    rebuildFts(entityId, entityName, previousObsText) {
+        if (previousObsText !== undefined) {
+            this.db
+                .prepare("INSERT INTO entities_fts (entities_fts, rowid, name, observations) VALUES('delete', ?, ?, ?)")
+                .run(entityId, entityName, previousObsText);
+        }
+        const allObs = this.db
+            .prepare('SELECT content FROM observations WHERE entity_id = ?')
+            .all(entityId);
+        const obsText = allObs.map((o) => o.content).join(' ');
+        this.db
+            .prepare('INSERT INTO entities_fts (rowid, name, observations) VALUES (?, ?, ?)')
+            .run(entityId, entityName, obsText);
+    }
+}
+//# sourceMappingURL=knowledge-graph.js.map

package/dist/knowledge-graph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-graph.js","sourceRoot":"","sources":["../src/knowledge-graph.ts"],"names":[],"mappings":"AAiCA,MAAM,OAAO,cAAc;IACL;IAApB,YAAoB,EAAqB;QAArB,OAAE,GAAF,EAAE,CAAmB;IAAG,CAAC;IAE7C,YAAY,CACV,IAAY,EACZ,IAAY,EACZ,IAAmE;QAGnE,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE;aACzB,OAAO,CACN,wEAAwE,CACzE;aACA,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1E,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,GAAG,CAAC,CAAC;QAE7C,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aAChB,OAAO,CAAC,wCAAwC,CAAC;aACjD,GAAG,CAAC,IAAI,CAAmB,CAAC;QAC/B,MAAM,QAAQ,GAAG,GAAG,CAAC,EAAE,CAAC;QAIxB,MAAM,OAAO,GAAG,WAAW;YACzB,CAAC,CAAC,EAAE;YACJ,CAAC,CAAE,IAAI,CAAC,EAAE;iBACL,OAAO,CAAC,sDAAsD,CAAC;iBAC/D,GAAG,CAAC,QAAQ,CAA2B,CAAC;QAC/C,MAAM,WAAW,GAAG,WAAW;YAC7B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAG5C,IAAI,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC/B,6DAA6D,CAC9D,CAAC;YACF,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;QAGD,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAG7C,IAAI,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACvB,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC/B,iDAAiD,CAClD,CAAC;YACF,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC5B,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,mBAAmB,CAAC,QAA6B;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YACnC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzB,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;oBAChC,YAAY,EAAE,CAAC,CAAC,YAAY;oBAC5B,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ;iBACrB,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QACH,GAAG,EAAE,CAAC;IACR,CAAC;IAED,cAAc,CACZ,QAAgB,EAChB,MAAc,EACd,YAAoB,EACpB,QAAc;QAEd,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE;aACpB,OAAO,CAAC,wCAAwC,CAAC;aACjD,GAAG,CAAC,QAAQ,CAA+B,CAAC;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE;aAClB,OAAO,CAAC,wCAAwC,CAAC;aACjD,GAAG,CAAC,MAAM,CAA+B,CAAC;QAE7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,EAAE,CAAC,CAAC;QACjD,CAAC;QAED,IAAI,CAAC,EAAE;aACJ,OAAO,CACN,6GAA6G,CAC9G;aACA,GAAG,CACF,OAAO,CAAC,EAAE,EACV,KAAK,CAAC,EAAE,EACR,YAAY,EACZ,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAC3C,CAAC;IACN,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aAChB,OAAO,CAAC,0EAA0E,CAAC;aACnF,GAAG,CAAC,IAAI,CAAoB,CAAC;QAEhC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QAEtB,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE;aACzB,OAAO,CAAC,kEAAkE,CAAC;aAC3E,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;aACX,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAE9B,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CAAC,0CAA0C,CAAC;aACnD,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;aACX,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAE1B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAE1C,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,QAAQ,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;YAC7D,YAAY;YACZ,IAAI;YACJ,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;SACxD,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,UAAkB;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CACN;;;;+BAIuB,CACxB;aACA,GAAG,CAAC,UAAU,CAAU,CAAC;QAE5B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;SAC1D,CAAC,CAAC,CAAC;IACN,CAAC;IAED,MAAM,CAAC,KAAc,EAAE,IAAoB;QACzC,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;QAEhC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAClC,OAAO,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAID,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAClE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvD,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE/D,IAAI,OAA2B,CAAC;QAChC,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,EAAE;iBACd,OAAO,CACN;;;mBAGS,CACV;iBACA,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAuB,CAAC;QAChD,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAElB,IAAI,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC;gBAAE,OAAO,EAAE,CAAC;YAC7C,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAGpC,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3C,IAAI,CAAC,MAAM;gBAAE,SAAS;YAEtB,IAAI,IAAI,EAAE,GAAG,EAAE,CAAC;gBACd,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC;oBAAE,SAAS;YAChD,CAAC;YAED,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,UAAU,CAAC,KAAc;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CAAC,oDAAoD,CAAC;aAC7D,GAAG,CAAC,KAAK,IAAI,EAAE,CAAuB,CAAC;QAE1C,OAAO,IAAI;aACR,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;aAClC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED,YAAY,CAAC,IAAY;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aAChB,OAAO,CAAC,wCAAwC,CAAC;aACjD,GAAG,CAAC,IAAI,CAA+B,CAAC;QAE3C,IAAI,CAAC,GAAG;YAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAIpC,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE;aACnB,OAAO,CAAC,sDAAsD,CAAC;aAC/D,GAAG,CAAC,GAAG,CAAC,EAAE,CAA0B,CAAC;QACxC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,CAAC,EAAE;aACJ,OAAO,CACN,8FAA8F,CAC/F;aACA,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAG9B,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEjE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU,CAChB,QAAgB,EAChB,UAAkB,EAClB,eAAwB;QAIxB,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;YAClC,IAAI,CAAC,EAAE;iBACJ,OAAO,CACN,8FAA8F,CAC/F;iBACA,GAAG,CAAC,QAAQ,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;QAChD,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE;aACnB,OAAO,CAAC,sDAAsD,CAAC;aAC/D,GAAG,CAAC,QAAQ,CAA0B,CAAC;QAC1C,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,CAAC,EAAE;aACJ,OAAO,CAAC,uEAAuE,CAAC;aAChF,GAAG,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IACxC,CAAC;CACF"}

package/dist/mcp/server.d.ts

@@ -1,27 +1,3 @@
 #!/usr/bin/env node
-export declare class ToolCallTimeoutError extends Error {
-    readonly toolName: string;
-    readonly timeoutMs: number;
-    readonly elapsedMs: number;
-    constructor(toolName: string, timeoutMs: number, elapsedMs: number);
-}
-declare class ClaudeCodeBuddyMCPServer {
-    private server;
-    private components;
-    private toolRouter;
-    private sessionBootstrapper;
-    private isShuttingDown;
-    private shutdownPromise;
-    get toolHandlers(): import("./handlers/ToolHandlers.js").ToolHandlers;
-    get buddyHandlers(): import("./handlers/BuddyHandlers.js").BuddyHandlers;
-    private constructor();
-    static create(): Promise<ClaudeCodeBuddyMCPServer>;
-    private setupHandlers;
-    start(): Promise<void>;
-    handleRequest(request: unknown): Promise<unknown>;
-    private setupSignalHandlers;
-    private shutdown;
-    private performShutdown;
-}
-export { ClaudeCodeBuddyMCPServer };
+export {};
 //# sourceMappingURL=server.d.ts.map

package/dist/mcp/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":";AA4DA,qBAAa,oBAAqB,SAAQ,KAAK;IAC7C,SAAgB,QAAQ,EAAE,MAAM,CAAC;IACjC,SAAgB,SAAS,EAAE,MAAM,CAAC;IAClC,SAAgB,SAAS,EAAE,MAAM,CAAC;gBAEtB,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;CAOnE;AA8BD,cAAM,wBAAwB;IAC5B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,UAAU,CAAmB;IACrC,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,cAAc,CAAS;IAE/B,OAAO,CAAC,eAAe,CAA8B;IAOrD,IAAW,YAAY,sDAEtB;IAOD,IAAW,aAAa,wDAEvB;IAiBD,OAAO;WAsDM,MAAM,IAAI,OAAO,CAAC,wBAAwB,CAAC;IAiBxD,OAAO,CAAC,aAAa;IAwGf,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBtB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IA0GvD,OAAO,CAAC,mBAAmB;YA0Bb,QAAQ;YAkBR,eAAe;CAiE9B;AAGD,OAAO,EAAE,wBAAwB,EAAE,CAAC"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":""}

package/dist/mcp/server.js

@@ -2,326 +2,40 @@
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
-import { ServerInitializer } from './ServerInitializer.js';
-import { ToolRouter } from './ToolRouter.js';
-import { getAllToolDefinitions } from './ToolDefinitions.js';
-import { setupResourceHandlers } from './handlers/index.js';
-import { SessionBootstrapper } from './SessionBootstrapper.js';
-import { logger } from '../utils/logger.js';
-import { readFileSync } from 'fs';
-import { fileURLToPath } from 'url';
-import { dirname, join } from 'path';
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = dirname(__filename);
-const packageJson = JSON.parse(readFileSync(join(__dirname, '../../package.json'), 'utf-8'));
-import { logError, formatMCPError } from '../utils/errorHandler.js';
-import { generateRequestId } from '../utils/requestId.js';
-const DEFAULT_TOOL_TIMEOUT_MS = 60000;
-const MIN_TOOL_TIMEOUT_MS = 1000;
-const parsedToolTimeoutMs = parseInt(process.env.MEMESH_TOOL_TIMEOUT_MS || '', 10);
-const toolTimeoutMs = Number.isFinite(parsedToolTimeoutMs) && parsedToolTimeoutMs > 0
-    ? Math.max(MIN_TOOL_TIMEOUT_MS, parsedToolTimeoutMs)
-    : DEFAULT_TOOL_TIMEOUT_MS;
-export class ToolCallTimeoutError extends Error {
-    toolName;
-    timeoutMs;
-    elapsedMs;
-    constructor(toolName, timeoutMs, elapsedMs) {
-        super(`Tool '${toolName}' timed out after ${elapsedMs}ms (limit: ${timeoutMs}ms)`);
-        this.name = 'ToolCallTimeoutError';
-        this.toolName = toolName;
-        this.timeoutMs = timeoutMs;
-        this.elapsedMs = elapsedMs;
-    }
+import { openDatabase, closeDatabase } from '../db.js';
+import { handleTool, TOOL_DEFINITIONS } from './tools.js';
+const server = new Server({ name: 'memesh', version: '2.11.0' }, { capabilities: { tools: {} } });
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+    tools: TOOL_DEFINITIONS.map((t) => ({
+        name: t.name,
+        description: t.description,
+        inputSchema: t.inputSchema,
+    })),
+}));
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    const { name, arguments: args } = request.params;
+    return handleTool(name, args);
+});
+async function main() {
+    openDatabase();
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
 }
-class ClaudeCodeBuddyMCPServer {
-    server;
-    components;
-    toolRouter;
-    sessionBootstrapper;
-    isShuttingDown = false;
-    shutdownPromise = null;
-    get toolHandlers() {
-        return this.components.toolHandlers;
-    }
-    get buddyHandlers() {
-        return this.components.buddyHandlers;
-    }
-    constructor(components) {
-        this.server = new Server({
-            name: 'memesh',
-            version: packageJson.version,
-        }, {
-            capabilities: {
-                tools: {},
-                resources: {
-                    subscribe: true,
-                    listChanged: false,
-                },
-            },
-        });
-        this.components = components;
-        this.toolRouter = new ToolRouter({
-            rateLimiter: this.components.rateLimiter,
-            toolHandlers: this.components.toolHandlers,
-            buddyHandlers: this.components.buddyHandlers,
-            knowledgeGraph: this.components.knowledgeGraph,
-        });
-        this.components.toolInterface.attachToolDispatcher(this.toolRouter);
-        this.sessionBootstrapper = new SessionBootstrapper(this.components.projectMemoryManager, undefined, this.components.sessionMemoryPipeline);
-        if (this.components.sessionMemoryPipeline) {
-            void this.components.sessionMemoryPipeline.start().catch(err => {
-                logger.warn('SessionMemoryPipeline failed to start:', err);
-            });
-        }
-        this.setupHandlers();
-        setupResourceHandlers(this.server);
-        this.setupSignalHandlers();
-    }
-    static async create() {
-        const components = await ServerInitializer.initialize();
-        return new ClaudeCodeBuddyMCPServer(components);
-    }
-    setupHandlers() {
-        this.server.setRequestHandler(ListToolsRequestSchema, async () => {
-            const tools = getAllToolDefinitions();
-            return { tools };
-        });
-        this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-            const requestId = generateRequestId();
-            const params = request.params;
-            const toolName = params.name || 'unknown';
-            const startTime = Date.now();
-            logger.debug('[MCP] Incoming tool call request', {
-                requestId,
-                toolName,
-                component: 'ClaudeCodeBuddyMCPServer',
-            });
-            try {
-                const toolPromise = this.toolRouter.routeToolCall(request.params, undefined, requestId);
-                let timeoutId;
-                const timeoutPromise = new Promise((_, reject) => {
-                    timeoutId = setTimeout(() => {
-                        const elapsed = Date.now() - startTime;
-                        reject(new ToolCallTimeoutError(toolName, toolTimeoutMs, elapsed));
-                    }, toolTimeoutMs);
-                });
-                let result;
-                try {
-                    result = await Promise.race([toolPromise, timeoutPromise]);
-                }
-                finally {
-                    if (timeoutId) {
-                        clearTimeout(timeoutId);
-                    }
-                }
-                return await this.sessionBootstrapper.maybePrepend(result);
-            }
-            catch (error) {
-                const elapsed = Date.now() - startTime;
-                if (error instanceof ToolCallTimeoutError) {
-                    logger.error('[MCP] Tool call timed out', {
-                        requestId,
-                        toolName,
-                        timeoutMs: toolTimeoutMs,
-                        elapsedMs: elapsed,
-                        component: 'ClaudeCodeBuddyMCPServer',
-                    });
-                }
-                logError(error, {
-                    component: 'ClaudeCodeBuddyMCPServer',
-                    method: 'CallToolRequestHandler',
-                    requestId,
-                    data: {
-                        toolName,
-                        elapsedMs: elapsed,
-                    },
-                });
-                return formatMCPError(error, {
-                    component: 'ClaudeCodeBuddyMCPServer',
-                    method: 'CallToolRequestHandler',
-                    requestId,
-                    data: {
-                        toolName,
-                        elapsedMs: elapsed,
-                    },
-                });
-            }
-        });
-    }
-    async start() {
-        const transport = new StdioServerTransport();
-        await this.server.connect(transport);
-    }
-    async handleRequest(request) {
-        const requestId = generateRequestId();
-        if (!request || typeof request !== 'object') {
-            return {
-                jsonrpc: '2.0',
-                id: null,
-                error: { code: -32600, message: 'Invalid Request' },
-            };
-        }
-        const req = request;
-        const method = req.method;
-        const params = req.params;
-        const id = req.id;
-        try {
-            if (method === 'tools/list') {
-                const tools = getAllToolDefinitions();
-                return {
-                    jsonrpc: '2.0',
-                    id,
-                    result: { tools },
-                };
-            }
-            if (method === 'tools/call') {
-                const startTime = Date.now();
-                const callParams = params;
-                const toolName = callParams?.name || 'unknown';
-                logger.debug('[MCP] Daemon handling tool call request', {
-                    requestId,
-                    toolName,
-                    component: 'ClaudeCodeBuddyMCPServer',
-                });
-                const toolPromise = this.toolRouter.routeToolCall(callParams, undefined, requestId);
-                let timeoutId;
-                const timeoutPromise = new Promise((_, reject) => {
-                    timeoutId = setTimeout(() => {
-                        const elapsed = Date.now() - startTime;
-                        reject(new ToolCallTimeoutError(toolName, toolTimeoutMs, elapsed));
-                    }, toolTimeoutMs);
-                });
-                let result;
-                try {
-                    result = await Promise.race([toolPromise, timeoutPromise]);
-                }
-                finally {
-                    if (timeoutId) {
-                        clearTimeout(timeoutId);
-                    }
-                }
-                const finalResult = await this.sessionBootstrapper.maybePrepend(result);
-                return {
-                    jsonrpc: '2.0',
-                    id,
-                    result: finalResult,
-                };
-            }
-            return {
-                jsonrpc: '2.0',
-                id,
-                error: { code: -32601, message: `Method not found: ${method}` },
-            };
-        }
-        catch (error) {
-            if (error instanceof ToolCallTimeoutError) {
-                logger.error('[MCP] Daemon tool call timed out', {
-                    requestId,
-                    toolName: error.toolName,
-                    timeoutMs: toolTimeoutMs,
-                    component: 'ClaudeCodeBuddyMCPServer',
-                });
-            }
-            logError(error, {
-                component: 'ClaudeCodeBuddyMCPServer',
-                method: 'handleRequest',
-                requestId,
-            });
-            const errorResult = formatMCPError(error, {
-                component: 'ClaudeCodeBuddyMCPServer',
-                method: 'handleRequest',
-                requestId,
-            });
-            return {
-                jsonrpc: '2.0',
-                id,
-                result: errorResult,
-            };
-        }
-    }
-    setupSignalHandlers() {
-        this.server.onclose = () => {
-            logger.warn('MCP transport closed');
-        };
-        this.server.onerror = (error) => {
-            logger.error('MCP server error:', error);
-        };
-        const signals = ['SIGINT', 'SIGTERM'];
-        for (const signal of signals) {
-            process.once(signal, () => {
-                void this.shutdown(signal);
-            });
-        }
-    }
-    async shutdown(reason) {
-        if (this.shutdownPromise) {
-            return this.shutdownPromise;
-        }
-        if (this.isShuttingDown)
-            return;
-        this.isShuttingDown = true;
-        this.shutdownPromise = this.performShutdown(reason);
-        return this.shutdownPromise;
-    }
-    async performShutdown(reason) {
-        logger.warn(`Shutting down MCP server (${reason})...`);
-        try {
-            logger.info('Stopping session memory pipeline...');
-            if (this.components.sessionMemoryPipeline) {
-                await this.components.sessionMemoryPipeline.stop();
-            }
-        }
-        catch (error) {
-            logError(error, {
-                component: 'ClaudeCodeBuddyMCPServer',
-                method: 'shutdown',
-                operation: 'stopping session memory pipeline',
-            });
-            logger.error('Failed to stop session memory pipeline cleanly:', error);
-        }
-        try {
-            logger.info('Closing knowledge graph database...');
-            if (this.components.knowledgeGraph) {
-                await this.components.knowledgeGraph.close();
-            }
-        }
-        catch (error) {
-            logError(error, {
-                component: 'ClaudeCodeBuddyMCPServer',
-                method: 'shutdown',
-                operation: 'closing knowledge graph',
-            });
-            logger.error('Failed to close knowledge graph cleanly:', error);
-        }
-        try {
-            logger.info('Stopping rate limiter...');
-            if (this.components.rateLimiter) {
-                this.components.rateLimiter.stop();
-            }
-        }
-        catch (error) {
-            logError(error, {
-                component: 'ClaudeCodeBuddyMCPServer',
-                method: 'shutdown',
-                operation: 'stopping rate limiter',
-            });
-            logger.error('Failed to stop rate limiter cleanly:', error);
-        }
-        try {
-            logger.info('Closing MCP server transport...');
-            await this.server.close();
-        }
-        catch (error) {
-            logError(error, {
-                component: 'ClaudeCodeBuddyMCPServer',
-                method: 'shutdown',
-                operation: 'closing MCP server',
-            });
-            logger.error('Failed to close MCP server cleanly:', error);
-        }
-        logger.info('Shutdown complete');
+function shutdown() {
+    try {
+        closeDatabase();
     }
+    catch { }
+    process.exit(0);
 }
-export { ClaudeCodeBuddyMCPServer };
+process.on('SIGINT', shutdown);
+process.on('SIGTERM', shutdown);
+main().catch((err) => {
+    console.error('MeMesh server error:', err instanceof Error ? err.message : String(err));
+    try {
+        closeDatabase();
+    }
+    catch { }
+    process.exit(1);
+});
 //# sourceMappingURL=server.js.map