agents-relay 1.0.138 → 1.0.139

package/dist/pool.js

@@ -28,6 +28,7 @@ export class RepositoryWorkerPool {
     reconcileLocks = new Map();
     targetedReconcileFlights = new Map();
     jobTargets = new Map();
+    targetSeeds = new Map();
     constructor(client, repository, trustedAuthors, concurrency, makeReconciler, isRateLimited = () => false) {
         this.client = client;
         this.trustedAuthors = trustedAuthors;
@@ -99,6 +100,11 @@ export class RepositoryWorkerPool {
             this.running = false;
         }
     }
+    seedTargetSnapshot(repository, prNumber, jobId, seed) {
+        if (!this.repositories.includes(repository))
+            return;
+        this.targetSeeds.set(`${repository}#${prNumber}:${jobId}`, seed);
+    }
     async reconcileTarget(repository, prNumber, jobId) {
         if (this.isRateLimited() || !this.repositories.includes(repository))
             return;
@@ -108,8 +114,11 @@ export class RepositoryWorkerPool {
         if (existing)
             return existing;
         const flight = this.reconcileExclusive(jobId, async () => {
-            const store = new GitHubStore(this.client, repository, prNumber, this.trustedAuthors);
-            await withGitHubRequestContext(this.client, 'pool.targeted-reconcile', () => this.makeReconciler(store, Math.max(1, this.concurrency)).reconcile(jobId));
+            const seedKey = `${repository}#${prNumber}:${jobId}`;
+            const seed = this.targetSeeds.get(seedKey) ?? null;
+            this.targetSeeds.delete(seedKey);
+            const store = new GitHubStore(this.client, repository, prNumber, this.trustedAuthors, seed);
+            await withGitHubRequestContext(this.client, seed ? 'pool.targeted-reconcile.seeded' : 'pool.targeted-reconcile', () => this.makeReconciler(store, Math.max(1, this.concurrency)).reconcile(jobId));
         });
         this.targetedReconcileFlights.set(key, flight);
         try {

package/dist/relayd.js

@@ -9,7 +9,7 @@ import { serveDashboard } from './dashboard.js';
 import { NatsEventBus } from './events.js';
 import { RepositoryWorkerPool, aggregateManagedGitHubJobs } from './pool.js';
 import { Reconciler } from './reconciler.js';
-import { dashboardManagedJobs, GitHubStore, InMemoryStore, loadManagedGitHubPullRequestGraphql, withGitHubRequestContext } from './store.js';
+import { dashboardManagedJobs, GitHubStore, InMemoryStore, loadManagedGitHubPullRequestSnapshotGraphql, withGitHubRequestContext } from './store.js';
 import { githubAuthContext } from './github-auth.js';
 import { dashboardJobAction, isEntrypoint, runtimePlanner, runtimeReviewer, startService, SERVICE_DEFAULTS } from './cli.js';
 import { dashboardTaskAction, dashboardTaskSetState } from './dashboard-actions.js';
@@ -143,21 +143,21 @@ export async function runDaemon(argv) {
         catch { /* cache is best effort */ }
     };
     const pullRequestRefreshFlights = new Map();
-    const refreshPullRequest = async (repository, prNumber) => {
+    const refreshPullRequestSnapshot = async (repository, prNumber) => {
         if (client.rateLimitStatus().limited)
-            return cachedOverviews.filter(item => item.job.repository === repository && item.job.prNumber === prNumber);
+            return { overviews: cachedOverviews.filter(item => item.job.repository === repository && item.job.prNumber === prNumber), seed: { prNode: {}, comments: null } };
         const key = `${repository}#${prNumber}`;
         const existing = pullRequestRefreshFlights.get(key);
         if (existing)
             return existing;
         const flight = (async () => {
             githubReadPaths.prRefreshes += 1;
-            const refreshed = await withGitHubRequestContext(client, 'relayd.pr-refresh', () => loadManagedGitHubPullRequestGraphql(client, repository, prNumber, trusted));
+            const snapshot = await withGitHubRequestContext(client, 'relayd.pr-refresh', () => loadManagedGitHubPullRequestSnapshotGraphql(client, repository, prNumber, trusted));
             cachedOverviews = cachedOverviews.filter(item => item.job.repository !== repository || item.job.prNumber !== prNumber);
-            cachedOverviews.push(...refreshed);
+            cachedOverviews.push(...snapshot.overviews);
             cachedOverviews = dashboardManagedJobs(cachedOverviews);
             await persistCache();
-            return refreshed;
+            return snapshot;
         })();
         pullRequestRefreshFlights.set(key, flight);
         try {
@@ -168,6 +168,7 @@ export async function runDaemon(argv) {
                 pullRequestRefreshFlights.delete(key);
         }
     };
+    const refreshPullRequest = async (repository, prNumber) => (await refreshPullRequestSnapshot(repository, prNumber)).overviews;
     const initial = cachedOverviews;
     const fallback = initial[0]?.job;
     const dashboardStore = fallback ? new GitHubStore(client, fallback.repository, fallback.prNumber, trusted) : new InMemoryStore();
@@ -206,8 +207,10 @@ export async function runDaemon(argv) {
         refresh: async (event) => {
             if (!event.repository || !event.pullRequest)
                 return [];
-            const refreshed = await refreshPullRequest(event.repository, event.pullRequest);
-            return refreshed.map(item => item.job.id);
+            const snapshot = await refreshPullRequestSnapshot(event.repository, event.pullRequest);
+            for (const item of snapshot.overviews)
+                pool.seedTargetSnapshot(event.repository, event.pullRequest, item.job.id, snapshot.seed);
+            return snapshot.overviews.map(item => item.job.id);
         },
         wake: async (event) => {
             if (!event.repository || !event.pullRequest)

package/dist/store-github.js

@@ -11,11 +11,17 @@ export class GitHubStore {
     commentsCache = null;
     commentsFlight = null;
     clearReadSnapshot() { this.prNodeCache = null; this.commentsCache = null; this.commentsFlight = null; }
-    constructor(client, repository, prNumber, trustedAuthors = new Set()) {
+    constructor(client, repository, prNumber, trustedAuthors = new Set(), seed = null) {
         this.client = client;
         this.repository = repository;
         this.prNumber = prNumber;
         this.trustedAuthors = trustedAuthors;
+        if (seed) {
+            if (Object.keys(seed.prNode).length > 0)
+                this.prNodeCache = Promise.resolve(seed.prNode);
+            if (seed.comments)
+                this.commentsCache = [...seed.comments];
+        }
     }
     taskKey(task) { return `${this.repository}#${this.prNumber}:${task.jobId ?? ''}/${task.id}`; }
     taskKeyFromBody(body) { const task = parseTask(body); return task ? this.taskKey(task) : null; }

package/dist/store.js

@@ -12,28 +12,39 @@ function publicationFromComments(rows, trustedAuthors) {
     });
     return allowed.map(item => parsePackagePublicationMarker(String(item.body ?? ''))).filter((item) => item !== null).at(-1) ?? null;
 }
-export async function loadManagedGitHubPullRequestGraphql(client, repository, prNumber, trustedAuthors) {
+export async function loadManagedGitHubPullRequestSnapshotGraphql(client, repository, prNumber, trustedAuthors) {
     const [owner, name] = repository.split('/', 2);
     if (!owner || !name || !Number.isInteger(prNumber) || prNumber <= 0 || !client.graphql)
-        return [];
-    const query = 'query($owner:String!,$name:String!,$number:Int!){repository(owner:$owner,name:$name){pullRequest(number:$number){state isDraft merged mergedAt closedAt body comments(last:100){nodes{databaseId url author{login} body}}}}}';
+        return { overviews: [], seed: { prNode: {}, comments: null } };
+    const query = 'query($owner:String!,$name:String!,$number:Int!){repository(owner:$owner,name:$name){pullRequest(number:$number){id state isDraft merged mergedAt closedAt body headRefName headRefOid comments(last:100){totalCount nodes{id databaseId url updatedAt author{login} body}}}}}';
     const data = await client.graphql(query, { owner, name, number: prNumber });
     const repo = data.repository;
     const pr = repo?.pullRequest;
     if (!pr)
-        return [];
-    const comments = pr.comments?.nodes;
-    const trusted = (Array.isArray(comments) ? comments : []).filter(item => {
-        const row = item;
+        return { overviews: [], seed: { prNode: {}, comments: null } };
+    const commentConnection = pr.comments;
+    const commentRows = (Array.isArray(commentConnection?.nodes) ? commentConnection.nodes : []);
+    const totalCount = Number(commentConnection?.totalCount ?? commentRows.length);
+    const seedComments = Number.isFinite(totalCount) && totalCount <= commentRows.length ? commentRows.map(row => {
         const author = row.author;
+        return {
+            id: Number(row.databaseId ?? 0),
+            nodeId: typeof row.id === 'string' ? row.id : undefined,
+            url: typeof row.url === 'string' ? row.url : undefined,
+            body: String(row.body ?? ''),
+            updatedAt: String(row.updatedAt ?? ''),
+            authorLogin: typeof author?.login === 'string' ? author.login : null,
+        };
+    }) : null;
+    const trusted = commentRows.filter(item => {
+        const author = item.author;
         return typeof author?.login === 'string' && trustedAuthors.has(author.login);
     });
     const bodyJob = parseJob(typeof pr.body === 'string' ? pr.body : '');
     const commentJobs = trusted.map(item => parseJob(String(item.body ?? ''))).filter((item) => item !== null);
     const jobs = bodyJob ? [bodyJob] : commentJobs;
-    const publication = publicationFromComments((Array.isArray(comments) ? comments : []), trustedAuthors);
-    const tasks = trusted.map(item => {
-        const row = item;
+    const publication = publicationFromComments(commentRows, trustedAuthors);
+    const tasks = trusted.map(row => {
         const task = parseTask(String(row.body ?? ''));
         if (!task)
             return null;
@@ -42,12 +53,18 @@ export async function loadManagedGitHubPullRequestGraphql(client, repository, pr
     }).filter((item) => item !== null);
     const state = String(pr.state ?? '').toUpperCase();
     const githubState = pr.merged === true ? 'MERGED' : state === 'CLOSED' ? 'CLOSED' : 'OPEN';
-    return jobs.map(job => {
+    const overviews = jobs.map(job => {
         job.tasks = tasks.filter(task => task.jobId === job.id || (!task.jobId && jobs.length === 1));
         if (!job.description && typeof pr.body === 'string' && pr.body)
             job.description = pr.body;
+        if (typeof pr.headRefOid === 'string' && pr.headRefOid)
+            job.headSha = pr.headRefOid;
         return { job, githubState, draft: pr.isDraft === true, mergedAt: typeof pr.mergedAt === 'string' ? pr.mergedAt : null, closedAt: typeof pr.closedAt === 'string' ? pr.closedAt : null, publication };
     });
+    return { overviews, seed: { prNode: pr, comments: seedComments } };
+}
+export async function loadManagedGitHubPullRequestGraphql(client, repository, prNumber, trustedAuthors) {
+    return (await loadManagedGitHubPullRequestSnapshotGraphql(client, repository, prNumber, trustedAuthors)).overviews;
 }
 export async function loadManagedGitHubJobGraphql(client, repository, prNumber, jobId, trustedAuthors) {
     return (await loadManagedGitHubPullRequestGraphql(client, repository, prNumber, trustedAuthors)).find(item => item.job.id === jobId) ?? null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agents-relay",
-  "version": "1.0.138",
+  "version": "1.0.139",
   "description": "Durable async agent jobs coordinated through GitHub pull requests",
   "type": "module",
   "bin": {