From 4e52dd8fb74c40726114e1dfacac5434737fca4a Mon Sep 17 00:00:00 2001
From: Paul Klein <paulleoklein@gmail.com>
Date: Sun, 10 Nov 2024 23:39:06 -0800
Subject: [PATCH] Bump Version to 1.1.2 (#178)

* Add init scripts to the context, not just the page

* 1.1.2
---
 lib/index.ts      | 57 +++++++++++++++++++++++++++++++++++++++--------
 package-lock.json |  4 ++--
 package.json      |  2 +-
 3 files changed, 51 insertions(+), 12 deletions(-)

diff --git a/lib/index.ts b/lib/index.ts
index eea3f954..78065247 100644
--- a/lib/index.ts
+++ b/lib/index.ts
@@ -293,8 +293,8 @@ export class Stagehand {
       llmProvider || new LLMProvider(this.logger, this.enableCaching);
     this.env = env;
     this.observations = {};
-    this.apiKey = apiKey || process.env.BROWSERBASE_API_KEY;
-    this.projectId = projectId || process.env.BROWSERBASE_PROJECT_ID;
+    this.apiKey = apiKey ?? process.env.BROWSERBASE_API_KEY;
+    this.projectId = projectId ?? process.env.BROWSERBASE_PROJECT_ID;
     this.verbose = verbose ?? 0;
     this.debugDom = debugDom ?? false;
     this.defaultModelName = "gpt-4o";
@@ -339,6 +339,9 @@ export class Stagehand {
     });
     this.context = context;
     this.page = context.pages()[0];
+    // Redundant but needed for users who are re-connecting to a previously-created session
+    await this.page.waitForLoadState("domcontentloaded");
+    await this._waitForSettledDom();
     this.defaultModelName = modelName;
     this.domSettleTimeoutMs = domSettleTimeoutMs ?? this.domSettleTimeoutMs;
 
@@ -358,16 +361,36 @@ export class Stagehand {
 
     // This can be greatly improved, but the tldr is we put our built web scripts in dist, which should always
     // be one level above our running directly across evals, example, and as a package
-    await this.page.addInitScript({
+    await this.context.addInitScript({
+      path: path.join(__dirname, "..", "dist", "dom", "build", "xpathUtils.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "xpathUtils.js"),
+        "utf8",
+      ),
+    });
+
+    await this.context.addInitScript({
       path: path.join(__dirname, "..", "dist", "dom", "build", "process.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "process.js"),
+        "utf8",
+      ),
     });
 
-    await this.page.addInitScript({
+    await this.context.addInitScript({
       path: path.join(__dirname, "..", "dist", "dom", "build", "utils.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "utils.js"),
+        "utf8",
+      ),
     });
 
-    await this.page.addInitScript({
+    await this.context.addInitScript({
       path: path.join(__dirname, "..", "dist", "dom", "build", "debug.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "debug.js"),
+        "utf8",
+      ),
     });
 
     return { debugUrl, sessionUrl };
@@ -395,20 +418,36 @@ export class Stagehand {
     }
 
     // Add initialization scripts
-    await this.page.addInitScript({
+    await this.context.addInitScript({
       path: path.join(__dirname, "..", "dist", "dom", "build", "xpathUtils.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "xpathUtils.js"),
+        "utf8",
+      ),
     });
 
-    await this.page.addInitScript({
+    await this.context.addInitScript({
       path: path.join(__dirname, "..", "dist", "dom", "build", "process.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "process.js"),
+        "utf8",
+      ),
     });
 
-    await this.page.addInitScript({
+    await this.context.addInitScript({
       path: path.join(__dirname, "..", "dist", "dom", "build", "utils.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "utils.js"),
+        "utf8",
+      ),
     });
 
-    await this.page.addInitScript({
+    await this.context.addInitScript({
       path: path.join(__dirname, "..", "dist", "dom", "build", "debug.js"),
+      content: fs.readFileSync(
+        path.join(__dirname, "..", "dist", "dom", "build", "debug.js"),
+        "utf8",
+      ),
     });
 
     return { context: this.context };
diff --git a/package-lock.json b/package-lock.json
index c4a6228b..31ce8042 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@browserbasehq/stagehand",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@browserbasehq/stagehand",
-      "version": "1.1.1",
+      "version": "1.1.2",
       "license": "MIT",
       "dependencies": {
         "@anthropic-ai/sdk": "^0.27.3",
diff --git a/package.json b/package.json
index 55a501a7..a692d91a 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@browserbasehq/stagehand",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "description": "An AI web browsing framework focused on simplicity and extensibility.",
   "main": "./dist/index.js",
   "module": "./dist/index.js",