diff --git a/airflow-core/src/airflow/ui/src/queries/useTrigger.ts b/airflow-core/src/airflow/ui/src/queries/useTrigger.ts index cce546e0ad320..e033fe31a1dac 100644 --- a/airflow-core/src/airflow/ui/src/queries/useTrigger.ts +++ b/airflow-core/src/airflow/ui/src/queries/useTrigger.ts @@ -26,6 +26,7 @@ import { useDagServiceRecentDagRunsKey, UseGridServiceGridDataKeyFn, UseTaskInstanceServiceGetTaskInstancesKeyFn, + UseGridServiceGetGridRunsKeyFn, } from "openapi/queries"; import type { DagRunTriggerParams } from "src/components/TriggerDag/TriggerDAGForm"; import { toaster } from "src/components/ui"; @@ -41,6 +42,7 @@ export const useTrigger = ({ dagId, onSuccessConfirm }: { dagId: string; onSucce UseDagRunServiceGetDagRunsKeyFn({ dagId }, [{ dagId }]), UseTaskInstanceServiceGetTaskInstancesKeyFn({ dagId, dagRunId: "~" }, [{ dagId, dagRunId: "~" }]), UseGridServiceGridDataKeyFn({ dagId }, [{ dagId }]), + UseGridServiceGetGridRunsKeyFn({ dagId }, [{ dagId }]), ]; await Promise.all(queryKeys.map((key) => queryClient.invalidateQueries({ queryKey: key })));