From 2e18cda2b80666b24352ee99c7ad7a8c72f9dfce Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 5 Aug 2023 13:37:10 -0700 Subject: [PATCH] good one --- litellm/tests/test_secrets.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_secrets.py b/litellm/tests/test_secrets.py index 474d6bde00..3123c4745f 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