From 6dbe3c34090b8c894f9808ef78e3bc697d2c97cd Mon Sep 17 00:00:00 2001 From: Jonas Widen Date: Sun, 16 Mar 2025 18:03:41 +0100 Subject: [PATCH] Keep which state chat was started in --- lua/gemini/init.lua | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/lua/gemini/init.lua b/lua/gemini/init.lua index 57e294c..82b101b 100644 --- a/lua/gemini/init.lua +++ b/lua/gemini/init.lua @@ -6,6 +6,7 @@ local M = {} -- Store the buffer number of the chat window local chat_bufnr = nil local chat_winnr = nil +local current_context = nil -- Store the current context local function get_current_buffer_content() local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false) @@ -109,11 +110,11 @@ local function update_chat_window(new_content) vim.cmd('wincmd p') end, { buffer = chat_bufnr, nowait = true }) - -- Add input mapping + -- Add input mapping with context awareness vim.keymap.set('n', 'i', function() vim.ui.input({ prompt = "Gemini: " }, function(input) if input then - M.gemini_query(input) + M.gemini_query(input, current_context) end end) end, { buffer = chat_bufnr, nowait = true }) @@ -157,6 +158,9 @@ local function update_chat_window(new_content) end local function gemini_query(prompt, context) + -- Store the context for subsequent queries + current_context = context + local response = api.get_response(prompt, context) if response then local formatted_content = "\nUser: " .. prompt .. "\n\nAssistant: " .. response