diff --git a/task-sdk/src/airflow/sdk/execution_time/supervisor.py b/task-sdk/src/airflow/sdk/execution_time/supervisor.py index 01e3094cb5929..55110da616705 100644 --- a/task-sdk/src/airflow/sdk/execution_time/supervisor.py +++ b/task-sdk/src/airflow/sdk/execution_time/supervisor.py @@ -1896,7 +1896,13 @@ def supervise( exit_code = process.wait() end = time.monotonic() - log.info("Task finished", exit_code=exit_code, duration=end - start, final_state=process.final_state) + log.info( + "Task finished", + task_instance_id=str(ti.id), + exit_code=exit_code, + duration=end - start, + final_state=process.final_state, + ) if log_path and log_file_descriptor: log_file_descriptor.close() return exit_code diff --git a/task-sdk/tests/task_sdk/execution_time/test_supervisor.py b/task-sdk/tests/task_sdk/execution_time/test_supervisor.py index 8a5e06e0e8445..107cf46e49e97 100644 --- a/task-sdk/tests/task_sdk/execution_time/test_supervisor.py +++ b/task-sdk/tests/task_sdk/execution_time/test_supervisor.py @@ -667,6 +667,7 @@ def test_supervise_handles_deferred_task( "level": "info", "logger": "supervisor", "loc": mocker.ANY, + "task_instance_id": str(ti.id), } in captured_logs def test_supervisor_handles_already_running_task(self):