From 3403de4bea2af51cbf759076a7ee98c6b840cc23 Mon Sep 17 00:00:00 2001
From: marginal23326 <58261815+marginal23326@users.noreply.github.com>
Date: Tue, 11 Feb 2025 16:59:30 +0600
Subject: [PATCH] feat: make Azure OpenAI `api_version` configurable

---
 .env.example       | 1 +
 src/utils/utils.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.env.example b/.env.example
index 299082d..9248ab6 100644
--- a/.env.example
+++ b/.env.example
@@ -8,6 +8,7 @@ GOOGLE_API_KEY=
 
 AZURE_OPENAI_ENDPOINT=
 AZURE_OPENAI_API_KEY=
+AZURE_OPENAI_API_VERSION=2025-01-01-preview
 
 DEEPSEEK_ENDPOINT=https://api.deepseek.com
 DEEPSEEK_API_KEY=
diff --git a/src/utils/utils.py b/src/utils/utils.py
index e32c114..dc949ce 100644
--- a/src/utils/utils.py
+++ b/src/utils/utils.py
@@ -128,10 +128,11 @@ def get_llm_model(provider: str, **kwargs):
             base_url = os.getenv("AZURE_OPENAI_ENDPOINT", "")
         else:
             base_url = kwargs.get("base_url")
+        api_version = kwargs.get("api_version", "") or os.getenv("AZURE_OPENAI_API_VERSION", "2025-01-01-preview")
         return AzureChatOpenAI(
             model=kwargs.get("model_name", "gpt-4o"),
             temperature=kwargs.get("temperature", 0.0),
-            api_version="2024-05-01-preview",
+            api_version=api_version,
             azure_endpoint=base_url,
             api_key=api_key,
         )