phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix _enforce_unique_key_alias

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-27 16:22:04 -08:00
parent 74c025c50c
commit 3e50a06fb4
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -593,6 +593,7 @@ async def update_key_fn(
await _enforce_unique_key_alias(
key_alias=non_default_values.get("key_alias", None),
prisma_client=prisma_client,
existing_key_id=key,
)
response = await prisma_client.update_data(
@ -1898,6 +1899,7 @@ async def test_key_logging(
async def _enforce_unique_key_alias(
key_alias: Optional[str],
prisma_client: Any,
existing_key_id: Optional[str] = None,
) -> None:
"""
Helper to enforce unique key aliases across all keys.
@ -1905,13 +1907,19 @@ async def _enforce_unique_key_alias(
Args:
key_alias (Optional[str]): The key alias to check
prisma_client (Any): Prisma client instance
existing_key_id (Optional[str]): ID of existing key being updated, to exclude from uniqueness check
Raises:
HTTPException: If key alias already exists
HTTPException: If key alias already exists on a different key
"""
if key_alias is not None and prisma_client is not None:
where_clause: dict[str, Any] = {"key_alias": key_alias}
if existing_key_id:
# Exclude the current key from the uniqueness check
where_clause["NOT"] = {"token": existing_key_id}
existing_key = await prisma_client.db.litellm_verificationtoken.find_first(
where={"key_alias": key_alias}
where=where_clause
)
if existing_key is not None:
raise HTTPException(