How do I use Java SDK to invoke an Bedrock Agent?

Hello,

I have created several bedrock agents, and using the console I'm able to chat with them to perform the actions I want. Now I want to invoke them from my Restful application.

I'm using the 2.29.6, and my pom.xml (maven) is as follows:

                <dependency>
			<groupId>software.amazon.awssdk</groupId>
			<artifactId>bedrockagentruntime</artifactId>
			<version>2.29.6</version>
		</dependency>
		<dependency>
			<groupId>software.amazon.awssdk</groupId>
			<artifactId>bedrock</artifactId>
			<version>2.29.6</version>
		</dependency>
		<dependency>
			<groupId>software.amazon.awssdk</groupId>
			<artifactId>bedrockruntime</artifactId>
			<version>2.29.6</version>
		</dependency>

I see in the docs that only the async client can be used to invoke agents, so I'm using a CompletableFuture<Void> type, then trying to call the method "invokeAgen" method. Below is a code example I'm trying but I can't seem to make this work and looking for guidance.

The two scenarios I'm running into is either the CompletableFuture blocks indefinitley when using the "get", or it fails with a connection pool

An Error Occurred While Invoking Agent. software.amazon.awssdk.core.exception.SdkClientException: Unable to execute HTTP request: Acquire operation took longer than the configured maximum time. This indicates that a request cannot get a connection from the pool within the specified maximum time. This can be due to high request rate.
Consider taking any of the following actions to mitigate the issue: increase max connections, increase acquire timeout, or slowing the request rate.
Increasing the max connections can increase client throughput (unless the network interface is already fully utilized), but can eventually start to hit operation system limitations on the number of file descriptors used by the process. If you already are fully utilizing your network interface or cannot further increase your connection count, increasing the acquire timeout gives extra time for requests to acquire a connection before timing out. If the connections doesn't free up, the subsequent requests will still timeout.
If the above mechanisms are not able to fix the issue, try smoothing out your requests so that large traffic bursts cannot overload the client, being more efficient with the number of times you need to call AWS, or by increasing the number of hosts sending requests.

Code Example is as follows:

import lombok.extern.log4j.Log4j2;
import org.springframework.stereotype.Service;
import software.amazon.awssdk.auth.credentials.ProfileCredentialsProvider;
import software.amazon.awssdk.core.async.SdkPublisher;
import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.bedrockagentruntime.BedrockAgentRuntimeAsyncClient;
import software.amazon.awssdk.services.bedrockagentruntime.BedrockAgentRuntimeClient;
import software.amazon.awssdk.services.bedrockagentruntime.model.*;
import software.amazon.awssdk.services.bedrockruntime.BedrockRuntimeClient;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;


@Log4j2
@Service
public class AWSBedRockAgentService {

    BedrockAgentRuntimeAsyncClient client;

    private AWSBedRockAgentService()
    {
        client = BedrockAgentRuntimeAsyncClient.builder()
                .region(Region.US_EAST_1) // Replace with your desired region
                .credentialsProvider(ProfileCredentialsProvider.create())
                .build();
    }

    public ChatResponse DoSomething(String prompt, String authToken)
    {
        ChatResponse chatResponse = new ChatResponse();

        // Prepare the request
        String agentId = "<myAgentId>";
        String agentAliasId = "<myAlias>";
        String sessionId = "sameSessionId"; 
        String inputText = "What customers do I have access to?";

        chatResponse.originalPrompt = inputText;
        chatResponse.totalTokensUsed = 9999;
        chatResponse.responses = new ArrayList<String>();

        var map = new HashMap<String, String>();
        map.put("Authorization", authToken);

        SessionState ss = SessionState.builder()
                .sessionAttributes(map)
                .build();

        InvokeAgentRequest request = InvokeAgentRequest.builder()
                .agentId(agentId)
                .agentAliasId(agentAliasId)
                .sessionId(sessionId)
                .inputText(inputText)
                .enableTrace(true)
                .sessionState(ss)
                //.endSession(true) //will need to determine when to do this and when not to.

                .build();

        try {

            InvokeAgentResponse[] resp = {null};

            InvokeAgentResponseHandler handler = new InvokeAgentResponseHandler() {
                @Override
                public void responseReceived(InvokeAgentResponse invokeAgentResponse) {
                    log.info("Response Received from Agent: {}", invokeAgentResponse);

                    resp[0] = invokeAgentResponse;

                    //this.complete();
                }

                @Override
                public void onEventStream(SdkPublisher<ResponseStream> sdkPublisher) {
                    log.info("Agent - On Event Stream: {}", sdkPublisher);
                }

                @Override
                public void exceptionOccurred(Throwable throwable) {
                    log.warn("An Error Occurred While Invoking Agent. {}", throwable.getMessage());

                    //this.complete();
                }

                @Override
                public void complete() {
                    log.info("Agent Processing Completed.");
                }
            };
            // Invoke the agent
            CompletableFuture<Void> agentInvocation = client.invokeAgent(request, handler);

            //Wait for it to complete
            agentInvocation.completeOnTimeout(null, 30000, TimeUnit.SECONDS);

            if(resp.length > 0)
            {
                var agentResponse = resp[0];
                log.info("*************Agent Response************");
                log.info(agentResponse);
            }

            chatResponse.responses.add("No Exception Happened");

        } catch (Exception e) {
            log.error("Error invoking Bedrock agent: {}", e.getMessage());

            chatResponse.responses.add(e.getMessage());

        } finally {
            // Close the client
            client.close();
        }

        return chatResponse;
    }
}

Any thoughts, directions, examples appreciated!