agentic-flow 2.0.7 → 3.0.0-alpha.2

package/dist/agentic-flow/src/reasoningbank/core/matts.js

@@ -0,0 +1,237 @@
+/**
+ * MaTTS: Memory-aware Test-Time Scaling
+ * Algorithm 5 from ReasoningBank paper
+ *
+ * Two modes:
+ * - Parallel: k independent rollouts with self-contrast aggregation
+ * - Sequential: r iterative refinements with check-and-correct
+ */
+import { readFileSync } from 'fs';
+import { join, dirname } from 'path';
+import { fileURLToPath } from 'url';
+import { ulid } from 'ulid';
+import { loadConfig } from '../utils/config.js';
+import { retrieveMemories } from './retrieve.js';
+import { judgeTrajectory } from './judge.js';
+import { distillMemories } from './distill.js';
+import { ModelRouter } from '../../router/router.js';
+import * as db from '../db/queries.js';
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+// Initialize ModelRouter once
+let routerInstance = null;
+function getRouter() {
+    if (!routerInstance) {
+        routerInstance = new ModelRouter();
+    }
+    return routerInstance;
+}
+/**
+ * Run MaTTS in parallel mode
+ * Execute k independent rollouts and aggregate via self-contrast
+ */
+export async function mattsParallel(taskFn, query, options = {}) {
+    const config = loadConfig();
+    const k = options.k || config.matts.parallel_k;
+    const runId = ulid();
+    const startTime = Date.now();
+    console.log(`[INFO] Starting MaTTS parallel mode with k=${k}`);
+    // Store MaTTS run
+    db.storeMattsRun({
+        run_id: runId,
+        task_id: options.taskId || 'matts-' + runId,
+        mode: 'parallel',
+        k,
+        status: 'running',
+        summary: undefined
+    });
+    const trajectories = [];
+    // Execute k independent rollouts
+    for (let i = 0; i < k; i++) {
+        console.log(`[INFO] MaTTS parallel rollout ${i + 1}/${k}`);
+        try {
+            const trajectory = await taskFn();
+            const verdict = await judgeTrajectory(trajectory, query);
+            trajectories.push({
+                id: ulid(),
+                verdict,
+                trajectory
+            });
+            // Store trajectory
+            db.storeTrajectory({
+                task_id: options.taskId || 'matts-' + runId,
+                agent_id: options.agentId || 'matts-agent',
+                query,
+                trajectory_json: JSON.stringify(trajectory),
+                started_at: new Date().toISOString(),
+                ended_at: new Date().toISOString(),
+                judge_label: verdict.label,
+                judge_conf: verdict.confidence,
+                judge_reasons: JSON.stringify(verdict.reasons),
+                matts_run_id: runId
+            });
+        }
+        catch (error) {
+            console.error(`[ERROR] MaTTS rollout ${i + 1} failed:`, error);
+        }
+    }
+    // Aggregate memories via self-contrast
+    const aggregatedMemories = await aggregateMemories(trajectories, query, options);
+    const successRate = trajectories.filter(t => t.verdict.label === 'Success').length / trajectories.length;
+    const duration = Date.now() - startTime;
+    console.log(`[INFO] MaTTS parallel complete: ${trajectories.length} trajectories, ${successRate * 100}% success in ${duration}ms`);
+    db.logMetric('rb.matts.parallel.duration_ms', duration);
+    db.logMetric('rb.matts.parallel.success_rate', successRate);
+    db.logMetric('rb.matts.parallel.memories', aggregatedMemories.length);
+    return {
+        runId,
+        mode: 'parallel',
+        k,
+        trajectories,
+        aggregatedMemories,
+        successRate,
+        duration
+    };
+}
+/**
+ * Run MaTTS in sequential mode
+ * Iterative refinement with check-and-correct
+ */
+export async function mattsSequential(taskFn, query, options = {}) {
+    const config = loadConfig();
+    const r = options.r || config.matts.sequential_r || config.matts.sequential_k;
+    const runId = ulid();
+    const startTime = Date.now();
+    console.log(`[INFO] Starting MaTTS sequential mode with r=${r}`);
+    db.storeMattsRun({
+        run_id: runId,
+        task_id: options.taskId || 'matts-seq-' + runId,
+        mode: 'sequential',
+        k: r,
+        status: 'running',
+        summary: undefined
+    });
+    const trajectories = [];
+    let previousMemories = [];
+    // Iterative refinement
+    for (let i = 0; i < r; i++) {
+        console.log(`[INFO] MaTTS sequential iteration ${i + 1}/${r}`);
+        try {
+            // Retrieve relevant memories (including from previous iterations)
+            const memories = await retrieveMemories(query, {
+                domain: options.domain
+            });
+            // Execute with memories
+            const trajectory = await taskFn([...memories, ...previousMemories]);
+            const verdict = await judgeTrajectory(trajectory, query);
+            trajectories.push({
+                id: ulid(),
+                verdict,
+                trajectory
+            });
+            // If success and stop_on_success is true, break early
+            if (verdict.label === 'Success' && (config.matts.sequential_stop_on_success ?? true)) {
+                console.log(`[INFO] Success achieved at iteration ${i + 1}, stopping early`);
+                break;
+            }
+            // Distill memories from this iteration
+            const newMemories = await distillMemories(trajectory, verdict, query, options);
+            previousMemories = [...previousMemories, ...newMemories];
+            // Store trajectory
+            db.storeTrajectory({
+                task_id: options.taskId || 'matts-seq-' + runId,
+                agent_id: options.agentId || 'matts-agent',
+                query,
+                trajectory_json: JSON.stringify(trajectory),
+                started_at: new Date().toISOString(),
+                ended_at: new Date().toISOString(),
+                judge_label: verdict.label,
+                judge_conf: verdict.confidence,
+                judge_reasons: JSON.stringify(verdict.reasons),
+                matts_run_id: runId
+            });
+        }
+        catch (error) {
+            console.error(`[ERROR] MaTTS iteration ${i + 1} failed:`, error);
+        }
+    }
+    const successRate = trajectories.filter(t => t.verdict.label === 'Success').length / trajectories.length;
+    const duration = Date.now() - startTime;
+    console.log(`[INFO] MaTTS sequential complete: ${trajectories.length} iterations, ${successRate * 100}% success in ${duration}ms`);
+    db.logMetric('rb.matts.sequential.duration_ms', duration);
+    db.logMetric('rb.matts.sequential.success_rate', successRate);
+    return {
+        runId,
+        mode: 'sequential',
+        k: r,
+        trajectories,
+        aggregatedMemories: previousMemories,
+        successRate,
+        duration
+    };
+}
+/**
+ * Aggregate memories from multiple trajectories using self-contrast
+ */
+async function aggregateMemories(trajectories, query, options) {
+    console.log('[INFO] Aggregating memories via self-contrast');
+    // Load aggregation prompt
+    const promptPath = join(__dirname, '../prompts', 'matts-aggregate.json');
+    const promptTemplate = JSON.parse(readFileSync(promptPath, 'utf-8'));
+    // Format trajectories for comparison
+    const trajectoryTexts = trajectories.map((t, i) => ({
+        id: t.id,
+        label: t.verdict.label,
+        confidence: t.verdict.confidence,
+        steps: JSON.stringify(t.trajectory.steps || [], null, 2)
+    }));
+    // Check if we have any API key configured
+    const hasApiKey = process.env.OPENROUTER_API_KEY ||
+        process.env.ANTHROPIC_API_KEY ||
+        process.env.GOOGLE_GEMINI_API_KEY;
+    if (!hasApiKey) {
+        console.warn('[WARN] No API key set, skipping aggregation');
+        return [];
+    }
+    try {
+        const prompt = promptTemplate.template
+            .replace('{{k}}', String(trajectories.length))
+            .replace('{{task_query}}', query)
+            .replace('{{trajectories}}', JSON.stringify(trajectoryTexts, null, 2));
+        // Use ModelRouter for multi-provider support
+        const router = getRouter();
+        const response = await router.chat({
+            model: promptTemplate.model,
+            messages: [
+                { role: 'system', content: promptTemplate.system },
+                { role: 'user', content: prompt }
+            ],
+            temperature: promptTemplate.temperature,
+            maxTokens: promptTemplate.max_tokens
+        }, 'reasoningbank-matts-aggregate');
+        // Extract content from router response
+        const content = response.content
+            .filter(block => block.type === 'text')
+            .map(block => block.text)
+            .join('\n');
+        // Parse and store aggregated memories
+        const jsonMatch = content.match(/\{[\s\S]*\}/);
+        if (jsonMatch) {
+            const parsed = JSON.parse(jsonMatch[0]);
+            const memories = parsed.memories || [];
+            // Store with boosted confidence
+            const memoryIds = [];
+            for (const mem of memories) {
+                const verdict = { label: 'Success', confidence: 0.9, reasons: [] };
+                const ids = await distillMemories({ steps: [] }, verdict, query, options);
+                memoryIds.push(...ids);
+            }
+            return memoryIds;
+        }
+    }
+    catch (error) {
+        console.error('[ERROR] Memory aggregation failed:', error);
+    }
+    return [];
+}
+//# sourceMappingURL=matts.js.map

package/dist/agentic-flow/src/reasoningbank/core/matts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"matts.js","sourceRoot":"","sources":["../../../../../src/reasoningbank/core/matts.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAGvC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC,8BAA8B;AAC9B,IAAI,cAAc,GAAuB,IAAI,CAAC;AAC9C,SAAS,SAAS;IAChB,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,cAAc,GAAG,IAAI,WAAW,EAAE,CAAC;IACrC,CAAC;IACD,OAAO,cAAc,CAAC;AACxB,CAAC;AAgBD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAiC,EACjC,KAAa,EACb,UAA8E,EAAE;IAEhF,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC;IAC/C,MAAM,KAAK,GAAG,IAAI,EAAE,CAAC;IACrB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,EAAE,CAAC,CAAC;IAE/D,kBAAkB;IAClB,EAAE,CAAC,aAAa,CAAC;QACf,MAAM,EAAE,KAAK;QACb,OAAO,EAAE,OAAO,CAAC,MAAM,IAAI,QAAQ,GAAG,KAAK;QAC3C,IAAI,EAAE,UAAU;QAChB,CAAC;QACD,MAAM,EAAE,SAAS;QACjB,OAAO,EAAE,SAAS;KACnB,CAAC,CAAC;IAEH,MAAM,YAAY,GAAgC,EAAE,CAAC;IAErD,iCAAiC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE3D,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,MAAM,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAEzD,YAAY,CAAC,IAAI,CAAC;gBAChB,EAAE,EAAE,IAAI,EAAE;gBACV,OAAO;gBACP,UAAU;aACX,CAAC,CAAC;YAEH,mBAAmB;YACnB,EAAE,CAAC,eAAe,CAAC;gBACjB,OAAO,EAAE,OAAO,CAAC,MAAM,IAAI,QAAQ,GAAG,KAAK;gBAC3C,QAAQ,EAAE,OAAO,CAAC,OAAO,IAAI,aAAa;gBAC1C,KAAK;gBACL,eAAe,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;gBAC3C,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACpC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBAClC,WAAW,EAAE,OAAO,CAAC,KAAK;gBAC1B,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;gBAC9C,YAAY,EAAE,KAAK;aACpB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED,uCAAuC;IACvC,MAAM,kBAAkB,GAAG,MAAM,iBAAiB,CAAC,YAAY,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;IAEjF,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;IACzG,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IAExC,OAAO,CAAC,GAAG,CAAC,mCAAmC,YAAY,CAAC,MAAM,kBAAkB,WAAW,GAAG,GAAG,gBAAgB,QAAQ,IAAI,CAAC,CAAC;IAEnI,EAAE,CAAC,SAAS,CAAC,+BAA+B,EAAE,QAAQ,CAAC,CAAC;IACxD,EAAE,CAAC,SAAS,CAAC,gCAAgC,EAAE,WAAW,CAAC,CAAC;IAC5D,EAAE,CAAC,SAAS,CAAC,4BAA4B,EAAE,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAEtE,OAAO;QACL,KAAK;QACL,IAAI,EAAE,UAAU;QAChB,CAAC;QACD,YAAY;QACZ,kBAAkB;QAClB,WAAW;QACX,QAAQ;KACT,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,MAAgD,EAChD,KAAa,EACb,UAA8E,EAAE;IAEhF,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,YAAY,IAAI,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC;IAC9E,MAAM,KAAK,GAAG,IAAI,EAAE,CAAC;IACrB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,EAAE,CAAC,CAAC;IAEjE,EAAE,CAAC,aAAa,CAAC;QACf,MAAM,EAAE,KAAK;QACb,OAAO,EAAE,OAAO,CAAC,MAAM,IAAI,YAAY,GAAG,KAAK;QAC/C,IAAI,EAAE,YAAY;QAClB,CAAC,EAAE,CAAC;QACJ,MAAM,EAAE,SAAS;QACjB,OAAO,EAAE,SAAS;KACnB,CAAC,CAAC;IAEH,MAAM,YAAY,GAAgC,EAAE,CAAC;IACrD,IAAI,gBAAgB,GAAU,EAAE,CAAC;IAEjC,uBAAuB;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE/D,IAAI,CAAC;YACH,kEAAkE;YAClE,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,KAAK,EAAE;gBAC7C,MAAM,EAAE,OAAO,CAAC,MAAM;aACvB,CAAC,CAAC;YAEH,wBAAwB;YACxB,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,CAAC,GAAG,QAAQ,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC;YACpE,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAEzD,YAAY,CAAC,IAAI,CAAC;gBAChB,EAAE,EAAE,IAAI,EAAE;gBACV,OAAO;gBACP,UAAU;aACX,CAAC,CAAC;YAEH,sDAAsD;YACtD,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,IAAI,CAAC,EAAE,CAAC;gBACrF,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;gBAC7E,MAAM;YACR,CAAC;YAED,uCAAuC;YACvC,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YAC/E,gBAAgB,GAAG,CAAC,GAAG,gBAAgB,EAAE,GAAG,WAAW,CAAC,CAAC;YAEzD,mBAAmB;YACnB,EAAE,CAAC,eAAe,CAAC;gBACjB,OAAO,EAAE,OAAO,CAAC,MAAM,IAAI,YAAY,GAAG,KAAK;gBAC/C,QAAQ,EAAE,OAAO,CAAC,OAAO,IAAI,aAAa;gBAC1C,KAAK;gBACL,eAAe,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;gBAC3C,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACpC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBAClC,WAAW,EAAE,OAAO,CAAC,KAAK;gBAC1B,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;gBAC9C,YAAY,EAAE,KAAK;aACpB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;IACzG,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IAExC,OAAO,CAAC,GAAG,CAAC,qCAAqC,YAAY,CAAC,MAAM,gBAAgB,WAAW,GAAG,GAAG,gBAAgB,QAAQ,IAAI,CAAC,CAAC;IAEnI,EAAE,CAAC,SAAS,CAAC,iCAAiC,EAAE,QAAQ,CAAC,CAAC;IAC1D,EAAE,CAAC,SAAS,CAAC,kCAAkC,EAAE,WAAW,CAAC,CAAC;IAE9D,OAAO;QACL,KAAK;QACL,IAAI,EAAE,YAAY;QAClB,CAAC,EAAE,CAAC;QACJ,YAAY;QACZ,kBAAkB,EAAE,gBAAgB;QACpC,WAAW;QACX,QAAQ;KACT,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAC9B,YAAyC,EACzC,KAAa,EACb,OAAY;IAEZ,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAE7D,0BAA0B;IAC1B,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,YAAY,EAAE,sBAAsB,CAAC,CAAC;IACzE,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;IAErE,qCAAqC;IACrC,MAAM,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAClD,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK;QACtB,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU;QAChC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;KACzD,CAAC,CAAC,CAAC;IAEJ,0CAA0C;IAC1C,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB;QAC9B,OAAO,CAAC,GAAG,CAAC,iBAAiB;QAC7B,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IAEpD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC5D,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ;aACnC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;aAC7C,OAAO,CAAC,gBAAgB,EAAE,KAAK,CAAC;aAChC,OAAO,CAAC,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAEzE,6CAA6C;QAC7C,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;YACjC,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,cAAc,CAAC,MAAM,EAAE;gBAClD,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;aAClC;YACD,WAAW,EAAE,cAAc,CAAC,WAAW;YACvC,SAAS,EAAE,cAAc,CAAC,UAAU;SACrC,EAAE,+BAA+B,CAAC,CAAC;QAEpC,uCAAuC;QACvC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO;aAC7B,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;aACtC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;aACxB,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,sCAAsC;QACtC,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC/C,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;YAEvC,gCAAgC;YAChC,MAAM,SAAS,GAAa,EAAE,CAAC;YAC/B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,SAAkB,EAAE,UAAU,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBAC5E,MAAM,GAAG,GAAG,MAAM,eAAe,CAC/B,EAAE,KAAK,EAAE,EAAE,EAAE,EACb,OAAO,EACP,KAAK,EACL,OAAO,CACR,CAAC;gBACF,SAAS,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC;YACzB,CAAC;YAED,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;IAC7D,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC","sourcesContent":["/**\n * MaTTS: Memory-aware Test-Time Scaling\n * Algorithm 5 from ReasoningBank paper\n *\n * Two modes:\n * - Parallel: k independent rollouts with self-contrast aggregation\n * - Sequential: r iterative refinements with check-and-correct\n */\n\nimport { readFileSync } from 'fs';\nimport { join, dirname } from 'path';\nimport { fileURLToPath } from 'url';\nimport { ulid } from 'ulid';\nimport { loadConfig } from '../utils/config.js';\nimport { retrieveMemories } from './retrieve.js';\nimport { judgeTrajectory } from './judge.js';\nimport { distillMemories } from './distill.js';\nimport { ModelRouter } from '../../router/router.js';\nimport * as db from '../db/queries.js';\nimport type { Trajectory } from '../db/schema.js';\n\nconst __filename = fileURLToPath(import.meta.url);\nconst __dirname = dirname(__filename);\n\n// Initialize ModelRouter once\nlet routerInstance: ModelRouter | null = null;\nfunction getRouter(): ModelRouter {\n  if (!routerInstance) {\n    routerInstance = new ModelRouter();\n  }\n  return routerInstance;\n}\n\nexport interface MattsResult {\n  runId: string;\n  mode: 'parallel' | 'sequential';\n  k: number;\n  trajectories: Array<{\n    id: string;\n    verdict: { label: string; confidence: number };\n    trajectory: Trajectory;\n  }>;\n  aggregatedMemories: string[];\n  successRate: number;\n  duration: number;\n}\n\n/**\n * Run MaTTS in parallel mode\n * Execute k independent rollouts and aggregate via self-contrast\n */\nexport async function mattsParallel(\n  taskFn: () => Promise<Trajectory>,\n  query: string,\n  options: { k?: number; taskId?: string; agentId?: string; domain?: string } = {}\n): Promise<MattsResult> {\n  const config = loadConfig();\n  const k = options.k || config.matts.parallel_k;\n  const runId = ulid();\n  const startTime = Date.now();\n\n  console.log(`[INFO] Starting MaTTS parallel mode with k=${k}`);\n\n  // Store MaTTS run\n  db.storeMattsRun({\n    run_id: runId,\n    task_id: options.taskId || 'matts-' + runId,\n    mode: 'parallel',\n    k,\n    status: 'running',\n    summary: undefined\n  });\n\n  const trajectories: MattsResult['trajectories'] = [];\n\n  // Execute k independent rollouts\n  for (let i = 0; i < k; i++) {\n    console.log(`[INFO] MaTTS parallel rollout ${i + 1}/${k}`);\n\n    try {\n      const trajectory = await taskFn();\n      const verdict = await judgeTrajectory(trajectory, query);\n\n      trajectories.push({\n        id: ulid(),\n        verdict,\n        trajectory\n      });\n\n      // Store trajectory\n      db.storeTrajectory({\n        task_id: options.taskId || 'matts-' + runId,\n        agent_id: options.agentId || 'matts-agent',\n        query,\n        trajectory_json: JSON.stringify(trajectory),\n        started_at: new Date().toISOString(),\n        ended_at: new Date().toISOString(),\n        judge_label: verdict.label,\n        judge_conf: verdict.confidence,\n        judge_reasons: JSON.stringify(verdict.reasons),\n        matts_run_id: runId\n      });\n    } catch (error) {\n      console.error(`[ERROR] MaTTS rollout ${i + 1} failed:`, error);\n    }\n  }\n\n  // Aggregate memories via self-contrast\n  const aggregatedMemories = await aggregateMemories(trajectories, query, options);\n\n  const successRate = trajectories.filter(t => t.verdict.label === 'Success').length / trajectories.length;\n  const duration = Date.now() - startTime;\n\n  console.log(`[INFO] MaTTS parallel complete: ${trajectories.length} trajectories, ${successRate * 100}% success in ${duration}ms`);\n\n  db.logMetric('rb.matts.parallel.duration_ms', duration);\n  db.logMetric('rb.matts.parallel.success_rate', successRate);\n  db.logMetric('rb.matts.parallel.memories', aggregatedMemories.length);\n\n  return {\n    runId,\n    mode: 'parallel',\n    k,\n    trajectories,\n    aggregatedMemories,\n    successRate,\n    duration\n  };\n}\n\n/**\n * Run MaTTS in sequential mode\n * Iterative refinement with check-and-correct\n */\nexport async function mattsSequential(\n  taskFn: (memories: any[]) => Promise<Trajectory>,\n  query: string,\n  options: { r?: number; taskId?: string; agentId?: string; domain?: string } = {}\n): Promise<MattsResult> {\n  const config = loadConfig();\n  const r = options.r || config.matts.sequential_r || config.matts.sequential_k;\n  const runId = ulid();\n  const startTime = Date.now();\n\n  console.log(`[INFO] Starting MaTTS sequential mode with r=${r}`);\n\n  db.storeMattsRun({\n    run_id: runId,\n    task_id: options.taskId || 'matts-seq-' + runId,\n    mode: 'sequential',\n    k: r,\n    status: 'running',\n    summary: undefined\n  });\n\n  const trajectories: MattsResult['trajectories'] = [];\n  let previousMemories: any[] = [];\n\n  // Iterative refinement\n  for (let i = 0; i < r; i++) {\n    console.log(`[INFO] MaTTS sequential iteration ${i + 1}/${r}`);\n\n    try {\n      // Retrieve relevant memories (including from previous iterations)\n      const memories = await retrieveMemories(query, {\n        domain: options.domain\n      });\n\n      // Execute with memories\n      const trajectory = await taskFn([...memories, ...previousMemories]);\n      const verdict = await judgeTrajectory(trajectory, query);\n\n      trajectories.push({\n        id: ulid(),\n        verdict,\n        trajectory\n      });\n\n      // If success and stop_on_success is true, break early\n      if (verdict.label === 'Success' && (config.matts.sequential_stop_on_success ?? true)) {\n        console.log(`[INFO] Success achieved at iteration ${i + 1}, stopping early`);\n        break;\n      }\n\n      // Distill memories from this iteration\n      const newMemories = await distillMemories(trajectory, verdict, query, options);\n      previousMemories = [...previousMemories, ...newMemories];\n\n      // Store trajectory\n      db.storeTrajectory({\n        task_id: options.taskId || 'matts-seq-' + runId,\n        agent_id: options.agentId || 'matts-agent',\n        query,\n        trajectory_json: JSON.stringify(trajectory),\n        started_at: new Date().toISOString(),\n        ended_at: new Date().toISOString(),\n        judge_label: verdict.label,\n        judge_conf: verdict.confidence,\n        judge_reasons: JSON.stringify(verdict.reasons),\n        matts_run_id: runId\n      });\n    } catch (error) {\n      console.error(`[ERROR] MaTTS iteration ${i + 1} failed:`, error);\n    }\n  }\n\n  const successRate = trajectories.filter(t => t.verdict.label === 'Success').length / trajectories.length;\n  const duration = Date.now() - startTime;\n\n  console.log(`[INFO] MaTTS sequential complete: ${trajectories.length} iterations, ${successRate * 100}% success in ${duration}ms`);\n\n  db.logMetric('rb.matts.sequential.duration_ms', duration);\n  db.logMetric('rb.matts.sequential.success_rate', successRate);\n\n  return {\n    runId,\n    mode: 'sequential',\n    k: r,\n    trajectories,\n    aggregatedMemories: previousMemories,\n    successRate,\n    duration\n  };\n}\n\n/**\n * Aggregate memories from multiple trajectories using self-contrast\n */\nasync function aggregateMemories(\n  trajectories: MattsResult['trajectories'],\n  query: string,\n  options: any\n): Promise<string[]> {\n  console.log('[INFO] Aggregating memories via self-contrast');\n\n  // Load aggregation prompt\n  const promptPath = join(__dirname, '../prompts', 'matts-aggregate.json');\n  const promptTemplate = JSON.parse(readFileSync(promptPath, 'utf-8'));\n\n  // Format trajectories for comparison\n  const trajectoryTexts = trajectories.map((t, i) => ({\n    id: t.id,\n    label: t.verdict.label,\n    confidence: t.verdict.confidence,\n    steps: JSON.stringify(t.trajectory.steps || [], null, 2)\n  }));\n\n  // Check if we have any API key configured\n  const hasApiKey = process.env.OPENROUTER_API_KEY ||\n                    process.env.ANTHROPIC_API_KEY ||\n                    process.env.GOOGLE_GEMINI_API_KEY;\n\n  if (!hasApiKey) {\n    console.warn('[WARN] No API key set, skipping aggregation');\n    return [];\n  }\n\n  try {\n    const prompt = promptTemplate.template\n      .replace('{{k}}', String(trajectories.length))\n      .replace('{{task_query}}', query)\n      .replace('{{trajectories}}', JSON.stringify(trajectoryTexts, null, 2));\n\n    // Use ModelRouter for multi-provider support\n    const router = getRouter();\n    const response = await router.chat({\n      model: promptTemplate.model,\n      messages: [\n        { role: 'system', content: promptTemplate.system },\n        { role: 'user', content: prompt }\n      ],\n      temperature: promptTemplate.temperature,\n      maxTokens: promptTemplate.max_tokens\n    }, 'reasoningbank-matts-aggregate');\n\n    // Extract content from router response\n    const content = response.content\n      .filter(block => block.type === 'text')\n      .map(block => block.text)\n      .join('\\n');\n\n    // Parse and store aggregated memories\n    const jsonMatch = content.match(/\\{[\\s\\S]*\\}/);\n    if (jsonMatch) {\n      const parsed = JSON.parse(jsonMatch[0]);\n      const memories = parsed.memories || [];\n\n      // Store with boosted confidence\n      const memoryIds: string[] = [];\n      for (const mem of memories) {\n        const verdict = { label: 'Success' as const, confidence: 0.9, reasons: [] };\n        const ids = await distillMemories(\n          { steps: [] },\n          verdict,\n          query,\n          options\n        );\n        memoryIds.push(...ids);\n      }\n\n      return memoryIds;\n    }\n  } catch (error) {\n    console.error('[ERROR] Memory aggregation failed:', error);\n  }\n\n  return [];\n}\n"]}

package/dist/agentic-flow/src/reasoningbank/core/retrieve.d.ts

@@ -0,0 +1,35 @@
+/**
+ * Memory Retrieval with MMR diversity
+ * Algorithm 1 from ReasoningBank paper
+ */
+export interface RetrievedMemory {
+    id: string;
+    title: string;
+    description: string;
+    content: string;
+    score: number;
+    components: {
+        similarity: number;
+        recency: number;
+        reliability: number;
+    };
+}
+/**
+ * Retrieve top-k memories with MMR diversity
+ *
+ * Scoring formula: score = α·sim + β·recency + γ·reliability
+ * Where:
+ * - sim: cosine similarity to query
+ * - recency: exp(-age_days / half_life)
+ * - reliability: min(confidence, 1.0)
+ */
+export declare function retrieveMemories(query: string, options?: {
+    k?: number;
+    domain?: string;
+    agent?: string;
+}): Promise<RetrievedMemory[]>;
+/**
+ * Format memories for injection into system prompt
+ */
+export declare function formatMemoriesForPrompt(memories: RetrievedMemory[]): string;
+//# sourceMappingURL=retrieve.d.ts.map

package/dist/agentic-flow/src/reasoningbank/core/retrieve.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieve.d.ts","sourceRoot":"","sources":["../../../../../src/reasoningbank/core/retrieve.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAOH,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE;QACV,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED;;;;;;;;GAQG;AACH,wBAAsB,gBAAgB,CACpC,KAAK,EAAE,MAAM,EACb,OAAO,GAAE;IAAE,CAAC,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAO,GAC5D,OAAO,CAAC,eAAe,EAAE,CAAC,CA8D5B;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,MAAM,CAkB3E"}

package/dist/agentic-flow/src/reasoningbank/core/retrieve.js

@@ -0,0 +1,87 @@
+/**
+ * Memory Retrieval with MMR diversity
+ * Algorithm 1 from ReasoningBank paper
+ */
+import { computeEmbedding } from '../utils/embeddings.js';
+import { mmrSelection, cosineSimilarity } from '../utils/mmr.js';
+import * as db from '../db/queries.js';
+import { loadConfig } from '../utils/config.js';
+/**
+ * Retrieve top-k memories with MMR diversity
+ *
+ * Scoring formula: score = α·sim + β·recency + γ·reliability
+ * Where:
+ * - sim: cosine similarity to query
+ * - recency: exp(-age_days / half_life)
+ * - reliability: min(confidence, 1.0)
+ */
+export async function retrieveMemories(query, options = {}) {
+    const config = loadConfig();
+    const k = options.k || config.retrieve.k;
+    const startTime = Date.now();
+    console.log(`[INFO] Retrieving memories for query: ${query.substring(0, 100)}...`);
+    // 1. Embed query
+    const queryEmbed = await computeEmbedding(query);
+    // 2. Fetch candidates from database
+    const candidates = db.fetchMemoryCandidates({
+        domain: options.domain,
+        agent: options.agent,
+        minConfidence: config.retrieve.min_score
+    });
+    if (candidates.length === 0) {
+        console.log('[INFO] No memory candidates found');
+        return [];
+    }
+    console.log(`[INFO] Found ${candidates.length} candidates`);
+    // 3. Score each candidate with 4-factor model
+    const scored = candidates.map(item => {
+        const similarity = cosineSimilarity(queryEmbed, item.embedding);
+        const recency = Math.exp(-item.age_days / config.retrieve.recency_half_life_days);
+        const reliability = Math.min(item.confidence, 1.0);
+        const baseScore = config.retrieve.alpha * similarity +
+            config.retrieve.beta * recency +
+            config.retrieve.gamma * reliability;
+        return {
+            ...item,
+            score: baseScore,
+            components: { similarity, recency, reliability }
+        };
+    });
+    // 4. MMR selection for diversity
+    const selected = mmrSelection(scored, queryEmbed, k, config.retrieve.delta);
+    // 5. Record usage for selected memories
+    for (const mem of selected) {
+        db.incrementUsage(mem.id);
+    }
+    const duration = Date.now() - startTime;
+    console.log(`[INFO] Retrieval complete: ${selected.length} memories in ${duration}ms`);
+    db.logMetric('rb.retrieve.latency_ms', duration);
+    return selected.map(item => ({
+        id: item.id,
+        title: item.pattern_data.title,
+        description: item.pattern_data.description,
+        content: item.pattern_data.content,
+        score: item.score,
+        components: item.components
+    }));
+}
+/**
+ * Format memories for injection into system prompt
+ */
+export function formatMemoriesForPrompt(memories) {
+    if (memories.length === 0) {
+        return '';
+    }
+    let formatted = '\n## Relevant Memories from Past Experience\n\n';
+    for (let i = 0; i < memories.length; i++) {
+        const mem = memories[i];
+        formatted += `### Memory ${i + 1}: ${mem.title}\n\n`;
+        formatted += `${mem.description}\n\n`;
+        formatted += `**Strategy:**\n${mem.content}\n\n`;
+        formatted += `*Confidence: ${(mem.score * 100).toFixed(1)}% | `;
+        formatted += `Similarity: ${(mem.components.similarity * 100).toFixed(1)}%*\n\n`;
+        formatted += '---\n\n';
+    }
+    return formatted;
+}
+//# sourceMappingURL=retrieve.js.map

package/dist/agentic-flow/src/reasoningbank/core/retrieve.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieve.js","sourceRoot":"","sources":["../../../../../src/reasoningbank/core/retrieve.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACjE,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAehD;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAAa,EACb,UAA2D,EAAE;IAE7D,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;IACzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,OAAO,CAAC,GAAG,CAAC,yCAAyC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;IAEnF,iBAAiB;IACjB,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,CAAC;IAEjD,oCAAoC;IACpC,MAAM,UAAU,GAAG,EAAE,CAAC,qBAAqB,CAAC;QAC1C,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,aAAa,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;KACzC,CAAC,CAAC;IAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;QACjD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,gBAAgB,UAAU,CAAC,MAAM,aAAa,CAAC,CAAC;IAE5D,8CAA8C;IAC9C,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QACnC,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;QAClF,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAEnD,MAAM,SAAS,GACb,MAAM,CAAC,QAAQ,CAAC,KAAK,GAAG,UAAU;YAClC,MAAM,CAAC,QAAQ,CAAC,IAAI,GAAG,OAAO;YAC9B,MAAM,CAAC,QAAQ,CAAC,KAAK,GAAG,WAAW,CAAC;QAEtC,OAAO;YACL,GAAG,IAAI;YACP,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,EAAE;SACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,iCAAiC;IACjC,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAE5E,wCAAwC;IACxC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,EAAE,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,8BAA8B,QAAQ,CAAC,MAAM,gBAAgB,QAAQ,IAAI,CAAC,CAAC;IACvF,EAAE,CAAC,SAAS,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC;IAEjD,OAAO,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,KAAK,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK;QAC9B,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,WAAW;QAC1C,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;QAClC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,UAAU,EAAE,IAAI,CAAC,UAAU;KAC5B,CAAC,CAAC,CAAC;AACN,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,QAA2B;IACjE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,SAAS,GAAG,iDAAiD,CAAC;IAElE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACxB,SAAS,IAAI,cAAc,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,KAAK,MAAM,CAAC;QACrD,SAAS,IAAI,GAAG,GAAG,CAAC,WAAW,MAAM,CAAC;QACtC,SAAS,IAAI,kBAAkB,GAAG,CAAC,OAAO,MAAM,CAAC;QACjD,SAAS,IAAI,gBAAgB,CAAC,GAAG,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;QAChE,SAAS,IAAI,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC;QACjF,SAAS,IAAI,SAAS,CAAC;IACzB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC","sourcesContent":["/**\n * Memory Retrieval with MMR diversity\n * Algorithm 1 from ReasoningBank paper\n */\n\nimport { computeEmbedding } from '../utils/embeddings.js';\nimport { mmrSelection, cosineSimilarity } from '../utils/mmr.js';\nimport * as db from '../db/queries.js';\nimport { loadConfig } from '../utils/config.js';\n\nexport interface RetrievedMemory {\n  id: string;\n  title: string;\n  description: string;\n  content: string;\n  score: number;\n  components: {\n    similarity: number;\n    recency: number;\n    reliability: number;\n  };\n}\n\n/**\n * Retrieve top-k memories with MMR diversity\n *\n * Scoring formula: score = α·sim + β·recency + γ·reliability\n * Where:\n * - sim: cosine similarity to query\n * - recency: exp(-age_days / half_life)\n * - reliability: min(confidence, 1.0)\n */\nexport async function retrieveMemories(\n  query: string,\n  options: { k?: number; domain?: string; agent?: string } = {}\n): Promise<RetrievedMemory[]> {\n  const config = loadConfig();\n  const k = options.k || config.retrieve.k;\n  const startTime = Date.now();\n\n  console.log(`[INFO] Retrieving memories for query: ${query.substring(0, 100)}...`);\n\n  // 1. Embed query\n  const queryEmbed = await computeEmbedding(query);\n\n  // 2. Fetch candidates from database\n  const candidates = db.fetchMemoryCandidates({\n    domain: options.domain,\n    agent: options.agent,\n    minConfidence: config.retrieve.min_score\n  });\n\n  if (candidates.length === 0) {\n    console.log('[INFO] No memory candidates found');\n    return [];\n  }\n\n  console.log(`[INFO] Found ${candidates.length} candidates`);\n\n  // 3. Score each candidate with 4-factor model\n  const scored = candidates.map(item => {\n    const similarity = cosineSimilarity(queryEmbed, item.embedding);\n    const recency = Math.exp(-item.age_days / config.retrieve.recency_half_life_days);\n    const reliability = Math.min(item.confidence, 1.0);\n\n    const baseScore =\n      config.retrieve.alpha * similarity +\n      config.retrieve.beta * recency +\n      config.retrieve.gamma * reliability;\n\n    return {\n      ...item,\n      score: baseScore,\n      components: { similarity, recency, reliability }\n    };\n  });\n\n  // 4. MMR selection for diversity\n  const selected = mmrSelection(scored, queryEmbed, k, config.retrieve.delta);\n\n  // 5. Record usage for selected memories\n  for (const mem of selected) {\n    db.incrementUsage(mem.id);\n  }\n\n  const duration = Date.now() - startTime;\n  console.log(`[INFO] Retrieval complete: ${selected.length} memories in ${duration}ms`);\n  db.logMetric('rb.retrieve.latency_ms', duration);\n\n  return selected.map(item => ({\n    id: item.id,\n    title: item.pattern_data.title,\n    description: item.pattern_data.description,\n    content: item.pattern_data.content,\n    score: item.score,\n    components: item.components\n  }));\n}\n\n/**\n * Format memories for injection into system prompt\n */\nexport function formatMemoriesForPrompt(memories: RetrievedMemory[]): string {\n  if (memories.length === 0) {\n    return '';\n  }\n\n  let formatted = '\\n## Relevant Memories from Past Experience\\n\\n';\n\n  for (let i = 0; i < memories.length; i++) {\n    const mem = memories[i];\n    formatted += `### Memory ${i + 1}: ${mem.title}\\n\\n`;\n    formatted += `${mem.description}\\n\\n`;\n    formatted += `**Strategy:**\\n${mem.content}\\n\\n`;\n    formatted += `*Confidence: ${(mem.score * 100).toFixed(1)}% | `;\n    formatted += `Similarity: ${(mem.components.similarity * 100).toFixed(1)}%*\\n\\n`;\n    formatted += '---\\n\\n';\n  }\n\n  return formatted;\n}\n"]}

package/dist/agentic-flow/src/reasoningbank/db/queries.d.ts

@@ -0,0 +1,88 @@
+/**
+ * Database queries for ReasoningBank
+ * Operates on Claude Flow's memory.db at .swarm/memory.db
+ */
+import Database from '../../db/sql-adapter.js';
+import type { ReasoningMemory, PatternEmbedding, TaskTrajectory, MattsRun } from './schema.js';
+/**
+ * Run database migrations (create tables)
+ */
+export declare function runMigrations(): Promise<void>;
+/**
+ * Get database connection (singleton)
+ */
+export declare function getDb(): Database;
+/**
+ * Fetch reasoning memory candidates for retrieval
+ */
+export declare function fetchMemoryCandidates(options: {
+    domain?: string;
+    agent?: string;
+    minConfidence?: number;
+}): Array<ReasoningMemory & {
+    embedding: Float32Array;
+    age_days: number;
+}>;
+/**
+ * Store a new reasoning memory
+ */
+export declare function upsertMemory(memory: Omit<ReasoningMemory, 'created_at' | 'last_used'>): string;
+/**
+ * Store embedding for a memory
+ */
+export declare function upsertEmbedding(embedding: PatternEmbedding): void;
+/**
+ * Increment usage count for a memory
+ */
+export declare function incrementUsage(memoryId: string): void;
+/**
+ * Store task trajectory
+ */
+export declare function storeTrajectory(trajectory: Omit<TaskTrajectory, 'created_at'>): void;
+/**
+ * Store MaTTS run
+ */
+export declare function storeMattsRun(run: Omit<MattsRun, 'created_at'>): void;
+/**
+ * Log performance metric
+ */
+export declare function logMetric(name: string, value: number): void;
+/**
+ * Count new memories since last consolidation
+ */
+export declare function countNewMemoriesSinceConsolidation(): number;
+/**
+ * Get all active reasoning memories
+ */
+export declare function getAllActiveMemories(): ReasoningMemory[];
+/**
+ * Store memory link (relationship)
+ */
+export declare function storeLink(srcId: string, dstId: string, relation: 'entails' | 'contradicts' | 'refines' | 'duplicate_of', weight: number): void;
+/**
+ * Get contradictions for a memory
+ */
+export declare function getContradictions(memoryId: string): string[];
+/**
+ * Store consolidation run
+ */
+export declare function storeConsolidationRun(run: {
+    run_id: string;
+    items_processed: number;
+    duplicates_found: number;
+    contradictions_found: number;
+    items_pruned: number;
+    duration_ms: number;
+}): void;
+/**
+ * Prune old, unused memories
+ */
+export declare function pruneOldMemories(options: {
+    maxAgeDays: number;
+    minConfidence: number;
+}): number;
+/**
+ * Close database connection
+ */
+export declare function closeDb(): void;
+//# sourceMappingURL=queries.d.ts.map

package/dist/agentic-flow/src/reasoningbank/db/queries.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"queries.d.ts","sourceRoot":"","sources":["../../../../../src/reasoningbank/db/queries.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,QAAQ,MAAM,yBAAyB,CAAC;AAG/C,OAAO,KAAK,EAAE,eAAe,EAAE,gBAAgB,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAU/F;;GAEG;AACH,wBAAsB,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC,CAmGnD;AAED;;GAEG;AACH,wBAAgB,KAAK,IAAI,QAAQ,CAgBhC;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,EAAE;IAC7C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,GAAG,KAAK,CAAC,eAAe,GAAG;IAAE,SAAS,EAAE,YAAY,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,CAqCzE;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,YAAY,GAAG,WAAW,CAAC,GAAG,MAAM,CAmB9F;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,SAAS,EAAE,gBAAgB,GAAG,IAAI,CAWjE;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CASrD;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC,cAAc,EAAE,YAAY,CAAC,GAAG,IAAI,CAoBpF;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,GAAG,IAAI,CAOrE;AAED;;GAEG;AACH,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAa3D;AAED;;GAEG;AACH,wBAAgB,kCAAkC,IAAI,MAAM,CA4B3D;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,eAAe,EAAE,CAexD;AAED;;GAEG;AACH,wBAAgB,SAAS,CACvB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,SAAS,GAAG,aAAa,GAAG,SAAS,GAAG,cAAc,EAChE,MAAM,EAAE,MAAM,GACb,IAAI,CAON;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,CAU5D;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,GAAG,EAAE;IACzC,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB,GAAG,IAAI,CAeP;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE;IACxC,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB,GAAG,MAAM,CAYT;AAED;;GAEG;AACH,wBAAgB,OAAO,IAAI,IAAI,CAM9B"}