diff --git a/home/common/core/nixvim/plugins/default.nix b/home/common/core/nixvim/plugins/default.nix index 348b687..fe13b79 100644 --- a/home/common/core/nixvim/plugins/default.nix +++ b/home/common/core/nixvim/plugins/default.nix @@ -1,10 +1,18 @@ { pkgs, config, + lib, + inputs, ... }: let user = config.home.username; + mistralAPIKey = lib.optionalString (lib.hasAttr "sops-nix" inputs) config.sops.secrets."mistral-api-key".path; in { + sops.secrets = { + "mistral-api-key" = { + }; + }; + imports = [ ./cmp.nix ./colorizer.nix @@ -32,6 +40,18 @@ in { pkgs.vimPlugins.vim-dadbod-ui pkgs.vimPlugins.vim-dadbod-completion pkgs.vimPlugins.fugitive + + (pkgs.vimUtils.buildVimPlugin + { + name = "parrot.nvim"; + src = pkgs.fetchFromGitHub { + owner = "frankroeder"; + repo = "parrot.nvim"; + rev = "c992483"; + sha256 = "sha256-P899qNY/h5LTW0pp3sAmcSDsDGWQedXa20J7EggL7F8="; + }; + }) + (pkgs.vimUtils.buildVimPlugin { name = "buffer_manager.nvim"; @@ -58,6 +78,18 @@ in { pkgs.vimPlugins.vim-devicons ]; programs.nixvim.extraConfigLua = '' + -- function to read api key from secrets file + local function read_api_key(file_path) + local file = io.open(file_path, "r") + if file then + local api_key = file:read("*all") + file:close() + return api_key + else + error("Failed to open file: " .. file_path) + end + end + -- buffer_manager.nvim local opts = {noremap = true} @@ -123,5 +155,71 @@ in { end require("conform").format({ async = true, lsp_format = "fallback", range = range }) end, { range = true }) + + -- parrot.nvim setup + require("parrot").setup { + toggle_target = "popup", + providers = { + mistral = { + api_key = read_api_key("${mistralAPIKey}"), + params = { + chat = { temperature = 0, top_p = 1, max_tokens = 16384 }, + command = { temperature = 0, top_p = 1, max_tokens = 16384 }, + }, + -- api_key = os.getenv "MISTRAL_API_KEY", + }, + }, + hooks = { + AskCoder = function(prt, params) + local prompt = [[ + You are a code assistent. Answer my question carefully and concisely. + Answer with code snippets only. Do not give a summary or include any + code comments in your response. Never repeate code from a prevous + response or write obvious or unnecessary code, instead refer to such + code like: "**rest of code here**", or something to that effect. + Tokens are expensive, to we need to be very consice. If you uses + excessive tokens, then I will send you the invoice to pay the bill. + Question: {{command}} + ``` + ]] + prt.ChatNew(params, prompt) + end, + + AskCoderSel = function(parrot, params) + local template = [[ + You are an expert programmer. Here is a code snippet along with a + question. Give me a short, accurate and consice answer. + + ```{{filetype}} + {{selection}} + ``` + ]] + local model_obj = parrot.get_model("command") + parrot.logger.info("Asking model: " .. model_obj.name) + parrot.Prompt(params, parrot.ui.Target.popup, model_obj, "🤖 Ask ~ ", template) + end, + + GitCommit = function(parrot, params) + local template = [[ + You are a git commit generator. You will be provided with a git diff file. + Your role is to create a concise commit message that outlines all of the + most relevant and impactful changes in the diff. You only output a commit + message, do not write anything else not related to the commit. Use + imperative mood for you writing style. Your message should include a + short summary in the first line followed by more detailed descriptions of + the changes on subsequent lines. The first line summary should be no more + than 70 characters, not end with a period, and a blank line should + separate the summary from the body of the message. There should only be 1 + summary line, and 1 body with the description. The body should include no + more than 3 bullet points with more detailed descriptions of the files + and functions that have been modified. Do not output any code. + Diff: {{selection}} + ]] + local model_obj = parrot.get_model("command") + parrot.logger.info("Asking model: " .. model_obj.name) + parrot.Prompt(params, parrot.ui.Target.popup, model_obj, nil, template) + end, + } + } ''; }