feat: history manager (#1644)

lua/avante/llm.lua

@@ -19,6 +19,44 @@ M.CANCEL_PATTERN = "AvanteLLMEscape"
 
 local group = api.nvim_create_augroup("avante_llm", { clear = true })
 
+---@param content AvanteLLMMessageContent
+---@param cb fun(title: string | nil): nil
+function M.summarize_chat_thread_title(content, cb)
+  local system_prompt =
+    [[Summarize the content as a title for the chat thread. The title should be a concise and informative summary of the conversation, capturing the main points and key takeaways. It should be no longer than 100 words and should be written in a clear and engaging style. The title should be suitable for use as the title of a chat thread on a messaging platform or other communication medium.]]
+  local response_content = ""
+  local provider = Providers[Config.memory_summary_provider or Config.provider]
+  M.curl({
+    provider = provider,
+    prompt_opts = {
+      system_prompt = system_prompt,
+      messages = {
+        { role = "user", content = content },
+      },
+    },
+    handler_opts = {
+      on_start = function(_) end,
+      on_chunk = function(chunk)
+        if not chunk then return end
+        response_content = response_content .. chunk
+      end,
+      on_stop = function(stop_opts)
+        if stop_opts.error ~= nil then
+          Utils.error(string.format("summarize failed: %s", vim.inspect(stop_opts.error)))
+          return
+        end
+        if stop_opts.reason == "complete" then
+          response_content = Utils.trim_think_content(response_content)
+          response_content = Utils.trim(response_content, { prefix = "\n", suffix = "\n" })
+          response_content = Utils.trim(response_content, { prefix = '"', suffix = '"' })
+          local title = response_content
+          cb(title)
+        end
+      end,
+    },
+  })
+end
+
 ---@param bufnr integer
 ---@param history avante.ChatHistory
 ---@param cb fun(memory: avante.ChatMemory | nil): nil