From 050f9bd3b5794890fb9e3cc7103385c8c8147fc4 Mon Sep 17 00:00:00 2001
From: Jonas Widen <jonas.widen@widens.eu>
Date: Tue, 18 Mar 2025 18:32:06 +0100
Subject: [PATCH] Review and cleanup

---
 lua/gemini/completion.lua | 35 +++++++++++++++++++----------------
 1 file changed, 19 insertions(+), 16 deletions(-)

diff --git a/lua/gemini/completion.lua b/lua/gemini/completion.lua
index 2c069d7..9d92f86 100644
--- a/lua/gemini/completion.lua
+++ b/lua/gemini/completion.lua
@@ -274,24 +274,30 @@ function M.trigger_completion()
     if not vim.b.gemini_completion_enabled then return end
     if vim.fn.pumvisible() ~= 0 then return end
     
-    -- Get current line and cursor position
     local cursor = vim.api.nvim_win_get_cursor(0)
     local line = cursor[1] - 1
     local col = cursor[2]
     local current_line = vim.api.nvim_get_current_line()
     
-    -- Check filetype exclusions
-    local config = require("gemini.config")
-    if vim.tbl_contains(config.options.completion.exclude_filetypes, vim.bo.filetype) then
-        return
-    end
-    
     -- Get context
-    local visible_lines, relative_cursor = get_visible_lines()
-    local context = get_context_around_cursor(visible_lines, relative_cursor, config.options.completion.max_context_lines)
+    local visible_lines = vim.api.nvim_buf_get_lines(0, math.max(0, line - 10), line + 10, false)
+    local context, detected_file_type = get_context_around_cursor(visible_lines, 10, current_suggestion.max_context_lines)
     
-    -- Create prompt for API
-    local prompt = table.concat(context, "\n") .. "\nPlease complete the following line:\n" .. current_line
+    -- Create a very specific prompt for raw completion
+    local prompt = string.format([[
+You are an autocomplete engine. Respond ONLY with the direct completion text.
+DO NOT include:
+- Explanations
+- Markdown formatting
+- Code block markers
+- Additional newlines
+- Any other text
+
+Language: %s
+Context:
+%s
+Complete this line:
+%s]], detected_file_type, table.concat(context, "\n"), current_line)
     
     -- Make API request
     api.get_response(prompt, nil, function(response, error)
@@ -301,15 +307,12 @@ function M.trigger_completion()
         end
         
         if response then
-            -- Extract the completion suggestion
-            local suggestion = response:gsub("^%s*(.-)%s*$", "%1") -- Trim whitespace
-            
-            -- Show the suggestion
+            -- Show the raw suggestion directly
             vim.schedule(function()
                 -- Only show if cursor position hasn't changed
                 local new_cursor = vim.api.nvim_win_get_cursor(0)
                 if new_cursor[1] - 1 == line and new_cursor[2] == col then
-                    show_suggestion(suggestion, col)
+                    show_suggestion(response, col)
                 end
             end)
         end