diff --git a/providers/standard/src/airflow/providers/standard/triggers/hitl.py b/providers/standard/src/airflow/providers/standard/triggers/hitl.py index 87dfc57ab076b..e65e683b8ff93 100644 --- a/providers/standard/src/airflow/providers/standard/triggers/hitl.py +++ b/providers/standard/src/airflow/providers/standard/triggers/hitl.py @@ -132,7 +132,7 @@ async def run(self) -> AsyncIterator[TriggerEvent]: yield TriggerEvent( HITLTriggerEventSuccessPayload( chosen_options=resp.chosen_options, - params_input=resp.params_input, + params_input=resp.params_input or {}, timedout=False, ) ) diff --git a/task-sdk/src/airflow/sdk/execution_time/hitl.py b/task-sdk/src/airflow/sdk/execution_time/hitl.py index 348f14b838ab3..1a97117a76dc9 100644 --- a/task-sdk/src/airflow/sdk/execution_time/hitl.py +++ b/task-sdk/src/airflow/sdk/execution_time/hitl.py @@ -27,7 +27,7 @@ ) if TYPE_CHECKING: - from airflow.api_fastapi.execution_api.datamodels.hitl import HITLDetailResponse + from airflow.sdk.api.datamodels._generated import HITLDetailResponse def upsert_hitl_detail(