diff --git a/litellm/tests/test_sentry.py b/litellm/tests/test_sentry.py index ca67258fca..b85336e019 100644 --- a/litellm/tests/test_sentry.py +++ b/litellm/tests/test_sentry.py @@ -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)