@mablhq/mabl-cli 2.30.0 → 2.31.2

package/api/featureSet.js

@@ -1,12 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.FeatureSet = exports.FindImplementationVersion = exports.ADVANCED_AUTO_HEALING_FEATURE_FLAG = exports.GENERATIVE_AI_FEATURE_FLAG = void 0;
+exports.FeatureSet = exports.FindImplementationVersion = exports.AI_ASSERTIONS_FEATURE_FLAG = exports.ADVANCED_AUTO_HEALING_FEATURE_FLAG = exports.GENERATIVE_AI_FEATURE_FLAG = void 0;
 const types_1 = require("../browserLauncher/types");
 const ACCESSIBILITY_CHECK_FEATURE_FLAG = 'accessibility_checks';
 const PERFORMANCE_TESTING_FEATURE_FLAG = 'performance_testing';
 const SMARTER_WAIT_FEATURE_FLAG = 'smarter_wait';
 exports.GENERATIVE_AI_FEATURE_FLAG = 'generative_ai';
 exports.ADVANCED_AUTO_HEALING_FEATURE_FLAG = 'advanced_auto_healing';
+exports.AI_ASSERTIONS_FEATURE_FLAG = 'ai_assertions';
 var FindImplementationVersion;
 (function (FindImplementationVersion) {
     FindImplementationVersion[FindImplementationVersion["V1"] = 0] = "V1";
@@ -30,6 +31,9 @@ class FeatureSet {
     hasPerformanceTestingFeatureEnabled() {
         return this.featureFlags.has(PERFORMANCE_TESTING_FEATURE_FLAG);
     }
+    hasAIAssertions() {
+        return this.featureFlags.has(exports.AI_ASSERTIONS_FEATURE_FLAG);
+    }
     allowGenerativeAi() {
         return this.featureFlags.has(exports.GENERATIVE_AI_FEATURE_FLAG);
     }

package/api/mablApiClient.js

@@ -569,6 +569,24 @@ class MablApiClient extends basicApiClient_1.BasicApiClient {
             throw toApiError(`Failed to update mailbox ${mailboxId} with error`, error);
         }
     }
+    async evaluateAiAssertion(workspaceId, testRunId, screenshot, userPrompt, metaPrompt) {
+        try {
+            const body = {
+                test_run_id: testRunId,
+                assertion_prompt: userPrompt,
+                screenshot: Buffer.from(screenshot).toString('base64'),
+            };
+            if (metaPrompt !== undefined) {
+                body.override = {
+                    meta_prompt: metaPrompt,
+                };
+            }
+            return await this.makePostRequestWithRetries(`${this.baseApiUrl}/analysis/${workspaceId}/assertion/evaluate`, body);
+        }
+        catch (error) {
+            throw toApiError('Failed to evaluate the AI prompt assertion', error);
+        }
+    }
     async createBranch(workspaceId, branchName) {
         try {
             const body = {

package/commands/tests/tests_cmds/import_cmds/import_playwright.js

@@ -39,11 +39,10 @@ const loggingProvider_1 = require("../../../../providers/logging/loggingProvider
 const chromiumBrowserEngine_1 = require("../../../../browserEngines/chromiumBrowserEngine");
 const RichPromise_1 = __importDefault(require("../../../../util/RichPromise"));
 const { spawn } = require('child_process');
-const execution_1 = require("../../../../execution");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const adm_zip_1 = __importDefault(require("adm-zip"));
-const playwrightToMablStepConverter_1 = require("../../../../execution/codeGenerators/playwrightTest/playwrightToMablStepConverter");
+const execution_1 = require("../../../../execution");
 const projectInfoTestFile = `
 import { test } from "@playwright/test";
 test("mablImportInfo", async ({baseURL, trace}) => {
@@ -151,7 +150,7 @@ async function importTests(args) {
         await runPlaywrightProject(args, tracesPath, cwd);
     }
     let importedTests = [];
-    const converter = new playwrightToMablStepConverter_1.PlaywrightToMablStepConverter();
+    const converter = new execution_1.PlaywrightToMablStepConverter();
     if (args[constants_1.CommandArgTraceFile]) {
         importedTests = processSingleTraceFile(args[constants_1.CommandArgTraceFile], playwrightProject, converter);
     }

package/commands/tests/tests_cmds/import_cmds/import_selenium.js

@@ -51,7 +51,7 @@ var PostImportActions;
     PostImportActions["View"] = "View the test description";
 })(PostImportActions || (PostImportActions = {}));
 exports.command = 'selenium';
-exports.describe = false;
+exports.describe = 'Import an existing Selenium-based test';
 exports.builder = (yargs) => {
     yargs
         .option(constants_1.CommandArgAuto, {