grepmax 0.26.43 → 0.26.44

package/dist/lib/daemon/vector-handler.js

@@ -1,4 +1,20 @@
 "use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runSimilar = runSimilar;
+exports.handleVectorSimilar = handleVectorSimilar;
+exports.runSurprises = runSurprises;
+exports.handleVectorSurprises = handleVectorSurprises;
+exports.registerVectorVerbs = registerVectorVerbs;
+const query_timeout_1 = require("../utils/query-timeout");
 /**
  * Daemon-side handlers for the `vector.*` read verbs — the two commands whose
  * store work is a vector scan rather than a row select.
@@ -16,21 +32,6 @@
  * the verb and the command's in-process fallback call, so the two paths cannot
  * produce different output.
  */
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.runSimilar = runSimilar;
-exports.handleVectorSimilar = handleVectorSimilar;
-exports.runSurprises = runSurprises;
-exports.handleVectorSurprises = handleVectorSurprises;
-exports.registerVectorVerbs = registerVectorVerbs;
 const surprising_connections_1 = require("../analysis/surprising-connections");
 const ann_config_1 = require("../store/ann-config");
 const arrow_1 = require("../utils/arrow");
@@ -55,7 +56,7 @@ function runSimilar(deps, req) {
             ? `path = '${(0, filter_builder_1.escapeSqlString)(req.absPath)}'`
             : `array_contains(defined_symbols, '${(0, filter_builder_1.escapeSqlString)((_b = req.symbol) !== null && _b !== void 0 ? _b : "")}') AND ${(0, filter_builder_1.pathStartsWith)(`${req.projectRoot}/`)}`)
             .limit(1)
-            .toArray();
+            .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
         if (sourceRows.length === 0)
             return { status: "not-found" };
         const source = sourceRows[0];
@@ -76,7 +77,7 @@ function runSimilar(deps, req) {
         ])
             .where(pathScope)
             .limit(req.limit + 5) // fetch extra to account for self-filtering
-            .toArray();
+            .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
         // Filter out self and apply threshold
         const filtered = results.filter((raw) => {
             const r = raw;

package/dist/lib/graph/graph-builder.js

@@ -8,6 +8,13 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GraphBuilder = void 0;
 const languages_1 = require("../core/languages");
@@ -83,7 +90,7 @@ class GraphBuilder {
             ])
                 .where(this.scopeWhere(`(array_contains(referenced_symbols, '${escaped}') OR array_contains(type_referenced_symbols, '${escaped}'))`))
                 .limit(100)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             // Cross-language phantom-edge guard. When the anchor family is known, drop
             // callers from a *different known* family; callers we can't classify (unknown
             // extension) are kept so a real edge is never lost. No anchor → no filter.
@@ -133,7 +140,7 @@ class GraphBuilder {
             ])
                 .where(this.scopeWhere(`array_contains(defined_symbols, '${escaped}')`))
                 .limit(1)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             if (rows.length === 0)
                 return [];
             const record = rows[0];
@@ -145,7 +152,7 @@ class GraphBuilder {
                     .query()
                     .select(["defined_symbols"])
                     .where(this.scopeWhere(`path = '${(0, filter_builder_1.escapeSqlString)(String(record.path))}'`))
-                    .toArray();
+                    .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
                 localDefinitions = localRows.flatMap((row) => (0, arrow_1.toArr)(row.defined_symbols));
             }
             return (0, arrow_1.toArr)(record.referenced_symbols).filter((name) => {
@@ -183,7 +190,7 @@ class GraphBuilder {
             ])
                 .where(this.scopeWhere(`array_contains(defined_symbols, '${escaped}')`))
                 .limit(1)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             const center = centerRows.length > 0
                 ? this.mapRowToNode(centerRows[0], symbol, "center")
                 : null;
@@ -214,7 +221,7 @@ class GraphBuilder {
                     "complexity",
                 ])
                     .limit(25)
-                    .toArray();
+                    .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
                 const memberNames = (0, arrow_1.toArr)((_a = centerRows[0]) === null || _a === void 0 ? void 0 : _a.member_referenced_symbols);
                 const content = String((_c = (_b = centerRows[0]) === null || _b === void 0 ? void 0 : _b.content) !== null && _c !== void 0 ? _c : "");
                 const bareCall = new RegExp(`(?<![\\w.$])${name.replace(/[.*+?^${}()|[\]\\]/g, "\\$&")}\\s*\\(`).test(content);
@@ -266,20 +273,30 @@ class GraphBuilder {
     }
     getImporters(symbol) {
         return __awaiter(this, void 0, void 0, function* () {
+            var _a, e_1, _b, _c;
             const table = yield this.db.ensureTable();
             const escaped = (0, filter_builder_1.escapeSqlString)(symbol);
-            // No .limit() here: LIKE + limit deadlocks in @lancedb 0.27.x when more
-            // rows match than the limit (verified). Unlimited scan is fast; cap in JS.
-            const rows = yield (0, query_timeout_1.withQueryTimeout)(table
+            const rows = (0, query_timeout_1.streamQueryRows)(table
                 .query()
                 .select(["path"])
-                .where(this.scopeWhere(`content LIKE '%import%${escaped}%'`))
-                .toArray(), `content LIKE %import%${symbol}% (importers)`);
+                .where(this.scopeWhere(`content LIKE '%import%${escaped}%'`)), `content LIKE %import%${symbol}% (importers)`);
             const files = new Set();
-            for (const row of rows) {
-                files.add(String(row.path || ""));
-                if (files.size >= 100)
-                    break;
+            try {
+                for (var _d = true, rows_1 = __asyncValues(rows), rows_1_1; rows_1_1 = yield rows_1.next(), _a = rows_1_1.done, !_a; _d = true) {
+                    _c = rows_1_1.value;
+                    _d = false;
+                    const row = _c;
+                    files.add(String(row.path || ""));
+                    if (files.size >= 100)
+                        break;
+                }
+            }
+            catch (e_1_1) { e_1 = { error: e_1_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = rows_1.return)) yield _b.call(rows_1);
+                }
+                finally { if (e_1) throw e_1.error; }
             }
             return Array.from(files);
         });
@@ -411,7 +428,7 @@ class GraphBuilder {
                 // No anchor → keep the cheap single-row fetch. With one, pull a few
                 // candidates so we can pick the same-family definition instead of guessing.
                 .limit(anchorFamily ? 25 : 1)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             if (rows.length === 0)
                 return null;
             let r = rows[0];
@@ -455,7 +472,7 @@ class GraphBuilder {
                 .select(["path", "defined_symbols", "referenced_symbols"])
                 .where(this.scopeWhere(`(${orClause})`))
                 .limit(100000)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             const toArray = (val) => {
                 if (val && typeof val.toArray === "function")
                     return val.toArray();

package/dist/lib/graph/impact.js

@@ -45,7 +45,7 @@ function resolveTargetSymbols(target, vectorDb, projectRoot) {
                 .query()
                 .select(["defined_symbols"])
                 .where(`path = '${(0, filter_builder_1.escapeSqlString)(absPath)}'`)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             const symbols = new Set();
             for (const chunk of chunks) {
                 for (const s of (0, arrow_1.toArr)(chunk.defined_symbols)) {
@@ -88,7 +88,7 @@ function resolveSymbolFamilies(symbols, vectorDb, projectRoot, excludePrefixes,
                 .select(["path"])
                 .where(`array_contains(defined_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}') AND ${pathScope}`)
                 .limit(25)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             const file = String(((_a = rows[0]) === null || _a === void 0 ? void 0 : _a.path) || "");
             families.set(sym, file ? (0, languages_1.languageFamilyForPath)(file) : null);
         }
@@ -117,7 +117,7 @@ function expandFileSymbols(symbols, vectorDb, projectRoot, excludePrefixes) {
             .select(["path"])
             .where(where)
             .limit(1)
-            .toArray();
+            .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
         if (defRows.length === 0)
             return symbols;
         const filePath = String(defRows[0].path);
@@ -126,7 +126,7 @@ function expandFileSymbols(symbols, vectorDb, projectRoot, excludePrefixes) {
             .query()
             .select(["defined_symbols"])
             .where(`path = '${(0, filter_builder_1.escapeSqlString)(filePath)}'`)
-            .toArray();
+            .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
         const expanded = new Set(symbols);
         for (const row of fileRows) {
             for (const s of (0, arrow_1.toArr)(row.defined_symbols)) {
@@ -211,13 +211,14 @@ function findImportFallbackTests(originalSymbols, vectorDb, projectRoot, exclude
         // that mentions them, drowning the answer in false positives.
         for (const sym of originalSymbols) {
             const family = (_a = symbolFamilies === null || symbolFamilies === void 0 ? void 0 : symbolFamilies.get(sym)) !== null && _a !== void 0 ? _a : null;
-            // No .limit() here: LIKE + limit deadlocks in @lancedb 0.27.x when more
-            // rows match than the limit (verified). Unlimited scan is fast; cap in JS.
+            // The pinned SDK passes the native LIKE+limit regression; bound rows at
+            // the same cap the old unlimited scan applied after materializing them.
             const rows = yield (0, query_timeout_1.withQueryTimeout)(table
                 .query()
                 .select(["path"])
                 .where(`content LIKE '%${(0, filter_builder_1.escapeSqlString)(sym)}%' AND ${pathScope}`)
-                .toArray(), `content LIKE %${sym}% (test fallback)`);
+                .limit(500)
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS), `content LIKE %${sym}% (test fallback)`);
             for (const row of rows.slice(0, 500)) {
                 const p = String(row.path || "");
                 if (isTestPath(p) && familyMatchesPath(family, p))
@@ -286,7 +287,7 @@ function findDependentsDetailed(symbols_1, vectorDb_1, projectRoot_1, excludePat
                 .select(["path"])
                 .where(`(array_contains(referenced_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}') OR array_contains(type_referenced_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}')) AND ${pathScope}`)
                 .limit(200)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             for (const row of rows) {
                 const p = String(row.path || "");
                 if (excludePaths === null || excludePaths === void 0 ? void 0 : excludePaths.has(p))

package/dist/lib/index/index-config.js

@@ -50,10 +50,10 @@ const fs = __importStar(require("node:fs"));
 const path = __importStar(require("node:path"));
 const config_1 = require("../../config");
 const GLOBAL_CONFIG_PATH = path.join(config_1.PATHS.globalRoot, "config.json");
-function readGlobalConfig() {
+function readGlobalConfig(home) {
     const defaultEmbedMode = process.arch === "arm64" && process.platform === "darwin" ? "gpu" : "cpu";
     try {
-        const raw = fs.readFileSync(GLOBAL_CONFIG_PATH, "utf-8");
+        const raw = fs.readFileSync(home ? path.join(home, "config.json") : GLOBAL_CONFIG_PATH, "utf-8");
         const parsed = JSON.parse(raw);
         // Ensure embedMode has a default even if missing from stored config
         if (!parsed.embedMode)

package/dist/lib/llm/config.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getLlmConfig = getLlmConfig;
+const node_net_1 = require("node:net");
 const DEFAULT_MODEL = "/Volumes/External/models/huggingface/hub/models--unsloth--Qwen3.5-35B-A3B-GGUF/Qwen3.5-35B-A3B-Q4_K_M.gguf";
 function envInt(key, fallback) {
     const v = process.env[key];
@@ -11,10 +12,15 @@ function envInt(key, fallback) {
 }
 function getLlmConfig() {
     var _a, _b, _c, _d;
+    const requestedHost = (_a = process.env.GMAX_LLM_HOST) !== null && _a !== void 0 ? _a : "127.0.0.1";
+    const host = requestedHost === "localhost" ? "127.0.0.1" : requestedHost;
+    if (!(host === "::1" || ((0, node_net_1.isIP)(host) === 4 && host.startsWith("127.")))) {
+        throw new Error("gmax is local-only: GMAX_LLM_HOST must be a loopback address");
+    }
     return {
-        model: (_a = process.env.GMAX_LLM_MODEL) !== null && _a !== void 0 ? _a : DEFAULT_MODEL,
-        binary: (_b = process.env.GMAX_LLM_BINARY) !== null && _b !== void 0 ? _b : "llama-server",
-        host: (_c = process.env.GMAX_LLM_HOST) !== null && _c !== void 0 ? _c : "127.0.0.1",
+        model: (_b = process.env.GMAX_LLM_MODEL) !== null && _b !== void 0 ? _b : DEFAULT_MODEL,
+        binary: (_c = process.env.GMAX_LLM_BINARY) !== null && _c !== void 0 ? _c : "llama-server",
+        host,
         port: envInt("GMAX_LLM_PORT", 8079),
         ctxSize: envInt("GMAX_LLM_CTX_SIZE", 16384),
         ngl: envInt("GMAX_LLM_NGL", 99),

package/dist/lib/llm/tools.js

@@ -49,6 +49,7 @@ const path = __importStar(require("node:path"));
 const impact_1 = require("../graph/impact");
 const arrow_1 = require("../utils/arrow");
 const filter_builder_1 = require("../utils/filter-builder");
+const query_timeout_1 = require("../utils/query-timeout");
 // ---------------------------------------------------------------------------
 // Tool definitions (OpenAI function-calling format)
 // ---------------------------------------------------------------------------
@@ -247,7 +248,7 @@ function executePeek(args, ctx) {
             .select(["is_exported", "start_line", "end_line"])
             .where(`array_contains(defined_symbols, '${(0, filter_builder_1.escapeSqlString)(symbol)}') AND ${(0, filter_builder_1.pathStartsWith)(prefix)}`)
             .limit(1)
-            .toArray();
+            .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
         const exported = metaRows.length > 0 && Boolean(metaRows[0].is_exported);
         const startLine = metaRows.length > 0
             ? Number(metaRows[0].start_line || 0)
@@ -333,7 +334,7 @@ function executeRelated(args, ctx) {
             "type_referenced_symbols",
         ])
             .where(`path = '${(0, filter_builder_1.escapeSqlString)(absPath)}'`)
-            .toArray();
+            .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
         if (fileChunks.length === 0)
             return "(file not indexed)";
         const definedHere = new Set();
@@ -356,7 +357,7 @@ function executeRelated(args, ctx) {
                 .select(["path"])
                 .where(`array_contains(defined_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}')`)
                 .limit(3)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             for (const row of rows) {
                 const p = String(row.path || "");
                 if (p === absPath)
@@ -372,7 +373,7 @@ function executeRelated(args, ctx) {
                 .select(["path"])
                 .where(`(array_contains(referenced_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}') OR array_contains(type_referenced_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}'))`)
                 .limit(20)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             for (const row of rows) {
                 const p = String(row.path || "");
                 if (p === absPath)

package/dist/lib/search/pagerank.js

@@ -56,6 +56,7 @@ const fs = __importStar(require("node:fs"));
 const path = __importStar(require("node:path"));
 const config_1 = require("../../config");
 const filter_builder_1 = require("../utils/filter-builder");
+const query_timeout_1 = require("../utils/query-timeout");
 const DEFAULT_DAMPING = 0.85;
 const DEFAULT_MAX_ITER = 50;
 const DEFAULT_TOL = 1e-6;
@@ -195,7 +196,7 @@ function buildGraphFromDb(db, pathPrefix) {
             .select(["defined_symbols", "referenced_symbols"])
             .where((0, filter_builder_1.pathStartsWith)(prefix))
             .limit(exports.PAGERANK_ROW_LIMIT)
-            .toArray();
+            .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
         const nodes = new Set();
         const edges = new Map();
         for (const row of rows) {

package/dist/lib/search/searcher.js

@@ -16,6 +16,7 @@ const config_1 = require("../../config");
 const ann_config_1 = require("../store/ann-config");
 const filter_builder_1 = require("../utils/filter-builder");
 const logger_1 = require("../utils/logger");
+const query_timeout_1 = require("../utils/query-timeout");
 const pool_1 = require("../workers/pool");
 const intent_1 = require("./intent");
 const pagerank_1 = require("./pagerank");
@@ -487,7 +488,7 @@ class Searcher {
             if (whereClause) {
                 vectorQuery = vectorQuery.where(whereClause);
             }
-            const vectorResults = (yield vectorQuery.toArray()).map((r) => (Object.assign({}, r)));
+            const vectorResults = (yield vectorQuery.toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS)).map((r) => (Object.assign({}, r)));
             let ftsResults = [];
             let ftsSearchFailed = false;
             if (this.ftsAvailable) {
@@ -499,7 +500,7 @@ class Searcher {
                     if (whereClause) {
                         ftsQuery = ftsQuery.where(whereClause);
                     }
-                    ftsResults = (yield ftsQuery.toArray()).map((r) => (Object.assign({}, r)));
+                    ftsResults = (yield ftsQuery.toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS)).map((r) => (Object.assign({}, r)));
                 }
                 catch (e) {
                     ftsSearchFailed = true;
@@ -742,7 +743,7 @@ class Searcher {
                         .select(["id", "colbert", "colbert_scale", "doc_token_ids"])
                         .where(`id IN (${rerankIds.join(",")})`)
                         .limit(rerankIds.length)
-                        .toArray();
+                        .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
                     const colbertMap = new Map(colbertRows.map((r) => [r.id, r]));
                     for (const doc of rerankCandidates) {
                         const extra = colbertMap.get(doc.id);
@@ -876,7 +877,7 @@ class Searcher {
                 ])
                     .where(`id IN (${finalIds.join(",")})`)
                     .limit(finalIds.length)
-                    .toArray();
+                    .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
                 const displayMap = new Map(displayRows.map((r) => [r.id, r]));
                 for (const item of diversified) {
                     const extra = displayMap.get(item.record.id);

package/dist/lib/skeleton/retriever.js

@@ -11,6 +11,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getStoredSkeleton = getStoredSkeleton;
 const filter_builder_1 = require("../utils/filter-builder");
+const query_timeout_1 = require("../utils/query-timeout");
 function getStoredSkeleton(db, filePath) {
     return __awaiter(this, void 0, void 0, function* () {
         try {
@@ -21,7 +22,7 @@ function getStoredSkeleton(db, filePath) {
                 .select(["file_skeleton"])
                 .where(`path = '${(0, filter_builder_1.escapeSqlString)(filePath)}' AND is_anchor = true`)
                 .limit(1)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             if (results.length > 0) {
                 const skel = results[0].file_skeleton;
                 if (typeof skel === "string" && skel.length > 0) {

package/dist/lib/store/lance-sdk.js

@@ -0,0 +1,46 @@
+"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;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Session = exports.Index = exports.connect = void 0;
+const path = __importStar(require("node:path"));
+// Source tests use the pinned dev dependency. Compiled consumers use the
+// unchanged official runtime copied at build time, with root native packages.
+const base = path.resolve(__dirname, "../..");
+const sdk = require(path.basename(base) === "src"
+    ? "@lancedb/lancedb"
+    : path.join(base, "vendor", "lancedb", "index.js"));
+exports.connect = sdk.connect;
+exports.Index = sdk.Index;
+exports.Session = sdk.Session;

package/dist/lib/store/vector-db.js

@@ -41,6 +41,13 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VectorDB = exports.DiskPressureError = void 0;
 exports.findStaleLanceTempFiles = findStaleLanceTempFiles;
@@ -48,7 +55,6 @@ exports.sweepStaleLanceTempFiles = sweepStaleLanceTempFiles;
 exports.isLanceCorruptionError = isLanceCorruptionError;
 exports.isMissingTableError = isMissingTableError;
 const fs = __importStar(require("node:fs"));
-const lancedb = __importStar(require("@lancedb/lancedb"));
 const apache_arrow_1 = require("apache-arrow");
 const config_1 = require("../../config");
 const embedding_reuse_1 = require("../index/embedding-reuse");
@@ -56,8 +62,10 @@ const index_config_1 = require("../index/index-config");
 const cleanup_1 = require("../utils/cleanup");
 const filter_builder_1 = require("../utils/filter-builder");
 const logger_1 = require("../utils/logger");
+const query_timeout_1 = require("../utils/query-timeout");
 const ann_config_1 = require("./ann-config");
 const compaction_result_1 = require("./compaction-result");
+const lancedb = __importStar(require("./lance-sdk"));
 const store_lease_1 = require("./store-lease");
 /**
  * Find abandoned Lance scratch files under `<lancedbDir>/ *.lance/data`.
@@ -369,7 +377,11 @@ class VectorDB {
             if (!this.db) {
                 fs.mkdirSync(this.lancedbDir, { recursive: true });
                 this.session = createLanceSession();
-                this.db = yield lancedb.connect(this.lancedbDir, {}, this.session);
+                // 0.38 accepts the legacy third argument in its types but drops it in
+                // the JS wrapper. Put the session in native ConnectionOptions instead.
+                this.db = yield lancedb.connect(this.lancedbDir, {
+                    session: this.session,
+                });
             }
             return this.db;
         });
@@ -1457,7 +1469,11 @@ class VectorDB {
     hasAnyRows() {
         return __awaiter(this, void 0, void 0, function* () {
             const table = yield this.ensureTable();
-            const rows = yield table.query().select(["id"]).limit(1).toArray();
+            const rows = yield table
+                .query()
+                .select(["id"])
+                .limit(1)
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             return rows.length > 0;
         });
     }
@@ -1470,7 +1486,7 @@ class VectorDB {
                 .select(["id"])
                 .where((0, filter_builder_1.pathStartsWith)(prefix))
                 .limit(1)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             return rows.length > 0;
         });
     }
@@ -1488,16 +1504,25 @@ class VectorDB {
     }
     getDistinctPathsForPrefix(pathPrefix) {
         return __awaiter(this, void 0, void 0, function* () {
+            var _a, e_1, _b, _c;
             const table = yield this.ensureTable();
             const prefix = pathPrefix.endsWith("/") ? pathPrefix : `${pathPrefix}/`;
-            const rows = yield table
-                .query()
-                .select(["path"])
-                .where((0, filter_builder_1.pathStartsWith)(prefix))
-                .toArray();
+            const rows = (0, query_timeout_1.streamQueryRows)(table.query().select(["path"]).where((0, filter_builder_1.pathStartsWith)(prefix)), "distinct project paths");
             const unique = new Set();
-            for (const r of rows) {
-                unique.add(String(r.path));
+            try {
+                for (var _d = true, rows_1 = __asyncValues(rows), rows_1_1; rows_1_1 = yield rows_1.next(), _a = rows_1_1.done, !_a; _d = true) {
+                    _c = rows_1_1.value;
+                    _d = false;
+                    const r = _c;
+                    unique.add(String(r.path));
+                }
+            }
+            catch (e_1_1) { e_1 = { error: e_1_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = rows_1.return)) yield _b.call(rows_1);
+                }
+                finally { if (e_1) throw e_1.error; }
             }
             return unique;
         });
@@ -1522,7 +1547,7 @@ class VectorDB {
                 "doc_token_ids",
             ])
                 .where(`path = '${(0, filter_builder_1.escapeSqlString)(filePath)}'`)
-                .toArray();
+                .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
             const reusable = new Map();
             for (const row of rows) {
                 const vector = (0, embedding_reuse_1.toFloat32)(row.vector);
@@ -1556,9 +1581,25 @@ class VectorDB {
     }
     getDistinctFileCount() {
         return __awaiter(this, void 0, void 0, function* () {
+            var _a, e_2, _b, _c;
             const table = yield this.ensureTable();
-            const rows = yield table.query().select(["path"]).toArray();
-            return new Set(rows.map((r) => r.path)).size;
+            const paths = new Set();
+            try {
+                for (var _d = true, _e = __asyncValues((0, query_timeout_1.streamQueryRows)(table.query().select(["path"]), "distinct file count")), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
+                    _c = _f.value;
+                    _d = false;
+                    const row = _c;
+                    paths.add(String(row.path));
+                }
+            }
+            catch (e_2_1) { e_2 = { error: e_2_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
+                }
+                finally { if (e_2) throw e_2.error; }
+            }
+            return paths.size;
         });
     }
     deletePaths(paths) {
@@ -1581,7 +1622,7 @@ class VectorDB {
                         .select(["id"])
                         .where(where)
                         .limit(1)
-                        .toArray();
+                        .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
                     if (existing.length > 0) {
                         yield table.delete(where);
                         this.markWriteCommitted();
@@ -1630,7 +1671,7 @@ class VectorDB {
                         .select(["id"])
                         .where(where)
                         .limit(1)
-                        .toArray();
+                        .toArray(query_timeout_1.QUERY_EXECUTION_OPTIONS);
                     if (existing.length > 0) {
                         yield table.delete(where);
                         this.markWriteCommitted();

package/dist/lib/utils/cross-project.js

@@ -15,13 +15,13 @@ exports.resolveCrossProjectScope = resolveCrossProjectScope;
 exports.projectForPath = projectForPath;
 exports.groupResultsByProject = groupResultsByProject;
 const project_registry_1 = require("./project-registry");
-function resolveCrossProjectScope(opts) {
+function resolveCrossProjectScope(opts, registeredProjects = (0, project_registry_1.listProjects)()) {
     const active = !!(opts.allProjects || opts.projects);
     if (!active) {
         return { active: false, roots: [], projectRoots: [], warnings: [] };
     }
     // Ignore "error"-status projects: the daemon won't search them anyway.
-    const all = (0, project_registry_1.listProjects)().filter((p) => p.status !== "error");
+    const all = registeredProjects.filter((p) => p.status !== "error");
     const byName = new Map(all.map((p) => [p.name, p]));
     const warnings = [];
     const resolveNames = (csv) => {