try to use nvim http client instead of curl

lua/gemini/api.lua

@@ -5,37 +5,92 @@ local M = {}
 -- Store conversation history
 local conversation_history = {}
 
+-- Helper function to get API key
 local function get_api_key()
-	-- Check for environment variable
+	return vim.g.gemini_api_key or os.getenv("GEMINI_API_KEY")
-	local api_key = os.getenv("GEMINI_API_KEY")
-	if api_key then
-		return api_key
 end
 
-	-- Check for Neovim global variable
+-- Async HTTP request function
-	api_key = vim.g.gemini_api_key
+local function async_request(url, payload, callback)
-	if api_key then
+	local host, port = vim.uri_from_fname(url):match("^https://([^/]+)(/.*)$")
-		return api_key
+	local client = vim.loop.new_tcp()
+	
+	client:connect(host, 443, function(err)
+		if err then 
+			vim.schedule(function()
+				callback(nil, "Connection error: " .. err)
+			end)
+			return
 		end
 
-	return nil -- API key not found
+		local ssl = vim.loop.new_tls()
+		ssl:wrap(client)
+		
+		local request = string.format(
+			"POST %s HTTP/1.1\r\n" ..
+			"Host: %s\r\n" ..
+			"Content-Type: application/json\r\n" ..
+			"Content-Length: %d\r\n" ..
+			"\r\n" ..
+			"%s",
+			port, host, #payload, payload
+		)
+		
+		local response = ""
+		
+		ssl:write(request, function(err)
+			if err then
+				vim.schedule(function()
+					callback(nil, "Write error: " .. err)
+				end)
+				return
 			end
 			
-local function make_request(prompt, context)
+			ssl:read_start(function(err, chunk)
+				if err then
+					vim.schedule(function()
+						callback(nil, "Read error: " .. err)
+					end)
+					return
+				end
+				
+				if chunk then
+					response = response .. chunk
+				else
+					-- Connection closed, process response
+					local body = response:match("\r\n\r\n(.+)$")
+					local success, decoded = pcall(vim.json.decode, body)
+					
+					vim.schedule(function()
+						if success then
+							callback(decoded)
+						else
+							callback(nil, "JSON decode error: " .. body)
+						end
+					end)
+					
+					ssl:close()
+					client:close()
+				end
+			end)
+		end)
+	end)
+end
+
+function M.get_response(prompt, context, callback)
 	local api_key = get_api_key()
 	
 	if not api_key then
-		vim.notify(
+		vim.schedule(function()
-			"Google AI API key not set. Set GEMINI_API_KEY environment variable or g.gemini_api_key in your init.vim or init.lua.",
+			callback(nil, "API key not set")
-			vim.log.levels.ERROR
+		end)
-		)
+		return
-		return nil
 	end
 
 	local model = "gemini-2.0-flash"
 	local contents = {}
 	
-	-- Add conversation history to the request
+	-- Add conversation history
 	for _, message in ipairs(conversation_history) do
 		table.insert(contents, {
 			role = message.role,
@@ -45,7 +100,7 @@ local function make_request(prompt, context)
 		})
 	end
 
-	-- Add the current prompt
+	-- Add current prompt
 	if context then
 		table.insert(contents, {
 			role = "user",
@@ -62,77 +117,53 @@ local function make_request(prompt, context)
 		})
 	end
 
-	local payload = vim.json.encode({
+	-- Store prompt in history
-		contents = contents,
-	})
-
-	-- Escape the payload for shell
-	payload = vim.fn.shellescape(payload)
-
-	local command = string.format(
-		"curl -s -X POST "
-			.. "'https://generativelanguage.googleapis.com/v1/models/%s:generateContent?key=%s' "
-			.. "-H 'Content-Type: application/json' "
-			.. "-d %s",
-		model,
-		api_key,
-		payload
-	)
-
-	local result = vim.fn.system(command)
-
-	-- Check for errors during the curl execution
-	if vim.v.shell_error ~= 0 then
-		vim.notify("Error executing curl. Check your command and ensure curl is installed.", vim.log.levels.ERROR)
-		return nil
-	end
-
-	local success, decoded_result = pcall(vim.json.decode, result)
-	if not success then
-		vim.notify("Failed to decode API response: " .. result, vim.log.levels.ERROR)
-		return nil
-	end
-
-	return decoded_result
-end
-
-function M.get_response(prompt, context)
-	-- Add user message to history
 	table.insert(conversation_history, {
 		role = "user",
 		content = prompt
 	})
 
-	local result = make_request(prompt, context)
+	local payload = vim.json.encode({
+		contents = contents,
+	})
 
-	if result then
+	local url = string.format(
-		if result.error then
+		"https://generativelanguage.googleapis.com/v1/models/%s:generateContent?key=%s",
-			vim.notify("API Error: " .. vim.inspect(result.error), vim.log.levels.ERROR)
+		model,
-			return nil
+		api_key
+	)
+
+	async_request(url, payload, function(result, error)
+		if error then
+			callback(nil, error)
+			return
 		end
 
-		if
+		if result.error then
-			result.candidates
+			callback(nil, "API Error: " .. vim.inspect(result.error))
-			and result.candidates[1]
+			return
-			and result.candidates[1].content
+		end
-			and result.candidates[1].content.parts
+
-			and result.candidates[1].content.parts[1]
+		if result.candidates and
-			and result.candidates[1].content.parts[1].text
+		   result.candidates[1] and
-		then
+		   result.candidates[1].content and
+		   result.candidates[1].content.parts and
+		   result.candidates[1].content.parts[1] and
+		   result.candidates[1].content.parts[1].text then
+			
 			local response_text = result.candidates[1].content.parts[1].text
-			-- Add assistant response to history
+			
+			-- Store response in history
 			table.insert(conversation_history, {
 				role = "model",
 				content = response_text
 			})
-			return response_text
-		end
 			
-		vim.notify("Unexpected response structure: " .. vim.inspect(result), vim.log.levels.ERROR)
+			callback(response_text)
+		else
+			callback(nil, "Unexpected response structure")
 		end
-	
+	end)
-	vim.notify("No response from Google AI API or malformed response.", vim.log.levels.ERROR)
-	return nil
 end
 
 -- Add function to clear conversation history

lua/gemini/init.lua

@@ -161,15 +161,31 @@ local function gemini_query(prompt, context)
     -- Store the context for subsequent queries
     current_context = context
     
-    -- Show initial message in chat window and ensure it's visible
+    -- Show initial message in chat window
     local initial_content = "User: " .. prompt .. "\n\nAssistant: Thinking..."
     update_chat_window(initial_content)
     
     -- Force Neovim to update the screen
     vim.cmd('redraw')
     
-    local response = api.get_response(prompt, context)
+    -- Make async request
-    if response then
+    api.get_response(prompt, context, function(response, error)
+        if error then
+            -- Replace "Thinking..." with error message
+            local lines = vim.api.nvim_buf_get_lines(chat_bufnr, 0, -1, false)
+            for i = 1, #lines do
+                if lines[i] == "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: Error - " .. error})
+                    vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', false)
+                    break
+                end
+            end
+            vim.notify("Failed to get response: " .. error, vim.log.levels.ERROR)
+            return
+        end
+
         -- Make buffer modifiable
         vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', true)
         
@@ -204,20 +220,7 @@ local function gemini_query(prompt, context)
         
         -- Clear the command line
         vim.cmd('echo ""')
-    else
+    end)
-        -- Replace "Thinking..." with error message
-        local lines = vim.api.nvim_buf_get_lines(chat_bufnr, 0, -1, false)
-        for i = 1, #lines do
-            if lines[i] == "Assistant: Thinking..." then
-                lines[i] = "Assistant: Failed to get response from Gemini API"
-                vim.api.nvim_buf_set_option(chat_bufnr, 'modifiable', true)
-                vim.api.nvim_buf_set_lines(chat_bufnr, 0, -1, false, lines)
-                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)
-    end
 end
 
 -- Make gemini_query available in M so it can be used by the setup function