From 4ca57f4cc05e560dee82e3b5d478ca65eb885fb8 Mon Sep 17 00:00:00 2001
From: Matt Hayden <1509079+matt-hayden@users.noreply.github.com>
Date: Wed, 4 Dec 2024 14:43:01 -0700
Subject: [PATCH] User can configure modal accept and dismiss keys

---
 lua/llm/config.lua  |  1 +
 lua/llm/keymaps.lua | 10 +++-------
 2 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/lua/llm/config.lua b/lua/llm/config.lua
index 36d4e19..9b0f0ed 100644
--- a/lua/llm/config.lua
+++ b/lua/llm/config.lua
@@ -21,6 +21,7 @@ local default_config = {
     suffix = "<fim_suffix>",
   },
   debounce_ms = 150,
+  keymap_modes = {"i", "n"},
   accept_keymap = "<Tab>",
   dismiss_keymap = "<S-Tab>",
   tls_skip_verify_insecure = false,
diff --git a/lua/llm/keymaps.lua b/lua/llm/keymaps.lua
index 1356aed..490f611 100644
--- a/lua/llm/keymaps.lua
+++ b/lua/llm/keymaps.lua
@@ -27,16 +27,12 @@ function M.setup()
     return
   end
 
+  local keymap_modes = config.get().keymap_modes
   local accept_keymap = config.get().accept_keymap
   local dismiss_keymap = config.get().dismiss_keymap
 
-  vim.keymap.set("i", accept_keymap, accept_suggestion, { expr = true })
-
-  vim.keymap.set("n", accept_keymap, accept_suggestion, { expr = true })
-
-  vim.keymap.set("i", dismiss_keymap, dismiss_suggestion, { expr = true })
-
-  vim.keymap.set("n", dismiss_keymap, dismiss_suggestion, { expr = true })
+  vim.keymap.set(keymap_modes, accept_keymap, accept_suggestion, { expr = true })
+  vim.keymap.set(keymap_modes, dismiss_keymap, dismiss_suggestion, { expr = true })
 
   M.setup_done = true
 end