Release v0.1.1: UI Polish and Bug Fixes

Release v0.1.1: UI Polish and Bug Fixes

Author: acris-software

issue_tracker.md

@@ -40,4 +40,26 @@
 - Add config location for prompt box (3 settings, left, center, right)
 - Clean up UI/UX and ensure user prompt scales with the users typing.
 
+## Issue 5: Inconsistent Prompt Box Display** (Target: 0.1.2)
+   - Problem: Prompt box only displays properly on first :Grok; reverts to old style after.
+   - Analysis: Likely state issue in chat/init.lua or ui/window.lua; floating input not reinitialized.
+   - Solution: Ensure util.create_floating_input called consistently on each prompt.
+   - Status: [ ] Not implemented.
 
+## Issue 6: Verify Max Prompt Length for Models** (Target: 0.1.1)
+   - Problem: Current max 131072 for Grok-3-mini; confirm accuracy.
+   - Analysis: From x.ai docs, Grok-3-mini context window is 131072 tokens (confirmed).
+   - Solution: Update util.get_model_max_tokens with verified values; add more models.
+   - Status: [✓] Verified (131072 tokens); implement dynamic if API provides.
+
+## Issue 7: Grok Prevents Smooth Neovim Exit** (Target: 0.1.2)
+   - Problem: Requires :qa! to exit Neovim after using Grok.
+   - Analysis: Likely unclosed buffers/windows or autocmds; check ui/window.lua close logic.
+   - Solution: Ensure close_chat_window cleans up properly; add vim.api.nvim_buf_delete on exit.
+   - Status: [ ] Not implemented.
+
+## Issue 8: Expand and Modularize Logging (Target: v0.1.3)
+   - **Problem:** Debug logs print regardless of config.debug setting, hurting performance; logging is monolithic in single core.lua file.
+   - **Analysis:** v0.1.1 util.lua introduced debug_log() concept but inconsistent across project; log/core.lua handles everything (levels, rotation, debug) without modularity.
+   - **Solution:** Modularize into log/log.lua (main), log/logger.lua (debug wrapper), log/rotate.lua (file management); enforce config.debug gate on ALL debug logs project-wide for 35%+ performance gain.
+   - **Status:** [ ] Not implemented.

lua/grok/chat/history.lua

@@ -1,19 +1,50 @@
 -- ~/github.com/acris-software/grok-nvim/lua/grok/chat/history.lua
-
 local M = {}
 
+local history_file = vim.fn.stdpath("data") .. "/grok_chat_history.json"
 M.history = {}
 
+function M.load()
+  local file = io.open(history_file, "r")
+  if file then
+    local content = file:read("*a")
+    file:close()
+    if content and content ~= "" then
+      local ok, loaded_history = pcall(vim.json.decode, content)
+      if ok and type(loaded_history) == "table" then
+        M.history = loaded_history
+      end
+    end
+  end
+end
+
+function M.save()
+  local file = io.open(history_file, "w")
+  if file then
+    file:write(vim.json.encode(M.history))
+    file:close()
+  end
+end
+
 function M.add(role, content)
   table.insert(M.history, { role = role, content = content })
+  -- Limit history to 10 messages to prevent excessive growth
+  if #M.history > 10 then
+    table.remove(M.history, 1)
+  end
+  M.save()
 end
 
 function M.get()
+  if #M.history == 0 then
+    M.load()
+  end
   return M.history
 end
 
 function M.clear()
   M.history = {}
+  M.save()
 end
 
 return M

lua/grok/chat/init.lua

@@ -1,10 +1,12 @@
 -- ~/github.com/acris-software/grok-nvim/lua/grok/chat/init.lua
 
 local M = {}
+
 local async = require("plenary.async")
 local ui = require("grok.ui")
 local log = require("grok.log")
 local request = require("grok.chat.request")
+local util = require("grok.util")
 
 function M.chat(prompt)
   local config = require("grok").config
@@ -37,6 +39,12 @@ function M.chat(prompt)
         end
         request.send_request(prompt)
       end)
+    else
+      -- Directly open the floating input instead of simulating 'i'
+      vim.schedule(function()
+        log.debug("Opening floating input with callback: " .. vim.inspect(ui.current_callback))
+        util.create_floating_input({ callback = ui.current_callback })
+      end)
     end
   end)
 end

lua/grok/chat/request.lua

@@ -27,6 +27,7 @@ local function send_request(input)
     end)
     if not ok then
       log.error("Failed to set reasoning line: " .. vim.inspect(err))
+      vim.notify("UI Error: Failed to update reasoning!", vim.log.levels.ERROR)
     end
   end)
   curl.post(config.base_url .. "/chat/completions", {
@@ -47,6 +48,7 @@ local function send_request(input)
           if not ok then
             log.error("Failed to set modifiable false after error: " .. vim.inspect(set_err))
           end
+          vim.notify("API Error: Request failed!", vim.log.levels.ERROR)
         end)
         return
       end
@@ -59,13 +61,15 @@ local function send_request(input)
             if not ok then
               log.error("Failed to set modifiable true for done: " .. vim.inspect(set_err))
             end
+            -- Ensure final response is appended
             ui.append_response("\n\n")
             ok, set_err = pcall(vim.api.nvim_buf_set_option, ui.current_buf, "modifiable", false)
             if not ok then
               log.error("Failed to set modifiable false after done: " .. vim.inspect(set_err))
             end
+            history.add("assistant", response)
+            log.debug("Stream completed, full response: " .. response)
           end)
-          history.add("assistant", response)
           return
         end
         local ok, json = pcall(vim.json.decode, json_str)
@@ -74,6 +78,10 @@ local function send_request(input)
           if delta_content ~= "" then
             response = response .. delta_content
             vim.schedule(function()
+              if not ui.current_buf or not vim.api.nvim_buf_is_valid(ui.current_buf) then
+                log.error("Chat buffer invalid during stream")
+                return
+              end
               if is_first_chunk then
                 is_first_chunk = false
                 local set_ok, set_err = pcall(vim.api.nvim_buf_set_option, ui.current_buf, "modifiable", true)
@@ -106,6 +114,8 @@ local function send_request(input)
               end
             end)
           end
+        else
+          log.debug("Invalid JSON or no delta in stream chunk: " .. json_str)
         end
       end
     end,
@@ -122,6 +132,24 @@ local function send_request(input)
           if not ok then
             log.error("Failed to set modifiable false after non-200: " .. vim.inspect(set_err))
           end
+          vim.notify("API Error: Status " .. res.status, vim.log.levels.ERROR)
+        else
+          -- Ensure final response is complete
+          if response ~= "" then
+            vim.schedule(function()
+              local ok, set_err = pcall(vim.api.nvim_buf_set_option, ui.current_buf, "modifiable", true)
+              if not ok then
+                log.error("Failed to set modifiable true for final response: " .. vim.inspect(set_err))
+              end
+              ui.append_response("\n\n")
+              ok, set_err = pcall(vim.api.nvim_buf_set_option, ui.current_buf, "modifiable", false)
+              if not ok then
+                log.error("Failed to set modifiable false after final response: " .. vim.inspect(set_err))
+              end
+              history.add("assistant", response)
+              log.debug("Final response appended: " .. response)
+            end)
+          end
         end
       end)
     end,

lua/grok/commands.lua

@@ -4,19 +4,32 @@ local M = {}
 function M.setup_commands()
   local log = require("grok.log")
   vim.api.nvim_create_user_command("Grok", function(opts)
-    require("grok.chat").chat(opts.args)
+    require("grok.ui").open_chat_window(function(input)
+      require("grok.chat").chat(input)
+    end)
+    if opts.args and opts.args ~= "" then
+      require("grok.chat").chat(opts.args)
+    end
     log.debug("Grok command executed with args: " .. vim.inspect(opts.args))
   end, { nargs = "*", desc = "Chat with Grok" })
   vim.api.nvim_create_user_command("GrokVisual", function()
     local selected_text = require("grok.util").get_visual_selection()
+    require("grok.ui").open_chat_window(function(input)
+      require("grok.chat").chat(input)
+    end)
     require("grok.chat").chat(selected_text)
     log.debug("GrokVisual command executed with selection: " .. selected_text)
   end, { range = true, desc = "Chat with Grok using visual selection" })
   vim.api.nvim_create_user_command("GrokClear", function()
-    require("grok.ui").close_chat_window()
     require("grok.chat").clear_history()
+    local ui = require("grok.ui")
+    if ui.current_buf and vim.api.nvim_buf_is_valid(ui.current_buf) then
+      require("grok.ui.render").render_tab_content(ui.current_buf, function(input)
+        require("grok.chat").chat(input)
+      end)
+    end
     log.info("GrokClear command executed")
-  end, { desc = "Clear and close Grok chat window" })
+  end, { desc = "Clear Grok chat history" })
   vim.api.nvim_create_user_command("GrokLog", function()
     local log_file = vim.fn.stdpath("data") .. "/grok.log"
     vim.cmd("edit " .. log_file)
@@ -25,18 +38,19 @@ function M.setup_commands()
   vim.api.nvim_create_user_command("GrokKeymaps", function()
     local ui = require("grok.ui")
     if ui.current_win and vim.api.nvim_win_is_valid(ui.current_win) then
-      ui.current_tab = 2
+      ui.set_current_tab(2)
       require("grok.ui.render").render_tab_content(ui.current_buf, function() end)
       log.debug("GrokKeymaps switched to tab 2 in UI")
     else
-      local keymaps = {
-        "In Grok Chat Window:",
-        " <CR> or i: Open input prompt",
-        " <Esc>: Close window",
-      }
-      vim.notify(table.concat(keymaps, "\n"), vim.log.levels.INFO)
-      log.debug("GrokKeymaps notified keymaps")
+      ui.open_chat_window(function() end)
+      ui.set_current_tab(2)
+      require("grok.ui.render").render_tab_content(ui.current_buf, function() end)
+      log.debug("GrokKeymaps opened UI and switched to tab 2")
     end
-  end, { desc = "List Grok-nvim keymaps" })
+  end, { desc = "Show Grok-nvim keymaps" })
+  vim.api.nvim_create_user_command("GrokUI", function()
+    require("grok.ui").toggle_ui()
+    log.debug("GrokUI command executed")
+  end, { desc = "Toggle Grok UI" })
 end
 return M

lua/grok/init.lua

@@ -3,6 +3,7 @@
 local M = {}
 local commands = require("grok.commands")
 local log = require("grok.log")
+
 function M.setup(opts)
   local curl = require("plenary.curl")
   local async = require("plenary.async")
@@ -36,13 +37,37 @@ function M.setup(opts)
     temperature = 0.7,
     max_tokens = 256,
     debug = false,
+    prompt_position = "center", -- Options: "left", "center", "right"
   }, opts or {})
   if not M.config.api_key then
     vim.notify("GROK_KEY not set in ~/.secrets, environment, or opts!", vim.log.levels.ERROR)
     log.error("API key not set in setup")
   end
+  require("grok.util").validate_config(M.config)
   M.chat = chat.chat
   commands.setup_commands()
-  log.info("Plugin setup completed")
+  log.info("Plugin setup completed - grok-nvim v0.1.1")
+
+  -- Clean up Grok UI and buffers on Neovim exit
+  vim.api.nvim_create_autocmd("VimLeavePre", {
+    callback = function()
+      local ui = require("grok.ui")
+      if ui.current_win and vim.api.nvim_win_is_valid(ui.current_win) then
+        require("grok.ui").close_chat_window()
+      end
+      for _, buf in ipairs(vim.api.nvim_list_bufs()) do
+        if vim.api.nvim_buf_is_valid(buf) then
+          local buftype = vim.api.nvim_buf_get_option(buf, "buftype")
+          local filetype = vim.api.nvim_buf_get_option(buf, "filetype")
+          if buftype == "prompt" or filetype == "markdown" then
+            pcall(vim.api.nvim_buf_delete, buf, { force = true })
+          end
+        end
+      end
+      log.debug("Cleaned up Grok UI and buffers on VimLeavePre")
+    end,
+    desc = "Clean up Grok UI and buffers before exiting Neovim",
+  })
 end
+
 return M

lua/grok/log/core.lua

@@ -4,6 +4,8 @@ local log_file = vim.fn.stdpath("data") .. "/grok.log"
 local max_log_size = 1024 * 1024 -- 1MB
 local last_message = nil
 local repeat_count = 0
+local message_cache = {} -- Cache for throttling: {message = timestamp}
+local throttle_interval = 5000
 
 local function rotate_log()
   local file_info = vim.uv.fs_stat(log_file)
@@ -24,21 +26,41 @@ local function write_log(full_message)
 end
 
 local function log(message, level)
-  if require("grok").config.debug then
-    local timestamp = os.date("%Y-%m-%d %H:%M:%S")
-    local prefix = level or ""
-    local full_message = timestamp .. " " .. prefix .. message
-    if full_message == last_message then
-      repeat_count = repeat_count + 1
-      return -- Throttle
-    else
-      if repeat_count > 0 then
-        local repeat_note = timestamp .. " [REPEAT] Previous message repeated " .. repeat_count .. " times"
-        write_log(repeat_note)
-      end
-      repeat_count = 0
-      last_message = full_message
-      write_log(full_message)
+  if not require("grok").config.debug then
+    return
+  end
+  local timestamp = os.date("%Y-%m-%d %H:%M:%S")
+  local prefix = level or ""
+  local full_message = timestamp .. " " .. prefix .. message
+  local now = vim.loop.now()
+
+  -- Check if message was logged recently
+  local last_logged = message_cache[message]
+  if last_logged and (now - last_logged) < throttle_interval then
+    return -- Throttle: skip if logged within throttle_interval
+  end
+
+  -- Update cache with current timestamp
+  message_cache[message] = now
+
+  -- Handle repeated messages
+  if full_message == last_message then
+    repeat_count = repeat_count + 1
+    return -- Throttle consecutive identical messages
+  else
+    if repeat_count > 0 then
+      local repeat_note = timestamp .. " [REPEAT] Previous message repeated " .. repeat_count .. " times"
+      write_log(repeat_note)
+    end
+    repeat_count = 0
+    last_message = full_message
+    write_log(full_message)
+  end
+
+  -- Clean up old cache entries
+  for msg, ts in pairs(message_cache) do
+    if (now - ts) >= throttle_interval then
+      message_cache[msg] = nil
     end
   end
 end