@agentrouter-top/relay-dsh-plugin-codex 0.2.2-agentrouter.4 → 0.2.2-agentrouter.6

package/lib/host-plugin.js

@@ -2908,7 +2908,7 @@ var CodexDshAdapter = class extends LlmAdapter {
 			this.appendActivity(agent, threadId, turnId, withCommandOutput(state, key, item), "completed", state);
 			return [];
 		}
-		if (isCodexActivityItem(item)) this.appendActivity(agent, threadId, turnId, item, "completed", state);
+		if (isCodexActivityItem(item) && item.type !== "imageGeneration") this.appendActivity(agent, threadId, turnId, item, "completed", state);
 		if (item.type === "mcpToolCall" && item.status === "completed") {
 			const images = (Array.isArray(item.result?.content) ? item.result.content : []).map((entry, contentIndex) => ({
 				entry,
@@ -2959,11 +2959,22 @@ var CodexDshAdapter = class extends LlmAdapter {
 			return chunks;
 		}
 		if (item.type === "imageGeneration" || item.type === "imageView") {
-			if (!this.attachments) return [];
+			if (!this.attachments || item.type === "imageGeneration" && [
+				"failed",
+				"cancelled",
+				"canceled"
+			].includes(item.status)) {
+				if (item.type === "imageGeneration") this.appendActivity(agent, threadId, turnId, {
+					...item,
+					status: "failed"
+				}, "completed", state);
+				return [];
+			}
 			const roots = [resolve(agent.session.header.cwd ?? process.cwd()), resolve(this.codexHome ?? resolve(homedir(), ".codex"), "generated_images")];
 			const index = state.nextIndex++;
 			try {
 				const attachment = item.type === "imageGeneration" ? await importCodexGeneratedImage(item, roots, this.attachments) : await importCodexImage(item.path, roots, this.attachments);
+				if (item.type === "imageGeneration") this.appendActivity(agent, threadId, turnId, item, "completed", state, attachment.attachmentId);
 				return [{
 					type: "block-start",
 					index,
@@ -2977,6 +2988,10 @@ var CodexDshAdapter = class extends LlmAdapter {
 					}
 				}];
 			} catch (error) {
+				if (item.type === "imageGeneration") this.appendActivity(agent, threadId, turnId, {
+					...item,
+					status: "failed"
+				}, "completed", state);
 				const label = basename(item.path ?? item.savedPath ?? `codex-${item.id}`);
 				const reason = imagePreviewFailureReason(error);
 				this.logger.warn?.("Codex image preview unavailable", {
@@ -3002,7 +3017,7 @@ var CodexDshAdapter = class extends LlmAdapter {
 		}
 		return [];
 	}
-	appendActivity(agent, threadId, turnId, item, phase, state) {
+	appendActivity(agent, threadId, turnId, item, phase, state, imageAttachmentId) {
 		const id = activityItemId(item);
 		if (!id) return;
 		const merged = mergeActivityItem(state.activityItems.get(id) ?? {}, item);
@@ -3041,7 +3056,7 @@ var CodexDshAdapter = class extends LlmAdapter {
 			state.startedActivities.add(id);
 		}
 		if (phase === "completed" && !state.completedActivities.has(id)) {
-			const payload = activityPayload(threadId, turnId, merged, "completed");
+			const payload = activityPayload(threadId, turnId, merged, "completed", imageAttachmentId);
 			const callId = toolCallId(`relay-codex:${JSON.stringify([
 				threadId,
 				turnId,
@@ -3379,8 +3394,12 @@ function permissionConfiguration(events) {
 function reasoningText(item) {
 	return [...item.summary ?? [], ...item.content ?? []].filter(Boolean).join("\n\n");
 }
-function activityPayload(threadId, turnId, item, phase) {
+function activityPayload(threadId, turnId, item, phase, imageAttachmentId) {
 	const activity = normalizeCodexActivity(item, phase);
+	if (item.type === "imageGeneration") activity.imageGeneration = {
+		version: 1,
+		...typeof imageAttachmentId === "string" ? { attachmentId: imageAttachmentId } : {}
+	};
 	return {
 		version: 1,
 		threadId,