remove extra lint

Author: filip-michalsky

CHANGELOG.md

@@ -233,13 +233,15 @@
   We're thrilled to announce the release of Stagehand 2.0, bringing significant improvements to make browser automation more powerful, faster, and easier to use than ever before.
 
   ### 🚀 New Features
+
   - **Introducing `stagehand.agent`**: A powerful new way to integrate SOTA Computer use models or Browserbase's [Open Operator](https://operator.browserbase.com) into Stagehand with one line of code! Perfect for multi-step workflows and complex interactions. [Learn more](https://docs.stagehand.dev/concepts/agent)
   - **Lightning-fast `act` and `extract`**: Major performance improvements to make your automations run significantly faster.
   - **Enhanced Logging**: Better visibility into what's happening during automation with improved logging and debugging capabilities.
   - **Comprehensive Documentation**: A completely revamped documentation site with better examples, guides, and best practices.
   - **Improved Error Handling**: More descriptive errors and better error recovery to help you debug issues faster.
 
   ### 🛠️ Developer Experience
+
   - **Better TypeScript Support**: Enhanced type definitions and better IDE integration
   - **Better Error Messages**: Clearer, more actionable error messages to help you debug faster
   - **Improved Caching**: More reliable action caching for better performance
@@ -500,6 +502,7 @@
 - [#316](https://github.com/browserbase/stagehand/pull/316) [`902e633`](https://github.com/browserbase/stagehand/commit/902e633e126a58b80b757ea0ecada01a7675a473) Thanks [@kamath](https://github.com/kamath)! - rename browserbaseResumeSessionID -> browserbaseSessionID
 
 - [#296](https://github.com/browserbase/stagehand/pull/296) [`f11da27`](https://github.com/browserbase/stagehand/commit/f11da27a20409c240ceeea2003d520f676def61a) Thanks [@kamath](https://github.com/kamath)! - - Deprecate fields in `init` in favor of constructor options
+
   - Deprecate `initFromPage` in favor of `browserbaseResumeSessionID` in constructor
   - Rename `browserBaseSessionCreateParams` -> `browserbaseSessionCreateParams`
 

lib/StagehandPage.ts

@@ -1,7 +1,11 @@
 import type { CDPSession, Page as PlaywrightPage, Frame } from "playwright";
 import { selectors } from "playwright";
 import { z } from "zod/v3";
-import { Page, defaultExtractSchema } from "../types/page";
+import {
+  Page,
+  defaultExtractSchema,
+  StagehandScreenshotOptions,
+} from "../types/page";
 import {
   ExtractOptions,
   ExtractResult,
@@ -415,37 +419,41 @@ ${scriptContent} \
           }
 
           // Handle screenshots with CDP
-          if (prop === "screenshot" && this.stagehand.env === "BROWSERBASE") {
-            return async (
-              options: {
-                type?: "png" | "jpeg";
-                quality?: number;
-                fullPage?: boolean;
-                clip?: { x: number; y: number; width: number; height: number };
-                omitBackground?: boolean;
-              } = {},
-            ) => {
-              const cdpOptions: Record<string, unknown> = {
-                format: options.type === "jpeg" ? "jpeg" : "png",
-                quality: options.quality,
-                clip: options.clip,
-                omitBackground: options.omitBackground,
-                fromSurface: true,
-              };
-
-              if (options.fullPage) {
-                cdpOptions.captureBeyondViewport = true;
-              }
+          if (prop === "screenshot") {
+            return async (options: StagehandScreenshotOptions = {}) => {
+              const rawScreenshot: typeof target.screenshot =
+                Object.getPrototypeOf(target).screenshot.bind(target);
+
+              const {
+                useCDP = this.stagehand.env === "BROWSERBASE",
+                ...playwrightOptions
+              } = options;
+
+              if (useCDP && this.stagehand.env === "BROWSERBASE") {
+                const cdpOptions: Record<string, unknown> = {
+                  format: options.type === "jpeg" ? "jpeg" : "png",
+                  quality: options.quality,
+                  clip: options.clip,
+                  omitBackground: options.omitBackground,
+                  fromSurface: true,
+                };
+
+                if (options.fullPage) {
+                  cdpOptions.captureBeyondViewport = true;
+                }
 
-              const data = await this.sendCDP<{ data: string }>(
-                "Page.captureScreenshot",
-                cdpOptions,
-              );
+                const data = await this.sendCDP<{ data: string }>(
+                  "Page.captureScreenshot",
+                  cdpOptions,
+                );
 
-              // Convert base64 to buffer
-              const buffer = Buffer.from(data.data, "base64");
+                // Convert base64 to buffer
+                const buffer = Buffer.from(data.data, "base64");
 
-              return buffer;
+                return buffer;
+              } else {
+                return await rawScreenshot(playwrightOptions);
+              }
             };
           }
 
@@ -739,7 +747,6 @@ ${scriptContent} \
             const result = await this.api.act({
               ...observeResult,
               frameId: this.rootFrameId,
-              modelClientOptions: this.stagehand["modelClientOptions"],
             });
             this.stagehand.addToHistory("act", observeResult, result);
             return result;
@@ -837,10 +844,7 @@ ${scriptContent} \
       if (!instructionOrOptions) {
         let result: ExtractResult<T>;
         if (this.api) {
-          result = await this.api.extract<T>({
-            frameId: this.rootFrameId,
-            modelClientOptions: this.stagehand["modelClientOptions"],
-          });
+          result = await this.api.extract<T>({ frameId: this.rootFrameId });
         } else {
           result = await this.extractHandler.extract();
         }