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

fix(proxy): fall back to authenticated agent ids

Browse source
This commit is contained in:
AlisinaDevelo 2026-09-27 06:06:43 +02:00
parent 420b737995
commit 57a9211b1b
2 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -210,7 +210,9 @@ class _PROXY_MaxBudgetPerSessionHandler(CustomLogger):
def _make_cache_key(self, session_id: str, agent_id: str) -> str:
from litellm.proxy.agent_endpoints.agent_registry import global_agent_registry
scope: Final = json.dumps((global_agent_registry.stable_agent_id(agent_id), session_id), separators=(",", ":"))
stable_agent_id: Final = global_agent_registry.stable_agent_id(agent_id)
canonical_agent_id: Final = stable_agent_id if isinstance(stable_agent_id, str) else agent_id
scope: Final = json.dumps((canonical_agent_id, session_id), separators=(",", ":"))
return f"{{agent_session_budget:{scope}}}:spend"
async def _get_current_spend(self, cache_key: str) -> float:

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

@ -184,7 +184,9 @@ class _PROXY_MaxIterationsHandler(CustomLogger):
return f"{{session_iterations:{session_id}}}:count"
from litellm.proxy.agent_endpoints.agent_registry import global_agent_registry
scope: Final = json.dumps((global_agent_registry.stable_agent_id(agent_id), session_id), separators=(",", ":"))
stable_agent_id: Final = global_agent_registry.stable_agent_id(agent_id)
canonical_agent_id: Final = stable_agent_id if isinstance(stable_agent_id, str) else agent_id
scope: Final = json.dumps((canonical_agent_id, session_id), separators=(",", ":"))
return f"{{agent_session_iterations:{scope}}}:count"
async def _increment_and_get(self, cache_key: str) -> int: