pulseed 0.5.2 → 0.5.3

package/dist/platform/code-search/orchestrator.js

@@ -0,0 +1,81 @@
+import * as path from "node:path";
+import { normalizeCandidates } from "./candidate-normalizer.js";
+import { fuseCandidates } from "./fusion.js";
+import { getCodeSearchIndexes } from "./indexes/index-store.js";
+import { planCodeSearchTask, taskFromVerificationSignal } from "./query-planner.js";
+import { rerankCandidates } from "./reranker.js";
+import { CallgraphRetriever } from "./retrievers/callgraph-retriever.js";
+import { ConfigRetriever } from "./retrievers/config-retriever.js";
+import { LexicalRetriever } from "./retrievers/lexical-retriever.js";
+import { PackageRetriever } from "./retrievers/package-retriever.js";
+import { RepoMapRetriever } from "./retrievers/repo-map-retriever.js";
+import { SemanticRetriever } from "./retrievers/semantic-retriever.js";
+import { StacktraceRetriever } from "./retrievers/stacktrace-retriever.js";
+import { SymbolRetriever } from "./retrievers/symbol-retriever.js";
+import { TestRetriever } from "./retrievers/test-retriever.js";
+import { createRetrievalTrace } from "./trace.js";
+const DEFAULT_RETRIEVERS = [
+    new StacktraceRetriever(),
+    new SymbolRetriever(),
+    new LexicalRetriever(),
+    new TestRetriever(),
+    new ConfigRetriever(),
+    new PackageRetriever(),
+    new RepoMapRetriever(),
+    new CallgraphRetriever(),
+    new SemanticRetriever(),
+];
+export class SearchOrchestrator {
+    root;
+    constructor(root = process.cwd()) {
+        this.root = root;
+    }
+    async search(task) {
+        return (await this.searchWithState(task)).candidates;
+    }
+    async searchWithState(task) {
+        const req = planCodeSearchTask({ ...task, cwd: task.cwd ?? this.root });
+        const indexes = await getCodeSearchIndexes(req.cwd, { maxFiles: req.budget.maxFiles });
+        const trace = createRetrievalTrace({ task: req.task, intent: req.intent });
+        trace.indexVersions.push(indexes.version);
+        const all = [];
+        for (const retriever of DEFAULT_RETRIEVERS) {
+            trace.retrieversUsed.push(retriever.name);
+            try {
+                const candidates = await retriever.search(req, indexes);
+                trace.candidatesReturnedByRetriever[retriever.name] = candidates.length;
+                all.push(...candidates);
+            }
+            catch (error) {
+                trace.warnings.push(`${retriever.name} retriever failed: ${error.message}`);
+            }
+        }
+        const normalized = normalizeCandidates(all);
+        const fused = fuseCandidates(normalized, req.budget.maxFusionCandidates);
+        const ranked = rerankCandidates(fused, req.intent, req.budget.maxRerankCandidates);
+        trace.fusedCandidates = fused.map((candidate) => candidate.id);
+        trace.rerankedCandidates = ranked.map((candidate) => candidate.id);
+        for (const candidate of ranked) {
+            trace.fileHashes[candidate.file] = candidate.fileHashAtIndex;
+            candidate.currentFileHash = candidate.fileHashAtIndex;
+        }
+        for (const candidate of ranked) {
+            candidate.reasons = [...new Set([...candidate.reasons, `trace:${trace.queryId}`])];
+        }
+        return {
+            queryId: trace.queryId,
+            task: { ...task, cwd: task.cwd ?? path.resolve(this.root) },
+            candidates: ranked,
+            trace,
+        };
+    }
+    async searchFromVerification(error, prior) {
+        const repairTask = taskFromVerificationSignal(error, prior.task);
+        const candidates = await this.search(repairTask);
+        return candidates.map((candidate) => ({
+            ...candidate,
+            reasons: [...new Set([...candidate.reasons, `repair from ${prior.queryId}`, `verification:${error.kind}`])],
+        }));
+    }
+}
+//# sourceMappingURL=orchestrator.js.map

package/dist/platform/code-search/orchestrator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"orchestrator.js","sourceRoot":"","sources":["../../../src/platform/code-search/orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,0BAA0B,EAAE,MAAM,oBAAoB,CAAC;AACpF,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,qCAAqC,CAAC;AACzE,OAAO,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,OAAO,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAC;AACnE,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAElD,MAAM,kBAAkB,GAAG;IACzB,IAAI,mBAAmB,EAAE;IACzB,IAAI,eAAe,EAAE;IACrB,IAAI,gBAAgB,EAAE;IACtB,IAAI,aAAa,EAAE;IACnB,IAAI,eAAe,EAAE;IACrB,IAAI,gBAAgB,EAAE;IACtB,IAAI,gBAAgB,EAAE;IACtB,IAAI,kBAAkB,EAAE;IACxB,IAAI,iBAAiB,EAAE;CACxB,CAAC;AAEF,MAAM,OAAO,kBAAkB;IACA;IAA7B,YAA6B,OAAO,OAAO,CAAC,GAAG,EAAE;QAApB,SAAI,GAAJ,IAAI,CAAgB;IAAG,CAAC;IAErD,KAAK,CAAC,MAAM,CAAC,IAAoB;QAC/B,OAAO,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAAoB;QACxC,MAAM,GAAG,GAAG,kBAAkB,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACxE,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACvF,MAAM,KAAK,GAAG,oBAAoB,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3E,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAE1C,MAAM,GAAG,GAAG,EAAE,CAAC;QACf,KAAK,MAAM,SAAS,IAAI,kBAAkB,EAAE,CAAC;YAC3C,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;gBACxD,KAAK,CAAC,6BAA6B,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC;gBACxE,GAAG,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;YAC1B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,IAAI,sBAAuB,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;YACzF,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,KAAK,GAAG,cAAc,CAAC,UAAU,EAAE,GAAG,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACzE,MAAM,MAAM,GAAG,gBAAgB,CAAC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACnF,KAAK,CAAC,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC/D,KAAK,CAAC,kBAAkB,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACnE,KAAK,MAAM,SAAS,IAAI,MAAM,EAAE,CAAC;YAC/B,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,eAAe,CAAC;YAC7D,SAAS,CAAC,eAAe,GAAG,SAAS,CAAC,eAAe,CAAC;QACxD,CAAC;QACD,KAAK,MAAM,SAAS,IAAI,MAAM,EAAE,CAAC;YAC/B,SAAS,CAAC,OAAO,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACrF,CAAC;QACD,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAC3D,UAAU,EAAE,MAAM;YAClB,KAAK;SACN,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,KAAyB,EAAE,KAAyB;QAC/E,MAAM,UAAU,GAAG,0BAA0B,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACjD,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACpC,GAAG,SAAS;YACZ,OAAO,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,OAAO,EAAE,eAAe,KAAK,CAAC,OAAO,EAAE,EAAE,gBAAgB,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;SAC5G,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}

package/dist/platform/code-search/path-policy.d.ts

@@ -0,0 +1,6 @@
+export declare function toRepoRelative(root: string, filePath: string): string;
+export declare function isPathInside(root: string, filePath: string): boolean;
+export declare function isSearchablePath(filePath: string): boolean;
+export declare function isConfigPath(filePath: string): boolean;
+export declare function isTestPath(filePath: string): boolean;
+//# sourceMappingURL=path-policy.d.ts.map

package/dist/platform/code-search/path-policy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"path-policy.d.ts","sourceRoot":"","sources":["../../../src/platform/code-search/path-policy.ts"],"names":[],"mappings":"AA4BA,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAGrE;AAED,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAGpE;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAQ1D;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAStD;AAED,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAIpD"}

package/dist/platform/code-search/path-policy.js

@@ -0,0 +1,60 @@
+import * as path from "node:path";
+import { classifyGeneratedPath } from "./generated-detector.js";
+const SOURCE_EXTENSIONS = new Set([
+    ".ts",
+    ".tsx",
+    ".js",
+    ".jsx",
+    ".mjs",
+    ".cjs",
+    ".json",
+    ".md",
+    ".yaml",
+    ".yml",
+    ".toml",
+    ".css",
+    ".scss",
+]);
+const CONFIG_FILE_NAMES = new Set([
+    "package.json",
+    "tsconfig.json",
+    "pnpm-workspace.yaml",
+    "eslint.config.mjs",
+    "vite.config.ts",
+    "vitest.config.ts",
+]);
+export function toRepoRelative(root, filePath) {
+    const absolute = path.isAbsolute(filePath) ? filePath : path.resolve(root, filePath);
+    return path.relative(root, absolute).split(path.sep).join("/");
+}
+export function isPathInside(root, filePath) {
+    const relative = path.relative(root, path.isAbsolute(filePath) ? filePath : path.resolve(root, filePath));
+    return relative === "" || (!relative.startsWith("..") && !path.isAbsolute(relative));
+}
+export function isSearchablePath(filePath) {
+    const normalized = filePath.split(path.sep).join("/");
+    const basename = path.basename(normalized);
+    const ext = path.extname(normalized);
+    if (basename.startsWith(".") && !CONFIG_FILE_NAMES.has(basename))
+        return false;
+    if (!SOURCE_EXTENSIONS.has(ext) && !CONFIG_FILE_NAMES.has(basename))
+        return false;
+    const classification = classifyGeneratedPath(normalized);
+    return !classification.vendor && !classification.buildArtifact;
+}
+export function isConfigPath(filePath) {
+    const basename = path.basename(filePath);
+    return CONFIG_FILE_NAMES.has(basename)
+        || /^tsconfig.*\.json$/.test(basename)
+        || /^vitest.*\.config\./.test(basename)
+        || /^eslint\.config\./.test(basename)
+        || basename.includes("workspace")
+        || basename.includes("docker")
+        || basename.includes("Dockerfile");
+}
+export function isTestPath(filePath) {
+    return /(?:^|[./_-])(?:test|spec)\.[cm]?[jt]sx?$/.test(filePath)
+        || filePath.includes("__tests__")
+        || filePath.includes("/tests/");
+}
+//# sourceMappingURL=path-policy.js.map

package/dist/platform/code-search/path-policy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"path-policy.js","sourceRoot":"","sources":["../../../src/platform/code-search/path-policy.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAEhE,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC;IAChC,KAAK;IACL,MAAM;IACN,KAAK;IACL,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,KAAK;IACL,OAAO;IACP,MAAM;IACN,OAAO;IACP,MAAM;IACN,OAAO;CACR,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC;IAChC,cAAc;IACd,eAAe;IACf,qBAAqB;IACrB,mBAAmB;IACnB,gBAAgB;IAChB,kBAAkB;CACnB,CAAC,CAAC;AAEH,MAAM,UAAU,cAAc,CAAC,IAAY,EAAE,QAAgB;IAC3D,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACrF,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,IAAY,EAAE,QAAgB;IACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC1G,OAAO,QAAQ,KAAK,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;AACvF,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAAgB;IAC/C,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACrC,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC;QAAE,OAAO,KAAK,CAAC;IAC/E,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC;QAAE,OAAO,KAAK,CAAC;IAClF,MAAM,cAAc,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;IACzD,OAAO,CAAC,cAAc,CAAC,MAAM,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,QAAgB;IAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACzC,OAAO,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC;WACjC,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAC;WACnC,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC;WACpC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC;WAClC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;WAC9B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;WAC3B,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,QAAgB;IACzC,OAAO,0CAA0C,CAAC,IAAI,CAAC,QAAQ,CAAC;WAC3D,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;WAC9B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;AACpC,CAAC"}

package/dist/platform/code-search/progressive-reader.d.ts

@@ -0,0 +1,8 @@
+import type { CodeSearchIndexes, ContextBundle, RankedCandidate, ReadRequest, ReadSessionState } from "./contracts.js";
+export declare class ProgressiveReader {
+    private readonly root;
+    private readonly indexes?;
+    constructor(root: string, indexes?: CodeSearchIndexes | undefined);
+    read(candidates: RankedCandidate[], request: ReadRequest, state?: ReadSessionState): Promise<ContextBundle>;
+}
+//# sourceMappingURL=progressive-reader.d.ts.map

package/dist/platform/code-search/progressive-reader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"progressive-reader.d.ts","sourceRoot":"","sources":["../../../src/platform/code-search/progressive-reader.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,iBAAiB,EACjB,aAAa,EACb,eAAe,EAGf,WAAW,EACX,gBAAgB,EACjB,MAAM,gBAAgB,CAAC;AAoGxB,qBAAa,iBAAiB;IAChB,OAAO,CAAC,QAAQ,CAAC,IAAI;IAAU,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAAvC,IAAI,EAAE,MAAM,EAAmB,OAAO,CAAC,EAAE,iBAAiB,YAAA;IAEjF,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC;CA+ElH"}

package/dist/platform/code-search/progressive-reader.js

@@ -0,0 +1,173 @@
+import * as fsp from "node:fs/promises";
+import * as path from "node:path";
+import { hashFileOrNull } from "./indexes/file-index.js";
+import { createRetrievalTrace, createQueryId } from "./trace.js";
+function estimateTokens(text) {
+    return Math.ceil(text.length / 4);
+}
+function overlaps(a, b) {
+    return a.file === b.file && a.startLine <= b.endLine && b.startLine <= a.endLine;
+}
+function expandRange(candidate, phase) {
+    const expansion = phase === "locate" ? 12
+        : phase === "understand" ? 40
+            : phase === "plan_edit" || phase === "edit" ? 80
+                : phase === "repair" ? 60
+                    : 30;
+    return {
+        file: candidate.file,
+        startLine: Math.max(1, candidate.range.startLine - expansion),
+        endLine: candidate.range.endLine + expansion,
+    };
+}
+function sliceIndexesForRanges(indexes, ranges) {
+    if (!indexes || ranges.length === 0)
+        return {};
+    const readFiles = new Set(ranges.map((range) => range.file));
+    const readPackages = new Set(indexes.packages.packages
+        .filter((pkg) => [...readFiles].some((file) => pkg.root === "" || file === pkg.root || file.startsWith(`${pkg.root}/`)))
+        .map((pkg) => pkg.root));
+    const repoFiles = indexes.repoMap.files
+        .filter((entry) => readFiles.has(entry.file)
+        || entry.imports.some((specifier) => [...readFiles].some((file) => file.includes(specifier.replace(/^\.\//, "").replace(/\.(js|ts|tsx|jsx)$/, "")))))
+        .slice(0, 12)
+        .map((entry) => ({
+        file: entry.file,
+        imports: entry.imports.slice(0, 12),
+        exports: entry.exports.slice(0, 12),
+    }));
+    const tests = indexes.tests.tests
+        .filter((test) => readFiles.has(test.file)
+        || test.imports.some((specifier) => [...readFiles].some((file) => file.includes(specifier.replace(/^\.\//, "").replace(/\.(js|ts|tsx|jsx)$/, "")))))
+        .slice(0, 8)
+        .map((test) => ({
+        file: test.file,
+        names: test.names.slice(0, 12),
+        imports: test.imports.slice(0, 12),
+    }));
+    const packages = indexes.packages.packages
+        .filter((pkg) => readPackages.has(pkg.root))
+        .slice(0, 4)
+        .map((pkg) => ({
+        name: pkg.name,
+        root: pkg.root,
+        dependencies: pkg.dependencies.slice(0, 24),
+    }));
+    const configs = indexes.configs.files
+        .filter((file) => file === "package.json" || file.includes("tsconfig") || file.includes("vitest"))
+        .slice(0, 8);
+    return {
+        repoMap: repoFiles.length > 0 ? { files: repoFiles } : undefined,
+        packageContext: packages.length > 0 ? { packages } : undefined,
+        testContext: tests.length > 0 ? { tests } : undefined,
+        configContext: configs.length > 0 ? { files: configs } : undefined,
+    };
+}
+async function readRange(root, candidate, key) {
+    try {
+        const absolute = path.resolve(root, key.file);
+        const lines = (await fsp.readFile(absolute, "utf8")).split("\n");
+        const start = Math.max(1, key.startLine);
+        const end = Math.min(lines.length, key.endLine);
+        const content = lines
+            .slice(start - 1, end)
+            .map((line, index) => `${start + index}\t${line}`)
+            .join("\n");
+        return {
+            ...key,
+            startLine: start,
+            endLine: end,
+            candidateId: candidate.id,
+            reason: candidate.reasons[0] ?? candidate.readRecommendation,
+            content,
+            tokenEstimate: estimateTokens(content),
+        };
+    }
+    catch {
+        return null;
+    }
+}
+export class ProgressiveReader {
+    root;
+    indexes;
+    constructor(root, indexes) {
+        this.root = root;
+        this.indexes = indexes;
+    }
+    async read(candidates, request, state) {
+        const queryId = request.queryId ?? state?.queryId ?? createQueryId("code-read");
+        const nextState = state ?? {
+            queryId,
+            readRanges: [],
+            rejectedRanges: [],
+            budget: {
+                maxReadRanges: request.maxReadRanges ?? 8,
+                maxReadTokens: request.maxReadTokens ?? 12_000,
+                usedReadRanges: 0,
+                estimatedTokens: 0,
+            },
+            phase: request.phase ?? "locate",
+        };
+        nextState.phase = request.phase ?? nextState.phase;
+        nextState.budget.maxReadRanges = request.maxReadRanges ?? nextState.budget.maxReadRanges;
+        nextState.budget.maxReadTokens = request.maxReadTokens ?? nextState.budget.maxReadTokens;
+        const requestedIds = new Set(request.candidateIds ?? []);
+        const pool = requestedIds.size > 0
+            ? candidates.filter((candidate) => requestedIds.has(candidate.id))
+            : candidates.filter((candidate) => candidate.readRecommendation === "read_now").slice(0, nextState.budget.maxReadRanges);
+        const ranges = [];
+        const warnings = [];
+        const omittedCandidates = [];
+        for (const candidate of pool) {
+            if (nextState.budget.usedReadRanges >= nextState.budget.maxReadRanges) {
+                omittedCandidates.push({ candidateId: candidate.id, reason: "read range budget exhausted" });
+                continue;
+            }
+            if (candidate.readRecommendation === "avoid_edit") {
+                omittedCandidates.push({ candidateId: candidate.id, reason: "vendor/build artifact candidate" });
+                continue;
+            }
+            const file = this.indexes?.files.find((entry) => entry.path === candidate.file);
+            if (file) {
+                const currentHash = await hashFileOrNull(file.absolutePath);
+                if (currentHash && currentHash !== candidate.fileHashAtIndex)
+                    warnings.push(`Stale candidate hash for ${candidate.file}`);
+            }
+            const key = expandRange(candidate, nextState.phase);
+            if (nextState.readRanges.some((read) => overlaps(read, key))) {
+                omittedCandidates.push({ candidateId: candidate.id, reason: "range already read" });
+                continue;
+            }
+            const result = await readRange(this.root, candidate, key);
+            if (!result) {
+                omittedCandidates.push({ candidateId: candidate.id, reason: "file could not be read" });
+                continue;
+            }
+            if (nextState.budget.estimatedTokens + result.tokenEstimate > nextState.budget.maxReadTokens) {
+                nextState.rejectedRanges.push({ key, reason: "token budget exhausted" });
+                omittedCandidates.push({ candidateId: candidate.id, reason: "token budget exhausted" });
+                continue;
+            }
+            ranges.push(result);
+            nextState.readRanges.push(key);
+            nextState.budget.usedReadRanges += 1;
+            nextState.budget.estimatedTokens += result.tokenEstimate;
+        }
+        const trace = createRetrievalTrace({ queryId, task: "code_read_context", intent: "unknown" });
+        trace.readCandidates = ranges.map((range) => range.candidateId);
+        trace.omittedCandidates = omittedCandidates;
+        trace.warnings = warnings;
+        const contextSlices = sliceIndexesForRanges(this.indexes, ranges);
+        return {
+            queryId,
+            state: nextState,
+            ranges,
+            ...contextSlices,
+            omittedCandidates,
+            warnings,
+            tokenEstimate: ranges.reduce((sum, range) => sum + range.tokenEstimate, 0),
+            trace,
+        };
+    }
+}
+//# sourceMappingURL=progressive-reader.js.map

package/dist/platform/code-search/progressive-reader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"progressive-reader.js","sourceRoot":"","sources":["../../../src/platform/code-search/progressive-reader.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,GAAG,MAAM,kBAAkB,CAAC;AACxC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAUlC,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,oBAAoB,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAEjE,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,QAAQ,CAAC,CAAe,EAAE,CAAe;IAChD,OAAO,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,CAAC;AACnF,CAAC;AAED,SAAS,WAAW,CAAC,SAA0B,EAAE,KAAgC;IAC/E,MAAM,SAAS,GACb,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE;QACvB,CAAC,CAAC,KAAK,KAAK,YAAY,CAAC,CAAC,CAAC,EAAE;YAC7B,CAAC,CAAC,KAAK,KAAK,WAAW,IAAI,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE;gBAChD,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE;oBACzB,CAAC,CAAC,EAAE,CAAC;IACP,OAAO;QACL,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7D,OAAO,EAAE,SAAS,CAAC,KAAK,CAAC,OAAO,GAAG,SAAS;KAC7C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAsC,EAAE,MAAyB;IAC9F,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAC/C,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7D,MAAM,YAAY,GAAG,IAAI,GAAG,CAC1B,OAAO,CAAC,QAAQ,CAAC,QAAQ;SACtB,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;SACvH,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAC1B,CAAC;IACF,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK;SACpC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAChB,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;WACtB,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CACrJ;SACA,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;SACZ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACf,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;QACnC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;KACpC,CAAC,CAAC,CAAC;IACN,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK;SAC9B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACf,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;WACrB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CACpJ;SACA,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACd,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;QAC9B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;KACnC,CAAC,CAAC,CAAC;IACN,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ;SACvC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;SAC3C,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;SACX,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACb,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,YAAY,EAAE,GAAG,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;KAC5C,CAAC,CAAC,CAAC;IACN,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK;SAClC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,cAAc,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;SACjG,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAEf,OAAO;QACL,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,SAAS;QAChE,cAAc,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS;QAC9D,WAAW,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS;QACrD,aAAa,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;KACnE,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,IAAY,EAAE,SAA0B,EAAE,GAAiB;IAClF,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,CAAC,MAAM,GAAG,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,OAAO,GAAG,KAAK;aAClB,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,GAAG,CAAC;aACrB,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,KAAK,GAAG,KAAK,KAAK,IAAI,EAAE,CAAC;aACjD,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,OAAO;YACL,GAAG,GAAG;YACN,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,GAAG;YACZ,WAAW,EAAE,SAAS,CAAC,EAAE;YACzB,MAAM,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,kBAAkB;YAC5D,OAAO;YACP,aAAa,EAAE,cAAc,CAAC,OAAO,CAAC;SACvC,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,OAAO,iBAAiB;IACC;IAA+B;IAA5D,YAA6B,IAAY,EAAmB,OAA2B;QAA1D,SAAI,GAAJ,IAAI,CAAQ;QAAmB,YAAO,GAAP,OAAO,CAAoB;IAAG,CAAC;IAE3F,KAAK,CAAC,IAAI,CAAC,UAA6B,EAAE,OAAoB,EAAE,KAAwB;QACtF,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,KAAK,EAAE,OAAO,IAAI,aAAa,CAAC,WAAW,CAAC,CAAC;QAChF,MAAM,SAAS,GAAqB,KAAK,IAAI;YAC3C,OAAO;YACP,UAAU,EAAE,EAAE;YACd,cAAc,EAAE,EAAE;YAClB,MAAM,EAAE;gBACN,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,CAAC;gBACzC,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,MAAM;gBAC9C,cAAc,EAAE,CAAC;gBACjB,eAAe,EAAE,CAAC;aACnB;YACD,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,QAAQ;SACjC,CAAC;QACF,SAAS,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,SAAS,CAAC,KAAK,CAAC;QACnD,SAAS,CAAC,MAAM,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC;QACzF,SAAS,CAAC,MAAM,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC;QAEzF,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;QACzD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,GAAG,CAAC;YAChC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,kBAAkB,KAAK,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAE3H,MAAM,MAAM,GAAsB,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,iBAAiB,GAAuC,EAAE,CAAC;QAEjE,KAAK,MAAM,SAAS,IAAI,IAAI,EAAE,CAAC;YAC7B,IAAI,SAAS,CAAC,MAAM,CAAC,cAAc,IAAI,SAAS,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBACtE,iBAAiB,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,6BAA6B,EAAE,CAAC,CAAC;gBAC7F,SAAS;YACX,CAAC;YACD,IAAI,SAAS,CAAC,kBAAkB,KAAK,YAAY,EAAE,CAAC;gBAClD,iBAAiB,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,iCAAiC,EAAE,CAAC,CAAC;gBACjG,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC;YAChF,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAC5D,IAAI,WAAW,IAAI,WAAW,KAAK,SAAS,CAAC,eAAe;oBAAE,QAAQ,CAAC,IAAI,CAAC,4BAA4B,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5H,CAAC;YACD,MAAM,GAAG,GAAG,WAAW,CAAC,SAAS,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;YACpD,IAAI,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;gBAC7D,iBAAiB,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,oBAAoB,EAAE,CAAC,CAAC;gBACpF,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;YAC1D,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,iBAAiB,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;gBACxF,SAAS;YACX,CAAC;YACD,IAAI,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,MAAM,CAAC,aAAa,GAAG,SAAS,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBAC7F,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;gBACzE,iBAAiB,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;gBACxF,SAAS;YACX,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpB,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC/B,SAAS,CAAC,MAAM,CAAC,cAAc,IAAI,CAAC,CAAC;YACrC,SAAS,CAAC,MAAM,CAAC,eAAe,IAAI,MAAM,CAAC,aAAa,CAAC;QAC3D,CAAC;QAED,MAAM,KAAK,GAAG,oBAAoB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9F,KAAK,CAAC,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAChE,KAAK,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC5C,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC1B,MAAM,aAAa,GAAG,qBAAqB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAElE,OAAO;YACL,OAAO;YACP,KAAK,EAAE,SAAS;YAChB,MAAM;YACN,GAAG,aAAa;YAChB,iBAAiB;YACjB,QAAQ;YACR,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC,CAAC;YAC1E,KAAK;SACN,CAAC;IACJ,CAAC;CACF"}

package/dist/platform/code-search/query-planner.d.ts

@@ -0,0 +1,4 @@
+import type { CodeSearchTask, SearchRequest, VerificationSignal } from "./contracts.js";
+export declare function planCodeSearchTask(task: CodeSearchTask): SearchRequest;
+export declare function taskFromVerificationSignal(signal: VerificationSignal, priorTask: CodeSearchTask): CodeSearchTask;
+//# sourceMappingURL=query-planner.d.ts.map

package/dist/platform/code-search/query-planner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-planner.d.ts","sourceRoot":"","sources":["../../../src/platform/code-search/query-planner.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAwB,aAAa,EAAc,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAqE1H,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,cAAc,GAAG,aAAa,CAiBtE;AAED,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,kBAAkB,EAAE,SAAS,EAAE,cAAc,GAAG,cAAc,CAiBhH"}

package/dist/platform/code-search/query-planner.js

@@ -0,0 +1,107 @@
+import * as path from "node:path";
+const DEFAULT_BUDGET = {
+    maxFiles: 1_500,
+    maxCandidatesPerRetriever: 100,
+    maxFusionCandidates: 400,
+    maxRerankCandidates: 200,
+};
+const STOPWORDS = new Set([
+    "the", "and", "for", "with", "from", "that", "this", "into", "when", "then", "have",
+    "has", "was", "were", "are", "can", "will", "should", "must", "not", "but", "fix",
+    "add", "use", "using", "make", "して", "する", "これ", "実装", "修正",
+]);
+const PATH_RE = /(?:^|[\s"'`(])((?:\.{1,2}\/)?[\w@./-]+\.(?:ts|tsx|js|jsx|json|md|yaml|yml|mjs|cjs))/g;
+const STACK_FRAME_RE = /(?:at\s+(?<symbol>[^\s(]+)\s+\()?((?<file>(?:\/|\.{1,2}\/)?[\w@./-]+\.(?:ts|tsx|js|jsx|mjs|cjs))):(?<line>\d+):(?<column>\d+)/g;
+function inferIntent(task, explicit) {
+    if (explicit)
+        return explicit;
+    const lower = task.toLowerCase();
+    if (/security|ssrf|xss|attack|攻撃|セキュリティ/.test(lower))
+        return "security_review";
+    if (/test|spec|assert|failure|failing|失敗/.test(lower))
+        return "test_failure";
+    if (/refactor|cleanup|整理|債務/.test(lower))
+        return "refactor";
+    if (/config|tsconfig|eslint|package|設定/.test(lower))
+        return "config_fix";
+    if (/explain|どう|なぜ|説明/.test(lower))
+        return "explain";
+    if (/api|schema|route|contract/.test(lower))
+        return "api_change";
+    if (/bug|fix|error|例外|壊/.test(lower))
+        return "bugfix";
+    if (/add|implement|feature|実装|追加/.test(lower))
+        return "feature_addition";
+    return "unknown";
+}
+function extractLikelyPaths(text) {
+    const paths = [];
+    for (const match of text.matchAll(PATH_RE)) {
+        paths.push(match[1]);
+    }
+    return [...new Set(paths.map((candidate) => candidate.replace(/^[\s"'`(]+/, "")))];
+}
+function extractStackFrames(text) {
+    const frames = [];
+    for (const match of text.matchAll(STACK_FRAME_RE)) {
+        const groups = match.groups;
+        if (!groups?.file)
+            continue;
+        frames.push({
+            file: groups.file,
+            line: groups.line ? Number(groups.line) : undefined,
+            column: groups.column ? Number(groups.column) : undefined,
+            symbol: groups.symbol,
+        });
+    }
+    return frames;
+}
+function extractTerms(text) {
+    const raw = text
+        .split(/[^A-Za-z0-9_$#./-]+/)
+        .map((term) => term.trim())
+        .filter((term) => term.length >= 3 && !STOPWORDS.has(term.toLowerCase()));
+    return [...new Set(raw)].slice(0, 40);
+}
+function extractSymbols(terms) {
+    return terms
+        .filter((term) => /^[A-Z_$][A-Za-z0-9_$]*$/.test(term) || /^[a-z_$][A-Za-z0-9_$]*[A-Z][A-Za-z0-9_$]*$/.test(term))
+        .slice(0, 30);
+}
+export function planCodeSearchTask(task) {
+    const combined = [task.task, task.stacktrace ?? "", ...(task.queryTerms ?? [])].join("\n");
+    const queryTerms = [...new Set([...(task.queryTerms ?? []), ...extractTerms(combined)])].slice(0, 50);
+    const stackFrames = extractStackFrames(combined);
+    return {
+        task: task.task,
+        intent: inferIntent(task.task, task.intent),
+        queryTerms,
+        stacktrace: task.stacktrace ?? "",
+        fileGlobs: task.fileGlobs ?? [],
+        packageScope: task.packageScope ?? "",
+        cwd: path.resolve(task.cwd ?? process.cwd()),
+        likelySymbols: extractSymbols(queryTerms),
+        likelyPaths: extractLikelyPaths(combined),
+        stackFrames,
+        budget: { ...DEFAULT_BUDGET, ...(task.budget ?? {}) },
+    };
+}
+export function taskFromVerificationSignal(signal, priorTask) {
+    switch (signal.kind) {
+        case "undefined_symbol":
+            return { ...priorTask, intent: "bugfix", task: `${priorTask.task}\nResolve undefined symbol ${signal.symbol}`, queryTerms: [signal.symbol, ...(priorTask.queryTerms ?? [])] };
+        case "type_error":
+            return { ...priorTask, intent: "test_failure", task: `${priorTask.task}\nType error ${signal.typeName ?? ""} ${signal.file ?? ""}`, queryTerms: [signal.typeName, signal.file].filter(Boolean) };
+        case "failing_test":
+            return { ...priorTask, intent: "test_failure", task: `${priorTask.task}\nFailing test ${signal.testName ?? ""} ${signal.assertion ?? ""}`, queryTerms: [signal.testFile, signal.testName, signal.assertion].filter(Boolean) };
+        case "runtime_stacktrace":
+            return { ...priorTask, intent: "bugfix", stacktrace: signal.stacktrace, task: `${priorTask.task}\nRuntime stacktrace:\n${signal.stacktrace}` };
+        case "lint_error":
+            return { ...priorTask, intent: "config_fix", task: `${priorTask.task}\nLint error ${signal.rule ?? ""} ${signal.file ?? ""}`, queryTerms: [signal.rule, signal.file].filter(Boolean) };
+        case "package_import_error":
+            return { ...priorTask, intent: "config_fix", task: `${priorTask.task}\nPackage import error ${signal.specifier ?? ""} ${signal.packageName ?? ""}`, queryTerms: [signal.specifier, signal.packageName].filter(Boolean) };
+        case "generated_file_selected":
+            return { ...priorTask, task: `${priorTask.task}\nFind source owner for generated file ${signal.file}`, queryTerms: [signal.file, "source", "template", "schema"] };
+    }
+}
+//# sourceMappingURL=query-planner.js.map

package/dist/platform/code-search/query-planner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-planner.js","sourceRoot":"","sources":["../../../src/platform/code-search/query-planner.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAGlC,MAAM,cAAc,GAAiB;IACnC,QAAQ,EAAE,KAAK;IACf,yBAAyB,EAAE,GAAG;IAC9B,mBAAmB,EAAE,GAAG;IACxB,mBAAmB,EAAE,GAAG;CACzB,CAAC;AAEF,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC;IACxB,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;IACnF,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK;IACjF,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI;CAC5D,CAAC,CAAC;AAEH,MAAM,OAAO,GAAG,sFAAsF,CAAC;AACvG,MAAM,cAAc,GAAG,gIAAgI,CAAC;AAExJ,SAAS,WAAW,CAAC,IAAY,EAAE,QAAiB;IAClD,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACjC,IAAI,oCAAoC,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,iBAAiB,CAAC;IAC/E,IAAI,qCAAqC,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,cAAc,CAAC;IAC7E,IAAI,wBAAwB,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,UAAU,CAAC;IAC5D,IAAI,mCAAmC,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,YAAY,CAAC;IACzE,IAAI,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACrD,IAAI,2BAA2B,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,YAAY,CAAC;IACjE,IAAI,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IACtD,IAAI,6BAA6B,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,kBAAkB,CAAC;IACzE,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3C,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;AACrF,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QAClD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,MAAM,EAAE,IAAI;YAAE,SAAS;QAC5B,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YACnD,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;YACzD,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,IAAY;IAChC,MAAM,GAAG,GAAG,IAAI;SACb,KAAK,CAAC,qBAAqB,CAAC;SAC5B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAC5E,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACxC,CAAC;AAED,SAAS,cAAc,CAAC,KAAe;IACrC,OAAO,KAAK;SACT,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,4CAA4C,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;SACjH,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,IAAoB;IACrD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3F,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,EAAE,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACtG,MAAM,WAAW,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACjD,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,MAAM,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC;QAC3C,UAAU;QACV,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;QACjC,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,EAAE;QAC/B,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,EAAE;QACrC,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;QAC5C,aAAa,EAAE,cAAc,CAAC,UAAU,CAAC;QACzC,WAAW,EAAE,kBAAkB,CAAC,QAAQ,CAAC;QACzC,WAAW;QACX,MAAM,EAAE,EAAE,GAAG,cAAc,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,EAAE;KACtD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,MAA0B,EAAE,SAAyB;IAC9F,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,kBAAkB;YACrB,OAAO,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,IAAI,8BAA8B,MAAM,CAAC,MAAM,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,SAAS,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAChL,KAAK,YAAY;YACf,OAAO,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,IAAI,gBAAgB,MAAM,CAAC,QAAQ,IAAI,EAAE,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAa,EAAE,CAAC;QAC/M,KAAK,cAAc;YACjB,OAAO,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,IAAI,kBAAkB,MAAM,CAAC,QAAQ,IAAI,EAAE,IAAI,MAAM,CAAC,SAAS,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,OAAO,CAAa,EAAE,CAAC;QAC5O,KAAK,oBAAoB;YACvB,OAAO,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,IAAI,0BAA0B,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC;QACjJ,KAAK,YAAY;YACf,OAAO,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,IAAI,gBAAgB,MAAM,CAAC,IAAI,IAAI,EAAE,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAa,EAAE,CAAC;QACrM,KAAK,sBAAsB;YACzB,OAAO,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,IAAI,0BAA0B,MAAM,CAAC,SAAS,IAAI,EAAE,IAAI,MAAM,CAAC,WAAW,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,OAAO,CAAa,EAAE,CAAC;QACvO,KAAK,yBAAyB;YAC5B,OAAO,EAAE,GAAG,SAAS,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,IAAI,0CAA0C,MAAM,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,CAAC;IACvK,CAAC;AACH,CAAC"}

package/dist/platform/code-search/reranker.d.ts

@@ -0,0 +1,4 @@
+import type { Intent, RankedCandidate } from "./contracts.js";
+import type { FusedCandidate } from "./fusion.js";
+export declare function rerankCandidates(candidates: FusedCandidate[], intent: Intent, limit: number): RankedCandidate[];
+//# sourceMappingURL=reranker.d.ts.map

package/dist/platform/code-search/reranker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.d.ts","sourceRoot":"","sources":["../../../src/platform/code-search/reranker.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,eAAe,EAAsB,MAAM,gBAAgB,CAAC;AAClF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAyClD,wBAAgB,gBAAgB,CAAC,UAAU,EAAE,cAAc,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,eAAe,EAAE,CAkB/G"}

package/dist/platform/code-search/reranker.js

@@ -0,0 +1,57 @@
+function signalScore(candidate, intent) {
+    const s = candidate.signals;
+    const intentBoost = intent === "test_failure" ? s.failingTestAffinity + s.stacktraceMatch + s.testRelation
+        : intent === "config_fix" ? s.configAffinity + s.packageBoundaryFit
+            : intent === "refactor" ? s.exactSymbolMatch + s.callgraphProximity + s.packageBoundaryFit
+                : intent === "explain" ? s.repoMapCentrality + s.semanticSimilarity + s.exactSymbolMatch
+                    : intent === "security_review" ? s.lexicalMatch + s.configAffinity + s.callgraphProximity
+                        : s.lexicalMatch + s.exactSymbolMatch + s.pathPrior;
+    return (candidate.rrfScore * 40
+        + s.lexicalMatch * 2.2
+        + s.exactSymbolMatch * 2.8
+        + s.stacktraceMatch * 3.5
+        + s.failingTestAffinity * 1.8
+        + s.testRelation * 1.4
+        + s.configAffinity * 1.4
+        + s.packageBoundaryFit * 1.1
+        + s.repoMapCentrality * 0.8
+        + s.callgraphProximity * 1.2
+        + s.pathPrior * 2
+        + s.semanticSimilarity * 0.7
+        + intentBoost);
+}
+function penaltyScore(candidate) {
+    const p = candidate.penalties;
+    return p.generated + p.vendor + p.buildArtifact + p.staleIndex + p.wrongPackageBoundary + p.largeFileWithoutSymbol;
+}
+function recommendation(candidate, score) {
+    if (candidate.penalties.vendor || candidate.penalties.buildArtifact)
+        return "avoid_edit";
+    if (candidate.penalties.generated)
+        return "reference_only";
+    if (score >= 3.5)
+        return "read_now";
+    if (score >= 1.4)
+        return "read_if_needed";
+    return "reference_only";
+}
+export function rerankCandidates(candidates, intent, limit) {
+    return candidates
+        .map((candidate) => {
+        const rerankScore = signalScore(candidate, intent) - penaltyScore(candidate);
+        const ranked = {
+            ...candidate,
+            rerankScore,
+            confidence: rerankScore >= 4 ? "high" : rerankScore >= 1.5 ? "medium" : "low",
+            readRecommendation: recommendation(candidate, rerankScore),
+        };
+        return ranked;
+    })
+        .sort((a, b) => b.rerankScore - a.rerankScore)
+        .map((candidate, index) => ({
+        ...candidate,
+        ranks: { ...candidate.ranks, finalRank: index + 1 },
+    }))
+        .slice(0, limit);
+}
+//# sourceMappingURL=reranker.js.map

package/dist/platform/code-search/reranker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.js","sourceRoot":"","sources":["../../../src/platform/code-search/reranker.ts"],"names":[],"mappings":"AAGA,SAAS,WAAW,CAAC,SAAyB,EAAE,MAAc;IAC5D,MAAM,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC;IAC5B,MAAM,WAAW,GACf,MAAM,KAAK,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,mBAAmB,GAAG,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,YAAY;QACtF,CAAC,CAAC,MAAM,KAAK,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,kBAAkB;YACnE,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,gBAAgB,GAAG,CAAC,CAAC,kBAAkB,GAAG,CAAC,CAAC,kBAAkB;gBAC1F,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,GAAG,CAAC,CAAC,kBAAkB,GAAG,CAAC,CAAC,gBAAgB;oBACxF,CAAC,CAAC,MAAM,KAAK,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,kBAAkB;wBACzF,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,gBAAgB,GAAG,CAAC,CAAC,SAAS,CAAC;IACtD,OAAO,CACL,SAAS,CAAC,QAAQ,GAAG,EAAE;UACrB,CAAC,CAAC,YAAY,GAAG,GAAG;UACpB,CAAC,CAAC,gBAAgB,GAAG,GAAG;UACxB,CAAC,CAAC,eAAe,GAAG,GAAG;UACvB,CAAC,CAAC,mBAAmB,GAAG,GAAG;UAC3B,CAAC,CAAC,YAAY,GAAG,GAAG;UACpB,CAAC,CAAC,cAAc,GAAG,GAAG;UACtB,CAAC,CAAC,kBAAkB,GAAG,GAAG;UAC1B,CAAC,CAAC,iBAAiB,GAAG,GAAG;UACzB,CAAC,CAAC,kBAAkB,GAAG,GAAG;UAC1B,CAAC,CAAC,SAAS,GAAG,CAAC;UACf,CAAC,CAAC,kBAAkB,GAAG,GAAG;UAC1B,WAAW,CACd,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,SAAyB;IAC7C,MAAM,CAAC,GAAG,SAAS,CAAC,SAAS,CAAC;IAC9B,OAAO,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,oBAAoB,GAAG,CAAC,CAAC,sBAAsB,CAAC;AACrH,CAAC;AAED,SAAS,cAAc,CAAC,SAAyB,EAAE,KAAa;IAC9D,IAAI,SAAS,CAAC,SAAS,CAAC,MAAM,IAAI,SAAS,CAAC,SAAS,CAAC,aAAa;QAAE,OAAO,YAAY,CAAC;IACzF,IAAI,SAAS,CAAC,SAAS,CAAC,SAAS;QAAE,OAAO,gBAAgB,CAAC;IAC3D,IAAI,KAAK,IAAI,GAAG;QAAE,OAAO,UAAU,CAAC;IACpC,IAAI,KAAK,IAAI,GAAG;QAAE,OAAO,gBAAgB,CAAC;IAC1C,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,UAA4B,EAAE,MAAc,EAAE,KAAa;IAC1F,OAAO,UAAU;SACd,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;QACjB,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,EAAE,MAAM,CAAC,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;QAC7E,MAAM,MAAM,GAAoB;YAC9B,GAAG,SAAS;YACZ,WAAW;YACX,UAAU,EAAE,WAAW,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK;YAC7E,kBAAkB,EAAE,cAAc,CAAC,SAAS,EAAE,WAAW,CAAC;SAC3D,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;SACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;SAC7C,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1B,GAAG,SAAS;QACZ,KAAK,EAAE,EAAE,GAAG,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,GAAG,CAAC,EAAE;KACpD,CAAC,CAAC;SACF,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;AACrB,CAAC"}

package/dist/platform/code-search/retrievers/callgraph-retriever.d.ts

@@ -0,0 +1,6 @@
+import type { CodeCandidate, CodeSearchIndexes, Retriever, SearchRequest } from "../contracts.js";
+export declare class CallgraphRetriever implements Retriever {
+    readonly name = "callgraph";
+    search(req: SearchRequest, indexes: CodeSearchIndexes): Promise<CodeCandidate[]>;
+}
+//# sourceMappingURL=callgraph-retriever.d.ts.map

package/dist/platform/code-search/retrievers/callgraph-retriever.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"callgraph-retriever.d.ts","sourceRoot":"","sources":["../../../../src/platform/code-search/retrievers/callgraph-retriever.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,iBAAiB,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAIlG,qBAAa,kBAAmB,YAAW,SAAS;IAClD,QAAQ,CAAC,IAAI,eAAe;IAEtB,MAAM,CAAC,GAAG,EAAE,aAAa,EAAE,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;CAuBvF"}

package/dist/platform/code-search/retrievers/callgraph-retriever.js

@@ -0,0 +1,31 @@
+import { makeCandidate } from "../candidate.js";
+import { directImportNeighbors } from "../indexes/call-graph.js";
+export class CallgraphRetriever {
+    name = "callgraph";
+    async search(req, indexes) {
+        if (req.intent !== "refactor" && req.intent !== "bugfix" && req.intent !== "explain")
+            return [];
+        const seedFiles = indexes.files
+            .filter((file) => req.likelyPaths.some((hint) => file.path.endsWith(hint) || hint.endsWith(file.path)))
+            .map((file) => file.path)
+            .slice(0, 5);
+        const neighbors = [...new Set(seedFiles.flatMap((file) => directImportNeighbors(file, indexes)))].slice(0, req.budget.maxCandidatesPerRetriever);
+        return neighbors.flatMap((file, index) => {
+            const matched = indexes.files.find((candidate) => candidate.path === file || candidate.path.endsWith(file));
+            if (!matched)
+                return [];
+            const candidate = makeCandidate({
+                req,
+                indexes,
+                file: matched.path,
+                retriever: this.name,
+                rank: index + 1,
+                preview: matched.path,
+                signals: { callgraphProximity: 0.7 },
+                reasons: ["direct import/reference neighbor"],
+            });
+            return candidate ? [candidate] : [];
+        });
+    }
+}
+//# sourceMappingURL=callgraph-retriever.js.map

package/dist/platform/code-search/retrievers/callgraph-retriever.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"callgraph-retriever.js","sourceRoot":"","sources":["../../../../src/platform/code-search/retrievers/callgraph-retriever.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAEjE,MAAM,OAAO,kBAAkB;IACpB,IAAI,GAAG,WAAW,CAAC;IAE5B,KAAK,CAAC,MAAM,CAAC,GAAkB,EAAE,OAA0B;QACzD,IAAI,GAAG,CAAC,MAAM,KAAK,UAAU,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,EAAE,CAAC;QAChG,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK;aAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;aACtG,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACf,MAAM,SAAS,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,qBAAqB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,yBAAyB,CAAC,CAAC;QACjJ,OAAO,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACvC,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,IAAI,IAAI,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YAC5G,IAAI,CAAC,OAAO;gBAAE,OAAO,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,aAAa,CAAC;gBAC9B,GAAG;gBACH,OAAO;gBACP,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,SAAS,EAAE,IAAI,CAAC,IAAI;gBACpB,IAAI,EAAE,KAAK,GAAG,CAAC;gBACf,OAAO,EAAE,OAAO,CAAC,IAAI;gBACrB,OAAO,EAAE,EAAE,kBAAkB,EAAE,GAAG,EAAE;gBACpC,OAAO,EAAE,CAAC,kCAAkC,CAAC;aAC9C,CAAC,CAAC;YACH,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}