testdriverai 5.1.1 → 5.1.2

package/agent.js

@@ -31,7 +31,7 @@ const sanitizeFilename = require("sanitize-filename");
 const macScreenPerms = require("mac-screen-capture-permissions");
 
 // local modules
-const wss = require("./lib/websockets.js");
+const websocketserver = require("./lib/websockets.js");
 const speak = require("./lib/speak.js");
 const analytics = require("./lib/analytics.js");
 const log = require("./lib/logger.js");
@@ -63,6 +63,7 @@ let checkCount = 0;
 let checkLimit = 7;
 let lastScreenshot = null;
 let rl;
+let wss; 
 
 // list of prompts that the user has given us
 let tasks = [];
@@ -837,7 +838,7 @@ const firstPrompt = async () => {
 
   rl.on("line", handleInput);
 
-  wss.addEventListener("input", async (message) => {
+  config.TD_VM && wss.addEventListener("input", async (message) => {
     handleInput(message.data);
   });
 
@@ -1081,7 +1082,7 @@ ${yaml.dump(step)}
 };
 
 const promptUser = () => {
-  wss.sendToClients("done");
+  config.TD_VM && wss.sendToClients("done");
   emitter.emit(events.interactive, true);
   rl.prompt(true);
 };
@@ -1144,6 +1145,9 @@ const embed = async (file, depth) => {
 
 const buildEnv = async () => {
   let win = await system.activeWin();
+  if (config.TD_VM) {
+    wss = websocketserver.create();
+  }
   setTerminalApp(win);
   await ensureMacScreenPerms();
   await makeSandbox();

package/lib/logger.js

@@ -10,6 +10,7 @@ const shouldLog =
 // responsible for rendering ai markdown output
 const { marked } = require("marked");
 const { markedTerminal } = require("marked-terminal");
+const { config } = require("dotenv");
 
 const { printf } = winston.format;
 
@@ -84,7 +85,9 @@ const createMarkdownStreamLogger = () => {
         return;
       }        
       
-      websockets.sendToClients("output", chunk);
+      if (config.TD_VM) {
+        websockets.sendToClients("output", chunk);
+      }
 
       const previousConsoleOutput = markedParsePartial(buffer, 0, -1);
 

package/lib/speak.js

@@ -6,7 +6,6 @@ const websockets = require("./websockets");
 module.exports = (message) => {
   if (config["TD_SPEAK"]) {
     say.stop();
-    // websockets.sendToClients("output", message);  
     if (process.platform === "darwin") {
       say.speak(message, "Fred", 1.2);
     } else {

package/lib/websockets.js

@@ -1,7 +1,10 @@
 const WebSocket = require('ws');
+const config = require('./config');
 
+let instance;
 class WebSocketServerSingleton {
   constructor() {
+
     if (!WebSocketServerSingleton.instance) {
       this.wss = new WebSocket.Server({ port: 8080 }, () => {
       });
@@ -72,7 +75,11 @@ class WebSocketServerSingleton {
   }
 }
 
-const instance = new WebSocketServerSingleton();
-Object.freeze(instance);
 
-module.exports = instance;
+module.exports = {
+  create: () => {
+    instance = new WebSocketServerSingleton();
+    Object.freeze(instance);  
+    return instance;
+  }
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "5.1.1",
+  "version": "5.1.2",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "index.js",
   "bin": {