feat: make Azure OpenAI api_version configurable

This commit is contained in:
marginal23326
2025-02-11 16:59:30 +06:00
parent 3a459a91ad
commit 3403de4bea
2 changed files with 3 additions and 1 deletions

View File

@@ -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=

View File

@@ -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,
)