Manually end output stream after Bedrock InvokeAgent

[tjcsrc16]

Hi all, I am using the Javascript API to invoke an agent from inside a Lambda, e.g.,

const response = await bedrockAgentRuntimeClient.send(invokeAgentCommand)
for await (const streamEvent of response.completion) {
...
}

Sometimes I want to stop and break the connection early before all the events are read. According to this, breaking out of the loop should close the connection. However, I noticed that when I do that and the lambda handler completes, the lambda stays active for about 2 minutes. I assume this is because the connection to the Bedrock stream stays open until a timeout on the bedrock client.
Is there a way to gracefully stop the connection without waiting for all of the events?