@reftrixmcp/ml 0.1.7

package/dist/embeddings/vision-embedding.service.js

@@ -0,0 +1,221 @@
+// SPDX-FileCopyrightText: 2026 TKMD and Reftrix Contributors
+// SPDX-License-Identifier: AGPL-3.0-only
+import { EmbeddingService } from "./service.js";
+/**
+ * デフォルトのキャッシュサイズ
+ */
+const DEFAULT_MAX_CACHE_SIZE = 5000;
+/**
+ * VisionFeaturesをテキスト表現に変換する
+ *
+ * @param features - 視覚特徴量
+ * @returns テキスト表現
+ */
+export function visionFeaturesToText(features) {
+    const parts = [
+        `visual_rhythm: ${features.rhythm}`,
+        `whitespace_ratio: ${features.whitespaceRatio}`,
+        `content_density: ${features.density}`,
+        `visual_gravity: ${features.gravity}`,
+        `color_theme: ${features.theme}`,
+    ];
+    // オプショナルフィールドを追加（存在する場合のみ）
+    if (features.mood !== undefined && features.mood !== null) {
+        parts.push(`mood: ${features.mood}`);
+    }
+    if (features.brandTone !== undefined && features.brandTone !== null) {
+        parts.push(`brandTone: ${features.brandTone}`);
+    }
+    return parts.join(", ");
+}
+/**
+ * VisionFeaturesからキャッシュキーを生成
+ */
+function getCacheKey(features) {
+    return JSON.stringify({
+        r: features.rhythm,
+        w: features.whitespaceRatio,
+        d: features.density,
+        g: features.gravity,
+        t: features.theme,
+        m: features.mood,
+        b: features.brandTone,
+    });
+}
+/**
+ * VisionEmbeddingService
+ *
+ * 視覚特徴量から768次元の埋め込みベクトルを生成するサービス
+ */
+export class VisionEmbeddingService {
+    embeddingService;
+    cache = new Map();
+    cacheHits = 0;
+    cacheMisses = 0;
+    cacheEvictions = 0;
+    maxCacheSize;
+    constructor(config = {}) {
+        this.maxCacheSize = config.maxCacheSize ?? DEFAULT_MAX_CACHE_SIZE;
+        this.embeddingService = new EmbeddingService();
+        if (process.env.NODE_ENV === "development") {
+            // eslint-disable-next-line no-console
+            console.log("[VisionEmbedding] Service created with maxCacheSize:", this.maxCacheSize);
+        }
+    }
+    /**
+     * VisionFeaturesから768次元の埋め込みベクトルを生成
+     *
+     * @param features - 視覚特徴量
+     * @returns 768次元のL2正規化されたベクトル
+     */
+    async generateVisionEmbedding(features) {
+        const cacheKey = getCacheKey(features);
+        // キャッシュチェック（LRU更新付き）
+        const cached = this.getCacheEntry(cacheKey);
+        if (cached) {
+            this.cacheHits++;
+            if (process.env.NODE_ENV === "development") {
+                // eslint-disable-next-line no-console
+                console.log("[VisionEmbedding] Cache hit");
+            }
+            return cached;
+        }
+        this.cacheMisses++;
+        // テキスト表現に変換
+        const text = visionFeaturesToText(features);
+        if (process.env.NODE_ENV === "development") {
+            // eslint-disable-next-line no-console
+            console.log("[VisionEmbedding] Generating embedding for:", text.substring(0, 80));
+        }
+        // EmbeddingServiceを使用して埋め込み生成
+        // 'passage'プレフィックスを使用（ドキュメント埋め込み）
+        const embedding = await this.embeddingService.generateEmbedding(text, "passage");
+        // キャッシュに保存（LRU eviction付き）
+        this.setCacheEntry(cacheKey, embedding);
+        return embedding;
+    }
+    /**
+     * 複数のVisionFeaturesから埋め込みベクトルを一括生成
+     *
+     * @param featuresArray - 視覚特徴量の配列
+     * @returns 768次元ベクトルの配列
+     */
+    async generateBatchVisionEmbeddings(featuresArray) {
+        if (featuresArray.length === 0) {
+            return [];
+        }
+        if (process.env.NODE_ENV === "development") {
+            // eslint-disable-next-line no-console
+            console.log("[VisionEmbedding] Batch processing", featuresArray.length, "items");
+        }
+        // キャッシュチェックと未キャッシュアイテムの特定
+        const results = new Array(featuresArray.length);
+        const uncachedIndices = [];
+        const uncachedTexts = [];
+        for (let i = 0; i < featuresArray.length; i++) {
+            const features = featuresArray[i];
+            if (!features)
+                continue;
+            const cacheKey = getCacheKey(features);
+            const cached = this.getCacheEntry(cacheKey);
+            if (cached) {
+                this.cacheHits++;
+                results[i] = cached;
+            }
+            else {
+                this.cacheMisses++;
+                uncachedIndices.push(i);
+                uncachedTexts.push(visionFeaturesToText(features));
+            }
+        }
+        // 未キャッシュアイテムの一括生成
+        if (uncachedTexts.length > 0) {
+            const embeddings = await this.embeddingService.generateBatchEmbeddings(uncachedTexts, "passage");
+            for (let j = 0; j < uncachedIndices.length; j++) {
+                const originalIndex = uncachedIndices[j];
+                const embedding = embeddings[j];
+                const features = featuresArray[originalIndex ?? 0];
+                if (originalIndex !== undefined && embedding && features) {
+                    results[originalIndex] = embedding;
+                    // キャッシュに保存
+                    const cacheKey = getCacheKey(features);
+                    this.setCacheEntry(cacheKey, embedding);
+                }
+            }
+        }
+        // undefinedを除去して返却
+        return results.filter((r) => r !== undefined);
+    }
+    /**
+     * キャッシュ統計を取得
+     */
+    getCacheStats() {
+        return {
+            hits: this.cacheHits,
+            misses: this.cacheMisses,
+            size: this.cache.size,
+            evictions: this.cacheEvictions,
+        };
+    }
+    /**
+     * キャッシュをクリア
+     */
+    clearCache() {
+        this.cache.clear();
+        this.cacheHits = 0;
+        this.cacheMisses = 0;
+        this.cacheEvictions = 0;
+        if (process.env.NODE_ENV === "development") {
+            // eslint-disable-next-line no-console
+            console.log("[VisionEmbedding] Cache cleared");
+        }
+    }
+    /**
+     * キャッシュエントリを取得（LRU更新）
+     */
+    getCacheEntry(key) {
+        const value = this.cache.get(key);
+        if (value !== undefined) {
+            // LRU: 最近使用したエントリを末尾に移動
+            this.cache.delete(key);
+            this.cache.set(key, value);
+        }
+        return value;
+    }
+    /**
+     * キャッシュエントリを設定（LRU eviction）
+     */
+    setCacheEntry(key, value) {
+        // キャッシュがいっぱいの場合、最も古いエントリを削除
+        while (this.cache.size >= this.maxCacheSize) {
+            const oldestKey = this.cache.keys().next().value;
+            if (oldestKey !== undefined) {
+                this.cache.delete(oldestKey);
+                this.cacheEvictions++;
+                if (process.env.NODE_ENV === "development") {
+                    // eslint-disable-next-line no-console
+                    console.log("[VisionEmbedding] Cache evicted oldest entry");
+                }
+            }
+            else {
+                break;
+            }
+        }
+        this.cache.set(key, value);
+    }
+}
+// シングルトンインスタンス
+export const visionEmbeddingService = new VisionEmbeddingService();
+/**
+ * 単一のVisionFeaturesから埋め込みを生成するヘルパー関数
+ */
+export async function createVisionEmbedding(features) {
+    return visionEmbeddingService.generateVisionEmbedding(features);
+}
+/**
+ * 複数のVisionFeaturesから埋め込みを一括生成するヘルパー関数
+ */
+export async function createBatchVisionEmbeddings(featuresArray) {
+    return visionEmbeddingService.generateBatchVisionEmbeddings(featuresArray);
+}
+//# sourceMappingURL=vision-embedding.service.js.map

package/dist/embeddings/vision-embedding.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vision-embedding.service.js","sourceRoot":"","sources":["../../src/embeddings/vision-embedding.service.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAC7D,yCAAyC;AAgBzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAEhD;;GAEG;AACH,MAAM,sBAAsB,GAAG,IAAI,CAAC;AAEpC;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAAC,QAAwB;IAC3D,MAAM,KAAK,GAAa;QACtB,kBAAkB,QAAQ,CAAC,MAAM,EAAE;QACnC,qBAAqB,QAAQ,CAAC,eAAe,EAAE;QAC/C,oBAAoB,QAAQ,CAAC,OAAO,EAAE;QACtC,mBAAmB,QAAQ,CAAC,OAAO,EAAE;QACrC,gBAAgB,QAAQ,CAAC,KAAK,EAAE;KACjC,CAAC;IAEF,2BAA2B;IAC3B,IAAI,QAAQ,CAAC,IAAI,KAAK,SAAS,IAAI,QAAQ,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;QAC1D,KAAK,CAAC,IAAI,CAAC,SAAS,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,IAAI,QAAQ,CAAC,SAAS,KAAK,IAAI,EAAE,CAAC;QACpE,KAAK,CAAC,IAAI,CAAC,cAAc,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,QAAwB;IAC3C,OAAO,IAAI,CAAC,SAAS,CAAC;QACpB,CAAC,EAAE,QAAQ,CAAC,MAAM;QAClB,CAAC,EAAE,QAAQ,CAAC,eAAe;QAC3B,CAAC,EAAE,QAAQ,CAAC,OAAO;QACnB,CAAC,EAAE,QAAQ,CAAC,OAAO;QACnB,CAAC,EAAE,QAAQ,CAAC,KAAK;QACjB,CAAC,EAAE,QAAQ,CAAC,IAAI;QAChB,CAAC,EAAE,QAAQ,CAAC,SAAS;KACtB,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,sBAAsB;IACzB,gBAAgB,CAAmB;IACnC,KAAK,GAA0B,IAAI,GAAG,EAAE,CAAC;IACzC,SAAS,GAAG,CAAC,CAAC;IACd,WAAW,GAAG,CAAC,CAAC;IAChB,cAAc,GAAG,CAAC,CAAC;IACnB,YAAY,CAAS;IAE7B,YAAY,SAAuC,EAAE;QACnD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,sBAAsB,CAAC;QAClE,IAAI,CAAC,gBAAgB,GAAG,IAAI,gBAAgB,EAAE,CAAC;QAE/C,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,EAAE,CAAC;YAC3C,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,sDAAsD,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACzF,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,uBAAuB,CAAC,QAAwB;QACpD,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;QAEvC,qBAAqB;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,EAAE,CAAC;gBAC3C,sCAAsC;gBACtC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;YAC7C,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,YAAY;QACZ,MAAM,IAAI,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE5C,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,EAAE,CAAC;YAC3C,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,6CAA6C,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACpF,CAAC;QAED,8BAA8B;QAC9B,kCAAkC;QAClC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAEjF,2BAA2B;QAC3B,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAExC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,6BAA6B,CAAC,aAA+B;QACjE,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,EAAE,CAAC;YAC3C,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,oCAAoC,EAAE,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACnF,CAAC;QAED,0BAA0B;QAC1B,MAAM,OAAO,GAA6B,IAAI,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC1E,MAAM,eAAe,GAAa,EAAE,CAAC;QACrC,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAClC,IAAI,CAAC,QAAQ;gBAAE,SAAS;YAExB,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;YAE5C,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,CAAC,SAAS,EAAE,CAAC;gBACjB,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACxB,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,kBAAkB;QAClB,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,uBAAuB,CACpE,aAAa,EACb,SAAS,CACV,CAAC;YAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAChD,MAAM,aAAa,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;gBAChC,MAAM,QAAQ,GAAG,aAAa,CAAC,aAAa,IAAI,CAAC,CAAC,CAAC;gBAEnD,IAAI,aAAa,KAAK,SAAS,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;oBACzD,OAAO,CAAC,aAAa,CAAC,GAAG,SAAS,CAAC;oBAEnC,WAAW;oBACX,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;oBACvC,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC;QAED,mBAAmB;QACnB,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAiB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,SAAS;YACpB,MAAM,EAAE,IAAI,CAAC,WAAW;YACxB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;YACrB,SAAS,EAAE,IAAI,CAAC,cAAc;SAC/B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;QAExB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,EAAE,CAAC;YAC3C,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,GAAW;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,wBAAwB;YACxB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACvB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,GAAW,EAAE,KAAe;QAChD,4BAA4B;QAC5B,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YACjD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC7B,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,EAAE,CAAC;oBAC3C,sCAAsC;oBACtC,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM;YACR,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAC7B,CAAC;CACF;AAED,eAAe;AACf,MAAM,CAAC,MAAM,sBAAsB,GAAG,IAAI,sBAAsB,EAAE,CAAC;AAEnE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,QAAwB;IAClE,OAAO,sBAAsB,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;AAClE,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,aAA+B;IAE/B,OAAO,sBAAsB,CAAC,6BAA6B,CAAC,aAAa,CAAC,CAAC;AAC7E,CAAC"}

package/dist/embeddings/vision-embedding.types.d.ts

@@ -0,0 +1,102 @@
+/**
+ * Vision Embedding Types
+ *
+ * Vision Embedding生成ロジック
+ *
+ * 視覚特徴量を表す型定義
+ */
+/**
+ * 視覚的リズムの種類
+ * - regular: 規則的なリズム（均等配置）
+ * - varied: 変化のあるリズム（意図的な強弱）
+ * - asymmetric: 非対称なリズム（アシンメトリック）
+ */
+export type VisionRhythm = "regular" | "varied" | "asymmetric";
+/**
+ * コンテンツ密度の種類
+ * - sparse: 疎（ホワイトスペースが多い）
+ * - moderate: 中程度
+ * - dense: 密集（コンテンツが多い）
+ */
+export type VisionDensity = "sparse" | "moderate" | "dense";
+/**
+ * 視覚的重心の位置
+ * - top: 上部
+ * - center: 中央
+ * - bottom: 下部
+ * - left: 左側
+ * - right: 右側
+ */
+export type VisionGravity = "top" | "center" | "bottom" | "left" | "right";
+/**
+ * カラーテーマの種類
+ * - light: ライトテーマ
+ * - dark: ダークテーマ
+ * - mixed: 混合テーマ
+ */
+export type VisionTheme = "light" | "dark" | "mixed";
+/**
+ * 視覚特徴量インターフェース
+ *
+ * Webページのレイアウト/デザインから抽出された視覚的特徴を表す
+ */
+export interface VisionFeatures {
+    /**
+     * 視覚的リズム
+     * 要素の配置パターン・繰り返しの規則性
+     */
+    rhythm: VisionRhythm;
+    /**
+     * ホワイトスペース比率 (0-1)
+     * 0: ホワイトスペースなし（完全にコンテンツで埋まっている）
+     * 1: 完全にホワイトスペース（コンテンツなし）
+     */
+    whitespaceRatio: number;
+    /**
+     * コンテンツ密度
+     * ページ上のコンテンツ要素の密集度
+     */
+    density: VisionDensity;
+    /**
+     * 視覚的重心
+     * コンテンツの視覚的な重心位置
+     */
+    gravity: VisionGravity;
+    /**
+     * カラーテーマ
+     * 全体的な色調
+     */
+    theme: VisionTheme;
+    /**
+     * ムード (オプション)
+     * Vision AIから取得したページの雰囲気/印象
+     * 例: 'professional', 'playful', 'elegant', 'minimal'
+     */
+    mood?: string;
+    /**
+     * ブランドトーン (オプション)
+     * Vision AIから取得したブランドの印象
+     * 例: 'corporate', 'startup', 'luxury', 'friendly'
+     */
+    brandTone?: string;
+}
+/**
+ * VisionEmbeddingServiceの設定オプション
+ */
+export interface VisionEmbeddingServiceConfig {
+    /**
+     * キャッシュの最大サイズ
+     * @default 5000
+     */
+    maxCacheSize?: number;
+}
+/**
+ * キャッシュ統計情報
+ */
+export interface VisionCacheStats {
+    hits: number;
+    misses: number;
+    size: number;
+    evictions: number;
+}
+//# sourceMappingURL=vision-embedding.types.d.ts.map

package/dist/embeddings/vision-embedding.types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vision-embedding.types.d.ts","sourceRoot":"","sources":["../../src/embeddings/vision-embedding.types.ts"],"names":[],"mappings":"AAGA;;;;;;GAMG;AAEH;;;;;GAKG;AACH,MAAM,MAAM,YAAY,GAAG,SAAS,GAAG,QAAQ,GAAG,YAAY,CAAC;AAE/D;;;;;GAKG;AACH,MAAM,MAAM,aAAa,GAAG,QAAQ,GAAG,UAAU,GAAG,OAAO,CAAC;AAE5D;;;;;;;GAOG;AACH,MAAM,MAAM,aAAa,GAAG,KAAK,GAAG,QAAQ,GAAG,QAAQ,GAAG,MAAM,GAAG,OAAO,CAAC;AAE3E;;;;;GAKG;AACH,MAAM,MAAM,WAAW,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,CAAC;AAErD;;;;GAIG;AACH,MAAM,WAAW,cAAc;IAC7B;;;OAGG;IACH,MAAM,EAAE,YAAY,CAAC;IAErB;;;;OAIG;IACH,eAAe,EAAE,MAAM,CAAC;IAExB;;;OAGG;IACH,OAAO,EAAE,aAAa,CAAC;IAEvB;;;OAGG;IACH,OAAO,EAAE,aAAa,CAAC;IAEvB;;;OAGG;IACH,KAAK,EAAE,WAAW,CAAC;IAEnB;;;;OAIG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;;OAIG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC3C;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACnB"}

package/dist/embeddings/vision-embedding.types.js

@@ -0,0 +1,4 @@
+// SPDX-FileCopyrightText: 2026 TKMD and Reftrix Contributors
+// SPDX-License-Identifier: AGPL-3.0-only
+export {};
+//# sourceMappingURL=vision-embedding.types.js.map

package/dist/embeddings/vision-embedding.types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vision-embedding.types.js","sourceRoot":"","sources":["../../src/embeddings/vision-embedding.types.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAC7D,yCAAyC"}

package/dist/embeddings/worker-thread-types.d.ts

@@ -0,0 +1,152 @@
+/**
+ * Worker Thread message types for ONNX embedding inference.
+ *
+ * Communication protocol between main thread (EmbeddingService) and
+ * worker thread (embedding-worker.ts). Messages are serialized via
+ * structured clone (MessagePort), so only plain objects and typed arrays
+ * are used — no ONNX Session/Tensor objects cross the boundary.
+ *
+ * @module embeddings/worker-thread-types
+ */
+/**
+ * Initialize the ONNX pipeline in the worker thread.
+ */
+export interface WorkerInitMessage {
+    type: "init";
+    requestId: string;
+    config: {
+        modelId: string;
+        cacheDir: string;
+        device: string;
+        dtype: string;
+        pipelineRecycleThreshold: number;
+    };
+}
+/**
+ * Generate embedding for a single prefixed text.
+ * The main thread is responsible for adding E5 prefix before sending.
+ */
+export interface WorkerGenerateMessage {
+    type: "generate";
+    requestId: string;
+    text: string;
+}
+/**
+ * Generate embeddings for a batch of prefixed texts.
+ * The main thread is responsible for adding E5 prefix before sending.
+ */
+export interface WorkerGenerateBatchMessage {
+    type: "generateBatch";
+    requestId: string;
+    texts: string[];
+}
+/**
+ * Dispose the ONNX pipeline to free native memory.
+ */
+export interface WorkerDisposeMessage {
+    type: "dispose";
+    requestId: string;
+}
+/**
+ * Terminate the worker thread gracefully.
+ */
+export interface WorkerTerminateMessage {
+    type: "terminate";
+    requestId: string;
+}
+/**
+ * Switch the ONNX execution provider at runtime (e.g. cpu → cuda).
+ * Disposes the current pipeline and re-initializes with the new provider.
+ */
+export interface WorkerSwitchProviderMessage {
+    type: "switch-provider";
+    requestId: string;
+    provider: "cpu" | "cuda";
+}
+/**
+ * Release GPU resources by disposing the pipeline and reverting to CPU.
+ * Safe to call even if already on CPU — acts as a no-op dispose.
+ */
+export interface WorkerReleaseGpuMessage {
+    type: "release-gpu";
+    requestId: string;
+}
+export type WorkerMessage = WorkerInitMessage | WorkerGenerateMessage | WorkerGenerateBatchMessage | WorkerDisposeMessage | WorkerTerminateMessage | WorkerSwitchProviderMessage | WorkerReleaseGpuMessage;
+/**
+ * Successful response from init.
+ */
+export interface WorkerInitResponse {
+    type: "init";
+    requestId: string;
+    success: true;
+    loadTimeMs: number;
+    /** The resolved ONNX execution provider ('cpu' or 'cuda'). */
+    executionProvider: string;
+}
+/**
+ * Successful response from generate.
+ * Embedding is a plain number[] (structured clone compatible).
+ */
+export interface WorkerGenerateResponse {
+    type: "generate";
+    requestId: string;
+    success: true;
+    embedding: number[];
+    inferenceTimeMs: number;
+}
+/**
+ * Successful response from generateBatch.
+ */
+export interface WorkerGenerateBatchResponse {
+    type: "generateBatch";
+    requestId: string;
+    success: true;
+    embeddings: number[][];
+    inferenceTimeMs: number;
+}
+/**
+ * Successful response from dispose.
+ */
+export interface WorkerDisposeResponse {
+    type: "dispose";
+    requestId: string;
+    success: true;
+}
+/**
+ * Successful response from terminate (sent just before process.exit).
+ */
+export interface WorkerTerminateResponse {
+    type: "terminate";
+    requestId: string;
+    success: true;
+}
+/**
+ * Successful response from switch-provider.
+ */
+export interface WorkerSwitchProviderResponse {
+    type: "switch-provider";
+    requestId: string;
+    success: true;
+    /** The provider now in effect after the switch. */
+    provider: "cpu" | "cuda";
+}
+/**
+ * Successful response from release-gpu.
+ */
+export interface WorkerReleaseGpuResponse {
+    type: "release-gpu";
+    requestId: string;
+    success: true;
+}
+/**
+ * Error response for any message type.
+ */
+export interface WorkerErrorResponse {
+    type: "error";
+    requestId: string;
+    success: false;
+    error: string;
+    originalType: WorkerMessage["type"];
+}
+export type WorkerResponse = WorkerInitResponse | WorkerGenerateResponse | WorkerGenerateBatchResponse | WorkerDisposeResponse | WorkerTerminateResponse | WorkerSwitchProviderResponse | WorkerReleaseGpuResponse | WorkerErrorResponse;
+//# sourceMappingURL=worker-thread-types.d.ts.map

package/dist/embeddings/worker-thread-types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-thread-types.d.ts","sourceRoot":"","sources":["../../src/embeddings/worker-thread-types.ts"],"names":[],"mappings":"AAGA;;;;;;;;;GASG;AAMH;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE;QACN,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;QACd,wBAAwB,EAAE,MAAM,CAAC;KAClC,CAAC;CACH;AAED;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,UAAU,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;;GAGG;AACH,MAAM,WAAW,0BAA0B;IACzC,IAAI,EAAE,eAAe,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,IAAI,EAAE,WAAW,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,MAAM,WAAW,2BAA2B;IAC1C,IAAI,EAAE,iBAAiB,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,KAAK,GAAG,MAAM,CAAC;CAC1B;AAED;;;GAGG;AACH,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,aAAa,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,aAAa,GACrB,iBAAiB,GACjB,qBAAqB,GACrB,0BAA0B,GAC1B,oBAAoB,GACpB,sBAAsB,GACtB,2BAA2B,GAC3B,uBAAuB,CAAC;AAM5B;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,IAAI,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,8DAA8D;IAC9D,iBAAiB,EAAE,MAAM,CAAC;CAC3B;AAED;;;GAGG;AACH,MAAM,WAAW,sBAAsB;IACrC,IAAI,EAAE,UAAU,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,IAAI,CAAC;IACd,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,2BAA2B;IAC1C,IAAI,EAAE,eAAe,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,IAAI,CAAC;IACd,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,IAAI,CAAC;CACf;AAED;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,WAAW,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,IAAI,CAAC;CACf;AAED;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC3C,IAAI,EAAE,iBAAiB,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,IAAI,CAAC;IACd,mDAAmD;IACnD,QAAQ,EAAE,KAAK,GAAG,MAAM,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC,IAAI,EAAE,aAAa,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,IAAI,CAAC;CACf;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,KAAK,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;CACrC;AAED,MAAM,MAAM,cAAc,GACtB,kBAAkB,GAClB,sBAAsB,GACtB,2BAA2B,GAC3B,qBAAqB,GACrB,uBAAuB,GACvB,4BAA4B,GAC5B,wBAAwB,GACxB,mBAAmB,CAAC"}

package/dist/embeddings/worker-thread-types.js

@@ -0,0 +1,4 @@
+// SPDX-FileCopyrightText: 2026 TKMD and Reftrix Contributors
+// SPDX-License-Identifier: AGPL-3.0-only
+export {};
+//# sourceMappingURL=worker-thread-types.js.map

package/dist/embeddings/worker-thread-types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-thread-types.js","sourceRoot":"","sources":["../../src/embeddings/worker-thread-types.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAC7D,yCAAyC"}

package/dist/embeddings/worker-thread.d.ts

@@ -0,0 +1,4 @@
+declare let inferencesSinceRecycle: number;
+declare let totalRecycles: number;
+export { totalRecycles, inferencesSinceRecycle };
+//# sourceMappingURL=worker-thread.d.ts.map

package/dist/embeddings/worker-thread.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-thread.d.ts","sourceRoot":"","sources":["../../src/embeddings/worker-thread.ts"],"names":[],"mappings":"AA2HA,QAAA,IAAI,sBAAsB,QAAI,CAAC;AAC/B,QAAA,IAAI,aAAa,QAAI,CAAC;AAoZtB,OAAO,EAAE,aAAa,EAAE,sBAAsB,EAAE,CAAC"}