diff --git a/lua/avante/config.lua b/lua/avante/config.lua index f53f53c..31031c4 100644 --- a/lua/avante/config.lua +++ b/lua/avante/config.lua @@ -537,6 +537,7 @@ M.BASE_PROVIDER_KEYS = { "allow_insecure", "api_key_name", "timeout", + "display_name", -- internal "local", "_shellenv", diff --git a/lua/avante/model_selector.lua b/lua/avante/model_selector.lua index 9229fa4..7afc4f0 100644 --- a/lua/avante/model_selector.lua +++ b/lua/avante/model_selector.lua @@ -8,11 +8,12 @@ local M = {} ---@param cfg table ---@return table? local function create_model_entry(provider, cfg) - return cfg.model and { - name = provider .. "/" .. cfg.model, - provider = provider, - model = cfg.model, - } + return cfg.model + and { + name = cfg.display_name or (provider .. "/" .. cfg.model), + provider = provider, + model = cfg.model, + } end function M.open() diff --git a/lua/avante/types.lua b/lua/avante/types.lua index ad5e4ae..646e572 100644 --- a/lua/avante/types.lua +++ b/lua/avante/types.lua @@ -215,6 +215,7 @@ vim.g.avante_login = vim.g.avante_login ---@field temperature? number ---@field max_tokens? number ---@field reasoning_effort? string +---@field display_name? string --- ---@class AvanteLLMUsage ---@field input_tokens number