From 08f2906f823761a6afc6a7a34810c34775b383a8 Mon Sep 17 00:00:00 2001
From: Aditya <bluenerd@protonmail.com>
Date: Sun, 10 Mar 2024 13:03:17 +0530
Subject: [PATCH] add ollama

---
 config/default.nix       |  2 ++
 config/ollama/ollama.nix | 40 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 config/ollama/ollama.nix

diff --git a/config/default.nix b/config/default.nix
index 076e050..36c4bf1 100644
--- a/config/default.nix
+++ b/config/default.nix
@@ -36,6 +36,8 @@
 
     ./none-ls/none-ls.nix
 
+    ./ollama/ollama.nix
+
     ./sets.nix
 
     ./snippets/luasnip.nix
diff --git a/config/ollama/ollama.nix b/config/ollama/ollama.nix
new file mode 100644
index 0000000..93b9fbd
--- /dev/null
+++ b/config/ollama/ollama.nix
@@ -0,0 +1,40 @@
+{
+  plugins.ollama = {
+    enable = true;
+    model = "tinyllama";
+    url = "http://127.0.0.11435";
+  };
+
+  keymaps = [
+    {
+      key = "<leader>o";
+      action = "<cmd>Ollama<cr>";
+      options = {
+        desc = "Ask Ollama";
+      };
+    }
+    {
+      mode = "v";
+      key = "<leader>oe";
+      action = "<cmd>Ollama Explain_Code<cr>";
+      options = {
+        desc = "Explain code snippet";
+      };
+    }
+    {
+      key = "<leader>oo";
+      action = "<cmd>Ollama Raw<cr>";
+      options = {
+        desc = "Ask Ollama";
+      };
+    }
+    {
+      mode = "v";
+      key = "<leader>os";
+      action = "<cmd>Ollama Simplify_Code<cr>";
+      options = {
+        desc = "Simplify code snippet";
+      };
+    }
+  ];
+}