@stupidloud/codegraph 0.7.7

package/dist/vectors/embedder.js

@@ -0,0 +1,237 @@
+"use strict";
+/**
+ * Text Embedder
+ *
+ * Generates vector embeddings through the Gemini API. Vector storage and
+ * similarity search remain local.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TextEmbedder = exports.EMBEDDING_DIMENSION = exports.DEFAULT_MODEL = void 0;
+exports.createEmbedder = createEmbedder;
+/**
+ * Default Gemini embedding model.
+ */
+exports.DEFAULT_MODEL = 'gemini-embedding-2';
+exports.EMBEDDING_DIMENSION = 768;
+const GEMINI_EMBEDDING_ENDPOINT = 'https://generativelanguage.googleapis.com/v1beta/models';
+const MAX_REQUESTS_PER_MINUTE = 100;
+const MIN_REQUEST_INTERVAL_MS = Math.ceil(60_000 / MAX_REQUESTS_PER_MINUTE);
+/**
+ * Text Embedder using Gemini.
+ *
+ * Batch generation uses batchEmbedContents with one request per node text. Do
+ * not combine multiple nodes into one content.parts array because that produces
+ * a single aggregate embedding instead of node-level vectors.
+ */
+class TextEmbedder {
+    apiKey;
+    modelId;
+    dimension;
+    initialized = false;
+    lastRequestAt = 0;
+    constructor(options = {}) {
+        this.apiKey = options.apiKey;
+        this.modelId = options.modelId || exports.DEFAULT_MODEL;
+        this.dimension = options.outputDimensionality || exports.EMBEDDING_DIMENSION;
+    }
+    /**
+     * Initialize the embedder by validating configuration.
+     */
+    async initialize() {
+        if (this.initialized) {
+            return;
+        }
+        if (!this.apiKey) {
+            throw new Error('Gemini API key is required for semantic search');
+        }
+        this.initialized = true;
+    }
+    /**
+     * Check if the embedder is initialized
+     */
+    isInitialized() {
+        return this.initialized;
+    }
+    /**
+     * Get the model ID being used
+     */
+    getModelId() {
+        return this.modelId;
+    }
+    /**
+     * Get the embedding dimension
+     */
+    getDimension() {
+        return this.dimension;
+    }
+    /**
+     * Generate embedding for a single text
+     */
+    async embed(text) {
+        const result = await this.embedBatch([text], 'document');
+        const embedding = result.embeddings[0];
+        if (!embedding) {
+            throw new Error('Gemini embedding response did not include values');
+        }
+        return {
+            embedding,
+            dimension: embedding.length,
+            model: result.model,
+        };
+    }
+    /**
+     * Generate embedding for a query.
+     */
+    async embedQuery(query) {
+        const result = await this.embedBatch([query], 'search_query');
+        const embedding = result.embeddings[0];
+        if (!embedding) {
+            throw new Error('Gemini embedding response did not include values');
+        }
+        return {
+            embedding,
+            dimension: embedding.length,
+            model: result.model,
+        };
+    }
+    /**
+     * Generate embeddings for multiple texts in node order.
+     */
+    async embedBatch(texts, type = 'document') {
+        if (!this.initialized) {
+            throw new Error('Embedder not initialized. Call initialize() first.');
+        }
+        if (texts.length === 0) {
+            return {
+                embeddings: [],
+                dimension: this.dimension,
+                model: this.modelId,
+                durationMs: 0,
+            };
+        }
+        const startTime = Date.now();
+        const embeddings = await this.embedPreparedBatch(texts.map((text) => this.prepareText(text)), type);
+        return {
+            embeddings,
+            dimension: embeddings[0]?.length ?? this.dimension,
+            model: this.modelId,
+            durationMs: Date.now() - startTime,
+        };
+    }
+    async embedPreparedBatch(texts, type) {
+        if (!this.initialized) {
+            throw new Error('Embedder not initialized. Call initialize() first.');
+        }
+        const url = `${GEMINI_EMBEDDING_ENDPOINT}/${encodeURIComponent(this.modelId)}:batchEmbedContents`;
+        const model = `models/${this.modelId}`;
+        const taskType = type === 'search_query' ? 'CODE_RETRIEVAL_QUERY' : 'RETRIEVAL_DOCUMENT';
+        await this.throttleRequests();
+        const response = await fetch(url, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'x-goog-api-key': this.apiKey,
+            },
+            body: JSON.stringify({
+                requests: texts.map((text) => ({
+                    model,
+                    content: {
+                        parts: [{ text }],
+                    },
+                    taskType,
+                    outputDimensionality: this.dimension,
+                })),
+            }),
+        });
+        if (!response.ok) {
+            const body = await response.text().catch(() => '');
+            throw new Error(`Gemini embedding request failed (${response.status}): ${body || response.statusText}`);
+        }
+        const json = await response.json();
+        const values = json.embeddings?.map((embedding) => embedding.values);
+        if (!values || values.length !== texts.length) {
+            throw new Error('Gemini batch embedding response count did not match request count');
+        }
+        const embeddings = values.map((embeddingValues) => {
+            if (!embeddingValues || embeddingValues.length === 0) {
+                throw new Error('Gemini embedding response did not include values');
+            }
+            return new Float32Array(embeddingValues);
+        });
+        if (embeddings.some((embedding) => embedding.length !== embeddings[0].length)) {
+            throw new Error('Gemini batch embedding response contained mixed dimensions');
+        }
+        return embeddings;
+    }
+    async throttleRequests() {
+        const now = Date.now();
+        const elapsed = now - this.lastRequestAt;
+        if (this.lastRequestAt > 0 && elapsed < MIN_REQUEST_INTERVAL_MS) {
+            await new Promise((resolve) => setTimeout(resolve, MIN_REQUEST_INTERVAL_MS - elapsed));
+        }
+        this.lastRequestAt = Date.now();
+    }
+    /**
+     * Prepare text for retrieval-oriented embedding.
+     */
+    prepareText(text) {
+        const maxLength = 8192;
+        return text.length > maxLength ? text.slice(0, maxLength) : text;
+    }
+    /**
+     * Create text representation of a code node for embedding.
+     */
+    static createNodeText(node) {
+        const parts = [];
+        parts.push(`${node.kind}: ${node.name}`);
+        if (node.qualifiedName && node.qualifiedName !== node.name) {
+            parts.push(`path: ${node.qualifiedName}`);
+        }
+        parts.push(`file: ${node.filePath}`);
+        if (node.signature) {
+            parts.push(`signature: ${node.signature}`);
+        }
+        if (node.docstring) {
+            parts.push(`documentation: ${node.docstring}`);
+        }
+        return parts.join('\n');
+    }
+    /**
+     * Compute cosine similarity between two embeddings
+     */
+    static cosineSimilarity(a, b) {
+        if (a.length !== b.length) {
+            throw new Error('Embeddings must have the same dimension');
+        }
+        let dotProduct = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < a.length; i++) {
+            const aVal = a[i];
+            const bVal = b[i];
+            dotProduct += aVal * bVal;
+            normA += aVal * aVal;
+            normB += bVal * bVal;
+        }
+        normA = Math.sqrt(normA);
+        normB = Math.sqrt(normB);
+        if (normA === 0 || normB === 0) {
+            return 0;
+        }
+        return dotProduct / (normA * normB);
+    }
+    /**
+     * Release resources
+     */
+    dispose() {
+        this.initialized = false;
+    }
+}
+exports.TextEmbedder = TextEmbedder;
+/**
+ * Create a text embedder instance
+ */
+function createEmbedder(options) {
+    return new TextEmbedder(options);
+}
+//# sourceMappingURL=embedder.js.map

package/dist/vectors/embedder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedder.js","sourceRoot":"","sources":["../../src/vectors/embedder.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AA0UH,wCAEC;AA1UD;;GAEG;AACU,QAAA,aAAa,GAAG,oBAAoB,CAAC;AACrC,QAAA,mBAAmB,GAAG,GAAG,CAAC;AACvC,MAAM,yBAAyB,GAAG,yDAAyD,CAAC;AAC5F,MAAM,uBAAuB,GAAG,GAAG,CAAC;AACpC,MAAM,uBAAuB,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,uBAAuB,CAAC,CAAC;AAsD5E;;;;;;GAMG;AACH,MAAa,YAAY;IACf,MAAM,CAAU;IAChB,OAAO,CAAS;IAChB,SAAS,CAAS;IAClB,WAAW,GAAG,KAAK,CAAC;IACpB,aAAa,GAAG,CAAC,CAAC;IAE1B,YAAY,UAA2B,EAAE;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,qBAAa,CAAC;QAChD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,oBAAoB,IAAI,2BAAmB,CAAC;IACvE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QACD,OAAO;YACL,SAAS;YACT,SAAS,EAAE,SAAS,CAAC,MAAM;YAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,KAAa;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,EAAE,cAAc,CAAC,CAAC;QAC9D,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QACD,OAAO;YACL,SAAS;YACT,SAAS,EAAE,SAAS,CAAC,MAAM;YAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CACd,KAAe,EACf,OAAoC,UAAU;QAE9C,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO;gBACL,UAAU,EAAE,EAAE;gBACd,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,KAAK,EAAE,IAAI,CAAC,OAAO;gBACnB,UAAU,EAAE,CAAC;aACd,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAC9C,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,EAC3C,IAAI,CACL,CAAC;QAEF,OAAO;YACL,UAAU;YACV,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,IAAI,CAAC,SAAS;YAClD,KAAK,EAAE,IAAI,CAAC,OAAO;YACnB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SACnC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,KAAe,EACf,IAAiC;QAEjC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;QACxE,CAAC;QAED,MAAM,GAAG,GAAG,GAAG,yBAAyB,IAAI,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC;QAClG,MAAM,KAAK,GAAG,UAAU,IAAI,CAAC,OAAO,EAAE,CAAC;QACvC,MAAM,QAAQ,GAAG,IAAI,KAAK,cAAc,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;QACzF,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,gBAAgB,EAAE,IAAI,CAAC,MAAO;aAC/B;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC7B,KAAK;oBACL,OAAO,EAAE;wBACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC;qBAClB;oBACD,QAAQ;oBACR,oBAAoB,EAAE,IAAI,CAAC,SAAS;iBACrC,CAAC,CAAC;aACJ,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,oCAAoC,QAAQ,CAAC,MAAM,MAAM,IAAI,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1G,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA6B,CAAC;QAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACrE,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;QACvF,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE;YAChD,IAAI,CAAC,eAAe,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrD,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;YACtE,CAAC;YACD,OAAO,IAAI,YAAY,CAAC,eAAe,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAE,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/E,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAChF,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC;QACzC,IAAI,IAAI,CAAC,aAAa,GAAG,CAAC,IAAI,OAAO,GAAG,uBAAuB,EAAE,CAAC;YAChE,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,uBAAuB,GAAG,OAAO,CAAC,CAAC,CAAC;QACzF,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAClC,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,IAAY;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC;QACvB,OAAO,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACnE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,cAAc,CAAC,IAOrB;QACC,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAEzC,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;YAC3D,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QAC5C,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAErC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACjD,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,gBAAgB,CAAC,CAAe,EAAE,CAAe;QACtD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC;YACnB,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC;YACnB,UAAU,IAAI,IAAI,GAAG,IAAI,CAAC;YAC1B,KAAK,IAAI,IAAI,GAAG,IAAI,CAAC;YACrB,KAAK,IAAI,IAAI,GAAG,IAAI,CAAC;QACvB,CAAC;QAED,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEzB,IAAI,KAAK,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,UAAU,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC3B,CAAC;CACF;AA/PD,oCA+PC;AAED;;GAEG;AACH,SAAgB,cAAc,CAAC,OAAyB;IACtD,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC"}

package/dist/vectors/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Vectors Module
+ *
+ * Provides text embedding and vector similarity search for semantic code search.
+ */
+export { TextEmbedder, createEmbedder, DEFAULT_MODEL, EMBEDDING_DIMENSION, EmbedderOptions, EmbeddingResult, BatchEmbeddingResult, } from './embedder';
+export { VectorSearchManager, createVectorSearch, VectorSearchOptions, } from './search';
+export { VectorManager, createVectorManager, VectorManagerOptions, EmbeddingProgress, } from './manager';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vectors/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,YAAY,EACZ,cAAc,EACd,aAAa,EACb,mBAAmB,EACnB,eAAe,EACf,eAAe,EACf,oBAAoB,GACrB,MAAM,YAAY,CAAC;AAEpB,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,aAAa,EACb,mBAAmB,EACnB,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,WAAW,CAAC"}

package/dist/vectors/index.js

@@ -0,0 +1,20 @@
+"use strict";
+/**
+ * Vectors Module
+ *
+ * Provides text embedding and vector similarity search for semantic code search.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createVectorManager = exports.VectorManager = exports.createVectorSearch = exports.VectorSearchManager = exports.EMBEDDING_DIMENSION = exports.DEFAULT_MODEL = exports.createEmbedder = exports.TextEmbedder = void 0;
+var embedder_1 = require("./embedder");
+Object.defineProperty(exports, "TextEmbedder", { enumerable: true, get: function () { return embedder_1.TextEmbedder; } });
+Object.defineProperty(exports, "createEmbedder", { enumerable: true, get: function () { return embedder_1.createEmbedder; } });
+Object.defineProperty(exports, "DEFAULT_MODEL", { enumerable: true, get: function () { return embedder_1.DEFAULT_MODEL; } });
+Object.defineProperty(exports, "EMBEDDING_DIMENSION", { enumerable: true, get: function () { return embedder_1.EMBEDDING_DIMENSION; } });
+var search_1 = require("./search");
+Object.defineProperty(exports, "VectorSearchManager", { enumerable: true, get: function () { return search_1.VectorSearchManager; } });
+Object.defineProperty(exports, "createVectorSearch", { enumerable: true, get: function () { return search_1.createVectorSearch; } });
+var manager_1 = require("./manager");
+Object.defineProperty(exports, "VectorManager", { enumerable: true, get: function () { return manager_1.VectorManager; } });
+Object.defineProperty(exports, "createVectorManager", { enumerable: true, get: function () { return manager_1.createVectorManager; } });
+//# sourceMappingURL=index.js.map

package/dist/vectors/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vectors/index.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,uCAQoB;AAPlB,wGAAA,YAAY,OAAA;AACZ,0GAAA,cAAc,OAAA;AACd,yGAAA,aAAa,OAAA;AACb,+GAAA,mBAAmB,OAAA;AAMrB,mCAIkB;AAHhB,6GAAA,mBAAmB,OAAA;AACnB,4GAAA,kBAAkB,OAAA;AAIpB,qCAKmB;AAJjB,wGAAA,aAAa,OAAA;AACb,8GAAA,mBAAmB,OAAA"}

package/dist/vectors/manager.d.ts

@@ -0,0 +1,124 @@
+/**
+ * Vector Manager
+ *
+ * High-level manager that coordinates embedding generation and vector search.
+ */
+import { SqliteDatabase } from '../db/sqlite-adapter';
+import { Node, SearchResult, SearchOptions } from '../types';
+import { EmbedderOptions } from './embedder';
+import { QueryBuilder } from '../db/queries';
+/**
+ * Progress callback for embedding generation
+ */
+export interface EmbeddingProgress {
+    /** Current node index */
+    current: number;
+    /** Total nodes to embed */
+    total: number;
+    /** Current node being embedded */
+    nodeName?: string;
+}
+/**
+ * Options for the vector manager
+ */
+export interface VectorManagerOptions {
+    /** Embedder options */
+    embedder?: EmbedderOptions;
+    /** Node kinds to embed (default: functions, methods, classes, interfaces) */
+    nodeKinds?: Node['kind'][];
+    /** Batch size for embedding generation */
+    batchSize?: number;
+}
+/**
+ * Vector Manager
+ *
+ * Provides high-level interface for semantic search:
+ * - Generates embeddings for code nodes
+ * - Stores embeddings in the database
+ * - Performs semantic similarity search
+ */
+export declare class VectorManager {
+    private embedder;
+    private searchManager;
+    private queries;
+    private nodeKinds;
+    private batchSize;
+    private initialized;
+    constructor(db: SqliteDatabase, queries: QueryBuilder, options?: VectorManagerOptions);
+    /**
+     * Initialize the vector manager
+     *
+     * Validates embedding configuration and initializes vector search.
+     */
+    initialize(): Promise<void>;
+    /**
+     * Check if the vector manager is initialized
+     */
+    isInitialized(): boolean;
+    /**
+     * Generate embeddings for all eligible nodes
+     *
+     * @param onProgress - Optional progress callback
+     * @returns Number of nodes embedded
+     */
+    embedAllNodes(onProgress?: (progress: EmbeddingProgress) => void): Promise<number>;
+    /**
+     * Generate embedding for a single node
+     *
+     * @param node - Node to embed
+     */
+    embedNode(node: Node): Promise<void>;
+    /**
+     * Semantic search for nodes matching a query
+     *
+     * @param query - Natural language query
+     * @param options - Search options
+     * @returns Array of search results with similarity scores
+     */
+    search(query: string, options?: SearchOptions): Promise<SearchResult[]>;
+    /**
+     * 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
+     */
+    findSimilar(nodeId: string, options?: SearchOptions): Promise<SearchResult[]>;
+    /**
+     * Delete embedding for a node
+     *
+     * @param nodeId - ID of the node
+     */
+    deleteNodeEmbedding(nodeId: string): void;
+    /**
+     * Get statistics about vector storage
+     */
+    getStats(): {
+        totalVectors: number;
+        vssEnabled: boolean;
+        modelId: string;
+        dimension: number;
+    };
+    /**
+     * Clear all vectors
+     */
+    clear(): void;
+    /**
+     * Delete vectors for nodes that no longer exist.
+     */
+    deleteStaleVectors(): number;
+    /**
+     * Rebuild the VSS index
+     */
+    rebuildIndex(): void;
+    /**
+     * Release resources
+     */
+    dispose(): void;
+    private hashText;
+}
+/**
+ * Create a vector manager
+ */
+export declare function createVectorManager(db: SqliteDatabase, queries: QueryBuilder, options?: VectorManagerOptions): VectorManager;
+//# sourceMappingURL=manager.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/vectors/manager.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAC7D,OAAO,EAAgC,eAAe,EAAE,MAAM,YAAY,CAAC;AAE3E,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAG7C;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,yBAAyB;IACzB,OAAO,EAAE,MAAM,CAAC;IAEhB,2BAA2B;IAC3B,KAAK,EAAE,MAAM,CAAC;IAEd,kCAAkC;IAClC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,uBAAuB;IACvB,QAAQ,CAAC,EAAE,eAAe,CAAC;IAE3B,6EAA6E;IAC7E,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;IAE3B,0CAA0C;IAC1C,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAeD;;;;;;;GAOG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,aAAa,CAAsB;IAC3C,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,SAAS,CAAiB;IAClC,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAS;gBAG1B,EAAE,EAAE,cAAc,EAClB,OAAO,EAAE,YAAY,EACrB,OAAO,GAAE,oBAAyB;IASpC;;;;OAIG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAcjC;;OAEG;IACH,aAAa,IAAI,OAAO;IAIxB;;;;;OAKG;IACG,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,iBAAiB,KAAK,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC;IAoExF;;;;OAIG;IACG,SAAS,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAU1C;;;;;;OAMG;IACG,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,aAAkB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IA0CjF;;;;;;OAMG;IACG,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,aAAkB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IA8DvF;;;;OAIG;IACH,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAIzC;;OAEG;IACH,QAAQ,IAAI;QACV,YAAY,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,OAAO,CAAC;QACpB,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,CAAC;KACnB;IASD;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;OAEG;IACH,kBAAkB,IAAI,MAAM;IAI5B;;OAEG;IACH,YAAY,IAAI,IAAI;IAIpB;;OAEG;IACH,OAAO,IAAI,IAAI;IAIf,OAAO,CAAC,QAAQ;CAGjB;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,EAAE,EAAE,cAAc,EAClB,OAAO,EAAE,YAAY,EACrB,OAAO,CAAC,EAAE,oBAAoB,GAC7B,aAAa,CAEf"}