[dags] update key mapping