Add ollama prompts

1 files changed, 43 insertions, 0 deletions

diff --git a/nvim/init.vim b/nvim/init.vim
index 28290e2..1b7fb4c 100644
--- a/nvim/init.vim
+++ b/nvim/init.vim
@@ -407,6 +407,49 @@ lua << EOF
     }
 EOF
 
+lua << EOF
+-- ollama.nvim integration
+do
+  local function ollama_loaded()
+    return package.loaded.ollama and require("ollama").status ~= nil
+  end
+  local ollama_status = {
+    IDLE = "🦙",
+    WORKING = "🤖"
+  }
+  local function get_status_icon()
+    return ollama_status[require("ollama").status()]
+  end
+  require("lualine").setup({sections = {lualine_x = {get_status_icon, ollama_loaded}}})
+  require("ollama").config.prompts.fgen = {
+    prompt = [[
+Generate $ftype code that follows this code, keep any comments:
+
+$sel
+
+Respond EXACTLY in this format:
+```$ftype
+<your code>
+```
+]],
+    action = "insert"
+  }
+  require("ollama").config.prompts.extend= {
+    prompt = [[
+Change this $ftype code to $input, keep any comments:
+
+$sel
+
+Respond EXACTLY in this format:
+```$ftype
+<your code>
+```
+]],
+    action = "display_replace"
+  }
+end
+
+EOF
 
 " ollama.nvim - ai code assist