require('neoai').setup({
    models = {
        {
            name = "openai",
            model = "gpt-3.5-turbo-1106", -- "gpt-4-1106-preview",
            params = nil,
        },
    },
    open_ai = {
        api_key = {
            get = function()
                local key = vim.fn.system("pass show openai/api_key")
                key = string.gsub(key, "\n", "")
                return key
            end,
        },
    },
    prompts = {
        context_prompt = function(context)
            return context
        end,
    },
    shortcuts = {
        {
            name = "textify",
            key = "<leader>as",
            desc = "fix text with AI",
            use_context = true,
            prompt = [[
                Please rewrite the text to make it more readable, clear,
                concise, and fix any grammatical, punctuation, or spelling
                errors
            ]],
            modes = { "v" },
            strip_function = nil,
        },
        {
            name = "gitcommit",
            key = "<leader>ag",
            desc = "generate git commit message",
            use_context = false,
            prompt = function()
                return [[
                    With specifications from conventionalcommits.org,
                    the commit message should be formatted as follows:
                    <type>[optional scope]: <description>

                    [optional body]

                    [optional footer(s)]
                    Here are some examples of commit messages:
                    =========================================
                    feat: holiday-themed devcard

                    Our DevCard now features a holiday theme for both Halloween and Christmas!
                    This also includes a link pointing to Chris' article on how to embed it on your GitHub Profile.

                    DD-267 #done
                    =========================================
                    build(extension): version 3.8.0
                    =========================================
                    chore!: drop Node 6 from testing matrix

                    BREAKING CHANGE: dropping Node 6 which hits end of life in April
                    =========================================
                    ci: use pulumi common new kubernetes helpers
                    =========================================
                    Using the following git diff in order to generate
                    a consise and clear git commit message:
                    ]] .. vim.fn.system("git diff --cached --ignore-space-at-eol")
            end,
            modes = { "n" },
            strip_function = nil,
        },
    },
})

-- Keybindings
vim.keymap.set("n", "<leader>at", "<cmd>NeoAIToggle<cr>",
  {silent = true, noremap = true}
)
vim.keymap.set("n", "<leader>ai", ":NeoAIInject YOUR MAY ONLY PROVIDE ONLY THE CODE SNIPPET AND WITHOUT THE CODE BLOCK",
  {silent = true, noremap = true}
)
vim.keymap.set("v", "<leader>ac", "<cmd>'<,'>NeoAIContext<cr>iPlease explain shortly",
  {silent = true, noremap = true}
)
vim.keymap.set("n", "<leader>ap", "<cmd>put c<cr><cmd>let @c=''<cr>",
  {silent = true, noremap = true}
)