A neovim plugin for LLM-assisted programming.
Add your API keys to your env (export it in zshrc or bashrc)
{
'domgdsman/llmcopilot.nvim',
dependencies = { 'nvim-lua/plenary.nvim' },
config = function()
local system_prompt =
'You should replace the code that you are sent, only following the comments. Do not talk at all. Only output valid code. Do not provide any backticks that surround the code. Never ever output backticks like this ```. Any comment that is asking you for something should be removed after you satisfy them. Other comments should left alone. Do not output backticks'
local helpful_prompt = 'You are a helpful assistant. What I have sent are my notes so far.'
local llmcopilot = require 'llmcopilot'
local function groq_replace()
llmcopilot.invoke_llm_and_stream_into_editor({
url = 'https://api.groq.com/openai/v1/chat/completions',
model = 'llama3-70b-8192',
api_key_name = 'GROQ_API_KEY',
system_prompt = system_prompt,
replace = true,
}, llmcopilot.make_openai_spec_curl_args, llmcopilot.handle_openai_spec_data)
end
local function groq_help()
llmcopilot.invoke_llm_and_stream_into_editor({
url = 'https://api.groq.com/openai/v1/chat/completions',
model = 'llama3-70b-8192',
api_key_name = 'GROQ_API_KEY',
system_prompt = helpful_prompt,
replace = false,
}, llmcopilot.make_openai_spec_curl_args, llmcopilot.handle_openai_spec_data)
end
local function openai_replace()
llmcopilot.invoke_llm_and_stream_into_editor({
url = 'https://api.openai.com/v1/chat/completions',
model = 'gpt-4o',
temp = 0.25,
api_key_name = 'OPENAI_API_KEY',
system_prompt = system_prompt,
replace = true,
}, llmcopilot.make_openai_spec_curl_args, llmcopilot.handle_openai_spec_data)
end
local function openai_help()
llmcopilot.invoke_llm_and_stream_into_editor({
url = 'https://api.openai.com/v1/chat/completions',
model = 'gpt-4o',
temp = 0.25,
api_key_name = 'OPENAI_API_KEY',
system_prompt = helpful_prompt,
replace = false,
}, llmcopilot.make_openai_spec_curl_args, llmcopilot.handle_openai_spec_data)
end
local function anthropic_help()
llmcopilot.invoke_llm_and_stream_into_editor({
url = 'https://api.anthropic.com/v1/messages',
model = 'claude-3-5-sonnet-20240620',
api_key_name = 'ANTHROPIC_API_KEY',
system_prompt = helpful_prompt,
replace = false,
}, llmcopilot.make_anthropic_spec_curl_args, llmcopilot.handle_anthropic_spec_data)
end
local function anthropic_replace()
llmcopilot.invoke_llm_and_stream_into_editor({
url = 'https://api.anthropic.com/v1/messages',
model = 'claude-3-5-sonnet-20240620',
api_key_name = 'ANTHROPIC_API_KEY',
system_prompt = system_prompt,
replace = true,
}, llmcopilot.make_anthropic_spec_curl_args, llmcopilot.handle_anthropic_spec_data)
end
vim.keymap.set({ 'n', 'v' }, '<leader>k', groq_replace, { desc = 'llm groq' })
vim.keymap.set({ 'n', 'v' }, '<leader>K', groq_help, { desc = 'llm groq_help' })
vim.keymap.set({ 'n', 'v' }, '<leader>L', openai_help, { desc = 'llm openai_help' })
vim.keymap.set({ 'n', 'v' }, '<leader>l', openai_replace, { desc = 'llm openai' })
vim.keymap.set({ 'n', 'v' }, '<leader>I', anthropic_help, { desc = 'llm anthropic_help' })
vim.keymap.set({ 'n', 'v' }, '<leader>i', anthropic_replace, { desc = 'llm anthropic' })
end,
},