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

fix: missing event_state in handler options of LLM stream function (#77)

Browse Source
This commit is contained in:
yetone 2024-08-19 11:18:59 +08:00 committed by GitHub
parent 29dabc4768
commit b71e43d69e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lua/avante/llm.lua
View File

@ -481,7 +481,6 @@ local active_job = nil
---@param on_complete fun(err: string|nil): any ---@param on_complete fun(err: string|nil): any
M.stream = function(question, code_lang, code_content, selected_content_content, on_chunk, on_complete) M.stream = function(question, code_lang, code_content, selected_content_content, on_chunk, on_complete)
local provider = Config.provider local provider = Config.provider
local event_state = nil
local code_opts = { local code_opts = {
question = question, question = question,
@ -489,7 +488,7 @@ M.stream = function(question, code_lang, code_content, selected_content_content,
code_content = code_content, code_content = code_content,
selected_code_content = selected_content_content, selected_code_content = selected_content_content,
} }
local handler_opts = vim.deepcopy({ on_chunk = on_chunk, on_complete = on_complete, event_state = event_state }, true) local handler_opts = { on_chunk = on_chunk, on_complete = on_complete, event_state = nil }
---@type AvanteCurlOutput ---@type AvanteCurlOutput
local spec = nil local spec = nil
@ -513,7 +512,7 @@ M.stream = function(question, code_lang, code_content, selected_content_content,
local function parse_and_call(line) local function parse_and_call(line)
local event = line:match("^event: (.+)$") local event = line:match("^event: (.+)$")
if event then if event then
event_state = event handler_opts.event_state = event
return return
end end
local data_match = line:match("^data: (.+)$") local data_match = line:match("^data: (.+)$")