fix: more readable value name (#1521)
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user