feat(gpt): track and manage job IDs for SSE requests

JavaHello · JavaHello · commit dfa2533ee940 · 2025-02-08T12:00:05.000+08:00
diff --git a/lua/kide/gpt/chat.lua b/lua/kide/gpt/chat.lua
@@ -7,6 +7,7 @@ local state = {
   cursormoved = false,
   chatruning = false,
   winleave = false,
+  job = nil,
 }
 local chat_last = {}
 
@@ -64,6 +65,10 @@ local close_gpt_win = function()
     state.cursormoved = false
     state.chatruning = false
     state.winleave = false
+    if state.job then
+      pcall(vim.fn.jobstop, state.job)
+      state.job = nil
+    end
   end
 end
 
@@ -258,7 +263,7 @@ M.gpt_chat = function()
     end
   end
 
-  require("kide.gpt.sse").request(json, callback)
+  state.job = require("kide.gpt.sse").request(json, callback)
 end
 
 M.gpt_last = function(buf)
diff --git a/lua/kide/gpt/commit.lua b/lua/kide/gpt/commit.lua
@@ -1,4 +1,5 @@
 local M = {}
+local job = nil
 local request_json = {
   messages = {
     {
@@ -33,7 +34,7 @@ function M.commit_message(diff, callback)
   json.messages[1].content =
     "I want you to act as a commit message generator. I will provide you with information about the task and the prefix for the task code, and I would like you to generate an appropriate commit message using the conventional commit format. Do not write any explanations or other words, just reply with the commit message."
   json.messages[2].content = diff
-  require("kide.gpt.sse").request(json, callback)
+  job = require("kide.gpt.sse").request(json, callback)
 end
 
 M.commit_diff_msg = function()
@@ -52,6 +53,10 @@ M.commit_diff_msg = function()
     buffer = codebuf,
     callback = function()
       closed = true
+      if job then
+        pcall(vim.fn.jobstop, job)
+        job = nil
+      end
     end,
   })
 
diff --git a/lua/kide/gpt/reasoner.lua b/lua/kide/gpt/reasoner.lua
@@ -7,6 +7,7 @@ local state = {
   cursormoved = false,
   chatruning = false,
   winleave = false,
+  job = nil,
 }
 local chat_last = {}
 
@@ -58,6 +59,10 @@ local close_gpt_win = function()
     state.cursormoved = false
     state.chatruning = false
     state.winleave = false
+    if state.job then
+      pcall(vim.fn.jobstop, state.job)
+      state.job = nil
+    end
   end
 end
 
@@ -268,7 +273,7 @@ M.gpt_chat = function()
     end
   end
 
-  require("kide.gpt.sse").request(json, callback)
+  state.job = require("kide.gpt.sse").request(json, callback)
 end
 
 M.gpt_last = function(buf)
diff --git a/lua/kide/gpt/sse.lua b/lua/kide/gpt/sse.lua
@@ -77,6 +77,7 @@ local function handle_sse_events(cmd, callback)
       })
     end,
   })
+  return job
 end
 
 function M.request(json, callback)
@@ -95,7 +96,7 @@ function M.request(json, callback)
     body,
     M.config.API_URL,
   }
-  handle_sse_events(cmd, callback)
+  return handle_sse_events(cmd, callback)
 end
 
 return M
diff --git a/lua/kide/gpt/translate.lua b/lua/kide/gpt/translate.lua
@@ -54,11 +54,12 @@ function M.translate(request, callback)
   local json = request_json
   json.messages[1].content = trans_system_prompt(request)
   json.messages[2].content = request.text
-  require("kide.gpt.sse").request(json, callback)
+  return require("kide.gpt.sse").request(json, callback)
 end
 
 local max_width = 120
 local max_height = 40
+local job = nil
 
 M.translate_float = function(request)
   local codebuf = vim.api.nvim_get_current_buf()
@@ -113,6 +114,10 @@ M.translate_float = function(request)
     buffer = buf,
     callback = function()
       closed = true
+      if job then
+        pcall(vim.fn.jobstop, job)
+        job = nil
+      end
     end,
   })
   vim.api.nvim_create_autocmd("WinLeave", {
@@ -172,7 +177,7 @@ M.translate_float = function(request)
       vim.api.nvim_put(put_data, "c", true, true)
     end
   end
-  M.translate(request, callback)
+  job = M.translate(request, callback)
 end
 
 return M
diff --git a/lua/mappings.lua b/lua/mappings.lua
@@ -21,7 +21,6 @@ map("v", "<A-i>", function()
   end, 500)
 end, { desc = "toggle term" })
 
-
 map("n", "<leader>gb", require("gitsigns").blame_line, { desc = "gitsigns blame line" })
 map("n", "<ESC>", "<CMD>noh<CR>", { desc = "Clear Highlight" })
 
