starred/servers-modelcontextprotocol
1
0
Fork 0
mirror of https://github.com/modelcontextprotocol/servers.git synced 2026-04-21 13:25:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

add optional encoded param to puppeteer_screenshot for base64 output

Browse Source
This commit is contained in:
Jun
2025-05-06 02:46:41 -05:00
parent 249024599f
commit 51593d9a9a
2 changed files with 7 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
src/puppeteer/index.ts
View File

@@ -242,6 +242,7 @@ async function handleToolCall(name: string, args: any): Promise<CallToolResult>
case "puppeteer_screenshot": {
const width = args.width ?? 800;
const height = args.height ?? 600;
const encoded = args.encoded ?? false;
await page.setViewport({ width, height });
const screenshot = await (args.selector ?
@@ -269,52 +270,14 @@ async function handleToolCall(name: string, args: any): Promise<CallToolResult>
type: "text",
text: `Screenshot '${args.name}' taken at ${width}x${height}`,
} as TextContent,
{
encoded ? ({
type: "text",
text: `data:image/png;base64,${screenshot}`,
} as TextContent) : ({
type: "image",
data: screenshot,
mimeType: "image/png",
} as ImageContent,
],
isError: false,
};
}
case "puppeteer_screenshot_encoded": {
const width = args.width ?? 800;
const height = args.height ?? 600;
await page.setViewport({ width, height });
const screenshot = await (args.selector
? (await page.$(args.selector))?.screenshot({ encoding: "base64" })
: page.screenshot({ encoding: "base64", fullPage: false }));
if (!screenshot) {
return {
content: [
{
type: "text",
text: args.selector ? `Element not found: ${args.selector}` : "Screenshot failed",
},
],
isError: true,
};
}
screenshots.set(args.name, screenshot as string);
server.notification({
method: "notifications/resources/list_changed",
});
return {
content: [
{
type: "text",
text: `Screenshot '${args.name}' taken at ${width}x${height}`,
} as TextContent,
{
type: "text",
text: `data:image/png;base64,${screenshot}`,
} as TextContent,
} as ImageContent),
],
isError: false,
};