aidex-graphra 1.0.0

package/dist/storage.js

@@ -0,0 +1,322 @@
+"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;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getDb = getDb;
+exports.closeDb = closeDb;
+exports.upsertChunk = upsertChunk;
+exports.upsertChunks = upsertChunks;
+exports.getChunk = getChunk;
+exports.getAllChunks = getAllChunks;
+exports.getChunkHash = getChunkHash;
+exports.getChunkCount = getChunkCount;
+exports.upsertEmbedding = upsertEmbedding;
+exports.upsertEmbeddings = upsertEmbeddings;
+exports.getEmbedding = getEmbedding;
+exports.getAllEmbeddings = getAllEmbeddings;
+exports.saveGraph = saveGraph;
+exports.loadGraph = loadGraph;
+exports.setMeta = setMeta;
+exports.getMeta = getMeta;
+exports.getFileMtime = getFileMtime;
+exports.upsertFile = upsertFile;
+exports.getTrackedFiles = getTrackedFiles;
+exports.removeFile = removeFile;
+exports.removeChunksForFile = removeChunksForFile;
+exports.getChunksWithoutEmbeddings = getChunksWithoutEmbeddings;
+exports.clearAll = clearAll;
+const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+/**
+ * SQLite-based storage for Graphra.
+ * Replaces the bloated chunks.json with a compact, fast database.
+ *
+ * Tables:
+ *   chunks     — id, file, type, name, signature, code, hash, summary
+ *   embeddings — chunk_id, vector (binary float32 blob)
+ *   graph      — source_file, target_file
+ *   meta       — key, value (for vocabulary, version, etc.)
+ */
+const CACHE_DIR = ".graphra";
+const DB_FILE = path.join(CACHE_DIR, "graphra.db");
+let _db = null;
+/** Get or create the database connection */
+function getDb() {
+    if (_db)
+        return _db;
+    if (!fs.existsSync(CACHE_DIR)) {
+        fs.mkdirSync(CACHE_DIR, { recursive: true });
+    }
+    _db = new better_sqlite3_1.default(DB_FILE);
+    _db.pragma("journal_mode = WAL");
+    _db.pragma("synchronous = NORMAL");
+    // Create tables
+    _db.exec(`
+    CREATE TABLE IF NOT EXISTS chunks (
+      id TEXT PRIMARY KEY,
+      file TEXT NOT NULL,
+      type TEXT NOT NULL,
+      name TEXT NOT NULL,
+      signature TEXT NOT NULL DEFAULT '',
+      code TEXT NOT NULL,
+      hash TEXT NOT NULL,
+      summary TEXT NOT NULL DEFAULT ''
+    );
+
+    CREATE TABLE IF NOT EXISTS embeddings (
+      chunk_id TEXT PRIMARY KEY,
+      vector BLOB,
+      FOREIGN KEY (chunk_id) REFERENCES chunks(id)
+    );
+
+    CREATE TABLE IF NOT EXISTS graph (
+      source_file TEXT NOT NULL,
+      target_file TEXT NOT NULL,
+      PRIMARY KEY (source_file, target_file)
+    );
+
+    CREATE TABLE IF NOT EXISTS meta (
+      key TEXT PRIMARY KEY,
+      value TEXT NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS files (
+      path TEXT PRIMARY KEY,
+      mtime REAL NOT NULL,
+      hash TEXT NOT NULL
+    );
+
+    CREATE INDEX IF NOT EXISTS idx_chunks_file ON chunks(file);
+    CREATE INDEX IF NOT EXISTS idx_chunks_name ON chunks(name);
+    CREATE INDEX IF NOT EXISTS idx_graph_source ON graph(source_file);
+    CREATE INDEX IF NOT EXISTS idx_graph_target ON graph(target_file);
+  `);
+    return _db;
+}
+/** Close the database */
+function closeDb() {
+    if (_db) {
+        _db.close();
+        _db = null;
+    }
+}
+// ============================================
+// Chunk Storage
+// ============================================
+/** Upsert a chunk into the database */
+function upsertChunk(chunk) {
+    const db = getDb();
+    db.prepare(`
+    INSERT OR REPLACE INTO chunks (id, file, type, name, signature, code, hash, summary)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(chunk.id, chunk.file, chunk.type, chunk.name, chunk.signature, chunk.code, chunk.hash ?? "", chunk.summary ?? "");
+}
+/** Batch upsert chunks (much faster) */
+function upsertChunks(chunks) {
+    const db = getDb();
+    const stmt = db.prepare(`
+    INSERT OR REPLACE INTO chunks (id, file, type, name, signature, code, hash, summary)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+  `);
+    const tx = db.transaction((items) => {
+        for (const c of items) {
+            stmt.run(c.id, c.file, c.type, c.name, c.signature, c.code, c.hash ?? "", c.summary ?? "");
+        }
+    });
+    tx(chunks);
+}
+/** Get a chunk by ID */
+function getChunk(id) {
+    const db = getDb();
+    return db.prepare("SELECT * FROM chunks WHERE id = ?").get(id);
+}
+/** Get all chunks */
+function getAllChunks() {
+    const db = getDb();
+    return db.prepare("SELECT * FROM chunks").all();
+}
+/** Get chunk hash for cache checking */
+function getChunkHash(id) {
+    const db = getDb();
+    const row = db.prepare("SELECT hash FROM chunks WHERE id = ?").get(id);
+    return row?.hash ?? null;
+}
+/** Get chunk count */
+function getChunkCount() {
+    const db = getDb();
+    return db.prepare("SELECT COUNT(*) as count FROM chunks").get().count;
+}
+// ============================================
+// Embedding Storage (binary float32 blobs)
+// ============================================
+/** Store an embedding vector as a binary blob */
+function upsertEmbedding(chunkId, vector) {
+    const db = getDb();
+    const buf = Buffer.from(new Float32Array(vector).buffer);
+    db.prepare("INSERT OR REPLACE INTO embeddings (chunk_id, vector) VALUES (?, ?)").run(chunkId, buf);
+}
+/** Batch upsert embeddings */
+function upsertEmbeddings(items) {
+    const db = getDb();
+    const stmt = db.prepare("INSERT OR REPLACE INTO embeddings (chunk_id, vector) VALUES (?, ?)");
+    const tx = db.transaction((entries) => {
+        for (const { chunkId, vector } of entries) {
+            const buf = Buffer.from(new Float32Array(vector).buffer);
+            stmt.run(chunkId, buf);
+        }
+    });
+    tx(items);
+}
+/** Get an embedding vector */
+function getEmbedding(chunkId) {
+    const db = getDb();
+    const row = db.prepare("SELECT vector FROM embeddings WHERE chunk_id = ?").get(chunkId);
+    if (!row?.vector)
+        return null;
+    return Array.from(new Float32Array(row.vector.buffer, row.vector.byteOffset, row.vector.byteLength / 4));
+}
+/** Get all embeddings as a map */
+function getAllEmbeddings() {
+    const db = getDb();
+    const rows = db.prepare("SELECT chunk_id, vector FROM embeddings").all();
+    const map = new Map();
+    for (const row of rows) {
+        if (row.vector) {
+            map.set(row.chunk_id, Array.from(new Float32Array(row.vector.buffer, row.vector.byteOffset, row.vector.byteLength / 4)));
+        }
+    }
+    return map;
+}
+// ============================================
+// Graph Storage
+// ============================================
+/** Save the dependency graph */
+function saveGraph(graph) {
+    const db = getDb();
+    db.prepare("DELETE FROM graph").run();
+    const stmt = db.prepare("INSERT INTO graph (source_file, target_file) VALUES (?, ?)");
+    const tx = db.transaction((g) => {
+        for (const [source, targets] of Object.entries(g)) {
+            for (const target of targets) {
+                stmt.run(source, target);
+            }
+        }
+    });
+    tx(graph);
+}
+/** Load the dependency graph */
+function loadGraph() {
+    const db = getDb();
+    const rows = db.prepare("SELECT source_file, target_file FROM graph").all();
+    const graph = {};
+    for (const row of rows) {
+        if (!graph[row.source_file])
+            graph[row.source_file] = [];
+        graph[row.source_file].push(row.target_file);
+    }
+    return graph;
+}
+// ============================================
+// Meta Storage
+// ============================================
+function setMeta(key, value) {
+    const db = getDb();
+    db.prepare("INSERT OR REPLACE INTO meta (key, value) VALUES (?, ?)").run(key, value);
+}
+function getMeta(key) {
+    const db = getDb();
+    const row = db.prepare("SELECT value FROM meta WHERE key = ?").get(key);
+    return row?.value ?? null;
+}
+// ============================================
+// File Tracking (for incremental re-indexing)
+// ============================================
+/** Get stored mtime for a file */
+function getFileMtime(filePath) {
+    const db = getDb();
+    const row = db.prepare("SELECT mtime FROM files WHERE path = ?").get(filePath);
+    return row?.mtime ?? null;
+}
+/** Upsert file tracking info */
+function upsertFile(filePath, mtime, hash) {
+    const db = getDb();
+    db.prepare("INSERT OR REPLACE INTO files (path, mtime, hash) VALUES (?, ?, ?)").run(filePath, mtime, hash);
+}
+/** Get all tracked file paths */
+function getTrackedFiles() {
+    const db = getDb();
+    return db.prepare("SELECT path FROM files").all().map((r) => r.path);
+}
+/** Remove a file and all its chunks/embeddings */
+function removeFile(filePath) {
+    const db = getDb();
+    // Get chunk IDs for this file
+    const chunkIds = db.prepare("SELECT id FROM chunks WHERE file = ?").all(filePath).map((r) => r.id);
+    // Delete embeddings for those chunks
+    if (chunkIds.length > 0) {
+        const placeholders = chunkIds.map(() => "?").join(",");
+        db.prepare(`DELETE FROM embeddings WHERE chunk_id IN (${placeholders})`).run(...chunkIds);
+    }
+    // Delete chunks
+    db.prepare("DELETE FROM chunks WHERE file = ?").run(filePath);
+    // Delete file record
+    db.prepare("DELETE FROM files WHERE path = ?").run(filePath);
+}
+/** Remove chunks for a file (before re-inserting updated ones).
+ *  Uses LIKE matching to handle both relative and absolute paths. */
+function removeChunksForFile(filePath) {
+    const db = getDb();
+    // Match by exact path OR by path ending (handles relative vs absolute)
+    const chunkIds = db.prepare("SELECT id FROM chunks WHERE file = ? OR file LIKE ?").all(filePath, `%${filePath.replace(/\\/g, "/")}`).map((r) => r.id);
+    if (chunkIds.length > 0) {
+        const placeholders = chunkIds.map(() => "?").join(",");
+        db.prepare(`DELETE FROM embeddings WHERE chunk_id IN (${placeholders})`).run(...chunkIds);
+        db.prepare(`DELETE FROM chunks WHERE id IN (${placeholders})`).run(...chunkIds);
+    }
+}
+/** Get chunk IDs that have no embedding yet */
+function getChunksWithoutEmbeddings() {
+    const db = getDb();
+    return db.prepare("SELECT c.id FROM chunks c LEFT JOIN embeddings e ON c.id = e.chunk_id WHERE e.chunk_id IS NULL").all().map((r) => r.id);
+}
+/** Clear all data (for full regeneration) */
+function clearAll() {
+    const db = getDb();
+    db.exec("DELETE FROM embeddings; DELETE FROM chunks; DELETE FROM graph; DELETE FROM meta; DELETE FROM files;");
+}
+//# sourceMappingURL=storage.js.map

package/dist/storage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.js","sourceRoot":"","sources":["../src/storage.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsBA,sBAsDC;AAGD,0BAKC;AAOD,kCAeC;AAGD,oCAcC;AAGD,4BAGC;AAGD,oCAGC;AAGD,oCAIC;AAGD,sCAGC;AAOD,0CAIC;AAGD,4CAYC;AAGD,oCAKC;AAGD,4CAUC;AAOD,8BAcC;AAGD,8BASC;AAMD,0BAGC;AAED,0BAIC;AAOD,oCAIC;AAGD,gCAGC;AAGD,0CAGC;AAGD,gCAaC;AAID,kDAWC;AAGD,gEAKC;AAGD,4BAGC;AAvTD,oEAAsC;AACtC,uCAAyB;AACzB,2CAA6B;AAG7B;;;;;;;;;GASG;AAEH,MAAM,SAAS,GAAG,UAAU,CAAC;AAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;AAEnD,IAAI,GAAG,GAA6B,IAAI,CAAC;AAEzC,4CAA4C;AAC5C,SAAgB,KAAK;IACnB,IAAI,GAAG;QAAE,OAAO,GAAG,CAAC;IAEpB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,GAAG,GAAG,IAAI,wBAAQ,CAAC,OAAO,CAAC,CAAC;IAC5B,GAAG,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;IACjC,GAAG,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC;IAEnC,gBAAgB;IAChB,GAAG,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuCR,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC;AAED,yBAAyB;AACzB,SAAgB,OAAO;IACrB,IAAI,GAAG,EAAE,CAAC;QACR,GAAG,CAAC,KAAK,EAAE,CAAC;QACZ,GAAG,GAAG,IAAI,CAAC;IACb,CAAC;AACH,CAAC;AAED,+CAA+C;AAC/C,gBAAgB;AAChB,+CAA+C;AAE/C,uCAAuC;AACvC,SAAgB,WAAW,CAAC,KAAoC;IAC9D,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,EAAE,CAAC,OAAO,CAAC;;;GAGV,CAAC,CAAC,GAAG,CACJ,KAAK,CAAC,EAAE,EACR,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,IAAI,IAAI,EAAE,EAChB,KAAK,CAAC,OAAO,IAAI,EAAE,CACpB,CAAC;AACJ,CAAC;AAED,wCAAwC;AACxC,SAAgB,YAAY,CAAC,MAAyC;IACpE,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;;GAGvB,CAAC,CAAC;IAEH,MAAM,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,KAAoB,EAAE,EAAE;QACjD,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAC7F,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,MAAM,CAAC,CAAC;AACb,CAAC;AAED,wBAAwB;AACxB,SAAgB,QAAQ,CAAC,EAAU;IACjC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,OAAO,EAAE,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAQ,CAAC;AACxE,CAAC;AAED,qBAAqB;AACrB,SAAgB,YAAY;IAC1B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,OAAO,EAAE,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,GAAG,EAAW,CAAC;AAC3D,CAAC;AAED,wCAAwC;AACxC,SAAgB,YAAY,CAAC,EAAU;IACrC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAQ,CAAC;IAC9E,OAAO,GAAG,EAAE,IAAI,IAAI,IAAI,CAAC;AAC3B,CAAC;AAED,sBAAsB;AACtB,SAAgB,aAAa;IAC3B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,OAAQ,EAAE,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC,GAAG,EAAU,CAAC,KAAK,CAAC;AACjF,CAAC;AAED,+CAA+C;AAC/C,2CAA2C;AAC3C,+CAA+C;AAE/C,iDAAiD;AACjD,SAAgB,eAAe,CAAC,OAAe,EAAE,MAAgB;IAC/D,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;IACzD,EAAE,CAAC,OAAO,CAAC,oEAAoE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;AACrG,CAAC;AAED,8BAA8B;AAC9B,SAAgB,gBAAgB,CAAC,KAA8C;IAC7E,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,oEAAoE,CAAC,CAAC;IAE9F,MAAM,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,OAAqB,EAAE,EAAE;QAClD,KAAK,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,OAAO,EAAE,CAAC;YAC1C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;YACzD,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,KAAK,CAAC,CAAC;AACZ,CAAC;AAED,8BAA8B;AAC9B,SAAgB,YAAY,CAAC,OAAe;IAC1C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,kDAAkD,CAAC,CAAC,GAAG,CAAC,OAAO,CAAQ,CAAC;IAC/F,IAAI,CAAC,GAAG,EAAE,MAAM;QAAE,OAAO,IAAI,CAAC;IAC9B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;AAC3G,CAAC;AAED,kCAAkC;AAClC,SAAgB,gBAAgB;IAC9B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,EAAW,CAAC;IAClF,MAAM,GAAG,GAAG,IAAI,GAAG,EAAoB,CAAC;IACxC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3H,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,+CAA+C;AAC/C,gBAAgB;AAChB,+CAA+C;AAE/C,gCAAgC;AAChC,SAAgB,SAAS,CAAC,KAAsB;IAC9C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,GAAG,EAAE,CAAC;IAEtC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,4DAA4D,CAAC,CAAC;IACtF,MAAM,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,CAAkB,EAAE,EAAE;QAC/C,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YAClD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,KAAK,CAAC,CAAC;AACZ,CAAC;AAED,gCAAgC;AAChC,SAAgB,SAAS;IACvB,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAC,GAAG,EAAW,CAAC;IACrF,MAAM,KAAK,GAAoB,EAAE,CAAC;IAClC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC;YAAE,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;QACzD,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,+CAA+C;AAC/C,eAAe;AACf,+CAA+C;AAE/C,SAAgB,OAAO,CAAC,GAAW,EAAE,KAAa;IAChD,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,EAAE,CAAC,OAAO,CAAC,wDAAwD,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;AACvF,CAAC;AAED,SAAgB,OAAO,CAAC,GAAW;IACjC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAQ,CAAC;IAC/E,OAAO,GAAG,EAAE,KAAK,IAAI,IAAI,CAAC;AAC5B,CAAC;AAED,+CAA+C;AAC/C,8CAA8C;AAC9C,+CAA+C;AAE/C,kCAAkC;AAClC,SAAgB,YAAY,CAAC,QAAgB;IAC3C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,wCAAwC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAQ,CAAC;IACtF,OAAO,GAAG,EAAE,KAAK,IAAI,IAAI,CAAC;AAC5B,CAAC;AAED,gCAAgC;AAChC,SAAgB,UAAU,CAAC,QAAgB,EAAE,KAAa,EAAE,IAAY;IACtE,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,EAAE,CAAC,OAAO,CAAC,mEAAmE,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;AAC7G,CAAC;AAED,iCAAiC;AACjC,SAAgB,eAAe;IAC7B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,OAAQ,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AAClF,CAAC;AAED,kDAAkD;AAClD,SAAgB,UAAU,CAAC,QAAgB;IACzC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,8BAA8B;IAC9B,MAAM,QAAQ,GAAI,EAAE,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC9G,qCAAqC;IACrC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,EAAE,CAAC,OAAO,CAAC,6CAA6C,YAAY,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IAC5F,CAAC;IACD,gBAAgB;IAChB,EAAE,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9D,qBAAqB;IACrB,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAC/D,CAAC;AAED;qEACqE;AACrE,SAAgB,mBAAmB,CAAC,QAAgB;IAClD,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,uEAAuE;IACvE,MAAM,QAAQ,GAAI,EAAE,CAAC,OAAO,CAC1B,qDAAqD,CACtD,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC/E,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,EAAE,CAAC,OAAO,CAAC,6CAA6C,YAAY,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC1F,EAAE,CAAC,OAAO,CAAC,mCAAmC,YAAY,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IAClF,CAAC;AACH,CAAC;AAED,+CAA+C;AAC/C,SAAgB,0BAA0B;IACxC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,OAAQ,EAAE,CAAC,OAAO,CAChB,gGAAgG,CACjG,CAAC,GAAG,EAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;AACrC,CAAC;AAED,6CAA6C;AAC7C,SAAgB,QAAQ;IACtB,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,EAAE,CAAC,IAAI,CAAC,qGAAqG,CAAC,CAAC;AACjH,CAAC"}

package/dist/tokenBudget.d.ts

@@ -0,0 +1,52 @@
+/**
+ * Token Budget — shared token estimation and packing logic.
+ *
+ * Used by CLI, REST server, and MCP server to ensure consistent
+ * token-aware context packing everywhere.
+ */
+/**
+ * Estimate token count for a string.
+ *
+ * Uses a refined heuristic based on OpenAI's tokenizer patterns:
+ *   - Code tokens average ~3.5 chars (shorter than English prose ~4.3)
+ *   - Identifiers like `getUserSession` count as 2-3 tokens
+ *   - Punctuation (brackets, colons, commas) each count as 1 token
+ *   - Whitespace is mostly free (merged with adjacent tokens)
+ *
+ * Accuracy: within ~15% of tiktoken for code, without the dependency.
+ */
+export declare function estimateTokens(text: string): number;
+/** Get the recommended token budget for a model */
+export declare function getTokenBudget(model?: string): number;
+export interface PackableEntry {
+    text: string;
+    score: number;
+    source: string;
+    [key: string]: any;
+}
+export interface PackResult {
+    entries: PackableEntry[];
+    totalTokens: number;
+    budgetUsed: number;
+    dropped: number;
+}
+/**
+ * Pack entries into a token budget, prioritizing by score.
+ *
+ * Strategy:
+ *   1. Sort by score (highest first) — most relevant entries get priority
+ *   2. Graph entries get a 1.5x score boost (architecture is always important)
+ *   3. Greedily add entries until budget is exhausted
+ *   4. Report how much budget was used and how many entries were dropped
+ */
+export declare function packEntries(entries: PackableEntry[], maxTokens: number): PackResult;
+/**
+ * Pack a raw text output into a token budget.
+ * Truncates lines from the bottom if over budget.
+ */
+export declare function packText(text: string, maxTokens: number): {
+    text: string;
+    tokens: number;
+    truncated: boolean;
+};
+//# sourceMappingURL=tokenBudget.d.ts.map

package/dist/tokenBudget.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenBudget.d.ts","sourceRoot":"","sources":["../src/tokenBudget.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH;;;;;;;;;;GAUG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CA4BnD;AAyDD,mDAAmD;AACnD,wBAAgB,cAAc,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,CAOrD;AAMD,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,aAAa,EAAE,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;;;;;;GAQG;AACH,wBAAgB,WAAW,CACzB,OAAO,EAAE,aAAa,EAAE,EACxB,SAAS,EAAE,MAAM,GAChB,UAAU,CA4BZ;AAED;;;GAGG;AACH,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,OAAO,CAAA;CAAE,CAqB9G"}

package/dist/tokenBudget.js

@@ -0,0 +1,175 @@
+"use strict";
+/**
+ * Token Budget — shared token estimation and packing logic.
+ *
+ * Used by CLI, REST server, and MCP server to ensure consistent
+ * token-aware context packing everywhere.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.estimateTokens = estimateTokens;
+exports.getTokenBudget = getTokenBudget;
+exports.packEntries = packEntries;
+exports.packText = packText;
+// ============================================
+// Token Estimation
+// ============================================
+/**
+ * Estimate token count for a string.
+ *
+ * Uses a refined heuristic based on OpenAI's tokenizer patterns:
+ *   - Code tokens average ~3.5 chars (shorter than English prose ~4.3)
+ *   - Identifiers like `getUserSession` count as 2-3 tokens
+ *   - Punctuation (brackets, colons, commas) each count as 1 token
+ *   - Whitespace is mostly free (merged with adjacent tokens)
+ *
+ * Accuracy: within ~15% of tiktoken for code, without the dependency.
+ */
+function estimateTokens(text) {
+    if (!text)
+        return 0;
+    // Count code-specific token patterns
+    let tokens = 0;
+    // Split on whitespace to get "words"
+    const words = text.split(/\s+/).filter((w) => w.length > 0);
+    for (const word of words) {
+        if (word.length <= 1) {
+            tokens += 1; // Single chars (punctuation, operators)
+        }
+        else if (word.length <= 4) {
+            tokens += 1; // Short words/identifiers
+        }
+        else if (word.length <= 10) {
+            tokens += 2; // Medium identifiers
+        }
+        else if (word.length <= 20) {
+            tokens += 3; // Long identifiers like "getUserSessionDetails"
+        }
+        else {
+            tokens += Math.ceil(word.length / 6); // Very long strings
+        }
+    }
+    // Add overhead for structural tokens (newlines, indentation)
+    const newlines = (text.match(/\n/g) || []).length;
+    tokens += Math.ceil(newlines * 0.5);
+    return Math.max(tokens, 1);
+}
+// ============================================
+// Model-Aware Token Budgets
+// ============================================
+/**
+ * Pre-configured token budgets for context injection.
+ * These are NOT the model's full context window — they're the
+ * recommended amount of codebase context to inject alongside
+ * the user's prompt and conversation history.
+ *
+ * Rule of thumb: use ~5-10% of the model's context window for
+ * codebase context, leaving room for conversation + response.
+ */
+const MODEL_TOKEN_BUDGETS = {
+    // OpenAI
+    "gpt-4": 6000,
+    "gpt-4o": 8000,
+    "gpt-4o-mini": 4000,
+    "gpt-4-turbo": 8000,
+    "gpt-3.5-turbo": 3000,
+    "o1": 8000,
+    "o1-mini": 4000,
+    "o3": 10000,
+    "o3-mini": 6000,
+    // Anthropic
+    "claude-3-opus": 10000,
+    "claude-3-sonnet": 8000,
+    "claude-3-haiku": 4000,
+    "claude-3.5-sonnet": 10000,
+    "claude-4-opus": 12000,
+    "claude-4-sonnet": 10000,
+    // Google
+    "gemini-pro": 6000,
+    "gemini-1.5-pro": 10000,
+    "gemini-2": 10000,
+    // Open source
+    "llama-3-8b": 2000,
+    "llama-3-70b": 4000,
+    "llama-3.1-405b": 6000,
+    "mistral-7b": 2000,
+    "mixtral": 4000,
+    "codellama-34b": 3000,
+    "deepseek-coder": 4000,
+    "deepseek-v2": 6000,
+    "qwen-2.5-coder": 4000,
+    "phi-3": 2000,
+    "starcoder2": 3000,
+    // Default
+    "default": 4000,
+};
+/** Get the recommended token budget for a model */
+function getTokenBudget(model) {
+    if (!model)
+        return MODEL_TOKEN_BUDGETS["default"];
+    const lower = model.toLowerCase();
+    for (const [key, budget] of Object.entries(MODEL_TOKEN_BUDGETS)) {
+        if (lower.includes(key))
+            return budget;
+    }
+    return MODEL_TOKEN_BUDGETS["default"];
+}
+/**
+ * Pack entries into a token budget, prioritizing by score.
+ *
+ * Strategy:
+ *   1. Sort by score (highest first) — most relevant entries get priority
+ *   2. Graph entries get a 1.5x score boost (architecture is always important)
+ *   3. Greedily add entries until budget is exhausted
+ *   4. Report how much budget was used and how many entries were dropped
+ */
+function packEntries(entries, maxTokens) {
+    // Boost graph entries and sort by score
+    const scored = entries.map((e) => ({
+        ...e,
+        _sortScore: e.source === "graph" ? e.score * 1.5 : e.score,
+    }));
+    scored.sort((a, b) => b._sortScore - a._sortScore);
+    const packed = [];
+    let totalTokens = 0;
+    let dropped = 0;
+    for (const entry of scored) {
+        const entryTokens = estimateTokens(entry.text);
+        if (totalTokens + entryTokens > maxTokens) {
+            dropped++;
+            continue; // Skip but keep trying smaller entries
+        }
+        packed.push(entry);
+        totalTokens += entryTokens;
+    }
+    return {
+        entries: packed,
+        totalTokens,
+        budgetUsed: Math.round((totalTokens / maxTokens) * 100),
+        dropped,
+    };
+}
+/**
+ * Pack a raw text output into a token budget.
+ * Truncates lines from the bottom if over budget.
+ */
+function packText(text, maxTokens) {
+    const tokens = estimateTokens(text);
+    if (tokens <= maxTokens) {
+        return { text, tokens, truncated: false };
+    }
+    // Truncate line by line from the bottom
+    const lines = text.split("\n");
+    let packed = "";
+    let count = 0;
+    for (const line of lines) {
+        const lineTokens = estimateTokens(line);
+        if (count + lineTokens > maxTokens - 5) { // Reserve 5 tokens for truncation notice
+            break;
+        }
+        packed += line + "\n";
+        count += lineTokens;
+    }
+    packed += `\n... (truncated, ${tokens - count} tokens omitted)`;
+    return { text: packed, tokens: count + 5, truncated: true };
+}
+//# sourceMappingURL=tokenBudget.js.map

package/dist/tokenBudget.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenBudget.js","sourceRoot":"","sources":["../src/tokenBudget.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAiBH,wCA4BC;AA0DD,wCAOC;AA6BD,kCA+BC;AAMD,4BAqBC;AAnMD,+CAA+C;AAC/C,mBAAmB;AACnB,+CAA+C;AAE/C;;;;;;;;;;GAUG;AACH,SAAgB,cAAc,CAAC,IAAY;IACzC,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,CAAC;IAEpB,qCAAqC;IACrC,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,qCAAqC;IACrC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE5D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,CAAC,CAAC,wCAAwC;QACvD,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,CAAC,CAAC,CAAC,0BAA0B;QACzC,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,CAAC,CAAC,CAAC,qBAAqB;QACpC,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,CAAC,CAAC,CAAC,gDAAgD;QAC/D,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,oBAAoB;QAC5D,CAAC;IACH,CAAC;IAED,6DAA6D;IAC7D,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;IAClD,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,GAAG,CAAC,CAAC;IAEpC,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC;AAED,+CAA+C;AAC/C,4BAA4B;AAC5B,+CAA+C;AAE/C;;;;;;;;GAQG;AACH,MAAM,mBAAmB,GAA2B;IAClD,SAAS;IACT,OAAO,EAAE,IAAI;IACb,QAAQ,EAAE,IAAI;IACd,aAAa,EAAE,IAAI;IACnB,aAAa,EAAE,IAAI;IACnB,eAAe,EAAE,IAAI;IACrB,IAAI,EAAE,IAAI;IACV,SAAS,EAAE,IAAI;IACf,IAAI,EAAE,KAAK;IACX,SAAS,EAAE,IAAI;IAEf,YAAY;IACZ,eAAe,EAAE,KAAK;IACtB,iBAAiB,EAAE,IAAI;IACvB,gBAAgB,EAAE,IAAI;IACtB,mBAAmB,EAAE,KAAK;IAC1B,eAAe,EAAE,KAAK;IACtB,iBAAiB,EAAE,KAAK;IAExB,SAAS;IACT,YAAY,EAAE,IAAI;IAClB,gBAAgB,EAAE,KAAK;IACvB,UAAU,EAAE,KAAK;IAEjB,cAAc;IACd,YAAY,EAAE,IAAI;IAClB,aAAa,EAAE,IAAI;IACnB,gBAAgB,EAAE,IAAI;IACtB,YAAY,EAAE,IAAI;IAClB,SAAS,EAAE,IAAI;IACf,eAAe,EAAE,IAAI;IACrB,gBAAgB,EAAE,IAAI;IACtB,aAAa,EAAE,IAAI;IACnB,gBAAgB,EAAE,IAAI;IACtB,OAAO,EAAE,IAAI;IACb,YAAY,EAAE,IAAI;IAElB,UAAU;IACV,SAAS,EAAE,IAAI;CAChB,CAAC;AAEF,mDAAmD;AACnD,SAAgB,cAAc,CAAC,KAAc;IAC3C,IAAI,CAAC,KAAK;QAAE,OAAO,mBAAmB,CAAC,SAAS,CAAC,CAAC;IAClD,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAClC,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAChE,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,OAAO,MAAM,CAAC;IACzC,CAAC;IACD,OAAO,mBAAmB,CAAC,SAAS,CAAC,CAAC;AACxC,CAAC;AAoBD;;;;;;;;GAQG;AACH,SAAgB,WAAW,CACzB,OAAwB,EACxB,SAAiB;IAEjB,wCAAwC;IACxC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACjC,GAAG,CAAC;QACJ,UAAU,EAAE,CAAC,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK;KAC3D,CAAC,CAAC,CAAC;IACJ,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;IAEnD,MAAM,MAAM,GAAoB,EAAE,CAAC;IACnC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,OAAO,GAAG,CAAC,CAAC;IAEhB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,WAAW,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,WAAW,GAAG,WAAW,GAAG,SAAS,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC;YACV,SAAS,CAAC,uCAAuC;QACnD,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnB,WAAW,IAAI,WAAW,CAAC;IAC7B,CAAC;IAED,OAAO;QACL,OAAO,EAAE,MAAM;QACf,WAAW;QACX,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,GAAG,CAAC;QACvD,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAgB,QAAQ,CAAC,IAAY,EAAE,SAAiB;IACtD,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IACpC,IAAI,MAAM,IAAI,SAAS,EAAE,CAAC;QACxB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IAC5C,CAAC;IAED,wCAAwC;IACxC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/B,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,KAAK,GAAG,UAAU,GAAG,SAAS,GAAG,CAAC,EAAE,CAAC,CAAC,yCAAyC;YACjF,MAAM;QACR,CAAC;QACD,MAAM,IAAI,IAAI,GAAG,IAAI,CAAC;QACtB,KAAK,IAAI,UAAU,CAAC;IACtB,CAAC;IAED,MAAM,IAAI,qBAAqB,MAAM,GAAG,KAAK,kBAAkB,CAAC;IAChE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,GAAG,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;AAC9D,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,62 @@
+/** Configuration for file scanning */
+export interface ScanConfig {
+    include: string[];
+    ignore: string[];
+}
+/** A code chunk extracted from AST parsing */
+export interface Chunk {
+    /** Unique identifier: file#name */
+    id: string;
+    /** Source file path */
+    file: string;
+    /** Type of code element */
+    type: "function" | "class" | "method" | "arrow-function" | "export" | "interface" | "type-alias" | "constant";
+    /** Name of the code element */
+    name: string;
+    /** Raw source code */
+    code: string;
+    /** Actual code signature (declaration line) — Aider-style */
+    signature?: string;
+    /** MD5 hash of the code (for cache invalidation) */
+    hash?: string;
+    /** Text summary (local or AI-generated) */
+    summary?: string;
+    /** Embedding vector */
+    embedding?: number[];
+}
+/** Cached chunk data stored in .graphra/ (legacy, kept for migration) */
+export interface ChunkCache {
+    [chunkId: string]: {
+        hash: string;
+        summary: string;
+        embedding: number[];
+    };
+}
+/** Dependency graph: file → list of imported files */
+export interface DependencyGraph {
+    [file: string]: string[];
+}
+/** Context output assembled for prompt building */
+export interface ContextResult {
+    /** Actual code signatures (Aider-style) — the primary context */
+    entries: {
+        chunkId: string;
+        file: string;
+        name: string;
+        signature: string;
+        summary: string;
+        importance: number;
+    }[];
+}
+/** Final prompt structure */
+export interface PromptData {
+    arch: string;
+    context: string;
+    task: string;
+}
+/** Search result from hybrid search */
+export interface SearchResult {
+    chunk: Chunk;
+    score: number;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAIA,sCAAsC;AACtC,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,8CAA8C;AAC9C,MAAM,WAAW,KAAK;IACpB,mCAAmC;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,uBAAuB;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,2BAA2B;IAC3B,IAAI,EAAE,UAAU,GAAG,OAAO,GAAG,QAAQ,GAAG,gBAAgB,GAAG,QAAQ,GAAG,WAAW,GAAG,YAAY,GAAG,UAAU,CAAC;IAC9G,+BAA+B;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,sBAAsB;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,6DAA6D;IAC7D,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oDAAoD;IACpD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,2CAA2C;IAC3C,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,uBAAuB;IACvB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,yEAAyE;AACzE,MAAM,WAAW,UAAU;IACzB,CAAC,OAAO,EAAE,MAAM,GAAG;QACjB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,EAAE,CAAC;KACrB,CAAC;CACH;AAED,sDAAsD;AACtD,MAAM,WAAW,eAAe;IAC9B,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAC1B;AAED,mDAAmD;AACnD,MAAM,WAAW,aAAa;IAC5B,iEAAiE;IACjE,OAAO,EAAE;QACP,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,EAAE,MAAM,CAAC;KACpB,EAAE,CAAC;CACL;AAED,6BAA6B;AAC7B,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,uCAAuC;AACvC,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf"}

package/dist/types.js

@@ -0,0 +1,6 @@
+"use strict";
+// ============================================
+// Graphra — Core Type Definitions (v2)
+// ============================================
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":";AAAA,+CAA+C;AAC/C,uCAAuC;AACvC,+CAA+C"}

package/dist/utils/hash.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Generate an MD5 hash of the given content.
+ * Used for cache invalidation of chunk summaries.
+ */
+export declare function md5(content: string): string;
+//# sourceMappingURL=hash.d.ts.map

package/dist/utils/hash.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hash.d.ts","sourceRoot":"","sources":["../../src/utils/hash.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,wBAAgB,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE3C"}