fix(llm): cover media mime aliases in tool replay

snowzlmbot · snowzlmbot · commit 887e20f898b2 · 2026-06-29T07:52:27.000+08:00
diff --git a/src/llm/providers/tool-result-text.test.ts b/src/llm/providers/tool-result-text.test.ts
@@ -82,18 +82,21 @@ describe("extractToolResultText", () => {
 
 
 
-  it("treats snake-case mime_type aliases as binary structured payloads", () => {
+  it.each([
+    ["mime_type", { mime_type: "application/octet-stream" }],
+    ["mediaType", { mediaType: "audio/mpeg" }],
+    ["contentType", { contentType: "application/pdf" }],
+    ["content_type", { content_type: "video/mp4" }],
+  ])("treats %s aliases as binary structured payloads", (_label, alias) => {
     expect(
       extractToolResultText([
         {
           type: "resource",
-          mime_type: "application/octet-stream",
+          ...alias,
           data: "binary-payload-bytes",
         },
       ]),
-    ).toBe(
-      '{"type":"resource","mime_type":"application/octet-stream","data":"[binary omitted: 20 chars]"}',
-    );
+    ).toBe(JSON.stringify({ type: "resource", ...alias, data: "[binary omitted: 20 chars]" }));
   });
 
   it("handles circular structured payloads without throwing", () => {
diff --git a/src/logging/tool-result-serialization.test.ts b/src/logging/tool-result-serialization.test.ts
@@ -0,0 +1,22 @@
+import { describe, expect, it } from "vitest";
+
+import { serializeStructuredToolResultBlock } from "./tool-result-serialization.js";
+
+describe("serializeStructuredToolResultBlock", () => {
+  it.each([
+    ["media_type", { media_type: "image/png" }],
+    ["mime_type", { mime_type: "application/octet-stream" }],
+    ["mimeType", { mimeType: "application/pdf" }],
+    ["mediaType", { mediaType: "audio/mpeg" }],
+    ["contentType", { contentType: "application/pdf" }],
+    ["content_type", { content_type: "video/mp4" }],
+  ])("omits binary data for %s media aliases", (_label, alias) => {
+    expect(
+      serializeStructuredToolResultBlock({
+        type: "resource",
+        ...alias,
+        data: "binary-payload-bytes",
+      }),
+    ).toBe(JSON.stringify({ type: "resource", ...alias, data: "[binary omitted: 20 chars]" }));
+  });
+});
diff --git a/src/logging/tool-result-serialization.ts b/src/logging/tool-result-serialization.ts
@@ -146,7 +146,14 @@ function carriesBinaryData(record: Record<string, unknown>): boolean {
   if (type === "audio" || type === "image" || type === "base64") {
     return true;
   }
-  const mediaType = readLowercaseString(record.media_type ?? record.mime_type ?? record.mimeType);
+  const mediaType = readLowercaseString(
+    record.media_type ??
+      record.mime_type ??
+      record.mimeType ??
+      record.mediaType ??
+      record.contentType ??
+      record.content_type,
+  );
   return (
     mediaType?.startsWith("image/") === true ||
     mediaType?.startsWith("audio/") === true ||
