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

src/stagehand/types/session_extract_params.py

@@ -44,6 +44,12 @@ class Options(TypedDict, total=False):
     model: OptionsModel
     """Model configuration object or model name string (e.g., 'openai/gpt-5-nano')"""
 
+    screenshot: bool
+    """
+    When true, include a screenshot of the current viewport in the extraction LLM
+    call. Defaults to false.
+    """
+
     selector: str
     """CSS selector to scope extraction to a specific element"""
 

tests/api_resources/test_sessions.py

@@ -593,6 +593,7 @@ def test_method_extract_with_all_params_overload_1(self, client: Stagehand) -> N
                     "project": "my-gcp-project",
                     "provider": "openai",
                 },
+                "screenshot": False,
                 "selector": "#main-content",
                 "timeout": 30000,
             },
@@ -682,6 +683,7 @@ def test_method_extract_with_all_params_overload_2(self, client: Stagehand) -> N
                     "project": "my-gcp-project",
                     "provider": "openai",
                 },
+                "screenshot": False,
                 "selector": "#main-content",
                 "timeout": 30000,
             },
@@ -1729,6 +1731,7 @@ async def test_method_extract_with_all_params_overload_1(self, async_client: Asy
                     "project": "my-gcp-project",
                     "provider": "openai",
                 },
+                "screenshot": False,
                 "selector": "#main-content",
                 "timeout": 30000,
             },
@@ -1818,6 +1821,7 @@ async def test_method_extract_with_all_params_overload_2(self, async_client: Asy
                     "project": "my-gcp-project",
                     "provider": "openai",
                 },
+                "screenshot": False,
                 "selector": "#main-content",
                 "timeout": 30000,
             },