tryaii-dre 0.2.0

package/dist/centroids/generator.js

@@ -0,0 +1,120 @@
+/**
+ * Centroid generator -- creates benchmark centroids from training queries.
+ *
+ * Centroids are the average embedding of all training queries for a benchmark.
+ * They are used by the EmbeddingClassifier to measure how similar a user's
+ * prompt is to each benchmark category.
+ *
+ * Centroids are regenerated when the embedding model changes, because different
+ * models produce different vector spaces.
+ */
+import { readFileSync, writeFileSync, mkdirSync } from 'node:fs';
+import { dirname, join } from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { vectorMean, vectorNormalize } from '../utils/math.js';
+const currentDir = dirname(fileURLToPath(import.meta.url));
+/** Path to bundled training queries. */
+export const TRAINING_QUERIES_PATH = join(currentDir, 'data', 'trainingQueries.json');
+/**
+ * Generates and manages benchmark centroids.
+ *
+ * Centroids are the average embedding vector of representative queries
+ * for each benchmark. When a user sends a prompt, we compute cosine
+ * similarity between their prompt's embedding and each centroid to
+ * determine what kind of task they're asking about.
+ */
+export class CentroidGenerator {
+    constructor(embeddingProvider) {
+        this._provider = embeddingProvider;
+    }
+    /**
+     * Generate centroids from training queries.
+     *
+     * @param trainingQueries - Dict of benchmark_name -> list of queries.
+     *                          If undefined, uses bundled default queries.
+     * @returns Dict of benchmark_name -> centroid vector (number array).
+     */
+    generate(trainingQueries) {
+        if (!trainingQueries) {
+            trainingQueries = this._loadDefaultQueries();
+        }
+        const centroids = {};
+        for (const [benchmark, queries] of Object.entries(trainingQueries)) {
+            // Embed all queries for this benchmark
+            const embeddings = this._provider.embedBatch(queries);
+            // Centroid = average of all embeddings, then normalize
+            const centroid = vectorNormalize(vectorMean(embeddings));
+            centroids[benchmark] = centroid;
+        }
+        return centroids;
+    }
+    /**
+     * Async version of `generate` -- routes through the provider's async path
+     * so it works with async-only providers like LocalEmbeddingProvider.
+     * Sync providers work too via the base class's default async fallback.
+     */
+    async generateAsync(trainingQueries) {
+        if (!trainingQueries) {
+            trainingQueries = this._loadDefaultQueries();
+        }
+        const centroids = {};
+        for (const [benchmark, queries] of Object.entries(trainingQueries)) {
+            const embeddings = await this._provider.embedBatchAsync(queries);
+            const centroid = vectorNormalize(vectorMean(embeddings));
+            centroids[benchmark] = centroid;
+        }
+        return centroids;
+    }
+    /**
+     * Generate a single centroid for a custom benchmark.
+     *
+     * @param benchmarkName - Name of the benchmark.
+     * @param queries - Representative queries for this benchmark.
+     * @returns Centroid vector (number array).
+     */
+    generateFromCustom(benchmarkName, queries) {
+        const embeddings = this._provider.embedBatch(queries);
+        return vectorNormalize(vectorMean(embeddings));
+    }
+    /**
+     * Async version of `generateFromCustom`. Works with any provider
+     * (sync via default fallback, async via its native async path).
+     */
+    async generateFromCustomAsync(benchmarkName, queries) {
+        const embeddings = await this._provider.embedBatchAsync(queries);
+        return vectorNormalize(vectorMean(embeddings));
+    }
+    /** Save centroids to a JSON file. */
+    save(centroids, path) {
+        mkdirSync(dirname(path), { recursive: true });
+        const data = {
+            metadata: {
+                model: this._provider.modelName,
+                dimension: this._provider.dimension,
+                benchmark_count: Object.keys(centroids).length,
+            },
+            centroids,
+        };
+        writeFileSync(path, JSON.stringify(data));
+    }
+    /** Load centroids from a JSON file. */
+    static load(path) {
+        const raw = readFileSync(path, 'utf-8');
+        const data = JSON.parse(raw);
+        return {
+            centroids: data.centroids,
+            metadata: data.metadata,
+        };
+    }
+    /** Load bundled training queries. */
+    _loadDefaultQueries() {
+        const raw = readFileSync(TRAINING_QUERIES_PATH, 'utf-8');
+        const data = JSON.parse(raw);
+        const queries = {};
+        for (const [name, benchData] of Object.entries(data.benchmarks)) {
+            queries[name] = benchData.queries;
+        }
+        return queries;
+    }
+}
+//# sourceMappingURL=generator.js.map

package/dist/centroids/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../src/centroids/generator.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACjE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAIzC,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAE/D,MAAM,UAAU,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAE3D,wCAAwC;AACxC,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,sBAAsB,CAAC,CAAC;AAEtF;;;;;;;GAOG;AACH,MAAM,OAAO,iBAAiB;IAG5B,YAAY,iBAAwC;QAClD,IAAI,CAAC,SAAS,GAAG,iBAAiB,CAAC;IACrC,CAAC;IAED;;;;;;OAMG;IACH,QAAQ,CAAC,eAA0C;QACjD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,eAAe,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC/C,CAAC;QAED,MAAM,SAAS,GAA6B,EAAE,CAAC;QAE/C,KAAK,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;YACnE,uCAAuC;YACvC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YAEtD,uDAAuD;YACvD,MAAM,QAAQ,GAAG,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;YACzD,SAAS,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC;QAClC,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,aAAa,CAAC,eAA0C;QAC5D,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,eAAe,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC/C,CAAC;QAED,MAAM,SAAS,GAA6B,EAAE,CAAC;QAE/C,KAAK,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;YACnE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACjE,MAAM,QAAQ,GAAG,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;YACzD,SAAS,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC;QAClC,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;;;OAMG;IACH,kBAAkB,CAAC,aAAqB,EAAE,OAAiB;QACzD,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACtD,OAAO,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;IACjD,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,uBAAuB,CAAC,aAAqB,EAAE,OAAiB;QACpE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QACjE,OAAO,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,qCAAqC;IACrC,IAAI,CAAC,SAAmC,EAAE,IAAY;QACpD,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE9C,MAAM,IAAI,GAAkB;YAC1B,QAAQ,EAAE;gBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS;gBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS;gBACnC,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM;aAC/C;YACD,SAAS;SACV,CAAC;QAEF,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,uCAAuC;IACvC,MAAM,CAAC,IAAI,CAAC,IAAY;QACtB,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACxC,MAAM,IAAI,GAAkB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAE5C,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;IACJ,CAAC;IAED,qCAAqC;IAC7B,mBAAmB;QACzB,MAAM,GAAG,GAAG,YAAY,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;QACzD,MAAM,IAAI,GAAwB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAElD,MAAM,OAAO,GAA6B,EAAE,CAAC;QAC7C,KAAK,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YAChE,OAAO,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC;QACpC,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/centroids/index.d.ts

@@ -0,0 +1,3 @@
+export { CentroidGenerator, TRAINING_QUERIES_PATH } from './generator.js';
+export { CentroidLoader } from './loader.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/centroids/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/centroids/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}

package/dist/centroids/index.js

@@ -0,0 +1,3 @@
+export { CentroidGenerator, TRAINING_QUERIES_PATH } from './generator.js';
+export { CentroidLoader } from './loader.js';
+//# sourceMappingURL=index.js.map

package/dist/centroids/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/centroids/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}

package/dist/centroids/loader.d.ts

@@ -0,0 +1,87 @@
+/**
+ * Centroid loader -- handles lazy initialization and model compatibility.
+ *
+ * Loading priority:
+ *   1. In-memory cache (already loaded)
+ *   2. User's cache directory (previously generated for their model)
+ *   3. Bundled static file (ships with package for default model -- zero delay)
+ *   4. Generate from training queries (only if using a non-default model)
+ */
+import { BaseEmbeddingProvider } from '../embeddings/base.js';
+/**
+ * Stable fingerprint of the benchmark set present in a centroid file.
+ *
+ * Defined as the sorted benchmark names joined by "|". Used to detect when a
+ * cached/bundled centroid file was generated against a different benchmark set
+ * and must be regenerated. Must stay identical to the Python SDK's
+ * `benchmark_fingerprint` (centroids/generator.py).
+ */
+export declare function benchmarkFingerprint(benchmarkNames: Iterable<string>): string;
+/**
+ * Manages centroid lifecycle: load, validate, regenerate.
+ *
+ * For the default embedding model (all-MiniLM-L6-v2), centroids are
+ * bundled with the package -- zero first-run delay. For other models,
+ * centroids are generated on first use and cached to disk.
+ */
+export declare class CentroidLoader {
+    private _provider;
+    private _centroids;
+    private _generator;
+    private _userCachePath;
+    constructor(embeddingProvider: BaseEmbeddingProvider, userCachePath?: string);
+    /**
+     * Get centroids, loading from best available source.
+     *
+     * Priority: memory > user cache > bundled static > generate fresh.
+     *
+     * Synchronous path -- if regeneration is needed (non-default model,
+     * first run) and the provider is async-only, this will throw. Async
+     * callers should use `getCentroidsAsync()` instead.
+     */
+    getCentroids(): Record<string, number[]>;
+    /**
+     * Async version of `getCentroids`. Works with any embedding provider --
+     * if centroid regeneration is needed, routes through the provider's
+     * async path.
+     */
+    getCentroidsAsync(): Promise<Record<string, number[]>>;
+    /**
+     * Try memory / user cache / bundled centroid file. Returns null if none
+     * of these sources have a valid file for the current provider's model.
+     * Shared between the sync and async load paths.
+     */
+    private _tryLoadFromFiles;
+    private _tryLoad;
+    private _regenerate;
+    private _regenerateAsync;
+    /**
+     * Force regeneration of centroids.
+     *
+     * @param customQueries - Optional custom training queries. If undefined, uses defaults.
+     */
+    regenerate(customQueries?: Record<string, string[]>): Record<string, number[]>;
+    /**
+     * Add a custom benchmark centroid to the existing set (sync).
+     *
+     * Requires a sync-capable embedding provider. Callers with async-only
+     * providers (e.g. LocalEmbeddingProvider) should use
+     * `addBenchmarkCentroidAsync()`.
+     *
+     * @param benchmarkName - Name of the new benchmark.
+     * @param queries - Representative queries for this benchmark.
+     * @returns The generated centroid vector.
+     */
+    addBenchmarkCentroid(benchmarkName: string, queries: string[]): number[];
+    /**
+     * Async version of `addBenchmarkCentroid`. Works with any provider.
+     * Mutates the in-memory centroid map so subsequent `getCentroids()` calls
+     * (including from classifiers sharing this loader) see the new benchmark.
+     */
+    addBenchmarkCentroidAsync(benchmarkName: string, queries: string[]): Promise<number[]>;
+    /** Remove a benchmark centroid. Returns true if it existed. */
+    removeBenchmark(benchmarkName: string): boolean;
+    /** List all available benchmark names. */
+    get availableBenchmarks(): string[];
+}
+//# sourceMappingURL=loader.d.ts.map

package/dist/centroids/loader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"loader.d.ts","sourceRoot":"","sources":["../../src/centroids/loader.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAMH,OAAO,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAc9D;;;;;;;GAOG;AACH,wBAAgB,oBAAoB,CAAC,cAAc,EAAE,QAAQ,CAAC,MAAM,CAAC,GAAG,MAAM,CAE7E;AA6BD;;;;;;GAMG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,SAAS,CAAwB;IACzC,OAAO,CAAC,UAAU,CAAyC;IAC3D,OAAO,CAAC,UAAU,CAAoB;IACtC,OAAO,CAAC,cAAc,CAAgB;gBAGpC,iBAAiB,EAAE,qBAAqB,EACxC,aAAa,CAAC,EAAE,MAAM;IAOxB;;;;;;;;OAQG;IACH,YAAY,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IAMxC;;;;OAIG;IACG,iBAAiB,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAM5D;;;;OAIG;IACH,OAAO,CAAC,iBAAiB;IAuBzB,OAAO,CAAC,QAAQ;IAoChB,OAAO,CAAC,WAAW;YAYL,gBAAgB;IAW9B;;;;OAIG;IACH,UAAU,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IAW9E;;;;;;;;;;OAUG;IACH,oBAAoB,CAAC,aAAa,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE;IAaxE;;;;OAIG;IACG,yBAAyB,CAAC,aAAa,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAY5F,+DAA+D;IAC/D,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO;IAY/C,0CAA0C;IAC1C,IAAI,mBAAmB,IAAI,MAAM,EAAE,CAElC;CACF"}

package/dist/centroids/loader.js

@@ -0,0 +1,236 @@
+/**
+ * Centroid loader -- handles lazy initialization and model compatibility.
+ *
+ * Loading priority:
+ *   1. In-memory cache (already loaded)
+ *   2. User's cache directory (previously generated for their model)
+ *   3. Bundled static file (ships with package for default model -- zero delay)
+ *   4. Generate from training queries (only if using a non-default model)
+ */
+import { existsSync, readFileSync } from 'node:fs';
+import { dirname, join } from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { CentroidGenerator, TRAINING_QUERIES_PATH } from './generator.js';
+const currentDir = dirname(fileURLToPath(import.meta.url));
+/** Path to bundled centroids (ships with the package). */
+const BUNDLED_CENTROIDS_DIR = join(currentDir, 'data');
+/** Get path to the bundled centroid file for a given model. */
+function bundledCentroidPath(modelName) {
+    const safeName = modelName.replace(/\//g, '__');
+    return join(BUNDLED_CENTROIDS_DIR, `centroids_${safeName}.json`);
+}
+/**
+ * Stable fingerprint of the benchmark set present in a centroid file.
+ *
+ * Defined as the sorted benchmark names joined by "|". Used to detect when a
+ * cached/bundled centroid file was generated against a different benchmark set
+ * and must be regenerated. Must stay identical to the Python SDK's
+ * `benchmark_fingerprint` (centroids/generator.py).
+ */
+export function benchmarkFingerprint(benchmarkNames) {
+    return [...benchmarkNames].sort().join('|');
+}
+/**
+ * Fingerprint of the bundled default benchmark set, derived from the shipped
+ * training queries. A centroid file whose benchmark set doesn't match this was
+ * built against a different benchmark set and must be regenerated. Mirrors the
+ * Python SDK's `CentroidGenerator.default_benchmark_fingerprint()`.
+ */
+function defaultBenchmarkFingerprint() {
+    const raw = readFileSync(TRAINING_QUERIES_PATH, 'utf-8');
+    const data = JSON.parse(raw);
+    return benchmarkFingerprint(Object.keys(data.benchmarks));
+}
+/**
+ * Whether the provider has reported a real embedding dimension yet.
+ *
+ * The default `LocalEmbeddingProvider` returns a hardcoded fallback (384)
+ * before its model is initialized, so comparing against it would wrongly
+ * reject/accept centroid files for non-default-dimension models. We detect the
+ * uninitialized state via its private `_dimension` marker; providers that
+ * always report a real dimension are treated as known.
+ */
+function providerDimensionKnown(provider) {
+    const dim = provider._dimension;
+    // Providers without the marker (custom providers) report a real dimension.
+    return dim === undefined ? true : dim !== null;
+}
+/**
+ * Manages centroid lifecycle: load, validate, regenerate.
+ *
+ * For the default embedding model (all-MiniLM-L6-v2), centroids are
+ * bundled with the package -- zero first-run delay. For other models,
+ * centroids are generated on first use and cached to disk.
+ */
+export class CentroidLoader {
+    constructor(embeddingProvider, userCachePath) {
+        this._centroids = null;
+        this._provider = embeddingProvider;
+        this._generator = new CentroidGenerator(embeddingProvider);
+        this._userCachePath = userCachePath ?? null;
+    }
+    /**
+     * Get centroids, loading from best available source.
+     *
+     * Priority: memory > user cache > bundled static > generate fresh.
+     *
+     * Synchronous path -- if regeneration is needed (non-default model,
+     * first run) and the provider is async-only, this will throw. Async
+     * callers should use `getCentroidsAsync()` instead.
+     */
+    getCentroids() {
+        const cached = this._tryLoadFromFiles();
+        if (cached !== null)
+            return cached;
+        return this._regenerate();
+    }
+    /**
+     * Async version of `getCentroids`. Works with any embedding provider --
+     * if centroid regeneration is needed, routes through the provider's
+     * async path.
+     */
+    async getCentroidsAsync() {
+        const cached = this._tryLoadFromFiles();
+        if (cached !== null)
+            return cached;
+        return this._regenerateAsync();
+    }
+    /**
+     * Try memory / user cache / bundled centroid file. Returns null if none
+     * of these sources have a valid file for the current provider's model.
+     * Shared between the sync and async load paths.
+     */
+    _tryLoadFromFiles() {
+        if (this._centroids !== null)
+            return this._centroids;
+        // 1. Try user's cached centroids
+        if (this._userCachePath) {
+            const loaded = this._tryLoad(this._userCachePath);
+            if (loaded !== null) {
+                this._centroids = loaded;
+                return this._centroids;
+            }
+        }
+        // 2. Try bundled static centroids (ships with package)
+        const bundledPath = bundledCentroidPath(this._provider.modelName);
+        const loaded = this._tryLoad(bundledPath);
+        if (loaded !== null) {
+            this._centroids = loaded;
+            return this._centroids;
+        }
+        return null;
+    }
+    _tryLoad(path) {
+        if (!existsSync(path))
+            return null;
+        try {
+            const { centroids, metadata } = CentroidGenerator.load(path);
+            const savedModel = metadata.model ?? '';
+            const savedDim = metadata.dimension ?? 0;
+            // Model name must always match.
+            if (savedModel !== this._provider.modelName) {
+                return null;
+            }
+            // Dimension check: skip while the provider hasn't reported a real
+            // dimension (it may still be returning a hardcoded fallback). Otherwise
+            // a non-default-dimension file would be wrongly rejected/accepted.
+            if (providerDimensionKnown(this._provider) && savedDim !== this._provider.dimension) {
+                return null;
+            }
+            // Benchmark-set check: fingerprint the benchmarks actually present in the
+            // file and compare against the expected default set. Computing from the
+            // file's own keys (rather than a stored metadata value that older/bundled
+            // files lack) lets pre-fingerprint bundled files load while still
+            // regenerating a file built against a different benchmark set.
+            if (benchmarkFingerprint(Object.keys(centroids)) !== defaultBenchmarkFingerprint()) {
+                return null;
+            }
+            return centroids;
+        }
+        catch {
+            return null;
+        }
+    }
+    _regenerate() {
+        const centroids = this._generator.generate();
+        // Save to user cache for future runs
+        if (this._userCachePath) {
+            this._generator.save(centroids, this._userCachePath);
+        }
+        this._centroids = centroids;
+        return centroids;
+    }
+    async _regenerateAsync() {
+        const centroids = await this._generator.generateAsync();
+        if (this._userCachePath) {
+            this._generator.save(centroids, this._userCachePath);
+        }
+        this._centroids = centroids;
+        return centroids;
+    }
+    /**
+     * Force regeneration of centroids.
+     *
+     * @param customQueries - Optional custom training queries. If undefined, uses defaults.
+     */
+    regenerate(customQueries) {
+        const centroids = this._generator.generate(customQueries);
+        if (this._userCachePath) {
+            this._generator.save(centroids, this._userCachePath);
+        }
+        this._centroids = centroids;
+        return centroids;
+    }
+    /**
+     * Add a custom benchmark centroid to the existing set (sync).
+     *
+     * Requires a sync-capable embedding provider. Callers with async-only
+     * providers (e.g. LocalEmbeddingProvider) should use
+     * `addBenchmarkCentroidAsync()`.
+     *
+     * @param benchmarkName - Name of the new benchmark.
+     * @param queries - Representative queries for this benchmark.
+     * @returns The generated centroid vector.
+     */
+    addBenchmarkCentroid(benchmarkName, queries) {
+        const centroids = this.getCentroids();
+        const newCentroid = this._generator.generateFromCustom(benchmarkName, queries);
+        centroids[benchmarkName] = newCentroid;
+        // Save updated centroids to user cache
+        if (this._userCachePath) {
+            this._generator.save(centroids, this._userCachePath);
+        }
+        return newCentroid;
+    }
+    /**
+     * Async version of `addBenchmarkCentroid`. Works with any provider.
+     * Mutates the in-memory centroid map so subsequent `getCentroids()` calls
+     * (including from classifiers sharing this loader) see the new benchmark.
+     */
+    async addBenchmarkCentroidAsync(benchmarkName, queries) {
+        const centroids = await this.getCentroidsAsync();
+        const newCentroid = await this._generator.generateFromCustomAsync(benchmarkName, queries);
+        centroids[benchmarkName] = newCentroid;
+        if (this._userCachePath) {
+            this._generator.save(centroids, this._userCachePath);
+        }
+        return newCentroid;
+    }
+    /** Remove a benchmark centroid. Returns true if it existed. */
+    removeBenchmark(benchmarkName) {
+        const centroids = this.getCentroids();
+        if (benchmarkName in centroids) {
+            delete centroids[benchmarkName];
+            if (this._userCachePath) {
+                this._generator.save(centroids, this._userCachePath);
+            }
+            return true;
+        }
+        return false;
+    }
+    /** List all available benchmark names. */
+    get availableBenchmarks() {
+        return Object.keys(this.getCentroids());
+    }
+}
+//# sourceMappingURL=loader.js.map

package/dist/centroids/loader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loader.js","sourceRoot":"","sources":["../../src/centroids/loader.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAGzC,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AAE1E,MAAM,UAAU,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAE3D,0DAA0D;AAC1D,MAAM,qBAAqB,GAAG,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;AAEvD,+DAA+D;AAC/D,SAAS,mBAAmB,CAAC,SAAiB;IAC5C,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAChD,OAAO,IAAI,CAAC,qBAAqB,EAAE,aAAa,QAAQ,OAAO,CAAC,CAAC;AACnE,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,oBAAoB,CAAC,cAAgC;IACnE,OAAO,CAAC,GAAG,cAAc,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC9C,CAAC;AAED;;;;;GAKG;AACH,SAAS,2BAA2B;IAClC,MAAM,GAAG,GAAG,YAAY,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;IACzD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4C,CAAC;IACxE,OAAO,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,sBAAsB,CAAC,QAA+B;IAC7D,MAAM,GAAG,GAAI,QAAsD,CAAC,UAAU,CAAC;IAC/E,2EAA2E;IAC3E,OAAO,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,CAAC;AACjD,CAAC;AAED;;;;;;GAMG;AACH,MAAM,OAAO,cAAc;IAMzB,YACE,iBAAwC,EACxC,aAAsB;QANhB,eAAU,GAAoC,IAAI,CAAC;QAQzD,IAAI,CAAC,SAAS,GAAG,iBAAiB,CAAC;QACnC,IAAI,CAAC,UAAU,GAAG,IAAI,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAC3D,IAAI,CAAC,cAAc,GAAG,aAAa,IAAI,IAAI,CAAC;IAC9C,CAAC;IAED;;;;;;;;OAQG;IACH,YAAY;QACV,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACxC,IAAI,MAAM,KAAK,IAAI;YAAE,OAAO,MAAM,CAAC;QACnC,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,iBAAiB;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACxC,IAAI,MAAM,KAAK,IAAI;YAAE,OAAO,MAAM,CAAC;QACnC,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACK,iBAAiB;QACvB,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC,UAAU,CAAC;QAErD,iCAAiC;QACjC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClD,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;gBACpB,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC;gBACzB,OAAO,IAAI,CAAC,UAAU,CAAC;YACzB,CAAC;QACH,CAAC;QAED,uDAAuD;QACvD,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAClE,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAC1C,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;YACpB,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC;YACzB,OAAO,IAAI,CAAC,UAAU,CAAC;QACzB,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,QAAQ,CAAC,IAAY;QAC3B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAEnC,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE7D,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC;YAEzC,gCAAgC;YAChC,IAAI,UAAU,KAAK,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;gBAC5C,OAAO,IAAI,CAAC;YACd,CAAC;YAED,kEAAkE;YAClE,wEAAwE;YACxE,mEAAmE;YACnE,IAAI,sBAAsB,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,QAAQ,KAAK,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;gBACpF,OAAO,IAAI,CAAC;YACd,CAAC;YAED,0EAA0E;YAC1E,wEAAwE;YACxE,0EAA0E;YAC1E,kEAAkE;YAClE,+DAA+D;YAC/D,IAAI,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,2BAA2B,EAAE,EAAE,CAAC;gBACnF,OAAO,IAAI,CAAC;YACd,CAAC;YAED,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,WAAW;QACjB,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAE7C,qCAAqC;QACrC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;QAExD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;OAIG;IACH,UAAU,CAAC,aAAwC;QACjD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAE1D,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;;;;;;;OAUG;IACH,oBAAoB,CAAC,aAAqB,EAAE,OAAiB;QAC3D,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACtC,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC/E,SAAS,CAAC,aAAa,CAAC,GAAG,WAAW,CAAC;QAEvC,uCAAuC;QACvC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,yBAAyB,CAAC,aAAqB,EAAE,OAAiB;QACtE,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACjD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,uBAAuB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC1F,SAAS,CAAC,aAAa,CAAC,GAAG,WAAW,CAAC;QAEvC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,+DAA+D;IAC/D,eAAe,CAAC,aAAqB;QACnC,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACtC,IAAI,aAAa,IAAI,SAAS,EAAE,CAAC;YAC/B,OAAO,SAAS,CAAC,aAAa,CAAC,CAAC;YAChC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACxB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;YACvD,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,0CAA0C;IAC1C,IAAI,mBAAmB;QACrB,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IAC1C,CAAC;CACF"}

package/dist/classifiers/base.d.ts

@@ -0,0 +1,56 @@
+/**
+ * Abstract base classifier.
+ *
+ * The embedding classifier implements this interface. The abstraction is
+ * kept so custom classifiers can be plugged in for tests or research.
+ */
+/** Output of any classifier. */
+export interface ClassificationResult {
+    /** Cosine similarity to each benchmark centroid (0-1). */
+    benchmarkScores: Record<string, number>;
+    /** Broad category (for display / debugging). */
+    broadCategory: string;
+    /** Subcategory refinement. */
+    subcategory: string;
+    /** Confidence of the classification (0-1). */
+    confidence: number;
+    /** Which classifier produced this result (always "embedding" in the current system). */
+    classifierUsed: string;
+    /** Whether this result came from cache. */
+    cacheHit: boolean;
+    /** How long classification took in milliseconds. */
+    processingTimeMs: number;
+}
+/** Create a default (empty) ClassificationResult. */
+export declare function emptyClassificationResult(): ClassificationResult;
+/** Get top benchmarks sorted by similarity score (descending). */
+export declare function topBenchmarks(result: ClassificationResult): Array<[string, number]>;
+/**
+ * Abstract base class for prompt classifiers.
+ *
+ * A classifier takes a user prompt and returns benchmark similarity scores.
+ * These scores tell us "what kind of task is this?" in terms of which
+ * AI benchmarks it most resembles.
+ */
+export declare abstract class BaseClassifier {
+    /**
+     * Classify a prompt synchronously.
+     *
+     * Implementations whose embedding backend is async-only should throw
+     * a clear error here -- callers should use `classifyAsync` instead.
+     *
+     * @param prompt - The user's input text.
+     * @returns ClassificationResult with benchmarkScores populated.
+     */
+    abstract classify(prompt: string): ClassificationResult;
+    /**
+     * Classify a prompt asynchronously.
+     *
+     * Default implementation wraps `classify()` so sync classifiers work
+     * out of the box on the async path. Async-only classifiers must override.
+     */
+    classifyAsync(prompt: string): Promise<ClassificationResult>;
+    /** Check if the classifier is initialized and ready to use. */
+    abstract isReady(): boolean;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/classifiers/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/classifiers/base.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,gCAAgC;AAChC,MAAM,WAAW,oBAAoB;IACnC,0DAA0D;IAC1D,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAExC,gDAAgD;IAChD,aAAa,EAAE,MAAM,CAAC;IAEtB,8BAA8B;IAC9B,WAAW,EAAE,MAAM,CAAC;IAEpB,8CAA8C;IAC9C,UAAU,EAAE,MAAM,CAAC;IAEnB,wFAAwF;IACxF,cAAc,EAAE,MAAM,CAAC;IAEvB,2CAA2C;IAC3C,QAAQ,EAAE,OAAO,CAAC;IAElB,oDAAoD;IACpD,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAED,qDAAqD;AACrD,wBAAgB,yBAAyB,IAAI,oBAAoB,CAUhE;AAED,kEAAkE;AAClE,wBAAgB,aAAa,CAAC,MAAM,EAAE,oBAAoB,GAAG,KAAK,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAGnF;AAED;;;;;;GAMG;AACH,8BAAsB,cAAc;IAClC;;;;;;;;OAQG;IACH,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,oBAAoB;IAEvD;;;;;OAKG;IACG,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAIlE,+DAA+D;IAC/D,QAAQ,CAAC,OAAO,IAAI,OAAO;CAC5B"}

package/dist/classifiers/base.js

@@ -0,0 +1,42 @@
+/**
+ * Abstract base classifier.
+ *
+ * The embedding classifier implements this interface. The abstraction is
+ * kept so custom classifiers can be plugged in for tests or research.
+ */
+/** Create a default (empty) ClassificationResult. */
+export function emptyClassificationResult() {
+    return {
+        benchmarkScores: {},
+        broadCategory: '',
+        subcategory: '',
+        confidence: 0,
+        classifierUsed: '',
+        cacheHit: false,
+        processingTimeMs: 0,
+    };
+}
+/** Get top benchmarks sorted by similarity score (descending). */
+export function topBenchmarks(result) {
+    return Object.entries(result.benchmarkScores)
+        .sort((a, b) => b[1] - a[1]);
+}
+/**
+ * Abstract base class for prompt classifiers.
+ *
+ * A classifier takes a user prompt and returns benchmark similarity scores.
+ * These scores tell us "what kind of task is this?" in terms of which
+ * AI benchmarks it most resembles.
+ */
+export class BaseClassifier {
+    /**
+     * Classify a prompt asynchronously.
+     *
+     * Default implementation wraps `classify()` so sync classifiers work
+     * out of the box on the async path. Async-only classifiers must override.
+     */
+    async classifyAsync(prompt) {
+        return this.classify(prompt);
+    }
+}
+//# sourceMappingURL=base.js.map

package/dist/classifiers/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/classifiers/base.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AA0BH,qDAAqD;AACrD,MAAM,UAAU,yBAAyB;IACvC,OAAO;QACL,eAAe,EAAE,EAAE;QACnB,aAAa,EAAE,EAAE;QACjB,WAAW,EAAE,EAAE;QACf,UAAU,EAAE,CAAC;QACb,cAAc,EAAE,EAAE;QAClB,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,CAAC;KACpB,CAAC;AACJ,CAAC;AAED,kEAAkE;AAClE,MAAM,UAAU,aAAa,CAAC,MAA4B;IACxD,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC;SAC1C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACjC,CAAC;AAED;;;;;;GAMG;AACH,MAAM,OAAgB,cAAc;IAYlC;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CAAC,MAAc;QAChC,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;CAIF"}

package/dist/classifiers/embedding.d.ts

@@ -0,0 +1,68 @@
+/**
+ * Neural embedding classifier.
+ *
+ * Classifies prompts by computing cosine similarity between the prompt's
+ * embedding vector and pre-computed benchmark centroids. This gives a
+ * semantic understanding of "what kind of task" a prompt represents.
+ *
+ * Exposes both sync (`classify`) and async (`classifyAsync`) entry points.
+ * The sync path requires an embedding provider whose `supportsSync` is true;
+ * the async path works with any provider.
+ */
+import { BaseClassifier, ClassificationResult } from './base.js';
+import { BaseEmbeddingProvider } from '../embeddings/base.js';
+import { CentroidLoader } from '../centroids/loader.js';
+/** Benchmark -> broad category mapping for display purposes. */
+export declare const BENCHMARK_CATEGORIES: Record<string, [string, string]>;
+/**
+ * Semantic classifier using embedding cosine similarity.
+ *
+ * Flow:
+ *   1. Embed the user prompt using the configured embedding provider
+ *   2. Compute cosine similarity against each benchmark centroid
+ *   3. Return similarity scores as the classification result
+ *
+ * Includes LRU caching for both embeddings and full classification results.
+ */
+export declare class EmbeddingClassifier extends BaseClassifier {
+    private _provider;
+    private _centroidLoader;
+    private _embeddingCache;
+    private _classificationCache;
+    constructor(embeddingProvider: BaseEmbeddingProvider, centroidLoader: CentroidLoader, opts?: {
+        embeddingCacheSize?: number;
+        classificationCacheSize?: number;
+        ttlSeconds?: number;
+    });
+    /**
+     * Synchronous classification. Requires the underlying provider to support
+     * `embed()` (`supportsSync === true`); otherwise the provider will throw.
+     */
+    classify(prompt: string): ClassificationResult;
+    /**
+     * Asynchronous classification. Works with any embedding provider; sync
+     * providers route through their default async fallback in BaseEmbeddingProvider.
+     */
+    classifyAsync(prompt: string): Promise<ClassificationResult>;
+    /** Try to return a cached classification result, stamped with fresh timing. */
+    private _readCache;
+    /** Score an embedding against the given centroids, cache the result, and return it. */
+    private _scoreAndCache;
+    private _getEmbeddingSync;
+    private _getEmbeddingAsync;
+    isReady(): boolean;
+    /**
+     * Embedding cache key. Includes the embedding model name and dimension so
+     * vectors from different models/dimensions never collide for the same prompt.
+     */
+    private _embeddingCacheKey;
+    /**
+     * Classification cache key. Includes the embedding model name, dimension, and
+     * benchmark-set fingerprint (not just md5(prompt)) so cached classifications
+     * are invalidated when the model, dimension, or benchmark set changes. Must
+     * stay identical to the Python SDK's classification cache key.
+     */
+    private _classificationCacheKey;
+    private static _hash;
+}
+//# sourceMappingURL=embedding.d.ts.map

package/dist/classifiers/embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../src/classifiers/embedding.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAKH,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AAEjE,OAAO,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAwB,MAAM,wBAAwB,CAAC;AAE9E,gEAAgE;AAChE,eAAO,MAAM,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAajE,CAAC;AAEF;;;;;;;;;GASG;AACH,qBAAa,mBAAoB,SAAQ,cAAc;IACrD,OAAO,CAAC,SAAS,CAAwB;IACzC,OAAO,CAAC,eAAe,CAAiB;IACxC,OAAO,CAAC,eAAe,CAAqB;IAC5C,OAAO,CAAC,oBAAoB,CAAiC;gBAG3D,iBAAiB,EAAE,qBAAqB,EACxC,cAAc,EAAE,cAAc,EAC9B,IAAI,CAAC,EAAE;QACL,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,uBAAuB,CAAC,EAAE,MAAM,CAAC;QACjC,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB;IAgBH;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,oBAAoB;IAgB9C;;;OAGG;IACG,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAelE,+EAA+E;IAC/E,OAAO,CAAC,UAAU;IAUlB,uFAAuF;IACvF,OAAO,CAAC,cAAc;IAwDtB,OAAO,CAAC,iBAAiB;YAUX,kBAAkB;IAUhC,OAAO,IAAI,OAAO;IAIlB;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAM1B;;;;;OAKG;IACH,OAAO,CAAC,uBAAuB;IAO/B,OAAO,CAAC,MAAM,CAAC,KAAK;CAGrB"}