From 8d82abe04358515d4bc8385acc6c5847befb717e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Thomas=20M=C3=BCller?= Date: Tue, 11 Feb 2025 04:35:28 +0100 Subject: [PATCH] feat: expose whether the llm is currently generating content (#1181) --- lua/avante/sidebar.lua | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/lua/avante/sidebar.lua b/lua/avante/sidebar.lua index 815529e..38a5a85 100644 --- a/lua/avante/sidebar.lua +++ b/lua/avante/sidebar.lua @@ -57,6 +57,7 @@ function Sidebar:new(id) selected_files_container = nil, input_container = nil, file_selector = FileSelector:new(id), + is_generating = false, }, { __index = self }) end @@ -1836,6 +1837,8 @@ function Sidebar:create_input_container(opts) ---@type AvanteLLMChunkCallback local on_chunk = function(chunk) + self.is_generating = true + original_response = original_response .. chunk local selected_files = self.file_selector:get_selected_files_contents() @@ -1870,6 +1873,8 @@ function Sidebar:create_input_container(opts) ---@type AvanteLLMStopCallback local on_stop = function(stop_opts) + self.is_generating = false + pcall(function() ---remove keymaps vim.keymap.del("n", "j", { buffer = self.result_container.bufnr })