Skip to content

Replace private mwaaserverless-internal API with public boto3 mwaa-serverless client#58

Merged
Shnekit merged 3 commits intomainfrom
fix/remove-private-airflow-api
Apr 15, 2026
Merged

Replace private mwaaserverless-internal API with public boto3 mwaa-serverless client#58
Shnekit merged 3 commits intomainfrom
fix/remove-private-airflow-api

Conversation

@Shnekit
Copy link
Copy Markdown
Contributor

@Shnekit Shnekit commented Feb 25, 2026

Summary

Replaces the pre-GA private mwaaserverless-internal service with the public mwaa-serverless boto3 client, and updates all SageMakerNotebookOperator tasks to use the new domain_id, project_id, domain_region parameters.

Changes

Core client replacement

  • Removed custom mwaaserverless-2024-07-26.normal-edited-2.json service model file
  • Replaced create_airflow_serverless_client() with a simple boto3.client('mwaa-serverless', ...)

GitHub workflows & git-templates

  • Removed all aws configure add-model --service-name mwaaserverless-internal steps from:
    • .github/workflows/smus-bundle-deploy.yml
    • .github/workflows/smus-direct-deploy.yml
    • .github/workflows/smus-multi-env-reusable.yml
    • .github/workflows/pr-tests.yml
    • git-templates/smus-deploy-reusable.yml
    • git-templates/smus-multi-env-reusable.yml

SageMakerNotebookOperator task params

  • Added domain_id, project_id, domain_region to all SageMakerNotebookOperator tasks in:
    • examples/analytic-workflow/data-notebooks/workflows/parallel_notebooks_workflow.yaml
    • examples/analytic-workflow/ml/training/workflows/ml_training_workflow.yaml
    • examples/analytic-workflow/ml/deployment/workflows/ml_deployment_workflow.yaml
    • examples/analytic-workflow/genai/workflows/genai_dev_workflow.yaml
    • tests/integration/basic_app/code/src/basic_test_workflow.yaml
    • tests/integration/basic_app/code/src/expected_failure_workflow.yaml
  • Removed now-redundant EnvironmentVariables block from create_workflow() (domain context is now passed directly via operator params)

Integration tests & scripts

  • Updated tests/integration/base.py, tests/scripts/download_workflow_outputs_from_xcom.py, tests/test_notebook_download.py, and tests/integration/examples-analytics-workflows/ml/test_ml_deployment_workflow.py to use boto3.client('mwaa-serverless', ...)
  • Fixed hardcoded endpoints in test scripts to use environment variables

Docs

  • Updated README.md and docs/getting-started/quickstart.md examples to include the new operator params
  • Updated developer/AmazonQ.md and .kiro/steering/aws-config.md to reflect new service name

@Shnekit Shnekit requested review from abaror75 and vasu2856 February 25, 2026 16:35
@Shnekit Shnekit self-assigned this Feb 25, 2026
Shnekit added 2 commits April 15, 2026 13:34
…rverless client

- Remove custom mwaaserverless-2024-07-26.normal-edited-2.json service model
- Replace create_airflow_serverless_client() with boto3.client('mwaa-serverless', ...)
- Remove all 'aws configure add-model --service-name mwaaserverless-internal' steps
  from GitHub workflows and git-templates
- Add domain_id, project_id, domain_region to all SageMakerNotebookOperator tasks
  in workflow YAMLs (examples, tests)
- Remove now-redundant EnvironmentVariables block from create_workflow()
- Update integration test helpers and scripts to use boto3 mwaa-serverless client
- Update docs and steering files to reflect new service name
@Shnekit Shnekit force-pushed the fix/remove-private-airflow-api branch from e837bb5 to 7e5587f Compare April 15, 2026 17:35
Comment thread src/smus_cicd/helpers/airflow_serverless.py Outdated
- Removed hardcoded 'us-east-1' fallback from create_airflow_serverless_client
- Now raises ValueError if region is not provided and not in boto session
@Shnekit Shnekit merged commit 9218b0c into main Apr 15, 2026
6 checks passed
@Shnekit Shnekit deleted the fix/remove-private-airflow-api branch April 21, 2026 15:38
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.

2 participants