vibe wip

Author: functora

pub/vi/vimrc.lua

@@ -6,6 +6,8 @@ end
 
 local ok, codecompanion = pcall(require, "codecompanion")
 if ok then
+	local rust =
+		"Read about the project in README.md. Your code must be in a strictly functional style while being highly efficient, avoid mutable variables (mut), avoid imperative statements like loops, for, while, and return, use functional style iterators instead, avoid redundant closures and redundant variables, use method and function chaining/compositions as much as possible, avoid the clone method, avoid redundant allocations, cover the code with tests, and follow the existing project code style. However, to achieve this, use only standard, simple Rust. Avoid relying on fancy custom traits, and write code in a simple, functional style. Ensure that all functions, variables, types, and other identifiers in the code are meaningful and clear in their context, while still preferring shorter names, as in the original code. Do not write any comments in the code, as good code should always be clear on its own. After making your changes, ensure that the code is formatted with rustfmt, compiles successfully, and passes all tests."
 	codecompanion.setup({
 		strategies = {
 			chat = {
@@ -21,32 +23,62 @@ if ok then
 				adapter = "gemini_cli",
 			},
 		},
+		prompt_library = {
+			rust = {
+				strategy = "chat",
+				prompts = { { role = "system", content = rust } },
+				opts = {},
+			},
+		},
 		adapters = {
 			acp = {
 				gemini_cli = function()
 					return require("codecompanion.adapters").extend("gemini_cli", {
 						defaults = {
 							auth_method = "gemini-api-key",
 							mcpServers = {},
-							timeout = 20000, -- 20 seconds
+							timeout = 50000,
+						},
+						commands = {
+							default = {
+								"gemini",
+								"--experimental-acp",
+								"-m",
+								"gemini-2.5-flash-lite",
+							},
+							lite = {
+								"gemini",
+								"--experimental-acp",
+								"-m",
+								"gemini-2.5-flash-lite",
+							},
+							flash = {
+								"gemini",
+								"--experimental-acp",
+								"-m",
+								"gemini-2.5-flash",
+							},
+							pro = {
+								"gemini",
+								"--experimental-acp",
+								"-m",
+								"gemini-2.5-pro",
+							},
 						},
 					})
 				end,
 			},
 		},
 	})
-	vim.keymap.set("n", "<leader>zz", "<CMD>CodeCompanionChat Toggle<CR>", {
-		desc = "ai: ch[a]t",
-	})
-	vim.keymap.set("n", "<leader>zx", "<CMD>CodeCompanionActions<CR>", {
-		desc = "ai: action[s]",
-	})
-	vim.keymap.set("n", "<leader>zc", "<CMD>CodeCompanionCmd<CR>", {
-		desc = "ai: cm[d]",
-	})
-	vim.keymap.set({ "n", "v" }, "<leader>zv", "<CMD>CodeCompanion<CR>", {
-		desc = "ai: [f]ile write with diff",
-	})
+	vim.api.nvim_create_user_command("CodeCompanionRust", function(opts)
+		local prev = opts.args
+		local next = rust .. (prev and " " .. prev or "")
+		vim.cmd("CodeCompanionChat " .. next)
+	end, { nargs = "*", complete = "file" })
+	vim.keymap.set("n", "<leader>zz", "<CMD>CodeCompanionChat Toggle<CR>", {})
+	vim.keymap.set("n", "<leader>zx", "<CMD>CodeCompanionActions<CR>", {})
+	vim.keymap.set("n", "<leader>zc", ":CodeCompanionRust ", {})
+	-- vim.g.codecompanion_yolo_mode = true
 end
 
 local ok, minuet = pcall(require, "minuet")