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

Update litellm/tests/test_client.py

Browse source
This commit is contained in:
sweep-ai[bot] 2023-08-01 16:03:05 +00:00 • committed by GitHub
parent 8cbc144ea8
commit 52b0cc0756
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_client.py
View file

@ -23,7 +23,7 @@ messages = [{ "content": user_message,"role": "user"}]
def test_completion_openai(): def test_completion_openai():
try: try:
response = completion(model="gpt-3.5-turbo", messages=messages, logger_fn=logger_fn) response = completion(model="updated-model", messages=messages, logger_fn=logger_fn)
# Add any assertions here to check the response # Add any assertions here to check the response
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
@ -37,7 +37,7 @@ def test_completion_non_openai():
def test_embedding_openai(): def test_embedding_openai():
try: try:
response = embedding(model='text-embedding-ada-002', input=[user_message], logger_fn=logger_fn) response = embedding(model='updated-model', input=[user_message], logger_fn=logger_fn)
# Add any assertions here to check the response # Add any assertions here to check the response
print(f"response: {str(response)[:50]}") print(f"response: {str(response)[:50]}")
except Exception as e: except Exception as e: