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

fix(claude): there can be no more than four cache controls (#425)

Browse Source
This commit is contained in:
yetone 2024-08-31 23:04:51 +08:00 committed by GitHub
parent d5a4db8321
commit 33192127a3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
lua/avante/providers/claude.lua
View File

@ -24,13 +24,35 @@ M.parse_message = function(prompt_opts)
end
end
for _, user_prompt in ipairs(prompt_opts.user_prompts) do
local user_prompts_with_length = {}
for idx, user_prompt in ipairs(prompt_opts.user_prompts) do
table.insert(user_prompts_with_length, { idx = idx, length = #user_prompt })
end
table.sort(user_prompts_with_length, function(a, b)
return a.length > b.length
end)
local top_three = vim.list_slice(user_prompts_with_length, 1, 3)
for idx, prompt_data in ipairs(prompt_opts.user_prompts) do
local user_prompt_obj = {
type = "text",
text = user_prompt,
cache_control = { type = "ephemeral" },
text = prompt_data,
}
local is_top_three = false
for _, top in ipairs(top_three) do
if top.idx == idx then
is_top_three = true
break
end
end
if is_top_three then
user_prompt_obj.cache_control = { type = "ephemeral" }
end
table.insert(message_content, user_prompt_obj)
end