kiri-mcp-server 0.16.0 → 0.17.0

package/dist/src/vss/provider.js

@@ -0,0 +1,254 @@
+/**
+ * VSS Embedding Provider Interface
+ *
+ * Defines the interface for generating semantic embeddings.
+ * Implementations can use different backends (LSH, Transformers.js, external API).
+ */
+import { generateEmbedding as generateLSHEmbedding } from "../shared/embedding.js";
+import { getModelPath, isModelInstalled } from "../shared/utils/global-dirs.js";
+import { loadVSSConfig, MODEL_METADATA } from "./config.js";
+/**
+ * LSH-based embedding provider (current implementation)
+ * Uses locality-sensitive hashing for structural similarity
+ */
+export class LSHEmbeddingProvider {
+    name = "lsh";
+    dims = 64;
+    ready = true;
+    async generateEmbedding(text) {
+        const result = generateLSHEmbedding(text);
+        if (!result)
+            return null;
+        return { dims: result.dims, values: result.values };
+    }
+    async generateBatch(items) {
+        return items.map((item) => {
+            const result = generateLSHEmbedding(item.content);
+            return {
+                path: item.path,
+                embedding: result ? { dims: 64, values: result.values } : null,
+            };
+        });
+    }
+    async initialize() {
+        // No initialization needed for LSH
+    }
+    async dispose() {
+        // No cleanup needed for LSH
+    }
+}
+/**
+ * Neural embedding provider (legacy, for non-ONNX models)
+ * Uses CodeBERT/UniXcoder for semantic embeddings
+ *
+ * @deprecated Use TransformersEmbeddingProvider for ONNX-ready models
+ */
+export class NeuralEmbeddingProvider {
+    name;
+    dims = 768; // CodeBERT/UniXcoder dimension
+    _ready = false;
+    modelPath;
+    model;
+    constructor(model) {
+        this.name = `neural:${model}`;
+        this.model = model;
+        this.modelPath = getModelPath(model);
+    }
+    get ready() {
+        return this._ready;
+    }
+    async generateEmbedding(text) {
+        if (!this._ready) {
+            throw new Error(`Neural provider not initialized. Call initialize() first.`);
+        }
+        // Legacy models require manual ONNX conversion, fall back to LSH
+        console.warn(`[VSS] Legacy model "${this.model}" has no ONNX, using LSH fallback`);
+        const lsh = new LSHEmbeddingProvider();
+        return lsh.generateEmbedding(text);
+    }
+    async generateBatch(items) {
+        if (!this._ready) {
+            throw new Error(`Neural provider not initialized. Call initialize() first.`);
+        }
+        const results = [];
+        for (const item of items) {
+            const embedding = await this.generateEmbedding(item.content);
+            results.push({ path: item.path, embedding });
+        }
+        return results;
+    }
+    async initialize() {
+        if (!isModelInstalled(this.model)) {
+            throw new Error(`Model "${this.model}" is not installed. Run "kiri model download ${this.model}" first.`);
+        }
+        console.info(`[VSS] Neural embedding provider initialized (model: ${this.model})`);
+        console.warn(`[VSS] Legacy model has no ONNX, will use LSH fallback`);
+        this._ready = true;
+    }
+    async dispose() {
+        this._ready = false;
+    }
+}
+// Lazy-loaded transformers pipeline to avoid startup cost
+let transformersPipeline = null;
+/**
+ * Transformers.js embedding provider
+ * Uses @huggingface/transformers for ONNX-based inference
+ */
+export class TransformersEmbeddingProvider {
+    name;
+    dims;
+    _ready = false;
+    model;
+    hfModelId;
+    extractor = null;
+    constructor(model) {
+        const metadata = MODEL_METADATA[model];
+        if (!metadata.hasOnnx) {
+            throw new Error(`Model "${model}" does not have ONNX support. Use NeuralEmbeddingProvider or choose an ONNX-ready model.`);
+        }
+        this.name = `transformers:${model}`;
+        this.model = model;
+        this.hfModelId = metadata.hfModelId;
+        this.dims = metadata.dims;
+    }
+    get ready() {
+        return this._ready;
+    }
+    async generateEmbedding(text) {
+        if (!this._ready || !this.extractor) {
+            throw new Error(`Transformers provider not initialized. Call initialize() first.`);
+        }
+        try {
+            const output = await this.extractor(text, { pooling: "mean", normalize: true });
+            // Feature extraction returns a Tensor with data property
+            const tensor = output;
+            const values = Array.from(tensor.data);
+            return { dims: this.dims, values };
+        }
+        catch (error) {
+            console.error(`[VSS] Embedding generation failed:`, error);
+            return null;
+        }
+    }
+    async generateBatch(items) {
+        if (!this._ready || !this.extractor) {
+            throw new Error(`Transformers provider not initialized. Call initialize() first.`);
+        }
+        const results = [];
+        // Process in batches to avoid memory issues
+        const batchSize = 8;
+        for (let i = 0; i < items.length; i += batchSize) {
+            const batch = items.slice(i, i + batchSize);
+            const texts = batch.map((item) => item.content);
+            try {
+                const subBatchStart = Date.now();
+                const output = await this.extractor(texts, { pooling: "mean", normalize: true });
+                const elapsed = ((Date.now() - subBatchStart) / 1000).toFixed(1);
+                console.info(`[VSS] Sub-batch ${Math.floor(i / batchSize) + 1}/${Math.ceil(items.length / batchSize)}: ${batch.length} texts in ${elapsed}s`);
+                // Feature extraction returns a Tensor with data property
+                const tensor = output;
+                // Output shape: [batch_size, dims]
+                for (let j = 0; j < batch.length; j++) {
+                    const start = j * this.dims;
+                    const end = start + this.dims;
+                    const values = Array.from(tensor.data.slice(start, end));
+                    results.push({
+                        path: batch[j].path,
+                        embedding: { dims: this.dims, values },
+                    });
+                }
+            }
+            catch (error) {
+                console.error(`[VSS] Batch embedding failed:`, error);
+                // Add null results for failed batch
+                for (const item of batch) {
+                    results.push({ path: item.path, embedding: null });
+                }
+            }
+        }
+        return results;
+    }
+    async initialize() {
+        console.info(`[VSS] Loading model: ${this.hfModelId}`);
+        // Lazy load transformers to reduce startup time
+        if (!transformersPipeline) {
+            const transformers = await import("@huggingface/transformers");
+            transformersPipeline = transformers.pipeline;
+        }
+        // Create feature extraction pipeline
+        this.extractor = await transformersPipeline("feature-extraction", this.hfModelId, {
+            dtype: "fp32", // Use fp32 for better compatibility
+        });
+        console.info(`[VSS] Transformers embedding provider ready (model: ${this.model})`);
+        this._ready = true;
+    }
+    async dispose() {
+        this.extractor = null;
+        this._ready = false;
+    }
+}
+/**
+ * Create an embedding provider based on configuration
+ */
+export async function createEmbeddingProvider(options) {
+    const config = loadVSSConfig(options?.config);
+    // Force LSH provider if specified
+    if (options?.forceProvider === "lsh") {
+        return new LSHEmbeddingProvider();
+    }
+    // If VSS is disabled, use LSH
+    if (!config.enabled) {
+        return new LSHEmbeddingProvider();
+    }
+    const metadata = MODEL_METADATA[config.model];
+    // Force transformers provider if specified
+    if (options?.forceProvider === "transformers") {
+        if (!metadata.hasOnnx) {
+            throw new Error(`Model "${config.model}" does not have ONNX support. Choose an ONNX-ready model.`);
+        }
+        const provider = new TransformersEmbeddingProvider(config.model);
+        await provider.initialize();
+        return provider;
+    }
+    // Force neural provider if specified (legacy)
+    if (options?.forceProvider === "neural") {
+        const provider = new NeuralEmbeddingProvider(config.model);
+        await provider.initialize();
+        return provider;
+    }
+    // Auto-select provider based on model capabilities
+    if (metadata.hasOnnx) {
+        // Use TransformersEmbeddingProvider for ONNX-ready models
+        // Models are auto-downloaded by @huggingface/transformers
+        const provider = new TransformersEmbeddingProvider(config.model);
+        await provider.initialize();
+        return provider;
+    }
+    // Legacy models: check if manually installed
+    if (!isModelInstalled(config.model)) {
+        switch (config.onMissing) {
+            case "error":
+                throw new Error(`VSS model "${config.model}" not installed. Run "kiri model download ${config.model}" first.`);
+            case "skip":
+                console.info(`[VSS] Model not installed, using LSH fallback`);
+                return new LSHEmbeddingProvider();
+            case "prompt":
+            case "download":
+                // These should be handled by the CLI before reaching here
+                console.warn(`[VSS] Model not installed, using LSH fallback`);
+                return new LSHEmbeddingProvider();
+        }
+    }
+    // Use legacy neural provider for non-ONNX models
+    const provider = new NeuralEmbeddingProvider(config.model);
+    await provider.initialize();
+    return provider;
+}
+/**
+ * Get provider info string for logging
+ */
+export function getProviderInfo(provider) {
+    return `${provider.name} (dims=${provider.dims}, ready=${provider.ready})`;
+}
+//# sourceMappingURL=provider.js.map

package/dist/src/vss/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../src/vss/provider.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,iBAAiB,IAAI,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AACnF,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAEhF,OAAO,EAAE,aAAa,EAAE,cAAc,EAAuC,MAAM,aAAa,CAAC;AAsDjG;;;GAGG;AACH,MAAM,OAAO,oBAAoB;IACtB,IAAI,GAAG,KAAK,CAAC;IACb,IAAI,GAAG,EAAE,CAAC;IACV,KAAK,GAAG,IAAI,CAAC;IAEtB,KAAK,CAAC,iBAAiB,CAAC,IAAY;QAClC,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QACzB,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,KAA+C;QAE/C,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClD,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI;aAC/D,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU;QACd,mCAAmC;IACrC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,4BAA4B;IAC9B,CAAC;CACF;AAED;;;;;GAKG;AACH,MAAM,OAAO,uBAAuB;IACzB,IAAI,CAAS;IACb,IAAI,GAAG,GAAG,CAAC,CAAC,+BAA+B;IAC5C,MAAM,GAAG,KAAK,CAAC;IACf,SAAS,CAAS;IAClB,KAAK,CAAiB;IAE9B,YAAY,KAAqB;QAC/B,IAAI,CAAC,IAAI,GAAG,UAAU,KAAK,EAAE,CAAC;QAC9B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,IAAY;QAClC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;QAED,iEAAiE;QACjE,OAAO,CAAC,IAAI,CAAC,uBAAuB,IAAI,CAAC,KAAK,mCAAmC,CAAC,CAAC;QACnF,MAAM,GAAG,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACvC,OAAO,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,KAA+C;QAE/C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7D,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CACb,UAAU,IAAI,CAAC,KAAK,gDAAgD,IAAI,CAAC,KAAK,UAAU,CACzF,CAAC;QACJ,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,uDAAuD,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;QACnF,OAAO,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAC;QACtE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;CACF;AAED,0DAA0D;AAC1D,IAAI,oBAAoB,GAA+D,IAAI,CAAC;AAE5F;;;GAGG;AACH,MAAM,OAAO,6BAA6B;IAC/B,IAAI,CAAS;IACb,IAAI,CAAS;IACd,MAAM,GAAG,KAAK,CAAC;IACf,KAAK,CAAiB;IACtB,SAAS,CAAS;IAClB,SAAS,GACf,IAAI,CAAC;IAEP,YAAY,KAAqB;QAC/B,MAAM,QAAQ,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,UAAU,KAAK,0FAA0F,CAC1G,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,IAAI,GAAG,gBAAgB,KAAK,EAAE,CAAC;QACpC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;QACpC,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IAC5B,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,IAAY;QAClC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QACrF,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAChF,yDAAyD;YACzD,MAAM,MAAM,GAAG,MAAgC,CAAC;YAChD,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACvC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;QACrC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC3D,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,KAA+C;QAE/C,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,4CAA4C;QAC5C,MAAM,SAAS,GAAG,CAAC,CAAC;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACjD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEhD,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACjF,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACjE,OAAO,CAAC,IAAI,CAAC,mBAAmB,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC,KAAK,KAAK,CAAC,MAAM,aAAa,OAAO,GAAG,CAAC,CAAC;gBAC9I,yDAAyD;gBACzD,MAAM,MAAM,GAAG,MAAgC,CAAC;gBAEhD,mCAAmC;gBACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACtC,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC;oBAC5B,MAAM,GAAG,GAAG,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC;oBAC9B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;oBACzD,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,KAAK,CAAC,CAAC,CAAE,CAAC,IAAI;wBACpB,SAAS,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE;qBACvC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;gBACtD,oCAAoC;gBACpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,CAAC,IAAI,CAAC,wBAAwB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAEvD,gDAAgD;QAChD,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC1B,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YAC/D,oBAAoB,GAAG,YAAY,CAAC,QAAQ,CAAC;QAC/C,CAAC;QAED,qCAAqC;QACrC,IAAI,CAAC,SAAS,GAAG,MAAM,oBAAoB,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,EAAE;YAChF,KAAK,EAAE,MAAM,EAAE,oCAAoC;SACpD,CAAC,CAAC;QAEH,OAAO,CAAC,IAAI,CAAC,uDAAuD,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;QACnF,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;CACF;AAYD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,OAAyB;IAEzB,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAE9C,kCAAkC;IAClC,IAAI,OAAO,EAAE,aAAa,KAAK,KAAK,EAAE,CAAC;QACrC,OAAO,IAAI,oBAAoB,EAAE,CAAC;IACpC,CAAC;IAED,8BAA8B;IAC9B,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,IAAI,oBAAoB,EAAE,CAAC;IACpC,CAAC;IAED,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAE9C,2CAA2C;IAC3C,IAAI,OAAO,EAAE,aAAa,KAAK,cAAc,EAAE,CAAC;QAC9C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,UAAU,MAAM,CAAC,KAAK,2DAA2D,CAClF,CAAC;QACJ,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,6BAA6B,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACjE,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,8CAA8C;IAC9C,IAAI,OAAO,EAAE,aAAa,KAAK,QAAQ,EAAE,CAAC;QACxC,MAAM,QAAQ,GAAG,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,mDAAmD;IACnD,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;QACrB,0DAA0D;QAC1D,0DAA0D;QAC1D,MAAM,QAAQ,GAAG,IAAI,6BAA6B,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACjE,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,6CAA6C;IAC7C,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACpC,QAAQ,MAAM,CAAC,SAAS,EAAE,CAAC;YACzB,KAAK,OAAO;gBACV,MAAM,IAAI,KAAK,CACb,cAAc,MAAM,CAAC,KAAK,6CAA6C,MAAM,CAAC,KAAK,UAAU,CAC9F,CAAC;YACJ,KAAK,MAAM;gBACT,OAAO,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;gBAC9D,OAAO,IAAI,oBAAoB,EAAE,CAAC;YACpC,KAAK,QAAQ,CAAC;YACd,KAAK,UAAU;gBACb,0DAA0D;gBAC1D,OAAO,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;gBAC9D,OAAO,IAAI,oBAAoB,EAAE,CAAC;QACtC,CAAC;IACH,CAAC;IAED,iDAAiD;IACjD,MAAM,QAAQ,GAAG,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC3D,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;IAC5B,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,QAA2B;IACzD,OAAO,GAAG,QAAQ,CAAC,IAAI,UAAU,QAAQ,CAAC,IAAI,WAAW,QAAQ,CAAC,KAAK,GAAG,CAAC;AAC7E,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kiri-mcp-server",
-  "version": "0.16.0",
+  "version": "0.17.0",
   "description": "KIRI context extraction platform",
   "type": "module",
   "packageManager": "pnpm@9.0.0",

package/dist/client/cli.js

@@ -1,68 +0,0 @@
-#!/usr/bin/env node
-import process from "node:process";
-import { bootstrapServer } from "../server/bootstrap.js";
-import { evaluateSecurityStatus, updateSecurityLock } from "../shared/security/config.js";
-function printUsage() {
-    console.info(`Usage: pnpm exec tsx src/client/cli.ts <command> [options]\n`);
-    console.info(`Commands:`);
-    console.info(`  security verify [--write-lock]    Verify security baseline matches lock file`);
-}
-function formatStatus() {
-    const status = evaluateSecurityStatus();
-    const lockInfo = status.lockHash ? `hash=${status.lockHash}` : "missing";
-    const matchState = status.matches ? "MATCH" : "MISMATCH";
-    return [
-        `config: ${status.configPath}`,
-        `lock: ${status.lockPath} (${lockInfo})`,
-        `state: ${matchState}`,
-    ].join("\n");
-}
-function handleSecurityVerify(argv) {
-    const writeLock = argv.includes("--write-lock");
-    const status = evaluateSecurityStatus();
-    if (!status.lockHash && writeLock) {
-        updateSecurityLock(status.hash);
-        console.info("Security lock created.");
-        const refreshed = evaluateSecurityStatus();
-        console.info([
-            `config: ${refreshed.configPath}`,
-            `lock: ${refreshed.lockPath} (hash=${refreshed.lockHash})`,
-            "state: MATCH",
-        ].join("\n"));
-        return 0;
-    }
-    try {
-        bootstrapServer({ allowWriteLock: writeLock });
-        console.info("Security baseline verified.");
-        console.info(formatStatus());
-        return 0;
-    }
-    catch (error) {
-        console.error(error instanceof Error ? error.message : String(error));
-        console.info(formatStatus());
-        return 1;
-    }
-}
-export function main(argv = process.argv.slice(2)) {
-    const [command, ...rest] = argv;
-    switch (command) {
-        case "security": {
-            const [subcommand, ...subArgs] = rest;
-            if (subcommand === "verify") {
-                return handleSecurityVerify(subArgs);
-            }
-            break;
-        }
-        case undefined:
-            printUsage();
-            return 1;
-        default:
-            break;
-    }
-    printUsage();
-    return 1;
-}
-if (process.argv[1] && new URL(import.meta.url).pathname === process.argv[1]) {
-    process.exitCode = main();
-}
-//# sourceMappingURL=cli.js.map

package/dist/client/cli.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../src/client/cli.ts"],"names":[],"mappings":";AACA,OAAO,OAAO,MAAM,cAAc,CAAC;AAEnC,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAE1F,SAAS,UAAU;IACjB,OAAO,CAAC,IAAI,CAAC,8DAA8D,CAAC,CAAC;IAC7E,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC1B,OAAO,CAAC,IAAI,CAAC,gFAAgF,CAAC,CAAC;AACjG,CAAC;AAED,SAAS,YAAY;IACnB,MAAM,MAAM,GAAG,sBAAsB,EAAE,CAAC;IACxC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACzE,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC;IACzD,OAAO;QACL,WAAW,MAAM,CAAC,UAAU,EAAE;QAC9B,SAAS,MAAM,CAAC,QAAQ,KAAK,QAAQ,GAAG;QACxC,UAAU,UAAU,EAAE;KACvB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAc;IAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,sBAAsB,EAAE,CAAC;IACxC,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,SAAS,EAAE,CAAC;QAClC,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACvC,MAAM,SAAS,GAAG,sBAAsB,EAAE,CAAC;QAC3C,OAAO,CAAC,IAAI,CACV;YACE,WAAW,SAAS,CAAC,UAAU,EAAE;YACjC,SAAS,SAAS,CAAC,QAAQ,UAAU,SAAS,CAAC,QAAQ,GAAG;YAC1D,cAAc;SACf,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;QACF,OAAO,CAAC,CAAC;IACX,CAAC;IACD,IAAI,CAAC;QACH,eAAe,CAAC,EAAE,cAAc,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/C,OAAO,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAC5C,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QAC7B,OAAO,CAAC,CAAC;IACX,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACtE,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QAC7B,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IAC/C,MAAM,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;IAChC,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,CAAC,UAAU,EAAE,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC;YACtC,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC5B,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;YACvC,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,SAAS;YACZ,UAAU,EAAE,CAAC;YACb,OAAO,CAAC,CAAC;QACX;YACE,MAAM;IACV,CAAC;IACD,UAAU,EAAE,CAAC;IACb,OAAO,CAAC,CAAC;AACX,CAAC;AAED,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IAC7E,OAAO,CAAC,QAAQ,GAAG,IAAI,EAAE,CAAC;AAC5B,CAAC"}

package/dist/client/index.js

@@ -1,5 +0,0 @@
-export function buildContextBundleRequest(snippets) {
-    // TODO: align with MCP client schema
-    return JSON.stringify({ snippets }, null, 2);
-}
-//# sourceMappingURL=index.js.map

package/dist/client/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,yBAAyB,CACvC,QAAkD;IAElD,qCAAqC;IACrC,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC/C,CAAC"}

package/dist/eval/metrics.js

@@ -1,47 +0,0 @@
-export function precisionAtK(retrievedIds, relevantIds, k) {
-    if (k <= 0 || retrievedIds.length === 0) {
-        return 0;
-    }
-    const relevantSet = new Set(relevantIds);
-    if (relevantSet.size === 0) {
-        return 0;
-    }
-    const limit = Math.min(k, retrievedIds.length);
-    let hits = 0;
-    for (let index = 0; index < limit; index += 1) {
-        const id = retrievedIds[index];
-        if (id !== undefined && relevantSet.has(id)) {
-            hits += 1;
-        }
-    }
-    return hits / limit;
-}
-export function timeToFirstUseful(events, options = {}) {
-    if (events.length === 0) {
-        return Number.POSITIVE_INFINITY;
-    }
-    const sorted = [...events].sort((a, b) => a.timestampMs - b.timestampMs);
-    const baseline = typeof options.startTimestampMs === "number"
-        ? options.startTimestampMs
-        : (sorted[0]?.timestampMs ?? 0);
-    for (const event of sorted) {
-        if (event.relevant) {
-            const deltaMs = event.timestampMs - baseline;
-            return Math.max(0, deltaMs) / 1000;
-        }
-    }
-    return Number.POSITIVE_INFINITY;
-}
-export function evaluateRetrieval(options) {
-    const { items, relevant, k } = options;
-    const ids = items.map((item) => item.id);
-    const precision = precisionAtK(ids, relevant, k);
-    const relevantSet = new Set(relevant);
-    const latencyEvents = items.map((item) => ({
-        timestampMs: item.timestampMs,
-        relevant: relevantSet.has(item.id),
-    }));
-    const ttff = timeToFirstUseful(latencyEvents);
-    return { precisionAtK: precision, timeToFirstUseful: ttff };
-}
-//# sourceMappingURL=metrics.js.map

package/dist/eval/metrics.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"metrics.js","sourceRoot":"","sources":["../../src/eval/metrics.ts"],"names":[],"mappings":"AAKA,MAAM,UAAU,YAAY,CAC1B,YAAsB,EACtB,WAA6B,EAC7B,CAAS;IAET,IAAI,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxC,OAAO,CAAC,CAAC;IACX,CAAC;IACD,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC;IACzC,IAAI,WAAW,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,CAAC,CAAC;IACX,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;IAC/C,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC9C,MAAM,EAAE,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QAC/B,IAAI,EAAE,KAAK,SAAS,IAAI,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,CAAC;QACZ,CAAC;IACH,CAAC;IACD,OAAO,IAAI,GAAG,KAAK,CAAC;AACtB,CAAC;AAOD,MAAM,UAAU,iBAAiB,CAC/B,MAAsB,EACtB,UAAyC,EAAE;IAE3C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,MAAM,CAAC,iBAAiB,CAAC;IAClC,CAAC;IACD,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC;IACzE,MAAM,QAAQ,GACZ,OAAO,OAAO,CAAC,gBAAgB,KAAK,QAAQ;QAC1C,CAAC,CAAC,OAAO,CAAC,gBAAgB;QAC1B,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,WAAW,IAAI,CAAC,CAAC,CAAC;IACpC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,GAAG,QAAQ,CAAC;YAC7C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;QACrC,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,iBAAiB,CAAC;AAClC,CAAC;AAaD,MAAM,UAAU,iBAAiB,CAAC,OAAiC;IACjE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzC,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;IACjD,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;IACtC,MAAM,aAAa,GAAmB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACzD,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,QAAQ,EAAE,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;KACnC,CAAC,CAAC,CAAC;IACJ,MAAM,IAAI,GAAG,iBAAiB,CAAC,aAAa,CAAC,CAAC;IAC9C,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;AAC9D,CAAC"}