@task def push_task(**context): context['task_instance'].xcom_push(key='record_count', value=100)
To avoid database overload for large payloads (e.g., DataFrames), you can configure a custom backend: airflow xcom
process(extract())