scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 136

Merge pull request #21431 from BerriAI/fix/isolate-litellm-transport-state

Browse source 
fix(tests): restore disable_aiohttp_transport and force_ipv4 in isolate_litellm_state
This commit is contained in:
jquinter 2026-02-17 21:22:41 -03:00 • committed by GitHub
commit 62c7ebfd2c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/conftest.py
View file

@ -49,6 +49,12 @@ def isolate_litellm_state():
if hasattr(litellm, '_async_failure_callback'):
original_state['_async_failure_callback'] = litellm._async_failure_callback.copy() if litellm._async_failure_callback else []
# Store transport/network globals — many tests set these without restoring,
# causing subsequent tests to get None from _create_async_transport()
for _attr in ('disable_aiohttp_transport', 'force_ipv4'):
if hasattr(litellm, _attr):
original_state[_attr] = getattr(litellm, _attr)
# Flush cache before test (critical for respx mocks)
if hasattr(litellm, "in_memory_llm_clients_cache"):
litellm.in_memory_llm_clients_cache.flush_cache()