Feat : Support AWS bedrock base models

[detunjiSamuel]

AWS Bedrock Provider Integration

Added support for AWS Bedrock as a new LLM provider:

• Anthropic Claude models
• Amazon Titan models (text & embeddings)
• AI21 Jurassic models

Key Changes

Added Bedrock provider implementation with model-specific handlers:

• Anthropic: Reused existing Anthropic provider code with Bedrock-specific modifications ( no actual changes)
• Titan: Added chat completion and embedding support
• AI21: Added completion and chat completion support
• Request/response mapping between Hub formats and Bedrock formats

Testing Notes

All tests pass using AWS credentials in us-east-1/2 regions
Verified error handling for invalid credentials/models
Tested non-streaming responses ( models in Bedrock don't seem to have streaming types )

Review notes

The model ID from AWS link does not work consistently.
Instead, use the Inference profile ARN or Inference profile ID from the cross-region reference tab as your model_id.

Issue: #20

/claim #20

[CLAassistant]

All committers have signed the CLA.

[detunjiSamuel]

Hey @nirga ,

Quick question about integrating Stability AI models in our Hub. I'm looking at AWS Bedrock's Stable Diffusion 3.5 integration (from their model catalog).

Not sure about the best format to implement this:

1. Should we use the existing chat_completions since it takes a similar prompt format?
2. Or would it make more sense to create a new Content type specifically for image generation?

Would appreciate your thoughts on this.

Thanks!

[nirga]

I think it should be in a new api @detunjiSamuel
But we can do it in a separate PR

[detunjiSamuel]

Hey @nirga , I have completed this PR and would appreciate your review.
I've verified the changes through:

• External Python file testing
• Manual console verification
• Updated test suite
• Results exported to Traceloop from the Pipeline.

Please let me know if any additional information or changes are needed.

Thank you!

[algora-pbc]

💵 To receive payouts, sign up on Algora, link your Github account and connect with Stripe.

[nirga]

Hey @detunjiSamuel, can you take a look at the tests? Seems to be failing

[detunjiSamuel]

@nirga , Thanks for pointing that out.
The error was caused by missing AWS credentials in the CI environment.
I've fixed this by adding recordings of my local AWS requests and using AWS Smithy to replay the responses.
The most recent commit resolves that issue.

Thanks again!

[detunjiSamuel]

Hi @nirga ,

I wanted to follow up on this PR as it's been a few weeks since the last update. As mentioned previously, I've addressed the test failures by implementing AWS Smithy to replay the test responses, which resolved the CI issues.

Everything is passing and ready for review.
Please let me know if you need any additional information or have any concerns I can address.