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(agents): only consult the identity store for managed targets

Browse source 
is_agent_allowed entered the identity-store path whenever a prisma client
was configured, so an ordinary agent paired with an internal user returned
503 instead of 200. Classify the target from the registry first and fall
back to the store only when the registry has no entry, so an unmanaged
target never depends on the store being reachable.
This commit is contained in:
Joshua Valluru 2026-09-29 15:29:24 -07:00
parent cab5213bf3
commit 4c4675744c
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/agent_endpoints/auth/agent_permission_handler.py
View file

@ -161,13 +161,15 @@ class AgentRequestHandler:
from litellm.types.proxy.agent_identity import AgentIdentityFailure
registered: Final = global_agent_registry.get_agent_by_id(agent_id)
if prisma_client is not None or (registered is not None and registered.identity_managed):
registry_managed: Final = isinstance(registered, AgentResponse) and registered.identity_managed
if registry_managed or (registered is None and prisma_client is not None):
target: Final = await AgentIdentityStore.from_client(prisma_client).agent(agent_id)
if isinstance(target, AgentIdentityFailure):
raise_identity_failure(target)
if target is None and registered is not None and registered.identity_managed:
if registry_managed:
raise_identity_failure(target)
elif target is None and registry_managed:
return False
if target is not None and target.identity_managed:
elif isinstance(target, AgentResponse) and target.identity_managed:
if (
not target.enabled
or target.identity is None