@mablhq/mabl-cli 2.79.1 → 2.79.7

package/api/mablApiClient.js

@@ -1257,25 +1257,25 @@ class MablApiClient extends basicApiClient_1.BasicApiClient {
             throw toApiError(`Failed to generate element description`, error);
         }
     }
-    async findElementImage(screenshot, findPrompt, workspaceId, context) {
+    async findElementImage(screenshot, findPrompt, workspaceId, includeElementImage, context) {
         try {
-            const { bounding_box: boundingBox, found } = await this.makePostRequest(`${this.baseApiAgentUrl}/agent/find/findElementImage?workspaceId=${workspaceId}`, {
+            const response = await this.makePostRequest(`${this.baseApiAgentUrl}/agent/find/findElementImage?workspaceId=${workspaceId}`, {
                 screenshot,
                 description: findPrompt,
                 context,
+                include_element_image: includeElementImage,
             });
-            if (!found) {
+            if (!response.found) {
                 return undefined;
             }
-            if (boundingBox?.x === undefined ||
-                boundingBox.y === undefined ||
-                boundingBox.width === undefined ||
-                boundingBox.height === undefined) {
-                console.debug(`Incomplete bounding box for ${findPrompt}:`, boundingBox);
+            if (response.bounding_box?.x === undefined ||
+                response.bounding_box?.y === undefined ||
+                response.bounding_box?.width === undefined ||
+                response.bounding_box?.height === undefined) {
+                console.debug(`Incomplete bounding box for ${findPrompt}:`, response.bounding_box);
                 return undefined;
             }
-            const { x, y, width, height } = boundingBox;
-            return { x, y, width, height };
+            return response;
         }
         catch (error) {
             throw toApiError(`Failed to find element image ${findPrompt}`, error);