Export keymaps.accept_suggestion and dismiss_suggestion for remapping

lua/tabnine/completion.lua

@@ -13,6 +13,9 @@ local function valid_response(response)
 end
 
 function M.accept()
+	if not state.completions_cache then
+		return
+	end
 	M.clear()
 	local lines = utils.str_to_lines(state.rendered_completion)
 
@@ -31,10 +34,14 @@ function M.accept()
 end
 
 function M.clear()
+	if not state.completions_cache then
+		return
+	end
 	if state.cancel_completion then
 		state.cancel_completion()
 	end
 	state.debounce_timer:stop()
+
 	api.nvim_buf_clear_namespace(0, consts.tabnine_namespace, 0, -1)
 end
 

lua/tabnine/keymaps.lua

@@ -3,25 +3,21 @@ local state = require("tabnine.state")
 local config = require("tabnine.config")
 local completion = require("tabnine.completion")
 
+function M.accept_suggestion()
+	vim.schedule(completion.accept)
+end
+function M.dismiss_suggestion()
+	vim.schedule(function()
+		completion.clear()
+		state.completions_cache = nil
+	end)
+end
+
 function M.setup()
 	local accept_keymap = config.get_config().accept_keymap
 	local dismiss_keymap = config.get_config().dismiss_keymap
-	vim.keymap.set("i", accept_keymap, function()
-		if not state.completions_cache then
-			return accept_keymap
-		end
-		vim.schedule(completion.accept)
-	end, { expr = true })
-
-	vim.keymap.set("i", dismiss_keymap, function()
-		if not state.completions_cache then
-			return dismiss_keymap
-		end
-		vim.schedule(function()
-			completion.clear()
-			state.completions_cache = nil
-		end)
-	end, { expr = true })
+	vim.keymap.set("i", accept_keymap, M.accept_suggestion, { expr = true })
+	vim.keymap.set("i", dismiss_keymap, M.dismiss_suggestion, { expr = true })
 end
 
 return M