From cd9412514ea013b6b0e884b44c14952c7734e2d1 Mon Sep 17 00:00:00 2001 From: Aaron Pham Date: Tue, 27 Aug 2024 18:37:19 -0400 Subject: [PATCH] fix(llm): local-first check (#297) Signed-off-by: Aaron Pham --- lua/avante/providers/init.lua | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/lua/avante/providers/init.lua b/lua/avante/providers/init.lua index 879fdf9..47336ac 100644 --- a/lua/avante/providers/init.lua +++ b/lua/avante/providers/init.lua @@ -169,16 +169,17 @@ end ---@param opts {refresh: boolean, provider: AvanteProviderFunctor} ---@private E.setup = function(opts) + if opts.provider["local"] then + vim.g.avante_login = true + return + end + local var = opts.provider.api_key_name opts.provider.setup() - if opts.provider["local"] then - vim.g.avante_login = true - end - -- check if var is a all caps string - if var == M.AVANTE_INTERNAL_KEY or var:match("^cmd:(.*)") or opts.provider["local"] == true then + if var == M.AVANTE_INTERNAL_KEY or var:match("^cmd:(.*)") then return end @@ -280,7 +281,9 @@ M = setmetatable(M, { if t[k].setup == nil then t[k].setup = function() - t[k].parse_api_key() + if not E.is_local(k) then + t[k].parse_api_key() + end end end