agentic-flow 2.0.7 → 2.0.9

package/dist/memory/SharedMemoryPool.d.ts

@@ -0,0 +1,103 @@
+/**
+ * SharedMemoryPool — singleton resource pool that backs HybridReasoningBank
+ * and AdvancedMemorySystem.
+ *
+ * Centralises a single SQLite database handle (better-sqlite3) and a single
+ * `EmbeddingService` instance so multiple memory consumers share the same
+ * underlying tables / embedding cache without conflicting writes.
+ *
+ * The pool is constructed lazily on first `getInstance()` call. All heavy
+ * resources (sqlite handle, embedder pipeline) are created on demand inside
+ * `ensureInitialized()` so simply importing this module is side-effect free.
+ *
+ * Used by:
+ *   - reasoningbank/HybridBackend.ts   (HybridReasoningBank)
+ *   - reasoningbank/AdvancedMemory.ts  (AdvancedMemorySystem)
+ *
+ * Fixes issue #102 — this file was previously imported but missing on disk,
+ * which broke `import 'agentic-flow'` at the top level.
+ */
+import { EmbeddingService } from 'agentdb';
+type DatabaseHandle = any;
+type EmbedderHandle = EmbeddingService;
+export interface SharedMemoryPoolOptions {
+    /** SQLite database path. Defaults to `~/.agentic-flow/reasoningbank.db`. */
+    dbPath?: string;
+    /** Embedding model id. Defaults to `Xenova/all-MiniLM-L6-v2`. */
+    embeddingModel?: string;
+    /** Embedding vector dimension. Defaults to 384 (MiniLM-L6). */
+    embeddingDimension?: number;
+    /** Embedding provider. Defaults to `'transformers'`. */
+    embeddingProvider?: 'transformers' | 'openai' | 'local';
+}
+export interface SharedMemoryPoolStats {
+    initialized: boolean;
+    dbPath: string;
+    embeddingModel: string;
+    embeddingDimension: number;
+    cache: {
+        entries: number;
+        hits: number;
+        misses: number;
+        evictions: number;
+    };
+}
+export declare class SharedMemoryPool {
+    private static _instance;
+    private readonly options;
+    private db;
+    private embedder;
+    private initPromise;
+    private cache;
+    private cacheStats;
+    private constructor();
+    /**
+     * Singleton accessor. The first caller wins for option overrides; later
+     * callers always get the existing pool. To reconfigure, call `reset()`.
+     */
+    static getInstance(options?: SharedMemoryPoolOptions): SharedMemoryPool;
+    /** Tear down the singleton — primarily for tests. */
+    static reset(): void;
+    /**
+     * Idempotently ensure the database and embedder are ready. Subsequent calls
+     * return the same in-flight promise so concurrent consumers share init.
+     */
+    ensureInitialized(): Promise<void>;
+    private initialize;
+    /** Apply the minimum schema that ReflexionMemory / SkillLibrary require. */
+    private applySchema;
+    /**
+     * Synchronous accessor for the database handle. Throws if init hasn't run —
+     * call `ensureInitialized()` first. Provided for compatibility with
+     * controllers that take a `Database` instance in their constructor.
+     */
+    getDatabase(): DatabaseHandle;
+    /**
+     * Synchronous accessor for the embedder. Throws if init hasn't run.
+     */
+    getEmbedder(): EmbedderHandle;
+    /**
+     * Cache a query result with a TTL (milliseconds). Keys are arbitrary
+     * strings; consumers (HybridReasoningBank) typically encode the query
+     * shape into the key.
+     */
+    cacheQuery<T>(key: string, value: T, ttlMs: number): void;
+    /**
+     * Read a cached query result. Returns the cached value if present and not
+     * expired; lazily evicts expired entries on lookup.
+     *
+     * The default `T` is `any` for ergonomic interop with the existing
+     * HybridReasoningBank call sites that expect a loose return type.
+     * Pass an explicit type parameter (`getCachedQuery<MyShape>(...)`) when
+     * you want stricter typing.
+     */
+    getCachedQuery<T = any>(key: string): T | undefined;
+    /** Drop all cached query results. */
+    invalidateCache(): void;
+    /** Diagnostic stats for telemetry / health endpoints. */
+    getStats(): SharedMemoryPoolStats;
+    /** Close the underlying database handle and clear cached state. */
+    close(): void;
+}
+export {};
+//# sourceMappingURL=SharedMemoryPool.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"SharedMemoryPool.d.ts","sourceRoot":"","sources":["../../src/memory/SharedMemoryPool.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAKH,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAE3C,KAAK,cAAc,GAAG,GAAG,CAAC;AAC1B,KAAK,cAAc,GAAG,gBAAgB,CAAC;AAEvC,MAAM,WAAW,uBAAuB;IACtC,4EAA4E;IAC5E,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,iEAAiE;IACjE,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,+DAA+D;IAC/D,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,wDAAwD;IACxD,iBAAiB,CAAC,EAAE,cAAc,GAAG,QAAQ,GAAG,OAAO,CAAC;CACzD;AAcD,MAAM,WAAW,qBAAqB;IACpC,WAAW,EAAE,OAAO,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;IACvB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,KAAK,EAAE;QACL,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAC,SAAS,CAAiC;IAEzD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoC;IAC5D,OAAO,CAAC,EAAE,CAA+B;IACzC,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,WAAW,CAA8B;IAIjD,OAAO,CAAC,KAAK,CAA0C;IACvD,OAAO,CAAC,UAAU,CAAwC;IAE1D,OAAO;IAIP;;;OAGG;IACH,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,uBAAuB,GAAG,gBAAgB;IAOvE,qDAAqD;IACrD,MAAM,CAAC,KAAK,IAAI,IAAI;IAOpB;;;OAGG;IACG,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;YAQ1B,UAAU;IAsBxB,4EAA4E;IAC5E,OAAO,CAAC,WAAW;IAoEnB;;;;OAIG;IACH,WAAW,IAAI,cAAc;IAS7B;;OAEG;IACH,WAAW,IAAI,cAAc;IAS7B;;;;OAIG;IACH,UAAU,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAQzD;;;;;;;;OAQG;IAEH,cAAc,CAAC,CAAC,GAAG,GAAG,EAAE,GAAG,EAAE,MAAM,GAAG,CAAC,GAAG,SAAS;IAgBnD,qCAAqC;IACrC,eAAe,IAAI,IAAI;IAKvB,yDAAyD;IACzD,QAAQ,IAAI,qBAAqB;IAejC,mEAAmE;IACnE,KAAK,IAAI,IAAI;CAYd"}

package/dist/memory/SharedMemoryPool.js

@@ -0,0 +1,263 @@
+/**
+ * SharedMemoryPool — singleton resource pool that backs HybridReasoningBank
+ * and AdvancedMemorySystem.
+ *
+ * Centralises a single SQLite database handle (better-sqlite3) and a single
+ * `EmbeddingService` instance so multiple memory consumers share the same
+ * underlying tables / embedding cache without conflicting writes.
+ *
+ * The pool is constructed lazily on first `getInstance()` call. All heavy
+ * resources (sqlite handle, embedder pipeline) are created on demand inside
+ * `ensureInitialized()` so simply importing this module is side-effect free.
+ *
+ * Used by:
+ *   - reasoningbank/HybridBackend.ts   (HybridReasoningBank)
+ *   - reasoningbank/AdvancedMemory.ts  (AdvancedMemorySystem)
+ *
+ * Fixes issue #102 — this file was previously imported but missing on disk,
+ * which broke `import 'agentic-flow'` at the top level.
+ */
+import { mkdirSync } from 'node:fs';
+import { dirname, join } from 'node:path';
+import { homedir } from 'node:os';
+import { EmbeddingService } from 'agentdb';
+const DEFAULT_OPTIONS = {
+    dbPath: join(homedir(), '.agentic-flow', 'reasoningbank.db'),
+    embeddingModel: 'Xenova/all-MiniLM-L6-v2',
+    embeddingDimension: 384,
+    embeddingProvider: 'transformers',
+};
+export class SharedMemoryPool {
+    static _instance = null;
+    options;
+    db = null;
+    embedder = null;
+    initPromise = null;
+    // Lightweight in-process query cache used by HybridReasoningBank to avoid
+    // re-running the same retrieval across consumers in one session.
+    cache = new Map();
+    cacheStats = { hits: 0, misses: 0, evictions: 0 };
+    constructor(options = {}) {
+        this.options = { ...DEFAULT_OPTIONS, ...options };
+    }
+    /**
+     * Singleton accessor. The first caller wins for option overrides; later
+     * callers always get the existing pool. To reconfigure, call `reset()`.
+     */
+    static getInstance(options) {
+        if (!SharedMemoryPool._instance) {
+            SharedMemoryPool._instance = new SharedMemoryPool(options);
+        }
+        return SharedMemoryPool._instance;
+    }
+    /** Tear down the singleton — primarily for tests. */
+    static reset() {
+        if (SharedMemoryPool._instance) {
+            SharedMemoryPool._instance.close();
+        }
+        SharedMemoryPool._instance = null;
+    }
+    /**
+     * Idempotently ensure the database and embedder are ready. Subsequent calls
+     * return the same in-flight promise so concurrent consumers share init.
+     */
+    async ensureInitialized() {
+        if (this.db && this.embedder)
+            return;
+        if (!this.initPromise) {
+            this.initPromise = this.initialize();
+        }
+        return this.initPromise;
+    }
+    async initialize() {
+        // Dynamic import keeps node:fs, better-sqlite3, and the embedding pipeline
+        // out of the import graph for callers that never use them.
+        const Database = await loadBetterSqlite3();
+        mkdirSync(dirname(this.options.dbPath), { recursive: true });
+        this.db = new Database(this.options.dbPath);
+        // Reasonable defaults for an embedded reasoning database.
+        this.db.pragma('journal_mode = WAL');
+        this.db.pragma('foreign_keys = ON');
+        this.db.pragma('synchronous = NORMAL');
+        this.applySchema(this.db);
+        this.embedder = new EmbeddingService({
+            model: this.options.embeddingModel,
+            dimension: this.options.embeddingDimension,
+            provider: this.options.embeddingProvider,
+        });
+        await this.embedder.initialize();
+    }
+    /** Apply the minimum schema that ReflexionMemory / SkillLibrary require. */
+    applySchema(db) {
+        db.exec(`
+      CREATE TABLE IF NOT EXISTS episodes (
+        id           INTEGER PRIMARY KEY AUTOINCREMENT,
+        ts           INTEGER NOT NULL DEFAULT (strftime('%s', 'now')),
+        session_id   TEXT NOT NULL,
+        task         TEXT NOT NULL,
+        input        TEXT,
+        output       TEXT,
+        critique     TEXT,
+        reward       REAL NOT NULL,
+        success      INTEGER NOT NULL,
+        latency_ms   INTEGER,
+        tokens_used  INTEGER,
+        tags         TEXT,
+        metadata     TEXT
+      );
+
+      CREATE TABLE IF NOT EXISTS episode_embeddings (
+        episode_id   INTEGER PRIMARY KEY,
+        embedding    BLOB NOT NULL,
+        FOREIGN KEY (episode_id) REFERENCES episodes(id) ON DELETE CASCADE
+      );
+
+      CREATE INDEX IF NOT EXISTS idx_episodes_task    ON episodes(task);
+      CREATE INDEX IF NOT EXISTS idx_episodes_ts      ON episodes(ts);
+      CREATE INDEX IF NOT EXISTS idx_episodes_success ON episodes(success);
+
+      CREATE TABLE IF NOT EXISTS skills (
+        id           INTEGER PRIMARY KEY AUTOINCREMENT,
+        ts           INTEGER NOT NULL DEFAULT (strftime('%s', 'now')),
+        name         TEXT NOT NULL,
+        description  TEXT,
+        precondition TEXT,
+        action       TEXT,
+        outcome      TEXT,
+        success_rate REAL,
+        uses         INTEGER DEFAULT 0,
+        tags         TEXT,
+        metadata     TEXT,
+        UNIQUE(name)
+      );
+
+      CREATE TABLE IF NOT EXISTS skill_embeddings (
+        skill_id     INTEGER PRIMARY KEY,
+        embedding    BLOB NOT NULL,
+        FOREIGN KEY (skill_id) REFERENCES skills(id) ON DELETE CASCADE
+      );
+
+      CREATE TABLE IF NOT EXISTS causal_edges (
+        id              INTEGER PRIMARY KEY AUTOINCREMENT,
+        ts              INTEGER NOT NULL DEFAULT (strftime('%s', 'now')),
+        from_memory_id   INTEGER NOT NULL,
+        from_memory_type TEXT NOT NULL,
+        to_memory_id     INTEGER NOT NULL,
+        to_memory_type   TEXT NOT NULL,
+        similarity      REAL,
+        uplift          REAL,
+        confidence      REAL,
+        sample_size     INTEGER,
+        metadata        TEXT
+      );
+
+      CREATE INDEX IF NOT EXISTS idx_causal_from ON causal_edges(from_memory_id, from_memory_type);
+      CREATE INDEX IF NOT EXISTS idx_causal_to   ON causal_edges(to_memory_id, to_memory_type);
+    `);
+    }
+    /**
+     * Synchronous accessor for the database handle. Throws if init hasn't run —
+     * call `ensureInitialized()` first. Provided for compatibility with
+     * controllers that take a `Database` instance in their constructor.
+     */
+    getDatabase() {
+        if (!this.db) {
+            throw new Error('SharedMemoryPool: database not initialised. Call ensureInitialized() first.');
+        }
+        return this.db;
+    }
+    /**
+     * Synchronous accessor for the embedder. Throws if init hasn't run.
+     */
+    getEmbedder() {
+        if (!this.embedder) {
+            throw new Error('SharedMemoryPool: embedder not initialised. Call ensureInitialized() first.');
+        }
+        return this.embedder;
+    }
+    /**
+     * Cache a query result with a TTL (milliseconds). Keys are arbitrary
+     * strings; consumers (HybridReasoningBank) typically encode the query
+     * shape into the key.
+     */
+    cacheQuery(key, value, ttlMs) {
+        if (!key || ttlMs <= 0)
+            return;
+        this.cache.set(key, {
+            value: value,
+            expiresAt: Date.now() + ttlMs,
+        });
+    }
+    /**
+     * Read a cached query result. Returns the cached value if present and not
+     * expired; lazily evicts expired entries on lookup.
+     *
+     * The default `T` is `any` for ergonomic interop with the existing
+     * HybridReasoningBank call sites that expect a loose return type.
+     * Pass an explicit type parameter (`getCachedQuery<MyShape>(...)`) when
+     * you want stricter typing.
+     */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    getCachedQuery(key) {
+        const entry = this.cache.get(key);
+        if (!entry) {
+            this.cacheStats.misses++;
+            return undefined;
+        }
+        if (entry.expiresAt < Date.now()) {
+            this.cache.delete(key);
+            this.cacheStats.evictions++;
+            this.cacheStats.misses++;
+            return undefined;
+        }
+        this.cacheStats.hits++;
+        return entry.value;
+    }
+    /** Drop all cached query results. */
+    invalidateCache() {
+        this.cacheStats.evictions += this.cache.size;
+        this.cache.clear();
+    }
+    /** Diagnostic stats for telemetry / health endpoints. */
+    getStats() {
+        return {
+            initialized: this.db !== null && this.embedder !== null,
+            dbPath: this.options.dbPath,
+            embeddingModel: this.options.embeddingModel,
+            embeddingDimension: this.options.embeddingDimension,
+            cache: {
+                entries: this.cache.size,
+                hits: this.cacheStats.hits,
+                misses: this.cacheStats.misses,
+                evictions: this.cacheStats.evictions,
+            },
+        };
+    }
+    /** Close the underlying database handle and clear cached state. */
+    close() {
+        try {
+            this.db?.close?.();
+        }
+        catch {
+            /* swallow — closing twice is benign */
+        }
+        this.db = null;
+        this.embedder = null;
+        this.initPromise = null;
+        this.cache.clear();
+        this.cacheStats = { hits: 0, misses: 0, evictions: 0 };
+    }
+}
+async function loadBetterSqlite3() {
+    // better-sqlite3 is a heavy native module; load it lazily so the rest of the
+    // package can be imported even when this optional dep is unavailable.
+    try {
+        const mod = await import('better-sqlite3');
+        return mod.default ?? mod;
+    }
+    catch (err) {
+        throw new Error(`SharedMemoryPool requires 'better-sqlite3' but it could not be loaded: ${err?.message || err}. ` +
+            `Install it with: npm install better-sqlite3`);
+    }
+}
+//# sourceMappingURL=SharedMemoryPool.js.map

package/dist/memory/SharedMemoryPool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SharedMemoryPool.js","sourceRoot":"","sources":["../../src/memory/SharedMemoryPool.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAgB3C,MAAM,eAAe,GAAsC;IACzD,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,eAAe,EAAE,kBAAkB,CAAC;IAC5D,cAAc,EAAE,yBAAyB;IACzC,kBAAkB,EAAE,GAAG;IACvB,iBAAiB,EAAE,cAAc;CAClC,CAAC;AAoBF,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAC,SAAS,GAA4B,IAAI,CAAC;IAExC,OAAO,CAAoC;IACpD,EAAE,GAA0B,IAAI,CAAC;IACjC,QAAQ,GAA0B,IAAI,CAAC;IACvC,WAAW,GAAyB,IAAI,CAAC;IAEjD,0EAA0E;IAC1E,iEAAiE;IACzD,KAAK,GAAG,IAAI,GAAG,EAA+B,CAAC;IAC/C,UAAU,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IAE1D,YAAoB,UAAmC,EAAE;QACvD,IAAI,CAAC,OAAO,GAAG,EAAE,GAAG,eAAe,EAAE,GAAG,OAAO,EAAE,CAAC;IACpD,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,WAAW,CAAC,OAAiC;QAClD,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAChC,gBAAgB,CAAC,SAAS,GAAG,IAAI,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,gBAAgB,CAAC,SAAS,CAAC;IACpC,CAAC;IAED,qDAAqD;IACrD,MAAM,CAAC,KAAK;QACV,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAC/B,gBAAgB,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACrC,CAAC;QACD,gBAAgB,CAAC,SAAS,GAAG,IAAI,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,iBAAiB;QACrB,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QACrC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACvC,CAAC;QACD,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,2EAA2E;QAC3E,2DAA2D;QAC3D,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAE3C,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC5C,0DAA0D;QAC1D,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QACrC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACpC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC;QAEvC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE1B,IAAI,CAAC,QAAQ,GAAG,IAAI,gBAAgB,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;YAClC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB;YAC1C,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;SACzC,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;IACnC,CAAC;IAED,4EAA4E;IACpE,WAAW,CAAC,EAAkB;QACpC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAgEP,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,WAAW;QACT,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,6EAA6E,CAC9E,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,WAAW;QACT,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CACb,6EAA6E,CAC9E,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;;;OAIG;IACH,UAAU,CAAI,GAAW,EAAE,KAAQ,EAAE,KAAa;QAChD,IAAI,CAAC,GAAG,IAAI,KAAK,IAAI,CAAC;YAAE,OAAO;QAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE;YAClB,KAAK,EAAE,KAAgB;YACvB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;SAC9B,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,8DAA8D;IAC9D,cAAc,CAAU,GAAW;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACzB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACvB,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;YAC5B,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACzB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;QACvB,OAAO,KAAK,CAAC,KAAU,CAAC;IAC1B,CAAC;IAED,qCAAqC;IACrC,eAAe;QACb,IAAI,CAAC,UAAU,CAAC,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;QAC7C,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,yDAAyD;IACzD,QAAQ;QACN,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI;YACvD,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC3B,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;YAC3C,kBAAkB,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB;YACnD,KAAK,EAAE;gBACL,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACxB,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;gBAC1B,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM;gBAC9B,SAAS,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS;aACrC;SACF,CAAC;IACJ,CAAC;IAED,mEAAmE;IACnE,KAAK;QACH,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,uCAAuC;QACzC,CAAC;QACD,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACf,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IACzD,CAAC;;AAGH,KAAK,UAAU,iBAAiB;IAC9B,6EAA6E;IAC7E,sEAAsE;IACtE,IAAI,CAAC;QACH,MAAM,GAAG,GAAQ,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAChD,OAAO,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC;IAC5B,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CACb,0EAA0E,GAAG,EAAE,OAAO,IAAI,GAAG,IAAI;YAC/F,6CAA6C,CAChD,CAAC;IACJ,CAAC;AACH,CAAC","sourcesContent":["/**\n * SharedMemoryPool — singleton resource pool that backs HybridReasoningBank\n * and AdvancedMemorySystem.\n *\n * Centralises a single SQLite database handle (better-sqlite3) and a single\n * `EmbeddingService` instance so multiple memory consumers share the same\n * underlying tables / embedding cache without conflicting writes.\n *\n * The pool is constructed lazily on first `getInstance()` call. All heavy\n * resources (sqlite handle, embedder pipeline) are created on demand inside\n * `ensureInitialized()` so simply importing this module is side-effect free.\n *\n * Used by:\n *   - reasoningbank/HybridBackend.ts   (HybridReasoningBank)\n *   - reasoningbank/AdvancedMemory.ts  (AdvancedMemorySystem)\n *\n * Fixes issue #102 — this file was previously imported but missing on disk,\n * which broke `import 'agentic-flow'` at the top level.\n */\n\nimport { mkdirSync } from 'node:fs';\nimport { dirname, join } from 'node:path';\nimport { homedir } from 'node:os';\nimport { EmbeddingService } from 'agentdb';\n\ntype DatabaseHandle = any;\ntype EmbedderHandle = EmbeddingService;\n\nexport interface SharedMemoryPoolOptions {\n  /** SQLite database path. Defaults to `~/.agentic-flow/reasoningbank.db`. */\n  dbPath?: string;\n  /** Embedding model id. Defaults to `Xenova/all-MiniLM-L6-v2`. */\n  embeddingModel?: string;\n  /** Embedding vector dimension. Defaults to 384 (MiniLM-L6). */\n  embeddingDimension?: number;\n  /** Embedding provider. Defaults to `'transformers'`. */\n  embeddingProvider?: 'transformers' | 'openai' | 'local';\n}\n\nconst DEFAULT_OPTIONS: Required<SharedMemoryPoolOptions> = {\n  dbPath: join(homedir(), '.agentic-flow', 'reasoningbank.db'),\n  embeddingModel: 'Xenova/all-MiniLM-L6-v2',\n  embeddingDimension: 384,\n  embeddingProvider: 'transformers',\n};\n\ninterface CacheEntry<T> {\n  value: T;\n  expiresAt: number;\n}\n\nexport interface SharedMemoryPoolStats {\n  initialized: boolean;\n  dbPath: string;\n  embeddingModel: string;\n  embeddingDimension: number;\n  cache: {\n    entries: number;\n    hits: number;\n    misses: number;\n    evictions: number;\n  };\n}\n\nexport class SharedMemoryPool {\n  private static _instance: SharedMemoryPool | null = null;\n\n  private readonly options: Required<SharedMemoryPoolOptions>;\n  private db: DatabaseHandle | null = null;\n  private embedder: EmbedderHandle | null = null;\n  private initPromise: Promise<void> | null = null;\n\n  // Lightweight in-process query cache used by HybridReasoningBank to avoid\n  // re-running the same retrieval across consumers in one session.\n  private cache = new Map<string, CacheEntry<unknown>>();\n  private cacheStats = { hits: 0, misses: 0, evictions: 0 };\n\n  private constructor(options: SharedMemoryPoolOptions = {}) {\n    this.options = { ...DEFAULT_OPTIONS, ...options };\n  }\n\n  /**\n   * Singleton accessor. The first caller wins for option overrides; later\n   * callers always get the existing pool. To reconfigure, call `reset()`.\n   */\n  static getInstance(options?: SharedMemoryPoolOptions): SharedMemoryPool {\n    if (!SharedMemoryPool._instance) {\n      SharedMemoryPool._instance = new SharedMemoryPool(options);\n    }\n    return SharedMemoryPool._instance;\n  }\n\n  /** Tear down the singleton — primarily for tests. */\n  static reset(): void {\n    if (SharedMemoryPool._instance) {\n      SharedMemoryPool._instance.close();\n    }\n    SharedMemoryPool._instance = null;\n  }\n\n  /**\n   * Idempotently ensure the database and embedder are ready. Subsequent calls\n   * return the same in-flight promise so concurrent consumers share init.\n   */\n  async ensureInitialized(): Promise<void> {\n    if (this.db && this.embedder) return;\n    if (!this.initPromise) {\n      this.initPromise = this.initialize();\n    }\n    return this.initPromise;\n  }\n\n  private async initialize(): Promise<void> {\n    // Dynamic import keeps node:fs, better-sqlite3, and the embedding pipeline\n    // out of the import graph for callers that never use them.\n    const Database = await loadBetterSqlite3();\n\n    mkdirSync(dirname(this.options.dbPath), { recursive: true });\n    this.db = new Database(this.options.dbPath);\n    // Reasonable defaults for an embedded reasoning database.\n    this.db.pragma('journal_mode = WAL');\n    this.db.pragma('foreign_keys = ON');\n    this.db.pragma('synchronous = NORMAL');\n\n    this.applySchema(this.db);\n\n    this.embedder = new EmbeddingService({\n      model: this.options.embeddingModel,\n      dimension: this.options.embeddingDimension,\n      provider: this.options.embeddingProvider,\n    });\n    await this.embedder.initialize();\n  }\n\n  /** Apply the minimum schema that ReflexionMemory / SkillLibrary require. */\n  private applySchema(db: DatabaseHandle): void {\n    db.exec(`\n      CREATE TABLE IF NOT EXISTS episodes (\n        id           INTEGER PRIMARY KEY AUTOINCREMENT,\n        ts           INTEGER NOT NULL DEFAULT (strftime('%s', 'now')),\n        session_id   TEXT NOT NULL,\n        task         TEXT NOT NULL,\n        input        TEXT,\n        output       TEXT,\n        critique     TEXT,\n        reward       REAL NOT NULL,\n        success      INTEGER NOT NULL,\n        latency_ms   INTEGER,\n        tokens_used  INTEGER,\n        tags         TEXT,\n        metadata     TEXT\n      );\n\n      CREATE TABLE IF NOT EXISTS episode_embeddings (\n        episode_id   INTEGER PRIMARY KEY,\n        embedding    BLOB NOT NULL,\n        FOREIGN KEY (episode_id) REFERENCES episodes(id) ON DELETE CASCADE\n      );\n\n      CREATE INDEX IF NOT EXISTS idx_episodes_task    ON episodes(task);\n      CREATE INDEX IF NOT EXISTS idx_episodes_ts      ON episodes(ts);\n      CREATE INDEX IF NOT EXISTS idx_episodes_success ON episodes(success);\n\n      CREATE TABLE IF NOT EXISTS skills (\n        id           INTEGER PRIMARY KEY AUTOINCREMENT,\n        ts           INTEGER NOT NULL DEFAULT (strftime('%s', 'now')),\n        name         TEXT NOT NULL,\n        description  TEXT,\n        precondition TEXT,\n        action       TEXT,\n        outcome      TEXT,\n        success_rate REAL,\n        uses         INTEGER DEFAULT 0,\n        tags         TEXT,\n        metadata     TEXT,\n        UNIQUE(name)\n      );\n\n      CREATE TABLE IF NOT EXISTS skill_embeddings (\n        skill_id     INTEGER PRIMARY KEY,\n        embedding    BLOB NOT NULL,\n        FOREIGN KEY (skill_id) REFERENCES skills(id) ON DELETE CASCADE\n      );\n\n      CREATE TABLE IF NOT EXISTS causal_edges (\n        id              INTEGER PRIMARY KEY AUTOINCREMENT,\n        ts              INTEGER NOT NULL DEFAULT (strftime('%s', 'now')),\n        from_memory_id   INTEGER NOT NULL,\n        from_memory_type TEXT NOT NULL,\n        to_memory_id     INTEGER NOT NULL,\n        to_memory_type   TEXT NOT NULL,\n        similarity      REAL,\n        uplift          REAL,\n        confidence      REAL,\n        sample_size     INTEGER,\n        metadata        TEXT\n      );\n\n      CREATE INDEX IF NOT EXISTS idx_causal_from ON causal_edges(from_memory_id, from_memory_type);\n      CREATE INDEX IF NOT EXISTS idx_causal_to   ON causal_edges(to_memory_id, to_memory_type);\n    `);\n  }\n\n  /**\n   * Synchronous accessor for the database handle. Throws if init hasn't run —\n   * call `ensureInitialized()` first. Provided for compatibility with\n   * controllers that take a `Database` instance in their constructor.\n   */\n  getDatabase(): DatabaseHandle {\n    if (!this.db) {\n      throw new Error(\n        'SharedMemoryPool: database not initialised. Call ensureInitialized() first.'\n      );\n    }\n    return this.db;\n  }\n\n  /**\n   * Synchronous accessor for the embedder. Throws if init hasn't run.\n   */\n  getEmbedder(): EmbedderHandle {\n    if (!this.embedder) {\n      throw new Error(\n        'SharedMemoryPool: embedder not initialised. Call ensureInitialized() first.'\n      );\n    }\n    return this.embedder;\n  }\n\n  /**\n   * Cache a query result with a TTL (milliseconds). Keys are arbitrary\n   * strings; consumers (HybridReasoningBank) typically encode the query\n   * shape into the key.\n   */\n  cacheQuery<T>(key: string, value: T, ttlMs: number): void {\n    if (!key || ttlMs <= 0) return;\n    this.cache.set(key, {\n      value: value as unknown,\n      expiresAt: Date.now() + ttlMs,\n    });\n  }\n\n  /**\n   * Read a cached query result. Returns the cached value if present and not\n   * expired; lazily evicts expired entries on lookup.\n   *\n   * The default `T` is `any` for ergonomic interop with the existing\n   * HybridReasoningBank call sites that expect a loose return type.\n   * Pass an explicit type parameter (`getCachedQuery<MyShape>(...)`) when\n   * you want stricter typing.\n   */\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  getCachedQuery<T = any>(key: string): T | undefined {\n    const entry = this.cache.get(key);\n    if (!entry) {\n      this.cacheStats.misses++;\n      return undefined;\n    }\n    if (entry.expiresAt < Date.now()) {\n      this.cache.delete(key);\n      this.cacheStats.evictions++;\n      this.cacheStats.misses++;\n      return undefined;\n    }\n    this.cacheStats.hits++;\n    return entry.value as T;\n  }\n\n  /** Drop all cached query results. */\n  invalidateCache(): void {\n    this.cacheStats.evictions += this.cache.size;\n    this.cache.clear();\n  }\n\n  /** Diagnostic stats for telemetry / health endpoints. */\n  getStats(): SharedMemoryPoolStats {\n    return {\n      initialized: this.db !== null && this.embedder !== null,\n      dbPath: this.options.dbPath,\n      embeddingModel: this.options.embeddingModel,\n      embeddingDimension: this.options.embeddingDimension,\n      cache: {\n        entries: this.cache.size,\n        hits: this.cacheStats.hits,\n        misses: this.cacheStats.misses,\n        evictions: this.cacheStats.evictions,\n      },\n    };\n  }\n\n  /** Close the underlying database handle and clear cached state. */\n  close(): void {\n    try {\n      this.db?.close?.();\n    } catch {\n      /* swallow — closing twice is benign */\n    }\n    this.db = null;\n    this.embedder = null;\n    this.initPromise = null;\n    this.cache.clear();\n    this.cacheStats = { hits: 0, misses: 0, evictions: 0 };\n  }\n}\n\nasync function loadBetterSqlite3(): Promise<any> {\n  // better-sqlite3 is a heavy native module; load it lazily so the rest of the\n  // package can be imported even when this optional dep is unavailable.\n  try {\n    const mod: any = await import('better-sqlite3');\n    return mod.default ?? mod;\n  } catch (err: any) {\n    throw new Error(\n      `SharedMemoryPool requires 'better-sqlite3' but it could not be loaded: ${err?.message || err}. ` +\n        `Install it with: npm install better-sqlite3`\n    );\n  }\n}\n"]}

package/dist/optimizations/agent-booster-migration.d.ts

@@ -38,9 +38,9 @@ interface EditResult {
 export declare class AgentBoosterMigration {
     private config;
     private boosterEngine;
+    private boosterEnginePromise;
     private stats;
     constructor(config?: Partial<AgentBoosterConfig>);
-    private ensureEngine;
     /**
      * Perform code edit using Agent Booster
      */
@@ -52,6 +52,7 @@ export declare class AgentBoosterMigration {
     /**
      * Edit using Agent Booster (352x faster)
      */
+    private getBoosterEngine;
     private editWithAgentBooster;
     /**
      * Traditional code edit (slow - 352ms average)

package/dist/optimizations/agent-booster-migration.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-booster-migration.d.ts","sourceRoot":"","sources":["../../src/optimizations/agent-booster-migration.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAkBH,UAAU,kBAAkB;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,OAAO,GAAG,QAAQ,CAAC;IAC/B,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,kBAAkB,EAAE,MAAM,EAAE,CAAC;IAC7B,WAAW,EAAE;QACX,mBAAmB,EAAE,MAAM,CAAC;QAC5B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,UAAU,QAAQ;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,UAAU;IAClB,OAAO,EAAE,OAAO,CAAC;IACjB,eAAe,EAAE,MAAM,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,eAAe,GAAG,aAAa,GAAG,UAAU,CAAC;IACrD,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;GAEG;AACH,qBAAa,qBAAqB;IAChC,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,aAAa,CAAa;IAClC,OAAO,CAAC,KAAK,CAMX;gBAEU,MAAM,GAAE,OAAO,CAAC,kBAAkB,CAAM;YA2BtC,YAAY;IAW1B;;OAEG;IACG,QAAQ,CAAC,IAAI,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IAcnD;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAoB1B;;OAEG;YACW,oBAAoB;IA0ClC;;OAEG;YACW,eAAe;IAyB7B;;;;OAIG;IACH,OAAO,CAAC,oBAAoB;IAM5B;;OAEG;IACH,OAAO,CAAC,eAAe;IAevB;;OAEG;IACH,QAAQ;;;;oBA9LM,MAAM;sBACJ,MAAM;0BACF,MAAM;wBACR,MAAM;qBACT,MAAM;;IA2MrB;;OAEG;IACH,cAAc,IAAI,MAAM;IAqCxB;;OAEG;IACH,OAAO,CAAC,KAAK;IAIb;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;IASzD;;OAEG;IACG,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC;QAChD,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,uBAAuB,EAAE,MAAM,CAAC;KACjC,CAAC;CAWH;AAED;;GAEG;AACH,eAAO,MAAM,qBAAqB,uBAA8B,CAAC;AAEjE;;GAEG;AACH,wBAAsB,QAAQ,CAC5B,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,UAAU,CAAC,CAOrB;AAED;;GAEG;AACH,wBAAsB,YAAY,kBAuDjC"}
+{"version":3,"file":"agent-booster-migration.d.ts","sourceRoot":"","sources":["../../src/optimizations/agent-booster-migration.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAmCH,UAAU,kBAAkB;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,OAAO,GAAG,QAAQ,CAAC;IAC/B,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,kBAAkB,EAAE,MAAM,EAAE,CAAC;IAC7B,WAAW,EAAE;QACX,mBAAmB,EAAE,MAAM,CAAC;QAC5B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,UAAU,QAAQ;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,UAAU;IAClB,OAAO,EAAE,OAAO,CAAC;IACjB,eAAe,EAAE,MAAM,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,eAAe,GAAG,aAAa,GAAG,UAAU,CAAC;IACrD,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;GAEG;AACH,qBAAa,qBAAqB;IAChC,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,aAAa,CAAmC;IACxD,OAAO,CAAC,oBAAoB,CAA4C;IACxE,OAAO,CAAC,KAAK,CAMX;gBAEU,MAAM,GAAE,OAAO,CAAC,kBAAkB,CAAM;IA6CpD;;OAEG;IACG,QAAQ,CAAC,IAAI,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IAcnD;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAoB1B;;OAEG;YACW,gBAAgB;YAahB,oBAAoB;IA8ClC;;OAEG;YACW,eAAe;IAyB7B;;;;OAIG;IACH,OAAO,CAAC,oBAAoB;IAM5B;;OAEG;IACH,OAAO,CAAC,eAAe;IAevB;;OAEG;IACH,QAAQ;;;;oBAtNM,MAAM;sBACJ,MAAM;0BACF,MAAM;wBACR,MAAM;qBACT,MAAM;;IAsOrB;;OAEG;IACH,cAAc,IAAI,MAAM;IAqCxB;;OAEG;IACH,OAAO,CAAC,KAAK;IAIb;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;IAOzD;;OAEG;IACG,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC;QAChD,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,uBAAuB,EAAE,MAAM,CAAC;KACjC,CAAC;CAWH;AAED;;GAEG;AACH,eAAO,MAAM,qBAAqB,uBAA8B,CAAC;AAEjE;;GAEG;AACH,wBAAsB,QAAQ,CAC5B,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,UAAU,CAAC,CAOrB;AAED;;GAEG;AACH,wBAAsB,YAAY,kBAuDjC"}

package/dist/optimizations/agent-booster-migration.js

@@ -9,19 +9,19 @@
  * Impact: All code editing operations
  */
 import { writeFileSync } from 'fs';
-// Optional: Agent Booster for local code editing (lazy loaded)
-let AgentBoosterEngine = null;
-async function getAgentBoosterEngine() {
-    if (AgentBoosterEngine)
-        return AgentBoosterEngine;
-    try {
-        const mod = await import('agent-booster');
-        AgentBoosterEngine = mod.AgentBooster;
-        return AgentBoosterEngine;
-    }
-    catch {
-        return null;
+let _AgentBoosterCtor = null;
+async function loadAgentBoosterCtor() {
+    if (_AgentBoosterCtor)
+        return _AgentBoosterCtor;
+    const mod = await import('agent-booster').catch((err) => {
+        throw new Error(`Optional package 'agent-booster' is not installed: ${err?.message || err}. ` +
+            `Run: npm install agent-booster`);
+    });
+    _AgentBoosterCtor = mod.AgentBooster ?? mod.default?.AgentBooster ?? mod.default;
+    if (!_AgentBoosterCtor) {
+        throw new Error("'agent-booster' loaded but does not export AgentBooster");
     }
+    return _AgentBoosterCtor;
 }
 /**
  * Agent Booster Migration Class
@@ -29,6 +29,7 @@ async function getAgentBoosterEngine() {
 export class AgentBoosterMigration {
     config;
     boosterEngine = null;
+    boosterEnginePromise = null;
     stats;
     constructor(config = {}) {
         this.config = {
@@ -37,36 +38,42 @@ export class AgentBoosterMigration {
             fallback: true,
             maxFileSize: 10 * 1024 * 1024, // 10MB
             supportedLanguages: [
-                'typescript', 'javascript', 'python', 'java', 'cpp', 'c',
-                'rust', 'go', 'ruby', 'php', 'swift', 'kotlin', 'scala',
-                'haskell', 'elixir', 'clojure', 'r', 'julia', 'dart'
+                'typescript',
+                'javascript',
+                'python',
+                'java',
+                'cpp',
+                'c',
+                'rust',
+                'go',
+                'ruby',
+                'php',
+                'swift',
+                'kotlin',
+                'scala',
+                'haskell',
+                'elixir',
+                'clojure',
+                'r',
+                'julia',
+                'dart',
             ],
             performance: {
                 targetSpeedupFactor: 352,
-                maxLatencyMs: 1
+                maxLatencyMs: 1,
             },
-            ...config
+            ...config,
         };
+        // Defer Agent Booster engine initialization until first use so the
+        // optional dependency can be loaded lazily.
         this.stats = {
             totalEdits: 0,
             boosterEdits: 0,
             traditionalEdits: 0,
             totalSavingsMs: 0,
-            costSavings: 0
+            costSavings: 0,
         };
     }
-    async ensureEngine() {
-        if (this.boosterEngine)
-            return true;
-        const Engine = await getAgentBoosterEngine();
-        if (!Engine)
-            return false;
-        this.boosterEngine = new Engine({
-            confidenceThreshold: 0.5,
-            maxChunks: 100
-        });
-        return true;
-    }
     /**
      * Perform code edit using Agent Booster
      */
@@ -104,14 +111,31 @@ export class AgentBoosterMigration {
     /**
      * Edit using Agent Booster (352x faster)
      */
+    async getBoosterEngine() {
+        if (this.boosterEngine)
+            return this.boosterEngine;
+        if (!this.boosterEnginePromise) {
+            this.boosterEnginePromise = (async () => {
+                const Ctor = await loadAgentBoosterCtor();
+                const engine = new Ctor({ confidenceThreshold: 0.5, maxChunks: 100 });
+                this.boosterEngine = engine;
+                return engine;
+            })();
+        }
+        return this.boosterEnginePromise;
+    }
     async editWithAgentBooster(edit, startTime) {
         try {
             const bytesProcessed = Buffer.byteLength(edit.newContent, 'utf8');
-            // Call REAL Agent Booster WASM engine
-            const result = await this.boosterEngine.apply({
+            // Call REAL Agent Booster WASM engine - use any for flexible signature
+            const engine = await this.getBoosterEngine();
+            const result = await engine.apply({
                 code: edit.oldContent,
                 edit: edit.newContent,
-                language: edit.language
+                language: edit.language,
+                target_filepath: edit.filePath || '',
+                instructions: edit.newContent,
+                code_edit: edit.newContent,
             });
             // Write the edit if successful
             if (result.success && edit.filePath) {
@@ -122,14 +146,14 @@ export class AgentBoosterMigration {
             const speedupFactor = traditionalTime / executionTimeMs;
             // Update stats
             this.stats.boosterEdits++;
-            this.stats.totalSavingsMs += (traditionalTime - executionTimeMs);
+            this.stats.totalSavingsMs += traditionalTime - executionTimeMs;
             this.stats.costSavings += this.calculateCostSavings(traditionalTime, executionTimeMs);
             return {
                 success: result.success,
                 executionTimeMs,
                 speedupFactor,
                 method: 'agent-booster',
-                bytesProcessed
+                bytesProcessed,
             };
         }
         catch (error) {
@@ -160,7 +184,7 @@ export class AgentBoosterMigration {
             executionTimeMs,
             speedupFactor: 1,
             method: 'traditional',
-            bytesProcessed
+            bytesProcessed,
         };
     }
     /**
@@ -204,7 +228,7 @@ export class AgentBoosterMigration {
             ...this.stats,
             avgSpeedupFactor,
             monthlySavings: monthlySavings.toFixed(2),
-            boosterAdoptionRate: ((this.stats.boosterEdits / this.stats.totalEdits) * 100).toFixed(1) + '%'
+            boosterAdoptionRate: ((this.stats.boosterEdits / this.stats.totalEdits) * 100).toFixed(1) + '%',
         };
     }
     /**
@@ -249,14 +273,14 @@ export class AgentBoosterMigration {
      * Sleep helper
      */
     sleep(ms) {
-        return new Promise(resolve => setTimeout(resolve, ms));
+        return new Promise((resolve) => setTimeout(resolve, ms));
     }
     /**
      * Batch edit multiple files
      */
     async batchEdit(edits) {
         // Process edits in parallel for maximum performance
-        const results = await Promise.all(edits.map(edit => this.editCode(edit)));
+        const results = await Promise.all(edits.map((edit) => this.editCode(edit)));
         return results;
     }
     /**
@@ -269,7 +293,7 @@ export class AgentBoosterMigration {
         return {
             filesProcessed: 1000,
             totalSpeedup: 352,
-            estimatedMonthlySavings: 240
+            estimatedMonthlySavings: 240,
         };
     }
 }
@@ -285,7 +309,7 @@ export async function editCode(filePath, oldContent, newContent, language) {
         filePath,
         oldContent,
         newContent,
-        language
+        language,
     });
 }
 /**
@@ -306,20 +330,20 @@ export async function exampleUsage() {
             filePath: '/tmp/file1.ts',
             oldContent: 'old1',
             newContent: 'new1',
-            language: 'typescript'
+            language: 'typescript',
         },
         {
             filePath: '/tmp/file2.js',
             oldContent: 'old2',
             newContent: 'new2',
-            language: 'javascript'
+            language: 'javascript',
         },
         {
             filePath: '/tmp/file3.py',
             oldContent: 'old3',
             newContent: 'new3',
-            language: 'python'
-        }
+            language: 'python',
+        },
     ];
     const batchResults = await agentBoosterMigration.batchEdit(batchEdits);
     console.log(`Batch Edit Results: ${batchResults.length} files processed`);