Description

Our integ tests are failing occassionally because of bedrock async guardrail behavior. This change accounts for the following

In async streaming: The buffering is non-blocking. Tokens are streamed while Guardrails processes the buffered content in the background. This means the response may be returned before Guardrails has finished processing. As a result, we cannot guarantee that the REDACT_MESSAGE is in the response

Bug fix

Testing

How have you tested the change? Verify that the changes do not break functionality or introduce warnings in consuming repositories: agents-docs, agents-tools, agents-cli

• I ran hatch run prepare

Checklist

• I have read the CONTRIBUTING document
• I have added any necessary tests that prove my fix is effective or my feature works
• I have updated the documentation accordingly
• I have added an appropriate example to the documentation to outline the feature, or no new docs are needed
• My changes generate no new warnings
• Any dependent changes have been merged and published

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.