File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -108,6 +108,19 @@ def initialize(
108108 "model_api_key is required, and can be set via environ: \" MODEL_API_KEY\" "
109109 end
110110
111+ headers = { }
112+ custom_headers_env = ENV [ "STAGEHAND_CUSTOM_HEADERS" ]
113+ unless custom_headers_env . nil?
114+ parsed = { }
115+ custom_headers_env . split ( "\n " ) . each do |line |
116+ colon = line . index ( ":" )
117+ unless colon . nil?
118+ parsed [ line [ 0 ...colon ] . strip ] = line [ ( colon + 1 ) ..] . strip
119+ end
120+ end
121+ headers = parsed . merge ( headers )
122+ end
123+
111124 @browserbase_api_key = browserbase_api_key . to_s
112125 @browserbase_project_id = browserbase_project_id . to_s
113126 @model_api_key = model_api_key . to_s
@@ -117,7 +130,8 @@ def initialize(
117130 timeout : timeout ,
118131 max_retries : max_retries ,
119132 initial_retry_delay : initial_retry_delay ,
120- max_retry_delay : max_retry_delay
133+ max_retry_delay : max_retry_delay ,
134+ headers : headers
121135 )
122136
123137 @sessions = Stagehand ::Resources ::Sessions . new ( client : self )
You can’t perform that action at this time.
0 commit comments