@xfabric/memory 0.1.0

package/dist/search/hybrid.js

@@ -0,0 +1,83 @@
+/**
+ * Build an FTS5 query from search text
+ * Extracts alphanumeric tokens and joins with AND
+ */
+export function buildFtsQuery(text) {
+    // Extract alphanumeric tokens (including underscores)
+    const tokens = text.match(/[A-Za-z0-9_]+/g);
+    if (!tokens || tokens.length === 0) {
+        return null;
+    }
+    // Quote each token and join with AND
+    return tokens.map((t) => `"${t}"`).join(" AND ");
+}
+/**
+ * Convert BM25 rank to a 0-1 score
+ * Lower rank = higher relevance, so we invert
+ */
+export function bm25RankToScore(rank) {
+    // Handle non-finite values
+    const normalized = Number.isFinite(rank) ? Math.max(0, rank) : 999;
+    // Convert rank to score: 1/(1+rank) maps 0->1, infinity->0
+    return 1 / (1 + normalized);
+}
+const DEFAULT_VECTOR_WEIGHT = 0.7;
+const DEFAULT_TEXT_WEIGHT = 0.3;
+/**
+ * Merge vector and keyword search results with weighted scoring
+ */
+export function mergeHybridResults(vectorResults, keywordResults, options = {}) {
+    const vectorWeight = options.vectorWeight ?? DEFAULT_VECTOR_WEIGHT;
+    const textWeight = options.textWeight ?? DEFAULT_TEXT_WEIGHT;
+    // Build a map of all results by ID
+    const resultMap = new Map();
+    // Add vector results
+    for (const result of vectorResults) {
+        resultMap.set(result.id, {
+            path: result.path,
+            startLine: result.startLine,
+            endLine: result.endLine,
+            source: result.source,
+            vectorSnippet: result.snippet,
+            vectorScore: result.vectorScore,
+            textScore: 0,
+        });
+    }
+    // Merge keyword results
+    for (const result of keywordResults) {
+        const existing = resultMap.get(result.id);
+        if (existing) {
+            existing.keywordSnippet = result.snippet;
+            existing.textScore = result.textScore;
+        }
+        else {
+            resultMap.set(result.id, {
+                path: result.path,
+                startLine: result.startLine,
+                endLine: result.endLine,
+                source: result.source,
+                keywordSnippet: result.snippet,
+                vectorScore: 0,
+                textScore: result.textScore,
+            });
+        }
+    }
+    // Convert to final results with combined scores
+    const results = [];
+    for (const [, value] of resultMap) {
+        const combinedScore = vectorWeight * value.vectorScore + textWeight * value.textScore;
+        results.push({
+            path: value.path,
+            startLine: value.startLine,
+            endLine: value.endLine,
+            source: value.source,
+            // Prefer keyword snippet (often more contextually relevant)
+            snippet: value.keywordSnippet || value.vectorSnippet || "",
+            score: combinedScore,
+        });
+    }
+    // Sort by combined score descending
+    results.sort((a, b) => b.score - a.score);
+    return results;
+}
+//# sourceMappingURL=hybrid.js.map

package/dist/search/hybrid.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid.js","sourceRoot":"","sources":["../../src/search/hybrid.ts"],"names":[],"mappings":"AAOA;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,IAAY;IACxC,sDAAsD;IACtD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IAE5C,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,qCAAqC;IACrC,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AACnD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,IAAY;IAC1C,2BAA2B;IAC3B,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IACnE,2DAA2D;IAC3D,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,qBAAqB,GAAG,GAAG,CAAC;AAClC,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAEhC;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAChC,aAAmC,EACnC,cAAqC,EACrC,UAA+B,EAAE;IAEjC,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,qBAAqB,CAAC;IACnE,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,mBAAmB,CAAC;IAE7D,mCAAmC;IACnC,MAAM,SAAS,GAAG,IAAI,GAAG,EAYtB,CAAC;IAEJ,qBAAqB;IACrB,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;QACnC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE;YACvB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,aAAa,EAAE,MAAM,CAAC,OAAO;YAC7B,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,SAAS,EAAE,CAAC;SACb,CAAC,CAAC;IACL,CAAC;IAED,wBAAwB;IACxB,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1C,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC;YACzC,QAAQ,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE;gBACvB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,cAAc,EAAE,MAAM,CAAC,OAAO;gBAC9B,WAAW,EAAE,CAAC;gBACd,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,gDAAgD;IAChD,MAAM,OAAO,GAAyB,EAAE,CAAC;IAEzC,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,SAAS,EAAE,CAAC;QAClC,MAAM,aAAa,GACjB,YAAY,GAAG,KAAK,CAAC,WAAW,GAAG,UAAU,GAAG,KAAK,CAAC,SAAS,CAAC;QAElE,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,4DAA4D;YAC5D,OAAO,EAAE,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,aAAa,IAAI,EAAE;YAC1D,KAAK,EAAE,aAAa;SACrB,CAAC,CAAC;IACL,CAAC;IAED,oCAAoC;IACpC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IAE1C,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/search/index.d.ts

@@ -0,0 +1,4 @@
+export { buildFtsQuery, bm25RankToScore, mergeHybridResults } from "./hybrid.js";
+export { searchFts, type FtsSearchOptions } from "./fts.js";
+export { searchVector, searchVectorWithSqliteVec, syncVectorTable, type VectorSearchOptions, } from "./vector.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/search/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,EAAE,SAAS,EAAE,KAAK,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EACL,YAAY,EACZ,yBAAyB,EACzB,eAAe,EACf,KAAK,mBAAmB,GACzB,MAAM,aAAa,CAAC"}

package/dist/search/index.js

@@ -0,0 +1,4 @@
+export { buildFtsQuery, bm25RankToScore, mergeHybridResults } from "./hybrid.js";
+export { searchFts } from "./fts.js";
+export { searchVector, searchVectorWithSqliteVec, syncVectorTable, } from "./vector.js";
+//# sourceMappingURL=index.js.map

package/dist/search/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/search/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,EAAE,SAAS,EAAyB,MAAM,UAAU,CAAC;AAC5D,OAAO,EACL,YAAY,EACZ,yBAAyB,EACzB,eAAe,GAEhB,MAAM,aAAa,CAAC"}

package/dist/search/vector.d.ts

@@ -0,0 +1,25 @@
+import type Database from "better-sqlite3";
+import type { VectorSearchResult, MemorySource } from "../types.js";
+export type VectorSearchOptions = {
+    maxResults?: number;
+    minScore?: number;
+    source?: MemorySource;
+};
+/**
+ * Search using vector similarity (cosine similarity) - JS fallback
+ */
+export declare function searchVector(db: Database.Database, queryEmbedding: number[], options?: VectorSearchOptions): VectorSearchResult[];
+/**
+ * Search using sqlite-vec for database-side ANN search
+ * Falls back to JS cosine similarity if vec_chunks table doesn't exist
+ */
+export declare function searchVectorWithSqliteVec(db: Database.Database, queryEmbedding: number[], options?: VectorSearchOptions): VectorSearchResult[];
+/**
+ * Sync vec_chunks virtual table with chunks table
+ * Call this after bulk inserts to ensure ANN search data is up to date
+ */
+export declare function syncVectorTable(db: Database.Database): {
+    synced: number;
+    errors: number;
+};
+//# sourceMappingURL=vector.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"vector.d.ts","sourceRoot":"","sources":["../../src/search/vector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAC3C,OAAO,KAAK,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAGpE,MAAM,MAAM,mBAAmB,GAAG;IAChC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB,CAAC;AAEF;;GAEG;AACH,wBAAgB,YAAY,CAC1B,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,cAAc,EAAE,MAAM,EAAE,EACxB,OAAO,GAAE,mBAAwB,GAChC,kBAAkB,EAAE,CA0DtB;AAED;;;GAGG;AACH,wBAAgB,yBAAyB,CACvC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,cAAc,EAAE,MAAM,EAAE,EACxB,OAAO,GAAE,mBAAwB,GAChC,kBAAkB,EAAE,CAmEtB;AAED;;;GAGG;AACH,wBAAgB,eAAe,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAkDzF"}

package/dist/search/vector.js

@@ -0,0 +1,152 @@
+import { cosineSimilarity } from "../storage/sqlite-vec.js";
+/**
+ * Search using vector similarity (cosine similarity) - JS fallback
+ */
+export function searchVector(db, queryEmbedding, options = {}) {
+    const { maxResults = 50, minScore = 0, source } = options;
+    // Build query with optional source filter
+    let sql = `
+    SELECT
+      id,
+      path,
+      source,
+      start_line,
+      end_line,
+      text,
+      embedding
+    FROM chunks
+  `;
+    const params = [];
+    if (source) {
+        sql += " WHERE source = ?";
+        params.push(source);
+    }
+    const stmt = db.prepare(sql);
+    const rows = (params.length > 0 ? stmt.all(...params) : stmt.all());
+    // Calculate similarities
+    const results = [];
+    for (const row of rows) {
+        const embedding = JSON.parse(row.embedding);
+        const similarity = cosineSimilarity(queryEmbedding, embedding);
+        if (similarity >= minScore) {
+            results.push({
+                id: row.id,
+                path: row.path,
+                startLine: row.start_line,
+                endLine: row.end_line,
+                source: row.source,
+                snippet: row.text,
+                vectorScore: similarity,
+            });
+        }
+    }
+    // Sort by similarity descending
+    results.sort((a, b) => b.vectorScore - a.vectorScore);
+    // Limit results
+    return results.slice(0, maxResults);
+}
+/**
+ * Search using sqlite-vec for database-side ANN search
+ * Falls back to JS cosine similarity if vec_chunks table doesn't exist
+ */
+export function searchVectorWithSqliteVec(db, queryEmbedding, options = {}) {
+    const { maxResults = 50, minScore = 0, source } = options;
+    try {
+        // Check if vec_chunks virtual table exists
+        const tableExists = db
+            .prepare("SELECT name FROM sqlite_master WHERE type='table' AND name='vec_chunks'")
+            .get();
+        if (!tableExists) {
+            // Fall back to JS-based search
+            return searchVector(db, queryEmbedding, options);
+        }
+        // Convert embedding to blob format for sqlite-vec
+        const embeddingBlob = Buffer.from(new Float32Array(queryEmbedding).buffer);
+        // Build query with ANN search
+        let sql = `
+      SELECT
+        c.id,
+        c.path,
+        c.source,
+        c.start_line,
+        c.end_line,
+        c.text,
+        1 - vec_distance_cosine(v.embedding, ?) as score
+      FROM vec_chunks v
+      JOIN chunks c ON v.id = c.id
+    `;
+        const params = [embeddingBlob];
+        if (source) {
+            sql += " WHERE c.source = ?";
+            params.push(source);
+        }
+        sql += " ORDER BY score DESC LIMIT ?";
+        params.push(maxResults);
+        const rows = db.prepare(sql).all(...params);
+        // Filter by min score and map to results
+        return rows
+            .filter((row) => row.score >= minScore)
+            .map((row) => ({
+            id: row.id,
+            path: row.path,
+            startLine: row.start_line,
+            endLine: row.end_line,
+            source: row.source,
+            snippet: row.text,
+            vectorScore: row.score,
+        }));
+    }
+    catch {
+        // Fall back to JS-based search on any error
+        return searchVector(db, queryEmbedding, options);
+    }
+}
+/**
+ * Sync vec_chunks virtual table with chunks table
+ * Call this after bulk inserts to ensure ANN search data is up to date
+ */
+export function syncVectorTable(db) {
+    let synced = 0;
+    let errors = 0;
+    try {
+        // Check if vec_chunks exists
+        const tableExists = db
+            .prepare("SELECT name FROM sqlite_master WHERE type='table' AND name='vec_chunks'")
+            .get();
+        if (!tableExists) {
+            return { synced: 0, errors: 0 };
+        }
+        // Get all chunks that don't have entries in vec_chunks
+        const missingRows = db
+            .prepare(`
+        SELECT c.id, c.embedding
+        FROM chunks c
+        LEFT JOIN vec_chunks v ON c.id = v.id
+        WHERE v.id IS NULL
+      `)
+            .all();
+        // Insert missing entries
+        const insertStmt = db.prepare("INSERT INTO vec_chunks (id, embedding) VALUES (?, ?)");
+        for (const row of missingRows) {
+            try {
+                const embedding = JSON.parse(row.embedding);
+                const embeddingBlob = Buffer.from(new Float32Array(embedding).buffer);
+                insertStmt.run(row.id, embeddingBlob);
+                synced++;
+            }
+            catch {
+                errors++;
+            }
+        }
+        // Remove orphaned entries
+        db.exec(`
+      DELETE FROM vec_chunks
+      WHERE id NOT IN (SELECT id FROM chunks)
+    `);
+    }
+    catch {
+        // Table doesn't exist or other error
+    }
+    return { synced, errors };
+}
+//# sourceMappingURL=vector.js.map

package/dist/search/vector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector.js","sourceRoot":"","sources":["../../src/search/vector.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAQ5D;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,EAAqB,EACrB,cAAwB,EACxB,UAA+B,EAAE;IAEjC,MAAM,EAAE,UAAU,GAAG,EAAE,EAAE,QAAQ,GAAG,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAE1D,0CAA0C;IAC1C,IAAI,GAAG,GAAG;;;;;;;;;;GAUT,CAAC;IAEF,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,MAAM,EAAE,CAAC;QACX,GAAG,IAAI,mBAAmB,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtB,CAAC;IAED,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAQ/D,CAAC;IAEJ,yBAAyB;IACzB,MAAM,OAAO,GAAyB,EAAE,CAAC;IAEzC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAa,CAAC;QACxD,MAAM,UAAU,GAAG,gBAAgB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAE/D,IAAI,UAAU,IAAI,QAAQ,EAAE,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,SAAS,EAAE,GAAG,CAAC,UAAU;gBACzB,OAAO,EAAE,GAAG,CAAC,QAAQ;gBACrB,MAAM,EAAE,GAAG,CAAC,MAAsB;gBAClC,OAAO,EAAE,GAAG,CAAC,IAAI;gBACjB,WAAW,EAAE,UAAU;aACxB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,gCAAgC;IAChC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC;IAEtD,gBAAgB;IAChB,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;AACtC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,yBAAyB,CACvC,EAAqB,EACrB,cAAwB,EACxB,UAA+B,EAAE;IAEjC,MAAM,EAAE,UAAU,GAAG,EAAE,EAAE,QAAQ,GAAG,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAE1D,IAAI,CAAC;QACH,2CAA2C;QAC3C,MAAM,WAAW,GAAG,EAAE;aACnB,OAAO,CAAC,yEAAyE,CAAC;aAClF,GAAG,EAAE,CAAC;QAET,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,+BAA+B;YAC/B,OAAO,YAAY,CAAC,EAAE,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;QACnD,CAAC;QAED,kDAAkD;QAClD,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,CAAC;QAE3E,8BAA8B;QAC9B,IAAI,GAAG,GAAG;;;;;;;;;;;KAWT,CAAC;QAEF,MAAM,MAAM,GAAiC,CAAC,aAAa,CAAC,CAAC;QAE7D,IAAI,MAAM,EAAE,CAAC;YACX,GAAG,IAAI,qBAAqB,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC;QAED,GAAG,IAAI,8BAA8B,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAExB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAQvC,CAAC;QAEJ,yCAAyC;QACzC,OAAO,IAAI;aACR,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,IAAI,QAAQ,CAAC;aACtC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACb,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,OAAO,EAAE,GAAG,CAAC,QAAQ;YACrB,MAAM,EAAE,GAAG,CAAC,MAAsB;YAClC,OAAO,EAAE,GAAG,CAAC,IAAI;YACjB,WAAW,EAAE,GAAG,CAAC,KAAK;SACvB,CAAC,CAAC,CAAC;IACR,CAAC;IAAC,MAAM,CAAC;QACP,4CAA4C;QAC5C,OAAO,YAAY,CAAC,EAAE,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;IACnD,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,EAAqB;IACnD,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,IAAI,CAAC;QACH,6BAA6B;QAC7B,MAAM,WAAW,GAAG,EAAE;aACnB,OAAO,CAAC,yEAAyE,CAAC;aAClF,GAAG,EAAE,CAAC;QAET,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QAClC,CAAC;QAED,uDAAuD;QACvD,MAAM,WAAW,GAAG,EAAE;aACnB,OAAO,CACN;;;;;OAKD,CACA;aACA,GAAG,EAAyC,CAAC;QAEhD,yBAAyB;QACzB,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,sDAAsD,CAAC,CAAC;QAEtF,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAa,CAAC;gBACxD,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC;gBACtE,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;gBACtC,MAAM,EAAE,CAAC;YACX,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,EAAE,CAAC;YACX,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,EAAE,CAAC,IAAI,CAAC;;;KAGP,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,qCAAqC;IACvC,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC5B,CAAC"}

package/dist/storage/index.d.ts

@@ -0,0 +1,4 @@
+export { ensureMemoryIndexSchema, getSchemaVersion, migrateSchema, ensureColumn, SCHEMA_VERSION, type SchemaStatus, } from "./schema.js";
+export { createDatabase, MemoryStorage, type StorageOptions, type CreateDatabaseResult, } from "./sqlite.js";
+export { loadSqliteVec, isSqliteVecAvailable, createVectorTable, cosineSimilarity, resetSqliteVecState, } from "./sqlite-vec.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,uBAAuB,EACvB,gBAAgB,EAChB,aAAa,EACb,YAAY,EACZ,cAAc,EACd,KAAK,YAAY,GAClB,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,cAAc,EACd,aAAa,EACb,KAAK,cAAc,EACnB,KAAK,oBAAoB,GAC1B,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,aAAa,EACb,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,GACpB,MAAM,iBAAiB,CAAC"}

package/dist/storage/index.js

@@ -0,0 +1,4 @@
+export { ensureMemoryIndexSchema, getSchemaVersion, migrateSchema, ensureColumn, SCHEMA_VERSION, } from "./schema.js";
+export { createDatabase, MemoryStorage, } from "./sqlite.js";
+export { loadSqliteVec, isSqliteVecAvailable, createVectorTable, cosineSimilarity, resetSqliteVecState, } from "./sqlite-vec.js";
+//# sourceMappingURL=index.js.map

package/dist/storage/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,uBAAuB,EACvB,gBAAgB,EAChB,aAAa,EACb,YAAY,EACZ,cAAc,GAEf,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,cAAc,EACd,aAAa,GAGd,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,aAAa,EACb,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,GACpB,MAAM,iBAAiB,CAAC"}

package/dist/storage/schema.d.ts

@@ -0,0 +1,24 @@
+import type Database from "better-sqlite3";
+export declare const SCHEMA_VERSION = 2;
+export type SchemaStatus = {
+    version: number;
+    ftsAvailable: boolean;
+    ftsError?: string;
+};
+/**
+ * Ensure a column exists in a table, adding it if necessary
+ */
+export declare function ensureColumn(db: Database.Database, table: string, column: string, definition: string): boolean;
+/**
+ * Ensure the memory index schema exists in the database
+ */
+export declare function ensureMemoryIndexSchema(db: Database.Database): SchemaStatus;
+/**
+ * Get the current schema version from the database
+ */
+export declare function getSchemaVersion(db: Database.Database): number;
+/**
+ * Migrate schema from one version to another
+ */
+export declare function migrateSchema(db: Database.Database, fromVersion: number): SchemaStatus;
+//# sourceMappingURL=schema.d.ts.map

package/dist/storage/schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../src/storage/schema.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAE3C,eAAO,MAAM,cAAc,IAAI,CAAC;AAEhC,MAAM,MAAM,YAAY,GAAG;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,OAAO,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,wBAAgB,YAAY,CAC1B,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,GACjB,OAAO,CAaT;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,YAAY,CAmH3E;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM,CAS9D;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,MAAM,GAAG,YAAY,CAuCtF"}

package/dist/storage/schema.js

@@ -0,0 +1,175 @@
+export const SCHEMA_VERSION = 2;
+/**
+ * Ensure a column exists in a table, adding it if necessary
+ */
+export function ensureColumn(db, table, column, definition) {
+    try {
+        // Check if column exists using PRAGMA table_info
+        const columns = db.prepare(`PRAGMA table_info(${table})`).all();
+        const exists = columns.some((col) => col.name === column);
+        if (!exists) {
+            db.exec(`ALTER TABLE ${table} ADD COLUMN ${column} ${definition}`);
+        }
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * Ensure the memory index schema exists in the database
+ */
+export function ensureMemoryIndexSchema(db) {
+    const status = {
+        version: SCHEMA_VERSION,
+        ftsAvailable: false,
+    };
+    // meta table - stores configuration and version info
+    db.exec(`
+    CREATE TABLE IF NOT EXISTS meta (
+      key TEXT PRIMARY KEY,
+      value TEXT NOT NULL
+    )
+  `);
+    // files table - tracks indexed files
+    db.exec(`
+    CREATE TABLE IF NOT EXISTS files (
+      path TEXT PRIMARY KEY,
+      source TEXT NOT NULL DEFAULT 'memory',
+      hash TEXT NOT NULL,
+      mtime INTEGER NOT NULL,
+      size INTEGER NOT NULL
+    )
+  `);
+    // Add session-specific columns to files table if needed
+    ensureColumn(db, "files", "byte_offset", "INTEGER DEFAULT 0");
+    ensureColumn(db, "files", "session_id", "TEXT");
+    // chunks table - stores text chunks with embeddings
+    db.exec(`
+    CREATE TABLE IF NOT EXISTS chunks (
+      id TEXT PRIMARY KEY,
+      path TEXT NOT NULL,
+      source TEXT NOT NULL DEFAULT 'memory',
+      start_line INTEGER NOT NULL,
+      end_line INTEGER NOT NULL,
+      hash TEXT NOT NULL,
+      model TEXT NOT NULL,
+      text TEXT NOT NULL,
+      embedding TEXT NOT NULL,
+      updated_at INTEGER NOT NULL
+    )
+  `);
+    // Add binary embedding column for sqlite-vec support
+    ensureColumn(db, "chunks", "embedding_blob", "BLOB");
+    // Add session-specific columns to chunks table
+    ensureColumn(db, "chunks", "session_id", "TEXT");
+    ensureColumn(db, "chunks", "role", "TEXT");
+    ensureColumn(db, "chunks", "byte_offset", "INTEGER DEFAULT 0");
+    // embedding_cache table - caches embeddings to avoid re-computation
+    db.exec(`
+    CREATE TABLE IF NOT EXISTS embedding_cache (
+      provider TEXT NOT NULL,
+      model TEXT NOT NULL,
+      provider_key TEXT NOT NULL,
+      hash TEXT NOT NULL,
+      embedding TEXT NOT NULL,
+      dims INTEGER,
+      updated_at INTEGER NOT NULL,
+      PRIMARY KEY (provider, model, provider_key, hash)
+    )
+  `);
+    // Add binary embedding column to cache
+    ensureColumn(db, "embedding_cache", "embedding_blob", "BLOB");
+    // FTS5 virtual table for full-text search
+    try {
+        db.exec(`
+      CREATE VIRTUAL TABLE IF NOT EXISTS chunks_fts USING fts5(
+        text,
+        id UNINDEXED,
+        path UNINDEXED,
+        source UNINDEXED,
+        model UNINDEXED,
+        start_line UNINDEXED,
+        end_line UNINDEXED
+      )
+    `);
+        status.ftsAvailable = true;
+    }
+    catch (err) {
+        status.ftsError = err instanceof Error ? err.message : String(err);
+    }
+    // Indexes
+    db.exec(`
+    CREATE INDEX IF NOT EXISTS idx_chunks_path ON chunks(path)
+  `);
+    db.exec(`
+    CREATE INDEX IF NOT EXISTS idx_chunks_source ON chunks(source)
+  `);
+    db.exec(`
+    CREATE INDEX IF NOT EXISTS idx_chunks_session ON chunks(session_id)
+  `);
+    db.exec(`
+    CREATE INDEX IF NOT EXISTS idx_embedding_cache_updated_at ON embedding_cache(updated_at)
+  `);
+    db.exec(`
+    CREATE INDEX IF NOT EXISTS idx_files_source ON files(source)
+  `);
+    // Store schema version
+    db.prepare(`
+    INSERT OR REPLACE INTO meta (key, value) VALUES (?, ?)
+  `).run("schema_version", String(SCHEMA_VERSION));
+    return status;
+}
+/**
+ * Get the current schema version from the database
+ */
+export function getSchemaVersion(db) {
+    try {
+        const row = db.prepare("SELECT value FROM meta WHERE key = ?").get("schema_version");
+        return row ? parseInt(row.value, 10) : 0;
+    }
+    catch {
+        return 0;
+    }
+}
+/**
+ * Migrate schema from one version to another
+ */
+export function migrateSchema(db, fromVersion) {
+    const status = {
+        version: SCHEMA_VERSION,
+        ftsAvailable: false,
+    };
+    // Version 1 -> 2: Add binary embedding columns and session support
+    if (fromVersion < 2) {
+        ensureColumn(db, "chunks", "embedding_blob", "BLOB");
+        ensureColumn(db, "chunks", "session_id", "TEXT");
+        ensureColumn(db, "chunks", "role", "TEXT");
+        ensureColumn(db, "chunks", "byte_offset", "INTEGER DEFAULT 0");
+        ensureColumn(db, "embedding_cache", "embedding_blob", "BLOB");
+        ensureColumn(db, "files", "byte_offset", "INTEGER DEFAULT 0");
+        ensureColumn(db, "files", "session_id", "TEXT");
+        // Create new indexes
+        try {
+            db.exec(`CREATE INDEX IF NOT EXISTS idx_chunks_session ON chunks(session_id)`);
+            db.exec(`CREATE INDEX IF NOT EXISTS idx_files_source ON files(source)`);
+        }
+        catch {
+            // Indexes might already exist
+        }
+    }
+    // Check FTS availability
+    try {
+        db.prepare("SELECT 1 FROM chunks_fts LIMIT 0").run();
+        status.ftsAvailable = true;
+    }
+    catch (err) {
+        status.ftsError = err instanceof Error ? err.message : String(err);
+    }
+    // Update schema version
+    db.prepare(`
+    INSERT OR REPLACE INTO meta (key, value) VALUES (?, ?)
+  `).run("schema_version", String(SCHEMA_VERSION));
+    return status;
+}
+//# sourceMappingURL=schema.js.map

package/dist/storage/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/storage/schema.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC;AAQhC;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,EAAqB,EACrB,KAAa,EACb,MAAc,EACd,UAAkB;IAElB,IAAI,CAAC;QACH,iDAAiD;QACjD,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,qBAAqB,KAAK,GAAG,CAAC,CAAC,GAAG,EAAwB,CAAC;QACtF,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAE1D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,EAAE,CAAC,IAAI,CAAC,eAAe,KAAK,eAAe,MAAM,IAAI,UAAU,EAAE,CAAC,CAAC;QACrE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,EAAqB;IAC3D,MAAM,MAAM,GAAiB;QAC3B,OAAO,EAAE,cAAc;QACvB,YAAY,EAAE,KAAK;KACpB,CAAC;IAEF,qDAAqD;IACrD,EAAE,CAAC,IAAI,CAAC;;;;;GAKP,CAAC,CAAC;IAEH,qCAAqC;IACrC,EAAE,CAAC,IAAI,CAAC;;;;;;;;GAQP,CAAC,CAAC;IAEH,wDAAwD;IACxD,YAAY,CAAC,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;IAC9D,YAAY,CAAC,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;IAEhD,oDAAoD;IACpD,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;GAaP,CAAC,CAAC;IAEH,qDAAqD;IACrD,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,CAAC,CAAC;IAErD,+CAA+C;IAC/C,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;IACjD,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3C,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;IAE/D,oEAAoE;IACpE,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;GAWP,CAAC,CAAC;IAEH,uCAAuC;IACvC,YAAY,CAAC,EAAE,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,CAAC,CAAC;IAE9D,0CAA0C;IAC1C,IAAI,CAAC;QACH,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;KAUP,CAAC,CAAC;QACH,MAAM,CAAC,YAAY,GAAG,IAAI,CAAC;IAC7B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,QAAQ,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACrE,CAAC;IAED,UAAU;IACV,EAAE,CAAC,IAAI,CAAC;;GAEP,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC;;GAEP,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC;;GAEP,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC;;GAEP,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC;;GAEP,CAAC,CAAC;IAEH,uBAAuB;IACvB,EAAE,CAAC,OAAO,CAAC;;GAEV,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;IAEjD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,EAAqB;IACpD,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAEtE,CAAC;QACd,OAAO,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,EAAqB,EAAE,WAAmB;IACtE,MAAM,MAAM,GAAiB;QAC3B,OAAO,EAAE,cAAc;QACvB,YAAY,EAAE,KAAK;KACpB,CAAC;IAEF,mEAAmE;IACnE,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACpB,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,CAAC,CAAC;QACrD,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QACjD,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3C,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;QAC/D,YAAY,CAAC,EAAE,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,CAAC,CAAC;QAC9D,YAAY,CAAC,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;QAC9D,YAAY,CAAC,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAEhD,qBAAqB;QACrB,IAAI,CAAC;YACH,EAAE,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;YAC/E,EAAE,CAAC,IAAI,CAAC,8DAA8D,CAAC,CAAC;QAC1E,CAAC;QAAC,MAAM,CAAC;YACP,8BAA8B;QAChC,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,IAAI,CAAC;QACH,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,EAAE,CAAC;QACrD,MAAM,CAAC,YAAY,GAAG,IAAI,CAAC;IAC7B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,QAAQ,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACrE,CAAC;IAED,wBAAwB;IACxB,EAAE,CAAC,OAAO,CAAC;;GAEV,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;IAEjD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/storage/sqlite-vec.d.ts

@@ -0,0 +1,22 @@
+import type Database from "better-sqlite3";
+/**
+ * Try to load the sqlite-vec extension
+ */
+export declare function loadSqliteVec(db: Database.Database): boolean;
+/**
+ * Check if sqlite-vec is available
+ */
+export declare function isSqliteVecAvailable(): boolean;
+/**
+ * Create a vec0 virtual table for vector search
+ */
+export declare function createVectorTable(db: Database.Database, tableName: string, dimensions: number): boolean;
+/**
+ * Pure JS cosine similarity fallback
+ */
+export declare function cosineSimilarity(a: number[], b: number[]): number;
+/**
+ * Reset extension state (for testing)
+ */
+export declare function resetSqliteVecState(): void;
+//# sourceMappingURL=sqlite-vec.d.ts.map

package/dist/storage/sqlite-vec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sqlite-vec.d.ts","sourceRoot":"","sources":["../../src/storage/sqlite-vec.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAK3C;;GAEG;AACH,wBAAgB,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,OAAO,CAmB5D;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,OAAO,CAW9C;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAC/B,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,GACjB,OAAO,CAgBT;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAiBjE;AAED;;GAEG;AACH,wBAAgB,mBAAmB,IAAI,IAAI,CAG1C"}

package/dist/storage/sqlite-vec.js

@@ -0,0 +1,85 @@
+let sqliteVecLoaded = false;
+let loadError = null;
+/**
+ * Try to load the sqlite-vec extension
+ */
+export function loadSqliteVec(db) {
+    if (loadError) {
+        return false;
+    }
+    if (sqliteVecLoaded) {
+        return true;
+    }
+    try {
+        // Dynamic import of sqlite-vec
+        const sqliteVec = require("sqlite-vec");
+        sqliteVec.load(db);
+        sqliteVecLoaded = true;
+        return true;
+    }
+    catch (err) {
+        loadError = err instanceof Error ? err : new Error(String(err));
+        return false;
+    }
+}
+/**
+ * Check if sqlite-vec is available
+ */
+export function isSqliteVecAvailable() {
+    if (loadError) {
+        return false;
+    }
+    try {
+        require("sqlite-vec");
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * Create a vec0 virtual table for vector search
+ */
+export function createVectorTable(db, tableName, dimensions) {
+    if (!loadSqliteVec(db)) {
+        return false;
+    }
+    try {
+        db.exec(`
+      CREATE VIRTUAL TABLE IF NOT EXISTS ${tableName} USING vec0(
+        id TEXT PRIMARY KEY,
+        embedding float[${dimensions}]
+      )
+    `);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * Pure JS cosine similarity fallback
+ */
+export function cosineSimilarity(a, b) {
+    if (a.length !== b.length || a.length === 0) {
+        return 0;
+    }
+    let dotProduct = 0;
+    let normA = 0;
+    let normB = 0;
+    for (let i = 0; i < a.length; i++) {
+        dotProduct += a[i] * b[i];
+        normA += a[i] * a[i];
+        normB += b[i] * b[i];
+    }
+    const magnitude = Math.sqrt(normA) * Math.sqrt(normB);
+    return magnitude === 0 ? 0 : dotProduct / magnitude;
+}
+/**
+ * Reset extension state (for testing)
+ */
+export function resetSqliteVecState() {
+    sqliteVecLoaded = false;
+    loadError = null;
+}
+//# sourceMappingURL=sqlite-vec.js.map

package/dist/storage/sqlite-vec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sqlite-vec.js","sourceRoot":"","sources":["../../src/storage/sqlite-vec.ts"],"names":[],"mappings":"AAEA,IAAI,eAAe,GAAG,KAAK,CAAC;AAC5B,IAAI,SAAS,GAAiB,IAAI,CAAC;AAEnC;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,EAAqB;IACjD,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACH,+BAA+B;QAC/B,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;QACxC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnB,eAAe,GAAG,IAAI,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,SAAS,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAChE,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,CAAC;QACH,OAAO,CAAC,YAAY,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAC/B,EAAqB,EACrB,SAAiB,EACjB,UAAkB;IAElB,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,EAAE,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,CAAC;QACH,EAAE,CAAC,IAAI,CAAC;2CAC+B,SAAS;;0BAE1B,UAAU;;KAE/B,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,CAAW,EAAE,CAAW;IACvD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5C,OAAO,CAAC,CAAC;IACX,CAAC;IAED,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1B,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtD,OAAO,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,SAAS,CAAC;AACtD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB;IACjC,eAAe,GAAG,KAAK,CAAC;IACxB,SAAS,GAAG,IAAI,CAAC;AACnB,CAAC"}