agent-relay-runner 0.78.5 → 0.78.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-relay-runner",
-  "version": "0.78.5",
+  "version": "0.78.7",
   "description": "Unified provider lifecycle runner for Agent Relay",
   "type": "module",
   "bin": {
@@ -20,7 +20,7 @@
     "directory": "runner"
   },
   "dependencies": {
-    "agent-relay-sdk": "0.2.56"
+    "agent-relay-sdk": "0.2.57"
   },
   "devDependencies": {
     "@types/bun": "latest",

package/plugins/claude/.claude-plugin/plugin.json

@@ -1,7 +1,7 @@
 {
   "name": "agent-relay-runner",
   "description": "Thin Agent Relay runner bridge for Claude Code",
-  "version": "0.78.5",
+  "version": "0.78.7",
   "agentRelayContracts": {
     "providerPluginProtocol": 1
   }

package/src/adapters/codex-agent-message-capture.ts

@@ -1,16 +1,20 @@
 type EmitAgentMessage = (body: string) => void;
+type ResponseEntry = { itemId?: string; body: string };
+type CompleteItemOptions = { includeInResponse?: boolean; phase?: unknown };
 
 export class CodexAgentMessageCapture {
-  private segments: string[][] = [];
-  private currentSegment: string[] = [];
+  private segments: ResponseEntry[][] = [];
+  private currentSegment: ResponseEntry[] = [];
   private readonly buffers = new Map<string, string>();
   private readonly flushedLengths = new Map<string, number>();
+  private readonly responseRecordedLengths = new Map<string, number>();
 
   reset(): void {
     this.segments = [];
     this.currentSegment = [];
     this.buffers.clear();
     this.flushedLengths.clear();
+    this.responseRecordedLengths.clear();
   }
 
   appendDelta(itemId: string | undefined, delta: string): void {
@@ -18,8 +22,14 @@ export class CodexAgentMessageCapture {
     this.buffers.set(itemId, `${this.buffers.get(itemId) ?? ""}${delta}`);
   }
 
-  completeItem(itemId: string | undefined, text: string | undefined, emit: EmitAgentMessage): void {
-    this.flushText(itemId, text ?? (itemId ? this.buffers.get(itemId) : undefined), emit);
+  completeItem(itemId: string | undefined, text: string | undefined, emit: EmitAgentMessage, options: CompleteItemOptions = {}): void {
+    const fullText = text ?? (itemId ? this.buffers.get(itemId) : undefined);
+    this.flushText(itemId, fullText, emit);
+    if (options.includeInResponse === false || options.phase === "commentary") {
+      this.forgetResponseItem(itemId);
+    } else {
+      this.recordResponseText(itemId, fullText);
+    }
     if (!itemId) return;
     this.buffers.delete(itemId);
     this.flushedLengths.delete(itemId);
@@ -36,11 +46,14 @@ export class CodexAgentMessageCapture {
     this.closeSegment(emit);
     const finalSegment = this.segments.at(-1);
     this.segments = [];
-    return finalSegment?.length ? finalSegment.join("\n\n").trim() : undefined;
+    return finalSegment?.length ? finalSegment.map((entry) => entry.body).join("\n\n").trim() : undefined;
   }
 
   private flushPending(emit: EmitAgentMessage): void {
-    for (const [itemId, text] of [...this.buffers.entries()]) this.flushText(itemId, text, emit);
+    for (const [itemId, text] of [...this.buffers.entries()]) {
+      this.flushText(itemId, text, emit);
+      this.recordResponseText(itemId, text);
+    }
   }
 
   private flushText(itemId: string | undefined, text: string | undefined, emit: EmitAgentMessage): void {
@@ -50,7 +63,30 @@ export class CodexAgentMessageCapture {
     if (itemId) this.flushedLengths.set(itemId, text.length);
     const body = chunk.trim();
     if (!body) return;
-    this.currentSegment.push(body);
     emit(body);
   }
+
+  private recordResponseText(itemId: string | undefined, text: string | undefined): void {
+    if (!text) return;
+    const body = text.trim();
+    if (!body) return;
+    if (itemId) {
+      const offset = this.responseRecordedLengths.get(itemId) ?? 0;
+      const chunk = text.slice(offset).trim();
+      this.responseRecordedLengths.set(itemId, text.length);
+      if (!chunk) return;
+      this.currentSegment.push({ itemId, body: chunk });
+      return;
+    }
+    this.currentSegment.push({ body });
+  }
+
+  private forgetResponseItem(itemId: string | undefined): void {
+    if (!itemId) return;
+    this.responseRecordedLengths.delete(itemId);
+    this.currentSegment = this.currentSegment.filter((entry) => entry.itemId !== itemId);
+    this.segments = this.segments
+      .map((segment) => segment.filter((entry) => entry.itemId !== itemId))
+      .filter((segment) => segment.length > 0);
+  }
 }

package/src/adapters/codex.ts

@@ -504,7 +504,7 @@ export class CodexAdapter implements ProviderAdapter {
     // buffered reasoning first so it lands in transcript order, ahead of this tool/message.
     if (type !== "reasoning") this.flushBufferedReasoning();
     if (type === "agentMessage") {
-      this.agentMessageCapture.completeItem(itemId, stringValue(item.text), (body) => this.emitAgentNarration(body));
+      this.agentMessageCapture.completeItem(itemId, stringValue(item.text), (body) => this.emitAgentNarration(body), { phase: item.phase });
       return;
     }
     if (type === "userMessage") {