@mduenas/codegraph 0.4.0

package/dist/vectors/manager.js

@@ -0,0 +1,274 @@
+"use strict";
+/**
+ * Vector Manager
+ *
+ * High-level manager that coordinates embedding generation and vector search.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectorManager = void 0;
+exports.createVectorManager = createVectorManager;
+const embedder_1 = require("./embedder");
+const search_1 = require("./search");
+/**
+ * Default node kinds to embed
+ */
+const DEFAULT_NODE_KINDS = [
+    'function',
+    'method',
+    'class',
+    'interface',
+    'type_alias',
+    'module',
+    'component',
+];
+/**
+ * Vector Manager
+ *
+ * Provides high-level interface for semantic search:
+ * - Generates embeddings for code nodes
+ * - Stores embeddings in the database
+ * - Performs semantic similarity search
+ */
+class VectorManager {
+    embedder;
+    searchManager;
+    queries;
+    nodeKinds;
+    batchSize;
+    initialized = false;
+    constructor(db, queries, options = {}) {
+        this.embedder = (0, embedder_1.createEmbedder)(options.embedder);
+        this.searchManager = (0, search_1.createVectorSearch)(db, embedder_1.EMBEDDING_DIMENSION);
+        this.queries = queries;
+        this.nodeKinds = options.nodeKinds || DEFAULT_NODE_KINDS;
+        this.batchSize = options.batchSize || 32;
+    }
+    /**
+     * Initialize the vector manager
+     *
+     * Loads the embedding model and initializes vector search.
+     */
+    async initialize() {
+        if (this.initialized) {
+            return;
+        }
+        // Initialize embedder (downloads model if needed)
+        await this.embedder.initialize();
+        // Initialize vector search (loads sqlite-vss if available)
+        await this.searchManager.initialize();
+        this.initialized = true;
+    }
+    /**
+     * Check if the vector manager is initialized
+     */
+    isInitialized() {
+        return this.initialized;
+    }
+    /**
+     * Generate embeddings for all eligible nodes
+     *
+     * @param onProgress - Optional progress callback
+     * @returns Number of nodes embedded
+     */
+    async embedAllNodes(onProgress) {
+        if (!this.initialized) {
+            throw new Error('VectorManager not initialized. Call initialize() first.');
+        }
+        // Get all nodes that should be embedded
+        const nodesToEmbed = [];
+        for (const kind of this.nodeKinds) {
+            const nodes = this.queries.getNodesByKind(kind);
+            nodesToEmbed.push(...nodes);
+        }
+        // Filter out nodes that already have embeddings
+        const existingIds = new Set(this.searchManager.getIndexedNodeIds());
+        const newNodes = nodesToEmbed.filter((n) => !existingIds.has(n.id));
+        if (newNodes.length === 0) {
+            return 0;
+        }
+        // Process in batches
+        let processed = 0;
+        const model = this.embedder.getModelId();
+        for (let i = 0; i < newNodes.length; i += this.batchSize) {
+            const batch = newNodes.slice(i, i + this.batchSize);
+            // Create text representations
+            const texts = batch.map((node) => embedder_1.TextEmbedder.createNodeText(node));
+            // Generate embeddings
+            const result = await this.embedder.embedBatch(texts, 'document');
+            // Store embeddings
+            const entries = [];
+            for (let idx = 0; idx < batch.length; idx++) {
+                const node = batch[idx];
+                const embedding = result.embeddings[idx];
+                if (node && embedding) {
+                    entries.push({ nodeId: node.id, embedding });
+                }
+            }
+            this.searchManager.storeVectorBatch(entries, model);
+            processed += batch.length;
+            // Report progress
+            if (onProgress) {
+                onProgress({
+                    current: processed,
+                    total: newNodes.length,
+                    nodeName: batch[batch.length - 1]?.name,
+                });
+            }
+        }
+        return processed;
+    }
+    /**
+     * Generate embedding for a single node
+     *
+     * @param node - Node to embed
+     */
+    async embedNode(node) {
+        if (!this.initialized) {
+            throw new Error('VectorManager not initialized. Call initialize() first.');
+        }
+        const text = embedder_1.TextEmbedder.createNodeText(node);
+        const result = await this.embedder.embed(text);
+        this.searchManager.storeVector(node.id, result.embedding, result.model);
+    }
+    /**
+     * Semantic search for nodes matching a query
+     *
+     * @param query - Natural language query
+     * @param options - Search options
+     * @returns Array of search results with similarity scores
+     */
+    async search(query, options = {}) {
+        if (!this.initialized) {
+            throw new Error('VectorManager not initialized. Call initialize() first.');
+        }
+        const { limit = 10, kinds } = options;
+        // Generate query embedding
+        const queryResult = await this.embedder.embedQuery(query);
+        // Search for similar vectors
+        const vectorResults = this.searchManager.search(queryResult.embedding, {
+            limit: limit * 2, // Get more results to filter
+            minScore: 0.3, // Minimum similarity threshold
+        });
+        // Get nodes and filter by kind if specified
+        const results = [];
+        for (const vr of vectorResults) {
+            const node = this.queries.getNodeById(vr.nodeId);
+            if (!node) {
+                continue;
+            }
+            // Filter by node kind if specified
+            if (kinds && kinds.length > 0 && !kinds.includes(node.kind)) {
+                continue;
+            }
+            results.push({
+                node,
+                score: vr.score,
+            });
+            if (results.length >= limit) {
+                break;
+            }
+        }
+        return results;
+    }
+    /**
+     * Find nodes similar to a given node
+     *
+     * @param nodeId - ID of the node to find similar nodes for
+     * @param options - Search options
+     * @returns Array of similar nodes with similarity scores
+     */
+    async findSimilar(nodeId, options = {}) {
+        if (!this.initialized) {
+            throw new Error('VectorManager not initialized. Call initialize() first.');
+        }
+        const { limit = 10, kinds } = options;
+        // Get the node's embedding
+        let embedding = this.searchManager.getVector(nodeId);
+        // If no embedding exists, generate one
+        if (!embedding) {
+            const node = this.queries.getNodeById(nodeId);
+            if (!node) {
+                throw new Error(`Node not found: ${nodeId}`);
+            }
+            await this.embedNode(node);
+            embedding = this.searchManager.getVector(nodeId);
+            if (!embedding) {
+                throw new Error(`Failed to generate embedding for node: ${nodeId}`);
+            }
+        }
+        // Search for similar vectors (excluding the source node)
+        const vectorResults = this.searchManager.search(embedding, {
+            limit: limit + 1, // Get one extra to exclude the source
+            minScore: 0.3,
+        });
+        // Get nodes and filter
+        const results = [];
+        for (const vr of vectorResults) {
+            // Skip the source node
+            if (vr.nodeId === nodeId) {
+                continue;
+            }
+            const node = this.queries.getNodeById(vr.nodeId);
+            if (!node) {
+                continue;
+            }
+            // Filter by node kind if specified
+            if (kinds && kinds.length > 0 && !kinds.includes(node.kind)) {
+                continue;
+            }
+            results.push({
+                node,
+                score: vr.score,
+            });
+            if (results.length >= limit) {
+                break;
+            }
+        }
+        return results;
+    }
+    /**
+     * Delete embedding for a node
+     *
+     * @param nodeId - ID of the node
+     */
+    deleteNodeEmbedding(nodeId) {
+        this.searchManager.deleteVector(nodeId);
+    }
+    /**
+     * Get statistics about vector storage
+     */
+    getStats() {
+        return {
+            totalVectors: this.searchManager.getVectorCount(),
+            vssEnabled: this.searchManager.isVssEnabled(),
+            modelId: this.embedder.getModelId(),
+            dimension: this.embedder.getDimension(),
+        };
+    }
+    /**
+     * Clear all vectors
+     */
+    clear() {
+        this.searchManager.clear();
+    }
+    /**
+     * Rebuild the VSS index
+     */
+    rebuildIndex() {
+        this.searchManager.rebuildVssIndex();
+    }
+    /**
+     * Release resources
+     */
+    dispose() {
+        this.embedder.dispose();
+    }
+}
+exports.VectorManager = VectorManager;
+/**
+ * Create a vector manager
+ */
+function createVectorManager(db, queries, options) {
+    return new VectorManager(db, queries, options);
+}
+//# sourceMappingURL=manager.js.map

package/dist/vectors/manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/vectors/manager.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAgWH,kDAMC;AAlWD,yCAAgG;AAChG,qCAAmE;AA+BnE;;GAEG;AACH,MAAM,kBAAkB,GAAmB;IACzC,UAAU;IACV,QAAQ;IACR,OAAO;IACP,WAAW;IACX,YAAY;IACZ,QAAQ;IACR,WAAW;CACZ,CAAC;AAEF;;;;;;;GAOG;AACH,MAAa,aAAa;IAChB,QAAQ,CAAe;IACvB,aAAa,CAAsB;IACnC,OAAO,CAAe;IACtB,SAAS,CAAiB;IAC1B,SAAS,CAAS;IAClB,WAAW,GAAG,KAAK,CAAC;IAE5B,YACE,EAAqB,EACrB,OAAqB,EACrB,UAAgC,EAAE;QAElC,IAAI,CAAC,QAAQ,GAAG,IAAA,yBAAc,EAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC,aAAa,GAAG,IAAA,2BAAkB,EAAC,EAAE,EAAE,8BAAmB,CAAC,CAAC;QACjE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,kBAAkB,CAAC;QACzD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC;IAC3C,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,kDAAkD;QAClD,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;QAEjC,2DAA2D;QAC3D,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAEtC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CAAC,UAAkD;QACpE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QAED,wCAAwC;QACxC,MAAM,YAAY,GAAW,EAAE,CAAC;QAChC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAClC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAChD,YAAY,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QAC9B,CAAC;QAED,gDAAgD;QAChD,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,iBAAiB,EAAE,CAAC,CAAC;QACpE,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEpE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,CAAC;QACX,CAAC;QAED,qBAAqB;QACrB,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;QAEzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACzD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;YAEpD,8BAA8B;YAC9B,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,uBAAY,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;YAErE,sBAAsB;YACtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAEjE,mBAAmB;YACnB,MAAM,OAAO,GAAuD,EAAE,CAAC;YACvE,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;gBAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;gBACxB,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;gBACzC,IAAI,IAAI,IAAI,SAAS,EAAE,CAAC;oBACtB,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;YACD,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAEpD,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC;YAE1B,kBAAkB;YAClB,IAAI,UAAU,EAAE,CAAC;gBACf,UAAU,CAAC;oBACT,OAAO,EAAE,SAAS;oBAClB,KAAK,EAAE,QAAQ,CAAC,MAAM;oBACtB,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI;iBACxC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,SAAS,CAAC,IAAU;QACxB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,IAAI,GAAG,uBAAY,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1E,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,UAAyB,EAAE;QACrD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAEtC,2BAA2B;QAC3B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAE1D,6BAA6B;QAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE;YACrE,KAAK,EAAE,KAAK,GAAG,CAAC,EAAE,6BAA6B;YAC/C,QAAQ,EAAE,GAAG,EAAE,+BAA+B;SAC/C,CAAC,CAAC;QAEH,4CAA4C;QAC5C,MAAM,OAAO,GAAmB,EAAE,CAAC;QACnC,KAAK,MAAM,EAAE,IAAI,aAAa,EAAE,CAAC;YAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;YACjD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YAED,mCAAmC;YACnC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5D,SAAS;YACX,CAAC;YAED,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI;gBACJ,KAAK,EAAE,EAAE,CAAC,KAAK;aAChB,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,MAAM,IAAI,KAAK,EAAE,CAAC;gBAC5B,MAAM;YACR,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,UAAyB,EAAE;QAC3D,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAEtC,2BAA2B;QAC3B,IAAI,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAErD,uCAAuC;QACvC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,mBAAmB,MAAM,EAAE,CAAC,CAAC;YAC/C,CAAC;YAED,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC3B,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAEjD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,0CAA0C,MAAM,EAAE,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;QAED,yDAAyD;QACzD,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE;YACzD,KAAK,EAAE,KAAK,GAAG,CAAC,EAAE,sCAAsC;YACxD,QAAQ,EAAE,GAAG;SACd,CAAC,CAAC;QAEH,uBAAuB;QACvB,MAAM,OAAO,GAAmB,EAAE,CAAC;QACnC,KAAK,MAAM,EAAE,IAAI,aAAa,EAAE,CAAC;YAC/B,uBAAuB;YACvB,IAAI,EAAE,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;gBACzB,SAAS;YACX,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;YACjD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YAED,mCAAmC;YACnC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5D,SAAS;YACX,CAAC;YAED,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI;gBACJ,KAAK,EAAE,EAAE,CAAC,KAAK;aAChB,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,MAAM,IAAI,KAAK,EAAE,CAAC;gBAC5B,MAAM;YACR,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CAAC,MAAc;QAChC,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,QAAQ;QAMN,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,cAAc,EAAE;YACjD,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE;YAC7C,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE;YACnC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE;SACxC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;CACF;AAlSD,sCAkSC;AAED;;GAEG;AACH,SAAgB,mBAAmB,CACjC,EAAqB,EACrB,OAAqB,EACrB,OAA8B;IAE9B,OAAO,IAAI,aAAa,CAAC,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC"}

package/dist/vectors/search.d.ts

@@ -0,0 +1,134 @@
+/**
+ * Vector Search
+ *
+ * Provides vector similarity search using sqlite-vss extension.
+ * Falls back to brute-force cosine similarity if sqlite-vss is not available.
+ */
+import Database from 'better-sqlite3';
+import { Node } from '../types';
+/**
+ * Options for vector search
+ */
+export interface VectorSearchOptions {
+    /** Maximum number of results to return */
+    limit?: number;
+    /** Minimum similarity score (0-1) */
+    minScore?: number;
+    /** Node kinds to filter results */
+    nodeKinds?: Node['kind'][];
+}
+/**
+ * Vector Search Manager
+ *
+ * Handles vector storage and similarity search for semantic code search.
+ */
+export declare class VectorSearchManager {
+    private db;
+    private vssEnabled;
+    private embeddingDimension;
+    constructor(db: Database.Database, dimension?: number);
+    /**
+     * Initialize vector search
+     *
+     * Attempts to load sqlite-vss extension. Falls back to brute-force
+     * search if the extension is not available.
+     */
+    initialize(): Promise<void>;
+    /**
+     * Load the sqlite-vss extension
+     */
+    private loadVssExtension;
+    /**
+     * Create the VSS virtual table for vector search
+     */
+    private createVssTable;
+    /**
+     * Ensure the basic vectors table exists (for fallback mode)
+     */
+    private ensureVectorsTable;
+    /**
+     * Check if VSS extension is enabled
+     */
+    isVssEnabled(): boolean;
+    /**
+     * Store a vector embedding for a node
+     *
+     * @param nodeId - ID of the node
+     * @param embedding - Vector embedding
+     * @param model - Model used to generate embedding
+     */
+    storeVector(nodeId: string, embedding: Float32Array, model: string): void;
+    /**
+     * Store vector in VSS virtual table
+     */
+    private storeInVss;
+    /**
+     * Store multiple vectors in a batch
+     *
+     * @param entries - Array of node IDs and embeddings
+     * @param model - Model used to generate embeddings
+     */
+    storeVectorBatch(entries: Array<{
+        nodeId: string;
+        embedding: Float32Array;
+    }>, model: string): void;
+    /**
+     * Get vector for a node
+     *
+     * @param nodeId - ID of the node
+     * @returns Embedding or null if not found
+     */
+    getVector(nodeId: string): Float32Array | null;
+    /**
+     * Delete vector for a node
+     *
+     * @param nodeId - ID of the node
+     */
+    deleteVector(nodeId: string): void;
+    /**
+     * Search for similar vectors
+     *
+     * @param queryEmbedding - Query vector to search for
+     * @param options - Search options
+     * @returns Array of node IDs with similarity scores
+     */
+    search(queryEmbedding: Float32Array, options?: VectorSearchOptions): Array<{
+        nodeId: string;
+        score: number;
+    }>;
+    /**
+     * Search using sqlite-vss KNN search
+     */
+    private searchWithVss;
+    /**
+     * Brute-force search using cosine similarity
+     */
+    private searchBruteForce;
+    /**
+     * Get count of stored vectors
+     */
+    getVectorCount(): number;
+    /**
+     * Check if a node has a vector
+     */
+    hasVector(nodeId: string): boolean;
+    /**
+     * Get all node IDs that have vectors
+     */
+    getIndexedNodeIds(): string[];
+    /**
+     * Clear all vectors
+     */
+    clear(): void;
+    /**
+     * Rebuild VSS index from vectors table
+     *
+     * Useful after bulk operations or if VSS index gets out of sync.
+     */
+    rebuildVssIndex(): void;
+}
+/**
+ * Create a vector search manager
+ */
+export declare function createVectorSearch(db: Database.Database, dimension?: number): VectorSearchManager;
+//# sourceMappingURL=search.d.ts.map

package/dist/vectors/search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../src/vectors/search.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,IAAI,EAAE,MAAM,UAAU,CAAC;AAGhC;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,0CAA0C;IAC1C,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,qCAAqC;IACrC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,mCAAmC;IACnC,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;CAC5B;AAED;;;;GAIG;AACH,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,EAAE,CAAoB;IAC9B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,kBAAkB,CAAS;gBAEvB,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,SAAS,GAAE,MAA4B;IAK1E;;;;;OAKG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAsBjC;;OAEG;YACW,gBAAgB;IAkB9B;;OAEG;IACH,OAAO,CAAC,cAAc;IA8BtB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;OAEG;IACH,YAAY,IAAI,OAAO;IAIvB;;;;;;OAMG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAoBzE;;OAEG;IACH,OAAO,CAAC,UAAU;IAwClB;;;;;OAKG;IACH,gBAAgB,CACd,OAAO,EAAE,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,YAAY,CAAA;KAAE,CAAC,EAC3D,KAAK,EAAE,MAAM,GACZ,IAAI;IAuBP;;;;;OAKG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI;IAe9C;;;;OAIG;IACH,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAgBlC;;;;;;OAMG;IACH,MAAM,CACJ,cAAc,EAAE,YAAY,EAC5B,OAAO,GAAE,mBAAwB,GAChC,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAU3C;;OAEG;IACH,OAAO,CAAC,aAAa;IA4CrB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA+BxB;;OAEG;IACH,cAAc,IAAI,MAAM;IAOxB;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAOlC;;OAEG;IACH,iBAAiB,IAAI,MAAM,EAAE;IAO7B;;OAEG;IACH,KAAK,IAAI,IAAI;IASb;;;;OAIG;IACH,eAAe,IAAI,IAAI;CAwBxB;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,SAAS,CAAC,EAAE,MAAM,GACjB,mBAAmB,CAErB"}