fix(client): allow updating header/query affecting fields in toBuilder()

Author: stainless-app[bot]

stagehand-java-core/src/main/kotlin/com/browserbase/api/core/ClientOptions.kt

@@ -485,23 +485,24 @@ private constructor(
             headers.put("X-Stainless-Runtime", "JRE")
             headers.put("X-Stainless-Runtime-Version", getJavaVersion())
             headers.put("X-Stainless-Kotlin-Version", KotlinVersion.CURRENT.toString())
+            // We replace after all the default headers to allow end-users to overwrite them.
+            headers.replaceAll(this.headers.build())
+            queryParams.replaceAll(this.queryParams.build())
             browserbaseApiKey.let {
                 if (!it.isEmpty()) {
-                    headers.put("x-bb-api-key", it)
+                    headers.replace("x-bb-api-key", it)
                 }
             }
             browserbaseProjectId.let {
                 if (!it.isEmpty()) {
-                    headers.put("x-bb-project-id", it)
+                    headers.replace("x-bb-project-id", it)
                 }
             }
             modelApiKey.let {
                 if (!it.isEmpty()) {
-                    headers.put("x-model-api-key", it)
+                    headers.replace("x-model-api-key", it)
                 }
             }
-            headers.replaceAll(this.headers.build())
-            queryParams.replaceAll(this.queryParams.build())
 
             return ClientOptions(
                 httpClient,

stagehand-java-core/src/test/kotlin/com/browserbase/api/core/ClientOptionsTest.kt

@@ -16,6 +16,37 @@ internal class ClientOptionsTest {
 
     private val httpClient = mock<HttpClient>()
 
+    @Test
+    fun putHeader_canOverwriteDefaultHeader() {
+        val clientOptions =
+            ClientOptions.builder()
+                .httpClient(httpClient)
+                .putHeader("User-Agent", "My User Agent")
+                .browserbaseApiKey("My Browserbase API Key")
+                .browserbaseProjectId("My Browserbase Project ID")
+                .modelApiKey("My Model API Key")
+                .build()
+
+        assertThat(clientOptions.headers.values("User-Agent")).containsExactly("My User Agent")
+    }
+
+    @Test
+    fun toBuilder_bbApiKeyAuthCanBeUpdated() {
+        var clientOptions =
+            ClientOptions.builder()
+                .httpClient(httpClient)
+                .browserbaseApiKey("My Browserbase API Key")
+                .browserbaseProjectId("My Browserbase Project ID")
+                .modelApiKey("My Model API Key")
+                .build()
+
+        clientOptions =
+            clientOptions.toBuilder().browserbaseApiKey("another My Browserbase API Key").build()
+
+        assertThat(clientOptions.headers.values("x-bb-api-key"))
+            .containsExactly("another My Browserbase API Key")
+    }
+
     @Test
     fun toBuilder_whenOriginalClientOptionsGarbageCollected_doesNotCloseOriginalClient() {
         var clientOptions =