From ad2b75a8ccdd827be56e5d28aa35cab22be784d3 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 30 Oct 2023 20:05:35 -0700 Subject: [PATCH] (test) fix test test_exception_tracking --- litellm/tests/test_sentry.py | 2 ++ 1 file changed, 2 insertions(+) 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)