@burtson-labs/host-kit 0.3.1

package/dist/memory.js

@@ -0,0 +1,151 @@
+"use strict";
+/**
+ * Memory — auto-load workspace-local context files (BANDIT.md / CLAUDE.md)
+ * and inline them into the system prompt so the agent follows project rules
+ * without being re-told on every turn.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadCombinedMemory = loadCombinedMemory;
+exports.loadMemory = loadMemory;
+exports.appendMemory = appendMemory;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const memoryIndex_1 = require("./memoryIndex");
+const CANDIDATES = [
+    'BANDIT.md',
+    'CLAUDE.md',
+    'AGENTS.md', // OpenAI Codex / GitHub Copilot convention — load alongside BANDIT.md, don't pick one
+    '.bandit/BANDIT.md',
+    '.bandit/memory.md'
+];
+const MAX_BYTES = 32 * 1024;
+/**
+ * loadMemory + MEMORY.md index, fused into a single bundle the host can
+ * inject as the system-prompt memory block. Always-loaded files come
+ * first; the index follows under its own source marker so the model
+ * knows which entries are full content vs. on-demand topic pointers.
+ *
+ * Returns sources from BOTH passes when present — callers use this for
+ * the `/memory` slash command and boot-status output.
+ */
+async function loadCombinedMemory(cwd, warn) {
+    const base = await loadMemory(cwd);
+    const index = await (0, memoryIndex_1.loadMemoryIndex)(cwd, warn);
+    const indexBlock = (0, memoryIndex_1.renderMemoryIndexBlock)(index);
+    if (!indexBlock)
+        return base;
+    const headed = `<!-- source: ${memoryIndex_1.MEMORY_INDEX_FILE} -->\n${indexBlock}`;
+    return {
+        content: base.content ? `${base.content}\n\n${headed}` : headed,
+        sources: index.source ? [...base.sources, memoryIndex_1.MEMORY_INDEX_FILE] : base.sources
+    };
+}
+async function loadMemory(cwd) {
+    const sections = [];
+    const sources = [];
+    for (const rel of CANDIDATES) {
+        const abs = path.resolve(cwd, rel);
+        try {
+            const raw = await fs.promises.readFile(abs);
+            if (raw.byteLength === 0)
+                continue;
+            const truncated = raw.byteLength > MAX_BYTES;
+            const text = raw.subarray(0, MAX_BYTES).toString('utf-8');
+            sections.push(`<!-- source: ${rel} -->\n${text}${truncated ? '\n… (truncated)' : ''}`);
+            sources.push(rel);
+        }
+        catch {
+            // Not present — that's fine.
+        }
+    }
+    return {
+        content: sections.join('\n\n'),
+        sources
+    };
+}
+/**
+ * Append a single fact to project memory so it survives across sessions.
+ *
+ * Persists to `BANDIT.md` at the workspace root (creates the file with a
+ * minimal frontmatter-free header when it doesn't exist). Bullets are
+ * appended under a `## Notes` heading so multiple `/remember` calls
+ * accumulate without scattering. CLAUDE.md-only repos still get found
+ * by loadMemory, but new facts always land in BANDIT.md to keep one
+ * canonical write target.
+ *
+ * Returns the absolute path written so the caller can echo it back to
+ * the user — visibility matters because the user typed "remember X"
+ * and needs to see WHERE the fact landed.
+ */
+async function appendMemory(cwd, fact) {
+    const trimmed = fact.trim();
+    if (!trimmed)
+        throw new Error('appendMemory: fact must be a non-empty string');
+    const abs = path.resolve(cwd, 'BANDIT.md');
+    let existing = '';
+    try {
+        existing = await fs.promises.readFile(abs, 'utf-8');
+    }
+    catch {
+        // File doesn't exist — start with a minimal scaffold so the file is
+        // self-explanatory if the user opens it later.
+        existing =
+            '# Project Memory\n\n' +
+                'Auto-loaded by Bandit on every prompt. Add facts the agent should\n' +
+                'remember across sessions — preferences, repo locations, conventions,\n' +
+                'incident notes. Use `/remember <fact>` to append a bullet under the\n' +
+                'Notes heading.\n';
+    }
+    // First bullet under "## Notes"? Append a new section. Otherwise
+    // tack the bullet onto the existing section.
+    const NOTES_HEADING = '## Notes';
+    const hasSection = existing.includes(`\n${NOTES_HEADING}`) || existing.startsWith(NOTES_HEADING);
+    const bullet = `- ${trimmed}`;
+    let next;
+    if (hasSection) {
+        // Insert at the end of the file — bullets stay chronological so the
+        // most recent context lands at the bottom where it's easiest to see
+        // when scrolling.
+        next = existing.endsWith('\n') ? `${existing}${bullet}\n` : `${existing}\n${bullet}\n`;
+    }
+    else {
+        const sep = existing.endsWith('\n\n') ? '' : existing.endsWith('\n') ? '\n' : '\n\n';
+        next = `${existing}${sep}${NOTES_HEADING}\n\n${bullet}\n`;
+    }
+    await fs.promises.writeFile(abs, next, 'utf-8');
+    return abs;
+}
+//# sourceMappingURL=memory.js.map

package/dist/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../src/memory.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCH,gDAUC;AAED,gCAsBC;AAgBD,oCAkCC;AArHD,uCAAyB;AACzB,2CAA6B;AAC7B,+CAKuB;AAEvB,MAAM,UAAU,GAAG;IACjB,WAAW;IACX,WAAW;IACX,WAAW,EAAE,sFAAsF;IACnG,mBAAmB;IACnB,mBAAmB;CACpB,CAAC;AACF,MAAM,SAAS,GAAG,EAAE,GAAG,IAAI,CAAC;AAQ5B;;;;;;;;GAQG;AACI,KAAK,UAAU,kBAAkB,CAAC,GAAW,EAAE,IAAmB;IACvE,MAAM,IAAI,GAAG,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,MAAM,IAAA,6BAAe,EAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAC/C,MAAM,UAAU,GAAG,IAAA,oCAAsB,EAAC,KAAK,CAAC,CAAC;IACjD,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,MAAM,MAAM,GAAG,gBAAgB,+BAAiB,SAAS,UAAU,EAAE,CAAC;IACtE,OAAO;QACL,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,OAAO,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM;QAC/D,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,+BAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;KAC5E,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,UAAU,CAAC,GAAW;IAC1C,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,OAAO,GAAa,EAAE,CAAC;IAE7B,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC5C,IAAI,GAAG,CAAC,UAAU,KAAK,CAAC;gBAAE,SAAS;YACnC,MAAM,SAAS,GAAG,GAAG,CAAC,UAAU,GAAG,SAAS,CAAC;YAC7C,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC1D,QAAQ,CAAC,IAAI,CAAC,gBAAgB,GAAG,SAAS,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvF,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;QAAC,MAAM,CAAC;YACP,6BAA6B;QAC/B,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC;QAC9B,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;GAaG;AACI,KAAK,UAAU,YAAY,CAAC,GAAW,EAAE,IAAY;IAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,CAAC,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IAC/E,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IAC3C,IAAI,QAAQ,GAAG,EAAE,CAAC;IAClB,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAAC,MAAM,CAAC;QACP,oEAAoE;QACpE,+CAA+C;QAC/C,QAAQ;YACN,sBAAsB;gBACtB,qEAAqE;gBACrE,wEAAwE;gBACxE,uEAAuE;gBACvE,kBAAkB,CAAC;IACvB,CAAC;IACD,iEAAiE;IACjE,6CAA6C;IAC7C,MAAM,aAAa,GAAG,UAAU,CAAC;IACjC,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,aAAa,EAAE,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;IACjG,MAAM,MAAM,GAAG,KAAK,OAAO,EAAE,CAAC;IAC9B,IAAI,IAAY,CAAC;IACjB,IAAI,UAAU,EAAE,CAAC;QACf,oEAAoE;QACpE,oEAAoE;QACpE,kBAAkB;QAClB,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,IAAI,CAAC,CAAC,CAAC,GAAG,QAAQ,KAAK,MAAM,IAAI,CAAC;IACzF,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC;QACrF,IAAI,GAAG,GAAG,QAAQ,GAAG,GAAG,GAAG,aAAa,OAAO,MAAM,IAAI,CAAC;IAC5D,CAAC;IACD,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAChD,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/memoryIndex.d.ts

@@ -0,0 +1,38 @@
+export declare const MAX_INDEX_BYTES: number;
+export declare const MAX_MEMORY_FILE_BYTES: number;
+export declare const MEMORY_DIR = "memory";
+export declare const MEMORY_INDEX_FILE = "MEMORY.md";
+export interface MemoryIndexEntry {
+    /** Slug derived from filename (e.g. "auth-conventions"). Used as the `name` arg to read_memory. */
+    name: string;
+    /** Human label from the markdown link. */
+    title: string;
+    /** One-line summary after the em-dash. */
+    hook: string;
+    /** Path relative to the workspace root, e.g. "memory/auth-conventions.md". */
+    relPath: string;
+    /** Resolved absolute path on disk. */
+    absPath: string;
+}
+export interface MemoryIndex {
+    /** The MEMORY.md content (trimmed to MAX_INDEX_BYTES). Empty when the file is missing. */
+    indexContent: string;
+    /** Parsed entries; dangling links are dropped (with a warning). */
+    entries: MemoryIndexEntry[];
+    /** Absolute path of MEMORY.md if found, else null. */
+    source: string | null;
+}
+/**
+ * Optional warn sink. Defaults to a single-line stderr write so the host
+ * sees dangling-link warnings without surfacing them to the model.
+ */
+export type MemoryWarnFn = (msg: string) => void;
+/**
+ * Render a MEMORY.md index as a system-prompt block. Includes a one-line
+ * nudge telling the model to call `read_memory(name=...)` when an entry's
+ * hook matches the task. Returns an empty string when the index has no
+ * entries so the caller can `if (block) ...` without a null check.
+ */
+export declare function renderMemoryIndexBlock(index: MemoryIndex): string;
+export declare function loadMemoryIndex(cwd: string, warn?: MemoryWarnFn): Promise<MemoryIndex>;
+//# sourceMappingURL=memoryIndex.d.ts.map

package/dist/memoryIndex.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memoryIndex.d.ts","sourceRoot":"","sources":["../src/memoryIndex.ts"],"names":[],"mappings":"AAmBA,eAAO,MAAM,eAAe,QAAW,CAAC;AACxC,eAAO,MAAM,qBAAqB,QAAY,CAAC;AAC/C,eAAO,MAAM,UAAU,WAAW,CAAC;AACnC,eAAO,MAAM,iBAAiB,cAAc,CAAC;AAE7C,MAAM,WAAW,gBAAgB;IAC/B,mGAAmG;IACnG,IAAI,EAAE,MAAM,CAAC;IACb,0CAA0C;IAC1C,KAAK,EAAE,MAAM,CAAC;IACd,0CAA0C;IAC1C,IAAI,EAAE,MAAM,CAAC;IACb,8EAA8E;IAC9E,OAAO,EAAE,MAAM,CAAC;IAChB,sCAAsC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,WAAW;IAC1B,0FAA0F;IAC1F,YAAY,EAAE,MAAM,CAAC;IACrB,mEAAmE;IACnE,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAC5B,sDAAsD;IACtD,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;CACvB;AAQD;;;GAGG;AACH,MAAM,MAAM,YAAY,GAAG,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAC;AAMjD;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,WAAW,GAAG,MAAM,CAWjE;AAED,wBAAsB,eAAe,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,GAAE,YAA0B,GAAG,OAAO,CAAC,WAAW,CAAC,CAuDzG"}

package/dist/memoryIndex.js

@@ -0,0 +1,142 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MEMORY_INDEX_FILE = exports.MEMORY_DIR = exports.MAX_MEMORY_FILE_BYTES = exports.MAX_INDEX_BYTES = void 0;
+exports.renderMemoryIndexBlock = renderMemoryIndexBlock;
+exports.loadMemoryIndex = loadMemoryIndex;
+/**
+ * Lazy-load topic memory.
+ *
+ * MEMORY.md at the workspace root is an INDEX — one bullet per topic, each
+ * pointing at a markdown file in a sibling `memory/` directory. The index
+ * itself goes into the system prompt every turn (small — capped at 4 KB).
+ * The linked files are NOT preloaded; the agent reads them via the
+ * `read_memory` tool when it decides a topic is relevant.
+ *
+ * Index entry format:
+ *
+ *   - [Topic title](memory/file.md) — one-line hook for relevance matching
+ *
+ * The hook is what the agent sees and reasons about. Pick hooks that name
+ * the situation ("when editing auth code", "when the user mentions X").
+ */
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+exports.MAX_INDEX_BYTES = 4 * 1024;
+exports.MAX_MEMORY_FILE_BYTES = 32 * 1024;
+exports.MEMORY_DIR = 'memory';
+exports.MEMORY_INDEX_FILE = 'MEMORY.md';
+const ENTRY_RE = /^[-*]\s*\[([^\]]+)\]\(memory\/([^)]+)\)\s*(?:[—–-]+)\s*(.+)$/;
+function slugFromFilename(file) {
+    return file.replace(/\.md$/i, '');
+}
+const defaultWarn = (msg) => {
+    process.stderr.write(`[memory-index] ${msg}\n`);
+};
+/**
+ * Render a MEMORY.md index as a system-prompt block. Includes a one-line
+ * nudge telling the model to call `read_memory(name=...)` when an entry's
+ * hook matches the task. Returns an empty string when the index has no
+ * entries so the caller can `if (block) ...` without a null check.
+ */
+function renderMemoryIndexBlock(index) {
+    if (index.entries.length === 0)
+        return '';
+    const lines = index.entries.map((e) => `- [${e.title}](${e.relPath}) — ${e.hook}`);
+    return [
+        '<!-- source: MEMORY.md (index — call read_memory(name="<slug>") to load a topic) -->',
+        'These are topic memories — full content is NOT preloaded. Read the hook; if it matches the current task, call `read_memory(name="<slug>")` BEFORE making changes. The slug is the part after `memory/` and before `.md` in the link.',
+        '',
+        ...lines
+    ].join('\n');
+}
+async function loadMemoryIndex(cwd, warn = defaultWarn) {
+    const abs = path.resolve(cwd, exports.MEMORY_INDEX_FILE);
+    let raw;
+    try {
+        raw = await fs.promises.readFile(abs);
+    }
+    catch {
+        return { indexContent: '', entries: [], source: null };
+    }
+    if (raw.byteLength === 0) {
+        return { indexContent: '', entries: [], source: abs };
+    }
+    const truncated = raw.byteLength > exports.MAX_INDEX_BYTES;
+    const text = raw.subarray(0, exports.MAX_INDEX_BYTES).toString('utf-8');
+    const indexContent = truncated ? `${text}\n… (truncated — MEMORY.md exceeds ${exports.MAX_INDEX_BYTES} bytes; split topics into smaller files)` : text;
+    const entries = [];
+    const seen = new Set();
+    for (const line of text.split(/\r?\n/)) {
+        const match = ENTRY_RE.exec(line.trim());
+        if (!match)
+            continue;
+        const [, title, fileRaw, hookRaw] = match;
+        const file = fileRaw.trim();
+        if (!file || file.includes('..') || path.isAbsolute(file)) {
+            warn(`skipped entry "${title}" — invalid memory/ path "${file}"`);
+            continue;
+        }
+        const relPath = path.posix.join(exports.MEMORY_DIR, file);
+        const absPath = path.resolve(cwd, exports.MEMORY_DIR, file);
+        let exists = false;
+        try {
+            const stat = await fs.promises.stat(absPath);
+            exists = stat.isFile();
+        }
+        catch {
+            exists = false;
+        }
+        if (!exists) {
+            warn(`dangling link: ${relPath} (referenced by MEMORY.md but no such file)`);
+            continue;
+        }
+        const name = slugFromFilename(file);
+        if (seen.has(name)) {
+            warn(`duplicate slug "${name}" — keeping first occurrence`);
+            continue;
+        }
+        seen.add(name);
+        entries.push({
+            name,
+            title: title.trim(),
+            hook: hookRaw.trim(),
+            relPath,
+            absPath
+        });
+    }
+    return { indexContent, entries, source: abs };
+}
+//# sourceMappingURL=memoryIndex.js.map

package/dist/memoryIndex.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memoryIndex.js","sourceRoot":"","sources":["../src/memoryIndex.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAoEA,wDAWC;AAED,0CAuDC;AAxID;;;;;;;;;;;;;;;GAeG;AACH,uCAAyB;AACzB,2CAA6B;AAEhB,QAAA,eAAe,GAAG,CAAC,GAAG,IAAI,CAAC;AAC3B,QAAA,qBAAqB,GAAG,EAAE,GAAG,IAAI,CAAC;AAClC,QAAA,UAAU,GAAG,QAAQ,CAAC;AACtB,QAAA,iBAAiB,GAAG,WAAW,CAAC;AAwB7C,MAAM,QAAQ,GAAG,8DAA8D,CAAC;AAEhF,SAAS,gBAAgB,CAAC,IAAY;IACpC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;AACpC,CAAC;AAQD,MAAM,WAAW,GAAiB,CAAC,GAAG,EAAE,EAAE;IACxC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,GAAG,IAAI,CAAC,CAAC;AAClD,CAAC,CAAC;AAEF;;;;;GAKG;AACH,SAAgB,sBAAsB,CAAC,KAAkB;IACvD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAC1C,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,OAAO,OAAO,CAAC,CAAC,IAAI,EAAE,CAClD,CAAC;IACF,OAAO;QACL,sFAAsF;QACtF,sOAAsO;QACtO,EAAE;QACF,GAAG,KAAK;KACT,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAEM,KAAK,UAAU,eAAe,CAAC,GAAW,EAAE,OAAqB,WAAW;IACjF,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,yBAAiB,CAAC,CAAC;IACjD,IAAI,GAAW,CAAC;IAChB,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,YAAY,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACzD,CAAC;IACD,IAAI,GAAG,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,EAAE,YAAY,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;IACxD,CAAC;IACD,MAAM,SAAS,GAAG,GAAG,CAAC,UAAU,GAAG,uBAAe,CAAC;IACnD,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,uBAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAChE,MAAM,YAAY,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,sCAAsC,uBAAe,0CAA0C,CAAC,CAAC,CAAC,IAAI,CAAC;IAE/I,MAAM,OAAO,GAAuB,EAAE,CAAC;IACvC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,KAAK,CAAC;QAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1D,IAAI,CAAC,kBAAkB,KAAK,6BAA6B,IAAI,GAAG,CAAC,CAAC;YAClE,SAAS;QACX,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,kBAAU,EAAE,IAAI,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,kBAAU,EAAE,IAAI,CAAC,CAAC;QACpD,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,GAAG,KAAK,CAAC;QACjB,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,IAAI,CAAC,kBAAkB,OAAO,6CAA6C,CAAC,CAAC;YAC7E,SAAS;QACX,CAAC;QACD,MAAM,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACnB,IAAI,CAAC,mBAAmB,IAAI,8BAA8B,CAAC,CAAC;YAC5D,SAAS;QACX,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACf,OAAO,CAAC,IAAI,CAAC;YACX,IAAI;YACJ,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE;YACnB,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE;YACpB,OAAO;YACP,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;AAChD,CAAC"}

package/dist/mentions.d.ts

@@ -0,0 +1,33 @@
+/**
+ * @-mentions — detect `@path/to/file` patterns in user input and inline the
+ * file contents into the prompt so the model sees them in the first turn.
+ *
+ * user> explain @src/auth.ts
+ * → prompt becomes: "explain @src/auth.ts\n\n[File: src/auth.ts]\n<content>"
+ *
+ * Images (png/jpg/etc.) are NOT inlined as text — doing so dumps raw binary
+ * bytes decoded as UTF-8 into the prompt, and the model interprets the
+ * resulting gibberish as corrupt data. Instead, image mentions are
+ * collected into `images[]` as base64 data URLs so the caller can
+ * forward them on the provider's `images` field (Ollama vision models
+ * accept this; Bandit cloud forwards through to the gateway).
+ */
+export interface ExpandedPrompt {
+    prompt: string;
+    mentions: {
+        path: string;
+        bytes: number;
+        ok: boolean;
+        kind?: 'text' | 'image';
+    }[];
+    /**
+     * Base64-encoded payloads for any image mentions in the prompt, in the
+     * order they appeared. Empty when no images were attached. Caller is
+     * responsible for forwarding these to the provider — the prompt text
+     * only contains a `[Image: path]` placeholder so the model has a label
+     * to anchor references to.
+     */
+    images: string[];
+}
+export declare function expandMentions(input: string, cwd: string): Promise<ExpandedPrompt>;
+//# sourceMappingURL=mentions.d.ts.map

package/dist/mentions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mentions.d.ts","sourceRoot":"","sources":["../src/mentions.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAwBH,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,OAAO,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,EAAE,CAAC;IAClF;;;;;;OAMG;IACH,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,wBAAsB,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CAwDxF"}

package/dist/mentions.js

@@ -0,0 +1,126 @@
+"use strict";
+/**
+ * @-mentions — detect `@path/to/file` patterns in user input and inline the
+ * file contents into the prompt so the model sees them in the first turn.
+ *
+ * user> explain @src/auth.ts
+ * → prompt becomes: "explain @src/auth.ts\n\n[File: src/auth.ts]\n<content>"
+ *
+ * Images (png/jpg/etc.) are NOT inlined as text — doing so dumps raw binary
+ * bytes decoded as UTF-8 into the prompt, and the model interprets the
+ * resulting gibberish as corrupt data. Instead, image mentions are
+ * collected into `images[]` as base64 data URLs so the caller can
+ * forward them on the provider's `images` field (Ollama vision models
+ * accept this; Bandit cloud forwards through to the gateway).
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.expandMentions = expandMentions;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const agent_core_1 = require("@burtson-labs/agent-core");
+const MENTION_REGEX = /(?:^|\s)@([^\s@]+)/g;
+/**
+ * redact secrets from @-mentioned file content with the
+ * same opt-out as the rest of the redactor (`BANDIT_NO_SECRET_REDACTION=1`).
+ * Pulled into a helper so the env check happens once per mention pass.
+ */
+function applyMentionSecretRedaction(text) {
+    if (/^(1|true)$/i.test(process.env.BANDIT_NO_SECRET_REDACTION ?? '')) {
+        return text;
+    }
+    if (!text || text.length === 0)
+        return text;
+    return (0, agent_core_1.redactSecretsString)(text);
+}
+const MAX_FILE_BYTES = 64 * 1024; // 64 KB per mention
+const MAX_MENTIONS = 8;
+const IMAGE_EXT = /\.(?:png|jpe?g|gif|webp|heic|bmp)$/i;
+async function expandMentions(input, cwd) {
+    const matches = [...input.matchAll(MENTION_REGEX)].slice(0, MAX_MENTIONS);
+    if (matches.length === 0)
+        return { prompt: input, mentions: [], images: [] };
+    const seen = new Set();
+    const sections = [];
+    const mentions = [];
+    const images = [];
+    for (const m of matches) {
+        const rel = m[1];
+        if (seen.has(rel))
+            continue;
+        seen.add(rel);
+        const abs = path.isAbsolute(rel) ? rel : path.resolve(cwd, rel);
+        // Guard: don't read outside the workspace.
+        if (!abs.startsWith(path.resolve(cwd))) {
+            mentions.push({ path: rel, bytes: 0, ok: false });
+            continue;
+        }
+        try {
+            const stat = await fs.promises.stat(abs);
+            if (!stat.isFile()) {
+                mentions.push({ path: rel, bytes: 0, ok: false });
+                continue;
+            }
+            const raw = await fs.promises.readFile(abs);
+            // Image branch: encode + register as attachment, emit a short
+            // placeholder so the prompt doesn't contain binary gibberish.
+            if (IMAGE_EXT.test(rel)) {
+                images.push(raw.toString('base64'));
+                sections.push(`[Image attached: ${rel} — ${raw.byteLength} bytes]`);
+                mentions.push({ path: rel, bytes: raw.byteLength, ok: true, kind: 'image' });
+                continue;
+            }
+            const truncated = raw.byteLength > MAX_FILE_BYTES;
+            const rawContent = raw.subarray(0, MAX_FILE_BYTES).toString('utf-8');
+            // redact secrets from @-mentioned file content before
+            // it lands in the user message. @-mentions inline file content
+            // into the prompt body verbatim; without this pass, dropping
+            // `@.env` or `@~/.aws/credentials` into the composer would ship
+            // every key in there to the model context and the session log.
+            // Same redactor that runs on tool-result output, applied here
+            // for parity. Opt-out via BANDIT_NO_SECRET_REDACTION env.
+            const content = applyMentionSecretRedaction(rawContent);
+            const suffix = truncated ? `\n… (truncated, full file is ${raw.byteLength} bytes)` : '';
+            sections.push(`[File: ${rel}]\n\`\`\`\n${content}${suffix}\n\`\`\``);
+            mentions.push({ path: rel, bytes: raw.byteLength, ok: true, kind: 'text' });
+        }
+        catch {
+            mentions.push({ path: rel, bytes: 0, ok: false });
+        }
+    }
+    const body = sections.length > 0 ? `${input}\n\n${sections.join('\n\n')}` : input;
+    return { prompt: body, mentions, images };
+}
+//# sourceMappingURL=mentions.js.map

package/dist/mentions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mentions.js","sourceRoot":"","sources":["../src/mentions.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;GAaG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqCH,wCAwDC;AA3FD,uCAAyB;AACzB,2CAA6B;AAC7B,yDAA+D;AAE/D,MAAM,aAAa,GAAG,qBAAqB,CAAC;AAE5C;;;;GAIG;AACH,SAAS,2BAA2B,CAAC,IAAY;IAC/C,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,IAAI,EAAE,CAAC,EAAE,CAAC;QACrE,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAC5C,OAAO,IAAA,gCAAmB,EAAC,IAAI,CAAC,CAAC;AACnC,CAAC;AACD,MAAM,cAAc,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,oBAAoB;AACtD,MAAM,YAAY,GAAG,CAAC,CAAC;AACvB,MAAM,SAAS,GAAG,qCAAqC,CAAC;AAejD,KAAK,UAAU,cAAc,CAAC,KAAa,EAAE,GAAW;IAC7D,MAAM,OAAO,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;IAC1E,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IAE7E,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,QAAQ,GAA+B,EAAE,CAAC;IAChD,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;QACxB,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACjB,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,SAAS;QAC5B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAEd,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChE,2CAA2C;QAC3C,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YACvC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAClD,SAAS;QACX,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;gBACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAClD,SAAS;YACX,CAAC;YACD,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC5C,8DAA8D;YAC9D,8DAA8D;YAC9D,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACpC,QAAQ,CAAC,IAAI,CAAC,oBAAoB,GAAG,MAAM,GAAG,CAAC,UAAU,SAAS,CAAC,CAAC;gBACpE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC7E,SAAS;YACX,CAAC;YACD,MAAM,SAAS,GAAG,GAAG,CAAC,UAAU,GAAG,cAAc,CAAC;YAClD,MAAM,UAAU,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACrE,sDAAsD;YACtD,+DAA+D;YAC/D,6DAA6D;YAC7D,gEAAgE;YAChE,+DAA+D;YAC/D,8DAA8D;YAC9D,0DAA0D;YAC1D,MAAM,OAAO,GAAG,2BAA2B,CAAC,UAAU,CAAC,CAAC;YACxD,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,gCAAgC,GAAG,CAAC,UAAU,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,QAAQ,CAAC,IAAI,CAAC,UAAU,GAAG,cAAc,OAAO,GAAG,MAAM,UAAU,CAAC,CAAC;YACrE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9E,CAAC;QAAC,MAAM,CAAC;YACP,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IAClF,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;AAC5C,CAAC"}

package/dist/ollamaModels.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Shared Ollama model-discovery helpers used by both the CLI and the
+ * VS Code extension. Both surfaces need to:
+ *   - list what's actually pulled on the local runtime (`/api/tags`),
+ *   - map an unpulled model to the closest installed match so users
+ *     don't have to know about `-it-qat` vs `-it-q4_K_M` suffixes.
+ *
+ * Keeping the logic here means CLI auto-switch and extension UI pickers
+ * give identical suggestions for the same installed set.
+ */
+export interface OllamaModelInfo {
+    /** Full tag as returned by `/api/tags` (e.g. "gemma3:12b-it-qat"). */
+    name: string;
+    /** Approximate on-disk size in bytes, when reported. */
+    size?: number;
+    /** Last modified ISO string, when reported. */
+    modifiedAt?: string;
+}
+/**
+ * Fetch the list of models installed on an Ollama runtime. Embedding-only
+ * models are NOT filtered out here — callers that want just chat-capable
+ * models should pass `{ chatOnly: true }` or filter via {@link isChatCapable}.
+ */
+export declare function listInstalledOllamaModels(baseUrl: string, options?: {
+    timeoutMs?: number;
+    chatOnly?: boolean;
+}): Promise<OllamaModelInfo[]>;
+/** Embedding-only models should be excluded from chat/agent picklists. */
+export declare function isChatCapable(name: string): boolean;
+/**
+ * Rank installed Ollama models by how well they match a requested name.
+ * Prefers: same family + same param-size > same family > any chat-capable
+ * model. Excludes embedding-only models from the returned list.
+ */
+export declare function suggestOllamaMatch(requested: string, installed: string[]): string[];
+//# sourceMappingURL=ollamaModels.d.ts.map

package/dist/ollamaModels.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollamaModels.d.ts","sourceRoot":"","sources":["../src/ollamaModels.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,MAAM,WAAW,eAAe;IAC9B,sEAAsE;IACtE,IAAI,EAAE,MAAM,CAAC;IACb,wDAAwD;IACxD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,+CAA+C;IAC/C,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;;;GAIG;AACH,wBAAsB,yBAAyB,CAC7C,OAAO,EAAE,MAAM,EACf,OAAO,GAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;CAAO,GACvD,OAAO,CAAC,eAAe,EAAE,CAAC,CAmB5B;AAED,0EAA0E;AAC1E,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAEnD;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE,CA0BnF"}

package/dist/ollamaModels.js

@@ -0,0 +1,83 @@
+"use strict";
+/**
+ * Shared Ollama model-discovery helpers used by both the CLI and the
+ * VS Code extension. Both surfaces need to:
+ *   - list what's actually pulled on the local runtime (`/api/tags`),
+ *   - map an unpulled model to the closest installed match so users
+ *     don't have to know about `-it-qat` vs `-it-q4_K_M` suffixes.
+ *
+ * Keeping the logic here means CLI auto-switch and extension UI pickers
+ * give identical suggestions for the same installed set.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.listInstalledOllamaModels = listInstalledOllamaModels;
+exports.isChatCapable = isChatCapable;
+exports.suggestOllamaMatch = suggestOllamaMatch;
+/**
+ * Fetch the list of models installed on an Ollama runtime. Embedding-only
+ * models are NOT filtered out here — callers that want just chat-capable
+ * models should pass `{ chatOnly: true }` or filter via {@link isChatCapable}.
+ */
+async function listInstalledOllamaModels(baseUrl, options = {}) {
+    const url = `${baseUrl.replace(/\/+$/, '')}/api/tags`;
+    const timeoutMs = options.timeoutMs ?? 3000;
+    try {
+        const res = await fetch(url, { signal: AbortSignal.timeout(timeoutMs) });
+        if (!res.ok)
+            return [];
+        const data = (await res.json());
+        const all = (data.models ?? [])
+            .filter((m) => typeof m?.name === 'string')
+            .map((m) => ({ name: m.name, size: m.size, modifiedAt: m.modified_at }));
+        if (options.chatOnly) {
+            return all.filter((m) => isChatCapable(m.name));
+        }
+        return all;
+    }
+    catch {
+        return [];
+    }
+}
+/** Embedding-only models should be excluded from chat/agent picklists. */
+function isChatCapable(name) {
+    return !/embed|embedding|nomic/i.test(name);
+}
+/**
+ * Rank installed Ollama models by how well they match a requested name.
+ * Prefers: same family + same param-size > same family > any chat-capable
+ * model. Excludes embedding-only models from the returned list.
+ */
+function suggestOllamaMatch(requested, installed) {
+    const [reqStem = '', reqTag = ''] = requested.split(':');
+    // "gemma4" → "gemma"; "qwen2.5-coder" → "qwen". Used to cross-match
+    // across major versions of the same family (gemma4:e4b → gemma3:12b-it-qat).
+    const familyBase = (stem) => stem.replace(/[\d.]+.*$/, '') || stem;
+    const reqFamily = familyBase(reqStem);
+    const score = (name) => {
+        if (!isChatCapable(name))
+            return -1;
+        const [stem, tag = ''] = name.split(':');
+        let s = 0;
+        if (stem === reqStem)
+            s += 20;
+        else if (stem.startsWith(reqStem) || reqStem.startsWith(stem))
+            s += 10;
+        else if (reqFamily.length >= 3 && familyBase(stem) === reqFamily)
+            s += 5;
+        if (s === 0)
+            return 0;
+        if (reqTag && tag.startsWith(reqTag))
+            s += 6;
+        else if (reqTag && tag.includes(reqTag))
+            s += 3;
+        if (/qat|q4|q5|q8/i.test(tag))
+            s += 1;
+        return s;
+    };
+    return installed
+        .map((name) => ({ name, s: score(name) }))
+        .filter((x) => x.s > 0)
+        .sort((a, b) => b.s - a.s)
+        .map((x) => x.name);
+}
+//# sourceMappingURL=ollamaModels.js.map

package/dist/ollamaModels.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollamaModels.js","sourceRoot":"","sources":["../src/ollamaModels.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;AAgBH,8DAsBC;AAGD,sCAEC;AAOD,gDA0BC;AAjED;;;;GAIG;AACI,KAAK,UAAU,yBAAyB,CAC7C,OAAe,EACf,UAAsD,EAAE;IAExD,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,WAAW,CAAC;IACtD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;IAC5C,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,OAAO,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAE7B,CAAC;QACF,MAAM,GAAG,GAAsB,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;aAC/C,MAAM,CAAC,CAAC,CAAC,EAA8D,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,KAAK,QAAQ,CAAC;aACtG,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QAC3E,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,0EAA0E;AAC1E,SAAgB,aAAa,CAAC,IAAY;IACxC,OAAO,CAAC,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC9C,CAAC;AAED;;;;GAIG;AACH,SAAgB,kBAAkB,CAAC,SAAiB,EAAE,SAAmB;IACvE,MAAM,CAAC,OAAO,GAAG,EAAE,EAAE,MAAM,GAAG,EAAE,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzD,oEAAoE;IACpE,6EAA6E;IAC7E,MAAM,UAAU,GAAG,CAAC,IAAY,EAAU,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC;IACnF,MAAM,SAAS,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;IAEtC,MAAM,KAAK,GAAG,CAAC,IAAY,EAAU,EAAE;QACrC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YAAE,OAAO,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,EAAE,GAAG,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,IAAI,IAAI,KAAK,OAAO;YAAE,CAAC,IAAI,EAAE,CAAC;aACzB,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;YAAE,CAAC,IAAI,EAAE,CAAC;aAClE,IAAI,SAAS,CAAC,MAAM,IAAI,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,SAAS;YAAE,CAAC,IAAI,CAAC,CAAC;QACzE,IAAI,CAAC,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QACtB,IAAI,MAAM,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,CAAC,IAAI,CAAC,CAAC;aACxC,IAAI,MAAM,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC;YAAE,CAAC,IAAI,CAAC,CAAC;QACtC,OAAO,CAAC,CAAC;IACX,CAAC,CAAC;IAEF,OAAO,SAAS;SACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;SACzC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;SACtB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACzB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AACxB,CAAC"}