tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-09 14:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
parent faa16ef29f d7b0a8c559
commit 62c7ebfd2c
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_litellm/conftest.py
+6
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()