ollama setup for neovim

This commit is contained in:
Barak Michener 2024-09-21 13:42:04 -07:00
parent e49b11da27
commit 88ab0c75e4
4 changed files with 58 additions and 12 deletions

View file

@ -0,0 +1,37 @@
require("avante_lib").load()
require("avante").setup({
provider = "ollama",
auto_suggestions_provider = "ollama",
behavior = {
auto_suggestions = false,
},
hints = {
enabled = false,
},
vendors = {
---@type AvanteProvider
ollama = {
["local"] = true,
endpoint = "127.0.0.1:11434/v1",
model = "qwen2.5-coder:7b",
parse_curl_args = function(opts, code_opts)
return {
url = opts.endpoint .. "/chat/completions",
headers = {
["Accept"] = "application/json",
["Content-Type"] = "application/json",
},
body = {
model = opts.model,
messages = require("avante.providers").copilot.parse_message(code_opts), -- you can make your own message, but this is very advanced
max_tokens = 2048,
stream = true,
},
}
end,
parse_response_data = function(data_stream, event_state, opts)
require("avante.providers").openai.parse_response(data_stream, event_state, opts)
end,
},
},
})

View file

@ -1,7 +1,10 @@
require("ollama").setup({
model = "codellama",
url = "http://ollama.ollama.svc.k8s.barakmich.com",
serve = {
on_start = false,
},
model = "codellama:34b",
url = "http://daystrom.home.0b100.net:11434",
serve = {
on_start = false,
},
})
vim.keymap.set("n", "<Leader>,,", ":<c-u>lua require('ollama').prompt()<cr>", { silent = true })
vim.keymap.set("v", "<Leader>,,", ":<c-u>lua require('ollama').prompt()<cr>", { silent = true })

View file

@ -624,6 +624,8 @@ end
require("hoversplit").setup({})
--require("murdock")
require("ollama_setup")
--require("ollama_setup")
--require("gen_setup")
require("lir_setup")
require("lualine_setup")
require("avante_setup")

View file

@ -233,7 +233,6 @@ Plug 'nvim-lua/popup.nvim'
Plug 'nvim-lua/plenary.nvim'
Plug 'nvim-telescope/telescope.nvim'
Plug 'sindrets/diffview.nvim'
Plug 'nomnivore/ollama.nvim'
Plug 'kyazdani42/nvim-web-devicons' " for file icons
"Plug 'kyazdani42/nvim-tree.lua'
@ -275,7 +274,12 @@ Plug 'Shougo/echodoc.vim'
"Plug 'python-rope/ropevim'
Plug 'windwp/nvim-autopairs'
Plug 'danymat/neogen'
"Plug 'David-Kunz/gen.nvim'
"Plug 'nomnivore/ollama.nvim'
Plug 'stevearc/dressing.nvim'
Plug 'MunifTanjim/nui.nvim'
Plug 'zbirenbaum/copilot.lua'
Plug 'yetone/avante.nvim', {'branch': 'main', 'do': 'make'}
" Unclear
"Plug 'ncm2/float-preview.nvim'
@ -1046,7 +1050,7 @@ call denite#custom#var('grep', {
\ })
"nnoremap <silent> <Leader>g :<C-u>Denite -buffer-name=search-buffer grep<CR>
"nnoremap <silent> <Leader>aa :<C-u>RipGrep!<CR>
nnoremap <silent> <Leader>aa :<C-u>RgRoot!<CR>
"nnoremap <silent> <Leader>aa :<C-u>RgRoot!<CR>
"nnoremap <silent> <Leader>aw :<C-u>RgRoot! <C-r><C-w><CR>
"nnoremap <C-l> :<C-u>Files! `git rev-parse --show-toplevel`<CR>
"nnoremap <C-k> :<C-u>GFiles<CR>
@ -1118,7 +1122,7 @@ map <F10> command! -nargs=0 Syn call Syn()
hi link @type.rust TSCSecondaryTypeParam
let s:gptool_temp = expand($HOME . '/src/gptool/bindings.vim')
if filereadable(s:gptool_temp)
exec ':so ' . s:gptool_temp
endif
"let s:gptool_temp = expand($HOME . '/src/gptool/bindings.vim')
"if filereadable(s:gptool_temp)
"exec ':so ' . s:gptool_temp
"endif