CodeCompanion test
This commit is contained in:
40
lua/config/codecompanion.lua
Normal file
40
lua/config/codecompanion.lua
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
require("codecompanion").setup({
|
||||||
|
interactions = {
|
||||||
|
chat = {
|
||||||
|
adapter = "auraid",
|
||||||
|
opts = {
|
||||||
|
completion_provider = "cmp",
|
||||||
|
context_management = { enabled = false },
|
||||||
|
},
|
||||||
|
},
|
||||||
|
inline = { adapter = "auraid" },
|
||||||
|
},
|
||||||
|
adapters = {
|
||||||
|
http = {
|
||||||
|
auraid = function()
|
||||||
|
return require("codecompanion.adapters").extend("ollama", {
|
||||||
|
env = {
|
||||||
|
url = "http://localhost:11434",
|
||||||
|
},
|
||||||
|
headers = {
|
||||||
|
["Content-Type"] = "application/json",
|
||||||
|
},
|
||||||
|
schema = {
|
||||||
|
model = { default = "qwen36-vision:latest" },
|
||||||
|
temperature = { default = 0.7 },
|
||||||
|
think = { default = false },
|
||||||
|
max_tokens = { default = 268288 },
|
||||||
|
},
|
||||||
|
})
|
||||||
|
end,
|
||||||
|
opts = {
|
||||||
|
show_model_choices = false,
|
||||||
|
show_presets = false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
opts = {
|
||||||
|
log_level = "TRACE",
|
||||||
|
send_code = true,
|
||||||
|
},
|
||||||
|
})
|
||||||
@ -4,7 +4,7 @@ require'nvim-treesitter'.setup {
|
|||||||
disable = function(lang, buf)
|
disable = function(lang, buf)
|
||||||
-- Проверяем, существует ли буфер и связан ли он с файлом
|
-- Проверяем, существует ли буфер и связан ли он с файлом
|
||||||
if not buf or buf == 0 then
|
if not buf or buf == 0 then
|
||||||
return true
|
--return true
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Получаем информацию о файле
|
-- Получаем информацию о файле
|
||||||
|
|||||||
@ -160,43 +160,15 @@ return {
|
|||||||
end,
|
end,
|
||||||
},
|
},
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
{
|
{
|
||||||
'gsuuon/model.nvim',
|
"olimorris/codecompanion.nvim",
|
||||||
config = function()
|
opts = {},
|
||||||
local bash_provider = {
|
dependencies = {
|
||||||
request_completion = function(handlers, params, options)
|
"nvim-lua/plenary.nvim",
|
||||||
local input = params.input
|
"nvim-treesitter/nvim-treesitter",
|
||||||
local command = "~/test '" .. input .. "'"
|
},
|
||||||
local handle = io.popen(command)
|
config = function()
|
||||||
local result = handle:read("*a")
|
require("config.codecompanion")
|
||||||
handle:close()
|
end,
|
||||||
|
},
|
||||||
local response = vim.json.decode(result)
|
|
||||||
handlers.on_partial(response.message.content)
|
|
||||||
handlers.on_finish()
|
|
||||||
end
|
|
||||||
}
|
|
||||||
|
|
||||||
require('model').setup({
|
|
||||||
chats = {
|
|
||||||
mybash = {
|
|
||||||
provider = bash_provider,
|
|
||||||
create = function(input, context)
|
|
||||||
return {
|
|
||||||
messages = {
|
|
||||||
{ role = 'user', content = input }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
end,
|
|
||||||
run = function(messages, config)
|
|
||||||
local last_message = messages[#messages]
|
|
||||||
return { input = last_message.content }
|
|
||||||
end
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
end
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user