From cf1402cbe82eb847fa08bf44f3871a0a1b5acb02 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 31 Jul 2023 18:01:06 -0700 Subject: [PATCH] try again with azure test --- litellm/tests/test_completion.py | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 93da24aa8..ee64099ce 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -6,9 +6,19 @@ from main import completion main.set_verbose = True -user_message = "Hello, how are you?" +user_message = "Hello, whats the weather in San Francisco??" messages = [{ "content": user_message,"role": "user"}] + +################# Test 3 ################# +# test on Azure Openai Completion Call +try: + response = completion(model="chatgpt-test", messages=messages, azure=True) + print(response) +except Exception as e: + print(f"error occurred: {traceback.format_exc()}") + raise e + ################# Test 1 ################# # test on openai completion call, with model and messages try: @@ -79,14 +89,7 @@ except Exception as e: raise e -################# Test 3 ################# -# test on Azure Openai Completion Call -try: - response = completion(model="chatgpt-test", messages=messages, azure=True) - print(response) -except Exception as e: - print(f"error occurred: {traceback.format_exc()}") - raise e + ################# Test 4 ################# # test on Claude Completion Call