scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

Replaced the async client violation

Browse source
This commit is contained in:
Sameer Kankute 2026-04-24 17:17:09 +05:30
parent c1d907f355
commit b05fcbdd76
No known key found for this signature in database
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/integrations/rubrik.py
View file

@ -112,10 +112,9 @@ class RubrikLogger(CustomGuardrail, CustomBatchLogger):
llm_provider=httpxSpecialProvider.LoggingCallback
)
# Dedicated client to avoid connection pooling issues with LiteLLM's shared client
self.tool_blocking_client = httpx.AsyncClient(
timeout=httpx.Timeout(5.0, connect=2.0),
limits=httpx.Limits(max_connections=10, max_keepalive_connections=5),
self.tool_blocking_client = get_async_httpx_client(
llm_provider=httpxSpecialProvider.LoggingCallback,
params={"timeout": httpx.Timeout(5.0, connect=2.0)},
)
self._headers: dict[str, str] = {"Content-Type": "application/json"}
@ -126,7 +125,7 @@ class RubrikLogger(CustomGuardrail, CustomBatchLogger):
async def aclose(self):
"""Close the dedicated tool blocking HTTP client."""
await self.tool_blocking_client.aclose()
await self.tool_blocking_client.close()
# -- Guardrail hook --------------------------------------------------------