kiokuko-dsh 0.1.0

package/dist/embedding/model-installation.d.ts

@@ -0,0 +1,17 @@
+import { type PathEnvironment } from '../config/paths.js';
+import { type ModelDownloader, type ModelDownloadProgress } from './model-download.js';
+import type { LocalEmbeddingPreset } from './presets/manifest.js';
+export interface InstalledModel {
+    readonly installation: 'installed' | 'reused';
+    readonly directory: string;
+    readonly relativePath: string;
+    readonly totalBytes: number;
+    readonly manifestHash: string;
+}
+export interface InstallModelOptions extends PathEnvironment {
+    readonly downloader?: ModelDownloader;
+    readonly signal?: AbortSignal;
+    readonly onProgress?: (progress: ModelDownloadProgress) => void;
+}
+export declare function installEmbeddingModel(preset: LocalEmbeddingPreset, options?: InstallModelOptions): Promise<InstalledModel>;
+//# sourceMappingURL=model-installation.d.ts.map

package/dist/embedding/model-installation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-installation.d.ts","sourceRoot":"","sources":["../../src/embedding/model-installation.ts"],"names":[],"mappings":"AAIA,OAAO,EAA+B,KAAK,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAEvF,OAAO,EAAoC,KAAK,eAAe,EAAE,KAAK,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AACzH,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAElE,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,YAAY,EAAE,WAAW,GAAG,QAAQ,CAAC;IAC9C,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,mBAAoB,SAAQ,eAAe;IAC1D,QAAQ,CAAC,UAAU,CAAC,EAAE,eAAe,CAAC;IACtC,QAAQ,CAAC,MAAM,CAAC,EAAE,WAAW,CAAC;IAC9B,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,qBAAqB,KAAK,IAAI,CAAC;CACjE;AAkBD,wBAAsB,qBAAqB,CACzC,MAAM,EAAE,oBAAoB,EAC5B,OAAO,GAAE,mBAAwB,GAChC,OAAO,CAAC,cAAc,CAAC,CA0CzB"}

package/dist/embedding/model-installation.js

@@ -0,0 +1,71 @@
+import { randomUUID } from 'node:crypto';
+import { chmod, lstat, mkdir, rename, rm, writeFile } from 'node:fs/promises';
+import path from 'node:path';
+import { KiokukoError } from '../errors.js';
+import { getEmbeddingPresetDirectory } from '../config/paths.js';
+import { assertNoUnexpectedModelFiles, MODEL_MANIFEST_FILENAME, serializeModelManifest, verifyModelDirectory } from './model-manifest.js';
+import { createHuggingFaceModelDownloader } from './model-download.js';
+function relativeModelPath(preset) {
+    return path.posix.join('models', 'embeddings', preset.id, preset.revision);
+}
+async function existingDirectory(pathname) {
+    try {
+        const stat = await lstat(pathname);
+        if (!stat.isDirectory() || stat.isSymbolicLink())
+            throw new KiokukoError('SECURITY_REJECTION', 'Existing model installation is not a private directory');
+        return true;
+    }
+    catch (error) {
+        if (error instanceof KiokukoError)
+            throw error;
+        if (error instanceof Error && 'code' in error && error.code === 'ENOENT')
+            return false;
+        throw error;
+    }
+}
+export async function installEmbeddingModel(preset, options = {}) {
+    const downloader = options.downloader ?? createHuggingFaceModelDownloader();
+    const finalDirectory = getEmbeddingPresetDirectory(preset.id, preset.revision, options);
+    if (await existingDirectory(finalDirectory)) {
+        const manifest = await verifyModelDirectory(finalDirectory, preset);
+        await assertNoUnexpectedModelFiles(finalDirectory, preset);
+        return { installation: 'reused', directory: finalDirectory, relativePath: relativeModelPath(preset), totalBytes: manifest.totalBytes, manifestHash: manifest.artifactManifestHash };
+    }
+    const stagingParent = path.dirname(path.dirname(finalDirectory));
+    const stagingDirectory = path.join(stagingParent, '.staging', randomUUID());
+    await mkdir(stagingDirectory, { recursive: true, mode: 0o700 });
+    try {
+        const downloaded = await downloader.download(preset, stagingDirectory, {
+            ...(options.signal === undefined ? {} : { signal: options.signal }),
+            ...(options.onProgress === undefined ? {} : { onProgress: options.onProgress }),
+        });
+        const manifest = await verifyModelDirectory(downloaded.directory, preset);
+        await assertNoUnexpectedModelFiles(downloaded.directory, preset);
+        await writeFile(path.join(downloaded.directory, MODEL_MANIFEST_FILENAME), serializeModelManifest(manifest), { mode: 0o600, flag: 'wx' });
+        await chmod(downloaded.directory, 0o700);
+        await mkdir(path.dirname(finalDirectory), { recursive: true, mode: 0o700 });
+        if (await existingDirectory(finalDirectory)) {
+            const existing = await verifyModelDirectory(finalDirectory, preset);
+            if (existing.artifactManifestHash !== manifest.artifactManifestHash)
+                throw new KiokukoError('CONFLICT', 'A different model installation already exists for this preset revision');
+            return { installation: 'reused', directory: finalDirectory, relativePath: relativeModelPath(preset), totalBytes: existing.totalBytes, manifestHash: existing.artifactManifestHash };
+        }
+        try {
+            await rename(downloaded.directory, finalDirectory);
+            return { installation: 'installed', directory: finalDirectory, relativePath: relativeModelPath(preset), totalBytes: manifest.totalBytes, manifestHash: manifest.artifactManifestHash };
+        }
+        catch (error) {
+            if (error instanceof Error && 'code' in error && error.code === 'ENOTEMPTY'
+                && await existingDirectory(finalDirectory)) {
+                const existing = await verifyModelDirectory(finalDirectory, preset);
+                return { installation: 'reused', directory: finalDirectory, relativePath: relativeModelPath(preset), totalBytes: existing.totalBytes, manifestHash: existing.artifactManifestHash };
+            }
+            throw error;
+        }
+    }
+    catch (error) {
+        await rm(stagingDirectory, { recursive: true, force: true }).catch(() => undefined);
+        throw error;
+    }
+}
+//# sourceMappingURL=model-installation.js.map

package/dist/embedding/model-installation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-installation.js","sourceRoot":"","sources":["../../src/embedding/model-installation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC9E,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,2BAA2B,EAAwB,MAAM,oBAAoB,CAAC;AACvF,OAAO,EAAE,4BAA4B,EAAE,uBAAuB,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC1I,OAAO,EAAE,gCAAgC,EAAoD,MAAM,qBAAqB,CAAC;AAiBzH,SAAS,iBAAiB,CAAC,MAA4B;IACrD,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;AAC7E,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,QAAgB;IAC/C,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,IAAI,CAAC,cAAc,EAAE;YAAE,MAAM,IAAI,YAAY,CAAC,oBAAoB,EAAE,wDAAwD,CAAC,CAAC;QACzJ,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,YAAY;YAAE,MAAM,KAAK,CAAC;QAC/C,IAAI,KAAK,YAAY,KAAK,IAAI,MAAM,IAAI,KAAK,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAClH,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,MAA4B,EAC5B,UAA+B,EAAE;IAEjC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,gCAAgC,EAAE,CAAC;IAC5E,MAAM,cAAc,GAAG,2BAA2B,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACxF,IAAI,MAAM,iBAAiB,CAAC,cAAc,CAAC,EAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QACpE,MAAM,4BAA4B,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC3D,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,SAAS,EAAE,cAAc,EAAE,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,EAAE,YAAY,EAAE,QAAQ,CAAC,oBAAoB,EAAE,CAAC;IACtL,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC;IACjE,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC;IAC5E,MAAM,KAAK,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAChE,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,MAAM,EAAE,gBAAgB,EAAE;YACrE,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;YACnE,GAAG,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC;SAChF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC1E,MAAM,4BAA4B,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACjE,MAAM,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,uBAAuB,CAAC,EAAE,sBAAsB,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACzI,MAAM,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACzC,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC5E,IAAI,MAAM,iBAAiB,CAAC,cAAc,CAAC,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;YACpE,IAAI,QAAQ,CAAC,oBAAoB,KAAK,QAAQ,CAAC,oBAAoB;gBAAE,MAAM,IAAI,YAAY,CAAC,UAAU,EAAE,wEAAwE,CAAC,CAAC;YAClL,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,SAAS,EAAE,cAAc,EAAE,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,EAAE,YAAY,EAAE,QAAQ,CAAC,oBAAoB,EAAE,CAAC;QACtL,CAAC;QACD,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,UAAU,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YACnD,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,SAAS,EAAE,cAAc,EAAE,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,EAAE,YAAY,EAAE,QAAQ,CAAC,oBAAoB,EAAE,CAAC;QACzL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,MAAM,IAAI,KAAK,IAAK,KAA+B,CAAC,IAAI,KAAK,WAAW;mBACjG,MAAM,iBAAiB,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC7C,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;gBACpE,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,SAAS,EAAE,cAAc,EAAE,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,EAAE,YAAY,EAAE,QAAQ,CAAC,oBAAoB,EAAE,CAAC;YACtL,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,EAAE,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACpF,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/embedding/model-manifest.d.ts

@@ -0,0 +1,18 @@
+import type { LocalEmbeddingPreset, LocalEmbeddingPresetFile } from './presets/manifest.js';
+export declare const MODEL_MANIFEST_FILENAME = "kiokuko-model-manifest.json";
+export declare const MAX_MODEL_BYTES: number;
+export interface VerifiedModelManifest {
+    readonly schemaVersion: 1;
+    readonly presetId: 'local-small';
+    readonly repositoryId: string;
+    readonly revision: string;
+    readonly artifactManifestHash: string;
+    readonly files: readonly LocalEmbeddingPresetFile[];
+    readonly totalBytes: number;
+}
+export declare function validatePresetManifest(preset: LocalEmbeddingPreset): void;
+export declare function createModelManifest(preset: LocalEmbeddingPreset): VerifiedModelManifest;
+export declare function serializeModelManifest(manifest: VerifiedModelManifest): string;
+export declare function verifyModelDirectory(directory: string, preset: LocalEmbeddingPreset): Promise<VerifiedModelManifest>;
+export declare function assertNoUnexpectedModelFiles(directory: string, preset: LocalEmbeddingPreset): Promise<void>;
+//# sourceMappingURL=model-manifest.d.ts.map

package/dist/embedding/model-manifest.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-manifest.d.ts","sourceRoot":"","sources":["../../src/embedding/model-manifest.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,oBAAoB,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AAG5F,eAAO,MAAM,uBAAuB,gCAAgC,CAAC;AACrE,eAAO,MAAM,eAAe,QAAoB,CAAC;AAEjD,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC;IAC1B,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC;IACjC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC;IACtC,QAAQ,CAAC,KAAK,EAAE,SAAS,wBAAwB,EAAE,CAAC;IACpD,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;CAC7B;AAUD,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,oBAAoB,GAAG,IAAI,CAuBzE;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,oBAAoB,GAAG,qBAAqB,CAYvF;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,qBAAqB,GAAG,MAAM,CAE9E;AAED,wBAAsB,oBAAoB,CACxC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,oBAAoB,GAC3B,OAAO,CAAC,qBAAqB,CAAC,CAUhC;AAED,wBAAsB,4BAA4B,CAChD,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,oBAAoB,GAC3B,OAAO,CAAC,IAAI,CAAC,CAWf"}

package/dist/embedding/model-manifest.js

@@ -0,0 +1,86 @@
+import { createHash } from 'node:crypto';
+import { lstat, readdir, readFile } from 'node:fs/promises';
+import path from 'node:path';
+import { KiokukoError } from '../errors.js';
+import { canonicalJson } from '../serialization/validate.js';
+import { presetManifestHash } from './presets/manifest.js';
+export const MODEL_MANIFEST_FILENAME = 'kiokuko-model-manifest.json';
+export const MAX_MODEL_BYTES = 512 * 1024 * 1024;
+function invalid(message) {
+    throw new KiokukoError('VALIDATION_ERROR', message);
+}
+function hashFile(bytes) {
+    return createHash('sha256').update(bytes).digest('hex');
+}
+export function validatePresetManifest(preset) {
+    if (preset.id !== 'local-small' || preset.schemaVersion !== 1
+        || preset.sourceModel !== 'intfloat/multilingual-e5-small'
+        || preset.artifactRepository !== 'Xenova/multilingual-e5-small'
+        || !/^[0-9a-f]{40}$/u.test(preset.revision)
+        || preset.dimensions !== 384 || preset.maximumTokens !== 512
+        || preset.dtype !== 'q8' || preset.pooling !== 'mean' || preset.normalize !== true
+        || preset.distanceMetric !== 'cosine' || preset.distanceCeiling <= 0 || preset.distanceCeiling >= 2
+        || preset.inputContract !== 'e5-query-passage-v1' || preset.queryPrefix !== 'query: '
+        || preset.documentPrefix !== 'passage: ' || !/^[0-9]+\.[0-9]+\.[0-9]+$/u.test(preset.transformersJsVersion)) {
+        invalid('Embedding preset manifest is invalid');
+    }
+    const paths = new Set();
+    let totalBytes = 0;
+    for (const file of preset.files) {
+        if (paths.has(file.path) || file.path.length === 0 || file.path.startsWith('/')
+            || file.path.split('/').some((part) => part === '' || part === '.' || part === '..')
+            || !Number.isSafeInteger(file.size) || file.size <= 0
+            || !/^[0-9a-f]{64}$/u.test(file.sha256))
+            invalid('Embedding preset file manifest is invalid');
+        paths.add(file.path);
+        totalBytes += file.size;
+    }
+    if (preset.files.length === 0 || totalBytes > MAX_MODEL_BYTES)
+        invalid('Embedding preset file size is unsupported');
+}
+export function createModelManifest(preset) {
+    validatePresetManifest(preset);
+    const totalBytes = preset.files.reduce((total, file) => total + file.size, 0);
+    return Object.freeze({
+        schemaVersion: 1,
+        presetId: preset.id,
+        repositoryId: preset.artifactRepository,
+        revision: preset.revision,
+        artifactManifestHash: presetManifestHash(preset),
+        files: Object.freeze(preset.files.map((file) => Object.freeze({ ...file }))),
+        totalBytes,
+    });
+}
+export function serializeModelManifest(manifest) {
+    return `${canonicalJson(manifest)}\n`;
+}
+export async function verifyModelDirectory(directory, preset) {
+    const expected = createModelManifest(preset);
+    for (const file of expected.files) {
+        const filePath = path.join(directory, file.path);
+        const stat = await lstat(filePath);
+        if (!stat.isFile() || stat.isSymbolicLink())
+            throw new KiokukoError('SECURITY_REJECTION', `Model artifact is not a regular file: ${file.path}`);
+        if (stat.size !== file.size)
+            throw new KiokukoError('INTEGRITY_ERROR', `Model artifact size mismatch: ${file.path}`);
+        if (hashFile(await readFile(filePath)) !== file.sha256)
+            throw new KiokukoError('SECURITY_REJECTION', `Model artifact hash mismatch: ${file.path}`);
+    }
+    return expected;
+}
+export async function assertNoUnexpectedModelFiles(directory, preset) {
+    const allowed = new Set([...preset.files.map((file) => file.path), MODEL_MANIFEST_FILENAME]);
+    const walk = async (current, prefix) => {
+        for (const entry of await readdir(current, { withFileTypes: true })) {
+            const relative = prefix.length === 0 ? entry.name : `${prefix}/${entry.name}`;
+            if (entry.isSymbolicLink())
+                throw new KiokukoError('SECURITY_REJECTION', `Model directory contains a symbolic link: ${relative}`);
+            if (entry.isDirectory())
+                await walk(path.join(current, entry.name), relative);
+            else if (!allowed.has(relative))
+                throw new KiokukoError('SECURITY_REJECTION', `Model directory contains an unexpected file: ${relative}`);
+        }
+    };
+    await walk(directory, '');
+}
+//# sourceMappingURL=model-manifest.js.map

package/dist/embedding/model-manifest.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-manifest.js","sourceRoot":"","sources":["../../src/embedding/model-manifest.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5D,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAE7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE3D,MAAM,CAAC,MAAM,uBAAuB,GAAG,6BAA6B,CAAC;AACrE,MAAM,CAAC,MAAM,eAAe,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;AAYjD,SAAS,OAAO,CAAC,OAAe;IAC9B,MAAM,IAAI,YAAY,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAiB;IACjC,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAA4B;IACjE,IAAI,MAAM,CAAC,EAAE,KAAK,aAAa,IAAI,MAAM,CAAC,aAAa,KAAK,CAAC;WACxD,MAAM,CAAC,WAAW,KAAK,gCAAgC;WACvD,MAAM,CAAC,kBAAkB,KAAK,8BAA8B;WAC5D,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;WACxC,MAAM,CAAC,UAAU,KAAK,GAAG,IAAI,MAAM,CAAC,aAAa,KAAK,GAAG;WACzD,MAAM,CAAC,KAAK,KAAK,IAAI,IAAI,MAAM,CAAC,OAAO,KAAK,MAAM,IAAI,MAAM,CAAC,SAAS,KAAK,IAAI;WAC/E,MAAM,CAAC,cAAc,KAAK,QAAQ,IAAI,MAAM,CAAC,eAAe,IAAI,CAAC,IAAI,MAAM,CAAC,eAAe,IAAI,CAAC;WAChG,MAAM,CAAC,aAAa,KAAK,qBAAqB,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS;WAClF,MAAM,CAAC,cAAc,KAAK,WAAW,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,EAAE,CAAC;QAC9G,OAAO,CAAC,sCAAsC,CAAC,CAAC;IAClD,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;IAChC,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;eAC1E,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,IAAI,CAAC;eACjF,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC;eAClD,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;YAAE,OAAO,CAAC,2CAA2C,CAAC,CAAC;QAChG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,UAAU,IAAI,IAAI,CAAC,IAAI,CAAC;IAC1B,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,GAAG,eAAe;QAAE,OAAO,CAAC,2CAA2C,CAAC,CAAC;AACtH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAA4B;IAC9D,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC9E,OAAO,MAAM,CAAC,MAAM,CAAC;QACnB,aAAa,EAAE,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC,EAAE;QACnB,YAAY,EAAE,MAAM,CAAC,kBAAkB;QACvC,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,oBAAoB,EAAE,kBAAkB,CAAC,MAAM,CAAC;QAChD,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC;QAC5E,UAAU;KACX,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAA+B;IACpE,OAAO,GAAG,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC;AACxC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,SAAiB,EACjB,MAA4B;IAE5B,MAAM,QAAQ,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC7C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,CAAC,cAAc,EAAE;YAAE,MAAM,IAAI,YAAY,CAAC,oBAAoB,EAAE,yCAAyC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAChJ,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,YAAY,CAAC,iBAAiB,EAAE,iCAAiC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACrH,IAAI,QAAQ,CAAC,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC,KAAK,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,YAAY,CAAC,oBAAoB,EAAE,iCAAiC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACrJ,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,SAAiB,EACjB,MAA4B;IAE5B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,uBAAuB,CAAC,CAAC,CAAC;IAC7F,MAAM,IAAI,GAAG,KAAK,EAAE,OAAe,EAAE,MAAc,EAAiB,EAAE;QACpE,KAAK,MAAM,KAAK,IAAI,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YACpE,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;YAC9E,IAAI,KAAK,CAAC,cAAc,EAAE;gBAAE,MAAM,IAAI,YAAY,CAAC,oBAAoB,EAAE,6CAA6C,QAAQ,EAAE,CAAC,CAAC;YAClI,IAAI,KAAK,CAAC,WAAW,EAAE;gBAAE,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC;iBACzE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;gBAAE,MAAM,IAAI,YAAY,CAAC,oBAAoB,EAAE,gDAAgD,QAAQ,EAAE,CAAC,CAAC;QAC5I,CAAC;IACH,CAAC,CAAC;IACF,MAAM,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;AAC5B,CAAC"}

package/dist/embedding/openai-compatible-provider.d.ts

@@ -0,0 +1,24 @@
+import type { EmbeddingProvider, EmbeddingProfileIdentity, EnabledEmbeddingConfig } from './types.js';
+export declare const MAX_EMBEDDING_PROVIDER_INPUTS = 64;
+export declare const MAX_EMBEDDING_PROVIDER_REQUEST_BYTES: number;
+export declare const MAX_EMBEDDING_PROVIDER_RESPONSE_BYTES: number;
+export declare const DEFAULT_EMBEDDING_PROVIDER_RETRIES = 2;
+export declare const MAX_EMBEDDING_PROVIDER_RETRY_DELAY_MS = 5000;
+export interface OpenAICompatibleEmbeddingProviderOptions {
+    readonly config: EnabledEmbeddingConfig;
+    readonly fetchImpl?: typeof fetch;
+    readonly timeoutMs?: number;
+    readonly maxRetries?: number;
+    readonly retryDelayMs?: number;
+    readonly sleepImpl?: (milliseconds: number, signal?: AbortSignal) => Promise<void>;
+}
+export declare class OpenAICompatibleEmbeddingProvider implements EmbeddingProvider {
+    #private;
+    readonly profile: EmbeddingProfileIdentity;
+    constructor(options: OpenAICompatibleEmbeddingProviderOptions);
+    embed(inputs: readonly string[], options?: {
+        signal?: AbortSignal;
+    }): Promise<readonly Float32Array[]>;
+}
+export { OpenAICompatibleEmbeddingProvider as OpenAICompatibleProvider };
+//# sourceMappingURL=openai-compatible-provider.d.ts.map

package/dist/embedding/openai-compatible-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-compatible-provider.d.ts","sourceRoot":"","sources":["../../src/embedding/openai-compatible-provider.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,iBAAiB,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAEtG,eAAO,MAAM,6BAA6B,KAAK,CAAC;AAChD,eAAO,MAAM,oCAAoC,QAAkB,CAAC;AACpE,eAAO,MAAM,qCAAqC,QAAkB,CAAC;AACrE,eAAO,MAAM,kCAAkC,IAAI,CAAC;AACpD,eAAO,MAAM,qCAAqC,OAAQ,CAAC;AAM3D,MAAM,WAAW,wCAAwC;IACvD,QAAQ,CAAC,MAAM,EAAE,sBAAsB,CAAC;IACxC,QAAQ,CAAC,SAAS,CAAC,EAAE,OAAO,KAAK,CAAC;IAClC,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACpF;AAqND,qBAAa,iCAAkC,YAAW,iBAAiB;;IACzE,QAAQ,CAAC,OAAO,EAAE,wBAAwB,CAAC;gBAS/B,OAAO,EAAE,wCAAwC;IAavD,KAAK,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,EAAE,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAO,GAAG,OAAO,CAAC,SAAS,YAAY,EAAE,CAAC;CAwEjH;AAED,OAAO,EAAE,iCAAiC,IAAI,wBAAwB,EAAE,CAAC"}

package/dist/embedding/openai-compatible-provider.js

@@ -0,0 +1,330 @@
+import { TextDecoder } from 'node:util';
+import { parseStrictJson } from '../setup/strict-json.js';
+import { findSecret } from '../memory/secrets.js';
+import { KiokukoError } from '../errors.js';
+import { normalizeEmbeddingBaseUrl, requireEnabledEmbeddingConfig } from './config.js';
+import { createEmbeddingProfileIdentity } from './profile.js';
+import { normalizeVector } from './vector.js';
+import { EmbeddingProviderError } from './provider.js';
+export const MAX_EMBEDDING_PROVIDER_INPUTS = 64;
+export const MAX_EMBEDDING_PROVIDER_REQUEST_BYTES = 4 * 1024 * 1024;
+export const MAX_EMBEDDING_PROVIDER_RESPONSE_BYTES = 4 * 1024 * 1024;
+export const DEFAULT_EMBEDDING_PROVIDER_RETRIES = 2;
+export const MAX_EMBEDDING_PROVIDER_RETRY_DELAY_MS = 5_000;
+const INVALID_UNICODE = /[\p{Cs}\uFFFD]/u;
+const INVALID_CONTROL = /[\u0000-\u0008\u000b\u000c\u000e-\u001f\u007f]/u;
+const decoder = new TextDecoder('utf-8', { fatal: true, ignoreBOM: true });
+function invalidResponse() {
+    throw new EmbeddingProviderError('invalid_response', false);
+}
+function requestInvalid(message) {
+    throw new KiokukoError('VALIDATION_ERROR', message);
+}
+function isRecord(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function boundedInteger(value, minimum, maximum, label) {
+    if (!Number.isSafeInteger(value) || value < minimum || value > maximum)
+        requestInvalid(`${label} is invalid`);
+    return value;
+}
+function endpointUrl(baseUrl) {
+    const url = new URL(baseUrl);
+    const path = url.pathname.replace(/\/+$/u, '');
+    url.pathname = `${path}/embeddings`;
+    return url;
+}
+function validateInputs(inputs, batchSize) {
+    if (!Array.isArray(inputs) || inputs.length < 1 || inputs.length > Math.min(MAX_EMBEDDING_PROVIDER_INPUTS, batchSize)) {
+        requestInvalid('Embedding provider input count is invalid');
+    }
+    const copied = [...inputs];
+    for (const input of copied) {
+        if (typeof input !== 'string' || input.length === 0 || INVALID_CONTROL.test(input) || INVALID_UNICODE.test(input)) {
+            requestInvalid('Embedding provider input is invalid');
+        }
+        if (findSecret(input) !== undefined)
+            throw new EmbeddingProviderError('secret_blocked', false);
+    }
+    return copied;
+}
+function requestBody(config, inputs) {
+    const body = JSON.stringify({
+        input: [...inputs],
+        model: config.model,
+        encoding_format: 'float',
+    });
+    if (Buffer.byteLength(body, 'utf8') > MAX_EMBEDDING_PROVIDER_REQUEST_BYTES) {
+        requestInvalid('Embedding provider request exceeds the byte limit');
+    }
+    return body;
+}
+function retryAfterMs(response) {
+    const value = response.headers.get('retry-after');
+    if (value === null || !/^\d+$/u.test(value))
+        return null;
+    const seconds = Number(value);
+    if (!Number.isSafeInteger(seconds))
+        return null;
+    return Math.min(MAX_EMBEDDING_PROVIDER_RETRY_DELAY_MS, seconds * 1_000);
+}
+function statusFailure(response) {
+    if (response.status === 429)
+        return new EmbeddingProviderError('rate_limited', true, retryAfterMs(response));
+    if (response.status === 408 || response.status === 425)
+        return new EmbeddingProviderError('timeout', true);
+    if (response.status >= 500 && response.status <= 599)
+        return new EmbeddingProviderError('provider_unavailable', true);
+    return new EmbeddingProviderError('provider_unavailable', false);
+}
+async function cancelResponse(response) {
+    if (response.body === null)
+        return;
+    try {
+        await response.body.cancel();
+    }
+    catch {
+        // The response is already being discarded; the public failure remains bounded.
+    }
+}
+async function readResponseText(response) {
+    const contentType = response.headers.get('content-type');
+    if (contentType === null || !/^application\/(?:[\w.+-]+\+)?json(?:\s*;|$)/iu.test(contentType))
+        invalidResponse();
+    const contentLength = response.headers.get('content-length');
+    if (contentLength !== null && !/^\d+$/u.test(contentLength))
+        invalidResponse();
+    const declared = contentLength === null ? null : Number(contentLength);
+    if (declared !== null && (!Number.isSafeInteger(declared) || declared > MAX_EMBEDDING_PROVIDER_RESPONSE_BYTES))
+        invalidResponse();
+    const reader = response.body?.getReader();
+    if (reader === undefined) {
+        let body;
+        try {
+            body = await response.text();
+        }
+        catch {
+            throw new EmbeddingProviderError('provider_unavailable', true);
+        }
+        if (Buffer.byteLength(body, 'utf8') > MAX_EMBEDDING_PROVIDER_RESPONSE_BYTES || INVALID_UNICODE.test(body))
+            invalidResponse();
+        return body;
+    }
+    const chunks = [];
+    let size = 0;
+    for (;;) {
+        let next;
+        try {
+            next = await reader.read();
+        }
+        catch {
+            throw new EmbeddingProviderError('provider_unavailable', true);
+        }
+        if (next.done)
+            break;
+        if (next.value === undefined)
+            invalidResponse();
+        const chunk = Buffer.from(next.value);
+        size += chunk.byteLength;
+        if (size > MAX_EMBEDDING_PROVIDER_RESPONSE_BYTES) {
+            try {
+                await reader.cancel();
+            }
+            catch {
+                // Preserve the bounded validation failure if cancellation itself fails.
+            }
+            invalidResponse();
+        }
+        chunks.push(chunk);
+    }
+    let body;
+    try {
+        body = decoder.decode(Buffer.concat(chunks));
+    }
+    catch {
+        invalidResponse();
+    }
+    if (INVALID_UNICODE.test(body))
+        invalidResponse();
+    return body;
+}
+function parseProviderResponse(value, config, expectedCount) {
+    if (!isRecord(value))
+        invalidResponse();
+    const allowedRootKeys = new Set(['object', 'data', 'model', 'usage']);
+    if (Object.keys(value).some((key) => !allowedRootKeys.has(key))
+        || !Object.prototype.hasOwnProperty.call(value, 'data')
+        || !Object.prototype.hasOwnProperty.call(value, 'model'))
+        invalidResponse();
+    const response = value;
+    if (response.object !== undefined && response.object !== 'list')
+        invalidResponse();
+    if (response.model !== config.model || !Array.isArray(response.data) || response.data.length !== expectedCount)
+        invalidResponse();
+    const vectors = new Array(expectedCount);
+    const seen = new Set();
+    for (const itemValue of response.data) {
+        if (!isRecord(itemValue))
+            invalidResponse();
+        const allowedItemKeys = new Set(['object', 'embedding', 'index']);
+        if (Object.keys(itemValue).some((key) => !allowedItemKeys.has(key))
+            || !Object.prototype.hasOwnProperty.call(itemValue, 'embedding')
+            || !Object.prototype.hasOwnProperty.call(itemValue, 'index'))
+            invalidResponse();
+        const item = itemValue;
+        if (item.object !== undefined && item.object !== 'embedding')
+            invalidResponse();
+        if (typeof item.index !== 'number' || !Number.isSafeInteger(item.index) || item.index < 0 || item.index >= expectedCount || seen.has(item.index))
+            invalidResponse();
+        if (!Array.isArray(item.embedding) || item.embedding.length !== config.dimensions) {
+            throw new EmbeddingProviderError('dimension_mismatch', false);
+        }
+        for (const valueAtIndex of item.embedding) {
+            if (typeof valueAtIndex !== 'number' || !Number.isFinite(valueAtIndex))
+                invalidResponse();
+        }
+        let vector;
+        try {
+            vector = normalizeVector(item.embedding, config.dimensions);
+        }
+        catch (error) {
+            if (error instanceof KiokukoError && error.code === 'VALIDATION_ERROR') {
+                throw new EmbeddingProviderError('dimension_mismatch', false);
+            }
+            throw error;
+        }
+        seen.add(item.index);
+        vectors[item.index] = vector;
+    }
+    if (seen.size !== expectedCount || vectors.some((vector) => vector === undefined))
+        invalidResponse();
+    return Object.freeze(vectors);
+}
+async function defaultSleep(milliseconds, signal) {
+    if (milliseconds <= 0)
+        return;
+    await new Promise((resolve, reject) => {
+        let timer;
+        const onAbort = () => {
+            clearTimeout(timer);
+            signal?.removeEventListener('abort', onAbort);
+            reject(new EmbeddingProviderError('timeout', false));
+        };
+        const finish = () => {
+            signal?.removeEventListener('abort', onAbort);
+            resolve();
+        };
+        timer = setTimeout(finish, milliseconds);
+        if (signal === undefined)
+            return;
+        if (signal.aborted) {
+            clearTimeout(timer);
+            reject(new EmbeddingProviderError('timeout', false));
+            return;
+        }
+        signal.addEventListener('abort', onAbort, { once: true });
+    });
+}
+function retryDelay(attempt, configuredDelayMs, retryAfter) {
+    const exponential = configuredDelayMs * (2 ** attempt);
+    return Math.min(MAX_EMBEDDING_PROVIDER_RETRY_DELAY_MS, retryAfter ?? exponential);
+}
+export class OpenAICompatibleEmbeddingProvider {
+    profile;
+    #endpoint;
+    #config;
+    #fetchImpl;
+    #timeoutMs;
+    #maxRetries;
+    #retryDelayMs;
+    #sleepImpl;
+    constructor(options) {
+        const config = requireEnabledEmbeddingConfig(options.config);
+        const baseUrl = normalizeEmbeddingBaseUrl(config.baseUrl, config.allowRemote);
+        this.#config = Object.freeze({ ...config, baseUrl });
+        this.profile = createEmbeddingProfileIdentity(this.#config);
+        this.#endpoint = endpointUrl(baseUrl);
+        this.#fetchImpl = options.fetchImpl ?? fetch;
+        this.#timeoutMs = boundedInteger(options.timeoutMs ?? config.timeoutMs, 100, 120_000, 'Embedding provider timeout');
+        this.#maxRetries = boundedInteger(options.maxRetries ?? DEFAULT_EMBEDDING_PROVIDER_RETRIES, 0, 3, 'Embedding provider retries');
+        this.#retryDelayMs = boundedInteger(options.retryDelayMs ?? 100, 0, MAX_EMBEDDING_PROVIDER_RETRY_DELAY_MS, 'Embedding provider retry delay');
+        this.#sleepImpl = options.sleepImpl ?? defaultSleep;
+    }
+    async embed(inputs, options = {}) {
+        const normalizedInputs = validateInputs(inputs, this.#config.batchSize);
+        const body = requestBody(this.#config, normalizedInputs);
+        if (options.signal?.aborted)
+            throw new EmbeddingProviderError('timeout', false);
+        for (let attempt = 0; attempt <= this.#maxRetries; attempt += 1) {
+            let timedOut = false;
+            const controller = new AbortController();
+            const onAbort = () => controller.abort();
+            if (options.signal?.aborted)
+                controller.abort();
+            else
+                options.signal?.addEventListener('abort', onAbort, { once: true });
+            const timer = setTimeout(() => {
+                timedOut = true;
+                controller.abort();
+            }, this.#timeoutMs);
+            try {
+                let response;
+                try {
+                    response = await this.#fetchImpl(this.#endpoint, {
+                        method: 'POST',
+                        redirect: 'manual',
+                        headers: {
+                            accept: 'application/json',
+                            'content-type': 'application/json',
+                            ...(this.#config.apiKey === undefined ? {} : { authorization: `Bearer ${this.#config.apiKey}` }),
+                        },
+                        body,
+                        signal: controller.signal,
+                    });
+                }
+                catch {
+                    throw timedOut || options.signal?.aborted === true
+                        ? new EmbeddingProviderError('timeout', timedOut)
+                        : new EmbeddingProviderError('provider_unavailable', true);
+                }
+                if (response.status !== 200) {
+                    const failure = statusFailure(response);
+                    await cancelResponse(response);
+                    throw failure;
+                }
+                const responseText = await readResponseText(response);
+                let parsed;
+                try {
+                    parsed = parseStrictJson(responseText, { allowTrailingComma: false, disallowComments: true, allowEmptyContent: false }, 'Embedding provider response is invalid');
+                }
+                catch {
+                    invalidResponse();
+                }
+                return parseProviderResponse(parsed, this.#config, normalizedInputs.length);
+            }
+            catch (error) {
+                const classified = error instanceof EmbeddingProviderError
+                    ? error
+                    : error instanceof KiokukoError && error.code === 'VALIDATION_ERROR'
+                        ? new EmbeddingProviderError('invalid_response', false)
+                        : new EmbeddingProviderError('provider_unavailable', true);
+                const failure = timedOut
+                    ? new EmbeddingProviderError('timeout', true)
+                    : options.signal?.aborted === true
+                        ? new EmbeddingProviderError('timeout', false)
+                        : classified;
+                if (!failure.retryable || attempt >= this.#maxRetries || options.signal?.aborted)
+                    throw failure;
+                const delay = retryDelay(attempt, this.#retryDelayMs, failure.retryAfterMs);
+                await this.#sleepImpl(delay, options.signal);
+            }
+            finally {
+                clearTimeout(timer);
+                options.signal?.removeEventListener('abort', onAbort);
+            }
+        }
+        throw new EmbeddingProviderError('provider_unavailable', false);
+    }
+}
+export { OpenAICompatibleEmbeddingProvider as OpenAICompatibleProvider };
+//# sourceMappingURL=openai-compatible-provider.js.map

package/dist/embedding/openai-compatible-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-compatible-provider.js","sourceRoot":"","sources":["../../src/embedding/openai-compatible-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,yBAAyB,EAAE,6BAA6B,EAAE,MAAM,aAAa,CAAC;AACvF,OAAO,EAAE,8BAA8B,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAGvD,MAAM,CAAC,MAAM,6BAA6B,GAAG,EAAE,CAAC;AAChD,MAAM,CAAC,MAAM,oCAAoC,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;AACpE,MAAM,CAAC,MAAM,qCAAqC,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;AACrE,MAAM,CAAC,MAAM,kCAAkC,GAAG,CAAC,CAAC;AACpD,MAAM,CAAC,MAAM,qCAAqC,GAAG,KAAK,CAAC;AAE3D,MAAM,eAAe,GAAG,iBAAiB,CAAC;AAC1C,MAAM,eAAe,GAAG,iDAAiD,CAAC;AAC1E,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;AAuB3E,SAAS,eAAe;IACtB,MAAM,IAAI,sBAAsB,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED,SAAS,cAAc,CAAC,OAAe;IACrC,MAAM,IAAI,YAAY,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,cAAc,CAAC,KAAa,EAAE,OAAe,EAAE,OAAe,EAAE,KAAa;IACpF,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,OAAO,IAAI,KAAK,GAAG,OAAO;QAAE,cAAc,CAAC,GAAG,KAAK,aAAa,CAAC,CAAC;IAC9G,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,WAAW,CAAC,OAAe;IAClC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;IAC7B,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC/C,GAAG,CAAC,QAAQ,GAAG,GAAG,IAAI,aAAa,CAAC;IACpC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,cAAc,CAAC,MAAyB,EAAE,SAAiB;IAClE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,6BAA6B,EAAE,SAAS,CAAC,EAAE,CAAC;QACtH,cAAc,CAAC,2CAA2C,CAAC,CAAC;IAC9D,CAAC;IACD,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;IAC3B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAClH,cAAc,CAAC,qCAAqC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,UAAU,CAAC,KAAK,CAAC,KAAK,SAAS;YAAE,MAAM,IAAI,sBAAsB,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;IACjG,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAAC,MAA8B,EAAE,MAAyB;IAC5E,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC;QAC1B,KAAK,EAAE,CAAC,GAAG,MAAM,CAAC;QAClB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,eAAe,EAAE,OAAO;KACzB,CAAC,CAAC;IACH,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,oCAAoC,EAAE,CAAC;QAC3E,cAAc,CAAC,mDAAmD,CAAC,CAAC;IACtE,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,QAAkB;IACtC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAClD,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACzD,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9B,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC;QAAE,OAAO,IAAI,CAAC;IAChD,OAAO,IAAI,CAAC,GAAG,CAAC,qCAAqC,EAAE,OAAO,GAAG,KAAK,CAAC,CAAC;AAC1E,CAAC;AAED,SAAS,aAAa,CAAC,QAAkB;IACvC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG;QAAE,OAAO,IAAI,sBAAsB,CAAC,cAAc,EAAE,IAAI,EAAE,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC7G,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG;QAAE,OAAO,IAAI,sBAAsB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC3G,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG;QAAE,OAAO,IAAI,sBAAsB,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;IACtH,OAAO,IAAI,sBAAsB,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;AACnE,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,QAAkB;IAC9C,IAAI,QAAQ,CAAC,IAAI,KAAK,IAAI;QAAE,OAAO;IACnC,IAAI,CAAC;QACH,MAAM,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,+EAA+E;IACjF,CAAC;AACH,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,QAAkB;IAChD,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACzD,IAAI,WAAW,KAAK,IAAI,IAAI,CAAC,+CAA+C,CAAC,IAAI,CAAC,WAAW,CAAC;QAAE,eAAe,EAAE,CAAC;IAClH,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC7D,IAAI,aAAa,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC;QAAE,eAAe,EAAE,CAAC;IAC/E,MAAM,QAAQ,GAAG,aAAa,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IACvE,IAAI,QAAQ,KAAK,IAAI,IAAI,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,QAAQ,GAAG,qCAAqC,CAAC;QAAE,eAAe,EAAE,CAAC;IAElI,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;IAC1C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,IAAI,IAAY,CAAC;QACjB,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,sBAAsB,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,qCAAqC,IAAI,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,eAAe,EAAE,CAAC;QAC7H,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,SAAS,CAAC;QACR,IAAI,IAA0C,CAAC;QAC/C,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC7B,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,sBAAsB,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,IAAI,CAAC,IAAI;YAAE,MAAM;QACrB,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS;YAAE,eAAe,EAAE,CAAC;QAChD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,IAAI,KAAK,CAAC,UAAU,CAAC;QACzB,IAAI,IAAI,GAAG,qCAAqC,EAAE,CAAC;YACjD,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;YACxB,CAAC;YAAC,MAAM,CAAC;gBACP,wEAAwE;YAC1E,CAAC;YACD,eAAe,EAAE,CAAC;QACpB,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,IAAI,IAAY,CAAC;IACjB,IAAI,CAAC;QACH,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC/C,CAAC;IAAC,MAAM,CAAC;QACP,eAAe,EAAE,CAAC;IACpB,CAAC;IACD,IAAI,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;QAAE,eAAe,EAAE,CAAC;IAClD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc,EAAE,MAA8B,EAAE,aAAqB;IAClG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,eAAe,EAAE,CAAC;IACxC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IACtE,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;WAC1D,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC;WACpD,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC;QAAE,eAAe,EAAE,CAAC;IAC9E,MAAM,QAAQ,GAAG,KAAoC,CAAC;IACtD,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM;QAAE,eAAe,EAAE,CAAC;IACnF,IAAI,QAAQ,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,MAAM,KAAK,aAAa;QAAE,eAAe,EAAE,CAAC;IAElI,MAAM,OAAO,GAAG,IAAI,KAAK,CAAe,aAAa,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,eAAe,EAAE,CAAC;QAC5C,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;QAClE,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;eAC9D,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC;eAC7D,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC;YAAE,eAAe,EAAE,CAAC;QAClF,MAAM,IAAI,GAAG,SAA4C,CAAC;QAC1D,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,WAAW;YAAE,eAAe,EAAE,CAAC;QAChF,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,IAAI,aAAa,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;YAAE,eAAe,EAAE,CAAC;QACpK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,MAAM,CAAC,UAAU,EAAE,CAAC;YAClF,MAAM,IAAI,sBAAsB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAChE,CAAC;QACD,KAAK,MAAM,YAAY,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1C,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAAE,eAAe,EAAE,CAAC;QAC5F,CAAC;QACD,IAAI,MAAoB,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;gBACvE,MAAM,IAAI,sBAAsB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;YAChE,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC;IAC/B,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,KAAK,SAAS,CAAC;QAAE,eAAe,EAAE,CAAC;IACrG,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAChC,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,YAAoB,EAAE,MAAoB;IACpE,IAAI,YAAY,IAAI,CAAC;QAAE,OAAO;IAC9B,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,IAAI,KAAoC,CAAC;QACzC,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,sBAAsB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC;QACF,MAAM,MAAM,GAAG,GAAG,EAAE;YAClB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9C,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,KAAK,GAAG,UAAU,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACzC,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO;QACjC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,IAAI,sBAAsB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;YACrD,OAAO;QACT,CAAC;QACD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,UAAU,CAAC,OAAe,EAAE,iBAAyB,EAAE,UAAyB;IACvF,MAAM,WAAW,GAAG,iBAAiB,GAAG,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC;IACvD,OAAO,IAAI,CAAC,GAAG,CAAC,qCAAqC,EAAE,UAAU,IAAI,WAAW,CAAC,CAAC;AACpF,CAAC;AAED,MAAM,OAAO,iCAAiC;IACnC,OAAO,CAA2B;IAClC,SAAS,CAAM;IACf,OAAO,CAAyB;IAChC,UAAU,CAAe;IACzB,UAAU,CAAS;IACnB,WAAW,CAAS;IACpB,aAAa,CAAS;IACtB,UAAU,CAAgE;IAEnF,YAAY,OAAiD;QAC3D,MAAM,MAAM,GAAG,6BAA6B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7D,MAAM,OAAO,GAAG,yBAAyB,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAC9E,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,GAAG,8BAA8B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5D,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;QACtC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,IAAI,KAAK,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,cAAc,CAAC,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,EAAE,4BAA4B,CAAC,CAAC;QACpH,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC,OAAO,CAAC,UAAU,IAAI,kCAAkC,EAAE,CAAC,EAAE,CAAC,EAAE,4BAA4B,CAAC,CAAC;QAChI,IAAI,CAAC,aAAa,GAAG,cAAc,CAAC,OAAO,CAAC,YAAY,IAAI,GAAG,EAAE,CAAC,EAAE,qCAAqC,EAAE,gCAAgC,CAAC,CAAC;QAC7I,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,IAAI,YAAY,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAyB,EAAE,UAAoC,EAAE;QAC3E,MAAM,gBAAgB,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACxE,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QACzD,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,sBAAsB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAEhF,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,WAAW,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;YAChE,IAAI,QAAQ,GAAG,KAAK,CAAC;YACrB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACzC,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO;gBAAE,UAAU,CAAC,KAAK,EAAE,CAAC;;gBAC3C,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YACxE,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,QAAQ,GAAG,IAAI,CAAC;gBAChB,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;YACpB,IAAI,CAAC;gBACH,IAAI,QAAkB,CAAC;gBACvB,IAAI,CAAC;oBACH,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE;wBAC/C,MAAM,EAAE,MAAM;wBACd,QAAQ,EAAE,QAAQ;wBAClB,OAAO,EAAE;4BACP,MAAM,EAAE,kBAAkB;4BAC1B,cAAc,EAAE,kBAAkB;4BAClC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;yBACjG;wBACD,IAAI;wBACJ,MAAM,EAAE,UAAU,CAAC,MAAM;qBAC1B,CAAC,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACP,MAAM,QAAQ,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI;wBAChD,CAAC,CAAC,IAAI,sBAAsB,CAAC,SAAS,EAAE,QAAQ,CAAC;wBACjD,CAAC,CAAC,IAAI,sBAAsB,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;gBAC/D,CAAC;gBACD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC5B,MAAM,OAAO,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;oBACxC,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAC;oBAC/B,MAAM,OAAO,CAAC;gBAChB,CAAC;gBACD,MAAM,YAAY,GAAG,MAAM,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBACtD,IAAI,MAAe,CAAC;gBACpB,IAAI,CAAC;oBACH,MAAM,GAAG,eAAe,CACtB,YAAY,EACZ,EAAE,kBAAkB,EAAE,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAC/E,wCAAwC,CACzC,CAAC;gBACJ,CAAC;gBAAC,MAAM,CAAC;oBACP,eAAe,EAAE,CAAC;gBACpB,CAAC;gBACD,OAAO,qBAAqB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC9E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,UAAU,GAAG,KAAK,YAAY,sBAAsB;oBACxD,CAAC,CAAC,KAAK;oBACP,CAAC,CAAC,KAAK,YAAY,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB;wBAClE,CAAC,CAAC,IAAI,sBAAsB,CAAC,kBAAkB,EAAE,KAAK,CAAC;wBACvD,CAAC,CAAC,IAAI,sBAAsB,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;gBAC/D,MAAM,OAAO,GAAG,QAAQ;oBACtB,CAAC,CAAC,IAAI,sBAAsB,CAAC,SAAS,EAAE,IAAI,CAAC;oBAC7C,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI;wBAChC,CAAC,CAAC,IAAI,sBAAsB,CAAC,SAAS,EAAE,KAAK,CAAC;wBAC9C,CAAC,CAAC,UAAU,CAAC;gBACjB,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,IAAI,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO;oBAAE,MAAM,OAAO,CAAC;gBAChG,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBAC5E,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;YAC/C,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QACD,MAAM,IAAI,sBAAsB,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;IAClE,CAAC;CACF;AAED,OAAO,EAAE,iCAAiC,IAAI,wBAAwB,EAAE,CAAC"}

package/dist/embedding/presets/local-small.d.ts

@@ -0,0 +1,8 @@
+import type { LocalEmbeddingPreset } from './manifest.js';
+/**
+ * The only public local model preset. Every artifact coordinate is immutable;
+ * model installation code must consume this value rather than user input.
+ */
+export declare const LOCAL_SMALL_PRESET: LocalEmbeddingPreset;
+export declare const LOCAL_SMALL_PRESET_MANIFEST_HASH = "85d724217ffdda4333b6c2892f5138754d716f62e08a82cb0093ca5baca24fe4";
+//# sourceMappingURL=local-small.d.ts.map

package/dist/embedding/presets/local-small.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"local-small.d.ts","sourceRoot":"","sources":["../../../src/embedding/presets/local-small.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAE1D;;;GAGG;AACH,eAAO,MAAM,kBAAkB,EAAE,oBA4D/B,CAAC;AAEH,eAAO,MAAM,gCAAgC,qEACuB,CAAC"}