Forwarding request timeout and adding more custom machters for tests

Author: jpalvarezl

sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/implementation/http/HttpClientHelper.java

@@ -12,6 +12,7 @@
 import com.azure.core.util.CoreUtils;
 import com.azure.core.util.logging.ClientLogger;
 import com.openai.core.RequestOptions;
+import com.openai.core.Timeout;
 import com.openai.core.http.Headers;
 import com.openai.core.http.HttpClient;
 import com.openai.core.http.HttpRequest;
@@ -21,6 +22,7 @@
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.io.UncheckedIOException;
+import java.time.Duration;
 import java.util.List;
 import java.util.Objects;
 import java.util.concurrent.CompletableFuture;
@@ -94,7 +96,13 @@ public CompletableFuture<HttpResponse> executeAsync(HttpRequest request, Request
                 return failedFuture(runtimeException);
             }
 
-            return this.httpPipeline.send(azureRequest, new Context("azure-eagerly-read-response", true))
+            Context requestContext = new Context("azure-eagerly-read-response", true);
+            Timeout timeout = requestOptions.getTimeout();
+            if (timeout != null && !timeout.connect().isZero() && !timeout.connect().isNegative()) {
+                requestContext.addData("azure-response-timeout", timeout.connect());
+            }
+
+            return this.httpPipeline.send(azureRequest, requestContext)
                 .map(response -> (HttpResponse) new AzureHttpResponseAdapter(response))
                 //                    .onErrorMap(t -> {
                 //                        // 2 or 3 from Azure Errors, should be mapped to Stainless Error.

sdk/ai/azure-ai-agents/src/test/java/com/azure/ai/agents/ClientTestBase.java

@@ -105,7 +105,16 @@ private void addTestRecordCustomSanitizers() {
     }
 
     private void addCustomMatchers() {
-        interceptorManager.addMatchers(new CustomMatcher().setExcludedHeaders(Arrays.asList("Cookie", "Set-Cookie")));
+        interceptorManager.addMatchers(new CustomMatcher().setExcludedHeaders(
+            Arrays.asList("Cookie",
+                "Set-Cookie",
+                "X-Stainless-Arch",
+                "X-Stainless-Lang",
+                "X-Stainless-OS",
+                "X-Stainless-OS-Version",
+                "X-Stainless-Package-Version",
+                "X-Stainless-Runtime",
+                "X-Stainless-Runtime-Version")));
     }
 
     protected void sleep(long millis) {