tmokmss / bedrock-pr-reviewer

AI-based Pull Request Summarizer and Reviewer with Chat Capabilities.
MIT License
49 stars 12 forks source link

Support Bedrock tags #15

Open matanbaruch opened 3 months ago

matanbaruch commented 3 months ago

Hey, I would be nice to support Bedrock tags from action inputs Something like

bedrock_tags: 
  required: false
  default: ''

And if user provide them as list

input.bedrock_tags: department=foo,team=bar

Ref: https://docs.aws.amazon.com/bedrock/latest/userguide/tagging.html

tmokmss commented 2 weeks ago

@matanbaruch Hi, sorry for the late reply.

This tool just calls specified models with invokeModel API, so as far as I read the doc, we cannot set the tag from our side.

I think you can set bedrock_heavy_model or bedrock_light_model to your models with tags configured, and then you can track model usage using tags.