Normalize agent attachment output (#122)

2026-05-02 19:50:32 +09:00
parent 9e45534de0
commit d3c02265e5
16 changed files with 414 additions and 124 deletions
--- a/runners/shared/test/normalize-agent-output.test.ts
+++ b/runners/shared/test/normalize-agent-output.test.ts
@@ -0,0 +1,121 @@
+import { describe, expect, it } from 'vitest';
+
+import {
+  extractMarkdownImageAttachments,
+  normalizeAgentOutput,
+} from '../src/agent-protocol.js';
+
+describe('normalizeAgentOutput', () => {
+  it('extracts markdown image attachments without rewriting normal links', () => {
+    expect(
+      extractMarkdownImageAttachments(
+        '결과입니다.\n![screenshot](/tmp/result.png)\n[code](/tmp/source.ts#L10)',
+      ),
+    ).toEqual({
+      cleanText: '결과입니다.\n\n[code](/tmp/source.ts#L10)',
+      attachments: [
+        {
+          path: '/tmp/result.png',
+          name: 'result.png',
+        },
+      ],
+    });
+  });
+
+  it('normalizes markdown image output into internal attachments', () => {
+    expect(
+      normalizeAgentOutput(
+        'TASK_DONE\n\n스크린샷입니다.\n![screenshot](/tmp/screenshot.png)',
+      ),
+    ).toEqual({
+      result: 'TASK_DONE\n\n스크린샷입니다.',
+      output: {
+        visibility: 'public',
+        text: 'TASK_DONE\n\n스크린샷입니다.',
+        attachments: [
+          {
+            path: '/tmp/screenshot.png',
+            name: 'screenshot.png',
+          },
+        ],
+      },
+      attachmentSource: 'markdown-image',
+    });
+  });
+
+  it('normalizes legacy image tags into internal attachments', () => {
+    expect(
+      normalizeAgentOutput(
+        'TASK_DONE\n\n스크린샷입니다.\n[Image: screenshot.png → /tmp/legacy.png]',
+      ),
+    ).toEqual({
+      result: 'TASK_DONE\n\n스크린샷입니다.',
+      output: {
+        visibility: 'public',
+        text: 'TASK_DONE\n\n스크린샷입니다.',
+        attachments: [
+          {
+            path: '/tmp/legacy.png',
+            name: 'legacy.png',
+          },
+        ],
+      },
+      attachmentSource: 'image-tag',
+    });
+  });
+
+  it('normalizes short image tags documented in prompts', () => {
+    expect(
+      normalizeAgentOutput('TASK_DONE\n\n[Image: /tmp/short-form.png]'),
+    ).toEqual({
+      result: 'TASK_DONE',
+      output: {
+        visibility: 'public',
+        text: 'TASK_DONE',
+        attachments: [
+          {
+            path: '/tmp/short-form.png',
+            name: 'short-form.png',
+          },
+        ],
+      },
+      attachmentSource: 'image-tag',
+    });
+  });
+
+  it('keeps legacy ejclaw JSON as compatibility input', () => {
+    expect(
+      normalizeAgentOutput(
+        JSON.stringify({
+          ejclaw: {
+            visibility: 'public',
+            text: '이미지를 첨부했습니다.',
+            verdict: 'done',
+            attachments: [
+              {
+                path: '/tmp/compat.png',
+                name: 'compat.png',
+                mime: 'image/png',
+              },
+            ],
+          },
+        }),
+      ),
+    ).toEqual({
+      result: '이미지를 첨부했습니다.',
+      output: {
+        visibility: 'public',
+        text: '이미지를 첨부했습니다.',
+        verdict: 'done',
+        attachments: [
+          {
+            path: '/tmp/compat.png',
+            name: 'compat.png',
+            mime: 'image/png',
+          },
+        ],
+      },
+      attachmentSource: 'legacy-ejclaw-json',
+    });
+  });
+});