From 9981f45c7450461d46e6b7ed3d44dbef2a62a5dd Mon Sep 17 00:00:00 2001
From: Jonas Widen <jonas.widen@widens.eu>
Date: Sun, 16 Mar 2025 18:19:25 +0100
Subject: [PATCH] Minor fix

---
 lua/gemini/init.lua | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lua/gemini/init.lua b/lua/gemini/init.lua
index 7e8928d..5914ad1 100644
--- a/lua/gemini/init.lua
+++ b/lua/gemini/init.lua
@@ -175,7 +175,10 @@ local function gemini_query(prompt, context)
         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})
+                -- Split response into lines and insert "Assistant: " at the start of first line
+                local response_lines = vim.split(response, "\n")
+                response_lines[1] = "Assistant: " .. response_lines[1]
+                vim.api.nvim_buf_set_lines(chat_bufnr, i, i + 1, false, response_lines)
                 vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', false)
                 break
             end