From cacfac572614aa6e0232735da0917b0c6f0f396c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 13 Oct 2023 22:19:34 -0700 Subject: [PATCH] (test) fix azure embedding test --- litellm/tests/test_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 4ca2a9eea54..3d79100eeca 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -56,9 +56,9 @@ def test_openai_azure_embedding(): print(response) - os.environ['AZURE_API_VERSION'] = api_key + os.environ['AZURE_API_VERSION'] = api_version os.environ['AZURE_API_BASE'] = api_base - os.environ['AZURE_API_KEY'] = api_base + os.environ['AZURE_API_KEY'] = api_key except Exception as e: pytest.fail(f"Error occurred: {e}")