From 94c85d4d33533eb9b92a23beb4e5f3169340c260 Mon Sep 17 00:00:00 2001
From: S4NKALP <sankalptharu50028@gmail.com>
Date: Thu, 4 Apr 2024 22:43:13 +0545
Subject: [PATCH] .

---
 config/nvim/lazyvim.json                      |  1 +
 config/nvim/lua/config/keymaps.lua            |  1 +
 .../nvim/lua/plugins/extras/coding/ai/gen.lua | 84 +++++++++++++------
 .../plugins/extras/lang/python-extended.lua   | 26 +++++-
 4 files changed, 86 insertions(+), 26 deletions(-)

diff --git a/config/nvim/lazyvim.json b/config/nvim/lazyvim.json
index 892366b9..585d4246 100644
--- a/config/nvim/lazyvim.json
+++ b/config/nvim/lazyvim.json
@@ -3,6 +3,7 @@
     "lazyvim.plugins.extras.ui.edgy",
     "lazyvim.plugins.extras.util.dot",
     "plugins.extras.coding.ai.codeium",
+    "plugins.extras.coding.ai.gen",
     "plugins.extras.coding.annotation",
     "plugins.extras.coding.multicursor",
     "plugins.extras.coding.recorder",
diff --git a/config/nvim/lua/config/keymaps.lua b/config/nvim/lua/config/keymaps.lua
index ed9817e1..1941689a 100644
--- a/config/nvim/lua/config/keymaps.lua
+++ b/config/nvim/lua/config/keymaps.lua
@@ -40,6 +40,7 @@ end
 -- Buffers
 map("n", "<leader>bf", "<cmd>bfirst<cr>", { desc = "First Buffer" })
 map("n", "<leader>ba", "<cmd>blast<cr>", { desc = "Last Buffer" })
+map("n", "<leader>b<tab>", "<cmd>tabnew %<cr>", { desc = "Current Buffer in New Tab" })
 
 -- Center the screen automatically
 map("n", "n", "nzzzv")
diff --git a/config/nvim/lua/plugins/extras/coding/ai/gen.lua b/config/nvim/lua/plugins/extras/coding/ai/gen.lua
index d97d70d9..f620fe56 100644
--- a/config/nvim/lua/plugins/extras/coding/ai/gen.lua
+++ b/config/nvim/lua/plugins/extras/coding/ai/gen.lua
@@ -1,29 +1,63 @@
 return {
-  "jellydn/gen.nvim",
-  opts = {
-    model = "codellama",
-    display_mode = "split",
-    show_prompt = true,
-    show_model = true,
-    no_auto_close = false,
-    init = function(options)
-      pcall(io.popen, "ollama serve > /dev/null 2>&1 &")
-    end,
-    command = "curl --silent --no-buffer -X POST http://localhost:11434/api/generate -d $body",
-  },
-  config = function(_, opts)
-    local gen = require("gen")
-    gen.setup(opts)
+  {
+    "jellydn/gen.nvim",
+    opts = {
+      model = "codellama",
+      host = "localhost",
+      port = "11434",
+      display_mode = "split",
+      show_prompt = true,
+      show_model = true,
+      no_auto_close = false,
+      init = function(options)
+        pcall(io.popen, "ollama serve > /dev/null 2>&1 &")
+      end,
+      command = function(options)
+        local body = { model = options.model, stream = true }
+        return "curl --silent --no-buffer -X POST http://"
+          .. options.host
+          .. ":"
+          .. options.port
+          .. "/api/generate -d $body"
+      end,
+    },
+    config = function(_, opts)
+      local gen = require("gen")
+      gen.setup(opts)
 
-    gen.prompts["Elaborate_Text"] = {
-      prompt = "Elaborate the following text:\n$text",
-    }
-    gen.prompts["Fix_Code"] = {
-      prompt = "Fix the following code. Only output the result in format ```$filetype\n...\n```:\n```$filetype\n$text\n```",
-      extract = "```$filetype\n(.-)```",
-    }
-  end,
-  keys = {
-    { "<leader>i", "<cmd>Gen<cr>", mode = { "n", "v" }, desc = "IA (Gen)" },
+      gen.prompts["Elaborate_Text"] = {
+        prompt = "Elaborate the following text:\n$text",
+      }
+      gen.prompts["Fix_Code"] = {
+        prompt = "Fix the following code. Only output the result in format ```$filetype\n...\n```:\n```$filetype\n$text\n```",
+        extract = "```$filetype\n(.-)```",
+      }
+    end,
+    keys = {
+      { "<leader>ii", "<cmd>Gen<cr>", mode = { "n", "v" }, desc = "IA (Gen)" },
+      { "<leader>ig", "<cmd>Gen Generate<cr>", mode = { "n" }, desc = "Generate" },
+      { "<leader>ic", "<cmd>Gen Chat<cr>", mode = { "n" }, desc = "Chat" },
+      { "<leader>is", "<cmd>Gen Summarize<cr>", mode = { "n", "v" }, desc = "Summarize" },
+      { "<leader>ia", "<cmd>Gen Ask<cr>", mode = { "v" }, desc = "Ask" },
+      { "<leader>iH", "<cmd>Gen Change<cr>", mode = { "v" }, desc = "Change" },
+      { "<leader>iG", "<cmd>Gen Enhance_Grammar_Spelling<cr>", mode = { "v" }, desc = "Enhance Grammar Spelling" },
+      { "<leader>iw", "<cmd>Gen Enhance_Wording<cr>", mode = { "v" }, desc = "Enhance Wording" },
+      { "<leader>iC", "<cmd>Gen Make_Concise<cr>", mode = { "v" }, desc = "Make Concise" },
+      { "<leader>il", "<cmd>Gen Make_List<cr>", mode = { "v" }, desc = "Make List" },
+      { "<leader>it", "<cmd>Gen Make_Table<cr>", mode = { "v" }, desc = "Make Table" },
+      { "<leader>ir", "<cmd>Gen Review_Code<cr>", mode = { "v" }, desc = "Review Code" },
+      { "<leader>ie", "<cmd>Gen Enhance_Code<cr>", mode = { "v" }, desc = "Enhance Code" },
+      { "<leader>ih", "<cmd>Gen Change_Code<cr>", mode = { "v" }, desc = "Change Code" },
+      { "<leader>if", "<cmd>Gen Fix_Code<cr>", mode = { "v" }, desc = "Fix Code" },
+      { "<leader>iE", "<cmd>Gen Elaborate_Text<cr>", mode = { "v" }, desc = "Elaborate Text" },
+    },
+  },
+  {
+    "folke/which-key.nvim",
+    opts = {
+      defaults = {
+        ["<leader>i"] = { name = "󱚦 IA (Gen)" },
+      },
+    },
   },
 }
diff --git a/config/nvim/lua/plugins/extras/lang/python-extended.lua b/config/nvim/lua/plugins/extras/lang/python-extended.lua
index e51ac712..683e69c0 100644
--- a/config/nvim/lua/plugins/extras/lang/python-extended.lua
+++ b/config/nvim/lua/plugins/extras/lang/python-extended.lua
@@ -1,10 +1,34 @@
+-- LSP Server to use for Python.
+-- Set to "basedpyright" to use basedpyright instead of pyright.
+vim.g.lazyvim_python_lsp = "basedpyright"
+
 return {
   { import = "lazyvim.plugins.extras.lang.python" },
-  { import = "lazyvim.plugins.extras.lang.python-semshi" },
   {
     "neovim/nvim-lspconfig",
     opts = {
       servers = {
+        ---@type lspconfig.options.basedpyright
+        basedpyright = {
+          settings = {
+            basedpyright = {
+              analysis = {
+                diagnosticSeverityOverrides = {
+                  reportUnusedCallResult = "information",
+                  reportUnusedExpression = "information",
+                  reportUnknownMemberType = "none",
+                  reportUnknownLambdaType = "none",
+                  reportUnknownParameterType = "none",
+                  reportMissingParameterType = "none",
+                  reportUnknownVariableType = "none",
+                  reportUnknownArgumentType = "none",
+                  reportAny = "none",
+                },
+              },
+            },
+          },
+        },
+        ---@type lspconfig.options.pyright
         pyright = {
           settings = {
             verboseOutput = true,