phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) reset callbacks in completion()

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 23:35:50 -08:00
parent 1d42967725
commit 01d5875426
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_completion.py
View file

@ -21,6 +21,13 @@ messages = [{"content": user_message, "role": "user"}]
def logger_fn(user_model_dict):
print(f"user_model_dict: {user_model_dict}")
@pytest.fixture(autouse=True)
def reset_callbacks():
print("\npytest fixture - resetting callbacks")
litellm.success_callback = []
litellm._async_success_callback = []
litellm.failure_callback = []
litellm.callbacks = []
def test_completion_custom_provider_model_name():
try: