phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_user_email_metrics

Browse source
This commit is contained in:
Ishaan Jaff 2025-02-25 10:47:09 -08:00
parent 1b2e3855cc
commit 55ae194912
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/otel_tests/test_prometheus.py
View file

@ -556,8 +556,9 @@ async def test_user_email_metrics():
""" """
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
# Create a user with user_email # Create a user with user_email
user_email = f"test-{uuid.uuid4()}@example.com"
user_data = { user_data = {
"user_email": "test@example.com", "user_email": user_email,
} }
user_info = await create_test_user(session, user_data) user_info = await create_test_user(session, user_data)
key = user_info["key"] key = user_info["key"]
@ -577,5 +578,5 @@ async def test_user_email_metrics():
metrics_after_first = await get_prometheus_metrics(session) metrics_after_first = await get_prometheus_metrics(session)
print("metrics_after_first request", metrics_after_first) print("metrics_after_first request", metrics_after_first)
assert ( assert (
"test@example.com" in metrics_after_first user_email in metrics_after_first
), "user_email should be tracked correctly" ), "user_email should be tracked correctly"