52 lines
1.4 KiB
Lua
52 lines
1.4 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, code_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 "o1" set the supported parameters only
|
|
local stream = true
|
|
if base.model and string.find(base.model, "o1") then
|
|
body_opts.max_tokens = nil
|
|
body_opts.temperature = 1
|
|
stream = false
|
|
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(code_opts),
|
|
stream = stream,
|
|
}, body_opts),
|
|
}
|
|
end
|
|
|
|
return M
|