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

12
lua/avante/llm.lua
View File

@@ -232,7 +232,7 @@ function M._stream(opts)
}
---@type AvanteCurlOutput
local spec = provider.parse_curl_args(provider, prompt_opts)
local spec = provider:parse_curl_args(provider, prompt_opts)
local resp_ctx = {}
@@ -244,11 +244,11 @@ function M._stream(opts)
return
end
local data_match = line:match("^data: (.+)$")
if data_match then provider.parse_response(resp_ctx, data_match, current_event_state, handler_opts) end
if data_match then provider:parse_response(resp_ctx, data_match, current_event_state, handler_opts) end
end
local function parse_response_without_stream(data)
provider.parse_response_without_stream(data, current_event_state, handler_opts)
provider:parse_response_without_stream(data, current_event_state, handler_opts)
end
local completed = false
@@ -287,10 +287,10 @@ function M._stream(opts)
{ once = true }
)
end
provider.parse_stream_data(resp_ctx, data, handler_opts)
provider:parse_stream_data(resp_ctx, data, handler_opts)
else
if provider.parse_stream_data ~= nil then
provider.parse_stream_data(resp_ctx, data, handler_opts)
provider:parse_stream_data(resp_ctx, data, handler_opts)
else
parse_stream_data(data)
end
@@ -357,7 +357,7 @@ function M._stream(opts)
end
-- If stream is not enabled, then handle the response here
if (spec.body.stream == nil or spec.body.stream == false) and result.status == 200 then
if provider:is_disable_stream() and result.status == 200 then
vim.schedule(function()
completed = true
parse_response_without_stream(result.body)