Skip to content

Add AWS Bedrock AI Model Support in Playbooks#2164

Open
moshloop wants to merge 11 commits intomainfrom
cosine/add-aws-bedrock-ai-support-z14bzw
Open

Add AWS Bedrock AI Model Support in Playbooks#2164
moshloop wants to merge 11 commits intomainfrom
cosine/add-aws-bedrock-ai-support-z14bzw

Conversation

@moshloop
Copy link
Copy Markdown
Member

This pull request introduces support for the AWS Bedrock AI model in the playbooks system. The key changes include:

  1. New API Constant: Added a new constant for LLMBackendBedrock in api/global.go to represent the Bedrock model.
  2. Playbook Action Modification: Updated Populate function in api/v1/playbook_actions.go to handle the AWS connection type and set the appropriate model, while using the 'region' property for the API URL.
  3. Bedrock Model Wrapper: Implemented a new file llm/bedrock.go which includes the BedrockModelWrapper class to handle interactions with the Bedrock API, including calling the model and processing responses.
  4. Cost Calculation: Modified cost calculation logic in llm/cost.go to accommodate Bedrock without any associated costs initially, and updated the tests to include cases for Bedrock.

These changes aim to enhance the agility of our playbook actions by incorporating AWS's AI capabilities, allowing users to leverage Bedrock's models.


This pull request was co-created with Cosine Genie

Original Task: mission-control/yf0cdttccsqs
Author: Moshe Immerman

@coderabbitai
Copy link
Copy Markdown
Contributor

coderabbitai bot commented Mar 17, 2026

Warning

Rate limit exceeded

@moshloop has exceeded the limit for the number of commits that can be reviewed per hour. Please wait 1 minutes and 8 seconds before requesting another review.

⌛ How to resolve this issue?

After the wait time has elapsed, a review can be triggered using the @coderabbitai review command as a PR comment. Alternatively, push new commits to this PR.

We recommend that you space out your commits to avoid hitting the rate limit.

🚦 How do rate limits work?

CodeRabbit enforces hourly rate limits for each developer per organization.

Our paid plans have higher rate limits than the trial, open-source and free plans. In all cases, we re-allow further reviews after a brief timeout.

Please see our FAQ for further information.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro

Run ID: c196b1d8-5b94-4fd5-9661-b657c696c2f6

📥 Commits

Reviewing files that changed from the base of the PR and between 8667d82 and af6d1ee.

📒 Files selected for processing (7)
  • api/global.go
  • api/v1/playbook_actions.go
  • go.mod
  • llm/bedrock.go
  • llm/cost.go
  • llm/cost_test.go
  • llm/llm.go
✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Commit unit tests in branch cosine/add-aws-bedrock-ai-support-z14bzw
📝 Coding Plan
  • Generate coding plan for human review comments

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands and usage tips.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant