diff --git a/src/tools/screenshot.ts b/src/tools/screenshot.ts
index 341e037..3e142a7 100644
--- a/src/tools/screenshot.ts
+++ b/src/tools/screenshot.ts
@@ -62,7 +62,7 @@ const screenshotSchema = z.object({
filename: z.string().optional().describe('File name to save the screenshot to. Defaults to `page-{timestamp}.{png|jpeg}` if not specified.'),
element: z.string().optional().describe('Human-readable element description used to obtain permission to screenshot the element. If not provided, the screenshot will be taken of viewport. If element is provided, ref must be provided too.'),
ref: z.string().optional().describe('Exact target element reference from the page snapshot. If not provided, the screenshot will be taken of viewport. If ref is provided, element must be provided too.'),
- fullPage: z.boolean().optional().describe('When true, takes a screenshot of the full scrollable page, instead of the currently visible viewport. Cannot be used with element screenshots.'),
+ fullPage: z.boolean().optional().describe('When true, takes a screenshot of the full scrollable page, instead of the currently visible viewport. Cannot be used with element screenshots. WARNING: Full page screenshots may exceed API size limits on long pages.'),
allowLargeImages: z.boolean().optional().describe('Allow images with dimensions exceeding 8000 pixels (API limit). Default false - will error if image is too large to prevent API failures.'),
}).refine(data => {
return !!data.element === !!data.ref;
@@ -160,10 +160,29 @@ const screenshot = defineTabTool({
}
response.addResult(resultMessage);
- response.addImage({
- contentType: fileType === 'png' ? 'image/png' : 'image/jpeg',
- data: buffer
- });
+
+ // Only add image to response if dimensions are safe or explicitly allowed
+ let addImageToResponse = true;
+ if (!params.allowLargeImages) {
+ try {
+ const { width, height } = getImageDimensions(buffer);
+ const maxDimension = 8000;
+ if (width > maxDimension || height > maxDimension) {
+ addImageToResponse = false;
+ }
+ } catch (dimensionError) {
+ // If we can't parse dimensions, continue and add the image
+ }
+ }
+
+ if (addImageToResponse) {
+ response.addImage({
+ contentType: fileType === 'png' ? 'image/png' : 'image/jpeg',
+ data: buffer
+ });
+ } else {
+ response.addResult(`\n\n๐ซ **Image not included in response**: Screenshot exceeds API size limits (8000px). Image saved to file only.`);
+ }
}
});
diff --git a/test-large-screenshot.js b/test-large-screenshot.js
new file mode 100755
index 0000000..73fd354
--- /dev/null
+++ b/test-large-screenshot.js
@@ -0,0 +1,64 @@
+#!/usr/bin/env node
+
+/**
+ * Test script to verify large screenshot handling
+ * Creates a very tall page and tests fullPage screenshot protection
+ */
+
+const { execSync } = require('child_process');
+const fs = require('fs');
+const path = require('path');
+
+async function testLargeScreenshot() {
+ console.log('๐งช Testing large screenshot protection...');
+
+ // Create a simple HTML page that will be very tall
+ const testHtml = `
+
+
+
+ Large Page Test
+
+
+
+
+ This is a very tall page (10000px height)
+ Should trigger large image protection
+
+
+
+ `;
+
+ const testFile = path.join(__dirname, 'test-large-page.html');
+ fs.writeFileSync(testFile, testHtml);
+
+ console.log(`๐ Created test file: ${testFile}`);
+ console.log('๐ง This test requires manual verification with an MCP client');
+ console.log('');
+ console.log('To test:');
+ console.log('1. Start MCP server: npm run build && node lib/index.js');
+ console.log(`2. Navigate to: file://${testFile}`);
+ console.log('3. Try: browser_take_screenshot {"fullPage": true}');
+ console.log('4. Verify: Image saved to file but NOT included in response');
+ console.log('5. Should see: "๐ซ **Image not included in response**" message');
+ console.log('');
+ console.log('Expected behavior:');
+ console.log('- Screenshot file should be created');
+ console.log('- No large image sent to API (prevents conversation issues)');
+ console.log('- Clear warning message displayed');
+
+ return testFile;
+}
+
+testLargeScreenshot().catch(console.error);
\ No newline at end of file