From 56576c2481d82af8710d2111fef6cb2ecc57da32 Mon Sep 17 00:00:00 2001 From: Sang <49085035+gitsang@users.noreply.github.com> Date: Fri, 9 May 2025 14:00:33 +0800 Subject: [PATCH] feature: Allow users to define custom parse_api_key functions. (#1914) Co-authored-by: yetone --- lua/avante/providers/init.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lua/avante/providers/init.lua b/lua/avante/providers/init.lua index 645aaed..7e1680c 100644 --- a/lua/avante/providers/init.lua +++ b/lua/avante/providers/init.lua @@ -183,12 +183,12 @@ M = setmetatable(M, { t[k] = Utils.deep_extend_with_metatable("keep", provider_config, module) end - t[k].parse_api_key = function() return E.parse_envvar(t[k]) end + if t[k].parse_api_key == nil then t[k].parse_api_key = function() return E.parse_envvar(t[k]) end end -- default to gpt-4o as tokenizer if t[k].tokenizer_id == nil then t[k].tokenizer_id = "gpt-4o" end - if t[k].is_env_set == nil then t[k].is_env_set = function() return E.parse_envvar(t[k]) ~= nil end end + if t[k].is_env_set == nil then t[k].is_env_set = function() return t[k].parse_api_key() ~= nil end end if t[k].setup == nil then local provider_conf = M.parse_config(t[k])