This is an automated email from the ASF dual-hosted git repository.
kaxilnaik pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
from 088242a911b Update Example URL in YDB docs (#45033)
add 4de24a18d0b AIP-72: Allow retrieving Connection from Task Context
(#45043)
No new revisions were added by this update.
Summary of changes:
task_sdk/src/airflow/sdk/__init__.py | 3 +
task_sdk/src/airflow/sdk/api/client.py | 17 +++-
.../src/airflow/sdk/definitions}/connection.py | 39 +++++---
task_sdk/src/airflow/sdk/exceptions.py | 21 +++++
task_sdk/src/airflow/sdk/execution_time/comms.py | 16 +++-
task_sdk/src/airflow/sdk/execution_time/context.py | 78 ++++++++++++++++
.../src/airflow/sdk/execution_time/supervisor.py | 9 +-
.../src/airflow/sdk/execution_time/task_runner.py | 13 ++-
task_sdk/tests/execution_time/test_context.py | 103 +++++++++++++++++++++
task_sdk/tests/execution_time/test_supervisor.py | 2 +-
task_sdk/tests/execution_time/test_task_runner.py | 68 +++++++++++++-
11 files changed, 349 insertions(+), 20 deletions(-)
copy {airflow/api_fastapi/execution_api/datamodels =>
task_sdk/src/airflow/sdk/definitions}/connection.py (52%)
create mode 100644 task_sdk/src/airflow/sdk/execution_time/context.py
create mode 100644 task_sdk/tests/execution_time/test_context.py
