fix: check on_chunk if it is a nil (#1998)

2025-05-07 01:02:11 +08:00
parent c29075ae02
commit 3ab2a5d68e
1 changed files with 2 additions and 1 deletions
--- a/lua/avante/providers/openai.lua
+++ b/lua/avante/providers/openai.lua
@@ -335,7 +335,8 @@ function M:parse_response_without_stream(data, _, opts)
  if json.choices and json.choices[1] then
    local choice = json.choices[1]
    if choice.message and choice.message.content then
-      opts.on_chunk(choice.message.content)
+      if opts.on_chunk then opts.on_chunk(choice.message.content) end
+      self:add_text_message({}, choice.message.content, "generated", opts)
      vim.schedule(function() opts.on_stop({ reason = "complete" }) end)
    end
  end