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

updates to tests

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-01 16:01:28 -07:00
parent ab35fdb635
commit 4d17c57373
3 changed files with 29 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_logging.py
View file

@ -213,13 +213,7 @@ try:
response = completion(model="claude-instant-1", messages=messages)
except AuthenticationError:
pass
# Restore stdout
sys.stdout = old_stdout
output = new_stdout.getvalue().strip()
print(output)
if "Logging Details Pre-API Call" not in output:
raise Exception("Required log message not found!")
elif "Logging Details Post-API Call" not in output:
@ -231,6 +225,11 @@ try:
score += 1
except Exception as e:
print(f"exception type: {type(e).__name__}")
# Restore stdout
sys.stdout = old_stdout
output = new_stdout.getvalue().strip()
print(output)
pytest.fail(f"Error occurred: {e}")