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

Replaced the async client violation

Browse Source
This commit is contained in:
Sameer Kankute
2026-05-20 14:24:33 +05:30
parent 1ca48e931e
commit 5c67cf0e0c
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/integrations/rubrik.py
+4 -5
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 --------------------------------------------------------