Module: Stagehand::Local::ClientPatch

Included in:

Client

Defined in:

lib/stagehand/local.rb

Instance Method Summary

• #close ⇒ Object
• #initialize(server: "remote", **kwargs) ⇒ Object
• #request(req) ⇒ Object

Instance Method Details

#close ⇒ Object

# File 'lib/stagehand/local.rb', line 391

def close
  super if defined?(super)
ensure
  @local_server_manager&.close
end

#initialize(server: "remote", **kwargs) ⇒ Object

# File 'lib/stagehand/local.rb', line 366

def initialize(server: "remote", **kwargs)
  @server_mode = server.to_s

  if @server_mode == "local"
    base_url = kwargs[:base_url]
    kwargs[:base_url] = base_url.nil? ? "http://#{DEFAULT_HOST}" : base_url
    kwargs[:browserbase_api_key] =
      kwargs[:browserbase_api_key] || ENV["BROWSERBASE_API_KEY"] || ""
  end

  super(**kwargs)

  return unless @server_mode == "local"

  @local_server_manager = Stagehand::Local::ServerManager.new(
    model_api_key: @model_api_key,
    browserbase_api_key: @browserbase_api_key
  )
end

#request(req) ⇒ Object

# File 'lib/stagehand/local.rb', line 386

def request(req)
  ensure_local_server!
  super
end