@browserbasehq/stagehand 1.8.0 → 1.8.1-alpha-5c6f14bade201e08cb86d2e14e246cb65707f7ee

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { Page as Page$1, BrowserContext, Browser } from '@playwright/test';
 import { z, ZodType } from 'zod';
+import { Page as Page$1, BrowserContext as BrowserContext$1, Browser } from '@playwright/test';
 import Browserbase from '@browserbasehq/sdk';
 import { ClientOptions as ClientOptions$2 } from '@anthropic-ai/sdk';
 import { ClientOptions as ClientOptions$1 } from 'openai';
@@ -138,7 +138,7 @@ interface ConstructorParams {
     debugDom?: boolean;
     llmProvider?: LLMProvider;
     headless?: boolean;
-    logger?: (message: LogLine) => void;
+    logger?: (message: LogLine) => void | Promise<void>;
     domSettleTimeoutMs?: number;
     browserbaseSessionCreateParams?: Browserbase.Sessions.SessionCreateParams;
     enableCaching?: boolean;
@@ -167,7 +167,7 @@ interface InitFromPageOptions {
     modelClientOptions?: ClientOptions;
 }
 interface InitFromPageResult {
-    context: BrowserContext;
+    context: BrowserContext$1;
 }
 interface ActOptions {
     action: string;
@@ -208,11 +208,12 @@ interface Page extends Page$1 {
     extract: <T extends z.AnyZodObject>(options: ExtractOptions<T>) => Promise<ExtractResult<T>>;
     observe: (options?: ObserveOptions) => Promise<ObserveResult[]>;
 }
+type BrowserContext = BrowserContext$1;
 
 interface BrowserResult {
     env: "LOCAL" | "BROWSERBASE";
     browser?: Browser;
-    context: BrowserContext;
+    context: BrowserContext$1;
     debugUrl?: string;
     sessionUrl?: string;
     contextPath?: string;
@@ -242,7 +243,6 @@ declare class Stagehand {
     verbose: 0 | 1 | 2;
     llmProvider: LLMProvider;
     enableCaching: boolean;
-    private internalLogger;
     private apiKey;
     private projectId;
     private externalLogger?;
@@ -252,8 +252,6 @@ declare class Stagehand {
     };
     private contextPath?;
     private llmClient;
-    private extractHandler?;
-    private observeHandler?;
     constructor({ env, apiKey, projectId, verbose, debugDom, llmProvider, headless, logger, browserbaseSessionCreateParams, domSettleTimeoutMs, enableCaching, browserbaseSessionID, modelName, modelClientOptions, }?: ConstructorParams);
     get logger(): (logLine: LogLine) => void;
     get page(): Page;
@@ -278,4 +276,4 @@ declare class Stagehand {
     close(): Promise<void>;
 }
 
-export { type ActOptions, type ActResult, type AnthropicJsonSchemaObject, type AnthropicTransformedResponse, type AvailableModel, AvailableModelSchema, type BrowserResult, type ClientOptions, type ConstructorParams, type ExtractOptions, type ExtractResult, type GotoOptions, type InitFromPageOptions, type InitFromPageResult, type InitOptions, type InitResult, type LogLine, type ModelProvider, type ObserveOptions, type ObserveResult, PlaywrightCommandException, PlaywrightCommandMethodNotSupportedException, Stagehand, type ToolCall };
+export { type ActOptions, type ActResult, type AnthropicJsonSchemaObject, type AnthropicTransformedResponse, type AvailableModel, AvailableModelSchema, type BrowserContext, type BrowserResult, type ClientOptions, type ConstructorParams, type ExtractOptions, type ExtractResult, type GotoOptions, type InitFromPageOptions, type InitFromPageResult, type InitOptions, type InitResult, type LogLine, type ModelProvider, type ObserveOptions, type ObserveResult, type Page, PlaywrightCommandException, PlaywrightCommandMethodNotSupportedException, Stagehand, type ToolCall };

package/dist/index.js

@@ -4780,6 +4780,9 @@ function applyStealthScripts(context) {
     });
   });
 }
+var defaultLogger = (logLine) => __async(void 0, null, function* () {
+  console.log(logLineToString(logLine));
+});
 var Stagehand = class {
   constructor({
     env,
@@ -4801,8 +4804,7 @@ var Stagehand = class {
   }) {
     this.pending_logs_to_send_to_browserbase = [];
     this.is_processing_browserbase_logs = false;
-    this.externalLogger = logger;
-    this.internalLogger = this.log.bind(this);
+    this.externalLogger = logger || defaultLogger;
     this.enableCaching = enableCaching != null ? enableCaching : process.env.ENABLE_CACHING && process.env.ENABLE_CACHING === "true";
     this.llmProvider = llmProvider || new LLMProvider(this.logger, this.enableCaching);
     this.intEnv = env;
@@ -4821,10 +4823,7 @@ var Stagehand = class {
   }
   get logger() {
     return (logLine) => {
-      this.internalLogger(logLine);
-      if (this.externalLogger) {
-        this.externalLogger(logLine);
-      }
+      this.log(logLine);
     };
   }
   get page() {
@@ -4842,6 +4841,11 @@ var Stagehand = class {
     return "LOCAL";
   }
   get context() {
+    if (!this.stagehandContext) {
+      throw new Error(
+        "Stagehand not initialized. Make sure to await stagehand.init() first."
+      );
+    }
     return this.stagehandContext.context;
   }
   init(initOptions) {
@@ -4928,9 +4932,6 @@ var Stagehand = class {
     logObj.level = logObj.level || 1;
     if (this.externalLogger) {
       this.externalLogger(logObj);
-    } else {
-      const logMessage = logLineToString(logObj);
-      console.log(logMessage);
     }
     this.pending_logs_to_send_to_browserbase.push(__spreadProps(__spreadValues({}, logObj), {
       id: (0, import_crypto2.randomUUID)()

package/dist/lib/index.d.ts

@@ -1,7 +1,6 @@
-import { type BrowserContext } from "@playwright/test";
 import { z } from "zod";
 import { LogLine } from "../types/log";
-import { Page } from "../types/page";
+import { Page, BrowserContext } from "../types/page";
 import { ActOptions, ActResult, ConstructorParams, ExtractOptions, ExtractResult, InitFromPageOptions, InitFromPageResult, InitOptions, InitResult, ObserveOptions, ObserveResult } from "../types/stagehand";
 import { LLMProvider } from "./llm/LLMProvider";
 export declare class Stagehand {
@@ -15,7 +14,6 @@ export declare class Stagehand {
     verbose: 0 | 1 | 2;
     llmProvider: LLMProvider;
     enableCaching: boolean;
-    private internalLogger;
     private apiKey;
     private projectId;
     private externalLogger?;
@@ -25,8 +23,6 @@ export declare class Stagehand {
     };
     private contextPath?;
     private llmClient;
-    private extractHandler?;
-    private observeHandler?;
     constructor({ env, apiKey, projectId, verbose, debugDom, llmProvider, headless, logger, browserbaseSessionCreateParams, domSettleTimeoutMs, enableCaching, browserbaseSessionID, modelName, modelClientOptions, }?: ConstructorParams);
     get logger(): (logLine: LogLine) => void;
     get page(): Page;
@@ -55,3 +51,4 @@ export * from "../types/log";
 export * from "../types/model";
 export * from "../types/playwright";
 export * from "../types/stagehand";
+export * from "../types/page";

package/dist/types/page.d.ts

@@ -1,4 +1,5 @@
 import type { Page as PlaywrightPage } from "@playwright/test";
+import type { BrowserContext as PlaywrightContext } from "@playwright/test";
 import type { ActResult } from "./act";
 import type { ActOptions, ExtractOptions, ExtractResult, ObserveOptions, ObserveResult } from "./stagehand";
 import type { z } from "zod";
@@ -7,3 +8,4 @@ export interface Page extends PlaywrightPage {
     extract: <T extends z.AnyZodObject>(options: ExtractOptions<T>) => Promise<ExtractResult<T>>;
     observe: (options?: ObserveOptions) => Promise<ObserveResult[]>;
 }
+export type BrowserContext = PlaywrightContext;

package/dist/types/stagehand.d.ts

@@ -12,7 +12,7 @@ export interface ConstructorParams {
     debugDom?: boolean;
     llmProvider?: LLMProvider;
     headless?: boolean;
-    logger?: (message: LogLine) => void;
+    logger?: (message: LogLine) => void | Promise<void>;
     domSettleTimeoutMs?: number;
     browserbaseSessionCreateParams?: Browserbase.Sessions.SessionCreateParams;
     enableCaching?: boolean;

package/lib/index.ts

@@ -1,5 +1,5 @@
 import { Browserbase } from "@browserbasehq/sdk";
-import { type BrowserContext, chromium } from "@playwright/test";
+import { chromium } from "@playwright/test";
 import { randomUUID } from "crypto";
 import dotenv from "dotenv";
 import fs from "fs";
@@ -9,7 +9,7 @@ import { z } from "zod";
 import { BrowserResult } from "../types/browser";
 import { LogLine } from "../types/log";
 import { GotoOptions } from "../types/playwright";
-import { Page } from "../types/page";
+import { Page, BrowserContext } from "../types/page";
 import {
   ActOptions,
   ActResult,
@@ -24,8 +24,6 @@ import {
   ObserveResult,
 } from "../types/stagehand";
 import { scriptContent } from "./dom/build/scriptContent";
-import { StagehandExtractHandler } from "./handlers/extractHandler";
-import { StagehandObserveHandler } from "./handlers/observeHandler";
 import { LLMClient } from "./llm/LLMClient";
 import { LLMProvider } from "./llm/LLMProvider";
 import { logLineToString } from "./utils";
@@ -303,6 +301,10 @@ async function applyStealthScripts(context: BrowserContext) {
   });
 }
 
+const defaultLogger = async (logLine: LogLine) => {
+  console.log(logLineToString(logLine));
+};
+
 export class Stagehand {
   private stagehandPage!: StagehandPage;
   private stagehandContext!: StagehandContext;
@@ -316,18 +318,15 @@ export class Stagehand {
   public llmProvider: LLMProvider;
   public enableCaching: boolean;
 
-  private internalLogger: (logLine: LogLine) => void;
   private apiKey: string | undefined;
   private projectId: string | undefined;
-  private externalLogger?: (logLine: LogLine) => void;
+  // We want external logger to accept async functions
+  private externalLogger?: (logLine: LogLine) => void | Promise<void>;
   private browserbaseSessionCreateParams?: Browserbase.Sessions.SessionCreateParams;
   public variables: { [key: string]: unknown };
   private contextPath?: string;
   private llmClient: LLMClient;
 
-  private extractHandler?: StagehandExtractHandler;
-  private observeHandler?: StagehandObserveHandler;
-
   constructor(
     {
       env,
@@ -348,8 +347,7 @@ export class Stagehand {
       env: "BROWSERBASE",
     },
   ) {
-    this.externalLogger = logger;
-    this.internalLogger = this.log.bind(this);
+    this.externalLogger = logger || defaultLogger;
     this.enableCaching =
       enableCaching ??
       (process.env.ENABLE_CACHING && process.env.ENABLE_CACHING === "true");
@@ -372,10 +370,7 @@ export class Stagehand {
 
   public get logger(): (logLine: LogLine) => void {
     return (logLine: LogLine) => {
-      this.internalLogger(logLine);
-      if (this.externalLogger) {
-        this.externalLogger(logLine);
-      }
+      this.log(logLine);
     };
   }
 
@@ -398,6 +393,11 @@ export class Stagehand {
   }
 
   public get context(): BrowserContext {
+    if (!this.stagehandContext) {
+      throw new Error(
+        "Stagehand not initialized. Make sure to await stagehand.init() first.",
+      );
+    }
     return this.stagehandContext.context;
   }
 
@@ -504,9 +504,6 @@ export class Stagehand {
     // Normal Logging
     if (this.externalLogger) {
       this.externalLogger(logObj);
-    } else {
-      const logMessage = logLineToString(logObj);
-      console.log(logMessage);
     }
 
     // Add the logs to the browserbase session
@@ -613,3 +610,4 @@ export * from "../types/log";
 export * from "../types/model";
 export * from "../types/playwright";
 export * from "../types/stagehand";
+export * from "../types/page";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@browserbasehq/stagehand",
-  "version": "1.8.0",
+  "version": "1.8.1-alpha-5c6f14bade201e08cb86d2e14e246cb65707f7ee",
   "description": "An AI web browsing framework focused on simplicity and extensibility.",
   "main": "./dist/index.js",
   "module": "./dist/index.js",