Skip to content

Commit 6bf38b7

Browse files
committed
Using bare minimum code
1 parent 867d0d6 commit 6bf38b7

File tree

4 files changed

+11
-4
lines changed

4 files changed

+11
-4
lines changed

sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/AgentsClientBuilder.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@
1919
import com.azure.core.http.HttpPipeline;
2020
import com.azure.core.http.HttpPipelineBuilder;
2121
import com.azure.core.http.HttpPipelinePosition;
22+
import com.azure.core.http.HttpRequest;
23+
import com.azure.core.http.HttpResponse;
2224
import com.azure.core.http.policy.AddDatePolicy;
2325
import com.azure.core.http.policy.AddHeadersFromContextPolicy;
2426
import com.azure.core.http.policy.AddHeadersPolicy;
@@ -43,6 +45,8 @@
4345
import com.openai.client.okhttp.OpenAIOkHttpClient;
4446
import com.openai.client.okhttp.OpenAIOkHttpClientAsync;
4547
import com.openai.credential.BearerTokenCredential;
48+
import reactor.core.publisher.Mono;
49+
4650
import java.time.Duration;
4751
import java.util.ArrayList;
4852
import java.util.Comparator;
@@ -407,7 +411,7 @@ private OpenAIOkHttpClientAsync.Builder getOpenAIAsyncClientBuilder() {
407411
}
408412

409413
private HttpClient getOpenAIHttpClient() {
410-
return new PolicyDecoratingHttpClient(createHttpPipeline());
414+
return createHttpPipeline()::send;
411415
}
412416

413417
private static final ClientLogger LOGGER = new ClientLogger(AgentsClientBuilder.class);

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ private static BinaryData toBinaryData(HttpRequestBody requestBody) {
154154
if (requestBody == null) {
155155
return null;
156156
}
157-
157+
158158
try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
159159
requestBody.writeTo(outputStream);
160160
return BinaryData.fromBytes(outputStream.toByteArray());

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,10 @@
2020
* the provided delegate client. This makes it possible to reuse the Azure pipeline policy system with HTTP clients that
2121
* originate outside of the Azure SDK (e.g. OpenAI generated clients).
2222
*/
23+
@Deprecated
2324
public final class PolicyDecoratingHttpClient implements HttpClient {
2425

25-
// private final HttpClient delegate;
26+
// private final HttpClient delegate;
2627
private final HttpPipeline pipeline;
2728

2829
/**
@@ -31,7 +32,7 @@ public final class PolicyDecoratingHttpClient implements HttpClient {
3132
* @param delegate Underlying HTTP client that performs the actual network I/O.
3233
* @param policies Policies that should run before the request reaches the delegate.
3334
*/
34-
@Deprecated(forRemoval = true)
35+
@Deprecated
3536
public PolicyDecoratingHttpClient(HttpClient delegate, List<HttpPipelinePolicy> policies) {
3637
Objects.requireNonNull(delegate, "delegate cannot be null");
3738

sdk/ai/azure-ai-agents/src/test/java/com/azure/ai/agents/implementation/http/PolicyDecoratingHttpClientTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import com.azure.core.http.policy.HttpPipelinePolicy;
1616
import com.azure.core.test.http.MockHttpResponse;
1717
import com.azure.core.util.Context;
18+
import org.junit.jupiter.api.Disabled;
1819
import org.junit.jupiter.api.Test;
1920
import reactor.core.publisher.Mono;
2021

@@ -29,6 +30,7 @@
2930
import static org.junit.jupiter.api.Assertions.assertThrows;
3031
import static org.junit.jupiter.api.Assertions.assertTrue;
3132

33+
@Disabled
3234
class PolicyDecoratingHttpClientTests {
3335

3436
private static final HttpHeaderName PER_CALL_HEADER = HttpHeaderName.fromString("X-Per-Call");

0 commit comments

Comments
 (0)