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

fix: bedrock (#1511)

Browse Source
This commit is contained in:
yetone
2025-03-07 00:30:30 +08:00
committed by GitHub
parent 8620ea3e12
commit 25dee44de9
3 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lua/avante/providers/bedrock.lua
View File

@@ -7,6 +7,13 @@ local M = {}
M.api_key_name = "BEDROCK_KEYS"
M.use_xml_format = true
setmetatable(M, {
__index = function(_, k)
local model_handler = M.load_model_handler()
return model_handler[k]
end,
})
function M.load_model_handler()
local provider_conf, _ = P.parse_config(P["bedrock"])
local bedrock_model = provider_conf.model
@@ -18,6 +25,11 @@ function M.load_model_handler()
error(error_msg)
end
function M:parse_messages(prompt_opts)
local model_handler = M.load_model_handler()
return model_handler.parse_messages(self, prompt_opts)
end
function M:parse_response(ctx, data_stream, event_state, opts)
local model_handler = M.load_model_handler()
return model_handler.parse_response(self, ctx, data_stream, event_state, opts)