agents-relay 1.0.116 → 1.0.118

package/dist/reconciler-runtime.js

@@ -7,6 +7,7 @@ import { workflowEffectFailureEvent } from './workflow/contracts.js';
 export function executionEventSource(jobId, taskId, executionId) { return `job/${jobId}/task/${taskId}/execution/${executionId}`; }
 function eventSourceId(event) { return typeof event.source?.id === 'string' ? event.source.id : undefined; }
 const terminalExecutionClaims = new Set();
+const activeExecutionOwners = new Map();
 function claimTerminalExecution(sourceId) {
     terminalExecutionClaims.add(sourceId);
     const cleanup = setTimeout(() => terminalExecutionClaims.delete(sourceId), 60000);
@@ -61,6 +62,12 @@ export class ReconcilerRuntimeBase {
         }
     }
     async handleEvent(event) {
+        const sourceId = eventSourceId(event);
+        const activeOwner = sourceId ? activeExecutionOwners.get(sourceId) : undefined;
+        if (activeOwner && activeOwner !== this) {
+            await activeOwner.handleEvent(event);
+            return;
+        }
         if (event.type === 'task.started') {
             await this.markWorkerStarted(event);
             return;
@@ -87,10 +94,13 @@ export class ReconcilerRuntimeBase {
             return async () => { };
         }
     }
-    async idle() { while (this.live.size > 0 || this.plannerLive.size > 0 || this.continuationLive.size > 0 || this.reconciling)
+    async idle() { while ([...this.live.values()].some(item => !item.processSettled) || this.plannerLive.size > 0 || this.continuationLive.size > 0 || this.reconciling)
         await new Promise(resolve => setTimeout(resolve, 25)); }
     async close() {
         for (const live of this.live.values()) {
+            const sourceId = executionEventSource(live.jobId, live.taskId, live.executionId);
+            if (activeExecutionOwners.get(sourceId) === this)
+                activeExecutionOwners.delete(sourceId);
             live.controller.abort();
             live.execution.cancel();
         }
@@ -144,10 +154,19 @@ export class ReconcilerRuntimeBase {
             clearTimeout(timer); await unsubscribe(); };
         return { ack, stop };
     }
+    registerExecutionOwner(jobId, taskId, executionId) {
+        activeExecutionOwners.set(executionEventSource(jobId, taskId, executionId), this);
+    }
     clearExecution(taskId, executionId) {
         const live = this.live.get(taskId);
-        if (!executionId || live?.executionId === executionId)
+        if (!executionId || live?.executionId === executionId) {
+            if (live) {
+                const sourceId = executionEventSource(live.jobId, live.taskId, live.executionId);
+                if (activeExecutionOwners.get(sourceId) === this)
+                    activeExecutionOwners.delete(sourceId);
+            }
             this.live.delete(taskId);
+        }
         const timer = this.terminalTimers.get(taskId);
         if (timer)
             clearTimeout(timer);
@@ -228,7 +247,7 @@ export class ReconcilerRuntimeBase {
         const live = this.live.get(taskId);
         const liveWorkerStarted = live?.executionId === executionId && live.workerStarted;
         if (live?.executionId === executionId)
-            this.live.delete(taskId);
+            live.processSettled = true;
         const job = await this.store.load(jobId);
         const task = job.tasks.find(item => item.id === taskId);
         const workerStarted = liveWorkerStarted || task?.state === 'RUNNING';
@@ -249,6 +268,12 @@ export class ReconcilerRuntimeBase {
     async failUnexpectedExit(jobId, taskId, executionId, workerStarted) {
         if (terminalExecutionClaimed(jobId, taskId, executionId))
             return;
+        const current = await this.store.load(jobId);
+        const currentTask = current.tasks.find(item => item.id === taskId);
+        if (!currentTask || currentTask.executionId !== executionId || ['SUCCEEDED', 'FAILED', 'BLOCKED', 'CANCELLED'].includes(currentTask.state)) {
+            this.clearExecution(taskId, executionId);
+            return;
+        }
         const reason = workerStarted ? 'worker_exit_without_terminal_event' : 'worker_exit_before_start';
         const message = workerStarted
             ? `Worker ${taskId} exited after task.started without publishing a terminal task event`

package/dist/reconciler.js

@@ -419,7 +419,8 @@ export class Reconciler extends ReconcilerReviewBase {
         void (execution.launched ?? Promise.resolve()).then(() => this.emit(eventFor(job.id, task.id, task.parentTaskId, 'task.process.launched', 'running', `Task ${task.id} process launched`, 'user', { adapter: adapter.id, model: task.routing?.model }, { agent: 'relay', component: 'agents-relay' }))).catch(error => this.failLaunch(job.id, task.id, durableExecutionId, error instanceof Error ? error.message : String(error), 'launch'));
         if (startAckSetup)
             void startAckSetup.ack.finally(() => { void startAckSetup.stop(); }).catch(() => { });
-        this.live.set(task.id, { jobId: job.id, taskId: task.id, executionId: durableExecutionId, execution, controller, launchBarrier: Date.now(), workerStarted: false });
+        this.live.set(task.id, { jobId: job.id, taskId: task.id, executionId: durableExecutionId, execution, controller, launchBarrier: Date.now(), workerStarted: false, processSettled: false });
+        this.registerExecutionOwner(job.id, task.id, durableExecutionId);
         const terminalTimer = setTimeout(() => {
             this.detach(this.timeoutTask(job.id, task.id, durableExecutionId), job.id, task.id, 'worker terminal-event timeout');
         }, Math.max(1, task.timeoutMs));

package/dist/store-github.js

@@ -1,6 +1,7 @@
 import { parseAgent, parseJob, parseTask, renderAgentComment, renderJobBody, renderTaskComment, JOB_MARKER } from './markers.js';
 import { isTaskForJob, sameTaskDefinition } from './store-task-definition.js';
 import { resolveBlockerDependencies } from './blockers.js';
+const TASK_WRITE_OVERLAY = new Map();
 export class GitHubStore {
     client;
     repository;
@@ -12,6 +13,29 @@ export class GitHubStore {
         this.prNumber = prNumber;
         this.trustedAuthors = trustedAuthors;
     }
+    taskKey(task) { return `${this.repository}#${this.prNumber}:${task.jobId ?? ''}/${task.id}`; }
+    taskKeyFromBody(body) { const task = parseTask(body); return task ? this.taskKey(task) : null; }
+    overlayTaskWrites(comments) {
+        if (TASK_WRITE_OVERLAY.size === 0)
+            return comments;
+        const result = [...comments];
+        for (const [key, overlay] of [...TASK_WRITE_OVERLAY]) {
+            const index = result.findIndex(comment => this.taskKeyFromBody(comment.body) === key);
+            if (index >= 0) {
+                const remote = result[index];
+                if (remote.updatedAt && overlay.updatedAt && remote.updatedAt >= overlay.updatedAt) {
+                    TASK_WRITE_OVERLAY.delete(key);
+                    continue;
+                }
+                result[index] = { ...remote, ...overlay, id: remote.id || overlay.id, nodeId: remote.nodeId ?? overlay.nodeId, url: remote.url ?? overlay.url, authorLogin: remote.authorLogin ?? overlay.authorLogin };
+            }
+            else {
+                TASK_WRITE_OVERLAY.delete(key);
+            }
+        }
+        return result;
+    }
+    rememberTaskWrite(task, comment, body) { TASK_WRITE_OVERLAY.set(this.taskKey(task), { ...comment, body }); }
     isTrusted(comment) { return comment.authorLogin !== null && this.trustedAuthors.has(comment.authorLogin); }
     repoParts() {
         const [owner, name] = this.repository.split('/', 2);
@@ -61,7 +85,7 @@ export class GitHubStore {
             const pageInfo = comments?.pageInfo;
             cursor = pageInfo?.hasNextPage === true && typeof pageInfo.endCursor === 'string' ? pageInfo.endCursor : null;
         } while (cursor);
-        return all;
+        return this.overlayTaskWrites(all);
     }
     async comments() { return this.list(); }
     async pullRequestState() {
@@ -151,12 +175,31 @@ export class GitHubStore {
     async listAgents() { return (await this.list()).filter(comment => this.isTrusted(comment)).map(x => parseAgent(x.body)).filter((x) => x !== null); }
     async comment(body) {
         const pr = await this.prNode();
-        await this.graphql('mutation($subjectId:ID!,$body:String!){addComment(input:{subjectId:$subjectId,body:$body}){commentEdge{node{id}}}}', { subjectId: pr.id, body });
+        const data = await this.graphql('mutation($subjectId:ID!,$body:String!){addComment(input:{subjectId:$subjectId,body:$body}){commentEdge{node{id databaseId url body updatedAt author{login}}}}}', { subjectId: pr.id, body });
+        const edge = data.addComment?.commentEdge;
+        const node = edge?.node;
+        return {
+            id: Number(node?.databaseId ?? 0),
+            nodeId: typeof node?.id === 'string' ? node.id : undefined,
+            url: typeof node?.url === 'string' ? node.url : undefined,
+            body: typeof node?.body === 'string' ? node.body : body,
+            updatedAt: typeof node?.updatedAt === 'string' ? node.updatedAt : new Date().toISOString(),
+            authorLogin: typeof node?.author?.login === 'string' ? String((node?.author).login) : [...this.trustedAuthors][0] ?? null,
+        };
     }
     async updateComment(comment, body) {
         if (!comment.nodeId)
             throw new Error('GitHub comment node id is required for GraphQL update');
-        await this.graphql('mutation($id:ID!,$body:String!){updateIssueComment(input:{id:$id,body:$body}){issueComment{id}}}', { id: comment.nodeId, body });
+        const data = await this.graphql('mutation($id:ID!,$body:String!){updateIssueComment(input:{id:$id,body:$body}){issueComment{id databaseId url body updatedAt author{login}}}}', { id: comment.nodeId, body });
+        const node = data.updateIssueComment?.issueComment;
+        return {
+            id: Number(node?.databaseId ?? comment.id),
+            nodeId: typeof node?.id === 'string' ? node.id : comment.nodeId,
+            url: typeof node?.url === 'string' ? node.url : comment.url,
+            body: typeof node?.body === 'string' ? node.body : body,
+            updatedAt: typeof node?.updatedAt === 'string' ? node.updatedAt : new Date().toISOString(),
+            authorLogin: typeof node?.author?.login === 'string' ? String((node?.author).login) : comment.authorLogin,
+        };
     }
     async deleteComment(comment) {
         if (!comment.nodeId)
@@ -190,7 +233,9 @@ export class GitHubStore {
             return existing !== null && existing.jobId === task.jobId && existing.id === task.id;
         });
         if (matches.length === 0) {
-            await this.comment(renderTaskComment(task));
+            const body = renderTaskComment(task);
+            const persisted = await this.comment(body);
+            this.rememberTaskWrite(task, persisted, body);
             return;
         }
         const parsed = matches.map(comment => ({ comment, task: parseTask(comment.body) }));
@@ -203,7 +248,9 @@ export class GitHubStore {
         const comments = (await this.list()).filter(comment => this.isTrusted(comment));
         const existing = comments.filter(comment => { const markerTask = parseTask(comment.body); return markerTask !== null && markerTask.jobId === task.jobId && markerTask.id === task.id; });
         if (existing.length > 0) {
-            await this.updateComment(existing[0], renderTaskComment(task));
+            const body = renderTaskComment(task);
+            const persisted = await this.updateComment(existing[0], body);
+            this.rememberTaskWrite(task, persisted, body);
             for (const duplicate of existing.slice(1))
                 await this.deleteComment(duplicate);
         }

package/package.json

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