diff --git a/assets/keymaps/vim.json b/assets/keymaps/vim.json
index d0cf4621a59d8614022f2a4ba176a79b40f8d331..78bd01b1ef9edeac1ebea4588f165987a5c04942 100644
--- a/assets/keymaps/vim.json
+++ b/assets/keymaps/vim.json
@@ -579,6 +579,13 @@
       "shift-u": "git::UnstageAndNext" // "d shift-u"
     }
   },
+  {
+    "context": "VimControl && (AgentDiff || editor_agent_diff)",
+    "bindings": {
+      "d p": "agent::Reject",
+      "d u": "agent::Keep"
+    }
+  },
   {
     "context": "vim_operator == gu",
     "bindings": {