diff --git a/airflow_monitor/client.py b/airflow_monitor/client.py index d19c7e8..5ffd20d 100644 --- a/airflow_monitor/client.py +++ b/airflow_monitor/client.py @@ -204,14 +204,14 @@ class AirflowClient: resp = self._request( "GET", f"/dags/{encoded_dag_id}/dagRuns", - params=[("limit", 5), ("order_by", "-start_date")] + params=[("limit", 1), ("order_by", "-start_date")] + [("state", s) for s in states], ) else: resp = self._request( "GET", f"/dags/{encoded_dag_id}/dagRuns", - params={"order_by": "-start_date", "limit": 5}, + params={"order_by": "-start_date", "limit": 1}, ) runs = resp.json().get("dag_runs", []) else: @@ -282,7 +282,7 @@ class AirflowClient: self._request( "POST", f"/dags/{encoded_dag_id}/dagRuns/{encoded_run_id}/clear", - json={"dry_run": False, "only_failed": True}, + json={"dry_run": False}, ) logger.info("Cleared DAG run %s/%s for retry", dag_id, dag_run_id) return True