@thoughtflow/core 0.1.2

package/dist/storage/vector/memory-vector-store.js

@@ -0,0 +1,101 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MemoryVectorStore = void 0;
+class MemoryVectorStore {
+    chunks = new Map();
+    initialized = false;
+    async initialize() {
+        this.initialized = true;
+    }
+    async insert(chunks) {
+        if (!this.initialized)
+            await this.initialize();
+        for (const chunk of chunks) {
+            this.chunks.set(chunk.id, chunk);
+        }
+    }
+    async delete(filter) {
+        let count = 0;
+        const toDelete = [];
+        for (const [id, chunk] of this.chunks) {
+            if (this.matchFilter(chunk.metadata, filter)) {
+                toDelete.push(id);
+                count++;
+            }
+        }
+        for (const id of toDelete) {
+            this.chunks.delete(id);
+        }
+        return count;
+    }
+    async search(queryVector, options) {
+        const results = [];
+        const limit = options?.limit ?? 10;
+        const offset = options?.offset ?? 0;
+        const minScore = options?.minScore ?? 0;
+        for (const chunk of this.chunks.values()) {
+            if (options?.filter && !this.matchFilter(chunk.metadata, options.filter)) {
+                continue;
+            }
+            const score = this.cosineSimilarity(queryVector, chunk.vector);
+            if (score >= minScore) {
+                results.push({ item: chunk, score });
+            }
+        }
+        results.sort((a, b) => b.score - a.score);
+        return results.slice(offset, offset + limit);
+    }
+    async stats() {
+        return {
+            totalVectors: this.chunks.size,
+            collections: ['default'],
+            indexStatus: 'ready',
+        };
+    }
+    async clear() {
+        this.chunks.clear();
+    }
+    async dispose() {
+        this.chunks.clear();
+    }
+    cosineSimilarity(a, b) {
+        if (a.length !== b.length)
+            return 0;
+        let dot = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < a.length; i++) {
+            dot += a[i] * b[i];
+            normA += a[i] * a[i];
+            normB += b[i] * b[i];
+        }
+        const denominator = Math.sqrt(normA) * Math.sqrt(normB);
+        if (denominator === 0)
+            return 0;
+        return dot / denominator;
+    }
+    matchFilter(metadata, filter) {
+        for (const [key, value] of Object.entries(filter)) {
+            if (value === undefined || value === null)
+                continue;
+            const metadataValue = metadata[key];
+            if (Array.isArray(value)) {
+                const metaArr = metadataValue;
+                if (!metaArr || !value.every((v) => metaArr.includes(v))) {
+                    return false;
+                }
+            }
+            else if (typeof value === 'string') {
+                if (metadataValue !== value)
+                    return false;
+            }
+            else {
+                if (metadataValue !== value)
+                    return false;
+            }
+        }
+        return true;
+    }
+}
+exports.MemoryVectorStore = MemoryVectorStore;
+//# sourceMappingURL=memory-vector-store.js.map

package/dist/storage/vector/memory-vector-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-vector-store.js","sourceRoot":"","sources":["../../../src/storage/vector/memory-vector-store.ts"],"names":[],"mappings":";;;AAEA,MAAa,iBAAiB;IACpB,MAAM,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC7C,WAAW,GAAG,KAAK,CAAC;IAE5B,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAqB;QAChC,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC/C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAmD;QAC9D,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACtC,IAAI,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC7C,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClB,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,WAAqB,EAAE,OAA6B;QAC/D,MAAM,OAAO,GAAyB,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,CAAC,CAAC;QAExC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,OAAO,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzE,SAAS;YACX,CAAC;YACD,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;YAC/D,IAAI,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACtB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAC1C,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YAC9B,WAAW,EAAE,CAAC,SAAS,CAAC;YACxB,WAAW,EAAE,OAAO;SACrB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAEO,gBAAgB,CAAC,CAAW,EAAE,CAAW;QAC/C,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC;QACpC,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,WAAW,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAChC,OAAO,GAAG,GAAG,WAAW,CAAC;IAC3B,CAAC;IAEO,WAAW,CAAC,QAAuB,EAAE,MAA+B;QAC1E,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAClD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI;gBAAE,SAAS;YACpD,MAAM,aAAa,GAAG,QAAQ,CAAC,GAA0B,CAAC,CAAC;YAC3D,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,MAAM,OAAO,GAAG,aAAqC,CAAC;gBACtD,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAW,CAAC,CAAC,EAAE,CAAC;oBAC5E,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;iBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACrC,IAAI,aAAa,KAAK,KAAK;oBAAE,OAAO,KAAK,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACN,IAAI,aAAa,KAAK,KAAK;oBAAE,OAAO,KAAK,CAAC;YAC5C,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAlGD,8CAkGC"}

package/dist/storage/vector/mock-embedding.d.ts

@@ -0,0 +1,10 @@
+import type { IEmbeddingProvider } from "../../contracts/vector-store.types";
+export declare class MockEmbeddingProvider implements IEmbeddingProvider {
+    readonly dimensions = 16;
+    readonly maxBatchSize = 100;
+    readonly modelName = "mock";
+    embed(texts: string[]): Promise<number[][]>;
+    embedQuery(text: string): Promise<number[]>;
+    private hashToVector;
+}
+//# sourceMappingURL=mock-embedding.d.ts.map

package/dist/storage/vector/mock-embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-embedding.d.ts","sourceRoot":"","sources":["../../../src/storage/vector/mock-embedding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAE7E,qBAAa,qBAAsB,YAAW,kBAAkB;IAC9D,QAAQ,CAAC,UAAU,MAAM;IACzB,QAAQ,CAAC,YAAY,OAAO;IAC5B,QAAQ,CAAC,SAAS,UAAU;IAEtB,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAI3C,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAIjD,OAAO,CAAC,YAAY;CAoBrB"}

package/dist/storage/vector/mock-embedding.js

@@ -0,0 +1,36 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MockEmbeddingProvider = void 0;
+class MockEmbeddingProvider {
+    dimensions = 16;
+    maxBatchSize = 100;
+    modelName = 'mock';
+    async embed(texts) {
+        return texts.map((text) => this.hashToVector(text, this.dimensions));
+    }
+    async embedQuery(text) {
+        return this.hashToVector(text, this.dimensions);
+    }
+    hashToVector(text, dims) {
+        const vector = new Array(dims);
+        let h = 0;
+        for (let i = 0; i < text.length; i++) {
+            h = ((h << 5) - h + text.charCodeAt(i)) | 0;
+        }
+        const absH = Math.abs(h);
+        for (let i = 0; i < dims; i++) {
+            const seed = absH + i * 2654435761;
+            const val = Math.sin(seed) * 10000;
+            vector[i] = val - Math.floor(val);
+        }
+        const norm = Math.sqrt(vector.reduce((sum, v) => sum + v * v, 0));
+        if (norm > 0) {
+            for (let i = 0; i < dims; i++) {
+                vector[i] /= norm;
+            }
+        }
+        return vector;
+    }
+}
+exports.MockEmbeddingProvider = MockEmbeddingProvider;
+//# sourceMappingURL=mock-embedding.js.map

package/dist/storage/vector/mock-embedding.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-embedding.js","sourceRoot":"","sources":["../../../src/storage/vector/mock-embedding.ts"],"names":[],"mappings":";;;AAEA,MAAa,qBAAqB;IACvB,UAAU,GAAG,EAAE,CAAC;IAChB,YAAY,GAAG,GAAG,CAAC;IACnB,SAAS,GAAG,MAAM,CAAC;IAE5B,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IAClD,CAAC;IAEO,YAAY,CAAC,IAAY,EAAE,IAAY;QAC7C,MAAM,MAAM,GAAG,IAAI,KAAK,CAAS,IAAI,CAAC,CAAC;QACvC,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,GAAG,UAAU,CAAC;YACnC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;YACnC,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAClE,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;YACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;YACpB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAjCD,sDAiCC"}

package/dist/storage/vector/ollama-embedding.d.ts

@@ -0,0 +1,17 @@
+import type { IEmbeddingProvider } from "../../contracts/vector-store.types";
+export interface OllamaEmbeddingConfig {
+    baseUrl?: string;
+    model?: string;
+    dimensions?: number;
+    maxBatchSize?: number;
+}
+export declare class OllamaEmbeddingProvider implements IEmbeddingProvider {
+    readonly dimensions: number;
+    readonly modelName: string;
+    readonly maxBatchSize: number;
+    private readonly baseUrl;
+    constructor(config?: OllamaEmbeddingConfig);
+    embed(texts: string[]): Promise<number[][]>;
+    embedQuery(text: string): Promise<number[]>;
+}
+//# sourceMappingURL=ollama-embedding.d.ts.map

package/dist/storage/vector/ollama-embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama-embedding.d.ts","sourceRoot":"","sources":["../../../src/storage/vector/ollama-embedding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAE7E,MAAM,WAAW,qBAAqB;IACpC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,uBAAwB,YAAW,kBAAkB;IAChE,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAErB,MAAM,GAAE,qBAA0B;IAOxC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAkB3C,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;CAIlD"}

package/dist/storage/vector/ollama-embedding.js

@@ -0,0 +1,38 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OllamaEmbeddingProvider = void 0;
+class OllamaEmbeddingProvider {
+    dimensions;
+    modelName;
+    maxBatchSize;
+    baseUrl;
+    constructor(config = {}) {
+        this.baseUrl = config.baseUrl ?? 'http://localhost:11434';
+        this.modelName = config.model ?? 'nomic-embed-text';
+        this.dimensions = config.dimensions ?? 768;
+        this.maxBatchSize = config.maxBatchSize ?? 32;
+    }
+    async embed(texts) {
+        const allEmbeddings = [];
+        for (let i = 0; i < texts.length; i += this.maxBatchSize) {
+            const batch = texts.slice(i, i + this.maxBatchSize);
+            const response = await fetch(`${this.baseUrl}/api/embed`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({ model: this.modelName, input: batch }),
+            });
+            if (!response.ok) {
+                throw new Error(`Ollama embed failed: ${response.status} ${response.statusText}`);
+            }
+            const data = await response.json();
+            allEmbeddings.push(...data.embeddings);
+        }
+        return allEmbeddings;
+    }
+    async embedQuery(text) {
+        const results = await this.embed([text]);
+        return results[0];
+    }
+}
+exports.OllamaEmbeddingProvider = OllamaEmbeddingProvider;
+//# sourceMappingURL=ollama-embedding.js.map

package/dist/storage/vector/ollama-embedding.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama-embedding.js","sourceRoot":"","sources":["../../../src/storage/vector/ollama-embedding.ts"],"names":[],"mappings":";;;AASA,MAAa,uBAAuB;IACzB,UAAU,CAAS;IACnB,SAAS,CAAS;IAClB,YAAY,CAAS;IACb,OAAO,CAAS;IAEjC,YAAY,SAAgC,EAAE;QAC5C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC;QAC1D,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,KAAK,IAAI,kBAAkB,CAAC;QACpD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,GAAG,CAAC;QAC3C,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,MAAM,aAAa,GAAe,EAAE,CAAC;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACzD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC;YACpD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,YAAY,EAAE;gBACxD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;aAC9D,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACpF,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAgC,CAAC;YACjE,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACzC,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;CACF;AAnCD,0DAmCC"}

package/dist/storage/vector/transformers-embedding.d.ts

@@ -0,0 +1,23 @@
+import type { IEmbeddingProvider } from "../../contracts/vector-store.types";
+export interface TransformersEmbeddingConfig {
+    model?: string;
+    dtype?: 'fp32' | 'q8' | 'q4';
+    maxBatchSize?: number;
+    normalize?: boolean;
+    pooling?: 'mean' | 'cls';
+}
+export declare class TransformersEmbeddingProvider implements IEmbeddingProvider {
+    private extractor;
+    readonly dimensions = 384;
+    readonly modelName: string;
+    readonly maxBatchSize: number;
+    private readonly dtype;
+    private readonly normalize;
+    private readonly pooling;
+    constructor(config?: TransformersEmbeddingConfig);
+    initialize(): Promise<void>;
+    embed(texts: string[]): Promise<number[][]>;
+    embedQuery(text: string): Promise<number[]>;
+    dispose(): Promise<void>;
+}
+//# sourceMappingURL=transformers-embedding.d.ts.map

package/dist/storage/vector/transformers-embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transformers-embedding.d.ts","sourceRoot":"","sources":["../../../src/storage/vector/transformers-embedding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAE7E,MAAM,WAAW,2BAA2B;IAC1C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;CAC1B;AAED,qBAAa,6BAA8B,YAAW,kBAAkB;IACtE,OAAO,CAAC,SAAS,CAAiB;IAElC,QAAQ,CAAC,UAAU,OAAO;IAC1B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAU;IACpC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAErB,MAAM,GAAE,2BAAgC;IAQ9C,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAS3B,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAiB3C,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAK3C,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAG/B"}

package/dist/storage/vector/transformers-embedding.js

@@ -0,0 +1,80 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TransformersEmbeddingProvider = void 0;
+class TransformersEmbeddingProvider {
+    extractor = null;
+    dimensions = 384;
+    modelName;
+    maxBatchSize;
+    dtype;
+    normalize;
+    pooling;
+    constructor(config = {}) {
+        this.modelName = config.model ?? 'Xenova/all-MiniLM-L6-v2';
+        this.maxBatchSize = config.maxBatchSize ?? 32;
+        this.dtype = config.dtype ?? 'q8';
+        this.normalize = config.normalize ?? true;
+        this.pooling = config.pooling ?? 'mean';
+    }
+    async initialize() {
+        const { pipeline } = await Promise.resolve().then(() => __importStar(require('@huggingface/transformers')));
+        this.extractor = await pipeline('feature-extraction', this.modelName, { dtype: this.dtype });
+    }
+    async embed(texts) {
+        if (!this.extractor)
+            await this.initialize();
+        const allEmbeddings = [];
+        for (let i = 0; i < texts.length; i += this.maxBatchSize) {
+            const batch = texts.slice(i, i + this.maxBatchSize);
+            const result = await this.extractor(batch, {
+                pooling: this.pooling,
+                normalize: this.normalize,
+            });
+            const batchEmbeddings = result.tolist();
+            allEmbeddings.push(...batchEmbeddings);
+        }
+        return allEmbeddings;
+    }
+    async embedQuery(text) {
+        const results = await this.embed([text]);
+        return results[0];
+    }
+    async dispose() {
+        this.extractor = null;
+    }
+}
+exports.TransformersEmbeddingProvider = TransformersEmbeddingProvider;
+//# sourceMappingURL=transformers-embedding.js.map

package/dist/storage/vector/transformers-embedding.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transformers-embedding.js","sourceRoot":"","sources":["../../../src/storage/vector/transformers-embedding.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAUA,MAAa,6BAA6B;IAChC,SAAS,GAAY,IAAI,CAAC;IAEzB,UAAU,GAAG,GAAG,CAAC;IACjB,SAAS,CAAS;IAClB,YAAY,CAAS;IACb,KAAK,CAAS;IACd,SAAS,CAAU;IACnB,OAAO,CAAS;IAEjC,YAAY,SAAsC,EAAE;QAClD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,KAAK,IAAI,yBAAyB,CAAC;QAC3D,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC;QAClC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC;QAC1C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,EAAE,QAAQ,EAAE,GAAG,wDAAa,2BAA2B,GAAC,CAAC;QAC/D,IAAI,CAAC,SAAS,GAAG,MAAM,QAAQ,CAC7B,oBAAoB,EACpB,IAAI,CAAC,SAAS,EACd,EAAE,KAAK,EAAE,IAAI,CAAC,KAAY,EAAE,CAC7B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,IAAI,CAAC,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAE7C,MAAM,aAAa,GAAe,EAAE,CAAC;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACzD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC;YACpD,MAAM,MAAM,GAAQ,MAAO,IAAI,CAAC,SAAiB,CAAC,KAAK,EAAE;gBACvD,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAC,CAAC;YACH,MAAM,eAAe,GAAe,MAAM,CAAC,MAAM,EAAE,CAAC;YACpD,aAAa,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;QACzC,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACzC,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IACxB,CAAC;CACF;AApDD,sEAoDC"}

package/dist/tools/base/path-utils.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Resolve a user-provided path against rootDir.
+ *
+ * Supports:
+ * 1. Relative paths (recommended): e.g. "backend/src/ads/ads.controller.ts"
+ * 2. Absolute paths: e.g. "/Users/.../ads-demo/backend/src/ads/ads.controller.ts"
+ * 3. Bare filename fallback: if the path doesn't exist as given, searches for the
+ *    filename recursively under rootDir and returns the first match.
+ *
+ * Returns null on path traversal (security).
+ */
+export declare function resolveSafePath(rootDir: string, relPath: string): string | null;
+//# sourceMappingURL=path-utils.d.ts.map

package/dist/tools/base/path-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"path-utils.d.ts","sourceRoot":"","sources":["../../../src/tools/base/path-utils.ts"],"names":[],"mappings":"AAGA;;;;;;;;;;GAUG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAwB/E"}

package/dist/tools/base/path-utils.js

@@ -0,0 +1,96 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveSafePath = resolveSafePath;
+const path = __importStar(require("path"));
+const fs = __importStar(require("fs"));
+/**
+ * Resolve a user-provided path against rootDir.
+ *
+ * Supports:
+ * 1. Relative paths (recommended): e.g. "backend/src/ads/ads.controller.ts"
+ * 2. Absolute paths: e.g. "/Users/.../ads-demo/backend/src/ads/ads.controller.ts"
+ * 3. Bare filename fallback: if the path doesn't exist as given, searches for the
+ *    filename recursively under rootDir and returns the first match.
+ *
+ * Returns null on path traversal (security).
+ */
+function resolveSafePath(rootDir, relPath) {
+    const root = path.resolve(rootDir);
+    const fullPath = path.resolve(root, relPath);
+    // Security: block path traversal
+    if (!fullPath.startsWith(root + path.sep) && fullPath !== root) {
+        return null;
+    }
+    // If the file exists at the exact path, return it
+    if (fs.existsSync(fullPath)) {
+        return fullPath;
+    }
+    // Fallback: search for the filename anywhere under rootDir
+    // (handles cases where model omits subdirectory prefix like "backend/")
+    const basename = path.basename(relPath);
+    if (basename && basename !== ".") {
+        const found = findFileRecursive(root, basename);
+        if (found)
+            return found;
+    }
+    // File not found — return the intended path so caller can show a helpful error
+    return fullPath;
+}
+/**
+ * Recursively search for a filename under dir.
+ * Returns the first match or null.
+ */
+function findFileRecursive(dir, filename) {
+    try {
+        const entries = fs.readdirSync(dir, { withFileTypes: true });
+        for (const entry of entries) {
+            const full = path.join(dir, entry.name);
+            if (entry.isFile() && entry.name === filename) {
+                return full;
+            }
+            if (entry.isDirectory() && !entry.name.startsWith(".") && entry.name !== "node_modules") {
+                const found = findFileRecursive(full, filename);
+                if (found)
+                    return found;
+            }
+        }
+    }
+    catch {
+        // permission denied or other error — skip
+    }
+    return null;
+}
+//# sourceMappingURL=path-utils.js.map

package/dist/tools/base/path-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"path-utils.js","sourceRoot":"","sources":["../../../src/tools/base/path-utils.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAcA,0CAwBC;AAtCD,2CAA6B;AAC7B,uCAAyB;AAEzB;;;;;;;;;;GAUG;AACH,SAAgB,eAAe,CAAC,OAAe,EAAE,OAAe;IAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAE7C,iCAAiC;IACjC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;QAC/D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kDAAkD;IAClD,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,2DAA2D;IAC3D,wEAAwE;IACxE,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,QAAQ,IAAI,QAAQ,KAAK,GAAG,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAChD,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC;IAC1B,CAAC;IAED,+EAA+E;IAC/E,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;GAGG;AACH,SAAS,iBAAiB,CAAC,GAAW,EAAE,QAAgB;IACtD,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACxC,IAAI,KAAK,CAAC,MAAM,EAAE,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC9C,OAAO,IAAI,CAAC;YACd,CAAC;YACD,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gBACxF,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAChD,IAAI,KAAK;oBAAE,OAAO,KAAK,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,0CAA0C;IAC5C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/tools/base/pii-guard.middleware.d.ts

@@ -0,0 +1,45 @@
+import { ToolExecutionResult, ToolMiddleware } from "../../contracts";
+export type PiiStrategy = "redact" | "mask" | "hash" | "block";
+export interface PiiRule {
+    name: string;
+    pattern: RegExp;
+}
+export interface PiiAuditEntry {
+    phase: "before" | "after";
+    toolName: string;
+    strategy: PiiStrategy;
+    findings: Array<{
+        rule: string;
+        value: string;
+    }>;
+}
+export interface PiiGuardOptions {
+    defaultStrategy?: PiiStrategy;
+    rules?: PiiRule[];
+    toolStrategy?: Record<string, PiiStrategy>;
+    onAudit?: (entry: PiiAuditEntry) => void;
+}
+export declare class PiiGuardMiddleware implements ToolMiddleware {
+    private readonly options;
+    readonly name = "pii_guard";
+    private readonly defaultStrategy;
+    private readonly rules;
+    private readonly toolStrategy;
+    constructor(options?: PiiGuardOptions);
+    before(call: {
+        toolName: string;
+        input: Record<string, unknown>;
+    }): Promise<void>;
+    after(call: {
+        toolName: string;
+        input: Record<string, unknown>;
+    }, result: ToolExecutionResult): Promise<void>;
+    private resolveStrategy;
+    private emitAudit;
+    private transformUnknown;
+    private transformString;
+    private applyStrategy;
+    private cloneRegExp;
+    private replaceObject;
+}
+//# sourceMappingURL=pii-guard.middleware.d.ts.map

package/dist/tools/base/pii-guard.middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pii-guard.middleware.d.ts","sourceRoot":"","sources":["../../../src/tools/base/pii-guard.middleware.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEtE,MAAM,MAAM,WAAW,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;AAE/D,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,WAAW,CAAC;IACtB,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAClD;AAED,MAAM,WAAW,eAAe;IAC9B,eAAe,CAAC,EAAE,WAAW,CAAC;IAC9B,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC3C,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,aAAa,KAAK,IAAI,CAAC;CAC1C;AAWD,qBAAa,kBAAmB,YAAW,cAAc;IAO3C,OAAO,CAAC,QAAQ,CAAC,OAAO;IANpC,QAAQ,CAAC,IAAI,eAAe;IAE5B,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAc;IAC9C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAY;IAClC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA8B;gBAE9B,OAAO,GAAE,eAAoB;IAMpD,MAAM,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAejF,KAAK,CACT,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAE,EAC1D,MAAM,EAAE,mBAAmB,GAC1B,OAAO,CAAC,IAAI,CAAC;IAuBhB,OAAO,CAAC,eAAe;IAIvB,OAAO,CAAC,SAAS;IAsBjB,OAAO,CAAC,gBAAgB;IAgCxB,OAAO,CAAC,eAAe;IAkBvB,OAAO,CAAC,aAAa;IAsBrB,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,aAAa;CAStB"}

package/dist/tools/base/pii-guard.middleware.js

@@ -0,0 +1,140 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PiiGuardMiddleware = void 0;
+const crypto_1 = require("crypto");
+const DEFAULT_RULES = [
+    { name: "email", pattern: /\b[A-Z0-9._%+-]+@[A-Z0-9.-]+\.[A-Z]{2,}\b/gi },
+    { name: "phone", pattern: /\b(?:\+?\d[\d\s\-()]{7,}\d)\b/g },
+    { name: "credit_card", pattern: /\b(?:\d[ -]*?){13,19}\b/g },
+    { name: "pesel", pattern: /\b\d{11}\b/g },
+    { name: "openai_key", pattern: /\bsk-[A-Za-z0-9]{20,}\b/g },
+    { name: "github_token", pattern: /\bghp_[A-Za-z0-9]{30,}\b/g },
+];
+class PiiGuardMiddleware {
+    options;
+    name = "pii_guard";
+    defaultStrategy;
+    rules;
+    toolStrategy;
+    constructor(options = {}) {
+        this.options = options;
+        this.defaultStrategy = options.defaultStrategy ?? "redact";
+        this.rules = options.rules ?? DEFAULT_RULES;
+        this.toolStrategy = options.toolStrategy ?? {};
+    }
+    async before(call) {
+        const strategy = this.resolveStrategy(call.toolName);
+        const transformed = this.transformUnknown(call.input, strategy);
+        this.emitAudit("before", call.toolName, strategy, transformed.findings);
+        if (strategy === "block" && transformed.findings.length > 0) {
+            throw new Error(`PII guard blocked tool '${call.toolName}' due to sensitive input.`);
+        }
+        if (strategy !== "block") {
+            this.replaceObject(call.input, transformed.value);
+        }
+    }
+    async after(call, result) {
+        const strategy = this.resolveStrategy(call.toolName);
+        const outputTransformed = this.transformUnknown(result.output, strategy);
+        const dataTransformed = this.transformUnknown(result.data, strategy);
+        const findings = [...outputTransformed.findings, ...dataTransformed.findings];
+        this.emitAudit("after", call.toolName, strategy, findings);
+        if (strategy === "block" && findings.length > 0) {
+            result.success = false;
+            result.error = `PII guard blocked tool '${call.toolName}' output.`;
+            result.output = undefined;
+            result.data = undefined;
+            return;
+        }
+        if (strategy !== "block") {
+            result.output = outputTransformed.value;
+            result.data = dataTransformed.value;
+        }
+    }
+    resolveStrategy(toolName) {
+        return this.toolStrategy[toolName] ?? this.defaultStrategy;
+    }
+    emitAudit(phase, toolName, strategy, findings) {
+        if (!this.options.onAudit) {
+            return;
+        }
+        if (findings.length === 0) {
+            return;
+        }
+        this.options.onAudit({
+            phase,
+            toolName,
+            strategy,
+            findings,
+        });
+    }
+    transformUnknown(value, strategy) {
+        if (typeof value === "string") {
+            return this.transformString(value, strategy);
+        }
+        if (Array.isArray(value)) {
+            const findings = [];
+            const output = value.map((entry) => {
+                const transformed = this.transformUnknown(entry, strategy);
+                findings.push(...transformed.findings);
+                return transformed.value;
+            });
+            return { value: output, findings };
+        }
+        if (value && typeof value === "object") {
+            const findings = [];
+            const out = {};
+            for (const [key, entry] of Object.entries(value)) {
+                const transformed = this.transformUnknown(entry, strategy);
+                findings.push(...transformed.findings);
+                out[key] = transformed.value;
+            }
+            return { value: out, findings };
+        }
+        return { value, findings: [] };
+    }
+    transformString(input, strategy) {
+        const findings = [];
+        let output = input;
+        for (const rule of this.rules) {
+            const regexp = this.cloneRegExp(rule.pattern);
+            output = output.replace(regexp, (match) => {
+                findings.push({ rule: rule.name, value: match });
+                return this.applyStrategy(match, rule.name, strategy);
+            });
+        }
+        return { value: output, findings };
+    }
+    applyStrategy(value, ruleName, strategy) {
+        if (strategy === "redact") {
+            return `[REDACTED:${ruleName}]`;
+        }
+        if (strategy === "mask") {
+            if (value.length <= 4) {
+                return "****";
+            }
+            const head = value.slice(0, 2);
+            const tail = value.slice(-2);
+            return `${head}****${tail}`;
+        }
+        if (strategy === "hash") {
+            const hash = (0, crypto_1.createHash)("sha256").update(value).digest("hex");
+            return `[HASH:${hash.slice(0, 12)}]`;
+        }
+        return value;
+    }
+    cloneRegExp(pattern) {
+        const flags = pattern.flags.includes("g") ? pattern.flags : `${pattern.flags}g`;
+        return new RegExp(pattern.source, flags);
+    }
+    replaceObject(target, source) {
+        for (const key of Object.keys(target)) {
+            delete target[key];
+        }
+        for (const [key, value] of Object.entries(source)) {
+            target[key] = value;
+        }
+    }
+}
+exports.PiiGuardMiddleware = PiiGuardMiddleware;
+//# sourceMappingURL=pii-guard.middleware.js.map