From 232d4b1340bf3d1d9a97a15fee80b46980320e19 Mon Sep 17 00:00:00 2001
From: Grimaldi Baptiste <79135575+GrimalDev@users.noreply.github.com>
Date: Wed, 5 Mar 2025 11:25:48 +0000
Subject: [PATCH] feature model selection custom display name (#1477)

* feature model selection custom display name

Added an option for a custom display name
I had the need for this option for multiple similar custom model
configurations where I wanted to have more indicative titles.

* linting error
---
 lua/avante/config.lua         |  1 +
 lua/avante/model_selector.lua | 11 ++++++-----
 lua/avante/types.lua          |  1 +
 3 files changed, 8 insertions(+), 5 deletions(-)

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