--[
-- A basic Model completion with ollama.
--]
return {
	"gsuuon/model.nvim",
	cmd = { "M", "Model", "Mchat" },
	init = function()
		vim.filetype.add({
			extension = {
				mchat = "mchat",
			},
		})
	end,
	ft = "mchat",

	keys = {
		{ "<C-m>d", ":Mdelete<cr>", mode = "n" },
		{ "<C-m>s", ":Mselect<cr>", mode = "n" },
		{ "<C-m><space>", ":Mchat<cr>", mode = "n" },
	},

	config = function()
		local model = require("model")
		local ollama = require("mchat.providers.ollama")

		model.setup({
			prompts = {
				["ollama:starling"] = {
					provider = ollama,
					params = {
						model = "starling-lm",
					},
					builder = function(input)
						return {
							prompt = "GPT4 Correct User: " .. input .. "<|end_of_turn|>GPT4 Correct Assistant: ",
						}
					end,
				},
			},
		})
	end,
}