ringofstorms/nvim
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update gen plugin a bit

Browse source
This commit is contained in:
RingOfStorms (Joshua Bell) 2023-12-06 17:08:01 -06:00
parent a92bb733db
commit eecea2aa7b
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
lua/plugins/gen-ollama.lua
View file

@ -13,6 +13,11 @@ return {
end,
})
end,
opts = {
model = "wizard-vicuna-uncensored",
-- show_prompt = true,
show_model = true,
},
config = function(_, opts)
local g = require("gen")
g.setup(opts)
@ -28,12 +33,14 @@ return {
g.prompts = {
-- https://github.com/David-Kunz/gen.nvim/blob/main/lua/gen/prompts.lua
Prompt_Code_Completion = {
prompt = "Write code that meets these requirements: $input\nOnly output the result in format ```$filetype\n...\n```",
prompt =
"Write code that meets these requirements: $input\nOnly output the result in format ```$filetype\n...\n```",
replace = true,
extract = "```$filetype\n(.-)```",
},
Replace_Code_Completion = {
prompt = "Rewrite the following code, follow any comment instructions.\nOnly output the result in format ```$filetype\n...\n```:\n```$filetype\n$text\n```",
prompt =
"Rewrite the following code, follow any comment instructions.\nOnly output the result in format ```$filetype\n...\n```:\n```$filetype\n$text\n```",
replace = true,
extract = "```$filetype\n(.-)```",
},
@ -71,7 +78,7 @@ return {
mode = { "n", "v", "x" },
},
{
"<leader>xcc",
"<leader>xi",
"<cmd>Gen Prompt_And_Answer_Inline<cr>",
desc = "Prompt and answer inline at cursor",
mode = { "n", "v", "x" },