phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

try again with azure test

Browse source
This commit is contained in:
ishaan-jaff 2023-07-31 18:01:06 -07:00
parent d08c38199d
commit cf1402cbe8
1 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

21
litellm/tests/test_completion.py
View file

@ -6,9 +6,19 @@ from main import completion
main.set_verbose = True 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"}] 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 1 #################
# test on openai completion call, with model and messages # test on openai completion call, with model and messages
try: try:
@ -79,14 +89,7 @@ except Exception as e:
raise 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 4 #################
# test on Claude Completion Call # test on Claude Completion Call