@mmerterden/multi-agent-pipeline 20.2.0 → 20.3.0

package/pipeline/scripts/triage-memory.mjs

@@ -24,6 +24,9 @@
  *               (--state fills task-id / task-title / stack when the flag is absent)
  *   query    --issue "<text>" [--file-glob "Auth*"] [--top N]
  *               [--repo-slug <slug>] [--json]
+ *   prior-art --findings <path|-> [--top N] [--repo-slug <slug>]
+ *               One `query` per finding (its issue, its file's directory),
+ *               printed as a JSON array of hit lists in finding order.
  *   show     --id <id> [--repo-slug <slug>]
  *   timeline --anchor <id> [--before N] [--after N] [--repo-slug <slug>]
  *               What surrounded a finding when it was recorded. The corpus is
@@ -51,7 +54,7 @@
 import { readFileSync, existsSync, mkdirSync, appendFileSync, statSync } from "node:fs";
 import { homedir } from "node:os";
 import { execFileSync } from "node:child_process";
-import { join, basename, dirname } from "node:path";
+import { join, basename, dirname, posix } from "node:path";
 import {
   bm25,
   recencyRank,
@@ -308,8 +311,6 @@ function ingest() {
 }
 
 function query() {
-  const issueText = flags.issue || "";
-  const fileGlob = flags["file-glob"] || null;
   const cfg = recallConfig();
   const topN = flags.top ? Math.max(1, Number(flags.top)) : cfg.maxResults;
   const slug = flags["repo-slug"] ? String(flags["repo-slug"]) : repoSlug();
@@ -319,6 +320,46 @@ function query() {
     process.exitCode = 2;
     return;
   }
+  const { total, hits } = rank(corpus, flags.issue || "", flags["file-glob"] || null, topN, cfg);
+  process.stdout.write(JSON.stringify({ ok: true, slug, total, hits }) + "\n");
+  process.exitCode = 0;
+  return;
+}
+
+// Prior art for every finding of a review in one call: for each finding, the
+// query `--issue <issue> --file-glob <dir of file>/*` would answer, in finding
+// order. Phase 3 hands the triage agent this list; one command instead of a
+// per-finding shell loop, which the unattended guard refuses. Reads a findings
+// array, or an object with `findings[]`, from --findings <path> or `-` (stdin).
+function priorArt() {
+  const src = flags.findings;
+  if (!src || src === true) die("prior-art needs --findings <path|->", 1);
+  let parsed;
+  try {
+    parsed = JSON.parse(readFileSync(src === "-" ? 0 : String(src), "utf8"));
+  } catch (e) {
+    die(`prior-art: cannot read findings: ${e.message}`, 1);
+  }
+  const list = Array.isArray(parsed)
+    ? parsed
+    : Array.isArray(parsed?.findings)
+      ? parsed.findings
+      : [];
+  const cfg = recallConfig();
+  const topN = flags.top ? Math.max(1, Number(flags.top)) : cfg.maxResults;
+  const slug = flags["repo-slug"] ? String(flags["repo-slug"]) : repoSlug();
+  const corpus = readCorpus(slug);
+  const out = list.map((f) => {
+    if (corpus.length === 0) return [];
+    const file = typeof f?.file === "string" && f.file ? f.file : null;
+    const glob = file ? `${posix.dirname(file)}/*` : null;
+    return rank(corpus, String(f?.issue ?? ""), glob, topN, cfg).hits;
+  });
+  process.stdout.write(JSON.stringify(out) + "\n");
+  process.exitCode = 0;
+}
+
+function rank(corpus, issueText, fileGlob, topN, cfg) {
   const qTerms = tokens(issueText);
   const fileRe = fileGlob ? globToRegExp(fileGlob) : null;
   // --file-glob stays a hard filter, not one more ranking in the fusion: a
@@ -386,11 +427,7 @@ function query() {
       score: Math.round(score * 1000) / 1000,
     };
   });
-  process.stdout.write(
-    JSON.stringify({ ok: true, slug, total: filtered.length, hits: ranked }) + "\n",
-  );
-  process.exitCode = 0;
-  return;
+  return { total: filtered.length, hits: ranked };
 }
 
 // Drill-down: a compact prior-art line carries only its id, so the agent that
@@ -493,6 +530,9 @@ switch (SUB) {
   case "query":
     query();
     break;
+  case "prior-art":
+    priorArt();
+    break;
   case "show":
     showCmd();
     break;
@@ -507,7 +547,7 @@ switch (SUB) {
     break;
   default:
     process.stderr.write(
-      "usage: triage-memory.mjs <ingest|query|show|timeline|path|stats> [flags]\n",
+      "usage: triage-memory.mjs <ingest|query|prior-art|show|timeline|path|stats> [flags]\n",
     );
     process.exit(64);
 }