From 499b7a854be3f5b1ad612b659a1abcf956403b13 Mon Sep 17 00:00:00 2001
From: kernitus <2789734+kernitus@users.noreply.github.com>
Date: Mon, 27 Jan 2025 12:29:29 +0000
Subject: [PATCH] chore: make azure o series models stream

---
 lua/avante/providers/azure.lua | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/lua/avante/providers/azure.lua b/lua/avante/providers/azure.lua
index 0760238..49ee887 100644
--- a/lua/avante/providers/azure.lua
+++ b/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