contextos-mcp 0.4.0

package/dist/labs/cli.js

@@ -0,0 +1,295 @@
+#!/usr/bin/env tsx
+/**
+ * Standalone RLM CLI — run Recursive Language Model queries from the terminal.
+ *
+ * Usage:
+ *   npx tsx src/cli.ts --file large-file.txt "What are the main themes?"
+ *   npx tsx src/cli.ts --url https://example.com/big.txt "Summarize this"
+ *   cat data.txt | npx tsx src/cli.ts --stdin "Count the errors"
+ *
+ * Environment (pick one):
+ *   ANTHROPIC_API_KEY — for Anthropic models
+ *   OPENAI_API_KEY    — for OpenAI models
+ *   GEMINI_API_KEY    — for Google models
+ */
+import "./env.js";
+import * as fs from "node:fs";
+// Dynamic imports — ensures env.js has set process.env BEFORE pi-ai loads
+const { getModels, getProviders } = await import("@mariozechner/pi-ai");
+const { PythonRepl } = await import("../core/repl.js");
+const { runRlmLoop } = await import("../core/rlm.js");
+// ── Arg parsing ─────────────────────────────────────────────────────────────
+function usage() {
+    console.error(`
+swarm run — Text processing mode (RLM)
+
+USAGE
+  swarm run [OPTIONS] "<query>"
+
+OPTIONS
+  --model <id>     Model ID (default: RLM_MODEL from .env)
+  --file <path>    Read context from a file
+  --url <url>      Fetch context from a URL
+  --stdin          Read context from stdin (pipe data in)
+  --allow-hooks    Enable lifecycle hooks execution (default: disabled)
+  --verbose        Show iteration progress
+
+EXAMPLES
+  swarm run --file big.txt "List all classes"
+  curl -s https://example.com/large.py | swarm run --stdin "Summarize"
+  swarm run --url https://raw.githubusercontent.com/.../typing.py "Count public classes"
+`.trim());
+    process.exit(1);
+}
+function parseArgs() {
+    const args = process.argv.slice(2);
+    let modelId;
+    let file;
+    let url;
+    let useStdin = false;
+    let allowHooks = false;
+    let verbose = false;
+    const positional = [];
+    for (let i = 0; i < args.length; i++) {
+        const arg = args[i];
+        if (arg === "--model" && i + 1 < args.length) {
+            modelId = args[++i];
+        }
+        else if (arg === "--file" && i + 1 < args.length) {
+            file = args[++i];
+        }
+        else if (arg === "--url" && i + 1 < args.length) {
+            url = args[++i];
+        }
+        else if (arg === "--stdin") {
+            useStdin = true;
+        }
+        else if (arg === "--allow-hooks") {
+            allowHooks = true;
+        }
+        else if (arg === "--verbose") {
+            verbose = true;
+        }
+        else if (arg === "--help" || arg === "-h") {
+            usage();
+        }
+        else if (!arg.startsWith("--")) {
+            positional.push(arg);
+        }
+        else {
+            console.error(`Unknown option: ${arg}`);
+            usage();
+        }
+    }
+    if (!modelId) {
+        modelId = process.env.RLM_MODEL || "claude-sonnet-4-6";
+    }
+    if (positional.length === 0) {
+        console.error("Error: query argument is required");
+        usage();
+    }
+    const query = positional.join(" ");
+    if (!file && !url && !useStdin) {
+        console.error("Error: one of --file, --url, or --stdin is required");
+        usage();
+    }
+    return { modelId, file, url, useStdin, allowHooks, verbose, query };
+}
+// ── Helpers ─────────────────────────────────────────────────────────────────
+const MAX_STDIN_BYTES = 50 * 1024 * 1024; // 50MB
+async function readStdin() {
+    const chunks = [];
+    let total = 0;
+    for await (const chunk of process.stdin) {
+        total += chunk.length;
+        if (total > MAX_STDIN_BYTES) {
+            throw new Error(`stdin exceeds ${MAX_STDIN_BYTES / 1024 / 1024}MB limit`);
+        }
+        chunks.push(chunk);
+    }
+    return Buffer.concat(chunks).toString("utf-8");
+}
+const FETCH_TIMEOUT_MS = 30_000;
+const MAX_RESPONSE_BYTES = 50 * 1024 * 1024; // 50MB
+async function fetchUrl(url) {
+    const resp = await fetch(url, { signal: AbortSignal.timeout(FETCH_TIMEOUT_MS) });
+    if (!resp.ok) {
+        throw new Error(`Failed to fetch ${url}: ${resp.status} ${resp.statusText}`);
+    }
+    const contentLength = resp.headers.get("content-length");
+    if (contentLength && parseInt(contentLength, 10) > MAX_RESPONSE_BYTES) {
+        throw new Error(`Response too large (${(parseInt(contentLength, 10) / 1024 / 1024).toFixed(1)}MB). Limit is ${MAX_RESPONSE_BYTES / 1024 / 1024}MB.`);
+    }
+    const text = await resp.text();
+    if (text.length > MAX_RESPONSE_BYTES) {
+        throw new Error(`Response too large (${(text.length / 1024 / 1024).toFixed(1)}MB). Limit is ${MAX_RESPONSE_BYTES / 1024 / 1024}MB.`);
+    }
+    return text;
+}
+// ── Main ────────────────────────────────────────────────────────────────────
+async function main() {
+    const args = parseArgs();
+    // Provider → env var mapping
+    const providerKeys = {
+        anthropic: "ANTHROPIC_API_KEY",
+        openai: "OPENAI_API_KEY",
+        google: "GEMINI_API_KEY",
+    };
+    const defaultModels = {
+        anthropic: "claude-sonnet-4-6",
+        openai: "gpt-4o",
+        google: "gemini-2.5-flash",
+    };
+    // Resolve model — ensure provider has an API key
+    // Prioritise well-known providers so e.g. "gpt-4o" picks "openai" not "azure-openai-responses"
+    let model;
+    let _resolvedProvider = "";
+    const allModelIds = [];
+    const knownProviders = new Set(Object.keys(providerKeys));
+    // First pass: only well-known providers
+    for (const provider of getProviders()) {
+        const providerModels = getModels(provider);
+        for (const m of providerModels) {
+            allModelIds.push(m.id);
+            if (!model && m.id === args.modelId && knownProviders.has(provider)) {
+                const key = providerKeys[provider];
+                if (process.env[key]) {
+                    model = m;
+                    _resolvedProvider = provider;
+                }
+            }
+        }
+    }
+    // Second pass: remaining providers (if not found above)
+    if (!model) {
+        for (const provider of getProviders()) {
+            if (knownProviders.has(provider))
+                continue;
+            for (const m of getModels(provider)) {
+                if (m.id === args.modelId) {
+                    const key = `${provider.toUpperCase().replace(/-/g, "_")}_API_KEY`;
+                    if (process.env[key]) {
+                        model = m;
+                        _resolvedProvider = provider;
+                        break;
+                    }
+                }
+            }
+            if (model)
+                break;
+        }
+    }
+    // Fallback: if default model's provider has no key, pick one that does
+    if (!model) {
+        for (const [prov, envKey] of Object.entries(providerKeys)) {
+            if (!process.env[envKey])
+                continue;
+            const fallbackId = defaultModels[prov];
+            if (!fallbackId)
+                continue;
+            for (const p of getProviders()) {
+                if (p !== prov)
+                    continue;
+                for (const m of getModels(p)) {
+                    if (m.id === fallbackId) {
+                        model = m;
+                        _resolvedProvider = prov;
+                        args.modelId = fallbackId;
+                        console.error(`Note: using ${fallbackId} (${prov}) — set RLM_MODEL to override`);
+                        break;
+                    }
+                }
+                if (model)
+                    break;
+            }
+            if (model)
+                break;
+        }
+    }
+    if (!model) {
+        console.error(`Error: unknown model "${args.modelId}"`);
+        console.error(`Available models: ${allModelIds.join(", ")}`);
+        process.exit(1);
+    }
+    // Load context
+    let context;
+    if (args.file) {
+        try {
+            const stat = fs.statSync(args.file);
+            if (stat.isDirectory()) {
+                console.error(`Error: "${args.file}" is a directory. Use the interactive mode (\`rlm\`) for directory loading.`);
+                process.exit(1);
+            }
+        }
+        catch (err) {
+            console.error(`Error: could not access "${args.file}": ${err.message}`);
+            process.exit(1);
+        }
+        console.error(`Reading context from file: ${args.file}`);
+        try {
+            context = fs.readFileSync(args.file, "utf-8");
+        }
+        catch (err) {
+            console.error(`Error: could not read file "${args.file}": ${err.message}`);
+            process.exit(1);
+        }
+    }
+    else if (args.url) {
+        console.error(`Fetching context from URL: ${args.url}`);
+        context = await fetchUrl(args.url);
+    }
+    else {
+        console.error("Reading context from stdin...");
+        context = await readStdin();
+    }
+    console.error(`Context loaded: ${context.length.toLocaleString()} characters`);
+    console.error(`Model: ${model.id}`);
+    console.error(`Query: ${args.query}`);
+    console.error("---");
+    // Start REPL: PythonRepl if explicitly specified; NodeVmRepl is deprecated/disabled
+    const usePython = process.argv.includes("--python-repl") || process.argv.includes("--repl=python");
+    if (!usePython) {
+        console.error("Error: In-process JavaScript execution has been permanently deprecated for security. Pass --python-repl for Python text processing.");
+        process.exit(1);
+    }
+    const repl = new PythonRepl();
+    const ac = new AbortController();
+    const abortAndExit = () => {
+        console.error("\nAborting...");
+        ac.abort();
+    };
+    process.on("SIGINT", abortAndExit);
+    process.on("SIGTERM", abortAndExit);
+    try {
+        await repl.start(ac.signal);
+        const startTime = Date.now();
+        const result = await runRlmLoop({
+            context,
+            query: args.query,
+            model,
+            repl,
+            signal: ac.signal,
+            onProgress: args.verbose
+                ? (info) => {
+                    const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
+                    console.error(`[${elapsed}s] Iteration ${info.iteration}/${info.maxIterations} | ` +
+                        `Sub-queries: ${info.subQueries} | Phase: ${info.phase}`);
+                }
+                : undefined,
+        });
+        const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
+        console.error("---");
+        console.error(`Completed in ${elapsed}s | ${result.iterations} iterations | ${result.totalSubQueries} sub-queries | ${result.completed ? "success" : "incomplete"}`);
+        console.error("---");
+        // Write the answer to stdout (not stderr) so it can be piped
+        console.log(result.answer);
+    }
+    finally {
+        repl.shutdown();
+    }
+}
+main().catch((err) => {
+    console.error("Fatal error:", err);
+    process.exit(1);
+});
+//# sourceMappingURL=cli.js.map

package/dist/labs/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../src/labs/cli.ts"],"names":[],"mappings":";AACA;;;;;;;;;;;;GAYG;AAEH,OAAO,UAAU,CAAC;AAClB,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAE9B,0EAA0E;AAC1E,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,CAAC;AACxE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;AACvD,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;AAItD,+EAA+E;AAE/E,SAAS,KAAK;IACb,OAAO,CAAC,KAAK,CACZ;;;;;;;;;;;;;;;;;;CAkBD,CAAC,IAAI,EAAE,CACN,CAAC;IACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC;AAYD,SAAS,SAAS;IACjB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACnC,IAAI,OAA2B,CAAC;IAChC,IAAI,IAAwB,CAAC;IAC7B,IAAI,GAAuB,CAAC;IAC5B,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,UAAU,GAAG,KAAK,CAAC;IACvB,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,GAAG,KAAK,SAAS,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YAC9C,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACrB,CAAC;aAAM,IAAI,GAAG,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YACpD,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAClB,CAAC;aAAM,IAAI,GAAG,KAAK,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YACnD,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACjB,CAAC;aAAM,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YAC9B,QAAQ,GAAG,IAAI,CAAC;QACjB,CAAC;aAAM,IAAI,GAAG,KAAK,eAAe,EAAE,CAAC;YACpC,UAAU,GAAG,IAAI,CAAC;QACnB,CAAC;aAAM,IAAI,GAAG,KAAK,WAAW,EAAE,CAAC;YAChC,OAAO,GAAG,IAAI,CAAC;QAChB,CAAC;aAAM,IAAI,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;YAC7C,KAAK,EAAE,CAAC;QACT,CAAC;aAAM,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,CAAC;YACP,OAAO,CAAC,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC,CAAC;YACxC,KAAK,EAAE,CAAC;QACT,CAAC;IACF,CAAC;IAED,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,mBAAmB,CAAC;IACxD,CAAC;IAED,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACnD,KAAK,EAAE,CAAC;IACT,CAAC;IAED,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEnC,IAAI,CAAC,IAAI,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChC,OAAO,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACrE,KAAK,EAAE,CAAC;IACT,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;AACrE,CAAC;AAED,+EAA+E;AAE/E,MAAM,eAAe,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO;AAEjD,KAAK,UAAU,SAAS;IACvB,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACzC,KAAK,IAAK,KAAgB,CAAC,MAAM,CAAC;QAClC,IAAI,KAAK,GAAG,eAAe,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,iBAAiB,eAAe,GAAG,IAAI,GAAG,IAAI,UAAU,CAAC,CAAC;QAC3E,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAe,CAAC,CAAC;IAC9B,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAChC,MAAM,kBAAkB,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO;AAEpD,KAAK,UAAU,QAAQ,CAAC,GAAW;IAClC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;IACjF,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,KAAK,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;IAC9E,CAAC;IACD,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACzD,IAAI,aAAa,IAAI,QAAQ,CAAC,aAAa,EAAE,EAAE,CAAC,GAAG,kBAAkB,EAAE,CAAC;QACvE,MAAM,IAAI,KAAK,CACd,uBAAuB,CAAC,QAAQ,CAAC,aAAa,EAAE,EAAE,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,IAAI,KAAK,CACnI,CAAC;IACH,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IAC/B,IAAI,IAAI,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CACd,uBAAuB,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,IAAI,KAAK,CACnH,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAED,+EAA+E;AAE/E,KAAK,UAAU,IAAI;IAClB,MAAM,IAAI,GAAG,SAAS,EAAE,CAAC;IAEzB,6BAA6B;IAC7B,MAAM,YAAY,GAA2B;QAC5C,SAAS,EAAE,mBAAmB;QAC9B,MAAM,EAAE,gBAAgB;QACxB,MAAM,EAAE,gBAAgB;KACxB,CAAC;IACF,MAAM,aAAa,GAA2B;QAC7C,SAAS,EAAE,mBAAmB;QAC9B,MAAM,EAAE,QAAQ;QAChB,MAAM,EAAE,kBAAkB;KAC1B,CAAC;IAEF,iDAAiD;IACjD,+FAA+F;IAC/F,IAAI,KAA6B,CAAC;IAClC,IAAI,iBAAiB,GAAG,EAAE,CAAC;IAC3B,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;IAE1D,wCAAwC;IACxC,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,EAAE,CAAC;QACvC,MAAM,cAAc,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC3C,KAAK,MAAM,CAAC,IAAI,cAAc,EAAE,CAAC;YAChC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACvB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,OAAO,IAAI,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrE,MAAM,GAAG,GAAG,YAAY,CAAC,QAAQ,CAAE,CAAC;gBACpC,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtB,KAAK,GAAG,CAAC,CAAC;oBACV,iBAAiB,GAAG,QAAQ,CAAC;gBAC9B,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IACD,wDAAwD;IACxD,IAAI,CAAC,KAAK,EAAE,CAAC;QACZ,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,EAAE,CAAC;YACvC,IAAI,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC;gBAAE,SAAS;YAC3C,KAAK,MAAM,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;oBAC3B,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,UAAU,CAAC;oBACnE,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;wBACtB,KAAK,GAAG,CAAC,CAAC;wBACV,iBAAiB,GAAG,QAAQ,CAAC;wBAC7B,MAAM;oBACP,CAAC;gBACF,CAAC;YACF,CAAC;YACD,IAAI,KAAK;gBAAE,MAAM;QAClB,CAAC;IACF,CAAC;IAED,uEAAuE;IACvE,IAAI,CAAC,KAAK,EAAE,CAAC;QACZ,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;gBAAE,SAAS;YACnC,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,CAAC,UAAU;gBAAE,SAAS;YAC1B,KAAK,MAAM,CAAC,IAAI,YAAY,EAAE,EAAE,CAAC;gBAChC,IAAI,CAAC,KAAK,IAAI;oBAAE,SAAS;gBACzB,KAAK,MAAM,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC9B,IAAI,CAAC,CAAC,EAAE,KAAK,UAAU,EAAE,CAAC;wBACzB,KAAK,GAAG,CAAC,CAAC;wBACV,iBAAiB,GAAG,IAAI,CAAC;wBACzB,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC;wBAC1B,OAAO,CAAC,KAAK,CAAC,eAAe,UAAU,KAAK,IAAI,+BAA+B,CAAC,CAAC;wBACjF,MAAM;oBACP,CAAC;gBACF,CAAC;gBACD,IAAI,KAAK;oBAAE,MAAM;YAClB,CAAC;YACD,IAAI,KAAK;gBAAE,MAAM;QAClB,CAAC;IACF,CAAC;IAED,IAAI,CAAC,KAAK,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CAAC,yBAAyB,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;QACxD,OAAO,CAAC,KAAK,CAAC,qBAAqB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,eAAe;IACf,IAAI,OAAe,CAAC;IACpB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACf,IAAI,CAAC;YACJ,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBACxB,OAAO,CAAC,KAAK,CACZ,WAAW,IAAI,CAAC,IAAI,6EAA6E,CACjG,CAAC;gBACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACjB,CAAC;QACF,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,4BAA4B,IAAI,CAAC,IAAI,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACxE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC;YACJ,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,+BAA+B,IAAI,CAAC,IAAI,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACF,CAAC;SAAM,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACrB,OAAO,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QACxD,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC;SAAM,CAAC;QACP,OAAO,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;QAC/C,OAAO,GAAG,MAAM,SAAS,EAAE,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,CAAC,MAAM,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IAC/E,OAAO,CAAC,KAAK,CAAC,UAAU,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;IACpC,OAAO,CAAC,KAAK,CAAC,UAAU,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;IACtC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAErB,oFAAoF;IACpF,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;IACnG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CACZ,qIAAqI,CACrI,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,UAAU,EAAE,CAAC;IAC9B,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;IAEjC,MAAM,YAAY,GAAG,GAAG,EAAE;QACzB,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;QAC/B,EAAE,CAAC,KAAK,EAAE,CAAC;IACZ,CAAC,CAAC;IACF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;IAEpC,IAAI,CAAC;QACJ,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;QAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC;YAC/B,OAAO;YACP,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK;YACL,IAAI;YACJ,MAAM,EAAE,EAAE,CAAC,MAAM;YACjB,UAAU,EAAE,IAAI,CAAC,OAAO;gBACvB,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE;oBACT,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;oBAC7D,OAAO,CAAC,KAAK,CACZ,IAAI,OAAO,gBAAgB,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,aAAa,KAAK;wBACnE,gBAAgB,IAAI,CAAC,UAAU,aAAa,IAAI,CAAC,KAAK,EAAE,CACzD,CAAC;gBACH,CAAC;gBACF,CAAC,CAAC,SAAS;SACZ,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC7D,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,CAAC,KAAK,CACZ,gBAAgB,OAAO,OAAO,MAAM,CAAC,UAAU,iBAAiB,MAAM,CAAC,eAAe,kBAAkB,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,EAAE,CACrJ,CAAC;QACF,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAErB,6DAA6D;QAC7D,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;YAAS,CAAC;QACV,IAAI,CAAC,QAAQ,EAAE,CAAC;IACjB,CAAC;AACF,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;IACnC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}

package/dist/labs/episodic-memory.d.ts

@@ -0,0 +1,132 @@
+/**
+ * Episodic memory — persists successful thread strategies to disk.
+ *
+ * Records episodes after successful thread completions:
+ *   - Task pattern (normalized keywords)
+ *   - Agent + model used
+ *   - Result quality (success, duration, cost, files changed)
+ *   - Task slot + complexity classification
+ *
+ * Recall: given a new task, find similar past episodes and return
+ * the strategies that worked. Used to inform agent/model selection
+ * and provide context hints to the orchestrator.
+ *
+ * Storage: JSON files in ~/.swarm/memory/episodes/ indexed by
+ * task similarity hash (trigram-based).
+ */
+export interface Episode {
+    /** Unique episode ID (SHA-256 of task + timestamp). */
+    id: string;
+    /** Original task description. */
+    task: string;
+    /** Normalized task keywords for similarity matching. */
+    taskKeywords: string[];
+    /** Agent backend used. */
+    agent: string;
+    /** Model used. */
+    model: string;
+    /** Task slot (execution/search/reasoning/planning). */
+    slot: string;
+    /** Task complexity (simple/medium/complex). */
+    complexity: string;
+    /** Whether the thread succeeded. */
+    success: boolean;
+    /** Duration in ms. */
+    durationMs: number;
+    /** Estimated cost in USD. */
+    estimatedCostUsd: number;
+    /** Number of files changed. */
+    filesChangedCount: number;
+    /** File paths changed (for pattern matching). */
+    filesChanged: string[];
+    /** Compressed result summary (the episode content). */
+    summary: string;
+    /** Timestamp. */
+    timestamp: number;
+}
+export interface EpisodeRecall {
+    episode: Episode;
+    /** Similarity score (0-1). */
+    similarity: number;
+}
+/** Per-agent aggregate statistics derived from episodic memory. */
+export interface AgentAggregateStats {
+    /** Total number of successful episodes. */
+    totalEpisodes: number;
+    /** Average duration in ms. */
+    avgDurationMs: number;
+    /** Average cost in USD. */
+    avgCostUsd: number;
+    /** Success count per slot (execution/search/reasoning/planning). */
+    slotCounts: Map<string, number>;
+    /** File extensions this agent has successfully worked with. */
+    fileExtensions: Set<string>;
+}
+/** Aggregate stats across all agents. */
+export interface AggregateStats {
+    /** Per-agent statistics. */
+    perAgent: Map<string, AgentAggregateStats>;
+    /**
+     * File extension → agent → success count.
+     * Used by the router to boost agents that handle specific file types well.
+     */
+    fileExtensions: Map<string, Map<string, number>>;
+}
+export declare class EpisodicMemory {
+    private memoryDir;
+    private episodes;
+    private loaded;
+    private maxEpisodes;
+    constructor(memoryDir: string, maxEpisodes?: number);
+    /** Initialize — create directory and load existing episodes. */
+    init(): Promise<void>;
+    /** Record a new episode from a completed thread. */
+    record(params: {
+        task: string;
+        agent: string;
+        model: string;
+        slot: string;
+        complexity: string;
+        success: boolean;
+        durationMs: number;
+        estimatedCostUsd: number;
+        filesChanged: string[];
+        summary: string;
+    }): Promise<Episode>;
+    /**
+     * Recall similar past episodes for a given task.
+     * Returns episodes sorted by similarity (highest first).
+     */
+    recall(task: string, maxResults?: number, minSimilarity?: number): EpisodeRecall[];
+    /**
+     * Get strategy recommendations based on past episodes.
+     * Returns a formatted string for inclusion in orchestrator context.
+     */
+    getStrategyHints(task: string): string | null;
+    /**
+     * Get the best agent/model recommendation for a task based on past episodes.
+     * Returns null if no relevant episodes found.
+     */
+    recommendStrategy(task: string): {
+        agent: string;
+        model: string;
+        confidence: number;
+    } | null;
+    /**
+     * Get aggregate statistics across all episodes, grouped by agent.
+     *
+     * Returns per-agent success rates, average costs/durations, slot distributions,
+     * and a file-extension-to-agent mapping. Used by the model router as a fallback
+     * when no high-confidence episodic match exists.
+     *
+     * Returns null if no episodes are loaded (graceful degradation).
+     */
+    getAggregateStats(): AggregateStats | null;
+    /** Get total episode count. */
+    get size(): number;
+    /** Get all episodes (for viewer). */
+    getAll(): Episode[];
+    private loadAll;
+    private saveEpisode;
+    private deleteFile;
+}