@esankhan3/code-search-mcp 0.0.2 → 0.2.0

package/dist/core/vector-store.js

@@ -1,259 +0,0 @@
-class VectorStore {
-  db;
-  // lancedb.Connection
-  table;
-  // lancedb.Table
-  dbPath;
-  initialized = false;
-  constructor(dbPath) {
-    this.dbPath = dbPath;
-  }
-  /** Initialize connection, create or open table */
-  async init() {
-    let lancedb;
-    try {
-      lancedb = await import("@lancedb/lancedb");
-    } catch {
-      throw new Error(
-        "@lancedb/lancedb is not installed. Install it with: npm install @lancedb/lancedb"
-      );
-    }
-    this.db = await lancedb.connect(this.dbPath);
-    try {
-      this.table = await this.db.openTable("chunks");
-      await this.ensureFtsIndex();
-      this.initialized = true;
-    } catch {
-      this.initialized = true;
-    }
-  }
-  /** Create or rebuild the full-text search index on contextualizedContent */
-  async ensureFtsIndex() {
-    if (!this.table) return;
-    try {
-      const lancedb = await import("@lancedb/lancedb");
-      await this.table.createIndex("contextualizedContent", {
-        config: lancedb.Index.fts(),
-        replace: true
-      });
-    } catch {
-    }
-  }
-  /** Insert or replace chunks with embeddings */
-  async upsertChunks(chunks) {
-    const rows = chunks.map((c) => ({
-      id: c.id,
-      vector: c.embedding,
-      // LanceDB uses 'vector' field for embeddings
-      content: c.content,
-      contextualizedContent: c.contextualizedContent,
-      contextPrefix: c.contextPrefix,
-      filePath: c.filePath,
-      repoName: c.repoName,
-      project: c.project,
-      entityType: c.entityType,
-      entityName: c.entityName ?? "",
-      parentEntity: c.parentEntity ?? "",
-      language: c.language,
-      startLine: c.startLine,
-      endLine: c.endLine,
-      tokens: c.tokens
-    }));
-    if (!this.table) {
-      this.table = await this.db.createTable("chunks", rows, { mode: "overwrite" });
-    } else {
-      const project = chunks[0]?.project;
-      if (project) {
-        try {
-          await this.table.delete(`project = '${project.replace(/'/g, "''")}'`);
-        } catch {
-        }
-      }
-      await this.table.add(rows);
-    }
-    await this.ensureFtsIndex();
-  }
-  /** Semantic vector search */
-  async vectorSearch(queryEmbedding, opts) {
-    if (!this.table) return [];
-    let query = this.table.search(queryEmbedding).limit(opts?.limit ?? 20);
-    if (opts?.filter) query = query.where(opts.filter);
-    const results = await query.toArray();
-    return results.map((r) => ({
-      chunk: rowToChunk(r),
-      score: r._distance != null ? 1 / (1 + r._distance) : 0.5,
-      source: "vector"
-    }));
-  }
-  /** Full-text BM25 search (LanceDB built-in FTS) */
-  async fullTextSearch(queryText, limit = 20) {
-    if (!this.table) return [];
-    try {
-      const results = await this.table.search(queryText, "fts", "contextualizedContent").limit(limit).toArray();
-      return results.map((r) => ({
-        chunk: rowToChunk(r),
-        score: r._relevance_score ?? 0.5,
-        source: "bm25"
-      }));
-    } catch {
-      return [];
-    }
-  }
-  /** Get specific chunks by their IDs */
-  async getByIds(ids) {
-    if (!this.table || ids.length === 0) return [];
-    const filter = ids.map((id) => `id = '${id}'`).join(" OR ");
-    try {
-      const results = await this.table.filter(filter).toArray();
-      return results.map((r) => rowToChunk(r));
-    } catch {
-      return [];
-    }
-  }
-  /** Look up chunks by repo + file + entity name — for direct AST graph expansion.
-   *  Returns ScoredChunks with a fixed graph-expansion score. */
-  async getChunksByEntity(lookups) {
-    if (!this.table || lookups.length === 0) return [];
-    const esc = (s) => s.replace(/'/g, "''");
-    const conditions = lookups.map((l) => {
-      const base = `repoName = '${esc(l.repoName)}' AND filePath = '${esc(l.filePath)}'`;
-      return l.entityName ? `(${base} AND entityName = '${esc(l.entityName)}')` : `(${base})`;
-    });
-    try {
-      const allResults = [];
-      const batchSize = 20;
-      for (let i = 0; i < conditions.length; i += batchSize) {
-        const batch = conditions.slice(i, i + batchSize).join(" OR ");
-        const results = await this.table.filter(batch).limit(batchSize * 2).toArray();
-        for (const r of results) {
-          allResults.push({
-            chunk: rowToChunk(r),
-            score: 0.75,
-            source: "graph"
-          });
-        }
-      }
-      return allResults;
-    } catch {
-      return [];
-    }
-  }
-  /** Get all chunks for a specific file in a repo (for incremental comparison) */
-  async getChunksByFile(repoName, filePath) {
-    if (!this.table) return [];
-    const esc = (s) => s.replace(/'/g, "''");
-    try {
-      const results = await this.table.filter(`repoName = '${esc(repoName)}' AND filePath = '${esc(filePath)}'`).toArray();
-      return results.map((r) => ({
-        id: r.id,
-        content: r.content,
-        entityName: r.entityName || "",
-        tokens: r.tokens || 0
-      }));
-    } catch {
-      return [];
-    }
-  }
-  /** Delete specific chunks by their IDs (for surgical updates) */
-  async deleteChunksByIds(ids) {
-    if (!this.table || ids.length === 0) return;
-    const batchSize = 50;
-    for (let i = 0; i < ids.length; i += batchSize) {
-      const batch = ids.slice(i, i + batchSize);
-      const filter = batch.map((id) => `id = '${id.replace(/'/g, "''")}'`).join(" OR ");
-      try {
-        await this.table.delete(filter);
-      } catch {
-      }
-    }
-  }
-  /** Get all chunk IDs for a project (for incremental diff) */
-  async getChunkIds(project) {
-    if (!this.table) return [];
-    try {
-      const escapedProject = project.replace(/'/g, "''");
-      const results = await this.table.query().where(`project = '${escapedProject}'`).select(["id"]).toArray();
-      return results.map((r) => r.id);
-    } catch {
-      return [];
-    }
-  }
-  /** Get index statistics */
-  async getStats() {
-    if (!this.table) return null;
-    try {
-      const count = await this.table.countRows();
-      return { rowCount: count };
-    } catch {
-      return null;
-    }
-  }
-  /** Check if the store has data */
-  async hasData() {
-    const stats = await this.getStats();
-    return (stats?.rowCount ?? 0) > 0;
-  }
-  /** Delete chunks for specific files within a repo (for incremental re-indexing) */
-  async deleteFileChunks(project, repoName, filePaths) {
-    if (!this.table || filePaths.length === 0) return;
-    const escapedProject = project.replace(/'/g, "''");
-    const escapedRepo = repoName.replace(/'/g, "''");
-    const fileFilter = filePaths.map((f) => `'${f.replace(/'/g, "''")}'`).join(", ");
-    try {
-      await this.table.delete(
-        `project = '${escapedProject}' AND repoName = '${escapedRepo}' AND filePath IN (${fileFilter})`
-      );
-    } catch {
-    }
-  }
-  /** Add chunks without deleting existing project data (for incremental updates) */
-  async addChunks(chunks) {
-    if (chunks.length === 0) return;
-    const rows = chunks.map((c) => ({
-      id: c.id,
-      vector: c.embedding,
-      content: c.content,
-      contextualizedContent: c.contextualizedContent,
-      contextPrefix: c.contextPrefix,
-      filePath: c.filePath,
-      repoName: c.repoName,
-      project: c.project,
-      entityType: c.entityType,
-      entityName: c.entityName ?? "",
-      parentEntity: c.parentEntity ?? "",
-      language: c.language,
-      startLine: c.startLine,
-      endLine: c.endLine,
-      tokens: c.tokens
-    }));
-    if (!this.table) {
-      this.table = await this.db.createTable("chunks", rows, { mode: "overwrite" });
-    } else {
-      await this.table.add(rows);
-    }
-    await this.ensureFtsIndex();
-  }
-}
-function rowToChunk(row) {
-  return {
-    id: row.id,
-    filePath: row.filePath,
-    repoName: row.repoName,
-    project: row.project,
-    startLine: row.startLine,
-    endLine: row.endLine,
-    content: row.content,
-    contextPrefix: row.contextPrefix,
-    contextualizedContent: row.contextualizedContent,
-    language: row.language,
-    entityType: row.entityType,
-    entityName: row.entityName || void 0,
-    parentEntity: row.parentEntity || void 0,
-    tokens: row.tokens,
-    imports: [],
-    exports: []
-  };
-}
-export {
-  VectorStore
-};