remove baidu
This commit is contained in:
parent
4d9201bbea
commit
ab9127fcbc
@ -143,6 +143,14 @@ M._defaults = {
|
|||||||
max_tokens = 8000,
|
max_tokens = 8000,
|
||||||
},
|
},
|
||||||
---@type AvanteSupportedProvider
|
---@type AvanteSupportedProvider
|
||||||
|
["baidu"] = {
|
||||||
|
__inherited_from = "openai",
|
||||||
|
model = "deekseek-v1",
|
||||||
|
timeout = 30000, -- Timeout in milliseconds
|
||||||
|
temperature = 0,
|
||||||
|
max_tokens = 8000,
|
||||||
|
},
|
||||||
|
---@type AvanteSupportedProvider
|
||||||
["claude-opus"] = {
|
["claude-opus"] = {
|
||||||
__inherited_from = "claude",
|
__inherited_from = "claude",
|
||||||
model = "claude-3-opus-20240229",
|
model = "claude-3-opus-20240229",
|
||||||
|
@ -1,83 +0,0 @@
|
|||||||
-- FILEPATH: avante.nvim/lua/avante/providers/baidu.lua
|
|
||||||
local Utils = require("avante.utils")
|
|
||||||
local Config = require("avante.config")
|
|
||||||
local P = require("avante.providers")
|
|
||||||
|
|
||||||
---@class AvanteProviderFunctor
|
|
||||||
local M = {}
|
|
||||||
|
|
||||||
M.api_key_name = "BAIDU_API_KEY"
|
|
||||||
|
|
||||||
M.role_map = {
|
|
||||||
user = "user",
|
|
||||||
assistant = "assistant",
|
|
||||||
}
|
|
||||||
|
|
||||||
---@param opts AvantePromptOptions
|
|
||||||
M.parse_messages = function(opts)
|
|
||||||
local messages = {}
|
|
||||||
|
|
||||||
table.insert(messages, { role = "user", content = opts.system_prompt })
|
|
||||||
|
|
||||||
vim
|
|
||||||
.iter(opts.messages)
|
|
||||||
:each(function(msg) table.insert(messages, { role = M.role_map[msg.role], content = msg.content }) end)
|
|
||||||
|
|
||||||
return messages
|
|
||||||
end
|
|
||||||
|
|
||||||
M.parse_response = function(ctx, data_stream, _, opts)
|
|
||||||
if data_stream:match('"%[DONE%]":') then
|
|
||||||
opts.on_stop({ reason = "complete" })
|
|
||||||
return
|
|
||||||
end
|
|
||||||
|
|
||||||
---@type BaiduChatResponse
|
|
||||||
local jsn = vim.json.decode(data_stream)
|
|
||||||
if jsn.result then
|
|
||||||
opts.on_chunk(jsn.result)
|
|
||||||
if jsn.is_end then opts.on_stop({ reason = "complete" }) end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local Log = require("avante.utils.log")
|
|
||||||
|
|
||||||
M.parse_curl_args = function(provider, prompt_opts)
|
|
||||||
local base, body_opts = P.parse_config(provider)
|
|
||||||
|
|
||||||
-- 验证 appid 是否存在
|
|
||||||
if not base.appid or base.appid == "" then error("Baidu provider requires appid to be set in config") end
|
|
||||||
|
|
||||||
local headers = {
|
|
||||||
["Content-Type"] = "application/json",
|
|
||||||
["appid"] = base.appid, -- 将 appid 加入请求头
|
|
||||||
}
|
|
||||||
|
|
||||||
if P.env.require_api_key(base) then
|
|
||||||
local api_key = provider.parse_api_key()
|
|
||||||
if api_key == nil then
|
|
||||||
error("Baidu API key is not set, please set BAIDU_API_KEY in your environment variable or config file")
|
|
||||||
end
|
|
||||||
headers["Authorization"] = "Bearer " .. api_key
|
|
||||||
end
|
|
||||||
|
|
||||||
local request = {
|
|
||||||
url = Utils.url_join(base.endpoint, "/chat/completions"),
|
|
||||||
proxy = base.proxy,
|
|
||||||
insecure = base.allow_insecure,
|
|
||||||
headers = headers,
|
|
||||||
body = vim.tbl_deep_extend("force", {
|
|
||||||
model = base.model,
|
|
||||||
messages = M.parse_messages(prompt_opts),
|
|
||||||
disable_search = base.disable_search,
|
|
||||||
enable_citation = base.enable_citation,
|
|
||||||
}, body_opts),
|
|
||||||
}
|
|
||||||
|
|
||||||
-- 记录请求详细信息
|
|
||||||
Log.log_request(request.url, request.headers, request.body)
|
|
||||||
|
|
||||||
return request
|
|
||||||
end
|
|
||||||
|
|
||||||
return M
|
|
@ -136,7 +136,6 @@ local DressingState = { winid = nil, input_winid = nil, input_bufnr = nil }
|
|||||||
---@field gemini AvanteProviderFunctor
|
---@field gemini AvanteProviderFunctor
|
||||||
---@field cohere AvanteProviderFunctor
|
---@field cohere AvanteProviderFunctor
|
||||||
---@field bedrock AvanteBedrockProviderFunctor
|
---@field bedrock AvanteBedrockProviderFunctor
|
||||||
---@field baidu AvanteProviderFunctor -- 新增百度 provider
|
|
||||||
|
|
||||||
local M = {}
|
local M = {}
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user