expose api_version

This commit is contained in:
Krrish Dholakia 2023-08-05 21:57:12 -07:00
parent 0a1525a670
commit 8c5118cae6
3 changed files with 3 additions and 2 deletions

View file

@ -67,7 +67,7 @@ def completion(
# azure configs
openai.api_type = "azure"
openai.api_base = litellm.api_base if litellm.api_base is not None else get_secret("AZURE_API_BASE")
openai.api_version = get_secret("AZURE_API_VERSION")
openai.api_version = litellm.api_version if litellm.api_version is not None else get_secret("AZURE_API_VERSION")
if api_key:
openai.api_key = api_key
elif litellm.azure_key: