From dbdee8bc02bc281d79238f889aab7d3c79d898f4 Mon Sep 17 00:00:00 2001
From: mukunda katta <mukunda.vjcs6@gmail.com>
Date: Sat, 18 Apr 2026 00:53:00 -0700
Subject: [PATCH] fix: save WebP responses with the right extension

---
 tests/daemon/client.test.ts | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/tests/daemon/client.test.ts b/tests/daemon/client.test.ts
index dede15875..8c59289cb 100644
--- a/tests/daemon/client.test.ts
+++ b/tests/daemon/client.test.ts
@@ -107,5 +107,20 @@ describe('daemon client', () => {
       const response = await handleResponse(unsupportedContentResponse, 'md');
       assert.ok(response.includes('.png'));
     });
+
+    it('uses the webp extension for WebP images', async () => {
+      const webpContentResponse = {
+        content: [
+          {
+            type: 'image' as const,
+            data: 'base64data',
+            mimeType: 'image/webp',
+          },
+        ],
+        structuredContent: {},
+      };
+      const response = await handleResponse(webpContentResponse, 'md');
+      assert.ok(response.includes('.webp'));
+    });
   });
 });