From 87109cc8dabf5848e10c4436bd4173bee9f696a1 Mon Sep 17 00:00:00 2001
From: Daniel Standish <15932138+dstandish@users.noreply.github.com>
Date: Wed, 17 Jul 2024 11:41:08 -0700
Subject: [PATCH] Add try_number and map_index as params for log event endpoint

---
 airflow/api_connexion/endpoints/event_log_endpoint.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/airflow/api_connexion/endpoints/event_log_endpoint.py b/airflow/api_connexion/endpoints/event_log_endpoint.py
index 3b3dbe6efd490..ef55ad145c755 100644
--- a/airflow/api_connexion/endpoints/event_log_endpoint.py
+++ b/airflow/api_connexion/endpoints/event_log_endpoint.py
@@ -58,6 +58,8 @@ def get_event_logs(
     dag_id: str | None = None,
     task_id: str | None = None,
     run_id: str | None = None,
+    map_index: int | None = None,
+    try_number: int | None = None,
     owner: str | None = None,
     event: str | None = None,
     excluded_events: str | None = None,
@@ -90,6 +92,10 @@ def get_event_logs(
         query = query.where(Log.task_id == task_id)
     if run_id:
         query = query.where(Log.run_id == run_id)
+    if map_index:
+        query = query.where(Log.map_index == map_index)
+    if try_number:
+        query = query.where(Log.try_number == try_number)
     if owner:
         query = query.where(Log.owner == owner)
     if event: