🚀 Feature: Tracing support for Amazon Bedrock knowledge bases

[tonykchen]

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!

[tonykchen]

I'll likely work on this one before #1647

[nirga]

Thanks @tonykchen! Let me know if I can help :)