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

fix: satisfy type-discipline gate

Browse source
This commit is contained in:
AlisinaDevelo 2026-09-27 09:09:26 +02:00
parent d21a5ec97f
commit b13ad9ce1f
2 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/hooks/max_budget_per_session_limiter.py
View file

@ -172,11 +172,11 @@ class _PROXY_MaxBudgetPerSessionHandler(CustomLogger):
if redis_cache is self._registered_redis_cache:
return
self.increment_script = cast(
self.increment_script = cast( # cast-ok: Redis registration returns the callable invoked below
Callable[..., Awaitable[object]],
redis_cache.async_register_script(MAX_BUDGET_SESSION_INCREMENT_SCRIPT),
)
self.get_agent_spend_script = cast(
self.get_agent_spend_script = cast( # cast-ok: Redis registration returns the callable invoked below
Callable[..., Awaitable[object]],
redis_cache.async_register_script(MAX_BUDGET_SESSION_GET_AGENT_SPEND_SCRIPT),
)
@ -314,7 +314,7 @@ class _PROXY_MaxBudgetPerSessionHandler(CustomLogger):
return f"{{session_budget:{session_id}}}:spend"
async def _get_local_scope(self, cache_key: str) -> dict[str, object] | None:
result: Final[object | None] = cast(
result: Final[object | None] = cast( # cast-ok: cache API returns Any; validate value before use
object | None,
await self.internal_usage_cache.async_get_cache(
key=cache_key,
@ -325,7 +325,7 @@ class _PROXY_MaxBudgetPerSessionHandler(CustomLogger):
if result is None:
return None
if isinstance(result, dict) and all(isinstance(key, str) for key in result):
return cast(dict[str, object], result)
return cast(dict[str, object], result) # cast-ok: keys are checked and values remain opaque
raise RuntimeError("Agent session scope cache has an invalid value")
@staticmethod
@ -376,7 +376,7 @@ class _PROXY_MaxBudgetPerSessionHandler(CustomLogger):
return max(float(legacy_value) - total_new_value, 0.0) + agent_value
async def _get_local_spend(self, cache_key: str) -> float | None:
result: Final[object | None] = cast(
result: Final[object | None] = cast( # cast-ok: cache API returns Any; validate value before use
object | None,
await self.internal_usage_cache.async_get_cache(
key=cache_key,

8
litellm/proxy/hooks/max_iterations_limiter.py
View file

@ -144,7 +144,7 @@ class _PROXY_MaxIterationsHandler(CustomLogger):
if redis_cache is self._registered_redis_cache:
return
self.increment_script = cast(
self.increment_script = cast( # cast-ok: Redis registration returns the callable invoked below
Callable[..., Awaitable[object]],
redis_cache.async_register_script(MAX_ITERATIONS_INCREMENT_SCRIPT),
)
@ -256,7 +256,7 @@ class _PROXY_MaxIterationsHandler(CustomLogger):
return f"agent:{stable_agent_id}"
async def _get_local_scope(self, cache_key: str) -> dict[str, object] | None:
result: Final[object | None] = cast(
result: Final[object | None] = cast( # cast-ok: cache API returns Any; validate value before use
object | None,
await self.internal_usage_cache.async_get_cache(
key=cache_key,
@ -267,11 +267,11 @@ class _PROXY_MaxIterationsHandler(CustomLogger):
if result is None:
return None
if isinstance(result, dict) and all(isinstance(key, str) for key in result):
return cast(dict[str, object], result)
return cast(dict[str, object], result) # cast-ok: keys are checked and values remain opaque
raise RuntimeError("Agent session scope cache has an invalid value")
async def _get_local_count(self, cache_key: str) -> int | None:
local_result: Final[object | None] = cast(
local_result: Final[object | None] = cast( # cast-ok: cache API returns Any; validate value below
object | None,
await self.internal_usage_cache.async_get_cache(
key=cache_key,