feat: Add screenshot option to Extract

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 8
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/browserbase/stagehand-e77d6b15f0a94b16a54ef87a84d2cabe49eb11cff5ceba76f00dd788ff483eab.yml
-openapi_spec_hash: a1dab7fe72a772d188a15305124ebd73
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/browserbase/stagehand-80502d74c1be605e77d45ff2b54297fe34ce85dbad1e8f2dfa30ba6d09601219.yml
+openapi_spec_hash: fd62f768756a400c3ecd695bfcf3845a
 config_hash: 1fb12ae9b478488bc1e56bfbdc210b01

lib/stagehand/models/session_extract_params.rb

@@ -72,6 +72,13 @@ class Options < Stagehand::Internal::Type::BaseModel
         #   @return [Stagehand::Models::ModelConfig, String, nil]
         optional :model, union: -> { Stagehand::SessionExtractParams::Options::Model }
 
+        # @!attribute screenshot
+        #   When true, include a screenshot of the current viewport in the extraction LLM
+        #   call. Defaults to false.
+        #
+        #   @return [Boolean, nil]
+        optional :screenshot, Stagehand::Internal::Type::Boolean
+
         # @!attribute selector
         #   CSS selector to scope extraction to a specific element
         #
@@ -84,11 +91,16 @@ class Options < Stagehand::Internal::Type::BaseModel
         #   @return [Float, nil]
         optional :timeout, Float
 
-        # @!method initialize(ignore_selectors: nil, model: nil, selector: nil, timeout: nil)
+        # @!method initialize(ignore_selectors: nil, model: nil, screenshot: nil, selector: nil, timeout: nil)
+        #   Some parameter documentations has been truncated, see
+        #   {Stagehand::Models::SessionExtractParams::Options} for more details.
+        #
         #   @param ignore_selectors [Array<String>] Selectors for elements and subtrees that should be excluded from extraction
         #
         #   @param model [Stagehand::Models::ModelConfig, String] Model configuration object or model name string (e.g., 'openai/gpt-5-nano')
         #
+        #   @param screenshot [Boolean] When true, include a screenshot of the current viewport in the extraction LLM ca
+        #
         #   @param selector [String] CSS selector to scope extraction to a specific element
         #
         #   @param timeout [Float] Timeout in ms for the extraction

rbi/stagehand/models/session_extract_params.rbi

@@ -127,6 +127,14 @@ module Stagehand
         end
         attr_writer :model
 
+        # When true, include a screenshot of the current viewport in the extraction LLM
+        # call. Defaults to false.
+        sig { returns(T.nilable(T::Boolean)) }
+        attr_reader :screenshot
+
+        sig { params(screenshot: T::Boolean).void }
+        attr_writer :screenshot
+
         # CSS selector to scope extraction to a specific element
         sig { returns(T.nilable(String)) }
         attr_reader :selector
@@ -145,6 +153,7 @@ module Stagehand
           params(
             ignore_selectors: T::Array[String],
             model: T.any(Stagehand::ModelConfig::OrHash, String),
+            screenshot: T::Boolean,
             selector: String,
             timeout: Float
           ).returns(T.attached_class)
@@ -154,6 +163,9 @@ module Stagehand
           ignore_selectors: nil,
           # Model configuration object or model name string (e.g., 'openai/gpt-5-nano')
           model: nil,
+          # When true, include a screenshot of the current viewport in the extraction LLM
+          # call. Defaults to false.
+          screenshot: nil,
           # CSS selector to scope extraction to a specific element
           selector: nil,
           # Timeout in ms for the extraction
@@ -166,6 +178,7 @@ module Stagehand
             {
               ignore_selectors: T::Array[String],
               model: T.any(Stagehand::ModelConfig, String),
+              screenshot: T::Boolean,
               selector: String,
               timeout: Float
             }

sig/stagehand/models/session_extract_params.rbs

@@ -63,6 +63,7 @@ module Stagehand
         {
           ignore_selectors: ::Array[String],
           model: Stagehand::Models::SessionExtractParams::Options::model,
+          screenshot: bool,
           selector: String,
           timeout: Float
         }
@@ -78,6 +79,10 @@ module Stagehand
           Stagehand::Models::SessionExtractParams::Options::model
         ) -> Stagehand::Models::SessionExtractParams::Options::model
 
+        attr_reader screenshot: bool?
+
+        def screenshot=: (bool) -> bool
+
         attr_reader selector: String?
 
         def selector=: (String) -> String
@@ -89,13 +94,15 @@ module Stagehand
         def initialize: (
           ?ignore_selectors: ::Array[String],
           ?model: Stagehand::Models::SessionExtractParams::Options::model,
+          ?screenshot: bool,
           ?selector: String,
           ?timeout: Float
         ) -> void
 
         def to_hash: -> {
           ignore_selectors: ::Array[String],
           model: Stagehand::Models::SessionExtractParams::Options::model,
+          screenshot: bool,
           selector: String,
           timeout: Float
         }