🚀 Feature: Tracing support for Amazon Bedrock knowledge bases

Which component is this feature for?

Bedrock Instrumentation

🔖 Feature description

Add tracing support for Amazon Bedrock knowledge bases.

🎤 Why is this feature needed ?

Developers may configure their RAG application to use the RetrieveAndGenerate API to query their Bedrock knowledge base and generate responses from the information it retrieves. They may also call the Retrieve API to query their knowledge base with information retrieved directly from the knowledge base.

✌️ How do you aim to achieve this?

1. Add instrumentation logic for the bedrock-agent-runtime service and patch the retreive and retrieve_and_generate methods.
2. Create spans for the requests.

🔄️ Additional Information

No response

👀 Have you spent some time to check if this feature request has been raised before?

• I checked and didn't find similar issue

Are you willing to submit PR?

Yes I am willing to submit a PR!

Start with the Bedrock Instrumentation entry point and the bedrock-agent-runtime service integration, then review the Retrieve and RetrieveAndGenerate API documentation linked in the issue. Trace both methods and define spans for their requests; the work is done when both knowledge-base operations are instrumented.