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]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
1 out of 2 committers have signed the CLA.

✅ detunjiSamuel
❌ nirga
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

[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 - I want to merge this PR, can you sign the CLA?

[nirga]

Ping @detunjiSamuel - can you sign the CLA?