scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(proxy): avoid module-global mutation in callback regression test

Browse source
This commit is contained in:
wadrzl55333 2026-08-28 16:00:47 +08:00
parent 4f06fa8a50
commit 817b0a9837
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/proxy_unit_tests/test_proxy_config_unit_test.py
View file

@ -159,7 +159,7 @@ async def test_multiple_includes():
@pytest.mark.asyncio
async def test_config_file_success_callback_registers_custom_logger_instance():
async def test_config_file_success_callback_registers_custom_logger_instance(monkeypatch):
"""Config-file success_callback must register a custom-logger instance, not just the
string.
@ -176,8 +176,8 @@ async def test_config_file_success_callback_registers_custom_logger_instance():
LangfusePromptManagement,
)
litellm.success_callback = []
litellm._async_success_callback = []
monkeypatch.setattr(litellm, "success_callback", [])
monkeypatch.setattr(litellm, "_async_success_callback", [])
config_content = {"litellm_settings": {"success_callback": ["langfuse"]}}
with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as temp_file:
@ -197,8 +197,6 @@ async def test_config_file_success_callback_registers_custom_logger_instance():
)
assert num_langfuse_instances == 1
finally:
litellm.success_callback = []
litellm._async_success_callback = []
os.unlink(temp_file_path)