6475407d0d
* fix(llm): make sure to allow passing custom module Signed-off-by: Aaron Pham <contact@aarnphm.xyz> * fix: correct custom class Signed-off-by: Aaron Pham <contact@aarnphm.xyz> * fix: correct attribute Signed-off-by: Aaron Pham <contact@aarnphm.xyz> --------- Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
42 lines
1.0 KiB
Lua
42 lines
1.0 KiB
Lua
local Utils = require("avante.utils")
|
|
local Config = require("avante.config")
|
|
local P = require("avante.providers")
|
|
local O = require("avante.providers").openai
|
|
|
|
---@class AvanteProviderFunctor
|
|
local M = {}
|
|
|
|
M.api_key_name = "GROQ_API_KEY"
|
|
|
|
M.has = function()
|
|
return os.getenv(M.api_key_name) and true or false
|
|
end
|
|
|
|
M.parse_message = O.parse_message
|
|
M.parse_response = O.parse_response
|
|
|
|
M.parse_curl_args = function(provider, code_opts)
|
|
local base, body_opts = P.parse_config(provider)
|
|
|
|
local headers = {
|
|
["Content-Type"] = "application/json",
|
|
}
|
|
if not P.env.is_local("groq") then
|
|
headers["Authorization"] = "Bearer " .. os.getenv(base.api_key_name or M.api_key_name)
|
|
end
|
|
|
|
return {
|
|
url = Utils.trim(base.endpoint, { suffix = "/" }) .. "/openai/v1/chat/completions",
|
|
proxy = base.proxy,
|
|
insecure = base.allow_insecure,
|
|
headers = headers,
|
|
body = vim.tbl_deep_extend("force", {
|
|
model = base.model,
|
|
messages = M.parse_message(code_opts),
|
|
stream = true,
|
|
}, body_opts),
|
|
}
|
|
end
|
|
|
|
return M
|