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

* 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

View File

@@ -80,7 +80,7 @@ local Dressing = require("avante.ui.dressing")
---@field parse_curl_args AvanteCurlArgsParser
---@field setup? fun(): nil
---@field has fun(): boolean
---@field API_KEY string
---@field api_key_name string
---@field parse_stream_data? AvanteStreamParser
---
---@class avante.Providers
@@ -102,13 +102,10 @@ setmetatable(M, {
local v = Config.vendors[k]
-- Patch from vendors similar to supported providers.
---@type AvanteProviderFunctor
t[k] = setmetatable({}, { __index = v })
t[k].API_KEY = v.api_key_name
-- Hack for aliasing and makes it sane for us.
t[k].parse_response = v.parse_response_data
t[k].has = function()
return os.getenv(v.api_key_name) and true or false
end
return t[k]
end
@@ -130,7 +127,7 @@ E._once = false
---@param opts {refresh: boolean, provider: AvanteProviderFunctor}
---@private
E.setup = function(opts)
local var = opts.provider.API_KEY
local var = opts.provider.api_key_name
if var == M.AVANTE_INTERNAL_KEY then
return