phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test completion

Browse source
This commit is contained in:
ishaan-jaff 2023-09-20 20:14:18 -07:00
parent c844d26ff1
commit 20e86e6a05
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_completion.py
View file

@ -147,6 +147,10 @@ def test_completion_with_litellm_call_id():
# ) # )
# # Add any assertions here to check the response # # Add any assertions here to check the response
# print(response) # print(response)
# except Exception as e:
# pytest.fail(f"Error occurred: {e}")
# hf_test_completion()
# this should throw an exception, to trigger https://logs.litellm.ai/ # this should throw an exception, to trigger https://logs.litellm.ai/
@ -368,7 +372,7 @@ def test_completion_azure():
try: try:
print("azure gpt-3.5 test\n\n") print("azure gpt-3.5 test\n\n")
response = completion( response = completion(
model="chatgpt-v-2", model="azure/chatgpt-v-2",
messages=messages, messages=messages,
) )
# Add any assertions here to check the response # Add any assertions here to check the response
@ -434,7 +438,7 @@ def test_completion_azure_with_litellm_key():
def test_completion_azure_deployment_id(): def test_completion_azure_deployment_id():
try: try:
response = completion( response = completion(
deployment_id="azure/chatgpt-v-2", deployment_id="chatgpt-v-2",
model="gpt-3.5-turbo", model="gpt-3.5-turbo",
messages=messages, messages=messages,
) )