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

(test) fix test test_exception_tracking

Browse source
This commit is contained in:
ishaan-jaff 2023-10-30 20:05:35 -07:00
parent 9c70da56df
commit ad2b75a8cc
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_sentry.py
View file

@ -14,6 +14,7 @@ import time
def test_exception_tracking():
print('expect this to fail and log to sentry')
litellm.set_verbose=True
old_api_key = os.environ["OPENAI_API_KEY"]
os.environ["OPENAI_API_KEY"] = "ss"
try:
response = completion(model="gpt-3.5-turbo",
@ -25,6 +26,7 @@ def test_exception_tracking():
temperature=0.2
)
print(response)
os.environ["OPENAI_API_KEY"] = old_api_key
except Exception as e:
print("got_exception")
print(e)