89 lines
1.8 KiB
Lua
89 lines
1.8 KiB
Lua
-- lua/gemini/api.lua
|
|
|
|
local M = {}
|
|
|
|
local function get_api_key()
|
|
-- Check for environment variable
|
|
local api_key = os.getenv("GEMINI_API_KEY")
|
|
if api_key then
|
|
return api_key
|
|
end
|
|
|
|
-- Check for Neovim global variable
|
|
api_key = vim.g.gemini_api_key
|
|
if api_key then
|
|
return api_key
|
|
end
|
|
|
|
return nil -- API key not found
|
|
end
|
|
|
|
local function make_request(prompt)
|
|
local api_key = get_api_key() -- Get API key from environment or global variable
|
|
|
|
if not api_key then
|
|
vim.notify(
|
|
"Google AI API key not set. Set GEMINI_API_KEY environment variable or g.gemini_api_key in your init.vim or init.lua.",
|
|
vim.log.levels.ERROR
|
|
)
|
|
return nil
|
|
end
|
|
|
|
local model = "gemini-2.0-flash" -- SPECIFY Gemini 2.0 Flash MODEL
|
|
-- Construct the JSON payload
|
|
local payload = vim.json.encode({
|
|
contents = {
|
|
{
|
|
parts = {
|
|
{
|
|
text = prompt,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
local command = string.format(
|
|
"curl -s -X POST "
|
|
.. "'https://generative-ai.googleapis.com/v1beta/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 decoded_result = vim.json.decode(result)
|
|
|
|
return decoded_result
|
|
end
|
|
|
|
function M.get_response(prompt)
|
|
local result = make_request(prompt)
|
|
|
|
if
|
|
result
|
|
and result.candidates
|
|
and result.candidates[1]
|
|
and result.candidates[1].content
|
|
and result.candidates[1].content.parts
|
|
then
|
|
return result.candidates[1].content.parts[1].text
|
|
else
|
|
vim.notify("No response from Google AI API or malformed response.", vim.log.levels.ERROR)
|
|
return nil
|
|
end
|
|
end
|
|
|
|
print("api.lua loaded")
|
|
|
|
return M
|