CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(llm): persistent key check for override class (#158)

Browse Source 
* fix(llm): make sure to allow passing custom module

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>

* fix: correct custom class

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>

* fix: correct attribute

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>

---------

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
This commit is contained in:
Aaron Pham
2024-08-22 23:52:49 -04:00
committed by GitHub
parent 49fabfc358
commit 6475407d0d
9 changed files with 24 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lua/avante/providers/claude.lua
View File

@@ -6,10 +6,10 @@ local P = require("avante.providers")
---@class AvanteProviderFunctor
local M = {}
M.API_KEY = "ANTHROPIC_API_KEY"
M.api_key_name = "ANTHROPIC_API_KEY"
M.has = function()
return os.getenv(M.API_KEY) and true or false
return os.getenv(M.api_key_name) and true or false
end
M.parse_message = function(opts)
@@ -93,7 +93,7 @@ M.parse_curl_args = function(provider, code_opts)
["anthropic-beta"] = "prompt-caching-2024-07-31",
}
if not P.env.is_local("claude") then
headers["x-api-key"] = os.getenv(M.API_KEY)
headers["x-api-key"] = os.getenv(base.api_key_name or M.api_key_name)
end
return {