Skip to content

Amazon Bedrock - Model Throughput Provisioning - #38850

Merged
ferruzzi merged 5 commits into
apache:mainfrom
aws-mwaa:ferruzzi/bedrock/m4-throughput
Apr 11, 2024
Merged

Amazon Bedrock - Model Throughput Provisioning#38850
ferruzzi merged 5 commits into
apache:mainfrom
aws-mwaa:ferruzzi/bedrock/m4-throughput

Conversation

@ferruzzi

Copy link
Copy Markdown
Contributor

Adds support for provisioning model throughput on Amazon Bedrock including: operators, sensors, waiters, and triggerer, as well as unit tests and system tests for all of the above, and doc updates.

Manually tested in Breeze with the Trigger and with the Sensor:
image

Sits on top of a change in #38849, please make sure that is merged before this one.


^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in newsfragments.

Comment threadairflow/providers/amazon/aws/operators/bedrock.py Outdated
Comment threadairflow/providers/amazon/aws/sensors/bedrock.py Outdated
Comment threadtests/system/providers/amazon/aws/example_bedrock.py Outdated
@ferruzzi
ferruzziforce-pushed the ferruzzi/bedrock/m4-throughput branch from 3d61fcc to 90eea2dCompareApril 9, 2024 18:54
@ferruzzi

Copy link
Copy Markdown
ContributorAuthor

Dependency merged; I'll get this rebased on top of that one and merge it this afternoon.

@ferruzzi
ferruzziforce-pushed the ferruzzi/bedrock/m4-throughput branch from bffa635 to fe2fd35CompareApril 10, 2024 20:06
@ferruzzi

Copy link
Copy Markdown
ContributorAuthor

Gah, static check is just a merge issue, imported the sensor twice. Easy fix.

@ferruzzi
ferruzzi merged commit c25d346 into apache:mainApr 11, 2024
@ferruzzi
ferruzzi deleted the ferruzzi/bedrock/m4-throughput branch April 11, 2024 18:11
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

@ferruzzi@o-nikolas@vincbeck