(test) fix test test_exception_tracking

2023-10-30 20:05:35 -07:00 · 2023-10-30 20:05:35 -07:00 · ad2b75a8cc
commit ad2b75a8cc
parent 9c70da56df
1 changed files with 2 additions and 0 deletions
--- 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)