@mnemo-mcp/core 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Omer Maksuti
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,123 @@
+# Mnemo
+
+> Persistent memory for Claude Code. Your AI never starts from scratch again.
+
+[![CI](https://github.com/omermaksutii/mnemo/actions/workflows/ci.yml/badge.svg)](https://github.com/omermaksutii/mnemo/actions)
+[![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
+
+Mnemo gives Claude Code a brain that survives across sessions. It captures decisions, conventions, and preferences — and recalls them by meaning, on demand, with sub-100ms semantic search.
+
+---
+
+## Install
+
+```bash
+# install the CLI
+npm install -g @mnemo-mcp/cli
+
+# install Mnemo into Claude Code (skill + MCP server)
+mnemo init
+
+# (optional) also wire auto-capture hooks
+mnemo init --with-hooks
+
+# verify
+mnemo doctor
+```
+
+After `mnemo init`, restart Claude Code. The `/mnemo` skill and the `mnemo_*` MCP tools are now available — Claude will use them automatically when relevant.
+
+## Use it from the terminal too
+
+```bash
+mnemo remember "our API auth uses OAuth2 with refresh tokens every 30min"
+mnemo remember --global "I prefer pnpm over npm"
+mnemo recall "what's our auth pattern?"
+mnemo list
+mnemo stats
+mnemo forget <id>
+mnemo export --out memories.json
+mnemo import memories.json
+```
+
+## What Mnemo solves
+
+Claude Code forgets everything when the session ends. `CLAUDE.md` partially helps — it's a static blob loaded on every turn — but it grows unwieldy fast and can't answer "do we have a precedent for X?"
+
+Mnemo gives you semantic, on-demand memory:
+
+| | `CLAUDE.md` | Mnemo |
+|---|---|---|
+| Capacity | A handful of paragraphs before token cost hurts | Tens of thousands of memories |
+| Retrieval | Always loaded, every turn | On demand, by meaning |
+| Updates | You edit a file by hand | Captured automatically or via `/teach` |
+| Cross-project | Per-project only | Project + global tiers |
+| Forgetting | Manually delete lines | `mnemo forget <id>` or `/forget` |
+
+## How it works
+
+```
+Claude Code session
+  ├── /mnemo skill              ← teaches Claude when to call the tools
+  ├── @mnemo-mcp/server server          ← exposes recall/remember/forget/list/stats
+  └── @mnemo-mcp/cli hooks           ← session-start, pre-task, post-edit auto-wiring
+              │
+              ▼
+       @mnemo-mcp/core
+       ├── ONNX all-MiniLM-L6-v2  (384-dim embeddings, ~25MB, lazy)
+       ├── HNSW vector index      (sub-100ms recall at 50k memories)
+       └── sql.js (WASM SQLite)   (~/.mnemo/memory.db)
+```
+
+Everything is local-first. No daemon. No cloud account. No telemetry.
+
+## Auto-capture rules
+
+By default the post-edit hook captures any change to:
+
+- `CLAUDE.md`, `AGENTS.md`, `GEMINI.md`
+- `*.adr.md`
+- `docs/decisions/**`
+- `docs/adr/**`
+
+Anything you write to those files becomes a project memory automatically. Disable with `mnemo init` (no `--with-hooks`).
+
+## Recall scoring
+
+```
+score = 0.7 × cosine_similarity + 0.2 × recency + 0.1 × access_boost
+```
+
+Recency decays with a 30-day half-life. Access boost saturates at 20 reads.
+
+## Layout
+
+```
+@mnemo-mcp/core   — pure TS memory engine
+@mnemo-mcp/server    — MCP server (drop into Claude Code)
+@mnemo-mcp/cli    — terminal commands + hook handlers + init
+```
+
+## Roadmap
+
+- ✅ **v0.1** — core engine + CLI
+- ✅ **v0.2** — MCP server (5 tools)
+- ✅ **v0.3** — hooks (session-start, pre-task, post-edit)
+- ✅ **v0.4** — `/mnemo` skill
+- ✅ **v1.0** — `mnemo init`, beefier doctor, CI, polish
+- ⏳ **v1.1** — team mode (git-synced shared memory)
+- ⏳ **v2.0** — opt-in hosted sync, web UI
+
+## Development
+
+```bash
+npm install
+npm test                                    # fast (44 tests, hash embedder)
+MNEMO_TEST_ONNX=1 MNEMO_E2E=1 npm test     # full incl. real ONNX (~5s)
+npm run build
+npm run lint
+```
+
+## License
+
+MIT — see [LICENSE](LICENSE).

package/dist/duration.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Parse a duration string like "30d", "12h", "45m", "10s", "2w" into milliseconds.
+ * Returns null if the input is invalid.
+ */
+export declare function parseDuration(input: string): number | null;
+export declare function expiresAtFromTtl(ttl: string | undefined, now?: number): number | null;
+export declare function sinceFromAgo(ago: string | undefined, now?: number): number | undefined;
+//# sourceMappingURL=duration.d.ts.map

package/dist/duration.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"duration.d.ts","sourceRoot":"","sources":["../src/duration.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAY1D;AAED,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,EAAE,GAAG,SAAa,GAAG,MAAM,GAAG,IAAI,CAIzF;AAED,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,EAAE,GAAG,SAAa,GAAG,MAAM,GAAG,SAAS,CAI1F"}

package/dist/duration.js

@@ -0,0 +1,31 @@
+/**
+ * Parse a duration string like "30d", "12h", "45m", "10s", "2w" into milliseconds.
+ * Returns null if the input is invalid.
+ */
+export function parseDuration(input) {
+    const m = /^(\d+)(s|m|h|d|w)$/i.exec(input.trim());
+    if (!m)
+        return null;
+    const n = Number(m[1]);
+    switch (m[2].toLowerCase()) {
+        case 's': return n * 1000;
+        case 'm': return n * 60_000;
+        case 'h': return n * 3_600_000;
+        case 'd': return n * 86_400_000;
+        case 'w': return n * 7 * 86_400_000;
+    }
+    return null;
+}
+export function expiresAtFromTtl(ttl, now = Date.now()) {
+    if (!ttl)
+        return null;
+    const ms = parseDuration(ttl);
+    return ms === null ? null : now + ms;
+}
+export function sinceFromAgo(ago, now = Date.now()) {
+    if (!ago)
+        return undefined;
+    const ms = parseDuration(ago);
+    return ms === null ? undefined : now - ms;
+}
+//# sourceMappingURL=duration.js.map

package/dist/duration.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"duration.js","sourceRoot":"","sources":["../src/duration.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,KAAa;IACzC,MAAM,CAAC,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACnD,IAAI,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IACpB,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,QAAQ,CAAC,CAAC,CAAC,CAAE,CAAC,WAAW,EAAE,EAAE,CAAC;QAC5B,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;QAC1B,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,MAAM,CAAC;QAC5B,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,SAAS,CAAC;QAC/B,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,UAAU,CAAC;QAChC,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC;IACtC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,GAAuB,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IACxE,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IACtB,MAAM,EAAE,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;IAC9B,OAAO,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,GAAuB,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IACpE,IAAI,CAAC,GAAG;QAAE,OAAO,SAAS,CAAC;IAC3B,MAAM,EAAE,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;IAC9B,OAAO,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC;AAC5C,CAAC"}

package/dist/embedder.d.ts

@@ -0,0 +1,19 @@
+export interface Embedder {
+    readonly dimension: number;
+    embed(text: string): Promise<Float32Array>;
+    embedBatch(texts: string[]): Promise<Float32Array[]>;
+}
+export type HashEmbedderOpts = {
+    dimension?: number;
+};
+/**
+ * Deterministic, dependency-free embedder for tests.
+ * Real semantic similarity is provided by OnnxEmbedder.
+ */
+export declare class HashEmbedder implements Embedder {
+    readonly dimension: number;
+    constructor(opts?: HashEmbedderOpts);
+    embed(text: string): Promise<Float32Array>;
+    embedBatch(texts: string[]): Promise<Float32Array[]>;
+}
+//# sourceMappingURL=embedder.d.ts.map

package/dist/embedder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedder.d.ts","sourceRoot":"","sources":["../src/embedder.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,QAAQ;IACvB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAC3C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;CACtD;AAED,MAAM,MAAM,gBAAgB,GAAG;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF;;;GAGG;AACH,qBAAa,YAAa,YAAW,QAAQ;IAC3C,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;gBAEf,IAAI,GAAE,gBAAqB;IAIjC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAiB1C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;CAG3D"}

package/dist/embedder.js

@@ -0,0 +1,33 @@
+import { createHash } from 'node:crypto';
+/**
+ * Deterministic, dependency-free embedder for tests.
+ * Real semantic similarity is provided by OnnxEmbedder.
+ */
+export class HashEmbedder {
+    dimension;
+    constructor(opts = {}) {
+        this.dimension = opts.dimension ?? 384;
+    }
+    async embed(text) {
+        const out = new Float32Array(this.dimension);
+        const tokens = text.toLowerCase().split(/\W+/).filter(Boolean);
+        for (const tok of tokens) {
+            const h = createHash('sha256').update(tok).digest();
+            for (let i = 0; i < this.dimension; i++) {
+                const byte = h[i % h.length] ?? 0;
+                out[i] += (byte / 255) * 2 - 1;
+            }
+        }
+        let mag = 0;
+        for (let i = 0; i < out.length; i++)
+            mag += out[i] * out[i];
+        mag = Math.sqrt(mag) || 1;
+        for (let i = 0; i < out.length; i++)
+            out[i] = out[i] / mag;
+        return out;
+    }
+    async embedBatch(texts) {
+        return Promise.all(texts.map(t => this.embed(t)));
+    }
+}
+//# sourceMappingURL=embedder.js.map

package/dist/embedder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedder.js","sourceRoot":"","sources":["../src/embedder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAYzC;;;GAGG;AACH,MAAM,OAAO,YAAY;IACd,SAAS,CAAS;IAE3B,YAAY,OAAyB,EAAE;QACrC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,GAAG,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/D,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;YACpD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBAClC,GAAG,CAAC,CAAC,CAAE,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QACD,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE;YAAE,GAAG,IAAI,GAAG,CAAC,CAAC,CAAE,GAAG,GAAG,CAAC,CAAC,CAAE,CAAC;QAC9D,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE;YAAE,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAE,GAAG,GAAG,CAAC;QAC5D,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,8 @@
+export declare const VERSION = "1.0.0";
+export * from './types.js';
+export { Mnemo } from './mnemo.js';
+export { HashEmbedder } from './embedder.js';
+export type { Embedder } from './embedder.js';
+export { resolveDataDir, projectHashOf, paths } from './paths.js';
+export { parseDuration, expiresAtFromTtl, sinceFromAgo } from './duration.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,UAAU,CAAC;AAC/B,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,YAAY,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC"}

package/dist/index.js

@@ -0,0 +1,7 @@
+export const VERSION = '1.0.0';
+export * from './types.js';
+export { Mnemo } from './mnemo.js';
+export { HashEmbedder } from './embedder.js';
+export { resolveDataDir, projectHashOf, paths } from './paths.js';
+export { parseDuration, expiresAtFromTtl, sinceFromAgo } from './duration.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC;AAC/B,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE7C,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC"}

package/dist/mnemo.d.ts

@@ -0,0 +1,22 @@
+import type { CaptureInput, ListFilter, MemoryHit, MemoryRecord, MnemoOpts, MnemoStats, PruneOpts, PruneResult, RecallOpts, UpdateInput } from './types.js';
+export declare class Mnemo {
+    private store;
+    private index;
+    private embedder;
+    private dataDir;
+    private constructor();
+    static open(opts?: MnemoOpts): Promise<Mnemo>;
+    /** True when the most recent `capture()` updated an existing memory instead of inserting a new one. */
+    lastCaptureDeduped: boolean;
+    capture(input: CaptureInput): Promise<MemoryRecord>;
+    recall(query: string, opts?: RecallOpts): Promise<MemoryHit[]>;
+    forget(id: string): Promise<void>;
+    update(id: string, fields: UpdateInput): Promise<MemoryRecord | null>;
+    list(filter?: ListFilter): Promise<MemoryRecord[]>;
+    stats(): Promise<MnemoStats>;
+    export(): Promise<MemoryRecord[]>;
+    import(records: MemoryRecord[]): Promise<void>;
+    prune(opts?: PruneOpts): Promise<PruneResult>;
+    close(): Promise<void>;
+}
+//# sourceMappingURL=mnemo.d.ts.map

package/dist/mnemo.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mnemo.d.ts","sourceRoot":"","sources":["../src/mnemo.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EACV,YAAY,EACZ,UAAU,EACV,SAAS,EACT,YAAY,EACZ,SAAS,EACT,UAAU,EACV,SAAS,EACT,WAAW,EACX,UAAU,EACV,WAAW,EACZ,MAAM,YAAY,CAAC;AAGpB,qBAAa,KAAK;IAEd,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,OAAO;IAJjB,OAAO;WAOM,IAAI,CAAC,IAAI,GAAE,SAAc,GAAG,OAAO,CAAC,KAAK,CAAC;IAiCvD,uGAAuG;IACvG,kBAAkB,UAAS;IAErB,OAAO,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAkDnD,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,UAAe,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IA8BlE,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMjC,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAWrE,IAAI,CAAC,MAAM,GAAE,UAAe,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAItD,KAAK,IAAI,OAAO,CAAC,UAAU,CAAC;IAiB5B,MAAM,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;IAIjC,MAAM,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAa9C,KAAK,CAAC,IAAI,GAAE,SAAc,GAAG,OAAO,CAAC,WAAW,CAAC;IAqEjD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAI7B"}

package/dist/mnemo.js

@@ -0,0 +1,266 @@
+import { randomUUID } from 'node:crypto';
+import { Store } from './store.js';
+import { VectorIndex } from './vector-index.js';
+import { HashEmbedder } from './embedder.js';
+import { score } from './ranker.js';
+import { paths, resolveDataDir } from './paths.js';
+import { stat } from 'node:fs/promises';
+export class Mnemo {
+    store;
+    index;
+    embedder;
+    dataDir;
+    constructor(store, index, embedder, dataDir) {
+        this.store = store;
+        this.index = index;
+        this.embedder = embedder;
+        this.dataDir = dataDir;
+    }
+    static async open(opts = {}) {
+        const dataDir = resolveDataDir(opts.dataDir);
+        const p = paths(dataDir);
+        let embedder;
+        if (opts.embedderType === 'onnx') {
+            const { OnnxEmbedder } = await import('./onnx-embedder.js');
+            embedder = await OnnxEmbedder.load(p.modelDir);
+        }
+        else {
+            embedder = new HashEmbedder({ dimension: 384 });
+        }
+        const store = await Store.open(p.dbFile);
+        const index = await VectorIndex.open({
+            path: p.indexFile,
+            dimension: embedder.dimension,
+            maxElements: 100_000,
+        });
+        // Rebuild missing index entries from store (covers cold start where the
+        // index was deleted but the database survived).
+        const all = await store.list({ includeExpired: true });
+        if (index.size() < all.length) {
+            for (const rec of all) {
+                const v = await embedder.embed(rec.content);
+                await index.add(rec.id, v);
+            }
+            await index.save();
+        }
+        return new Mnemo(store, index, embedder, dataDir);
+    }
+    /** True when the most recent `capture()` updated an existing memory instead of inserting a new one. */
+    lastCaptureDeduped = false;
+    async capture(input) {
+        const dedupThreshold = input.dedupThreshold ?? 0.95;
+        const v = await this.embedder.embed(input.content);
+        this.lastCaptureDeduped = false;
+        if (dedupThreshold > 0 && this.index.size() > 0) {
+            const candidates = await this.index.query(v, 5);
+            for (const cand of candidates) {
+                if (cand.similarity < dedupThreshold)
+                    break;
+                const existing = await this.store.get(cand.id);
+                if (!existing)
+                    continue;
+                const sameScope = existing.scope === (input.scope ?? 'project') &&
+                    existing.projectHash === (input.projectHash ?? null);
+                if (!sameScope)
+                    continue;
+                const updated = await this.store.update(existing.id, {
+                    content: input.content,
+                    tags: dedupeTags([...(existing.tags ?? []), ...(input.tags ?? [])]),
+                    expiresAt: input.expiresAt !== undefined ? input.expiresAt : existing.expiresAt,
+                });
+                if (updated) {
+                    const v2 = await this.embedder.embed(updated.content);
+                    await this.index.add(updated.id, v2);
+                    await this.index.save();
+                    this.lastCaptureDeduped = true;
+                    return updated;
+                }
+            }
+        }
+        const now = Date.now();
+        const rec = {
+            id: randomUUID(),
+            scope: input.scope ?? 'project',
+            projectHash: input.projectHash ?? null,
+            source: input.source ?? 'manual',
+            content: input.content,
+            tags: input.tags ?? [],
+            createdAt: now,
+            updatedAt: now,
+            accessCount: 0,
+            lastAccessedAt: now,
+            expiresAt: input.expiresAt ?? null,
+        };
+        await this.store.upsert(rec);
+        await this.index.add(rec.id, v);
+        await this.index.save();
+        return rec;
+    }
+    async recall(query, opts = {}) {
+        const k = opts.k ?? 10;
+        const minScore = opts.minScore ?? 0;
+        const v = await this.embedder.embed(query);
+        const candidates = await this.index.query(v, Math.max(k * 4, 20));
+        const requiredTags = opts.tags ?? [];
+        const sources = opts.source ? (Array.isArray(opts.source) ? opts.source : [opts.source]) : null;
+        const since = opts.since ?? 0;
+        const now = Date.now();
+        const out = [];
+        for (const cand of candidates) {
+            const rec = await this.store.get(cand.id);
+            if (!rec)
+                continue;
+            if (opts.scope && opts.scope !== 'all' && rec.scope !== opts.scope)
+                continue;
+            if (opts.projectHash !== undefined && rec.projectHash !== opts.projectHash)
+                continue;
+            if (!opts.includeExpired && rec.expiresAt !== null && rec.expiresAt <= now)
+                continue;
+            if (since && rec.updatedAt < since)
+                continue;
+            if (sources && !sources.includes(rec.source))
+                continue;
+            if (requiredTags.length && !requiredTags.every(t => rec.tags.includes(t)))
+                continue;
+            const s = score(cand.similarity, rec);
+            if (s < minScore)
+                continue;
+            out.push({ record: rec, score: s, similarity: cand.similarity });
+        }
+        out.sort((a, b) => b.score - a.score);
+        const top = out.slice(0, k);
+        for (const hit of top)
+            await this.store.bumpAccess(hit.record.id);
+        return top;
+    }
+    async forget(id) {
+        await this.store.delete(id);
+        await this.index.remove(id);
+        await this.index.save();
+    }
+    async update(id, fields) {
+        const updated = await this.store.update(id, fields);
+        if (!updated)
+            return null;
+        if (fields.content !== undefined) {
+            const v = await this.embedder.embed(updated.content);
+            await this.index.add(updated.id, v);
+            await this.index.save();
+        }
+        return updated;
+    }
+    async list(filter = {}) {
+        return this.store.list(filter);
+    }
+    async stats() {
+        const counts = await this.store.count();
+        let storageBytes = 0;
+        try {
+            const s = await stat(paths(this.dataDir).dbFile);
+            storageBytes = s.size;
+        }
+        catch { }
+        return {
+            totalMemories: counts.total,
+            byScope: counts.byScope,
+            indexSize: this.index.size(),
+            embeddingDimension: this.embedder.dimension,
+            storageBytes,
+            expired: counts.expired,
+        };
+    }
+    async export() {
+        return this.store.list({ includeExpired: true });
+    }
+    async import(records) {
+        for (const rec of records) {
+            await this.store.upsert({
+                ...rec,
+                source: 'imported',
+                expiresAt: rec.expiresAt ?? null,
+            });
+            const v = await this.embedder.embed(rec.content);
+            await this.index.add(rec.id, v);
+        }
+        await this.index.save();
+    }
+    async prune(opts = {}) {
+        const expiredOn = opts.expired !== false;
+        const dupThr = opts.duplicateThreshold ?? 0.97;
+        const minAccess = opts.minAccessCount ?? 0;
+        const staleDays = opts.staleAfterDays ?? 0;
+        const dryRun = !!opts.dryRun;
+        const now = Date.now();
+        const all = await this.store.list({ includeExpired: true });
+        const expired = [];
+        const stale = [];
+        const duplicates = [];
+        if (expiredOn) {
+            for (const r of all) {
+                if (r.expiresAt !== null && r.expiresAt <= now)
+                    expired.push(r);
+            }
+        }
+        if (minAccess > 0 && staleDays > 0) {
+            const cutoff = now - staleDays * 86400_000;
+            for (const r of all) {
+                if (r.accessCount < minAccess && r.updatedAt < cutoff && !expired.includes(r)) {
+                    stale.push(r);
+                }
+            }
+        }
+        if (dupThr > 0) {
+            const seen = new Set();
+            // Walk by most-recently-accessed first; a more recently accessed record wins.
+            const sorted = [...all].sort((a, b) => b.lastAccessedAt - a.lastAccessedAt);
+            for (const keep of sorted) {
+                if (seen.has(keep.id) || expired.includes(keep))
+                    continue;
+                const v = await this.embedder.embed(keep.content);
+                const peers = await this.index.query(v, 8);
+                for (const peer of peers) {
+                    if (peer.id === keep.id)
+                        continue;
+                    if (peer.similarity < dupThr)
+                        continue;
+                    const dropCandidate = await this.store.get(peer.id);
+                    if (!dropCandidate)
+                        continue;
+                    if (dropCandidate.scope !== keep.scope || dropCandidate.projectHash !== keep.projectHash)
+                        continue;
+                    if (dropCandidate.lastAccessedAt > keep.lastAccessedAt)
+                        continue;
+                    if (seen.has(dropCandidate.id) || expired.includes(dropCandidate))
+                        continue;
+                    duplicates.push({ kept: keep, dropped: dropCandidate });
+                    seen.add(dropCandidate.id);
+                }
+                seen.add(keep.id);
+            }
+        }
+        let totalDeleted = 0;
+        if (!dryRun) {
+            const toDelete = new Set([
+                ...expired.map(r => r.id),
+                ...stale.map(r => r.id),
+                ...duplicates.map(d => d.dropped.id),
+            ]);
+            for (const id of toDelete) {
+                await this.store.delete(id);
+                await this.index.remove(id);
+            }
+            if (toDelete.size > 0)
+                await this.index.save();
+            totalDeleted = toDelete.size;
+        }
+        return { expired, duplicates, stale, totalDeleted };
+    }
+    async close() {
+        await this.index.save();
+        this.store.close();
+    }
+}
+function dedupeTags(tags) {
+    return Array.from(new Set(tags.map(t => t.trim()).filter(Boolean)));
+}
+//# sourceMappingURL=mnemo.js.map

package/dist/mnemo.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mnemo.js","sourceRoot":"","sources":["../src/mnemo.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAiB,MAAM,eAAe,CAAC;AAC5D,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAanD,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAExC,MAAM,OAAO,KAAK;IAEN;IACA;IACA;IACA;IAJV,YACU,KAAY,EACZ,KAAkB,EAClB,QAAkB,EAClB,OAAe;QAHf,UAAK,GAAL,KAAK,CAAO;QACZ,UAAK,GAAL,KAAK,CAAa;QAClB,aAAQ,GAAR,QAAQ,CAAU;QAClB,YAAO,GAAP,OAAO,CAAQ;IACtB,CAAC;IAEJ,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAkB,EAAE;QACpC,MAAM,OAAO,GAAG,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;QAEzB,IAAI,QAAkB,CAAC;QACvB,IAAI,IAAI,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;YACjC,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;YAC5D,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QACjD,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,IAAI,YAAY,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,MAAM,WAAW,CAAC,IAAI,CAAC;YACnC,IAAI,EAAE,CAAC,CAAC,SAAS;YACjB,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,WAAW,EAAE,OAAO;SACrB,CAAC,CAAC;QAEH,wEAAwE;QACxE,gDAAgD;QAChD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,IAAI,KAAK,CAAC,IAAI,EAAE,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC;YAC9B,KAAK,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;gBACtB,MAAM,CAAC,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC5C,MAAM,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAC7B,CAAC;YACD,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC;QAED,OAAO,IAAI,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED,uGAAuG;IACvG,kBAAkB,GAAG,KAAK,CAAC;IAE3B,KAAK,CAAC,OAAO,CAAC,KAAmB;QAC/B,MAAM,cAAc,GAAG,KAAK,CAAC,cAAc,IAAI,IAAI,CAAC;QACpD,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnD,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC;QAEhC,IAAI,cAAc,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;YAChD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAChD,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,IAAI,IAAI,CAAC,UAAU,GAAG,cAAc;oBAAE,MAAM;gBAC5C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC/C,IAAI,CAAC,QAAQ;oBAAE,SAAS;gBACxB,MAAM,SAAS,GACb,QAAQ,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,KAAK,IAAI,SAAS,CAAC;oBAC7C,QAAQ,CAAC,WAAW,KAAK,CAAC,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC;gBACvD,IAAI,CAAC,SAAS;oBAAE,SAAS;gBACzB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE;oBACnD,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,IAAI,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC;oBACnE,SAAS,EAAE,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS;iBAChF,CAAC,CAAC;gBACH,IAAI,OAAO,EAAE,CAAC;oBACZ,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBACtD,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;oBACrC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;oBACxB,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;oBAC/B,OAAO,OAAO,CAAC;gBACjB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,GAAG,GAAiB;YACxB,EAAE,EAAE,UAAU,EAAE;YAChB,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,SAAS;YAC/B,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,IAAI;YACtC,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,QAAQ;YAChC,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;YACtB,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;YACd,WAAW,EAAE,CAAC;YACd,cAAc,EAAE,GAAG;YACnB,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,IAAI;SACnC,CAAC;QACF,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAChC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACxB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAmB,EAAE;QAC/C,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAClE,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAChG,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,GAAG,GAAgB,EAAE,CAAC;QAE5B,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC1C,IAAI,CAAC,GAAG;gBAAE,SAAS;YACnB,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK,IAAI,GAAG,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK;gBAAE,SAAS;YAC7E,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,IAAI,GAAG,CAAC,WAAW,KAAK,IAAI,CAAC,WAAW;gBAAE,SAAS;YACrF,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,GAAG,CAAC,SAAS,KAAK,IAAI,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG;gBAAE,SAAS;YACrF,IAAI,KAAK,IAAI,GAAG,CAAC,SAAS,GAAG,KAAK;gBAAE,SAAS;YAC7C,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC;gBAAE,SAAS;YACvD,IAAI,YAAY,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;gBAAE,SAAS;YACpF,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;YACtC,IAAI,CAAC,GAAG,QAAQ;gBAAE,SAAS;YAC3B,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QACnE,CAAC;QACD,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,KAAK,MAAM,GAAG,IAAI,GAAG;YAAE,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAClE,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC5B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC5B,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,MAAmB;QAC1C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAC1B,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACjC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YACpC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QAC1B,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAqB,EAAE;QAChC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACxC,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC;YACjD,YAAY,GAAG,CAAC,CAAC,IAAI,CAAC;QACxB,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;QACV,OAAO;YACL,aAAa,EAAE,MAAM,CAAC,KAAK;YAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;YAC5B,kBAAkB,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS;YAC3C,YAAY;YACZ,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAuB;QAClC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;gBACtB,GAAG,GAAG;gBACN,MAAM,EAAE,UAAU;gBAClB,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,IAAI;aACjC,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAClC,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAAkB,EAAE;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,IAAI,IAAI,CAAC;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,IAAI,CAAC,CAAC;QAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,IAAI,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;QAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAmB,EAAE,CAAC;QACnC,MAAM,KAAK,GAAmB,EAAE,CAAC;QACjC,MAAM,UAAU,GAAoD,EAAE,CAAC;QAEvE,IAAI,SAAS,EAAE,CAAC;YACd,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;gBACpB,IAAI,CAAC,CAAC,SAAS,KAAK,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,GAAG;oBAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClE,CAAC;QACH,CAAC;QAED,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,GAAG,GAAG,SAAS,GAAG,SAAS,CAAC;YAC3C,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;gBACpB,IAAI,CAAC,CAAC,WAAW,GAAG,SAAS,IAAI,CAAC,CAAC,SAAS,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC9E,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,GAAG,CAAC,EAAE,CAAC;YACf,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/B,8EAA8E;YAC9E,MAAM,MAAM,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC,CAAC;YAC5E,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;gBAC1B,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;oBAAE,SAAS;gBAC1D,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC3C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,IAAI,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE;wBAAE,SAAS;oBAClC,IAAI,IAAI,CAAC,UAAU,GAAG,MAAM;wBAAE,SAAS;oBACvC,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACpD,IAAI,CAAC,aAAa;wBAAE,SAAS;oBAC7B,IAAI,aAAa,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,IAAI,aAAa,CAAC,WAAW,KAAK,IAAI,CAAC,WAAW;wBAAE,SAAS;oBACnG,IAAI,aAAa,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc;wBAAE,SAAS;oBACjE,IAAI,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;wBAAE,SAAS;oBAC5E,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;oBACxD,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;gBAC7B,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACpB,CAAC;QACH,CAAC;QAED,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAS;gBAC/B,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzB,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;aACrC,CAAC,CAAC;YACH,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;gBAC1B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC5B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;YACD,IAAI,QAAQ,CAAC,IAAI,GAAG,CAAC;gBAAE,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YAC/C,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC;QAC/B,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACxB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;CACF;AAED,SAAS,UAAU,CAAC,IAAc;IAChC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AACtE,CAAC"}

package/dist/onnx-embedder.d.ts

@@ -0,0 +1,10 @@
+import type { Embedder } from './embedder.js';
+export declare class OnnxEmbedder implements Embedder {
+    private pipe;
+    readonly dimension = 384;
+    private constructor();
+    static load(modelCacheDir: string): Promise<OnnxEmbedder>;
+    embed(text: string): Promise<Float32Array>;
+    embedBatch(texts: string[]): Promise<Float32Array[]>;
+}
+//# sourceMappingURL=onnx-embedder.d.ts.map

package/dist/onnx-embedder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"onnx-embedder.d.ts","sourceRoot":"","sources":["../src/onnx-embedder.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAS9C,qBAAa,YAAa,YAAW,QAAQ;IAGvB,OAAO,CAAC,IAAI;IAFhC,QAAQ,CAAC,SAAS,OAAO;IAEzB,OAAO;WAEM,IAAI,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAYzD,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAK1C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;CAQ3D"}

package/dist/onnx-embedder.js

@@ -0,0 +1,30 @@
+import { mkdir } from 'node:fs/promises';
+const MODEL_ID = 'Xenova/all-MiniLM-L6-v2';
+export class OnnxEmbedder {
+    pipe;
+    dimension = 384;
+    constructor(pipe) {
+        this.pipe = pipe;
+    }
+    static async load(modelCacheDir) {
+        await mkdir(modelCacheDir, { recursive: true });
+        const transformers = (await import('@huggingface/transformers'));
+        transformers.env.cacheDir = modelCacheDir;
+        transformers.env.allowLocalModels = false;
+        const pipe = await transformers.pipeline('feature-extraction', MODEL_ID);
+        return new OnnxEmbedder(pipe);
+    }
+    async embed(text) {
+        const out = await this.pipe(text, { pooling: 'mean', normalize: true });
+        return new Float32Array(out.data.slice(0, this.dimension));
+    }
+    async embedBatch(texts) {
+        const out = await this.pipe(texts, { pooling: 'mean', normalize: true });
+        const result = [];
+        for (let i = 0; i < texts.length; i++) {
+            result.push(new Float32Array(out.data.slice(i * this.dimension, (i + 1) * this.dimension)));
+        }
+        return result;
+    }
+}
+//# sourceMappingURL=onnx-embedder.js.map

package/dist/onnx-embedder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"onnx-embedder.js","sourceRoot":"","sources":["../src/onnx-embedder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAGzC,MAAM,QAAQ,GAAG,yBAAyB,CAAC;AAO3C,MAAM,OAAO,YAAY;IAGK;IAFnB,SAAS,GAAG,GAAG,CAAC;IAEzB,YAA4B,IAA+B;QAA/B,SAAI,GAAJ,IAAI,CAA2B;IAAG,CAAC;IAE/D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,aAAqB;QACrC,MAAM,KAAK,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,MAAM,YAAY,GAAG,CAAC,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAG9D,CAAC;QACF,YAAY,CAAC,GAAG,CAAC,QAAQ,GAAG,aAAa,CAAC;QAC1C,YAAY,CAAC,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAC;QACzE,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxE,OAAO,IAAI,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzE,MAAM,MAAM,GAAmB,EAAE,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC9F,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}