test(test_completion.py): fix azure read from env variable test

This commit is contained in:
Krrish Dholakia 2023-11-23 16:58:00 -08:00
parent 60dab0d700
commit abe2a009f9
2 changed files with 2 additions and 2 deletions

View file

@ -3,7 +3,7 @@ import Tabs from '@theme/Tabs';
import TabItem from '@theme/TabItem';
# Router - Load Balancing, Queueing
# Router - Load Balancing, Fallbacks
LiteLLM manages:
- Load-balance across multiple deployments (e.g. Azure/OpenAI)

View file

@ -606,7 +606,7 @@ def test_completion_azure():
response = completion(
model="azure/chatgpt-v-2",
messages=messages,
api_key="os.environ/AZURE_API_KEY_NA"
api_key="os.environ/AZURE_API_KEY"
)
print(f"response: {response}")
## Test azure flag for backwards compatibility