From 7bfeb48f7d45c49b089839eb228ecbe9f68ab657 Mon Sep 17 00:00:00 2001
From: Jonas Widen <jonas.widen@widens.eu>
Date: Sun, 16 Mar 2025 14:14:12 +0100
Subject: [PATCH] Fixed issues

---
 lua/gemini/api.lua | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/lua/gemini/api.lua b/lua/gemini/api.lua
index 1e157a6..f79c0da 100644
--- a/lua/gemini/api.lua
+++ b/lua/gemini/api.lua
@@ -29,7 +29,7 @@ local function make_request(prompt)
 		return nil
 	end
 
-	local model = "gemini-2.0-flash" -- SPECIFY Gemini 2.0 Flash MODEL
+	local model = "gemini-pro" -- Using the standard model instead of flash
 	-- Construct the JSON payload
 	local payload = vim.json.encode({
 		contents = {
@@ -43,11 +43,14 @@ local function make_request(prompt)
 		},
 	})
 
+	-- Escape the payload for shell
+	payload = vim.fn.shellescape(payload)
+
 	local command = string.format(
 		"curl -s -X POST "
-			.. "'https://generative-ai.googleapis.com/v1beta/models/%s:generateContent?key=%s' "
+			.. "'https://generative-ai.googleapis.com/v1/models/%s:generateContent?key=%s' "
 			.. "-H 'Content-Type: application/json' "
-			.. "-d '%s'",
+			.. "-d %s",
 		model,
 		api_key,
 		payload
@@ -61,7 +64,11 @@ local function make_request(prompt)
 		return nil
 	end
 
-	local decoded_result = vim.json.decode(result)
+	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
@@ -83,6 +90,4 @@ function M.get_response(prompt)
 	end
 end
 
-print("api.lua loaded")
-
 return M