50 lines
1.3 KiB
Lua
50 lines
1.3 KiB
Lua
---@class AvanteAzureProvider: AvanteDefaultBaseProvider
|
|
---@field deployment string
|
|
---@field api_version string
|
|
---@field temperature number
|
|
---@field max_tokens number
|
|
|
|
local Utils = require("avante.utils")
|
|
local P = require("avante.providers")
|
|
local O = require("avante.providers").openai
|
|
|
|
---@class AvanteProviderFunctor
|
|
local M = {}
|
|
|
|
M.api_key_name = "AZURE_OPENAI_API_KEY"
|
|
|
|
M.parse_messages = O.parse_messages
|
|
M.parse_response = O.parse_response
|
|
M.parse_response_without_stream = O.parse_response_without_stream
|
|
|
|
M.parse_curl_args = function(provider, prompt_opts)
|
|
local base, body_opts = P.parse_config(provider)
|
|
|
|
local headers = {
|
|
["Content-Type"] = "application/json",
|
|
}
|
|
if P.env.require_api_key(base) then headers["api-key"] = provider.parse_api_key() end
|
|
|
|
-- NOTE: When using "o" series set the supported parameters only
|
|
if O.is_o_series_model(base.model) then
|
|
body_opts.max_tokens = nil
|
|
body_opts.temperature = 1
|
|
end
|
|
|
|
return {
|
|
url = Utils.url_join(
|
|
base.endpoint,
|
|
"/openai/deployments/" .. base.deployment .. "/chat/completions?api-version=" .. base.api_version
|
|
),
|
|
proxy = base.proxy,
|
|
insecure = base.allow_insecure,
|
|
headers = headers,
|
|
body = vim.tbl_deep_extend("force", {
|
|
messages = M.parse_messages(prompt_opts),
|
|
stream = true,
|
|
}, body_opts),
|
|
}
|
|
end
|
|
|
|
return M
|