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: remove mutable agent identity registration constructions

Browse source
This commit is contained in:
Joshua Valluru 2026-09-30 15:05:40 -07:00
parent a168eaa98b
commit d6c1a9210b
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/proxy/agent_endpoints/agent_registry.py
View file

@ -32,6 +32,7 @@ from litellm.types.proxy.agent_identity import AgentIdentityFailure
if TYPE_CHECKING:
from prisma import models as prisma_models
from prisma.types import LiteLLM_RetiredAgentIdentityWhereUniqueInput
class AgentObjectPermissionRecord(Protocol):
@ -178,20 +179,19 @@ async def _managed_fields(
if history is None:
return result
entry: Final = history["create"]
prior: Final = await RetiredAgentIdentityRepository(client, use_writer=True).table.find_unique(
where={
"provider_tenant_id_client_id": {
"provider": entry["provider"],
"tenant_id": entry["tenant_id"],
"client_id": entry["client_id"],
}
where: Final[LiteLLM_RetiredAgentIdentityWhereUniqueInput] = {
"provider_tenant_id_client_id": {
"provider": entry["provider"],
"tenant_id": entry["tenant_id"],
"client_id": entry["client_id"],
}
)
}
prior: Final = await RetiredAgentIdentityRepository(client, use_writer=True).table.find_unique(where=where)
if prior is None:
return result
if existing is None or prior.agent_id != existing.agent_id:
raise HTTPException(409, "Entra application was already registered to another agent")
return {key: value for key, value in result.items() if key != "retired_identities"}
return MappingProxyType({key: value for key, value in result.items() if key != "retired_identities"})
def _dump_agent_params(raw: Mapping[str, object]) -> dict[str, object]: