feat: STG-1756 add Vertex auth params to Stagehand spec

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-eae8400fade7b2c8329c4148f56de92e147c34c0feecb420c015aab6544a9acc.yml
-openapi_spec_hash: 0a9eff1ac1d464e89cbd9db64709b08a
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/browserbase/stagehand-e77d6b15f0a94b16a54ef87a84d2cabe49eb11cff5ceba76f00dd788ff483eab.yml
+openapi_spec_hash: a1dab7fe72a772d188a15305124ebd73
 config_hash: 1fb12ae9b478488bc1e56bfbdc210b01

src/stagehand/types/model_config_param.py

@@ -2,12 +2,55 @@
 
 from __future__ import annotations
 
-from typing import Dict
+from typing import Dict, Union
 from typing_extensions import Literal, Required, Annotated, TypedDict
 
+from .._types import SequenceNotStr
 from .._utils import PropertyInfo
 
-__all__ = ["ModelConfigParam"]
+__all__ = ["ModelConfigParam", "GoogleAuthOptions", "GoogleAuthOptionsCredentials"]
+
+
+class GoogleAuthOptionsCredentials(TypedDict, total=False):
+    """Google Cloud service account credentials"""
+
+    client_email: Required[str]
+
+    private_key: Required[str]
+
+    auth_provider_x509_cert_url: str
+
+    auth_uri: str
+
+    client_id: str
+
+    client_x509_cert_url: str
+
+    private_key_id: str
+
+    project_id: str
+
+    token_uri: str
+
+    type: Literal["service_account"]
+
+    universe_domain: str
+
+
+class GoogleAuthOptions(TypedDict, total=False):
+    """google-auth-library options used to authenticate Vertex AI models"""
+
+    credentials: GoogleAuthOptionsCredentials
+    """Google Cloud service account credentials"""
+
+    project_id: Annotated[str, PropertyInfo(alias="projectId")]
+    """Google Cloud project ID used by google-auth-library"""
+
+    scopes: Union[str, SequenceNotStr[str]]
+    """Google auth scopes for the desired API request"""
+
+    universe_domain: Annotated[str, PropertyInfo(alias="universeDomain")]
+    """Google Cloud universe domain"""
 
 
 class ModelConfigParam(TypedDict, total=False):
@@ -20,8 +63,17 @@ class ModelConfigParam(TypedDict, total=False):
     base_url: Annotated[str, PropertyInfo(alias="baseURL")]
     """Base URL for the model provider"""
 
+    google_auth_options: Annotated[GoogleAuthOptions, PropertyInfo(alias="googleAuthOptions")]
+    """google-auth-library options used to authenticate Vertex AI models"""
+
     headers: Dict[str, str]
     """Custom headers sent with every request to the model provider"""
 
-    provider: Literal["openai", "anthropic", "google", "microsoft", "bedrock"]
+    location: str
+    """Google Cloud location for Vertex AI models"""
+
+    project: str
+    """Google Cloud project ID for Vertex AI models"""
+
+    provider: Literal["openai", "anthropic", "google", "microsoft", "bedrock", "vertex"]
     """AI provider for the model (or provide a baseURL endpoint instead)"""