From 0cb99b04c08c470bb0564ebff002448ecc65b829 Mon Sep 17 00:00:00 2001
From: Mohammad Baghaei <mbaghaei42@yahoo.com>
Date: Tue, 11 Nov 2025 12:34:45 +0330
Subject: [PATCH] fix: enhance screenshot functionality to support multiple
 image formats

---
 packages/render-fidelity-tools/src/artifact-creator.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/render-fidelity-tools/src/artifact-creator.ts b/packages/render-fidelity-tools/src/artifact-creator.ts
index b11bebf510..2685aae39d 100644
--- a/packages/render-fidelity-tools/src/artifact-creator.ts
+++ b/packages/render-fidelity-tools/src/artifact-creator.ts
@@ -382,7 +382,9 @@ export class ArtifactCreator {
     }
 
     const screenshot =
-        await page.screenshot({path: outputPath as `${string}.png`, omitBackground: true});
+        outputPath && (outputPath.endsWith('.png') || outputPath.endsWith('.jpeg') || outputPath.endsWith('.webp'))
+            ? await page.screenshot({path: outputPath as `${string}.png` | `${string}.jpeg` | `${string}.webp`, omitBackground: true})
+            : await page.screenshot({omitBackground: true});
 
     page.close();