Nvim: many small changes around whole project

.config/nvim/lua/plugins/nvim-cmp.lua

@@ -1,4 +1,7 @@
 return {
+{ 'tzachar/cmp-ai', dependencies = 'nvim-lua/plenary.nvim'},
+{ 'petertriho/cmp-git'},
+{
   'hrsh7th/nvim-cmp',
   dependencies = {
     'dcampos/cmp-snippy',
@@ -10,11 +13,71 @@ return {
     'hrsh7th/cmp-nvim-lsp-signature-help',
     'hrsh7th/cmp-path',
     'hrsh7th/cmp-calc',
+    'tzachar/cmp-ai'
   },
+
   event = "InsertEnter",
   config = function()
     local cmp = require'cmp'
     local snippy = require'snippy'
+    local cmp_ai = require('cmp_ai.config')
+
+    cmp_ai:setup({
+      max_lines = 100,
+      provider = 'Ollama',
+      provider_options = {
+        model = 'deepseek-coder:6.7b',
+        base_url = 'http://batman.local:11434/api/generate',
+        prompt = function(lines_before, lines_after)
+          -- prompt depends on the model you use. Here is an example for deepseek coder
+          return '<PRE> ' .. lines_before .. ' <SUF>' .. lines_after .. ' <MID>' -- for codellama
+        end,
+      },
+      debounce_delay = 600, -- ms llama may be GPU hungry, wait x ms after last key input, before sending request to it
+      notify = true,
+      notify_callback = function(msg)
+        vim.notify(msg)
+      end,
+      run_on_every_keystroke = true,
+      ignored_file_types = {
+        -- default is not to ignore
+        -- uncomment to ignore in lua:
+        -- lua = true
+      },
+    })
+
+    local default_cmp_sources = cmp.config.sources({
+        { name = 'nvim_lsp' },
+        { name = 'nvim_lsp_signature_help' },
+        { name = 'path' },
+        { name = 'snippy' },
+        { name = 'calc' },
+        --{ name = 'cmp_git' },
+        --{ name = 'cmp_ai' },
+    })
+
+    local bufIsBig = function(bufnr)
+      local max_filesize = 100 * 1024 -- 100 KB
+      local ok, stats = vim.loop.fs_stat(vim.api.nvim_buf_get_name(bufnr))
+      if ok and stats and stats.size > max_filesize then
+        return true
+      else
+        return false
+      end
+    end
+    -- If a file is too large, I don't want to add to it's cmp sources treesitter, see:
+    -- https://github.com/hrsh7th/nvim-cmp/issues/1522
+    vim.api.nvim_create_autocmd('BufReadPre', {
+      callback = function(t)
+        local sources = default_cmp_sources
+        if not bufIsBig(t.buf) then
+          sources[#sources+1] = {name = 'treesitter', group_index = 2}
+        end
+      cmp.setup.buffer {
+        sources = sources
+      }
+      end
+    })
 
     cmp.setup({
         snippet = {
@@ -48,8 +111,10 @@ return {
         }, {
             { name = 'buffer', keyword_length = 5, max_item_count = 10, priority = -5 },
         }),
+        sources = sources,
         sorting = {
             comparators = {
+                require('cmp_ai.compare'),
                 cmp.config.compare.offset,
                 cmp.config.compare.exact,
                 cmp.config.compare.recently_used,
@@ -89,3 +154,4 @@ return {
     require("cmp_git").setup()
   end
 }
+}