fix: bedrock (#1524)
* fix: bedrock * fix: bad variable name * fix: missing metatable
This commit is contained in:
@@ -23,19 +23,19 @@ M.parse_response = Claude.parse_response
|
||||
|
||||
---@param provider AvanteProviderFunctor
|
||||
---@param prompt_opts AvantePromptOptions
|
||||
---@param body_opts table
|
||||
---@param request_body table
|
||||
---@return table
|
||||
function M.build_bedrock_payload(provider, prompt_opts, body_opts)
|
||||
function M.build_bedrock_payload(provider, prompt_opts, request_body)
|
||||
local system_prompt = prompt_opts.system_prompt or ""
|
||||
local messages = provider:parse_messages(prompt_opts)
|
||||
local max_tokens = body_opts.max_tokens or 2000
|
||||
local max_tokens = request_body.max_tokens or 2000
|
||||
local payload = {
|
||||
anthropic_version = "bedrock-2023-05-31",
|
||||
max_tokens = max_tokens,
|
||||
messages = messages,
|
||||
system = system_prompt,
|
||||
}
|
||||
return vim.tbl_deep_extend("force", payload, body_opts or {})
|
||||
return vim.tbl_deep_extend("force", payload, request_body or {})
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
Reference in New Issue
Block a user