Skip to content

Add Bedrock Batch Inference Operator and accompanying parts#48468

Merged
o-nikolas merged 21 commits into
apache:mainfrom
aws-mwaa:ferruzzi/system-tests/bedrock-batch-inference
Apr 7, 2025
Merged

Add Bedrock Batch Inference Operator and accompanying parts#48468
o-nikolas merged 21 commits into
apache:mainfrom
aws-mwaa:ferruzzi/system-tests/bedrock-batch-inference

Conversation

@ferruzzi

Copy link
Copy Markdown
Contributor

Add a new operator for Bedrock Batch Inference Jobs along with all the sensor/waiter/trigger/etc support, docs, and testing.


^ 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 airflow-core/newsfragments.

@boring-cyborg boring-cyborg Bot added area:providers provider:amazon AWS/Amazon - related issues labels Mar 27, 2025
@ferruzzi
ferruzzi force-pushed the ferruzzi/system-tests/bedrock-batch-inference branch from 7315b9b to d1b033b Compare March 27, 2025 22:32
Comment thread providers/amazon/docs/operators/bedrock.rst Outdated
Comment thread providers/amazon/tests/unit/amazon/aws/operators/test_bedrock.py Outdated
Comment thread providers/amazon/src/airflow/providers/amazon/aws/operators/bedrock.py Outdated
@ferruzzi
ferruzzi force-pushed the ferruzzi/system-tests/bedrock-batch-inference branch from 0747f5f to 7ef33ce Compare April 2, 2025 21:14
@ferruzzi

ferruzzi commented Apr 2, 2025

Copy link
Copy Markdown
Contributor Author

Had to rebase for the latest batch of relocated files, sorry about that.

Comment thread providers/amazon/src/airflow/providers/amazon/aws/waiters/bedrock.json Outdated

@o-nikolas o-nikolas left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Already approved before, but lgtm!

Comment thread providers/amazon/src/airflow/providers/amazon/aws/sensors/bedrock.py Outdated
@ferruzzi

ferruzzi commented Apr 4, 2025

Copy link
Copy Markdown
Contributor Author

It was green! All I did was fix an import path! gaaah

@o-nikolas
o-nikolas merged commit 7bd1aa6 into apache:main Apr 7, 2025
@o-nikolas
o-nikolas deleted the ferruzzi/system-tests/bedrock-batch-inference branch April 7, 2025 19:47
simonprydden pushed a commit to simonprydden/airflow that referenced this pull request Apr 8, 2025
…8468)

Add a new operator for Bedrock Batch Inference Jobs along with all the sensor/waiter/trigger/etc support, docs, and testing.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:providers provider:amazon AWS/Amazon - related issues

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants