feat(nvim): set deepseek-coder-v2 as default ollama local llm

This commit is contained in:
Sergio Laín 2024-08-31 00:43:03 +02:00
parent efef17fa50
commit 6df3bd8bb6
No known key found for this signature in database
GPG key ID: 8429B2EE312F8150

View file

@ -1,14 +1,36 @@
local prefix = "<leader>a" local prefix = "<leader>a"
local user = vim.env.USER or "User" local user = vim.env.USER or "User"
vim.api.nvim_create_autocmd("User", {
pattern = "CodeCompanionChatAdapter",
callback = function(args)
if args.data.adapter == nil or vim.tbl_isempty(args.data) then
return
end
vim.g.llm_name = args.data.adapter.name
end,
})
return { return {
{ {
"olimorris/codecompanion.nvim", "olimorris/codecompanion.nvim",
cmd = { "CodeCompanion", "CodeCompanionActions", "CodeCompanionToggle", "CodeCompanionAdd", "CodeCompanionChat" }, cmd = { "CodeCompanion", "CodeCompanionActions", "CodeCompanionToggle", "CodeCompanionAdd", "CodeCompanionChat" },
opts = { opts = {
adapters = {
deepseek_coder = function()
return require("codecompanion.adapters").extend("ollama", {
name = "deepseek_coder",
schema = {
model = {
default = "deepseek-coder-v2:latest",
},
},
})
end,
},
strategies = { strategies = {
chat = { chat = {
adapter = "ollama", adapter = "deepseek_coder",
roles = { roles = {
llm = " CodeCompanion", llm = " CodeCompanion",
user = "" .. user:sub(1, 1):upper() .. user:sub(2), user = "" .. user:sub(1, 1):upper() .. user:sub(2),
@ -18,8 +40,8 @@ return {
stop = { modes = { n = "<C-c>" } }, stop = { modes = { n = "<C-c>" } },
}, },
}, },
inline = { adapter = { adapter = "ollama" } }, inline = { adapter = "deepseek_coder" },
agent = { adapter = { adapter = "ollama" } }, agent = { adapter = "deepseek_coder" },
}, },
}, },
keys = { keys = {