CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore(llm): cut-back support (#159)

Browse Source 
Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
This commit is contained in:
Aaron Pham
2024-08-23 00:21:30 -04:00
committed by GitHub
parent 6475407d0d
commit c6d5073945
5 changed files with 10 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
lua/avante/providers/deepseek.lua
View File

@@ -1,41 +0,0 @@
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 = "DEEPSEEK_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("deepseek") then
headers["Authorization"] = "Bearer " .. os.getenv(base.api_key_name or M.api_key_name)
end
return {
url = Utils.trim(base.endpoint, { suffix = "/" }) .. "/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

41
lua/avante/providers/groq.lua
View File

@@ -1,41 +0,0 @@
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

4
lua/avante/providers/init.lua
View File

@@ -88,9 +88,7 @@ local Dressing = require("avante.ui.dressing")
---@field copilot AvanteProviderFunctor
---@field claude AvanteProviderFunctor
---@field azure AvanteProviderFunctor
---@field deepseek AvanteProviderFunctor
---@field gemini AvanteProviderFunctor
---@field groq AvanteProviderFunctor
local M = {}
setmetatable(M, {
@@ -224,7 +222,7 @@ function M.refresh(provider)
require("avante.config").override({ provider = provider })
end
local default_providers = { "openai", "claude", "azure", "deepseek", "groq", "gemini", "copilot" }
local default_providers = { "openai", "claude", "azure", "gemini", "copilot" }
---@private
M.commands = function()