feat: keep the history content in the resp buffer (#807)

This commit is contained in:
yetone
2024-11-05 21:02:38 +08:00
committed by GitHub
parent 0aa2659580
commit 9fbe0256f2
2 changed files with 39 additions and 13 deletions

View File

@@ -1,5 +1,6 @@
local fn, api = vim.fn, vim.api
local Utils = require("avante.utils")
local LRUCache = require("avante.utils.lru_cache")
local Path = require("plenary.path")
local Scan = require("plenary.scandir")
local Config = require("avante.config")
@@ -27,6 +28,8 @@ end
---@param mode LlmMode
H.get_mode_file = function(mode) return string.format("custom.%s.avanterules", mode) end
local history_file_cache = LRUCache:new(12)
-- History path
local History = {}
@@ -39,11 +42,16 @@ History.get = function(bufnr) return Path:new(Config.history.storage_path):joinp
---@param bufnr integer
History.load = function(bufnr)
local history_file = History.get(bufnr)
local cached_key = tostring(history_file:absolute())
local cached_value = history_file_cache:get(cached_key)
if cached_value ~= nil then return cached_value end
local value = {}
if history_file:exists() then
local content = history_file:read()
return content ~= nil and vim.json.decode(content) or {}
value = content ~= nil and vim.json.decode(content) or {}
end
return {}
history_file_cache:set(cached_key, value)
return value
end
-- Saves the chat history for the given buffer.
@@ -51,7 +59,9 @@ end
---@param history table
History.save = vim.schedule_wrap(function(bufnr, history)
local history_file = History.get(bufnr)
local cached_key = tostring(history_file:absolute())
history_file:write(vim.json.encode(history), "w")
history_file_cache:set(cached_key, history)
end)
P.history = History