testdriverai 7.8.0-test.52 → 7.8.0-test.54

package/agent/index.js

@@ -70,6 +70,7 @@ class TestDriverAgent extends EventEmitter2 {
     this.sandboxId = flags["sandbox-id"] || null;
     this.sandboxAmi = flags["sandbox-ami"] || null;
     this.sandboxInstance = flags["sandbox-instance"] || null;
+    this.e2bTemplateId = flags["e2b-template-id"] || null;
     this.sandboxOs = flags.os || "linux";
     this.ip = flags.ip || null;
     this.workingDir = flags.workingDir || process.cwd();
@@ -2188,6 +2189,9 @@ Please check your network connection, TD_API_KEY, or the service status.`,
     if (this.sandboxInstance) {
       sandboxConfig.instanceType = this.sandboxInstance;
     }
+    if (this.e2bTemplateId) {
+      sandboxConfig.e2bTemplateId = this.e2bTemplateId;
+    }
     // Add keepAlive TTL if specified
     if (this.keepAlive !== undefined && this.keepAlive !== null) {
       sandboxConfig.keepAlive = this.keepAlive;

package/agent/lib/logger.js

@@ -7,6 +7,7 @@
  */
 
 const useStderr = process.env.TD_STDIO === 'stderr';
+const isDebug = process.env.DEBUG === 'true';
 
 /**
  * Log a message - uses stdout by default, stderr if TD_STDIO=stderr
@@ -40,6 +41,19 @@ function warn(...args) {
   }
 }
 
+/**
+ * Log a debug message - only outputs when DEBUG=true
+ * @param {...any} args - Arguments to log
+ */
+function debug(...args) {
+  if (!isDebug) return;
+  if (useStderr) {
+    console.error(...args);
+  } else {
+    console.log(...args);
+  }
+}
+
 /**
  * Check if logger is configured to use stderr
  * @returns {boolean}
@@ -50,6 +64,7 @@ function isStderrMode() {
 
 module.exports = {
   log,
+  debug,
   error,
   warn,
   isStderrMode,

package/agent/lib/sandbox.js

@@ -85,7 +85,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         suspendedRetryTimeout: 15000,     // retry from suspended every 15s (default 30s)
       });
 
-      logger.log(`[realtime] Connecting as sdk-${this._sandboxId}...`);
+      logger.debug(`[realtime] Connecting as sdk-${this._sandboxId}...`);
 
       await new Promise(function (resolve, reject) {
         self._ably.connection.on("connected", resolve);
@@ -99,7 +99,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
 
       this._sessionChannel = this._ably.channels.get(channelName);
 
-      logger.log(`[realtime] Channel initialized: ${channelName}`);
+      logger.debug(`[realtime] Channel initialized: ${channelName}`);
 
       // Enter presence on the session channel so the API can count connected SDK clients
       try {
@@ -107,7 +107,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
           sandboxId: this._sandboxId,
           connectedAt: Date.now(),
         });
-        logger.log(`[realtime] Entered presence on session channel (sandbox=${this._sandboxId})`);
+        logger.debug(`[realtime] Entered presence on session channel (sandbox=${this._sandboxId})`);
       } catch (e) {
         // Non-fatal — presence is used for concurrency counting, not critical path
         logger.warn("Failed to enter presence on session channel: " + (e.message || e));
@@ -118,7 +118,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         var message = msg.data;
         if (!message) return;
 
-        logger.log(`[realtime] Received response: type=${message.type || 'unknown'} (requestId=${message.requestId || 'none'})`);
+        logger.debug(`[realtime] Received response: type=${message.type || 'unknown'} (requestId=${message.requestId || 'none'})`);
 
         if (message.type === "sandbox.progress") {
           emitter.emit(events.sandbox.progress, {
@@ -212,7 +212,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
             var resolveAge = resolveEntry.startTime
               ? ((Date.now() - resolveEntry.startTime) / 1000).toFixed(1) + 's'
               : '?';
-            logger.log(
+            logger.debug(
               '[realtime] Promise RESOLVED: requestId=' + message.requestId +
               ' | type=' + (resolveEntry.message ? resolveEntry.message.type : 'unknown') +
               ' | age=' + resolveAge
@@ -241,7 +241,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
       this._onFileMsg = function (msg) {
         var message = msg.data;
         if (!message) return;
-        logger.log(`[realtime] Received file: type=${message.type || 'unknown'} (requestId=${message.requestId || 'none'})`);
+        logger.debug(`[realtime] Received file: type=${message.type || 'unknown'} (requestId=${message.requestId || 'none'})`);
         if (message.requestId && self.ps[message.requestId]) {
           emitter.emit(events.sandbox.received);
           self.ps[message.requestId].resolve(message);
@@ -260,7 +260,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         const chState = this._sessionChannel ? this._sessionChannel.state : 'null';
         const pendingIds = Object.keys(this.ps);
         const pending = pendingIds.length;
-        logger.log(`[realtime][stats] connection=${connState} | sandbox=${this._sandboxId} | pending=${pending} | channel=${chState}`);
+        logger.debug(`[realtime][stats] connection=${connState} | sandbox=${this._sandboxId} | pending=${pending} | channel=${chState}`);
         if (pending > 0) {
           const now = Date.now();
           for (const rid of pendingIds) {
@@ -268,20 +268,20 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
             if (!entry) continue;
             const type = entry.message ? entry.message.type : 'unknown';
             const ageSec = ((now - (entry.startTime || now)) / 1000).toFixed(1);
-            logger.log(`[realtime][stats]   pending: requestId=${rid} | type=${type} | age=${ageSec}s`);
+            logger.debug(`[realtime][stats]   pending: requestId=${rid} | type=${type} | age=${ageSec}s`);
           }
         }
       }, 10000);
       if (this._statsInterval.unref) this._statsInterval.unref();
 
       this._ably.connection.on("disconnected", function () {
-        logger.log("[realtime] Connection: disconnected - will auto-reconnect");
+        logger.debug("[realtime] Connection: disconnected - will auto-reconnect");
         self._disconnectedAt = Date.now();
       });
 
       this._ably.connection.on("connected", function () {
         // Log reconnection so the user knows the blip was recovered
-        logger.log("[realtime] Connection: reconnected");
+        logger.debug("[realtime] Connection: reconnected");
         // Extend any pending command timeouts by the disconnection duration so
         // commands whose timer was counting down while the connection was down
         // don't get incorrectly timed out.
@@ -290,7 +290,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
           self._disconnectedAt = null;
           var pendingIds = Object.keys(self.ps);
           if (pendingIds.length > 0) {
-            logger.log(
+            logger.debug(
               '[realtime] Extending ' + pendingIds.length + ' pending timeout(s) by ' +
               disconnectionDurationMs + 'ms after disconnection'
             );
@@ -353,7 +353,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         var entry = subs[i];
         if (!entry.sub) continue;
         try {
-          logger.log('[realtime] Discontinuity recovery: fetching historyBeforeSubscribe for ' + entry.name + '...');
+          logger.debug('[realtime] Discontinuity recovery: fetching historyBeforeSubscribe for ' + entry.name + '...');
           var page = await entry.sub.historyBeforeSubscribe({ limit: 100 });
           var recovered = 0;
           while (page) {
@@ -363,7 +363,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
               recovered++;
               try {
                 if (entry.handler) {
-                  logger.log('[realtime] Replaying recovered ' + entry.name + ' message (requestId=' + (page.items[j].data && page.items[j].data.requestId || 'none') + ')');
+                  logger.debug('[realtime] Replaying recovered ' + entry.name + ' message (requestId=' + (page.items[j].data && page.items[j].data.requestId || 'none') + ')');
                   entry.handler(page.items[j]);
                 }
               } catch (replayErr) {
@@ -373,7 +373,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
             page = page.hasNext() ? await page.next() : null;
           }
           totalRecovered += recovered;
-          logger.log('[realtime] Discontinuity recovery: replayed ' + recovered + ' ' + entry.name + ' message(s) from gap');
+          logger.debug('[realtime] Discontinuity recovery: replayed ' + recovered + ' ' + entry.name + ' message(s) from gap');
         } catch (err) {
           logger.error('[realtime] Discontinuity recovery failed for ' + entry.name + ': ' + (err.message || err));
         }
@@ -381,7 +381,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
       if (totalRecovered > 0) {
         logger.warn('[realtime] Recovered and replayed ' + totalRecovered + ' message(s) that were missed during connection interruption');
       } else {
-        logger.log('[realtime] Discontinuity recovery: no missed messages found');
+        logger.debug('[realtime] Discontinuity recovery: no missed messages found');
       }
     }
 
@@ -489,6 +489,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         body.ci = message.ci;
         if (message.ami) body.ami = message.ami;
         if (message.instanceType) body.instanceType = message.instanceType;
+        if (message.e2bTemplateId) body.e2bTemplateId = message.e2bTemplateId;
         if (message.keepAlive !== undefined) body.keepAlive = message.keepAlive;
       }
 
@@ -540,6 +541,15 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         const noVncPort = reply.runner && reply.runner.noVncPort;
         const runnerVncUrl = reply.runner && reply.runner.vncUrl;
 
+        // Log image version info (AMI for Windows, E2B template for Linux)
+        if (reply.imageVersion) {
+          if (isE2B) {
+            logger.log('E2B image version: v' + reply.imageVersion + (reply.e2bTemplateId ? ' (template: ' + reply.e2bTemplateId + ')' : ''));
+          } else {
+            logger.log('AMI image version: v' + reply.imageVersion + (reply.amiId ? ' (ami: ' + reply.amiId + ')' : ''));
+          }
+        }
+
         if (!isE2B) {
           logger.log(`Runner claimed — ip=${runnerIp || 'none'}, os=${reply.runner?.os || 'unknown'}, noVncPort=${noVncPort || 'not reported'}, vncUrl=${runnerVncUrl || 'not reported'}`);
         }
@@ -570,6 +580,13 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
               if (data && data.ip && reply.runner) reply.runner.ip = data.ip;
               if (data && data.runnerVersion && reply.runner) reply.runner.version = data.runnerVersion;
               logger.log('Runner agent ready (os=' + ((data && data.os) || 'unknown') + ', runner v' + ((data && data.runnerVersion) || 'unknown') + ')');
+              // Show upgrade info: if the runner's npm version differs from the baked image version,
+              // the runner was upgraded during provisioning.
+              var runnerVer = data && data.runnerVersion;
+              var imageVer = reply.imageVersion;
+              if (runnerVer && imageVer && runnerVer !== imageVer) {
+                logger.log('Runner upgraded during provisioning: v' + imageVer + ' \u2192 v' + runnerVer);
+              }
               if (data && data.update) {
                 var u = data.update;
                 if (u.status === 'up-to-date') {
@@ -1002,7 +1019,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
       }
 
       return channel.publish(eventName, message).then(function () {
-        logger.log(`[realtime] Published: channel=${channel.name.split(':').pop()}, event=${eventName}, type=${message.type || 'unknown'} (requestId=${message.requestId || 'none'})`);
+        logger.debug(`[realtime] Published: channel=${channel.name.split(':').pop()}, event=${eventName}, type=${message.type || 'unknown'} (requestId=${message.requestId || 'none'})`);
       });
     }
 

package/examples/scroll-keyboard.test.mjs

@@ -16,7 +16,7 @@ describe("Scroll Keyboard Test", () => {
     // Navigate to https://www.webhamster.com/
     await testdriver.focusApplication("Google Chrome");
     const urlBar = await testdriver.find(
-      "testdriver-sandbox.vercel.app/login, the URL in the omnibox showing the current page", {zoom: true}
+      "the URL in the omnibox", {zoom: true}
     );
     await urlBar.click();
     await testdriver.pressKeys(["ctrl", "a"]);

package/lib/vitest/hooks.mjs

@@ -403,6 +403,11 @@ export function TestDriver(context, options = {}) {
     config.apiRoot = process.env.TD_API_ROOT;
   }
 
+  // Use TD_E2B_TEMPLATE_ID from environment if not provided in config
+  if (!config.e2bTemplateId && process.env.TD_E2B_TEMPLATE_ID) {
+    config.e2bTemplateId = process.env.TD_E2B_TEMPLATE_ID;
+  }
+
   const testdriver = new TestDriverSDK(apiKey, config);
   testdriver.__vitestContext = context.task;
   testdriver._debugOnFailure = mergedOptions.debugOnFailure || false;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "7.8.0-test.52",
+  "version": "7.8.0-test.54",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "sdk.js",
   "types": "sdk.d.ts",

package/sdk.d.ts

@@ -273,6 +273,8 @@ export interface TestDriverOptions {
   sandboxAmi?: string;
   /** EC2 instance type for sandbox (e.g., 'i3.metal') */
   sandboxInstance?: string;
+  /** E2B template ID to use when creating the sandbox (e.g., 'my-template-id') */
+  e2bTemplateId?: string;
   /** Cache key for element finding operations. If provided, enables caching tied to this key */
   cacheKey?: string;
   /** Reconnect to the last used sandbox instead of creating a new one. When true, provision methods (chrome, vscode, installer, etc.) will be skipped since the application is already running. Throws error if no previous sandbox exists. */
@@ -327,6 +329,8 @@ export interface ConnectOptions {
   sandboxAmi?: string;
   /** EC2 instance type for sandbox (e.g., 'i3.metal') */
   sandboxInstance?: string;
+  /** E2B template ID to use when creating the sandbox (e.g., 'my-template-id') */
+  e2bTemplateId?: string;
   /** Operating system for the sandbox (default: 'linux') */
   os?: "windows" | "linux";
   /**

package/sdk.js

@@ -568,7 +568,7 @@ class Element {
         cacheKey: cacheKey,
         os: this.sdk.os,
         resolution: this.sdk.resolution,
-        zoom: zoom,
+        zoom: zoom === true ? 1 : zoom === false ? 0 : zoom,
         confidence: minConfidence,
         type: elementType,
         ai: {
@@ -1498,6 +1498,7 @@ class TestDriverSDK {
     // Store sandbox configuration options
     this.sandboxAmi = options.sandboxAmi || null;
     this.sandboxInstance = options.sandboxInstance || null;
+    this.e2bTemplateId = options.e2bTemplateId || null;
 
     // Store reconnect preference from options
     this.reconnect =
@@ -2716,6 +2717,7 @@ CAPTCHA_SOLVER_EOF`,
    * @param {string} options.ip - Direct IP address to connect to
    * @param {string} options.sandboxAmi - AMI to use for the sandbox
    * @param {string} options.sandboxInstance - Instance type for the sandbox
+   * @param {string} options.e2bTemplateId - E2B template ID to use when creating the sandbox
    * @param {string} options.os - Operating system for the sandbox (windows or linux)
    * @param {boolean} options.reuseConnection - Reuse recent connection if available (default: true)
    * @returns {Promise<Object>} Sandbox instance details
@@ -2804,6 +2806,12 @@ CAPTCHA_SOLVER_EOF`,
     } else if (this.sandboxInstance) {
       this.agent.sandboxInstance = this.sandboxInstance;
     }
+    // Use e2bTemplateId from connectOptions if provided, otherwise fall back to constructor value
+    if (connectOptions.e2bTemplateId !== undefined) {
+      this.agent.e2bTemplateId = connectOptions.e2bTemplateId;
+    } else if (this.e2bTemplateId) {
+      this.agent.e2bTemplateId = this.e2bTemplateId;
+    }
     // Use os from connectOptions if provided, otherwise fall back to this.os
     if (connectOptions.os !== undefined) {
       this.agent.sandboxOs = connectOptions.os;