kiri-mcp-server 0.17.0 → 0.18.0

package/dist/src/vss/model-downloader.js

@@ -1,320 +0,0 @@
-/**
- * VSS Model Downloader
- *
- * Downloads and manages VSS models from HuggingFace Hub.
- * Models are stored in the global ~/.kiri/models/ directory.
- */
-import { createWriteStream, existsSync, mkdirSync, readdirSync, rmSync, statSync } from "node:fs";
-import { rename, rm } from "node:fs/promises";
-import { join } from "node:path";
-import { Readable } from "node:stream";
-import { pipeline } from "node:stream/promises";
-import { getModelPath, ensureModelsDir, isModelInstalled } from "../shared/utils/global-dirs.js";
-import { SUPPORTED_MODELS } from "./config.js";
-/**
- * Model registry with download information
- *
- * Note: ONNX-ready models (jina-embeddings, all-MiniLM) are auto-downloaded
- * by @huggingface/transformers and don't need manual download via this registry.
- */
-export const MODEL_REGISTRY = {
-    "codebert-base": {
-        repo: "microsoft/codebert-base",
-        files: ["config.json", "tokenizer.json", "vocab.txt", "pytorch_model.bin"],
-        sizeBytes: 500_000_000,
-        sizeHuman: "~500MB",
-        description: "Microsoft CodeBERT base model for code understanding",
-    },
-    "unixcoder-base": {
-        repo: "microsoft/unixcoder-base",
-        files: ["config.json", "tokenizer.json", "vocab.txt", "pytorch_model.bin"],
-        sizeBytes: 500_000_000,
-        sizeHuman: "~500MB",
-        description: "Microsoft UniXcoder base model for code representation",
-    },
-    // ONNX-ready models: auto-downloaded by @huggingface/transformers
-    "jina-embeddings-v2-base-code": {
-        repo: "jinaai/jina-embeddings-v2-base-code",
-        files: [], // Auto-managed by transformers.js
-        sizeBytes: 322_000_000,
-        sizeHuman: "~322MB",
-        description: "Jina code embedding model (ONNX ready, 30+ languages)",
-    },
-    "all-MiniLM-L6-v2": {
-        repo: "Xenova/all-MiniLM-L6-v2",
-        files: [], // Auto-managed by transformers.js
-        sizeBytes: 23_000_000,
-        sizeHuman: "~23MB",
-        description: "Lightweight general embedding model (ONNX ready)",
-    },
-};
-/**
- * Get model information
- */
-export function getModelInfo(model) {
-    return MODEL_REGISTRY[model];
-}
-/**
- * List all available models
- */
-export function listAvailableModels() {
-    return SUPPORTED_MODELS.map((name) => ({
-        name,
-        info: MODEL_REGISTRY[name],
-        installed: isModelInstalled(name),
-    }));
-}
-/**
- * Build HuggingFace download URL
- */
-function buildHuggingFaceUrl(repo, file) {
-    return `https://huggingface.co/${repo}/resolve/main/${file}`;
-}
-/**
- * Download a single file with progress reporting and resume support
- */
-async function downloadFile(url, destPath, onProgress) {
-    // Ensure parent directory exists
-    const dir = join(destPath, "..");
-    if (!existsSync(dir)) {
-        mkdirSync(dir, { recursive: true });
-    }
-    // Check for existing partial download (temp file with PID)
-    const tempPath = `${destPath}.${process.pid}.tmp`;
-    let resumeFrom = 0;
-    if (existsSync(tempPath)) {
-        try {
-            const tempStat = statSync(tempPath);
-            resumeFrom = tempStat.size;
-        }
-        catch {
-            // Ignore stat errors, start fresh
-            resumeFrom = 0;
-        }
-    }
-    // Build request headers
-    const headers = {
-        "User-Agent": "kiri-mcp-server",
-    };
-    // Add Range header for resume if we have partial content
-    if (resumeFrom > 0) {
-        headers["Range"] = `bytes=${resumeFrom}-`;
-    }
-    const response = await fetch(url, { headers });
-    // Handle response status
-    if (!response.ok && response.status !== 206) {
-        // 206 = Partial Content (resume success)
-        throw new Error(`Failed to download ${url}: ${response.status} ${response.statusText}`);
-    }
-    // Calculate total size
-    let total = 0;
-    const contentLength = response.headers.get("content-length");
-    const contentRange = response.headers.get("content-range");
-    if (response.status === 206 && contentRange) {
-        // Parse Content-Range: bytes 1000-9999/10000
-        const match = contentRange.match(/\/(\d+)$/);
-        if (match?.[1]) {
-            total = parseInt(match[1], 10);
-        }
-    }
-    else if (contentLength) {
-        total = parseInt(contentLength, 10);
-        // If server doesn't support resume, start fresh
-        if (resumeFrom > 0 && response.status === 200) {
-            resumeFrom = 0;
-            // Remove partial file since we're starting over
-            try {
-                rmSync(tempPath);
-            }
-            catch {
-                // Ignore
-            }
-        }
-    }
-    // Open file stream (append if resuming, otherwise create new)
-    const fileStream = createWriteStream(tempPath, {
-        flags: resumeFrom > 0 && response.status === 206 ? "a" : "w",
-    });
-    let downloaded = resumeFrom;
-    const body = response.body;
-    if (!body) {
-        throw new Error(`No response body for ${url}`);
-    }
-    // Convert Web ReadableStream to Node.js Readable
-    const reader = body.getReader();
-    const nodeStream = new Readable({
-        async read() {
-            const { done, value } = await reader.read();
-            if (done) {
-                this.push(null);
-            }
-            else {
-                downloaded += value.length;
-                onProgress?.(downloaded, total);
-                this.push(Buffer.from(value));
-            }
-        },
-    });
-    await pipeline(nodeStream, fileStream);
-    // Rename temp file to final destination
-    await rename(tempPath, destPath);
-}
-/**
- * Download a model from HuggingFace
- */
-export async function downloadModel(model, onProgress) {
-    const info = getModelInfo(model);
-    if (!info) {
-        return {
-            success: false,
-            modelPath: "",
-            error: `Unknown model: ${model}. Supported models: ${SUPPORTED_MODELS.join(", ")}`,
-        };
-    }
-    // Ensure models directory exists
-    ensureModelsDir();
-    const modelPath = getModelPath(model);
-    // Create model directory
-    if (!existsSync(modelPath)) {
-        mkdirSync(modelPath, { recursive: true });
-    }
-    try {
-        // Download each file
-        for (const file of info.files) {
-            const url = buildHuggingFaceUrl(info.repo, file);
-            const destPath = join(modelPath, file);
-            // Skip if file already exists
-            if (existsSync(destPath)) {
-                onProgress?.({
-                    file,
-                    downloaded: 0,
-                    total: 0,
-                    percentage: 100,
-                    phase: "complete",
-                });
-                continue;
-            }
-            onProgress?.({
-                file,
-                downloaded: 0,
-                total: 0,
-                percentage: 0,
-                phase: "downloading",
-            });
-            await downloadFile(url, destPath, (downloaded, total) => {
-                const percentage = total > 0 ? Math.round((downloaded / total) * 100) : 0;
-                onProgress?.({
-                    file,
-                    downloaded,
-                    total,
-                    percentage,
-                    phase: "downloading",
-                });
-            });
-            onProgress?.({
-                file,
-                downloaded: 0,
-                total: 0,
-                percentage: 100,
-                phase: "complete",
-            });
-        }
-        // Verify installation
-        if (!isModelInstalled(model)) {
-            throw new Error("Model verification failed after download");
-        }
-        return {
-            success: true,
-            modelPath,
-        };
-    }
-    catch (error) {
-        // Clean up partial download
-        try {
-            if (existsSync(modelPath)) {
-                rmSync(modelPath, { recursive: true, force: true });
-            }
-        }
-        catch {
-            // Ignore cleanup errors
-        }
-        const errorMessage = error instanceof Error ? error.message : String(error);
-        onProgress?.({
-            file: "",
-            downloaded: 0,
-            total: 0,
-            percentage: 0,
-            phase: "error",
-            error: errorMessage,
-        });
-        return {
-            success: false,
-            modelPath: "",
-            error: errorMessage,
-        };
-    }
-}
-/**
- * Remove an installed model
- */
-export async function removeModel(model) {
-    const modelPath = getModelPath(model);
-    if (!existsSync(modelPath)) {
-        return {
-            success: false,
-            error: `Model "${model}" is not installed`,
-        };
-    }
-    try {
-        await rm(modelPath, { recursive: true, force: true });
-        return { success: true };
-    }
-    catch (error) {
-        return {
-            success: false,
-            error: error instanceof Error ? error.message : String(error),
-        };
-    }
-}
-/**
- * Get status of a specific model
- */
-export function getModelStatus(model) {
-    const modelPath = getModelPath(model);
-    const installed = isModelInstalled(model);
-    const status = {
-        name: model,
-        installed,
-        path: modelPath,
-    };
-    if (installed && existsSync(modelPath)) {
-        const files = readdirSync(modelPath);
-        status.files = files;
-        let totalSize = 0;
-        for (const file of files) {
-            try {
-                const fileStat = statSync(join(modelPath, file));
-                totalSize += fileStat.size;
-            }
-            catch {
-                // Ignore stat errors
-            }
-        }
-        status.sizeOnDisk = totalSize;
-    }
-    return status;
-}
-/**
- * Format bytes to human-readable string
- */
-export function formatBytes(bytes) {
-    if (!Number.isFinite(bytes) || bytes < 0)
-        return "0 B";
-    if (bytes === 0)
-        return "0 B";
-    const k = 1024;
-    const sizes = ["B", "KB", "MB", "GB", "TB"];
-    const i = Math.min(Math.floor(Math.log(bytes) / Math.log(k)), sizes.length - 1);
-    return `${parseFloat((bytes / Math.pow(k, i)).toFixed(1))} ${sizes[i]}`;
-}
-//# sourceMappingURL=model-downloader.js.map

package/dist/src/vss/model-downloader.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"model-downloader.js","sourceRoot":"","sources":["../../../src/vss/model-downloader.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAClG,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEhD,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAEjG,OAAO,EAAuB,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAkBpE;;;;;GAKG;AACH,MAAM,CAAC,MAAM,cAAc,GAAsC;IAC/D,eAAe,EAAE;QACf,IAAI,EAAE,yBAAyB;QAC/B,KAAK,EAAE,CAAC,aAAa,EAAE,gBAAgB,EAAE,WAAW,EAAE,mBAAmB,CAAC;QAC1E,SAAS,EAAE,WAAW;QACtB,SAAS,EAAE,QAAQ;QACnB,WAAW,EAAE,sDAAsD;KACpE;IACD,gBAAgB,EAAE;QAChB,IAAI,EAAE,0BAA0B;QAChC,KAAK,EAAE,CAAC,aAAa,EAAE,gBAAgB,EAAE,WAAW,EAAE,mBAAmB,CAAC;QAC1E,SAAS,EAAE,WAAW;QACtB,SAAS,EAAE,QAAQ;QACnB,WAAW,EAAE,wDAAwD;KACtE;IACD,kEAAkE;IAClE,8BAA8B,EAAE;QAC9B,IAAI,EAAE,qCAAqC;QAC3C,KAAK,EAAE,EAAE,EAAE,kCAAkC;QAC7C,SAAS,EAAE,WAAW;QACtB,SAAS,EAAE,QAAQ;QACnB,WAAW,EAAE,uDAAuD;KACrE;IACD,kBAAkB,EAAE;QAClB,IAAI,EAAE,yBAAyB;QAC/B,KAAK,EAAE,EAAE,EAAE,kCAAkC;QAC7C,SAAS,EAAE,UAAU;QACrB,SAAS,EAAE,OAAO;QAClB,WAAW,EAAE,kDAAkD;KAChE;CACF,CAAC;AAkCF;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,KAAqB;IAChD,OAAO,cAAc,CAAC,KAAK,CAAC,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB;IAKjC,OAAO,gBAAgB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACrC,IAAI;QACJ,IAAI,EAAE,cAAc,CAAC,IAAI,CAAC;QAC1B,SAAS,EAAE,gBAAgB,CAAC,IAAI,CAAC;KAClC,CAAC,CAAC,CAAC;AACN,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,IAAY,EAAE,IAAY;IACrD,OAAO,0BAA0B,IAAI,iBAAiB,IAAI,EAAE,CAAC;AAC/D,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,YAAY,CACzB,GAAW,EACX,QAAgB,EAChB,UAAwD;IAExD,iCAAiC;IACjC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACjC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACrB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACtC,CAAC;IAED,2DAA2D;IAC3D,MAAM,QAAQ,GAAG,GAAG,QAAQ,IAAI,OAAO,CAAC,GAAG,MAAM,CAAC;IAClD,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACpC,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC;QAC7B,CAAC;QAAC,MAAM,CAAC;YACP,kCAAkC;YAClC,UAAU,GAAG,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAED,wBAAwB;IACxB,MAAM,OAAO,GAA2B;QACtC,YAAY,EAAE,iBAAiB;KAChC,CAAC;IAEF,yDAAyD;IACzD,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;QACnB,OAAO,CAAC,OAAO,CAAC,GAAG,SAAS,UAAU,GAAG,CAAC;IAC5C,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAE/C,yBAAyB;IACzB,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC5C,yCAAyC;QACzC,MAAM,IAAI,KAAK,CAAC,sBAAsB,GAAG,KAAK,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;IAC1F,CAAC;IAED,uBAAuB;IACvB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC7D,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IAE3D,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAC;QAC5C,6CAA6C;QAC7C,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC7C,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACf,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;SAAM,IAAI,aAAa,EAAE,CAAC;QACzB,KAAK,GAAG,QAAQ,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;QACpC,gDAAgD;QAChD,IAAI,UAAU,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC9C,UAAU,GAAG,CAAC,CAAC;YACf,gDAAgD;YAChD,IAAI,CAAC;gBACH,MAAM,CAAC,QAAQ,CAAC,CAAC;YACnB,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAED,8DAA8D;IAC9D,MAAM,UAAU,GAAG,iBAAiB,CAAC,QAAQ,EAAE;QAC7C,KAAK,EAAE,UAAU,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG;KAC7D,CAAC,CAAC;IAEH,IAAI,UAAU,GAAG,UAAU,CAAC;IAC5B,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IAC3B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,iDAAiD;IACjD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC;QAC9B,KAAK,CAAC,IAAI;YACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,UAAU,IAAI,KAAK,CAAC,MAAM,CAAC;gBAC3B,UAAU,EAAE,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAChC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IAEvC,wCAAwC;IACxC,MAAM,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,KAAqB,EACrB,UAA6B;IAE7B,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,OAAO,EAAE,KAAK;YACd,SAAS,EAAE,EAAE;YACb,KAAK,EAAE,kBAAkB,KAAK,uBAAuB,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SACnF,CAAC;IACJ,CAAC;IAED,iCAAiC;IACjC,eAAe,EAAE,CAAC;IAClB,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IAEtC,yBAAyB;IACzB,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,IAAI,CAAC;QACH,qBAAqB;QACrB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAEvC,8BAA8B;YAC9B,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACzB,UAAU,EAAE,CAAC;oBACX,IAAI;oBACJ,UAAU,EAAE,CAAC;oBACb,KAAK,EAAE,CAAC;oBACR,UAAU,EAAE,GAAG;oBACf,KAAK,EAAE,UAAU;iBAClB,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YAED,UAAU,EAAE,CAAC;gBACX,IAAI;gBACJ,UAAU,EAAE,CAAC;gBACb,KAAK,EAAE,CAAC;gBACR,UAAU,EAAE,CAAC;gBACb,KAAK,EAAE,aAAa;aACrB,CAAC,CAAC;YAEH,MAAM,YAAY,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE;gBACtD,MAAM,UAAU,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1E,UAAU,EAAE,CAAC;oBACX,IAAI;oBACJ,UAAU;oBACV,KAAK;oBACL,UAAU;oBACV,KAAK,EAAE,aAAa;iBACrB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,UAAU,EAAE,CAAC;gBACX,IAAI;gBACJ,UAAU,EAAE,CAAC;gBACb,KAAK,EAAE,CAAC;gBACR,UAAU,EAAE,GAAG;gBACf,KAAK,EAAE,UAAU;aAClB,CAAC,CAAC;QACL,CAAC;QAED,sBAAsB;QACtB,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,SAAS;SACV,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,4BAA4B;QAC5B,IAAI,CAAC;YACH,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC1B,MAAM,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;QAED,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,UAAU,EAAE,CAAC;YACX,IAAI,EAAE,EAAE;YACR,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,CAAC;YACR,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,OAAO;YACd,KAAK,EAAE,YAAY;SACpB,CAAC,CAAC;QAEH,OAAO;YACL,OAAO,EAAE,KAAK;YACd,SAAS,EAAE,EAAE;YACb,KAAK,EAAE,YAAY;SACpB,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,KAAqB;IAErB,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IAEtC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,UAAU,KAAK,oBAAoB;SAC3C,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,EAAE,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACtD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC;AAaD;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,KAAqB;IAClD,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACtC,MAAM,SAAS,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IAE1C,MAAM,MAAM,GAAgB;QAC1B,IAAI,EAAE,KAAK;QACX,SAAS;QACT,IAAI,EAAE,SAAS;KAChB,CAAC;IAEF,IAAI,SAAS,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;QACrC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QAErB,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;gBACjD,SAAS,IAAI,QAAQ,CAAC,IAAI,CAAC;YAC7B,CAAC;YAAC,MAAM,CAAC;gBACP,qBAAqB;YACvB,CAAC;QACH,CAAC;QACD,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC;IAChC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,KAAa;IACvC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IACvD,IAAI,KAAK,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC9B,MAAM,CAAC,GAAG,IAAI,CAAC;IACf,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAC5C,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAChF,OAAO,GAAG,UAAU,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AAC1E,CAAC"}

package/dist/src/vss/provider.d.ts

@@ -1,132 +0,0 @@
-/**
- * VSS Embedding Provider Interface
- *
- * Defines the interface for generating semantic embeddings.
- * Implementations can use different backends (LSH, Transformers.js, external API).
- */
-import { type SupportedModel, type VSSConfig } from "./config.js";
-/**
- * Embedding result for a single text
- */
-export interface EmbeddingResult {
-    /** Embedding dimensions */
-    dims: number;
-    /** Embedding vector values */
-    values: number[];
-}
-/**
- * Batch embedding result
- */
-export interface BatchEmbeddingResult {
-    /** Path to the file */
-    path: string;
-    /** Embedding result (null if generation failed) */
-    embedding: EmbeddingResult | null;
-}
-/**
- * Embedding provider interface
- */
-export interface EmbeddingProvider {
-    /** Provider name */
-    readonly name: string;
-    /** Embedding dimensions */
-    readonly dims: number;
-    /** Whether the provider is ready to use */
-    readonly ready: boolean;
-    /**
-     * Generate embedding for a single text
-     */
-    generateEmbedding(text: string): Promise<EmbeddingResult | null>;
-    /**
-     * Generate embeddings for multiple texts (batch processing)
-     */
-    generateBatch(items: Array<{
-        path: string;
-        content: string;
-    }>): Promise<BatchEmbeddingResult[]>;
-    /**
-     * Initialize the provider (e.g., load model)
-     */
-    initialize(): Promise<void>;
-    /**
-     * Clean up resources
-     */
-    dispose(): Promise<void>;
-}
-/**
- * LSH-based embedding provider (current implementation)
- * Uses locality-sensitive hashing for structural similarity
- */
-export declare class LSHEmbeddingProvider implements EmbeddingProvider {
-    readonly name = "lsh";
-    readonly dims = 64;
-    readonly ready = true;
-    generateEmbedding(text: string): Promise<EmbeddingResult | null>;
-    generateBatch(items: Array<{
-        path: string;
-        content: string;
-    }>): Promise<BatchEmbeddingResult[]>;
-    initialize(): Promise<void>;
-    dispose(): Promise<void>;
-}
-/**
- * Neural embedding provider (legacy, for non-ONNX models)
- * Uses CodeBERT/UniXcoder for semantic embeddings
- *
- * @deprecated Use TransformersEmbeddingProvider for ONNX-ready models
- */
-export declare class NeuralEmbeddingProvider implements EmbeddingProvider {
-    readonly name: string;
-    readonly dims = 768;
-    private _ready;
-    private modelPath;
-    private model;
-    constructor(model: SupportedModel);
-    get ready(): boolean;
-    generateEmbedding(text: string): Promise<EmbeddingResult | null>;
-    generateBatch(items: Array<{
-        path: string;
-        content: string;
-    }>): Promise<BatchEmbeddingResult[]>;
-    initialize(): Promise<void>;
-    dispose(): Promise<void>;
-}
-/**
- * Transformers.js embedding provider
- * Uses @huggingface/transformers for ONNX-based inference
- */
-export declare class TransformersEmbeddingProvider implements EmbeddingProvider {
-    readonly name: string;
-    readonly dims: number;
-    private _ready;
-    private model;
-    private hfModelId;
-    private extractor;
-    constructor(model: SupportedModel);
-    get ready(): boolean;
-    generateEmbedding(text: string): Promise<EmbeddingResult | null>;
-    generateBatch(items: Array<{
-        path: string;
-        content: string;
-    }>): Promise<BatchEmbeddingResult[]>;
-    initialize(): Promise<void>;
-    dispose(): Promise<void>;
-}
-/**
- * Provider factory options
- */
-export interface ProviderOptions {
-    /** Force use of a specific provider type */
-    forceProvider?: "lsh" | "neural" | "transformers";
-    /** VSS configuration override */
-    config?: Partial<VSSConfig>;
-}
-/**
- * Create an embedding provider based on configuration
- */
-export declare function createEmbeddingProvider(options?: ProviderOptions): Promise<EmbeddingProvider>;
-/**
- * Get provider info string for logging
- */
-export declare function getProviderInfo(provider: EmbeddingProvider): string;
-//# sourceMappingURL=provider.d.ts.map

package/dist/src/vss/provider.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../../src/vss/provider.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAKH,OAAO,EAAiC,KAAK,cAAc,EAAE,KAAK,SAAS,EAAE,MAAM,aAAa,CAAC;AAEjG;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,2BAA2B;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,8BAA8B;IAC9B,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,uBAAuB;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,mDAAmD;IACnD,SAAS,EAAE,eAAe,GAAG,IAAI,CAAC;CACnC;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,oBAAoB;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,2BAA2B;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,2CAA2C;IAC3C,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;IAExB;;OAEG;IACH,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,CAAC;IAEjE;;OAEG;IACH,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAAC;IAEhG;;OAEG;IACH,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAE5B;;OAEG;IACH,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1B;AAED;;;GAGG;AACH,qBAAa,oBAAqB,YAAW,iBAAiB;IAC5D,QAAQ,CAAC,IAAI,SAAS;IACtB,QAAQ,CAAC,IAAI,MAAM;IACnB,QAAQ,CAAC,KAAK,QAAQ;IAEhB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;IAMhE,aAAa,CACjB,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAC9C,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAU5B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAG/B;AAED;;;;;GAKG;AACH,qBAAa,uBAAwB,YAAW,iBAAiB;IAC/D,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,OAAO;IACpB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,KAAK,CAAiB;gBAElB,KAAK,EAAE,cAAc;IAMjC,IAAI,KAAK,IAAI,OAAO,CAEnB;IAEK,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;IAWhE,aAAa,CACjB,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAC9C,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAa5B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAY3B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAG/B;AAKD;;;GAGG;AACH,qBAAa,6BAA8B,YAAW,iBAAiB;IACrE,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAiB;IAC9B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,SAAS,CACV;gBAEK,KAAK,EAAE,cAAc;IAajC,IAAI,KAAK,IAAI,OAAO,CAEnB;IAEK,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;IAiBhE,aAAa,CACjB,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAC9C,OAAO,CAAC,oBAAoB,EAAE,CAAC;IA2C5B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAkB3B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAI/B;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,4CAA4C;IAC5C,aAAa,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,cAAc,CAAC;IAClD,iCAAiC;IACjC,MAAM,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;CAC7B;AAED;;GAEG;AACH,wBAAsB,uBAAuB,CAC3C,OAAO,CAAC,EAAE,eAAe,GACxB,OAAO,CAAC,iBAAiB,CAAC,CAiE5B;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,iBAAiB,GAAG,MAAM,CAEnE"}