Update: Add Mistral API key and Parrot.nvim plugin
- Introduce Mistral API key - Add Parrot.nvim plugin - Implement a function to read the API key from a secrets file - Set up Parrot.nvim with Mistral provider and custom hooks for code assistance and git commit generation
This commit is contained in:
parent
a9f321a847
commit
13dad7b532
|
@ -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,
|
||||
}
|
||||
}
|
||||
'';
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue