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 20:15:43 +01:00
parent b950be8cd0
commit 12c973e6a7
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lua/gemini/chat.lua
View File

@ -121,16 +121,16 @@ function M.update_content(content, is_new_chat, is_thinking)
else else
-- For actual messages -- For actual messages
local display_lines = {} local display_lines = {}
local last_message_start_line = 1 local current_query_line = 1
-- Add all messages from history -- Add all messages from history
for i, msg in ipairs(state.conversation_history) do for i, msg in ipairs(state.conversation_history) do
if i > 1 then -- Add separator before messages (except first) if i > 1 then -- Add separator before messages (except first)
table.insert(display_lines, "━━━━━━━━━━━━━━━━━━━━━━━━━━") table.insert(display_lines, "━━━━━━━━━━━━━━━━━━━━━━━━━━")
end end
-- Keep track of where the last message starts -- Keep track of where the current query starts
if i == #state.conversation_history then if i == #state.conversation_history - 1 then -- User's query is second-to-last message
last_message_start_line = #display_lines + 1 current_query_line = #display_lines + 1
end end
-- Add prefix based on role -- Add prefix based on role
local prefix = msg.role == "user" and "User: " or "Assistant: " local prefix = msg.role == "user" and "User: " or "Assistant: "
@ -149,8 +149,8 @@ function M.update_content(content, is_new_chat, is_thinking)
-- Set the buffer content -- Set the buffer content
if #display_lines > 0 then if #display_lines > 0 then
vim.api.nvim_buf_set_lines(state.bufnr, 0, -1, false, display_lines) vim.api.nvim_buf_set_lines(state.bufnr, 0, -1, false, display_lines)
-- Scroll to the start of the last message -- Scroll to the current query
vim.api.nvim_win_set_cursor(state.winnr, {last_message_start_line, 0}) vim.api.nvim_win_set_cursor(state.winnr, {current_query_line, 0})
vim.cmd('normal! zt') vim.cmd('normal! zt')
end end
end end