feat: Include LLM headers in ModelConfig

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%2Fstagehand-a4a5ea048bb50d06460d81d6828b53b12b19e9224121ee6338dcd1f0781e22a1.yml
-openapi_spec_hash: 9b81c0ae04576318d13d7a80d4ab7b5a
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/browserbase%2Fstagehand-bc309fd00fe0507f4cbe3bc77fa27d0fbffeaa6e71998778da34de42608a67e8.yml
+openapi_spec_hash: 1db1af5c1b068bba1d652102f4454668
 config_hash: d6c6f623d03971bdba921650e5eb7e5f

lib/stagehand/models/model_config.rb

@@ -21,19 +21,27 @@ class ModelConfig < Stagehand::Internal::Type::BaseModel
       #   @return [String, nil]
       optional :base_url, String, api_name: :baseURL
 
+      # @!attribute headers
+      #   Custom headers sent with every request to the model provider
+      #
+      #   @return [Hash{Symbol=>String}, nil]
+      optional :headers, Stagehand::Internal::Type::HashOf[String]
+
       # @!attribute provider
       #   AI provider for the model (or provide a baseURL endpoint instead)
       #
       #   @return [Symbol, Stagehand::Models::ModelConfig::Provider, nil]
       optional :provider, enum: -> { Stagehand::ModelConfig::Provider }
 
-      # @!method initialize(model_name:, api_key: nil, base_url: nil, provider: nil)
+      # @!method initialize(model_name:, api_key: nil, base_url: nil, headers: nil, provider: nil)
       #   @param model_name [String] Model name string with provider prefix (e.g., 'openai/gpt-5-nano')
       #
       #   @param api_key [String] API key for the model provider
       #
       #   @param base_url [String] Base URL for the model provider
       #
+      #   @param headers [Hash{Symbol=>String}] Custom headers sent with every request to the model provider
+      #
       #   @param provider [Symbol, Stagehand::Models::ModelConfig::Provider] AI provider for the model (or provide a baseURL endpoint instead)
 
       # AI provider for the model (or provide a baseURL endpoint instead)

rbi/stagehand/models/model_config.rbi

@@ -26,6 +26,13 @@ module Stagehand
       sig { params(base_url: String).void }
       attr_writer :base_url
 
+      # Custom headers sent with every request to the model provider
+      sig { returns(T.nilable(T::Hash[Symbol, String])) }
+      attr_reader :headers
+
+      sig { params(headers: T::Hash[Symbol, String]).void }
+      attr_writer :headers
+
       # AI provider for the model (or provide a baseURL endpoint instead)
       sig { returns(T.nilable(Stagehand::ModelConfig::Provider::OrSymbol)) }
       attr_reader :provider
@@ -38,6 +45,7 @@ module Stagehand
           model_name: String,
           api_key: String,
           base_url: String,
+          headers: T::Hash[Symbol, String],
           provider: Stagehand::ModelConfig::Provider::OrSymbol
         ).returns(T.attached_class)
       end
@@ -48,6 +56,8 @@ module Stagehand
         api_key: nil,
         # Base URL for the model provider
         base_url: nil,
+        # Custom headers sent with every request to the model provider
+        headers: nil,
         # AI provider for the model (or provide a baseURL endpoint instead)
         provider: nil
       )
@@ -59,6 +69,7 @@ module Stagehand
             model_name: String,
             api_key: String,
             base_url: String,
+            headers: T::Hash[Symbol, String],
             provider: Stagehand::ModelConfig::Provider::OrSymbol
           }
         )

sig/stagehand/models/model_config.rbs

@@ -5,6 +5,7 @@ module Stagehand
         model_name: String,
         api_key: String,
         base_url: String,
+        headers: ::Hash[Symbol, String],
         provider: Stagehand::Models::ModelConfig::provider
       }
 
@@ -19,6 +20,10 @@ module Stagehand
 
       def base_url=: (String) -> String
 
+      attr_reader headers: ::Hash[Symbol, String]?
+
+      def headers=: (::Hash[Symbol, String]) -> ::Hash[Symbol, String]
+
       attr_reader provider: Stagehand::Models::ModelConfig::provider?
 
       def provider=: (
@@ -29,13 +34,15 @@ module Stagehand
         model_name: String,
         ?api_key: String,
         ?base_url: String,
+        ?headers: ::Hash[Symbol, String],
         ?provider: Stagehand::Models::ModelConfig::provider
       ) -> void
 
       def to_hash: -> {
         model_name: String,
         api_key: String,
         base_url: String,
+        headers: ::Hash[Symbol, String],
         provider: Stagehand::Models::ModelConfig::provider
       }