fix: make O1 work with new change

yetone · Nov 6, 2024 · 0534ff5 · 0534ff5
1 parent c7ff2a9
commit 0534ff5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lua/avante/sidebar.lua b/lua/avante/sidebar.lua
@@ -1354,10 +1354,11 @@ function Sidebar:create_input(opts)
       if is_first_chunk then
         is_first_chunk = false
         self:update_content(content_prefix .. cur_displayed_response, { stream = false, scroll = true })
-        return
+      else
+        local suffix = get_display_content_suffix(transformed)
+        self:update_content(content_prefix .. cur_displayed_response .. suffix, { scroll = true })
       end
-      local suffix = get_display_content_suffix(transformed)
-      self:update_content(content_prefix .. cur_displayed_response .. suffix, { scroll = true })
+
       vim.schedule(function() vim.cmd("redraw") end)
       displayed_response = cur_displayed_response
     end