free/gemini
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Restore chat history

Browse Source
This commit is contained in:
Jonas Widen 2025-03-16 19:30:15 +01:00
parent fa6a890db8
commit 34c22ed63b
1 changed files with 29 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
lua/gemini/chat.lua
View File

@ -113,29 +113,39 @@ end
function M.update_content(content, is_new_chat, is_thinking)
vim.api.nvim_buf_set_option(state.bufnr, 'modifiable', true)
local lines = vim.split(content, "\n")
-- Only add separator if:
-- 1. Not the first message
-- 2. Not a thinking message
-- 3. There's existing content in the buffer
if not is_thinking and not is_new_chat and vim.api.nvim_buf_line_count(state.bufnr) > 0 then
local separator = "━━━━━━━━━━━━━━━━━━━━━━━━━━"
table.insert(lines, 1, separator)
end
if is_new_chat then
-- If it's a new chat but we have history, append to existing content
if #state.conversation_history > 0 and not is_thinking then
local current_lines = vim.api.nvim_buf_get_lines(state.bufnr, 0, -1, false)
for i = #current_lines, 1, -1 do
table.insert(lines, 1, current_lines[i])
end
end
if is_thinking then
-- For thinking message, just replace/append without affecting history
vim.api.nvim_buf_set_lines(state.bufnr, 0, -1, false, lines)
else
vim.api.nvim_buf_set_lines(state.bufnr, -1, -1, false, lines)
if is_new_chat then
-- For new messages, show the complete history
local display_lines = {}
for i, msg in ipairs(state.conversation_history) do
if i > 1 then -- Add separator before messages (except first)
table.insert(display_lines, "━━━━━━━━━━━━━━━━━━━━━━━━━━")
end
local msg_lines = vim.split(msg.content, "\n")
for _, line in ipairs(msg_lines) do
table.insert(display_lines, line)
end
end
-- Add the new content
if #state.conversation_history > 0 then
table.insert(display_lines, "━━━━━━━━━━━━━━━━━━━━━━━━━━")
end
for _, line in ipairs(lines) do
table.insert(display_lines, line)
end
vim.api.nvim_buf_set_lines(state.bufnr, 0, -1, false, display_lines)
else
-- For subsequent messages in the same chat
if #state.conversation_history > 0 then
table.insert(lines, 1, "━━━━━━━━━━━━━━━━━━━━━━━━━━")
end
vim.api.nvim_buf_set_lines(state.bufnr, -1, -1, false, lines)
end
end
vim.api.nvim_buf_set_option(state.bufnr, 'modifiable', false)