From 4118c52c12eb954f47ecd594580d7d3573c6862c Mon Sep 17 00:00:00 2001
From: Hawtian Wang <twistoy.wang@gmail.com>
Date: Wed, 17 Apr 2024 18:11:51 +0800
Subject: [PATCH] ...

---
 lua/dotvim/bootstrap.lua                      |  2 +-
 .../pkgs/extra/misc/{copilot.lua => ai.lua}   | 19 ++++++++++++++++++-
 2 files changed, 19 insertions(+), 2 deletions(-)
 rename lua/dotvim/pkgs/extra/misc/{copilot.lua => ai.lua} (88%)

diff --git a/lua/dotvim/bootstrap.lua b/lua/dotvim/bootstrap.lua
index 19a2beab..9e052839 100644
--- a/lua/dotvim/bootstrap.lua
+++ b/lua/dotvim/bootstrap.lua
@@ -48,7 +48,7 @@ local enabled_packages = {
   "extra.languages.yaml",
   "extra.languages.toml",
   "extra.misc.competitive-programming",
-  "extra.misc.copilot",
+  "extra.misc.ai",
   "extra.misc.rime",
 }
 
diff --git a/lua/dotvim/pkgs/extra/misc/copilot.lua b/lua/dotvim/pkgs/extra/misc/ai.lua
similarity index 88%
rename from lua/dotvim/pkgs/extra/misc/copilot.lua
rename to lua/dotvim/pkgs/extra/misc/ai.lua
index c0dfe7a6..4f394bbf 100644
--- a/lua/dotvim/pkgs/extra/misc/copilot.lua
+++ b/lua/dotvim/pkgs/extra/misc/ai.lua
@@ -2,7 +2,7 @@ local _copilot_setup_done = false
 
 ---@type dotvim.core.package.PackageOption
 return {
-  name = "extra.misc.copilot",
+  name = "extra.misc.ai",
   deps = {
     "coding",
     "editor",
@@ -218,5 +218,22 @@ return {
         table.insert(opts.sections.lualine_y, 1, component)
       end,
     },
+    {
+      "David-Kunz/gen.nvim",
+      cmd = "Gen",
+      opts = {
+        model = "mistral", -- The default model to use.
+        host = "localhost", -- The host running the Ollama service.
+        port = "11434", -- The port on which the Ollama service is listening.
+        quit_map = "q",
+        retry_map = "<c-r>",
+        init = function() end, -- do nothing!
+        display_mode = "float", -- The display mode. Can be "float" or "split".
+        show_prompt = false, -- Shows the prompt submitted to Ollama.
+        show_model = false, -- Displays which model you are using at the beginning of your chat session.
+        no_auto_close = false, -- Never closes the window automatically.
+        debug = false, -- Prints errors and the command which is run.
+      },
+    },
   },
 }