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

chore: make azure o series models stream

Browse Source
This commit is contained in:
kernitus 2025-01-27 12:29:29 +00:00 committed by yetone
parent 2ead26f809
commit 499b7a854b
1 changed files with 3 additions and 5 deletions

8
lua/avante/providers/azure.lua

@ -25,12 +25,10 @@ M.parse_curl_args = function(provider, code_opts)
}
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
-- 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
stream = false
end
return {
@ -43,7 +41,7 @@ M.parse_curl_args = function(provider, code_opts)
headers = headers,
body = vim.tbl_deep_extend("force", {
messages = M.parse_messages(code_opts),
stream = stream,
stream = true,
}, body_opts),
}
end