fix: more readable value name (#1521)

This commit is contained in:
yetone
2025-03-07 20:27:11 +08:00
committed by GitHub
parent 6c7c5464ca
commit 4d983532a1
6 changed files with 39 additions and 35 deletions

View File

@@ -4,14 +4,14 @@ local Config = require("avante.config")
---@class avante.ModelSelector
local M = {}
---@param provider string
---@param provider_name string
---@param cfg table
---@return table?
local function create_model_entry(provider, cfg)
local function create_model_entry(provider_name, cfg)
return cfg.model
and {
name = cfg.display_name or (provider .. "/" .. cfg.model),
provider = provider,
name = cfg.display_name or (provider_name .. "/" .. cfg.model),
provider_name = provider_name,
model = cfg.model,
}
end
@@ -20,8 +20,8 @@ function M.open()
local models = {}
-- Collect models from main providers and vendors
for _, provider in ipairs(Config.providers) do
local entry = create_model_entry(provider, Config.get_provider(provider))
for _, provider_name in ipairs(Config.provider_names) do
local entry = create_model_entry(provider_name, Config.get_provider(provider_name))
if entry then table.insert(models, entry) end
end
@@ -37,11 +37,15 @@ function M.open()
if not choice then return end
-- Switch provider if needed
if choice.provider ~= Config.provider then require("avante.providers").refresh(choice.provider) end
if choice.provider_name ~= Config.provider then require("avante.providers").refresh(choice.provider_name) end
-- Update config with new model
Config.override({
[choice.provider] = vim.tbl_deep_extend("force", Config.get_provider(choice.provider), { model = choice.model }),
[choice.provider_name] = vim.tbl_deep_extend(
"force",
Config.get_provider(choice.provider_name),
{ model = choice.model }
),
})
Utils.info("Switched to model: " .. choice.name)