scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

style: apply black formatting to fix CI lint check

Browse source
This commit is contained in:
voidborne-d 2026-03-17 18:52:57 +00:00
parent ef22144854
commit ca8f5cffa0
2 changed files with 6 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/route_llm_request.py
View file

@ -185,7 +185,9 @@ async def add_shared_session_to_data(data: dict) -> None:
"SESSION REUSE: Shared aiohttp session is None after re-check, recreating..."
)
try:
new_session = await proxy_server._initialize_shared_aiohttp_session()
new_session = (
await proxy_server._initialize_shared_aiohttp_session()
)
except Exception:
verbose_proxy_logger.exception(
"SESSION REUSE: Exception during shared session recreation"

12
tests/test_litellm/proxy/test_aiohttp_session_recovery.py
View file

@ -22,9 +22,7 @@ async def test_add_shared_session_attaches_open_session():
mock_session = MagicMock()
mock_session.closed = False
with patch(
"litellm.proxy.proxy_server.shared_aiohttp_session", mock_session
):
with patch("litellm.proxy.proxy_server.shared_aiohttp_session", mock_session):
data = {}
await add_shared_session_to_data(data)
assert data["shared_session"] is mock_session
@ -129,9 +127,7 @@ async def test_add_shared_session_no_session_available():
"""When no session was ever created, data should not contain shared_session."""
from litellm.proxy.route_llm_request import add_shared_session_to_data
with patch(
"litellm.proxy.proxy_server.shared_aiohttp_session", None
):
with patch("litellm.proxy.proxy_server.shared_aiohttp_session", None):
data = {}
await add_shared_session_to_data(data)
assert "shared_session" not in data
@ -175,9 +171,7 @@ async def test_add_shared_session_concurrent_recreation_uses_lock():
):
# Launch 5 concurrent calls
results = [{} for _ in range(5)]
await asyncio.gather(
*(add_shared_session_to_data(d) for d in results)
)
await asyncio.gather(*(add_shared_session_to_data(d) for d in results))
# Only 1 coroutine should have called _initialize (the rest see the
# re-checked session as open under the lock)