diff --git a/lua/avante/config.lua b/lua/avante/config.lua index 609892a..6ba3cc1 100644 --- a/lua/avante/config.lua +++ b/lua/avante/config.lua @@ -574,6 +574,7 @@ M.BASE_PROVIDER_KEYS = { "__inherited_from", "disable_tools", "entra", + "hide_in_model_selector", } return M diff --git a/lua/avante/model_selector.lua b/lua/avante/model_selector.lua index 2b1616f..76b7736 100644 --- a/lua/avante/model_selector.lua +++ b/lua/avante/model_selector.lua @@ -21,8 +21,11 @@ function M.open() -- Collect models from main providers and vendors for _, provider_name in ipairs(Config.provider_names) do - local entry = create_model_entry(provider_name, Config.get_provider_config(provider_name)) + local cfg = Config.get_provider_config(provider_name) + if cfg.hide_in_model_selector then goto continue end + local entry = create_model_entry(provider_name, cfg) if entry then table.insert(models, entry) end + ::continue:: end if #models == 0 then diff --git a/lua/avante/types.lua b/lua/avante/types.lua index 3d152cf..e4b1dda 100644 --- a/lua/avante/types.lua +++ b/lua/avante/types.lua @@ -205,6 +205,7 @@ vim.g.avante_login = vim.g.avante_login ---@field _shellenv? string ---@field disable_tools? boolean ---@field entra? boolean +---@field hide_in_model_selector? boolean --- ---@class AvanteSupportedProvider: AvanteDefaultBaseProvider ---@field __inherited_from? string