@midscene/web 0.16.6 → 0.16.7

package/dist/lib/puppeteer.js

@@ -145,6 +145,46 @@ var ScriptPlayer = class {
         );
         const queryResult = await agent.aiQuery(prompt);
         this.setResult(queryTask.name, queryResult);
+      } else if (flowItem.aiNumber) {
+        const numberTask = flowItem;
+        const prompt = numberTask.aiNumber;
+        (0, import_utils.assert)(prompt, "missing prompt for number");
+        (0, import_utils.assert)(
+          typeof prompt === "string",
+          "prompt for number must be a string"
+        );
+        const numberResult = await agent.aiNumber(prompt);
+        this.setResult(numberTask.name, numberResult);
+      } else if (flowItem.aiString) {
+        const stringTask = flowItem;
+        const prompt = stringTask.aiString;
+        (0, import_utils.assert)(prompt, "missing prompt for string");
+        (0, import_utils.assert)(
+          typeof prompt === "string",
+          "prompt for string must be a string"
+        );
+        const stringResult = await agent.aiString(prompt);
+        this.setResult(stringTask.name, stringResult);
+      } else if (flowItem.aiBoolean) {
+        const booleanTask = flowItem;
+        const prompt = booleanTask.aiBoolean;
+        (0, import_utils.assert)(prompt, "missing prompt for boolean");
+        (0, import_utils.assert)(
+          typeof prompt === "string",
+          "prompt for boolean must be a string"
+        );
+        const booleanResult = await agent.aiBoolean(prompt);
+        this.setResult(booleanTask.name, booleanResult);
+      } else if (flowItem.aiLocate) {
+        const locateTask = flowItem;
+        const prompt = locateTask.aiLocate;
+        (0, import_utils.assert)(prompt, "missing prompt for aiLocate");
+        (0, import_utils.assert)(
+          typeof prompt === "string",
+          "prompt for aiLocate must be a string"
+        );
+        const locateResult = await agent.aiLocate(prompt);
+        this.setResult(locateTask.name, locateResult);
       } else if (flowItem.aiWaitFor) {
         const waitForTask = flowItem;
         const prompt = waitForTask.aiWaitFor;