Skip to content

Commit 5515959

Browse files
feat: support setting headers via env
1 parent 23c3eef commit 5515959

1 file changed

Lines changed: 15 additions & 1 deletion

File tree

lib/stagehand/client.rb

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)