mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-17 22:48:35 +00:00
Krish Dholakia
39de3610be
* fix(internal_user_endpoints.py): support user with `+` in email on user info ensures user is correctly parsed from input * fix(factory.py): support vertex function call args as None handles empty string in args for vertex gemini calls * docs(langfuse_integration.md): pin langfuse sdk version on docs * fix(vertex_ai/): return empty dict, instead of none when empty string given * refactor: reduce function size * fix: fix linting errors * fix: revert check * fix(internal_user_endpoints.py): fix check * test: update tests * test: update tests
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
# Test URL encoding handling for emails with + characters
|
|
import re
|
|
from urllib.parse import unquote
|
|
|
|
def test_user_id_parsing():
|
|
# Simulate the raw query string that would come from the URL
|
|
# When user calls: http://0.0.0.0:4000/user/info?user_id=machine-user+alp-air-admin-b58-b@tempus.com
|
|
# The query string would be: user_id=machine-user+alp-air-admin-b58-b@tempus.com
|
|
|
|
test_cases = [
|
|
"user_id=machine-user+alp-air-admin-b58-b@tempus.com",
|
|
"user_id=machine-user%2Balp-air-admin-b58-b@tempus.com", # URL encoded +
|
|
"user_id=regular@email.com",
|
|
"user_id=test-user@domain.com&other_param=value"
|
|
]
|
|
|
|
for query_string in test_cases:
|
|
print(f"\nTesting query string: {query_string}")
|
|
|
|
if 'user_id=' in query_string:
|
|
match = re.search(r'user_id=([^&]*)', query_string)
|
|
if match:
|
|
raw_user_id = unquote(match.group(1))
|
|
print(f"Extracted user_id: {raw_user_id}")
|
|
else:
|
|
print("No match found")
|
|
else:
|
|
print("user_id not found in query string")
|
|
|
|
if __name__ == "__main__":
|
|
test_user_id_parsing() |