ollama setup for neovim

dot_vim/lua/avante_setup.lua

@@ -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,
+		},
+	},
+})

dot_vim/lua/ollama_setup.lua

@@ -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 })

dot_vim/nvim.lua

@@ -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")