mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
Add User ID validation to ensure it is not an email or phone number
This commit is contained in:
parent
8be8022914
commit
57f98f2dc6
1 changed files with 18 additions and 0 deletions
|
@ -1,5 +1,6 @@
|
|||
import json
|
||||
import time
|
||||
import re
|
||||
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union, cast
|
||||
|
||||
import httpx
|
||||
|
@ -527,6 +528,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"]}
|
||||
|
||||
|
@ -784,3 +786,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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue