test(test_completion.py): fix test case

This commit is contained in:
Krrish Dholakia 2023-11-09 17:43:07 -08:00
parent e355a0674f
commit 62e83ce75a

View file

@ -440,10 +440,12 @@ def test_completion_openai_litellm_key():
##### unset litellm var ##### unset litellm var
litellm.api_key = None litellm.api_key = None
except Timeout as e:
pass
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_completion_openai_litellm_key() test_completion_openai_litellm_key()
def test_completion_openrouter1(): def test_completion_openrouter1():
try: try: