@debugg-ai/debugg-ai-mcp 2.6.0 → 2.6.1

package/dist/handlers/probePageHandler.js

@@ -24,6 +24,7 @@ import { extractLocalhostPort } from '../utils/urlParser.js';
 import { buildContext, findExistingTunnel, ensureTunnel, sanitizeResponseUrls, touchTunnelById, } from '../utils/tunnelContext.js';
 import { getCachedTemplateUuid, invalidateTemplateCache } from '../utils/handlerCaches.js';
 import { reaggregateByOriginPath, mapConsoleSlice } from '../utils/harSummarizer.js';
+import { fetchImageAsBase64, imageContentBlock } from '../utils/imageUtils.js';
 const logger = new Logger({ module: 'probePageHandler' });
 const TEMPLATE_KEYWORD = 'page probe';
 export async function probePageHandler(input, context, rawProgressCallback) {
@@ -256,12 +257,6 @@ export async function probePageHandler(input, context, rawProgressCallback) {
                 result.surferPageUuid = data.surferPageUuid;
             }
             results.push(result);
-            // Backend stores screenshots on the SurferPage row referenced by
-            // surfer_page_uuid; the inline screenshotB64 is no longer in capture
-            // output. v1: skip the per-result image content block; the screenshot
-            // is reachable via search_executions detail's surfer_page_uuid → SurferPage.
-            // (Future enhancement: fetch the SurferPage's presigned screenshot_url
-            // when input.captureScreenshots is true.)
         }
         const responsePayload = {
             executionId: executionUuid,
@@ -284,11 +279,35 @@ export async function probePageHandler(input, context, rawProgressCallback) {
             }
         }
         logger.toolComplete('probe_page', duration);
-        return {
-            content: [
-                { type: 'text', text: JSON.stringify(sanitizedPayload, null, 2) },
-            ],
-        };
+        const responseContent = [
+            { type: 'text', text: JSON.stringify(sanitizedPayload, null, 2) },
+        ];
+        // Embed screenshots when captureScreenshots is true. The backend may return
+        // screenshotB64 or a URL-keyed field on browser.capture outputData.
+        if (input.captureScreenshots) {
+            const SCREENSHOT_URL_KEYS = ['screenshotB64', 'screenshot', 'screenshotUrl', 'screenshotUri', 'finalScreenshot'];
+            for (const node of captureNodes) {
+                const data = node?.outputData ?? {};
+                if (typeof data.screenshotB64 === 'string' && data.screenshotB64) {
+                    responseContent.push(imageContentBlock(data.screenshotB64, 'image/png'));
+                }
+                else {
+                    let screenshotUrl = null;
+                    for (const key of SCREENSHOT_URL_KEYS) {
+                        if (key !== 'screenshotB64' && typeof data[key] === 'string' && data[key]) {
+                            screenshotUrl = data[key];
+                            break;
+                        }
+                    }
+                    if (screenshotUrl) {
+                        const img = await fetchImageAsBase64(screenshotUrl).catch(() => null);
+                        if (img)
+                            responseContent.push(imageContentBlock(img.data, img.mimeType));
+                    }
+                }
+            }
+        }
+        return { content: responseContent };
     }
     catch (error) {
         const duration = Date.now() - startTime;

package/dist/handlers/searchExecutionsHandler.js

@@ -13,6 +13,7 @@ import { handleExternalServiceError } from '../utils/errors.js';
 import { DebuggAIServerClient } from '../services/index.js';
 import { config } from '../config/index.js';
 import { toPaginationParams } from '../utils/pagination.js';
+import { fetchImageAsBase64, imageContentBlock } from '../utils/imageUtils.js';
 const logger = new Logger({ module: 'searchExecutionsHandler' });
 function notFound(uuid) {
     return {
@@ -38,7 +39,53 @@ export async function searchExecutionsHandler(input, _context) {
                     executions: [execution],
                 };
                 logger.toolComplete('search_executions', Date.now() - start);
-                return { content: [{ type: 'text', text: JSON.stringify(payload, null, 2) }] };
+                const content = [
+                    { type: 'text', text: JSON.stringify(payload, null, 2) },
+                ];
+                const SCREENSHOT_URL_KEYS = ['finalScreenshot', 'screenshot', 'screenshotUrl', 'screenshotUri'];
+                const GIF_KEYS = ['runGif', 'gifUrl', 'gif', 'videoUrl', 'recordingUrl'];
+                const nodes = execution.nodeExecutions ?? [];
+                const subworkflowNode = nodes.find((n) => n.nodeType === 'subworkflow.run');
+                let screenshotEmbedded = false;
+                let screenshotUrl = null;
+                let gifUrl = null;
+                const screenshotB64 = subworkflowNode?.outputData?.screenshotB64;
+                if (typeof screenshotB64 === 'string' && screenshotB64) {
+                    content.push(imageContentBlock(screenshotB64, 'image/png'));
+                    screenshotEmbedded = true;
+                }
+                for (const node of nodes) {
+                    const data = node.outputData ?? {};
+                    if (!screenshotEmbedded && !screenshotUrl) {
+                        for (const key of SCREENSHOT_URL_KEYS) {
+                            if (typeof data[key] === 'string' && data[key]) {
+                                screenshotUrl = data[key];
+                                break;
+                            }
+                        }
+                    }
+                    if (!gifUrl) {
+                        for (const key of GIF_KEYS) {
+                            if (typeof data[key] === 'string' && data[key]) {
+                                gifUrl = data[key];
+                                break;
+                            }
+                        }
+                    }
+                    if ((screenshotEmbedded || screenshotUrl) && gifUrl)
+                        break;
+                }
+                if (!screenshotEmbedded && screenshotUrl) {
+                    const img = await fetchImageAsBase64(screenshotUrl).catch(() => null);
+                    if (img)
+                        content.push(imageContentBlock(img.data, img.mimeType));
+                }
+                if (gifUrl) {
+                    const gif = await fetchImageAsBase64(gifUrl).catch(() => null);
+                    if (gif)
+                        content.push(imageContentBlock(gif.data, 'image/gif'));
+                }
+                return { content };
             }
             catch (err) {
                 if (err?.statusCode === 404 || err?.response?.status === 404)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@debugg-ai/debugg-ai-mcp",
-  "version": "2.6.0",
+  "version": "2.6.1",
   "description": "Zero-Config, Fully AI-Managed End-to-End Testing for all code gen platforms.",
   "type": "module",
   "bin": {