dotfiles/nvim/lua/plugins/nvim-cmp.lua

return {
  { -- Autocompletion
    'hrsh7th/nvim-cmp',
    event = 'InsertEnter',
    dependencies = {
      -- Snippet Engine & its associated nvim-cmp source
      {
        'L3MON4D3/LuaSnip',
        build = (function()
          -- Build Step is needed for regex support in snippets.
          -- This step is not supported in many windows environments.
          -- Remove the below condition to re-enable on windows.
          if vim.fn.has 'win32' == 1 or vim.fn.executable 'make' == 0 then
            return
          end
          return 'make install_jsregexp'
        end)(),
        dependencies = {
          -- `friendly-snippets` contains a variety of premade snippets.
          --    See the README about individual language/framework/plugin snippets:
          --    https://github.com/rafamadriz/friendly-snippets
          -- {
          --   'rafamadriz/friendly-snippets',
          --   config = function()
          --     require('luasnip.loaders.from_vscode').lazy_load()
          --   end,
          -- },
        },
      },
      'saadparwaiz1/cmp_luasnip',

      -- Adds other completion capabilities.
      --  nvim-cmp does not ship with all sources by default. They are split
      --  into multiple repos for maintenance purposes.
      'hrsh7th/cmp-nvim-lsp',
      'hrsh7th/cmp-path',
      'hrsh7th/cmp-nvim-lsp-signature-help',

      { dir = '~/code/cmp-ai' },
    },
    config = function()
      -- See `:help cmp`
      local cmp = require 'cmp'
      local types = require 'cmp.types'
      local luasnip = require 'luasnip'
      luasnip.config.setup {}

      local cmp_ai = require 'cmp_ai.config'
      cmp_ai:setup {
        provider_options = {
          model = 'codellama:13b-code',
          -- model = 'codellama:7b-code',
          -- model = 'qwen2.5-coder:7b-base',
          -- model = 'starcoder2:7b',
          -- model = 'codestral:22b',
          -- model = 'codegemma:7b-code',
          base_url = os.getenv 'LLAMA_API_BASE' or 'http://localhost:8080/ollama',
          api_key = os.getenv 'LLAMA_API_KEY',
          options = {
            stop = { '\n\n', '<EOT>' }, -- codellama
            -- stop = { '<|endoftext|>' }, -- qwen-coder
            -- stop = { '<file_sep>' }, -- starcoder2
            -- stop = { '[INST]', '[/INST]', '[PREFIX]', '[MIDDLE]', '[SUFFIX]' }, -- codeestral
            -- stop = { '<|fim_prefix|>', '<|fim_suffix|>', '<|fim_middle|>', '<|file_separator|>' }, -- codegemma
          },
          prompt = function(lines_before, lines_after)
            return lines_before
            -- return '<PRE> ' .. lines_before .. ' <SUF>' .. lines_after .. ' <MID>'
            -- return '<fim_prefix>' .. lines_before .. '<fim_suffix>' .. lines_after .. '<fim_middle>'
            -- return '<|fim_prefix|>' .. lines_before .. '<|fim_suffix|>' .. lines_after .. '<|fim_middle|>'
          end,
          suffix = function(lines_after)
            return lines_after
          end,
        },
        max_lines = 500,
        run_on_every_keystroke = false,
        notify = true,
      }
      vim.api.nvim_set_hl(0, 'CmpItemKindOllama', { fg = '#6cc644' })

      cmp.setup {
        snippet = {
          expand = function(args)
            luasnip.lsp_expand(args.body)
          end,
        },
        completion = { completeopt = 'menu,menuone,noinsert' },

        -- For an understanding of why these mappings were
        -- chosen, you will need to read `:help ins-completion`
        --
        -- No, but seriously. Please read `:help ins-completion`, it is really good!
        mapping = cmp.mapping.preset.insert {
          -- Disable use of <Up> and <Down> for navigation within a completion list.
          ['<Down>'] = cmp.mapping.close(),
          ['<Up>'] = cmp.mapping.close(),

          -- Identical to old <Down> key behavior
          ['<C-j>'] = cmp.mapping.select_next_item { behavior = types.cmp.SelectBehavior.Select },
          -- Identical to old <Up> key behavior
          ['<C-k>'] = cmp.mapping.select_prev_item { behavior = types.cmp.SelectBehavior.Select },

          -- Select the [n]ext item
          ['<C-n>'] = cmp.mapping.select_next_item(),
          -- Select the [p]revious item
          ['<C-p>'] = cmp.mapping.select_prev_item(),

          -- Scroll the documentation window [b]ack / [f]orward
          ['<C-b>'] = cmp.mapping.scroll_docs(-4),
          ['<C-f>'] = cmp.mapping.scroll_docs(4),

          -- Accept ([y]es) the completion.
          --  This will auto-import if your LSP supports it.
          --  This will expand snippets if the LSP sent a snippet.
          ['<C-y>'] = cmp.mapping.confirm { select = true },

          -- Cancel/abort any selected/partial completion and close the completion list.
          ['<C-c>'] = cmp.mapping.abort(),

          -- If you prefer more traditional completion keymaps,
          -- you can uncomment the following lines
          --['<CR>'] = cmp.mapping.confirm { select = true },
          --['<Tab>'] = cmp.mapping.select_next_item(),
          --['<S-Tab>'] = cmp.mapping.select_prev_item(),

          -- Manually trigger a completion from nvim-cmp.
          --  Generally you don't need this, because nvim-cmp will display
          --  completions whenever it has completion options available.
          ['<C-Space>'] = cmp.mapping.complete {},

          -- Think of <c-l> as moving to the right of your snippet expansion.
          --  So if you have a snippet that's like:
          --  function $name($args)
          --    $body
          --  end
          --
          -- <c-l> will move you to the right of each of the expansion locations.
          -- <c-h> is similar, except moving you backwards.
          ['<C-l>'] = cmp.mapping(function()
            if luasnip.expand_or_locally_jumpable() then
              luasnip.expand_or_jump()
            end
          end, { 'i', 's' }),
          ['<C-h>'] = cmp.mapping(function()
            if luasnip.locally_jumpable(-1) then
              luasnip.jump(-1)
            end
          end, { 'i', 's' }),

          ['<C-x>'] = cmp.mapping.complete {
            config = {
              sources = cmp.config.sources {
                { name = 'cmp_ai' },
              },
            },
          },

          -- For more advanced Luasnip keymaps (e.g. selecting choice nodes, expansion) see:
          --    https://github.com/L3MON4D3/LuaSnip?tab=readme-ov-file#keymaps
        },
        sources = {
          {
            name = 'lazydev',
            -- set group index to 0 to skip loading LuaLS completions as lazydev recommends it
            group_index = 0,
          },
          {
            name = 'nvim_lsp',
            entry_filter = function(entry, ctx)
              -- do not show snippets in the completion list
              return require('cmp').lsp.CompletionItemKind.Snippet ~= entry:get_kind()
            end,
          },
          { name = 'luasnip' },
          { name = 'path' },
          { name = 'nvim_lsp_signature_help' },
        },
      }
    end,
  },
}

-- vim: ts=2 sts=2 sw=2 et