diff --git a/litellm/tests/test_secrets.py b/litellm/tests/test_secrets.py index 474d6bde0..3123c4745 100644 --- a/litellm/tests/test_secrets.py +++ b/litellm/tests/test_secrets.py @@ -17,6 +17,14 @@ litellm.secret_manager_client = InfisicalClient(token=infisical_token) user_message = "Hello, whats the weather in San Francisco??" messages = [{ "content": user_message,"role": "user"}] +def test_completion_azure(): + try: + response = completion(model="chatgpt-test", messages=messages, azure=True) + # Add any assertions here to check the response + print(response) + except Exception as e: + pytest.fail(f"Error occurred: {e}") + def test_completion_openai(): try: response = completion(model="gpt-3.5-turbo", messages=messages) @@ -31,4 +39,4 @@ def test_completion_openai_with_optional_params(): # Add any assertions here to check the response print(response) except Exception as e: - pytest.fail(f"Error occurred: {e}") + pytest.fail(f"Error occurred: {e}") \ No newline at end of file