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

fix: bedrock claude do not support prompt caching (#1507)

Browse Source
This commit is contained in:
yetone
2025-03-06 18:31:56 +08:00
committed by GitHub
parent 6d39e06f57
commit 5aa55689ff
12 changed files with 85 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lua/avante/providers/cohere.lua
View File

@@ -47,7 +47,9 @@ M.role_map = {
assistant = "assistant",
}
function M.parse_messages(opts)
function M:is_disable_stream() return false end
function M:parse_messages(opts)
local messages = {
{ role = "system", content = opts.system_prompt },
}
@@ -57,7 +59,7 @@ function M.parse_messages(opts)
return { messages = messages }
end
function M.parse_stream_data(ctx, data, opts)
function M:parse_stream_data(ctx, data, opts)
---@type CohereChatResponse
local json = vim.json.decode(data)
if json.type ~= nil then
@@ -69,7 +71,7 @@ function M.parse_stream_data(ctx, data, opts)
end
end
function M.parse_curl_args(provider, prompt_opts)
function M:parse_curl_args(provider, prompt_opts)
local provider_conf, request_body = P.parse_config(provider)
local headers = {
@@ -92,7 +94,7 @@ function M.parse_curl_args(provider, prompt_opts)
body = vim.tbl_deep_extend("force", {
model = provider_conf.model,
stream = true,
}, M.parse_messages(prompt_opts), request_body),
}, self:parse_messages(prompt_opts), request_body),
}
end