scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-30 01:52:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

refactor(keys): freeze the service account route tuple and stamp metadata without seeding mutable literals

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
ryan 2026-09-18 09:34:44 +00:00
parent e68ea077f1
commit b50265d75f
2 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/_types.py
View file

@ -660,10 +660,10 @@ class LiteLLMRoutes(enum.Enum):
KeyManagementRoutes.AUTO_ROUTER_MANAGE.value,
]
team_service_account_key_routes = [
team_service_account_key_routes = (
KeyManagementRoutes.KEY_GENERATE.value,
KeyManagementRoutes.KEY_UPDATE.value,
]
)
management_routes = (
[

9
litellm/proxy/management_endpoints/key_management_endpoints.py
View file

@ -2240,13 +2240,12 @@ async def generate_service_account_key_fn(
)
if data.metadata is None or data.metadata.get("service_account_id") is None:
service_account_id: Final = (
(data.metadata or {}).get("service_account_id") or data.key_alias or str(uuid.uuid4())
)
data.metadata = { # rebind-ok: stamp the service_account_id onto the request so it persists on the key
**(data.metadata or {}),
service_account_id: Final = data.key_alias or str(uuid.uuid4())
stamped_metadata: Final = { # mutable-ok: GenerateKeyRequest.metadata is a plain dict field
**(data.metadata or MappingProxyType({})),
"service_account_id": service_account_id,
}
data.metadata = stamped_metadata # rebind-ok: the request carries the stamp so it persists on the key
verbose_proxy_logger.debug("entered /key/generate")