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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lua/avante/providers/azure.lua
View File

@ -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 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 -- NOTE: When using "o" series set the supported parameters only
local stream = true if O.is_o_series_model(base.model) then
if base.model and string.find(base.model, "o1") then
body_opts.max_tokens = nil body_opts.max_tokens = nil
body_opts.temperature = 1 body_opts.temperature = 1
stream = false
end end
return { return {
@ -43,7 +41,7 @@ M.parse_curl_args = function(provider, code_opts)
headers = headers, headers = headers,
body = vim.tbl_deep_extend("force", { body = vim.tbl_deep_extend("force", {
messages = M.parse_messages(code_opts), messages = M.parse_messages(code_opts),
stream = stream, stream = true,
}, body_opts), }, body_opts),
} }
end end