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

Minor fix

Browse Source
This commit is contained in:
Jonas Widen 2025-03-16 18:18:14 +01:00
parent 03502d4601
commit 554dbab2ed
1 changed files with 24 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
lua/gemini/init.lua
View File

@ -161,14 +161,25 @@ local function gemini_query(prompt, context)
-- Store the context for subsequent queries -- Store the context for subsequent queries
current_context = context current_context = context
-- Show initial message in chat window -- Show initial message in chat window and ensure it's visible
local initial_content = "\nUser: " .. prompt .. "\n\nAssistant: Thinking..." local initial_content = "\nUser: " .. prompt .. "\n\nAssistant: Thinking..."
update_chat_window(initial_content) update_chat_window(initial_content)
-- Force Neovim to update the screen
vim.cmd('redraw')
local response = api.get_response(prompt, context) local response = api.get_response(prompt, context)
if response then if response then
local formatted_content = "\nUser: " .. prompt .. "\n\nAssistant: " .. response -- Replace "Thinking..." with the actual response
update_chat_window(formatted_content) local lines = vim.api.nvim_buf_get_lines(chat_bufnr, 0, -1, false)
for i = #lines, 1, -1 do
if lines[i]:match("^Assistant: Thinking...") then
vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', true)
vim.api.nvim_buf_set_lines(chat_bufnr, i, i + 1, false, {"Assistant: " .. response})
vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', false)
break
end
end
-- Move focus to chat window -- Move focus to chat window
if chat_winnr and vim.api.nvim_win_is_valid(chat_winnr) then if chat_winnr and vim.api.nvim_win_is_valid(chat_winnr) then
@ -178,9 +189,16 @@ local function gemini_query(prompt, context)
-- Clear the command line -- Clear the command line
vim.cmd('echo ""') vim.cmd('echo ""')
else else
-- Update the chat window with error message -- Replace "Thinking..." with error message
local error_content = "\nUser: " .. prompt .. "\n\nAssistant: Failed to get response from Gemini API" local lines = vim.api.nvim_buf_get_lines(chat_bufnr, 0, -1, false)
update_chat_window(error_content) for i = #lines, 1, -1 do
if lines[i]:match("^Assistant: Thinking...") then
vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', true)
vim.api.nvim_buf_set_lines(chat_bufnr, i, i + 1, false, {"Assistant: Failed to get response from Gemini API"})
vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', false)
break
end
end
vim.notify("Failed to get a response from Gemini API", vim.log.levels.ERROR) vim.notify("Failed to get a response from Gemini API", vim.log.levels.ERROR)
end end
end end