diff --git a/lua/lazyvim/plugins/extras/coding/neogen.lua b/lua/lazyvim/plugins/extras/coding/neogen.lua index c8916676..856b7e9a 100644 --- a/lua/lazyvim/plugins/extras/coding/neogen.lua +++ b/lua/lazyvim/plugins/extras/coding/neogen.lua @@ -5,11 +5,37 @@ return { keys = { { "cn", - function() - require("neogen").generate() - end, + "", desc = "Generate Annotations (Neogen)", }, + { + "cnf", + function() + require("neogen").generate({ type = "func" }) + end, + desc = "Generate func Annotations (Neogen)", + }, + { + "cnc", + function() + require("neogen").generate({ type = "class" }) + end, + desc = "Generate class Annotations (Neogen)", + }, + { + "cnt", + function() + require("neogen").generate({ type = "type" }) + end, + desc = "Generate type Annotations (Neogen)", + }, + { + "cne", + function() + require("neogen").generate({ type = "file" }) + end, + desc = "Generate file Annotations (Neogen)", + }, }, opts = function(_, opts) if opts.snippet_engine ~= nil then