Changes

Author: ralismark

init/10-local:cody.lua

@@ -1 +1,3 @@
-require "cody"
+local cody = require "cody"
+local c = cody.config
+c.autocomplete.enable = false

init/50-maps-misc.lua

@@ -158,6 +158,7 @@ vim.keymap.set("i", "<c-y>", "<c-o><c-y>")
 
 -- readline
 vim.keymap.set({"i", "c"}, "<c-a>", "<Home>")
+vim.keymap.set({"i", "c"}, "", "<c-w>")
 
 -- Motions & Text Objects {{{1
 

init/50-maps-misc.vim

@@ -61,8 +61,8 @@ nnoremap <silent> [T <cmd>tabmove -1<cr>
 nnoremap <silent> ]T <cmd>tabmove +1<cr>
 
 " quickfix browse
-nnoremap <silent> [c <cmd>cprev<cr>
-nnoremap <silent> ]c <cmd>cnext<cr>
+nnoremap <silent> [q <cmd>cprev<cr>
+nnoremap <silent> ]q <cmd>cnext<cr>
 nnoremap <silent> [l <cmd>lprev<cr>
 nnoremap <silent> ]l <cmd>lnext<cr>
 nnoremap <silent> ]d <cmd>lua vim.diagnostic.goto_next()<cr>

init/51-leader.lua

@@ -15,15 +15,12 @@ vim.keymap.set("n", "<leader>os", [[<cmd>set spell! | set spell?<cr>]])
 vim.keymap.set("n", "<leader>on", [[<cmd>set relativenumber! | set relativenumber?<cr>]])
 vim.keymap.set("n", "<leader>od", [[<cmd>if &diff | diffoff | else | diffthis | endif | set diff?<cr>]])
 vim.keymap.set("n", "<leader>oq", [[<cmd>if getqflist({"winid":0}).winid | cclose | else | botright copen | endif<cr>]])
-vim.keymap.set("n", "<leader>oc", [[<cmd>if getloclist(0, {"winid":0}).winid | lclose | else | botright lopen | endif<cr>]])
--- vim.keymap.set("n", "<leader>ol", function()
--- 	if vim.fn.getqflist({ winid = 0 }).winid ~= 0 then
--- 		vim.cmd("cclose")
--- 	else
--- 		require "vimrc.diagnostic".load_qf()
--- 		vim.cmd("botright copen")
--- 	end
--- end)
+vim.keymap.set("n", "<leader>ol", [[<cmd>if getloclist(0, {"winid":0}).winid | lclose | else | botright lopen | endif<cr>]])
+vim.keymap.set("n", "<leader>oa", function()
+	local cody = require "cody"
+	cody.config.autocomplete.enable = not cody.config.autocomplete.enable
+	print((cody.config.autocomplete.enable and "  " or "no") .. "aicomplete")
+end)
 
 -- Splits
 vim.keymap.set("n", "<leader>s", "<nop>")
@@ -55,7 +52,7 @@ vim.keymap.set("n", "<leader>fu", function()
 		["”"] = '"',
 	}
 	for k, v in pairs(reps) do
-		vim.cmd("keepjumps keeppatterns %s/" .. k .. "/" .. v "/ge")
+		vim.cmd("keepjumps keeppatterns %s/" .. k .. "/" .. v .. "/ge")
 	end
 end)
 

lua/cody/api.lua

@@ -58,7 +58,11 @@ local function handle_syscomplete(context, results)
 	end
 end
 
--- See https://github.com/sourcegraph/openapi for docs
+---@class cody.ApiResponse
+---@field wait fun(): table
+
+--- See https://github.com/sourcegraph/openapi for docs
+---@return cody.ApiResponse
 function M.request(method, path, data, cb)
 	if vim.env.SRC_ENDPOINT == nil or vim.env.SRC_ACCESS_TOKEN == nil then
 		error("SRC_ENDPOINT and SRC_ACCESS_TOKEN must be set")

lua/cody/autocomplete.lua

@@ -6,7 +6,7 @@ local M = {}
 ---@return cody.PromptParams
 local function gather_context(buf, linenr, colnr, tokens)
 	local path = vim.api.nvim_buf_get_name(buf)
-	local context_bytes = tokens * config.chars_per_token * config.context_proportion
+	local context_bytes = tokens * config.chars_per_token * config.autocomplete.context_proportion
 
 	-- Context around the cursor
 	local affix_size = math.floor(context_bytes / 2)
@@ -30,16 +30,34 @@ local function gather_context(buf, linenr, colnr, tokens)
 	}
 end
 
-local function current_model_config()
-	if not (vim.env.SRC_ENDPOINT and vim.env.SRC_ACCESS_TOKEN) then
-		return nil
+---@return string|nil name
+---@return cody.ModelConfig|nil mc
+function M.get_model(setdefault)
+	if not (vim.env.SRC_ENDPOINT and vim.env.SRC_ACCESS_TOKEN and config.autocomplete.enable) then
+		return nil, nil
 	end
-	local cody_models = require "cody.models"
-	return cody_models.models[config.autocomplete_model]
+
+	if setdefault and config.autocomplete.model == nil then
+		local req = api.request("GET", "/.api/modelconfig/supported-models.json")
+		local ok, response = pcall(req.wait, req)
+		if ok and response.defaultModels and response.defaultModels.codeCompletion then
+			config.autocomplete.model = response.defaultModels.codeCompletion
+		else
+			config.autocomplete.model = "(could not load default)"
+		end
+	end
+
+	local mc = (require "cody.models").models[config.autocomplete.model]
+	if not mc then
+		return nil, nil
+	end
+
+	return config.autocomplete.model, mc
 end
 
 function M:is_available()
-	return not not current_model_config()
+	local name, mc = M.get_model(false)
+	return mc ~= nil
 end
 
 function M:get_trigger_characters()
@@ -57,16 +75,16 @@ function M:complete(params, callback)
 	local linenr = params.context.cursor.row
 	local colnr = params.context.cursor.col
 
+	local name, mc = M.get_model(true)
 	---@type cody.ModelConfig|nil
-	local mc = current_model_config()
 	if not mc then
 		return
 	end
 
 	-- Generate request body
 	local prompt_params = gather_context(params.context.bufnr, linenr, colnr, mc.contextTokens)
 	local body = {
-		model = config.autocomplete_model,
+		model = name,
 
 		stopSequences = mc.stopSequences,
 		maxTokensToSample = mc.autoCompleteMultilineMaxTokens,

lua/cody/config.lua

@@ -1,12 +1,19 @@
-local M = {}
+return {
 
-M.autocomplete_model = "fireworks::v1::deepseek-coder-v2-lite-base" --[[@as cody.ModelName]]
+	-- For Claude, a token approximately represents 3.5 English characters.
+	-- see: https://docs.anthropic.com/claude/docs/glossary#tokens
+	chars_per_token = 3.5,
 
--- For Claude, a token approximately represents 3.5 English characters.
--- see: https://docs.anthropic.com/claude/docs/glossary#tokens
-M.chars_per_token = 3.5
+	-- Settings for autocomplete
+	autocomplete = {
+		enable = true,
 
--- The proportion of the token limit that should used for context.
-M.context_proportion = 0.9
+		-- cody.ModelName, or nil to fetch the default complete model from
+		-- sourcegraph
+		model = "fireworks::v1::deepseek-coder-v2-lite-base",
 
-return M
+		-- The proportion of the token limit that should used for context.
+		context_proportion = 0.9,
+	},
+
+}