forked from CopilotC-Nvim/CopilotChat.nvim
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.lua
351 lines (331 loc) · 12.1 KB
/
config.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
local prompts = require('CopilotChat.prompts')
local context = require('CopilotChat.context')
local select = require('CopilotChat.select')
--- @class CopilotChat.config.source
--- @field bufnr number
--- @field winnr number
---@class CopilotChat.config.selection.diagnostic
---@field message string
---@field severity string
---@field start_row number
---@field start_col number
---@field end_row number
---@field end_col number
---@class CopilotChat.config.selection
---@field lines string
---@field diagnostics table<CopilotChat.config.selection.diagnostic>?
---@field filename string?
---@field filetype string?
---@field start_row number?
---@field start_col number?
---@field end_row number?
---@field end_col number?
---@class CopilotChat.config.context
---@field description string?
---@field input fun(callback: fun(input: string?))?
---@field resolve fun(input: string?, source: CopilotChat.config.source):table<CopilotChat.copilot.embed>
---@class CopilotChat.config.prompt
---@field prompt string?
---@field description string?
---@field kind string?
---@field mapping string?
---@field system_prompt string?
---@field callback fun(response: string, source: CopilotChat.config.source)?
---@field selection nil|fun(source: CopilotChat.config.source):CopilotChat.config.selection?
---@class CopilotChat.config.window
---@field layout string?
---@field relative string?
---@field border string?
---@field width number?
---@field height number?
---@field row number?
---@field col number?
---@field title string?
---@field footer string?
---@field zindex number?
---@class CopilotChat.config.mapping
---@field normal string?
---@field insert string?
---@field detail string?
---@class CopilotChat.config.mappings
---@field complete CopilotChat.config.mapping?
---@field close CopilotChat.config.mapping?
---@field reset CopilotChat.config.mapping?
---@field submit_prompt CopilotChat.config.mapping?
---@field toggle_sticky CopilotChat.config.mapping?
---@field accept_diff CopilotChat.config.mapping?
---@field yank_diff CopilotChat.config.mapping?
---@field show_diff CopilotChat.config.mapping?
---@field show_system_prompt CopilotChat.config.mapping?
---@field show_user_selection CopilotChat.config.mapping?
---@field show_help CopilotChat.config.mapping?
--- CopilotChat default configuration
---@class CopilotChat.config
---@field debug boolean?
---@field log_level string?
---@field proxy string?
---@field allow_insecure boolean?
---@field system_prompt string?
---@field model string?
---@field agent string?
---@field context string?
---@field temperature number?
---@field question_header string?
---@field answer_header string?
---@field error_header string?
---@field separator string?
---@field chat_autocomplete boolean?
---@field show_folds boolean?
---@field show_help boolean?
---@field auto_follow_cursor boolean?
---@field auto_insert_mode boolean?
---@field clear_chat_on_new_prompt boolean?
---@field highlight_selection boolean?
---@field highlight_headers boolean?
---@field history_path string?
---@field callback fun(response: string, source: CopilotChat.config.source)?
---@field selection nil|fun(source: CopilotChat.config.source):CopilotChat.config.selection?
---@field contexts table<string, CopilotChat.config.context>?
---@field prompts table<string, CopilotChat.config.prompt|string>?
---@field window CopilotChat.config.window?
---@field mappings CopilotChat.config.mappings?
return {
debug = false, -- Enable debug logging (same as 'log_level = 'debug')
log_level = 'info', -- Log level to use, 'trace', 'debug', 'info', 'warn', 'error', 'fatal'
proxy = nil, -- [protocol://]host[:port] Use this proxy
allow_insecure = false, -- Allow insecure server connections
system_prompt = prompts.COPILOT_INSTRUCTIONS, -- System prompt to use (can be specified manually in prompt via /).
model = 'gpt-4o', -- Default model to use, see ':CopilotChatModels' for available models (can be specified manually in prompt via $).
agent = 'copilot', -- Default agent to use, see ':CopilotChatAgents' for available agents (can be specified manually in prompt via @).
context = nil, -- Default context to use (can be specified manually in prompt via #).
temperature = 0.1, -- GPT result temperature
question_header = '## User ', -- Header to use for user questions
answer_header = '## Copilot ', -- Header to use for AI answers
error_header = '## Error ', -- Header to use for errors
separator = '───', -- Separator to use in chat
chat_autocomplete = true, -- Enable chat autocompletion (when disabled, requires manual `mappings.complete` trigger)
show_folds = true, -- Shows folds for sections in chat
show_help = true, -- Shows help message as virtual lines when waiting for user input
auto_follow_cursor = true, -- Auto-follow cursor in chat
auto_insert_mode = false, -- Automatically enter insert mode when opening window and on new prompt
insert_at_end = false, -- Move cursor to end of buffer when inserting text
clear_chat_on_new_prompt = false, -- Clears chat on every new prompt
highlight_selection = true, -- Highlight selection
highlight_headers = true, -- Highlight headers in chat, disable if using markdown renderers (like render-markdown.nvim)
history_path = vim.fn.stdpath('data') .. '/copilotchat_history', -- Default path to stored history
callback = nil, -- Callback to use when ask response is received
-- default selection
selection = function(source)
return select.visual(source) or select.buffer(source)
end,
-- default contexts
contexts = {
buffer = {
description = 'Includes specified buffer in chat context (default current). Supports input.',
input = function(callback)
vim.ui.select(
vim.tbl_map(
function(buf)
return { id = buf, name = vim.api.nvim_buf_get_name(buf) }
end,
vim.tbl_filter(function(buf)
return vim.api.nvim_buf_is_loaded(buf) and vim.fn.buflisted(buf) == 1
end, vim.api.nvim_list_bufs())
),
{
prompt = 'Select a buffer> ',
format_item = function(item)
return item.name
end,
},
function(choice)
callback(choice and choice.id)
end
)
end,
resolve = function(input, source)
return {
context.outline(input and tonumber(input) or source.bufnr),
}
end,
},
buffers = {
description = 'Includes all buffers in chat context (default listed). Supports input.',
input = function(callback)
vim.ui.select({ 'listed', 'visible' }, {
prompt = 'Select buffer scope> ',
}, callback)
end,
resolve = function(input)
input = input or 'listed'
return vim.tbl_map(
context.outline,
vim.tbl_filter(function(b)
return vim.api.nvim_buf_is_loaded(b)
and vim.fn.buflisted(b) == 1
and (input == 'listed' or #vim.fn.win_findbuf(b) > 0)
end, vim.api.nvim_list_bufs())
)
end,
},
file = {
description = 'Includes content of provided file in chat context. Supports input.',
input = function(callback)
local files = vim.tbl_filter(function(file)
return vim.fn.isdirectory(file) == 0
end, vim.fn.glob('**/*', false, true))
vim.ui.select(files, {
prompt = 'Select a file> ',
}, callback)
end,
resolve = function(input)
return {
context.file(input),
}
end,
},
files = {
description = 'Includes all non-hidden filenames in the current workspace in chat context. Supports input.',
input = function(callback)
vim.ui.input({
prompt = 'Enter a file pattern> ',
default = '**/*',
}, callback)
end,
resolve = function(input)
return context.files(input)
end,
},
git = {
description = 'Includes current git diff in chat context (default unstaged). Supports input.',
input = function(callback)
vim.ui.select({ 'unstaged', 'staged' }, {
prompt = 'Select diff type> ',
}, callback)
end,
resolve = function(input, source)
return {
context.gitdiff(input, source.bufnr),
}
end,
},
},
-- default prompts
prompts = {
Explain = {
prompt = '> /COPILOT_EXPLAIN\n\nWrite an explanation for the selected code and diagnostics as paragraphs of text.',
},
Review = {
prompt = '> /COPILOT_REVIEW\n\nReview the selected code.',
callback = function(response, source)
local diagnostics = {}
for line in response:gmatch('[^\r\n]+') do
if line:find('^line=') then
local start_line = nil
local end_line = nil
local message = nil
local single_match, message_match = line:match('^line=(%d+): (.*)$')
if not single_match then
local start_match, end_match, m_message_match = line:match('^line=(%d+)-(%d+): (.*)$')
if start_match and end_match then
start_line = tonumber(start_match)
end_line = tonumber(end_match)
message = m_message_match
end
else
start_line = tonumber(single_match)
end_line = start_line
message = message_match
end
if start_line and end_line then
table.insert(diagnostics, {
lnum = start_line - 1,
end_lnum = end_line - 1,
col = 0,
message = message,
severity = vim.diagnostic.severity.WARN,
source = 'Copilot Review',
})
end
end
end
vim.diagnostic.set(
vim.api.nvim_create_namespace('copilot_diagnostics'),
source.bufnr,
diagnostics
)
end,
},
Fix = {
prompt = '> /COPILOT_GENERATE\n\nThere is a problem in this code. Rewrite the code to show it with the bug fixed.',
},
Optimize = {
prompt = '> /COPILOT_GENERATE\n\nOptimize the selected code to improve performance and readability.',
},
Docs = {
prompt = '> /COPILOT_GENERATE\n\nPlease add documentation comments to the selected code.',
},
Tests = {
prompt = '> /COPILOT_GENERATE\n\nPlease generate tests for my code.',
},
Commit = {
prompt = '> #git:staged\n\nWrite commit message for the change with commitizen convention. Make sure the title has maximum 50 characters and message is wrapped at 72 characters. Wrap the whole message in code block with language gitcommit.',
},
},
-- default window options
window = {
layout = 'vertical', -- 'vertical', 'horizontal', 'float', 'replace'
width = 0.5, -- fractional width of parent, or absolute width in columns when > 1
height = 0.5, -- fractional height of parent, or absolute height in rows when > 1
-- Options below only apply to floating windows
relative = 'editor', -- 'editor', 'win', 'cursor', 'mouse'
border = 'single', -- 'none', single', 'double', 'rounded', 'solid', 'shadow'
row = nil, -- row position of the window, default is centered
col = nil, -- column position of the window, default is centered
title = 'Copilot Chat', -- title of chat window
footer = nil, -- footer of chat window
zindex = 1, -- determines if window is on top or below other floating windows
},
-- default mappings
mappings = {
complete = {
insert = '<Tab>',
},
close = {
normal = 'q',
insert = '<C-c>',
},
reset = {
normal = '<C-l>',
insert = '<C-l>',
},
submit_prompt = {
normal = '<CR>',
insert = '<C-s>',
},
toggle_sticky = {
detail = 'Makes line under cursor sticky or deletes sticky line.',
normal = 'gr',
},
accept_diff = {
normal = '<C-y>',
insert = '<C-y>',
},
yank_diff = {
normal = 'gy',
register = '"',
},
show_diff = {
normal = 'gd',
},
show_system_prompt = {
normal = 'gp',
},
show_user_selection = {
normal = 'gs',
},
show_help = {
normal = 'gh',
},
},
}