mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-10 16:22:17 +00:00
Add User ID validation to ensure it is not an email or phone number (#10102)
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
import json
|
||||
import time
|
||||
import re
|
||||
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union, cast
|
||||
|
||||
import httpx
|
||||
@@ -641,6 +642,7 @@ class AnthropicConfig(AnthropicModelInfo, BaseConfig):
|
||||
_litellm_metadata
|
||||
and isinstance(_litellm_metadata, dict)
|
||||
and "user_id" in _litellm_metadata
|
||||
and not _valid_user_id(_litellm_metadata.get("user_id", None))
|
||||
):
|
||||
optional_params["metadata"] = {"user_id": _litellm_metadata["user_id"]}
|
||||
|
||||
@@ -948,3 +950,19 @@ class AnthropicConfig(AnthropicModelInfo, BaseConfig):
|
||||
message=error_message,
|
||||
headers=cast(httpx.Headers, headers),
|
||||
)
|
||||
|
||||
|
||||
def _valid_user_id(user_id: str) -> bool:
|
||||
"""
|
||||
Validate that user_id is not an email or phone number.
|
||||
Returns: bool: True if valid (not email or phone), False otherwise
|
||||
"""
|
||||
email_pattern = r"^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$"
|
||||
phone_pattern = r"^\+?[\d\s\(\)-]{7,}$"
|
||||
|
||||
if re.match(email_pattern, user_id):
|
||||
return False
|
||||
if re.match(phone_pattern, user_id):
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
Reference in New Issue
Block a user