dots/chezmoi/dot_config/nvim/lua/aleidk/plugins/ai.lua

return {
	"jackMort/ChatGPT.nvim",
	event = "VeryLazy",
	opts = {
		api_key_cmd = nil,
		yank_register = "+",
		edit_with_instructions = {
			diff = false,
			keymaps = {
				-- close = "<C-c>",
				-- accept = "<C-y>",
				-- toggle_diff = "<C-d>",
				-- toggle_settings = "<C-o>",
				-- toggle_help = "<C-h>",
				-- cycle_windows = "<Tab>",
				-- use_output_as_input = "<C-i>",
				submit = "<C-a>",
			},
		},
		chat = {
			-- welcome_message = WELCOME_MESSAGE,
			loading_text = "Loading, please wait ...",
			question_sign = "", -- 🙂
			answer_sign = "ﮧ", -- 🤖
			border_left_sign = "",
			border_right_sign = "",
			max_line_length = 120,
			sessions_window = {
				active_sign = "  ",
				inactive_sign = "  ",
				current_line_sign = "",
				border = {
					style = "rounded",
					text = {
						top = " Sessions ",
					},
				},
				win_options = {
					winhighlight = "Normal:Normal,FloatBorder:FloatBorder",
				},
			},
			keymaps = {
				-- 	close = "<C-c>",
				-- 	yank_last = "<C-y>",
				-- 	yank_last_code = "<C-k>",
				-- 	scroll_up = "<C-u>",
				-- 	scroll_down = "<C-d>",
				-- 	new_session = "<C-n>",
				-- 	cycle_windows = "<Tab>",
				-- 	cycle_modes = "<C-f>",
				-- 	next_message = "<C-j>",
				-- 	prev_message = "<C-k>",
				-- 	select_session = "<Space>",
				-- 	rename_session = "r",
				-- 	delete_session = "d",
				-- 	draft_message = "<C-r>",
				-- 	edit_message = "e",
				-- 	delete_message = "d",
				-- 	toggle_settings = "<C-o>",
				-- 	toggle_sessions = "<C-p>",
				-- 	toggle_help = "<C-h>",
				-- 	toggle_message_role = "<C-r>",
				-- 	toggle_system_role_open = "<C-s>",
				-- 	stop_generating = "<C-x>",
				submit = "<C-a>",
			},
		},
		popup_layout = {
			default = "center",
			center = {
				width = "80%",
				height = "80%",
			},
			right = {
				width = "30%",
				width_settings_open = "50%",
			},
		},
		popup_window = {
			border = {
				highlight = "FloatBorder",
				style = "rounded",
				text = {
					top = " ChatGPT ",
				},
			},
			win_options = {
				wrap = true,
				linebreak = true,
				foldcolumn = "1",
				winhighlight = "Normal:Normal,FloatBorder:FloatBorder",
			},
			buf_options = {
				filetype = "markdown",
			},
		},
		system_window = {
			border = {
				highlight = "FloatBorder",
				style = "rounded",
				text = {
					top = " SYSTEM ",
				},
			},
			win_options = {
				wrap = true,
				linebreak = true,
				foldcolumn = "2",
				winhighlight = "Normal:Normal,FloatBorder:FloatBorder",
			},
		},
		popup_input = {
			prompt = "  ",
			border = {
				highlight = "FloatBorder",
				style = "rounded",
				text = {
					top_align = "center",
					top = " Prompt ",
				},
			},
			win_options = {
				winhighlight = "Normal:Normal,FloatBorder:FloatBorder",
			},
			submit = "<C-Enter>",
			submit_n = "<Enter>",
			max_visible_lines = 20,
		},
		settings_window = {
			setting_sign = "  ",
			border = {
				style = "rounded",
				text = {
					top = " Settings ",
				},
			},
			win_options = {
				winhighlight = "Normal:Normal,FloatBorder:FloatBorder",
			},
		},
		help_window = {
			setting_sign = "  ",
			border = {
				style = "rounded",
				text = {
					top = " Help ",
				},
			},
			win_options = {
				winhighlight = "Normal:Normal,FloatBorder:FloatBorder",
			},
		},
		openai_params = {
			model = "gpt-3.5-turbo",
			frequency_penalty = 0,
			presence_penalty = 0,
			max_tokens = 300,
			temperature = 0,
			top_p = 1,
			n = 1,
		},
		openai_edit_params = {
			model = "gpt-3.5-turbo",
			frequency_penalty = 0,
			presence_penalty = 0,
			temperature = 0,
			top_p = 1,
			n = 1,
		},
		use_openai_functions_for_edits = false,
		actions_paths = {},
		show_quickfixes_cmd = "Trouble quickfix",
		predefined_chat_gpt_prompts = "https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv",
		highlights = {
			help_key = "@symbol",
			help_description = "@comment",
		},
	},
	config = function()
		require("chatgpt").setup()
	end,
	dependencies = {
		"MunifTanjim/nui.nvim",
		"nvim-lua/plenary.nvim",
		"folke/trouble.nvim",
		"nvim-telescope/telescope.nvim",
	},
	keys = {
		{ "<leader>ap", "<cmd>ChatGPT<CR>", desc = "ChatGPT" },
		{ "<leader>ae", "<cmd>ChatGPTEditWithInstruction<CR>", desc = "Edit with instruction", mode = { "n", "v" } },
		{ "<leader>aa", "<cmd>ChatGPTActAs<CR>", desc = "Act as", mode = { "n", "v" } },
		{ "<leader>ag", "<cmd>ChatGPTRun grammar_correction<CR>", desc = "Grammar Correction", mode = { "n", "v" } },
		{ "<leader>aT", "<cmd>ChatGPTRun translate<CR>", desc = "Translate", mode = { "n", "v" } },
		{ "<leader>ak", "<cmd>ChatGPTRun keywords<CR>", desc = "Keywords", mode = { "n", "v" } },
		{ "<leader>ad", "<cmd>ChatGPTRun docstring<CR>", desc = "Docstring", mode = { "n", "v" } },
		{ "<leader>at", "<cmd>ChatGPTRun add_tests<CR>", desc = "Add Tests", mode = { "n", "v" } },
		{ "<leader>ao", "<cmd>ChatGPTRun optimize_code<CR>", desc = "Optimize Code", mode = { "n", "v" } },
		{ "<leader>as", "<cmd>ChatGPTRun summarize<CR>", desc = "Summarize", mode = { "n", "v" } },
		{ "<leader>af", "<cmd>ChatGPTRun fix_bugs<CR>", desc = "Fix Bugs", mode = { "n", "v" } },
		{ "<leader>ax", "<cmd>ChatGPTRun explain_code<CR>", desc = "Explain Code", mode = { "n", "v" } },
		{ "<leader>ar", "<cmd>ChatGPTRun roxygen_edit<CR>", desc = "Roxygen Edit", mode = { "n", "v" } },
		{
			"<leader>al",
			"<cmd>ChatGPTRun code_readability_analysis<CR>",
			desc = "Code Readability Analysis",
			mode = { "n", "v" },
		},
	},
}