feat: implement auto_unload for ollama backend. (#38)

* feat: implement auto_unload for ollama backend.

* docs(ollama): change the documentation to the default behaviour.

Author: Davidyz

README.md

@@ -181,6 +181,8 @@ cmp_ai:setup({
   provider = 'Ollama',
   provider_options = {
     model = 'codellama:7b-code',
+    auto_unload = false, -- Set to true to automatically unload the model when
+                        -- exiting nvim.
   },
   notify = true,
   notify_callback = function(msg)

lua/cmp_ai/backends/ollama.lua

@@ -13,7 +13,14 @@ function Ollama:new(o)
       temperature = 0.2,
     },
   })
-
+  if self.params.auto_unload then
+    vim.api.nvim_create_autocmd('VimLeave', {
+      callback = function()
+        self:Get(self.params.base_url, {}, { model = self.params.model, keep_alive = 0 }, function() end)
+      end,
+      group = vim.api.nvim_create_augroup('CmpAIOllama', { clear = true }),
+    })
+  end
   return o
 end