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

fix(cohere): fix history messages (#2031)

Browse Source
This commit is contained in:
Maxime Brunet
2025-05-12 11:22:56 +00:00
committed by GitHub
parent b6a3c925a4
commit 786d95464c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -64,10 +64,15 @@ function M:parse_stream_data(ctx, data, opts)
local json = vim.json.decode(data) local json = vim.json.decode(data)
if json.type ~= nil then if json.type ~= nil then
if json.type == "message-end" and json.delta.finish_reason == "COMPLETE" then if json.type == "message-end" and json.delta.finish_reason == "COMPLETE" then
P.openai:finish_pending_messages(ctx, opts)
opts.on_stop({ reason = "complete" }) opts.on_stop({ reason = "complete" })
return return
end end
if json.type == "content-delta" then opts.on_chunk(json.delta.message.content.text) end if json.type == "content-delta" then
local content = json.delta.message.content.text
P.openai:add_text_message(ctx, content, "generating", opts)
if content and content ~= "" and opts.on_chunk then opts.on_chunk(content) end
end
end end
end end